KMPrintPresenter.swift 41 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999
  1. //
  2. // KMPrintPresenter.swift
  3. // PDF Master
  4. //
  5. // Created by lizhe on 2022/12/21.
  6. //
  7. import Cocoa
  8. import PDFKit
  9. //MARK: CPDFKit page 方法无法使用 暂时使用系统方法
  10. class KMPrintPresenter: NSObject {
  11. lazy var printData: KMPrintModel = KMPrintModel() {
  12. didSet {
  13. self.reloadData()
  14. }
  15. }
  16. var document: CPDFDocument?
  17. fileprivate weak var delegate: KMPrintPresenterDeleage?
  18. /**
  19. 初始化presenter 绑定数据
  20. */
  21. func initPresenter(delegate: KMPrintPresenterDeleage, data: KMPrintModel, document: CPDFDocument) {
  22. self.delegate = delegate
  23. self.document = document
  24. self.printData = data
  25. // DispatchQueue.main.async {
  26. // let pdfDocument = self.updatePrintDocument(documentURL: document.documentURL, data: self.printData)
  27. // self.printData.url = pdfDocument.documentURL
  28. // }
  29. }
  30. /**
  31. 刷新数据
  32. */
  33. func reloadData() {
  34. guard let document = document else { return }
  35. let pdfDocument = self.updatePrintDocument(documentURL: document.documentURL, data: self.printData)
  36. self.printData.url = pdfDocument.documentURL
  37. }
  38. /**
  39. @abstract 解除绑定
  40. */
  41. func free() {
  42. delegate = nil
  43. }
  44. }
  45. protocol KMPrintPresenterDeleage: NSObject {
  46. func showData(presenter: KMPrintPresenter, document: CPDFDocument)
  47. }
  48. protocol KMPrintPresenterDocument: NSObject {}
  49. extension KMPrintPresenter: KMPrintPresenterDocument {
  50. /**
  51. @abstract 获取打印document
  52. @param url 源文件url
  53. @param data 数据
  54. @retrun document
  55. */
  56. func updatePrintDocument(documentURL: URL, data: KMPrintModel) -> CPDFDocument {
  57. // 获取基本参数
  58. let printModel: KMPrintModel = data
  59. //获取总page
  60. let pages = self.fetchPages(documentURL, printModel.page)
  61. //绘制PDF
  62. let filePath = self.drawPages(nil, printModel, pages)
  63. let result = CPDFDocument(url: URL(fileURLWithPath: filePath))!
  64. if self.delegate != nil {
  65. self.delegate?.showData(presenter: self, document: result)
  66. }
  67. KMPrint("保存地址" + filePath)
  68. return result
  69. }
  70. /**
  71. @abstract 插入page
  72. @param paperSet 纸张设置
  73. @param pageSet page设置
  74. @param pages page数组
  75. */
  76. func drawPages(_ toFilePath: String?,
  77. _ printModel: KMPrintModel,
  78. _ pages: [KMPrintDrawPage]) -> String {
  79. /**
  80. 参数
  81. */
  82. //纸张大小
  83. let paperSize: CGSize = self.fetchPaperSize(printModel.paper)
  84. //总页数
  85. let paperCount: Int = self.fetchTotalPaperCount(paperSize, pages, printModel.page)
  86. //每页page数
  87. let pageOfPaperCount: Int = self.fetchPageOfPaper(printModel.page)
  88. //获取每张纸的page
  89. let drawPages: [[KMPrintDrawPage]] = self.fetchDrawPages(paperSize, printModel.page, paperCount, pageOfPaperCount, pages)
  90. //导出地址
  91. let filePath = KMPrintPresenter.fetchSaveFilePath(toFilePath)
  92. /**
  93. 绘制每张纸的内容
  94. */
  95. //创建画布
  96. let context: CGContext = self.createContext(filePath, paperSize)
  97. for drawPage in drawPages {
  98. context.beginPDFPage(nil)
  99. self.drawPageToContext(context, drawPage, printModel)
  100. context.endPDFPage()
  101. }
  102. context.closePDF()
  103. return filePath
  104. }
  105. /**
  106. 获取绘制的pages
  107. @pageModel page参数
  108. @param paperCount 纸张数量
  109. @param pageOfPaperCount 每张纸的page数量
  110. @param pages 所有page数量
  111. */
  112. func fetchDrawPages(_ paperSize: CGSize, _ pageModel: KMPrintPageModel,_ paperCount: Int, _ pageOfPaperCount: Int, _ pages: [KMPrintDrawPage]) -> [[KMPrintDrawPage]] {
  113. guard pages.count != 0 else {
  114. return []
  115. }
  116. //一个page重复获取次数
  117. var pageRepetitionCount = 1
  118. if (pageModel.operation.type == .poster) {
  119. if (pageModel.operation.poster.type == .tile) {
  120. pageRepetitionCount = Int(pageModel.operation.poster.tilePoint.x * pageModel.operation.poster.tilePoint.y)
  121. } else if (pageModel.operation.poster.type == .breakUp) {
  122. pageRepetitionCount = Int(pageModel.operation.poster.pageOfPaper.point.x * pageModel.operation.poster.pageOfPaper.point.y)
  123. }
  124. }
  125. var drawPages:[[KMPrintDrawPage]] = []
  126. for i in 0...(paperCount - 1) {
  127. //获取多页page
  128. var tempPags: [KMPrintDrawPage] = []
  129. for j in 0...(pageOfPaperCount - 1) {
  130. let pageIndex = i / pageRepetitionCount
  131. if (pageIndex * pageOfPaperCount + j < pages.count) {
  132. let originDrawPage = (pages[pageIndex * pageOfPaperCount + j])
  133. let drawPage = KMPrintDrawPage()
  134. drawPage.page = originDrawPage.page
  135. var pageCropRect = self.fetchPageCropRect(paperSize,i % pageRepetitionCount, pageModel, drawPage)
  136. var pageShowRect = pageCropRect
  137. if (pageModel.operation.type == .poster) {
  138. if (pageModel.operation.poster.type == .tile) {
  139. pageShowRect = self.fetchPageShowRect(paperSize, i % pageRepetitionCount, pageModel, drawPage)
  140. } else if (pageModel.operation.poster.type == .breakUp) {
  141. pageShowRect = pageCropRect
  142. }
  143. }
  144. drawPage.cropRect = pageCropRect
  145. drawPage.showRect = pageShowRect
  146. tempPags.append(drawPage)
  147. }
  148. }
  149. drawPages.append(tempPags)
  150. }
  151. return drawPages
  152. }
  153. /**
  154. 获取pages
  155. @param type 页面类型
  156. @param contentType annoation类型
  157. @param selectPages 当type 为custom时 传入选中page的下标
  158. */
  159. static func fetchSaveFilePath(_ filePath: String?) -> String {
  160. var saveFilePath = filePath ?? ""
  161. if saveFilePath.count == 0 {
  162. saveFilePath = NSTemporaryDirectory() + "/PDFMasterTest/test2.pdf"
  163. }
  164. if !FileManager.default.fileExists(atPath: NSTemporaryDirectory() + "/PDFMasterTest") {
  165. try?FileManager.default.createDirectory(atPath: NSTemporaryDirectory() + "/PDFMasterTest", withIntermediateDirectories: true)
  166. }
  167. if FileManager.default.fileExists(atPath: saveFilePath) {
  168. try?FileManager.default.removeItem(atPath: saveFilePath)
  169. }
  170. return saveFilePath
  171. }
  172. /**
  173. 获取pages
  174. @param type 页面类型
  175. @param contentType annoation类型
  176. @param selectPages 当type 为custom时 传入选中page的下标
  177. */
  178. static func creatDocument(_ url: URL) -> CPDFDocument {
  179. if FileManager.default.fileExists(atPath: NSTemporaryDirectory() + "/PDFMasterTest") {
  180. try?FileManager.default.createDirectory(atPath: NSTemporaryDirectory() + "/PDFMasterTest", withIntermediateDirectories: true)
  181. }
  182. let document = CPDFDocument(url: url)!
  183. // document.importPages(IndexSet(integer: 0), from: document, at: 0)
  184. let count = document.pageCount
  185. for _ in 0...(count - 1) {
  186. document.removePage(at: 0)
  187. }
  188. return document
  189. }
  190. /**
  191. 获取pages
  192. @param type 页面类型
  193. @param contentType annoation类型
  194. @param selectPages 当type 为custom时 传入选中page的下标
  195. */
  196. func fetchPages(_ documentURL: URL, _ pageModel: KMPrintPageModel) -> [KMPrintDrawPage] {
  197. let document = PDFDocument.init(url: documentURL)!
  198. var pageIndexs: [Int] = []
  199. let range = pageModel.range
  200. let contentType = pageModel.contentType
  201. let reversePrintOrder = range.reversePrintOrder
  202. switch range.type {
  203. case .allPage:
  204. for index in 0...document.pageCount - 1 {
  205. pageIndexs.append(index)
  206. }
  207. case .evenPage:
  208. for index in 0...document.pageCount - 1 {
  209. if index % 2 == 0 {
  210. pageIndexs.append(index)
  211. }
  212. }
  213. case .oddPage:
  214. for index in 0...document.pageCount - 1 {
  215. if index % 2 != 0 {
  216. pageIndexs.append(index)
  217. }
  218. }
  219. case .currentPage:
  220. pageIndexs.append(0)
  221. case .custom:
  222. pageIndexs.append(0)
  223. default:
  224. pageIndexs.append(0)
  225. }
  226. var pagesArray: [KMPrintDrawPage] = []
  227. for index in pageIndexs {
  228. let page = document.page(at: index)!
  229. let drawPage = KMPrintDrawPage()
  230. drawPage.page = page
  231. self.dealPageContent(contentType, [drawPage])
  232. if reversePrintOrder {
  233. pagesArray.insert(drawPage, at: 0)
  234. } else {
  235. pagesArray.append(drawPage)
  236. }
  237. }
  238. return pagesArray
  239. }
  240. /**
  241. 处理page annoation 内容
  242. @param contentType annoation类型
  243. @param pages page
  244. */
  245. func dealPageContent (_ contentType: KMPrintContentType, _ pages: [KMPrintDrawPage]) -> Void {
  246. for page in pages {
  247. let annoations: [PDFAnnotation] = page.page.annotations
  248. //内容处理
  249. switch contentType {
  250. case .document:
  251. for annoation in annoations {
  252. annoation.page!.removeAnnotation(annoation)
  253. }
  254. case .documentAndStamp:
  255. for annoation in annoations {
  256. if !self.isAnnoationStamp(type: annoation.type!) {
  257. annoation.page!.removeAnnotation(annoation)
  258. }
  259. }
  260. case .documentAndMarkup:
  261. for annoation in annoations {
  262. if !self.isAnnoationMarkup(type: annoation.type!) {
  263. annoation.page!.removeAnnotation(annoation)
  264. }
  265. }
  266. case .documentAndForm:
  267. for annoation in annoations {
  268. if !self.isAnnoationForm(type: annoation.type!) {
  269. annoation.page!.removeAnnotation(annoation)
  270. }
  271. }
  272. default:
  273. KMPrint("未找到")
  274. break
  275. }
  276. }
  277. }
  278. /**
  279. @abstract 获取context
  280. @param size纸张大小
  281. */
  282. func createContext(_ saveFilePath: String, _ size: CGSize) -> CGContext {
  283. var mediaBox: CGRect = NSMakeRect(0, 0, size.width, size.height)
  284. let url = CFURLCreateWithFileSystemPath(nil, saveFilePath as CFString, .cfurlposixPathStyle, false)
  285. let content: CGContext = CGContext.init(url!, mediaBox: &mediaBox, nil)!
  286. return content
  287. }
  288. /**
  289. @abstract 绘制page
  290. @param context
  291. @pages page数组 [CPDFPage]
  292. */
  293. func drawPageToContext(_ context: CGContext, _ pages: [KMPrintDrawPage], _ data: KMPrintModel, _ drawPageRect: CGRect = NSZeroRect) {
  294. //左下角有坐标系原点
  295. /**
  296. paper
  297. */
  298. let paperSize: CGSize = self.fetchPaperSize(data.paper)//纸张大小
  299. let paperItemSize: CGSize = self.fetchPaperItemSize(data.paper) //页面paper大小(去除边框)
  300. let paperInset: NSEdgeInsets = data.paper.info.inset //绘制paper大小
  301. let border: Bool = true //是否存在边框
  302. /**
  303. page
  304. */
  305. let pageOrder: KMPrintPageOperation.Multipage.Order = .horizontal //页面顺序
  306. let pageSize: CGSize = self.fetchPageItemSize(data.page, paperItemSize) //page大小
  307. let showModel: KMPrintPageOperation.Size = self.fetchShowModel(data.page)
  308. let autoRotate = self.fetchAutoRotate(data.page)
  309. let autoSize: Bool = self.fetchAutoSize(data.page)
  310. //行列
  311. let columnAndRow = self.fetchPageColumnAndRow(data.page) //行 列数量
  312. let columnAndRowSpace = CGPoint(x: 2, y: 2) //行 列之间的空间
  313. for i in 0...Int(columnAndRow.x) - 1 {
  314. for j in 0...(Int(columnAndRow.y) - 1) {
  315. let index = j + i * Int(columnAndRow.x)
  316. if index < pages.count {
  317. //参数
  318. let page: KMPrintDrawPage = pages[index]
  319. let rect = page.showRect
  320. //裁剪当前Page
  321. page.page.setBounds(page.cropRect, for: .cropBox)
  322. let pageItemSize = rect.size
  323. let rotate = page.page.rotation
  324. var scale = self.fetchPageScale(page, pageSize, autoRotate, autoSize)
  325. if data.page.operation.type == .size {
  326. if showModel.model == .custom {
  327. scale *= showModel.scale
  328. } else if showModel.model == .full {
  329. scale = 1
  330. }
  331. } else if (data.page.operation.type == .poster) {
  332. if (data.page.operation.poster.type == .tile) {
  333. scale = data.page.operation.poster.scale
  334. }
  335. }
  336. //当前item的自身中心点
  337. let center = CGPoint(x: (pageSize.width - pageItemSize.width * scale) / 2.0 ,
  338. y: (pageSize.height - pageItemSize.height * scale) / 2.0)
  339. var origin = rect.origin
  340. //多页Page自动旋转
  341. if autoSize {
  342. switch pageOrder {
  343. case .horizontal:
  344. origin.x = (pageSize.width + columnAndRowSpace.x) * CGFloat(j) + paperInset.left + center.x
  345. //页面内容高度 + 下边的行间距 - 第几个cell的高度 +居中
  346. origin.y = paperSize.height - (pageSize.height + columnAndRowSpace.y) * (CGFloat(i) + 1) + center.y + columnAndRowSpace.y
  347. case .horizontalReverseSequence:
  348. origin.x = paperSize.width - (pageSize.width + columnAndRowSpace.x) * (CGFloat(j) + 1) + center.x + columnAndRowSpace.x
  349. origin.y = paperSize.height - (pageSize.height + columnAndRowSpace.y) * (CGFloat(i) + 1) + center.y + paperInset.bottom + columnAndRowSpace.y
  350. case .vertical:
  351. origin.x = (pageSize.width + columnAndRowSpace.x) * CGFloat(i) + paperInset.left + center.x
  352. origin.y = paperSize.height - (pageSize.height + columnAndRowSpace.y) * (CGFloat(j) + 1) + center.y + paperInset.bottom + columnAndRowSpace.y
  353. case .verticalReverseSequence:
  354. origin.x = paperSize.width - (pageSize.width + columnAndRowSpace.x) * (CGFloat(i) + 1) + center.x + columnAndRowSpace.x
  355. origin.y = paperSize.height - (pageSize.height + columnAndRowSpace.y) * (CGFloat(j) + 1) + center.y + paperInset.bottom + columnAndRowSpace.y
  356. default:
  357. KMPrint("未找到")
  358. break
  359. }
  360. }
  361. NSGraphicsContext.saveGraphicsState()
  362. //平移
  363. context.translateBy(x: origin.x, y: origin.y)
  364. //缩放
  365. context.scaleBy(x: CGFloat(scale), y: CGFloat(scale))
  366. page.page.draw(with: PDFDisplayBox.cropBox, to: context)
  367. // page.transform(context, for: PDFDisplayBox.cropBox)
  368. if border {
  369. var dirtyRect = rect
  370. if rotate == 90 ||
  371. rotate == 270 {
  372. dirtyRect = NSMakeRect(dirtyRect.origin.x, dirtyRect.origin.y, dirtyRect.size.height, dirtyRect.size.width)
  373. }
  374. context.addRect(dirtyRect)
  375. context.setStrokeColor(red: 1.0, green: 1.0, blue: 1.0, alpha: 1.0)
  376. context.strokePath()
  377. }
  378. NSGraphicsContext.restoreGraphicsState()
  379. // page.setBounds(NSRect(x: 0, y: 0, width: pageRect.size.width, height: pageRect.size.height), for: .cropBox)
  380. }
  381. }
  382. }
  383. }
  384. }
  385. protocol KMPrintPresenterPage {}
  386. extension KMPrintPresenter: KMPrintPresenterPage {
  387. /**
  388. 获取pages
  389. @param type 页面类型
  390. @param contentType annoation类型
  391. @param selectPages 当type 为custom时 传入选中page的下标
  392. */
  393. func fetchPageCropRect(_ paperSize: CGSize, _ index: Int, _ pageModel: KMPrintPageModel, _ page: KMPrintDrawPage) -> CGRect {
  394. var newRect = page.page.bounds(for: .cropBox)
  395. if (pageModel.operation.type == .poster) {
  396. let originSize = newRect.size
  397. var cropPoint = CGPoint(x: 1, y: 1)
  398. var scale = 1.0
  399. if (pageModel.operation.poster.type == .tile) {
  400. cropPoint = pageModel.operation.poster.tilePoint
  401. // scale = pageModel.operation.poster.scale
  402. } else if (pageModel.operation.poster.type == .breakUp) {
  403. cropPoint = pageModel.operation.poster.pageOfPaper.point
  404. }
  405. let width = originSize.width / cropPoint.x
  406. let height = originSize.height / cropPoint.y
  407. let column: Int = Int(cropPoint.x) //行
  408. let row: Int = Int(cropPoint.y) //列
  409. for i in 0...column - 1 {
  410. for j in 0...row - 1 {
  411. if i + j + i * (row - 1) == index {
  412. newRect.origin.x = CGFloat(i) * width * scale
  413. newRect.origin.y = ((originSize.height - CGFloat(j) * height - height)) * scale
  414. newRect.size.width = width * scale
  415. newRect.size.height = height * scale
  416. return newRect
  417. }
  418. }
  419. }
  420. }
  421. return newRect
  422. }
  423. func fetchPageShowRect(_ paperSize: CGSize, _ index: Int, _ pageModel: KMPrintPageModel, _ page: KMPrintDrawPage) -> CGRect {
  424. var newRect = NSZeroRect
  425. var pageRect = CGRect(x: 0, y: 0, width: paperSize.width, height: paperSize.height)
  426. if (pageModel.operation.type == .poster) {
  427. let pageSize = page.page.bounds(for: .cropBox)
  428. var cropPoint = CGPoint(x: 1, y: 1)
  429. var scale = 1.0
  430. if (pageModel.operation.poster.type == .tile) {
  431. cropPoint = pageModel.operation.poster.tilePoint
  432. scale = pageModel.operation.poster.scale
  433. } else if (pageModel.operation.poster.type == .breakUp) {
  434. cropPoint = pageModel.operation.poster.pageOfPaper.point
  435. }
  436. if (cropPoint.x == 1 && cropPoint.y == 1) {
  437. } else {
  438. let originPaperSize = CGSize(width: paperSize.width * cropPoint.x, height: paperSize.height * cropPoint.y)
  439. let pageWidth = pageSize.width * scale
  440. let pageHeight = pageSize.height * scale
  441. let pageOrigin = CGPoint(x: (originPaperSize.width - pageWidth) / 2, y: (originPaperSize.height - pageHeight) / 2)
  442. let width = originPaperSize.width / cropPoint.x
  443. let height = originPaperSize.height / cropPoint.y
  444. let column: Int = Int(cropPoint.x) //行
  445. let row: Int = Int(cropPoint.y) //列
  446. for i in 0...column - 1 {
  447. for j in 0...row - 1 {
  448. if i + j + i * (row - 1) == index {
  449. newRect.origin.x = CGFloat(i) * width
  450. newRect.origin.y = ((originPaperSize.height - CGFloat(j) * height - height))
  451. newRect.size.width = width
  452. newRect.size.height = height
  453. if (pageOrigin.x > newRect.origin.x) {
  454. pageRect.origin.x = pageOrigin.x
  455. pageRect.size.width = newRect.size.width - pageOrigin.x
  456. } else if (originPaperSize.width - pageOrigin.x > newRect.origin.x) {
  457. pageRect.origin.x = 0
  458. pageRect.size.width = newRect.size.width - pageOrigin.x
  459. } else {
  460. pageRect.origin.x = 0
  461. pageRect.size.width = newRect.size.width
  462. }
  463. if (originPaperSize.height - pageOrigin.y < newRect.origin.y + newRect.size.height) {
  464. pageRect.origin.y = 0
  465. pageRect.size.height = newRect.size.height - pageOrigin.y
  466. } else if (pageOrigin.y > newRect.origin.y) {
  467. pageRect.origin.y = pageOrigin.y
  468. pageRect.size.height = newRect.size.height - pageOrigin.y
  469. } else {
  470. pageRect.origin.y = newRect.origin.y - pageOrigin.y
  471. pageRect.size.height = newRect.size.height
  472. }
  473. return pageRect
  474. }
  475. }
  476. }
  477. }
  478. }
  479. return pageRect
  480. }
  481. func fetchPageColumnAndRow(_ pageModel: KMPrintPageModel) -> CGPoint {
  482. var point = NSZeroPoint
  483. let pageOrder: KMPrintPageOperation.Multipage.Order = pageModel.operation.multipage.orderType //页面顺序
  484. switch pageModel.operation.type {
  485. case .multipage:
  486. point.x = pageModel.operation.multipage.pageOfPaper.point.x
  487. point.y = pageModel.operation.multipage.pageOfPaper.point.y
  488. default:
  489. point.x = 1
  490. point.y = 1
  491. }
  492. //如果是横向参数需切换
  493. if pageOrder == .horizontal ||
  494. pageOrder == .horizontalReverseSequence {
  495. let temp = point.x
  496. point.x = point.y
  497. point.y = temp
  498. }
  499. return point
  500. }
  501. func fetchAutoRotate(_ pageModel: KMPrintPageModel) -> Bool {
  502. var autoRotate = false
  503. switch pageModel.operation.type {
  504. case .multipage:
  505. autoRotate = pageModel.operation.multipage.isAutoRotate
  506. case .pamphlet:
  507. autoRotate = pageModel.operation.pamphlet.isAutoRotate
  508. default:
  509. autoRotate = false
  510. }
  511. return autoRotate
  512. }
  513. func fetchAutoSize(_ pageModel: KMPrintPageModel) -> Bool {
  514. var autoSize = false
  515. switch pageModel.operation.type {
  516. case .size:
  517. autoSize = true
  518. default:
  519. autoSize = false
  520. }
  521. return autoSize
  522. }
  523. func fetchShowModel(_ pageModel: KMPrintPageModel) -> KMPrintPageOperation.Size {
  524. var model = KMPrintPageOperation.Size()
  525. switch pageModel.operation.type {
  526. case .size:
  527. model = pageModel.operation.size
  528. default:
  529. model = KMPrintPageOperation.Size()
  530. }
  531. return model
  532. }
  533. func fetchPageItemSize(_ pageModel: KMPrintPageModel, _ paperSize: CGSize) -> CGSize {
  534. var size = NSZeroSize
  535. let columnAndRow = self.fetchPageColumnAndRow(pageModel) //行 列数量
  536. let columnAndRowSpace = CGPoint(x: 2, y: 2) //行 列之间的空间
  537. //page大小
  538. size = CGSize(width: (paperSize.width - CGFloat((columnAndRow.x - 1)) * CGFloat(columnAndRowSpace.x)) / columnAndRow.x,
  539. height: (paperSize.height - CGFloat((columnAndRow.y - 1)) * CGFloat(columnAndRowSpace.y)) / columnAndRow.y)
  540. return size
  541. }
  542. func fetchPageScale(_ page: KMPrintDrawPage, _ pageItemSize: CGSize, _ autoRotate: Bool, _ autoSize: Bool) -> CGFloat {
  543. var scale = 1.0
  544. let originSize = page.page.bounds(for: .cropBox)
  545. var rotate = page.page.rotation
  546. //取出page横竖时能显示的最大Rect
  547. if autoRotate {
  548. //page旋转度数为0度或者180度时,在指定的大小内,能显示的比例
  549. let scale1 = min(pageItemSize.width / originSize.width, pageItemSize.height / originSize.height)
  550. //page旋转度数为90度或者270度时,在指定的大小内,能显示的比例
  551. let scale2 = min(pageItemSize.width / originSize.height, pageItemSize.height / originSize.width)
  552. scale = max(scale1, scale2)
  553. if scale1 > scale2 {
  554. //高为竖直排列时,显示最大,则需将page时90度或者270度需要进行旋转
  555. if rotate == 90 || rotate == 270 {
  556. rotate = rotate - 90
  557. }
  558. } else {
  559. //宽为竖直排列时,显示最大,则需将page时0度或者180度需要进行旋转
  560. if rotate == 0 || rotate == 180 {
  561. rotate = rotate - 90
  562. }
  563. }
  564. } else {
  565. scale = min(pageItemSize.width / originSize.width, pageItemSize.height / originSize.height)
  566. }
  567. if (autoSize) {
  568. } else {
  569. scale = min(scale, 1)
  570. }
  571. return scale
  572. }
  573. func fetchPageRotate(_ page: KMPrintDrawPage, _ pageItemSize: CGSize, _ autoRotate: Bool) -> CGFloat {
  574. var scale = 1.0
  575. let originSize = page.page.bounds(for: .cropBox)
  576. var rotate = page.page.rotation
  577. //取出page横竖时能显示的最大Rect
  578. if autoRotate {
  579. //page旋转度数为0度或者180度时,在指定的大小内,能显示的比例
  580. let scale1 = min(pageItemSize.width / originSize.width, pageItemSize.height / originSize.height)
  581. //page旋转度数为90度或者270度时,在指定的大小内,能显示的比例
  582. let scale2 = min(pageItemSize.width / originSize.height, pageItemSize.height / originSize.width)
  583. scale = max(scale1, scale2)
  584. if scale1 > scale2 {
  585. //高为竖直排列时,显示最大,则需将page时90度或者270度需要进行旋转
  586. if rotate == 90 || rotate == 270 {
  587. rotate = rotate - 90
  588. }
  589. } else {
  590. //宽为竖直排列时,显示最大,则需将page时0度或者180度需要进行旋转
  591. if rotate == 0 || rotate == 180 {
  592. rotate = rotate - 90
  593. }
  594. }
  595. } else {
  596. scale = min(pageItemSize.width / originSize.width, pageItemSize.height / originSize.height)
  597. }
  598. return scale
  599. }
  600. func fetchPageLite(pageModel: KMPrintPageModel) {
  601. }
  602. }
  603. protocol KMPrintPresenterPaper {}
  604. extension KMPrintPresenter: KMPrintPresenterPaper {
  605. /**
  606. 获取每张纸的page
  607. @param type 页面类型
  608. @param contentType annoation类型
  609. @param selectPages 当type 为custom时 传入选中page的下标
  610. */
  611. func fetchPageOfPaper(_ pageModel: KMPrintPageModel) -> Int {
  612. var count = 1
  613. switch pageModel.operation.type {
  614. case .multipage:
  615. count = Int(pageModel.operation.multipage.pageOfPaper.point.x * pageModel.operation.multipage.pageOfPaper.point.y)
  616. case .poster:
  617. count = Int(pageModel.operation.multipage.pageOfPaper.point.x * pageModel.operation.multipage.pageOfPaper.point.y)
  618. default:
  619. count = 1
  620. }
  621. return count
  622. }
  623. /**
  624. 获取总纸张数
  625. @param pages page总数
  626. @param contentType annoation类型
  627. @param selectPages 当type 为custom时 传入选中page的下标
  628. */
  629. func fetchTotalPaperCount (_ paperSize: CGSize, _ pages: [KMPrintDrawPage], _ pageModel: KMPrintPageModel) -> Int {
  630. var count = 1
  631. let pageOfPaper = self.fetchPageOfPaper(pageModel)
  632. switch pageModel.operation.type {
  633. case .multipage:
  634. count = Int(ceilf(Float(pages.count / pageOfPaper)))
  635. case .poster:
  636. if (pageModel.operation.poster.type == .tile) {
  637. //1 2 4 9 16
  638. let scale = pageModel.operation.poster.scale
  639. let pageSize = pages.first?.page.bounds(for: .cropBox).size ?? paperSize
  640. let point = self.fetchPosterPageCount(paperSize: paperSize, pageSize: pageSize, scale: scale)
  641. pageModel.operation.poster.tilePoint = point
  642. count = Int((point.x * point.y)) * pages.count
  643. } else if (pageModel.operation.poster.type == .breakUp) {
  644. count = Int((pageModel.operation.poster.pageOfPaper.point.x * pageModel.operation.poster.pageOfPaper.point.y)) * pages.count
  645. } else {
  646. count = Int(ceilf(Float(pages.count / pageOfPaper)))
  647. }
  648. default:
  649. count = Int(ceilf(Float(pages.count / pageOfPaper)))
  650. }
  651. return count
  652. }
  653. func fetchPosterPageCount(paperSize: CGSize, pageSize: CGSize, scale: CGFloat) -> CGPoint {
  654. var xCount: Int = 1
  655. var yCount: Int = 1
  656. var contain: Bool = true
  657. while (contain) {
  658. if (pageSize.width * scale < CGFloat(xCount) * paperSize.width &&
  659. pageSize.height * scale < CGFloat(yCount) * paperSize.height) {
  660. contain = false
  661. break
  662. }
  663. //增加行数 和 列数
  664. if xCount == yCount {
  665. xCount += 1
  666. } else {
  667. yCount += 1
  668. }
  669. }
  670. return CGPoint(x: xCount, y: yCount)
  671. }
  672. /**
  673. 获取pages
  674. @param type 页面类型
  675. @param contentType annoation类型
  676. @param selectPages 当type 为custom时 传入选中page的下标
  677. */
  678. func fetchPaperSize(_ paperModel: KMPrintPaperModel) -> CGSize {
  679. var paperSize = paperModel.info.size
  680. let direction = paperModel.direction
  681. if direction == .vertical {
  682. paperSize = CGSize(width: paperModel.info.size.width, height: paperModel.info.size.height)
  683. } else if direction == .horizontal {
  684. paperSize = CGSize(width: paperModel.info.size.height, height: paperModel.info.size.width)
  685. }
  686. return paperSize
  687. }
  688. func fetchPaperItemSize(_ paperModel: KMPrintPaperModel) -> CGSize {
  689. var paperSize = self.fetchPaperSize(paperModel)
  690. let paperInset = paperModel.info.inset
  691. paperSize = CGSize(width: paperSize.width - paperInset.left - paperInset.right,
  692. height: paperSize.height - paperInset.bottom - paperInset.top)
  693. return paperSize
  694. }
  695. }
  696. protocol KMPrintPresenterDraw {}
  697. extension KMPrintPresenter: KMPrintPresenterDraw {
  698. // -(void)drawCutMarkContext:(CGContextRef)context contextSize:(CGSize)size
  699. // {
  700. // CGContextSetStrokeColorWithColor(context, [NSColor blackColor].CGColor);
  701. // CGContextSetLineWidth(context, 1.0);
  702. // CGContextSetLineCap(context, kCGLineCapSquare);
  703. //
  704. // if(self.columnIndex == 1 && self.lineIndex == 1) {
  705. // [self drawLeftBottomCutMarks:context size:size];
  706. // [self drawRightTopCutMarks:context size:size];
  707. // [self drawRightBottomCutMarks:context size:size];
  708. // } else if (self.lineIndex == 1 && self.columnIndex == self.vertArray.count) {
  709. // [self drawLeftTopCutMarks:context size:size];
  710. // [self drawRightTopCutMarks:context size:size];
  711. // [self drawRightBottomCutMarks:context size:size];
  712. // } else if (self.columnIndex == 1 && self.lineIndex == self.hourArray.count) {
  713. // [self drawLeftTopCutMarks:context size:size];
  714. // [self drawLeftBottomCutMarks:context size:size];
  715. // [self drawRightBottomCutMarks:context size:size];
  716. // } else if (self.columnIndex == self.vertArray.count && self.hourArray.count == self.lineIndex) {
  717. // [self drawLeftTopCutMarks:context size:size];
  718. // [self drawLeftBottomCutMarks:context size:size];
  719. // [self drawRightTopCutMarks:context size:size];
  720. // } else {
  721. // [self drawLeftTopCutMarks:context size:size];
  722. // [self drawLeftBottomCutMarks:context size:size];
  723. // [self drawRightTopCutMarks:context size:size];
  724. // [self drawRightBottomCutMarks:context size:size];
  725. // }
  726. // //绘制完成
  727. // CGContextStrokePath(context);
  728. // }
  729. //
  730. // //左上角切割标记
  731. // -(void)drawLeftTopCutMarks:(CGContextRef)context size:(CGSize)size
  732. // {
  733. // CGFloat KBlankA4H =size.height;
  734. // CGPoint point_LeftTop = CGPointZero;
  735. //
  736. // if (self.PDFPrint.splitType == kKMPDFPosterSplitType_PageNumber) {
  737. // point_LeftTop.x = self.PDFPrint.fullPageEdgeInsets.left;
  738. // } else {
  739. // point_LeftTop.x = self.PDFPrint.edgeInsets.left;
  740. // }
  741. //
  742. // point_LeftTop.y = KBlankA4H - self.PDFPrint.edgeInsets.top;
  743. //
  744. // CGContextMoveToPoint(context, 10,point_LeftTop.y);
  745. // CGContextAddLineToPoint(context,point_LeftTop.x, point_LeftTop.y);
  746. //
  747. // CGContextMoveToPoint(context, point_LeftTop.x - 10,KBlankA4H -10);
  748. // CGContextAddLineToPoint(context,point_LeftTop.x - 10,point_LeftTop.y);
  749. // }
  750. //
  751. // //右上角切割标记
  752. // -(void)drawRightTopCutMarks:(CGContextRef)context size:(CGSize)size
  753. // {
  754. // CGFloat KBlankA4W =size.width;
  755. // CGFloat KBlankA4H =size.height;
  756. //
  757. // CGPoint point_RightTop = CGPointZero;//右上角
  758. //
  759. // if (self.PDFPrint.splitType == kKMPDFPosterSplitType_PageNumber) {
  760. // point_RightTop.x = KBlankA4W - self.PDFPrint.fullPageEdgeInsets.right;
  761. // point_RightTop.y = KBlankA4H - self.PDFPrint.fullPageEdgeInsets.top;
  762. // } else {
  763. // point_RightTop.x = KBlankA4W - self.PDFPrint.edgeInsets.right;
  764. // point_RightTop.y = KBlankA4H - self.PDFPrint.edgeInsets.top;
  765. // }
  766. //
  767. // CGContextMoveToPoint(context,point_RightTop.x,point_RightTop.y);
  768. // CGContextAddLineToPoint(context,KBlankA4W - 10,point_RightTop.y);
  769. //
  770. // CGContextMoveToPoint(context,point_RightTop.x + 10,KBlankA4H - 10);
  771. // CGContextAddLineToPoint(context,point_RightTop.x + 10,point_RightTop.y);
  772. // }
  773. //
  774. // //左下角切割标记
  775. // -(void)drawLeftBottomCutMarks:(CGContextRef)context size:(CGSize)size
  776. // {
  777. // CGPoint point_LeftBottom = CGPointZero;//左下角
  778. // if (self.PDFPrint.splitType == kKMPDFPosterSplitType_PageNumber) {
  779. // point_LeftBottom.x = self.PDFPrint.fullPageEdgeInsets.left;
  780. // point_LeftBottom.y = self.PDFPrint.fullPageEdgeInsets.bottom;
  781. // } else {
  782. // point_LeftBottom.x = self.PDFPrint.edgeInsets.left;
  783. // point_LeftBottom.y = self.PDFPrint.edgeInsets.bottom;
  784. // }
  785. //
  786. // //左下角
  787. // CGContextMoveToPoint(context, 10,point_LeftBottom.y);
  788. // CGContextAddLineToPoint(context,point_LeftBottom.x, point_LeftBottom.y);
  789. //
  790. // CGContextMoveToPoint(context, point_LeftBottom.x- 10,10);
  791. // CGContextAddLineToPoint(context,point_LeftBottom.x - 10,point_LeftBottom.y);
  792. // }
  793. //
  794. // //右下角切割标记
  795. // -(void)drawRightBottomCutMarks:(CGContextRef)context size:(CGSize)size
  796. // {
  797. // CGFloat KBlankA4W =size.width;
  798. //
  799. // CGPoint point_RightBottom = CGPointZero;//右下角
  800. // if (self.PDFPrint.splitType == kKMPDFPosterSplitType_PageNumber) {
  801. // point_RightBottom.x = KBlankA4W - self.PDFPrint.fullPageEdgeInsets.right;
  802. // point_RightBottom.y = self.PDFPrint.fullPageEdgeInsets.bottom;
  803. // } else {
  804. // point_RightBottom.x = KBlankA4W - self.PDFPrint.edgeInsets.right;
  805. // point_RightBottom.y = self.PDFPrint.edgeInsets.bottom;
  806. // }
  807. //
  808. // CGContextMoveToPoint(context,KBlankA4W - 10,point_RightBottom.y);
  809. // CGContextAddLineToPoint(context,point_RightBottom.x,point_RightBottom.y);
  810. //
  811. // CGContextMoveToPoint(context,point_RightBottom.x+ 10,10);
  812. // CGContextAddLineToPoint(context,point_RightBottom.x+ 10,point_RightBottom.y);
  813. // }
  814. func drawString(_ pageModel: KMPrintPageModel, _ contextSize: CGSize) {
  815. var string = pageModel.operation.poster.tags.first ?? ""
  816. // if string.isEmpty {
  817. // string =
  818. // }
  819. }
  820. //
  821. // - (void)drawLabelTextContextSize:(CGSize)contextSize
  822. // {
  823. // CGFloat KBlankA4W =contextSize.width;
  824. // CGFloat KBlankA4H =contextSize.height;
  825. //
  826. // NSString *contextString = nil;
  827. // if (self.PDFPrint.labelString && self.PDFPrint.labelString.length > 0) {
  828. // contextString = self.PDFPrint.labelString;
  829. // } else {
  830. // NSDate *date = [NSDate date];
  831. // NSDateFormatter *formatter = [[[NSDateFormatter alloc] init] autorelease];
  832. // [formatter setDateFormat:@"YYYY-MM-dd hh:mm:ss"];
  833. // contextString = [NSString stringWithFormat:@"(%ld,%ld) %@ %@",self.columnIndex,self.lineIndex,[self.filePath lastPathComponent],[formatter stringFromDate:date]];
  834. // }
  835. //
  836. // CGFloat fontSize = 12.0 * (MAX(KBlankA4W, KBlankA4H)/842);
  837. // NSFont *font = [NSFont systemFontOfSize:fontSize];
  838. // NSColor *color = [NSColor blackColor];
  839. //
  840. // NSSize size = NSZeroSize;
  841. // NSMutableParagraphStyle *style = [[[NSMutableParagraphStyle alloc] init] autorelease];
  842. // [style setAlignment:NSCenterTextAlignment];
  843. // [style setLineBreakMode:NSLineBreakByCharWrapping];
  844. // NSMutableDictionary *dictionary = [NSMutableDictionary dictionary];
  845. // [dictionary setObject:style forKey:NSParagraphStyleAttributeName];
  846. // [dictionary setObject:color forKey:NSForegroundColorAttributeName];
  847. // [dictionary setObject:font forKey:NSFontAttributeName];
  848. // size = [contextString boundingRectWithSize:CGSizeMake(MAXFLOAT, MAXFLOAT)
  849. // options:NSStringDrawingUsesLineFragmentOrigin | NSStringDrawingUsesFontLeading
  850. // attributes:dictionary].size;
  851. //
  852. // if (self.PDFPrint.splitType == kKMPDFPosterSplitType_PageNumber) {
  853. // [contextString drawInRect:CGRectMake(self.PDFPrint.fullPageEdgeInsets.left +10,
  854. // KBlankA4H - self.PDFPrint.fullPageEdgeInsets.top + size.height,
  855. // size.width, size.height)
  856. // withAttributes:dictionary];
  857. // } else {
  858. // [contextString drawInRect:CGRectMake(self.PDFPrint.edgeInsets.left +10,
  859. // KBlankA4H - self.PDFPrint.edgeInsets.top + size.height,
  860. // size.width, size.height)
  861. // withAttributes:dictionary];
  862. // }
  863. //
  864. // }
  865. }
  866. protocol KMPrintPresenterPrivate {}
  867. extension KMPrintPresenter: KMPrintPresenterPrivate {
  868. func isAnnoationStamp(type: String) -> Bool {
  869. let annotationStamp: [String] = ["Square"]
  870. return annotationStamp.contains(type)
  871. }
  872. func isAnnoationMarkup(type: String) -> Bool {
  873. let annotationStamp: [String] = ["Widget", "Freehand"]
  874. return annotationStamp.contains(type)
  875. }
  876. func isAnnoationForm(type: String) -> Bool {
  877. let annotationStamp: [String] = ["FreeText"]
  878. return annotationStamp.contains(type)
  879. }
  880. }
  881. protocol KMPrintPresenterProtocol: NSObject {
  882. }
  883. ///**
  884. // @abstract 获取context
  885. // @param size纸张大小
  886. // */
  887. //func createContext(_ saveFilePath: String, size: CGSize) -> CGContext {
  888. // let s = CGSize(width: nearbyint(size.width), height: nearbyint(size.height))
  889. // let rep = NSBitmapImageRep.init(bitmapDataPlanes: nil,
  890. // pixelsWide: Int(s.width),
  891. // pixelsHigh: Int(s.height),
  892. // bitsPerSample: 8,
  893. // samplesPerPixel: 4,
  894. // hasAlpha: true,
  895. // isPlanar: false,
  896. // colorSpaceName: NSColorSpaceName.calibratedRGB,
  897. // bytesPerRow: 0,
  898. // bitsPerPixel: 0)!
  899. // let context: CGContext = NSGraphicsContext.init(bitmapImageRep: rep)!.cgContext
  900. // return context
  901. //}