123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756 |
- import Cocoa
- import PDFKit
- extension CPDFPage {
- var cropDrawingPage: CPDFPage? {
- get {
- return (objc_getAssociatedObject(self, "drawingPage") as! CPDFPage)
- }
- set {
- objc_setAssociatedObject(self, "drawingPage", newValue, objc_AssociationPolicy.OBJC_ASSOCIATION_RETAIN_NONATOMIC)
- }
- }
-
- var cropOffsetX: CGFloat {
- get {
- return objc_getAssociatedObject(self, "cropOffsetX") as! CGFloat
- }
- set {
- objc_setAssociatedObject(self, "cropOffsetX", newValue, objc_AssociationPolicy.OBJC_ASSOCIATION_ASSIGN)
- }
- }
-
- var cropOffsetY: CGFloat {
- get {
- return objc_getAssociatedObject(self, "cropOffsetY") as! CGFloat
- }
- set {
- objc_setAssociatedObject(self, "cropOffsetY", newValue, objc_AssociationPolicy.OBJC_ASSOCIATION_ASSIGN)
- }
- }
-
- var isChangePageSize: Bool {
- get {
- return objc_getAssociatedObject(self, "isChangePageSize") as! Bool
- }
- set {
- objc_setAssociatedObject(self, "isChangePageSize", newValue, objc_AssociationPolicy.OBJC_ASSOCIATION_ASSIGN)
- }
- }
-
- var cropRect: NSRect {
- get {
- return objc_getAssociatedObject(self, "cropRect") as! NSRect
- }
- set {
- objc_setAssociatedObject(self, "cropRect", newValue, objc_AssociationPolicy.OBJC_ASSOCIATION_ASSIGN)
- }
- }
- }
- class KMCropInfo: NSObject {
- var drawPage: CPDFPage!
- var cropOffsetX: CGFloat = 0
- var cropOffsetY: CGFloat = 0
- var isChangePageSize: Bool = false
- var cropRect: NSRect!
- }
- class KMCropPDFView: CPDFView {
-
- var drawInfo: Array<KMCropInfo> = []
-
- override func draw(_ page: CPDFPage!, to context: CGContext!) {
- super.draw(page, to: context)
-
- if (self.drawInfo.count > 0) {
- drawCropPage(page, to: context)
- }
- }
-
- private func drawCropPage(_ cropPage: CPDFPage, to context: CGContext!) {
- let info: KMCropInfo = self.drawInfo[Int(self.document.index(for: cropPage))]
- if (info.drawPage != nil) {
- let originalRect = cropPage.bounds(for: .mediaBox)
- let drawPage = info.drawPage
- let cropRect = drawPage!.bounds(for: .mediaBox)
-
- let orgPageWidth: CGFloat = NSWidth(cropRect)
-
- let orgPageHeight: CGFloat = NSHeight(cropRect)
-
- let spaceX = info.cropOffsetX
- let spaceY = info.cropOffsetY
-
- let scale: CGFloat = min(NSWidth(originalRect)/orgPageWidth, NSHeight(originalRect)/orgPageHeight)
-
- NSGraphicsContext.saveGraphicsState()
-
- context.translateBy(x: spaceX,y: spaceY)
- context.scaleBy(x: scale, y: scale)
-
- if #available(macOS 10.12, *) {
- drawPage?.draw(with: .cropBox, to: context)
- drawPage?.transform(context, for: .cropBox)
- } else {
- NSGraphicsContext.saveGraphicsState()
- NSGraphicsContext.current = NSGraphicsContext(cgContext: context, flipped: false)
- drawPage?.draw(with: .cropBox, to: context)
- drawPage?.transform(context, for: .cropBox)
- }
-
- if (info.isChangePageSize) {
- let tCropRect: NSRect = cropPage.cropRect
- let w1: CGFloat = orgPageWidth-(tCropRect.size.width)-(tCropRect.origin.x)
-
- context.setFillColor(red: 0, green: 0, blue: 0, alpha: 0.3)
- let rect1: NSRect = NSMakeRect(0, 0, orgPageWidth, (tCropRect.origin.y))
- let rect2: NSRect = NSMakeRect(0, (tCropRect.origin.y)+(tCropRect.size.height), orgPageWidth, orgPageHeight - tCropRect.origin.y-tCropRect.size.height)
- let rect3: NSRect = NSMakeRect(0, tCropRect.origin.y,tCropRect.origin.x,tCropRect.size.height);
- var rect4Y = w1
- if (w1 > 0) {
- rect4Y = 0
- }
- let rect4: NSRect = NSMakeRect(tCropRect.origin.x+tCropRect.size.width, tCropRect.origin.y,rect4Y,tCropRect.size.height)
-
- context.fill(rect1)
- context.fill(rect2)
- context.fill(rect3)
- context.fill(rect4)
- }
-
- NSGraphicsContext.restoreGraphicsState()
- }
- }
- }
- class KMCropPDFPage: CPDFPage {
-
-
- override func draw(with box: CPDFDisplayBox, to context: CGContext!) {
- super.draw(with: box, to: context)
-
- let pageSize: NSSize = self.bounds(for: .mediaBox).size
- drawPage(with: context, page: self.cropDrawingPage!, pageSize: pageSize)
- }
-
- private func drawPage(with context: CGContext, page: CPDFPage, pageSize: NSSize) {
- let myPage: KMCropPDFPage = page as! KMCropPDFPage
-
- var originalSize: NSSize = page.bounds(for: .mediaBox).size
-
-
- if (page.rotation % 180 != 0) {
- originalSize = NSMakeSize(originalSize.height, originalSize.width)
- }
-
- context.saveGState()
- context.translateBy(x: myPage.cropOffsetX, y: myPage.cropOffsetY)
-
- if #available(macOS 10.12, *) {
- page.draw(with: .cropBox, to: context)
- page.transform(context, for: .cropBox)
- } else {
- NSGraphicsContext.saveGraphicsState()
- NSGraphicsContext.current = NSGraphicsContext(cgContext: context, flipped: true)
- page.draw(with: .cropBox, to: context)
-
- NSGraphicsContext.restoreGraphicsState()
- page.transform(context, for: .cropBox)
- }
-
- context.restoreGState()
- }
- }
- class KMCropPreviewController: KMWatermarkAdjectivePreViewBaseController {
-
- var tipView = KMCropTipView()
-
- var windowController: KMCropSettingWindowController!
-
- override func viewDidLoad() {
- super.viewDidLoad()
-
- let itemTitles = [["裁剪"]]
- var itemModels: Array<Array<KMWatermarkAdjectiveTopBarItemModel>> = []
- for items in itemTitles {
- var array: Array<KMWatermarkAdjectiveTopBarItemModel> = []
- for title in items {
- let model = KMWatermarkAdjectiveTopBarItemModel()
- model.iconName = ""
- model.itemTitle = title
- array.append(model)
- }
- itemModels.append(array)
- }
-
- self.topBarView.initItemData(itemArrays: itemModels)
-
- let preView: CPDFListView = CPDFListView()
- self.preView = preView
- self.preView.frame = self.preViewBox.contentView!.bounds
- self.preView.autoresizingMask = NSView.AutoresizingMask(rawValue: 18)
- self.preViewBox.contentView?.addSubview(self.preView)
- self.preView.autoScales = true
- self.preView.displaysAsBook = true
- self.preView.delegate = self
-
- let myPreView: CPDFListView = self.preView as! CPDFListView
- myPreView.pdfListViewDelegate = self
-
-
-
- self.topBarView.isCanApply(can: false)
-
- self.tipView.frame = self.tipBox.contentView!.frame
- self.tipView.autoresizingMask = NSView.AutoresizingMask(rawValue: 18)
- self.tipBox.addSubview(self.tipView)
- self.tipBox.isHidden = true
-
- self.tipView.enterAction = {
- () in
- let myPreView: CPDFListView = self.preView as! CPDFListView
- var rect = NSIntegralRect(myPreView.currentSelectionRect())
- if (NSIsEmptyRect(rect)) {
- return
- }
-
- let window = KMCropSettingWindowController(windowNibName: "KMCropSettingWindowController")
- self.view.window?.beginSheet(window.window!)
- self.windowController = window
-
- window.itemClick = { [self]
- (index: Int) in
- if (index == 1) {
- self.view.window?.endSheet((self.windowController?.window)!)
- self.windowController = nil
- return
- }
- let pageRangeType = self.windowController.pageRangeIndex
- let pageCount: Int = Int(self.preView.document.pageCount)
- var pages: Array<Int> = []
- if (pageRangeType == 0) {
- pages.append(self.preView.currentPageIndex)
- } else if (pageRangeType == 1) {
- for i in 0 ..< pageCount {
- pages.append(i)
- }
- } else if (pageRangeType == 2) {
- var string: String = ""
- for i in 0 ..< pageCount {
- if (i % 2 == 1) {
- continue
- }
-
- pages.append(i)
- }
- } else if (pageRangeType == 3) {
- var string: String = ""
- for i in 0 ..< pageCount {
- if (i % 2 == 0) {
- continue
- }
-
- pages.append(i)
- }
- } else {
- for i in self.windowController.pageRangePages {
- pages.append(i)
- }
- }
-
- if (pages.count < 0) {
- let alert = NSAlert()
- alert.messageText = "请选择页面"
- alert.runModal()
- return
- }
-
- var pageSize: NSSize = NSZeroSize
- if (self.windowController.pageSize == "None") {
-
- } else {
- pageSize = KMCropTools.getPageSizeValue(self.windowController.pageSize)
- }
-
- for i in pages {
- let myPreView: CPDFListView = self.preView as! CPDFListView
- var page: CPDFPage = self.preView.document.page(at: UInt(i))
- var rect = NSIntegralRect(myPreView.selectionRect)
- if (NSIsEmptyRect(rect)) {
- rect = getPageForegroundBox(page)
- }
-
- var newRect = NSIntersectionRect(rect, (page.bounds(for: .mediaBox)))
- page.setBounds(newRect, for: .cropBox)
-
- if (pageSize.width == 0 && pageSize.height == 0) {
-
- } else {
- let tiffData = page.PDFListViewTIFFData(for: rect)
- let index: UInt = (page.pageIndex())
-
- self.preView.document.removePage(at: index)
-
- let result = self.preView.document.insertPage(pageSize, at: index)
- }
- }
-
-
- let toPath: String = self.preView.document.documentURL.path
- let documentPath = NSTemporaryDirectory()
- let tempPath: String = "\(documentPath)/\(toPath.lastPathComponent)"
- if (FileManager.default.fileExists(atPath: tempPath)) {
- try?FileManager.default.removeItem(atPath: tempPath)
- }
- let result = self.preView.document.write(to: URL(fileURLWithPath: tempPath))
- if (result) {
- if (FileManager.default.fileExists(atPath: toPath)) {
- try?FileManager.default.removeItem(atPath: toPath)
- }
-
- try?FileManager.default.moveItem(atPath: tempPath, toPath: toPath)
- } else {
- try?FileManager.default.removeItem(atPath: tempPath)
- }
- DispatchQueue.main.async {
- let myPreView: CPDFListView = self.preView as! CPDFListView
- myPreView.toolMode = .textToolMode
-
- self.preView.layoutDocumentView()
-
- self.preView.displayBox = .cropBox
- }
-
-
- self.view.window?.endSheet((self.windowController?.window)!)
- self.windowController = nil
-
-
- let count: Int = Int(self.preView.document!.pageCount)
-
-
- }
-
- }
- }
-
-
- override func viewDidAppear() {
- super.viewDidAppear()
-
- }
-
- override func topItemClick(index: Int) {
- let menu = NSMenu()
-
- let titles = ["裁剪当前页面 - 白边距", "裁剪所有页面 - 自动", "自定义裁剪区域"]
- for title in titles {
- let item = NSMenuItem(title: title, action: #selector(itemAction), target: self)
- item.tag = titles.firstIndex(of: title)!
- menu.addItem(item)
- }
-
- menu.popUp(positioning: nil, at: NSPoint(x: self.topBarView.frame.midX-100, y: 0), in: self.topBarView)
- }
-
- @objc private func itemAction(sender: NSMenuItem) {
- if (sender.tag == 0) {
-
- self.tipBox.isHidden = true
- cropCurrentPage()
- return
- }
-
- if (sender.tag == 1) {
-
- self.tipBox.isHidden = true
- cropAllPate()
- return
- }
-
-
- let myPreView: CPDFListView = self.preView as! CPDFListView
- myPreView.toolMode = .selectToolMode
-
- self.preView.autoScales = true
- self.preView.autoScales = false
- var pageHeight: CGFloat = NSHeight(self.preView.currentPage().bounds(for: self.preView.displayBox))
- if (self.preView.displaysPageBreaks) {
- pageHeight += 8
- }
-
- var scaleFactor: CGFloat = fmax(self.preView.minimumScaleFactor, NSHeight(self.preView.frame)/pageHeight)
- self.preView.scaleFactor = scaleFactor
-
- self.tipBox.isHidden = false
- self.tipView.setString(string: "请框选裁剪区域")
- }
-
- private func cropCurrentPage(in pageSize: NSSize) {
- let myPreView: CPDFListView = self.preView as! CPDFListView
- var rect = NSIntegralRect(myPreView.currentSelectionRect())
- var page: CPDFPage?
- if ((myPreView.currentSelectionPage()) != nil) {
- page = myPreView.currentSelectionPage()
- } else {
- page = myPreView.currentPage()
- }
- if (NSIsEmptyRect(rect)) {
- rect = getPageForegroundBox(page!)
- }
-
- let tiffData = page!.PDFListViewTIFFData(for: rect)
-
- let index: UInt = (page?.pageIndex())!
-
- var newRect = NSIntersectionRect(rect, (page?.bounds(for: .mediaBox))!)
- page?.setBounds(newRect, for: .cropBox)
-
- self.preView.document.removePage(at: index)
-
-
- let result = self.preView.document.insertPage(pageSize, at: index)
-
-
- self.preView.layoutDocumentView()
-
- self.preView.displayBox = .cropBox
- }
-
- private func cropAllPate() {
- var size = NSZeroSize
- for i in 0 ..< self.preView.document.pageCount {
- let page = self.preView.document.page(at: i)
- var rect = getPageForegroundBox(page!)
- size.width = fmax(size.width, NSWidth(rect))
- size.height = fmax(size.height, NSHeight(rect))
- }
-
- var rectArray: Array<NSRect> = []
- for i in 0 ..< self.preView.document.pageCount {
- let page = self.preView.document.page(at: i)
- var rect = getPageForegroundBox(page!)
- var bounds: NSRect = (page?.bounds(for: .mediaBox))!
- if (rect.minX - bounds.minX > bounds.maxX-rect.maxX) {
- rect.origin.x = rect.maxX-size.width
- }
-
- rect.origin.y = rect.maxY-size.height
- rect.size = size
-
- if (NSWidth(rect) > NSWidth(bounds)) {
- rect.size.width = NSWidth(bounds)
- }
- if (NSHeight(rect) > NSHeight(bounds)) {
- rect.size.height = NSHeight(bounds)
- }
-
- if (NSMinX(rect) < NSMinX(bounds)) {
- rect.origin.x = NSMinX(bounds)
- } else if (NSMaxX(rect) > NSMaxX(bounds)) {
- rect.origin.x = NSMaxX(bounds) - NSWidth(rect)
- }
-
- if (NSMinY(rect) < NSMinY(bounds)) {
- rect.origin.y = NSMinY(bounds)
- } else if (NSMaxY(rect) > NSMaxY(bounds)) {
- rect.origin.y = NSMaxY(bounds) - NSHeight(rect)
- }
-
- rectArray.append(rect)
- }
-
- cropPages(to: rectArray)
- }
-
- private func cropPages(to rects: Array<NSRect>) {
- let currentPage = self.preView.currentPage()
- let visibleRect: NSRect = self.preView.convert(self.preView.convert(self.preView.documentView().visibleRect, from: self.preView.documentView()), to: self.preView.currentPage())
-
- var oldRectArray: Array<NSRect> = []
- for i in 0 ..< self.preView.document.pageCount {
- let page = self.preView.document.page(at: i)
- var rect = NSIntersectionRect(rects[Int(i)], (page?.bounds(for: .mediaBox))!)
- let oldRect = page?.bounds(for: .cropBox)
- oldRectArray.append(oldRect!)
-
- page?.setBounds(rect, for: .cropBox)
- }
-
-
- self.preView.layoutDocumentView()
-
- self.preView.displayBox = .cropBox
-
- self.preView.go(to: currentPage)
- self.preView.go(to: visibleRect, on: currentPage)
- }
-
- private func cropCurrentPage() {
- let myPreView: CPDFListView = self.preView as! CPDFListView
- var rect = NSIntegralRect(myPreView.currentSelectionRect())
- var page: CPDFPage?
- if ((myPreView.currentSelectionPage()) != nil) {
- page = myPreView.currentSelectionPage()
- } else {
- page = myPreView.currentPage()
- }
- if (NSIsEmptyRect(rect)) {
- rect = getPageForegroundBox(page!)
- }
-
- let index: UInt = (page?.pageIndex())!
- cropPage(at: index, in: rect)
- }
-
- func cropPage(at index: UInt, in rect: NSRect) {
- let oldRect = self.preView.document.page(at: index)?.bounds(for: .cropBox)
-
-
-
- var page = self.preView.document.page(at: index)
- var newRect = NSIntersectionRect(rect, (page?.bounds(for: .mediaBox))!)
- page?.setBounds(newRect, for: .cropBox)
-
-
- self.preView.layoutDocumentView()
-
- self.preView.displayBox = .cropBox
- }
- private func getPageForegroundBox(_ page: CPDFPage) -> NSRect {
- let marginWidth: CGFloat = 10
- let marginHeight: CGFloat = 10
-
- let imageRep = newBitmapImageRepForBox(page, .mediaBox)
- let bounds = page.bounds(for: .mediaBox)
- var foregroundBox = imageRep.foregroundRect()
- if (imageRep == nil) {
- foregroundBox = bounds
- } else if (NSIsEmptyRect(foregroundBox)) {
- let centerPoint = NSPoint(x: bounds.midX, y: bounds.midY)
- let origin = NSPoint(x: round(centerPoint.x), y: round(centerPoint.y))
- foregroundBox.origin = origin
- foregroundBox.size = NSZeroSize
- } else {
- let origin = NSPoint(x: foregroundBox.origin.x+bounds.origin.x, y: foregroundBox.origin.y+bounds.origin.y)
- foregroundBox.origin = origin
- }
-
- return NSIntegralRect(NSInsetRect(foregroundBox, -marginWidth, -marginHeight))
- }
-
- private func newBitmapImageRepForBox(_ page: CPDFPage, _ box: CPDFDisplayBox) -> NSBitmapImageRep {
- let bounds = page.bounds(for: box)
- var imageRep = NSBitmapImageRep(bitmapDataPlanes: nil,
- pixelsWide: Int(NSWidth(bounds)),
- pixelsHigh: Int(NSHeight(bounds)),
- bitsPerSample: 8,
- samplesPerPixel: 4,
- hasAlpha: true,
- isPlanar: false,
- colorSpaceName: .calibratedRGB,
- bitmapFormat: NSBitmapImageRep.Format(rawValue: 0),
- bytesPerRow: 0,
- bitsPerPixel: 32)
-
- if (imageRep != nil) {
- NSGraphicsContext.saveGraphicsState()
- NSGraphicsContext.current = NSGraphicsContext.init(bitmapImageRep: imageRep!)
- NSGraphicsContext.current?.imageInterpolation = .none
- NSGraphicsContext.current?.shouldAntialias = false
- if (page.rotation != 0) {
- var transform = NSAffineTransform()
- if (page.rotation == 90) {
- transform.translateX(by: NSWidth(bounds), yBy: 0)
- } else if (page.rotation == 180) {
- transform.translateX(by: NSHeight(bounds), yBy: NSWidth(bounds))
- } else if (page.rotation == 270) {
- transform.translateX(by: 0, yBy: NSHeight(bounds))
- }
-
- transform.rotate(byDegrees: CGFloat(page.rotation))
- transform.concat()
- }
-
- page.draw(with: box, to: (NSGraphicsContext.current?.cgContext as! CGContext))
- NSGraphicsContext.current?.imageInterpolation = .default
- NSGraphicsContext.restoreGraphicsState()
- }
-
- return imageRep!
- }
- }
- extension KMCropPreviewController: CPDFViewDelegate {
- func pdfViewEditingSelectionDidChanged(_ pdfView: CPDFView!) {
-
- }
- }
- extension KMCropPreviewController: CPDFListViewDelegate {
- func pdfListViewDidSelectionEnd(_ pdfListView: CPDFListView!) {
- if (self.preView.isEqual(to: pdfListView)) {
- self.tipView.setString(string: "请按 Enter 键确定裁剪区域")
- }
- }
- }
-
|