// // CompareViewController.swift // KdanAuto // // Created by 朱东勇 on 2023/2/2. // import Foundation import AppKit import PDFKit class CompareViewController: NSViewController, NSTableViewDataSource, NSTableViewDelegate, NSPopoverDelegate { @IBOutlet var listView : NSTableView! @IBOutlet var filterBtn : NSButton! @IBOutlet var closeBtn : NSButton! @IBOutlet var progressIndicator : NSProgressIndicator! var _files : NSArray! var _showFiles : NSMutableArray! var _popover: NSPopover! static var sharedCViewController : CompareViewController? = nil class func shared() -> CompareViewController { if nil == sharedCViewController { var objects : NSArray! Bundle.main.loadNibNamed("CompareViewController", owner: nil, topLevelObjects: &objects) for tView in objects { if let tv = tView as? CompareViewController { sharedCViewController = tv } } } return sharedCViewController! } override func viewDidLoad() { self.listView.register(NSNib.init(nibNamed: NSNib.Name.init(NSString(string: "ImageCompareCellView")), bundle: Bundle.main), forIdentifier: NSUserInterfaceItemIdentifier.init("ImageCompareCellView")) } // IBAction @IBAction func filterSimilarItemAction(_ sender:NSButton) { self.setFiles(_files); } @IBAction func exportAction(_ sender:NSButton) { if (_files.count == 0) { let alert = NSAlert.init() alert.messageText = "无测试对比项导出"; alert.runModal(); return }else if (_showFiles.count == 0) { let alert = NSAlert.init() alert.messageText = "所有对比项均无差异"; alert.runModal(); return } let savePanel = NSSavePanel.init(); let checkBox = NSButton(checkboxWithTitle: "过滤无差异项", target: self, action: nil); checkBox.state = self.filterBtn.state; checkBox.frame = CGRectMake(0, 0, 150, 45); savePanel.allowedContentTypes = [.pdf] savePanel.accessoryView = checkBox if (savePanel.runModal() == .OK) { var exportFiles = NSMutableArray() if (checkBox.state == .on) { for tfile in NSArray(array: _files) { let file = tfile as! NSMutableDictionary; let degree = file.degree() if (fabs(degree - 100) > 0) { exportFiles.add(file); } } if (exportFiles.count == 0) { let alert = NSAlert.init() alert.messageText = "所有对比项均无差异"; alert.runModal(); return } }else { exportFiles = NSMutableArray(array: _files) } let url = savePanel.url!; NSLog("\(url)") let pdf = PDFDocument.init(); let needFilter = checkBox.state == .on self.progressIndicator.isHidden = false; self.progressIndicator.doubleValue = 0; processNextImages(pdf, index: 0, files:exportFiles, url: url, needFilter: needFilter) } } // Save PDF func processNextImages(_ pdf:PDFDocument, index:Int32, files:NSMutableArray, url:URL, needFilter:Bool) -> Void { let file = files[Int(index)] as! NSMutableDictionary NSLog("处理第\(index)页") // 所有文档均导出 let cell = ImageCompareCellView.shared() cell?.frame = CGRectMake(0, 0, CGRectGetWidth(self.listView.frame), 900); cell?.setFileInfo(file); self.closeBtn.isEnabled = false; DispatchQueue.global().async { autoreleasepool { let image = cell?.processCompareImage() if (image != nil) { let page = PDFPage.init(image: image!) let nPage = page?.copy() as! PDFPage pdf.insert(nPage, at: pdf.pageCount) } } let degree = file.degree() if (abs(degree - 100) > 0) { autoreleasepool { let image = cell?.processCoverImage() if (image != nil) { let page = PDFPage.init(image: image!) let nPage = page?.copy() as! PDFPage pdf.insert(nPage, at: pdf.pageCount) } } } if ((index+1) < files.count) { DispatchQueue.main.asyncAfter(deadline: DispatchTime.init(uptimeNanoseconds: 100)) { self.progressIndicator.doubleValue = Double(Int(index) * 100/files.count); self.processNextImages(pdf, index: index+1, files: files, url: url, needFilter: needFilter); }; }else { DispatchQueue.global().async { autoreleasepool { pdf.write(to: url) DispatchQueue.main.async { self.progressIndicator.isHidden = true; self.progressIndicator.doubleValue = 100.0; self.closeBtn.isEnabled = true; NSWorkspace.shared.activateFileViewerSelecting([url]) } } } } } } // Show func showIn(_ view:NSView, rect:NSRect) { if _popover == nil { _popover = NSPopover.init() _popover.contentViewController = self; _popover.delegate = self } _popover.contentSize = self.view.frame.size _popover.show(relativeTo: rect, of: view, preferredEdge: NSRectEdge.minY) } // Getter & Setter func setFiles(_ files:NSArray) { _files = files if (self.filterBtn.state == .on) { _showFiles = NSMutableArray() for tfile in NSArray(array: _files) { let file = tfile as! NSMutableDictionary; let degree = file.degree() if (fabs(degree - 100) > 0) { _showFiles.add(file); } } }else { _showFiles = NSMutableArray (array: _files); } listView.reloadData() } // IBAction @IBAction func closeAction(_ sender: NSButton) { _popover.close() } // TableView Delegate func tableView(_ tableView: NSTableView, viewFor tableColumn: NSTableColumn?, row: Int) -> NSView? { return autoreleasepool { var cellView = tableView.makeView(withIdentifier: .init("ImageCompareCellView"), owner: nil) as! ImageCompareCellView? if (nil == cellView) { cellView = ImageCompareCellView.shared() ?? nil } if (_showFiles.count >= row) { let fileInfo = _showFiles[row] as! NSMutableDictionary cellView?.setFileInfo(fileInfo) } return cellView } } func selectionShouldChange(in tableView: NSTableView) -> Bool { return true } func tableView(_ tableView: NSTableView, shouldSelectRow row: Int) -> Bool { return false } func tableView(_ tableView: NSTableView, shouldSelect tableColumn: NSTableColumn?) -> Bool { return false } func tableView(_ tableView: NSTableView, mouseDownInHeaderOf tableColumn: NSTableColumn) { } func tableView(_ tableView: NSTableView, didClick tableColumn: NSTableColumn) { } func tableView(_ tableView: NSTableView, didDrag tableColumn: NSTableColumn) { } func tableView(_ tableView: NSTableView, heightOfRow row: Int) -> CGFloat { return 900 } func tableView(_ tableView: NSTableView, isGroupRow row: Int) -> Bool { return false } func tableView(_ tableView: NSTableView, sizeToFitWidthOfColumn column: Int) -> CGFloat { return tableView.frame.width } func tableView(_ tableView: NSTableView, rowActionsForRow row: Int, edge: NSTableView.RowActionEdge) -> [NSTableViewRowAction] { return [] } func tableViewSelectionDidChange(_ notification: Notification) { } // TableView Data Source func numberOfRows(in tableView: NSTableView) -> Int { if nil == _showFiles { return 0 } return _showFiles.count } // func popoverShouldClose(_ popover: NSPopover) -> Bool { return true } }