// // KMPrintPresenter_C.swift // PDF Reader Pro // // Created by lizhe on 2022/12/21. // // import Cocoa class KMPrintPresenter_C: NSObject { lazy var printData: KMPrintModel! = KMPrintModel() var document: CPDFDocument? fileprivate weak var delegate: KMPrintPresenter_CDeleage? /** 初始化presenter */ func initPresenter(delegate: KMPrintPresenter_CDeleage, data: KMPrintModel, document: CPDFDocument) { self.delegate = delegate self.printData = data self.updatePrintDocument(documentURL: document.documentURL, data: self.printData) } func reloadData() { self.updatePrintDocument(documentURL: self.document!.documentURL, data: self.printData) } /** @abstract 解除绑定 */ func free() { delegate = nil } } protocol KMPrintPresenter_CDeleage: NSObject { func showData(presenter: KMPrintPresenter_C, document: CPDFDocument) } protocol KMPrintPresenter_CDocument: NSObject {} extension KMPrintPresenter_C: KMPrintPresenter_CDocument { /** @abstract 获取打印document @param url 源文件url @param data 数据 @retrun document */ func updatePrintDocument(documentURL: URL, data: KMPrintModel) -> CPDFDocument { /** 获取基本参数 */ let printModel: KMPrintModel = data /** 绘制document */ //创建预览document let previewDocument = self.creatDocument(url: documentURL) //获取总page let pages = self.fetchPages(documentURL: documentURL, range: printModel.page.range) //插入page let insertSuccess = self.insertPages(document: previewDocument, printModel: printModel, pages: pages) if insertSuccess { KMPrint("插入成功") } else { KMPrint("插入失败") } //重置document let filePath = self.saveDocument(document: previewDocument) if self.delegate != nil { self.delegate?.showData(presenter: self, document: previewDocument) } KMPrint("保存地址" + filePath) return previewDocument } /** @abstract 插入page @param paperSet 纸张设置 @param pageSet page设置 @param pages page数组 */ func insertPages(document: CPDFDocument, printModel: KMPrintModel, pages: [CPDFPage]) -> Bool { //每张纸包含的apge数量 var pageOfPaperCount: Int = 1 switch printModel.page.operation.type { case .multipage: pageOfPaperCount = Int(printModel.page.operation.pageOfPaper.point.x * printModel.page.operation.pageOfPaper.point.y) case .poster: pageOfPaperCount = Int(printModel.page.operation.pageOfPaper.point.x * printModel.page.operation.pageOfPaper.point.y) default: pageOfPaperCount = 1 } //总页数 let count: Int = Int(ceil(Double(pages.count / pageOfPaperCount))) let size = printModel.page.operation.size var insertPages: [CPDFPage] = [] for i in 0...count { //获取多页page var drawPages: [CPDFPage] = [] for j in 0...pageOfPaperCount { if i * pageOfPaperCount + j < pages.count { let drawPage = pages[i * pageOfPaperCount + j] drawPages.append(drawPage) } } //TODO: 暂时屏蔽 let insertPage = KMPrintPage_C.init() insertPage.pages = drawPages insertPage.data = printModel insertPage.setBounds(CGRect(x: 0, y: 0, width: printModel.paper.info.size.width, height: printModel.paper.info.size.height), for: CPDFDisplayBox.mediaBox) insertPage.drawBlock = { [weak self] box, context, pages, data in //绘制page self?.drawPageToContext(context: context, pages: pages, data: data) } insertPages.append(insertPage) } var result = true if insertPages.count != 0 { for index in 0...insertPages.count - 1 { let page = insertPages[index] var success = false if (printModel.page.range.reversePrintOrder) { success = document.insertPageObject(page, at: 0) } else { success = document.insertPageObject(page, at: UInt(index)) } if success { KMPrint("插入成功") } else { KMPrint("插入失败") result = false } } } // var result = true // if pages.count != 0 { // for index in 0...pages.count - 1 { // let page = pages[index] // let success = document.insertPageObject(page, at: UInt(index)) // if success { // print("插入成功") // } else { // print("插入失败") // result = false // } // } // } return result } func creatDocument(url: URL) -> CPDFDocument { if FileManager.default.fileExists(atPath: NSTemporaryDirectory() + "/PDFReaderProTest") { try?FileManager.default.createDirectory(atPath: NSTemporaryDirectory() + "/PDFReaderProTest", withIntermediateDirectories: true) } let document = CPDFDocument(url: url)! // document.importPages(IndexSet(integer: 0), from: document, at: 0) let count = document.pageCount for _ in 0...count { document.removePage(at: 0) } return document } func saveDocument(document: CPDFDocument) -> String { let filePath = NSTemporaryDirectory() + "/PDFReaderProTest/test2.pdf" if FileManager.default.fileExists(atPath: filePath) { try?FileManager.default.removeItem(atPath: filePath) } let url: URL = NSURL(string: filePath)! as URL var success = document.write(to: url) if !success { success = NSData(data: document.dataRepresentation()).write(to: url, atomically: true) } if success { KMPrint("保存成功" + filePath) } else { KMPrint("保存失败") } return filePath } /** 获取pages @param type 页面类型 @param selectPages 当type 为custom时 传入选中page的下标 */ func fetchPages(documentURL: URL, range: KMPrintPageRange) -> [CPDFPage] { let document = CPDFDocument.init(url: documentURL)! if document.pageCount < 1 { let page: CPDFPage = document.page(at: 0) return [page] } else { var pagesArray: [CPDFPage] = [] switch range.type { case .allPage: for index in 0...document.pageCount - 1 { let page: CPDFPage = document.page(at: index)! pagesArray.append(page) } case .evenPage: for index in 0...document.pageCount - 1 { if index % 2 == 0 { let page: CPDFPage = document.page(at: index)! pagesArray.append(page) } } case .oddPage: for index in 0...document.pageCount - 1 { if index % 2 != 0 { let page: CPDFPage = document.page(at: index)! pagesArray.append(page) } } // case .currentPage: // let page = self.pdfView.currentPage() // if page != nil { // pagesArray.append(page!) // } else { // print("page不存在") // } case .custom: for index in range.selectPages { let page: CPDFPage = document.page(at: UInt(index - 1)) pagesArray.append(page) } default: KMPrint("无匹配选项") } return pagesArray } } /** @abstract 获取context @param size纸张大小 */ func createContext(size: CGSize) -> CGContext { let s = CGSize(width: nearbyint(size.width), height: nearbyint(size.height)) let rep = NSBitmapImageRep.init(bitmapDataPlanes: nil, pixelsWide: Int(s.width), pixelsHigh: Int(s.height), bitsPerSample: 8, samplesPerPixel: 4, hasAlpha: true, isPlanar: false, colorSpaceName: NSColorSpaceName.calibratedRGB, bytesPerRow: 0, bitsPerPixel: 0)! let context: CGContext = NSGraphicsContext.init(bitmapImageRep: rep)!.cgContext return context } /** @abstract 绘制page @param context @pages page数组 [CPDFPage] */ func drawPageToContext(context: CGContext, pages: [CPDFPage], data: KMPrintModel) { /** paper */ let paperDirection: KMPrintPaperDirectionType = .vertical //打印方向 let paperSize: CGSize = CGSize(width: 595, height: 842) //页面paper大小 var dealPaperSize: CGSize = paperSize //页面paper大小 let paperInset: NSEdgeInsets = NSEdgeInsets(top: 0, left: 0, bottom: 0, right: 0) //绘制paper大小 var columnCount: Int = 2 //行 var rowCount: Int = 2 //列 let columnSpace: CGFloat = 2.0 //行间距 let rowSpace: CGFloat = 2.0 //列间距 let border: Bool = true /** page */ let pageOrder: KMPrintPageOperation.Multipage.Order = .horizontal //页面顺序 var pageSize: CGSize = CGSize(width: 100, height: 100) //page大小 let contentType: KMPrintContentType = .document var scale: Float = 1.0 //缩放尺寸 let autoRotate: Bool = false /** 计算 */ if paperDirection == .horizontal { dealPaperSize = CGSize(width: dealPaperSize.height, height: dealPaperSize.width) } //减去inset dealPaperSize = CGSize(width: dealPaperSize.width - paperInset.left - paperInset.right, height: dealPaperSize.height - paperInset.bottom - paperInset.top) //计算page尺寸(减去page之间的间隙) pageSize = CGSize(width: (dealPaperSize.width - CGFloat((columnCount - 1)) * CGFloat(columnSpace)) / CGFloat(columnCount), height: (dealPaperSize.height - CGFloat((rowCount - 1)) * CGFloat(rowSpace)) / CGFloat(rowCount)) //如果是横向参数需切换 if pageOrder == .horizontal || pageOrder == .horizontalReverseSequence { let temp = columnCount columnCount = rowCount rowCount = temp } for i in 0...columnCount { for j in 0...rowCount { let index = j + i * columnCount if index < pages.count { //参数 let page = pages[index] let rect = page.bounds(for: CPDFDisplayBox.cropBox) let originSize = rect.size var dealSize = originSize var rotate = page.rotation var annoations: [CPDFAnnotation] = page.annotations //处理 switch contentType { case .document: annoations.removeAll() case .documentAndStamp: for annoation in annoations { if !(annoation.isKind(of: CPDFStampAnnotation.self)) { annoation.page.removeAnnotation(annoation) } } case .documentAndMarkup: for annoation in annoations { if !(annoation.isKind(of: CPDFMarkupAnnotation.self)) { annoation.page.removeAnnotation(annoation) } } default: break } if rotate == 90 || rotate == 270 { dealSize = CGSize(width: originSize.height, height: originSize.width) } //取出page横竖时能显示的最大Rect if autoRotate { //page旋转度数为0度或者180度时，在指定的大小内，能显示的比例 let scale1 = Float(min(pageSize.width / originSize.width, pageSize.height / originSize.height)) //page旋转度数为90度或者270度时，在指定的大小内，能显示的比例 let scale2 = Float(min(pageSize.width / originSize.height, pageSize.height / originSize.width)) scale = max(scale1, scale2) if scale1 > scale2 { //高为竖直排列时，显示最大，则需将page时90度或者270度需要进行旋转 if rotate == 90 || rotate == 270 { rotate = rotate - 90 } } else { //宽为竖直排列时，显示最大，则需将page时0度或者180度需要进行旋转 if rotate == 0 || rotate == 180 { rotate = rotate - 90 } } if rotate == 90 || rotate == 270 { dealSize = CGSize(width: originSize.height, height: originSize.width) } else { dealSize = CGSize(width: originSize.width, height: originSize.height) } } else { scale = Float(min(pageSize.width / dealSize.width, pageSize.height / dealSize.height)) } scale = min(scale, 1) //居中 let spaceX: CGFloat = (pageSize.width - dealSize.width * CGFloat(scale)) / 2 let spaceY: CGFloat = (pageSize.height - dealSize.height * CGFloat(scale))/2 var translateX: CGFloat = 0.0 var translateY: CGFloat = 0.0 switch pageOrder { case .horizontal: translateX = (pageSize.width + columnSpace) + CGFloat(j) + paperInset.left + spaceX //页面内容高度 + 下边的行间距 - 第几个cell的高度 +居中 translateY = dealPaperSize.height + paperInset.bottom - (pageSize.height + rowSpace) * (CGFloat(i) + 1) + spaceY + rowSpace case .horizontalReverseSequence: translateX = dealPaperSize.width + paperInset.left - (pageSize.width + columnSpace) * (CGFloat(j) + 1) + spaceX + columnSpace translateY = dealPaperSize.height - (pageSize.height + rowSpace) * (CGFloat(i) + 1) + spaceY + paperInset.bottom + rowSpace case .vertical: translateX = (pageSize.width + columnSpace) * CGFloat(i) + paperInset.left + spaceX translateY = dealPaperSize.height - (pageSize.height + rowSpace) * (CGFloat(j) + 1) + spaceY + paperInset.bottom + rowSpace case .verticalReverseSequence: translateX = dealPaperSize.width + paperInset.left - (pageSize.width + columnSpace) * (CGFloat(i) + 1) + spaceX + columnSpace translateY = dealPaperSize.height - (pageSize.height + rowSpace) * (CGFloat(j) + 1) + spaceY + paperInset.bottom + rowSpace } context.saveGState() //平移 context.translateBy(x: translateX, y: translateY) //缩放 context.scaleBy(x: CGFloat(scale), y: CGFloat(scale)) page.draw(with: CPDFDisplayBox.cropBox, to: context) page.transform(context, for: CPDFDisplayBox.cropBox) if border { var dirtyRect = rect if rotate == 90 || rotate == 270 { dirtyRect = NSMakeRect(dirtyRect.origin.x, dirtyRect.origin.y, dirtyRect.size.height, dirtyRect.size.width) } context.addRect(dirtyRect) context.setStrokeColor(red: 0.0, green: 0.0, blue: 0.0, alpha: 1.0) context.strokePath() } context.restoreGState() } } } } } protocol KMPrintPresenter_CProtocol: NSObject { }