KMConvertBaseWindowController.swift 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810
  1. //
  2. // KMConvertBaseWindowController.swift
  3. // PDF Master
  4. //
  5. // Created by tangchao on 2022/12/5.
  6. //
  7. import Cocoa
  8. import PDFKit
  9. let kKMConvertLanugageSelectedIndex: String = "KMConvertLanugageSelectedIndex"
  10. typealias KMConvertSettingViewPageRangeDidChangeCallback = (_ pageRange: KMPageRange) -> Void
  11. typealias KMConvertSettingViewPageRangeDidInputFinishCallback = (_ pageRangeString: String) -> ()
  12. typealias KMConvertSettingViewLanugageDidSelectedCallback = (_ index: Int) -> ()
  13. class KMConvertSettingView: NSView {
  14. private var myCurrentLanguage: String!
  15. var currentLanguage: String {
  16. get {
  17. return myCurrentLanguage
  18. }
  19. set {
  20. myCurrentLanguage = newValue
  21. let index = UserDefaults.standard.string(forKey: kKMConvertLanugageSelectedIndex)
  22. if (index != nil && !index!.isEmpty) {
  23. ocrLanuguageIndex = Int(index!)!
  24. return
  25. }
  26. if (newValue.hasPrefix("zh-Hans")) {
  27. ocrLanuguageIndex = 0
  28. } else if (newValue.hasPrefix("en")) {
  29. ocrLanuguageIndex = 2
  30. } else if (newValue.hasPrefix("zh-Hant")) {
  31. ocrLanuguageIndex = 1
  32. } else if (newValue.hasPrefix("fr")) {
  33. ocrLanuguageIndex = 3
  34. } else if (newValue.hasPrefix("de")) {
  35. ocrLanuguageIndex = 4
  36. } else if (newValue.hasPrefix("ht")) { /// 海地克里奥尔语 Haitian
  37. ocrLanuguageIndex = 5
  38. } else if (newValue.hasPrefix("ja")) {
  39. ocrLanuguageIndex = 6
  40. } else if (newValue.hasPrefix("ko")) {
  41. ocrLanuguageIndex = 7
  42. } else if (newValue.hasPrefix("la")) { /// 拉丁语 Latin
  43. ocrLanuguageIndex = 8
  44. } else {
  45. ocrLanuguageIndex = 2
  46. }
  47. }
  48. }
  49. var ocrLanuguageIndex: Int = 3
  50. var pageRangeSelectedIndex: Int = 0
  51. var pageRangeDidChange: KMConvertSettingViewPageRangeDidChangeCallback?
  52. var pageRangeDidInputFinishCallback: KMConvertSettingViewPageRangeDidInputFinishCallback!
  53. var lanugageDidSelected: KMConvertSettingViewLanugageDidSelectedCallback!
  54. var ocrItemView: KMConvertOCRSettingItemView!
  55. var pageRangeItemView: KMConvertPageRangeSettingItemView?
  56. var pageCount = 0 {
  57. didSet {
  58. self.pageRangeItemView?.pageCount = self.pageCount
  59. }
  60. }
  61. func getPageRangeString() -> String {
  62. return ""
  63. }
  64. }
  65. typealias KMConvertBaseWindowControllerItemClick = (Int) -> ()
  66. class KMConvertBaseWindowController: NSWindowController {
  67. @IBOutlet weak var titleLabel: NSTextField!
  68. @IBOutlet weak var leftBox: NSBox!
  69. @IBOutlet weak var prePDFView: KMCustomPDFView?
  70. @IBOutlet weak var numberBox: NSView!
  71. @IBOutlet weak var numberTextField: NSTextField!
  72. @IBOutlet weak var perLabel: NSTextField!
  73. @IBOutlet weak var totalNumberLabel: NSTextField!
  74. @IBOutlet weak var backBox: NSBox!
  75. @IBOutlet weak var nextBox: NSBox!
  76. @IBOutlet weak var rightScrollView: NSScrollView!
  77. @IBOutlet weak var batchButton: NSButton!
  78. @IBOutlet weak var canelBox: NSBox!
  79. @IBOutlet weak var convertBox: NSBox!
  80. var canelButtonVC: KMDesignButton!
  81. var convertButtonVC: KMDesignButton!
  82. var backButtonVC: KMDesignButton!
  83. var nextButtonVC: KMDesignButton!
  84. var itemClick: KMConvertBaseWindowControllerItemClick!
  85. var documentModel: KMDocumentModel!
  86. var currentPageIndex: Int = 1
  87. var pageRange = KMPageRange.all
  88. var pageRangeString: String = ""
  89. var settingView: KMConvertSettingView!
  90. private var maskView: NSButton!
  91. private var indicator: NSProgressIndicator!
  92. var subType: Int = 0
  93. deinit {
  94. KMPrint("KMConvertBaseWindowController 已释放")
  95. self.prePDFView?.removeFromSuperview()
  96. self.prePDFView = nil
  97. }
  98. var fileExtension: String {
  99. get {
  100. return ""
  101. }
  102. }
  103. var progressController: SKProgressController?
  104. var convert: KMPDFConvert?
  105. convenience init() {
  106. self.init(windowNibName: "KMConvertBaseWindowController")
  107. }
  108. override func awakeFromNib() {
  109. super.awakeFromNib()
  110. DispatchQueue.main.async {
  111. self.window?.makeFirstResponder(nil)
  112. }
  113. }
  114. override func windowDidLoad() {
  115. super.windowDidLoad()
  116. self.window?.appearance = NSAppearance(named: .aqua)
  117. canelButtonVC = KMDesignButton.init(withType: .Text)
  118. convertButtonVC = KMDesignButton.init(withType: .Text)
  119. backButtonVC = KMDesignButton.init(withType: .Image)
  120. nextButtonVC = KMDesignButton.init(withType: .Image)
  121. self.totalNumberLabel.lineBreakMode = .byTruncatingTail
  122. if documentModel != nil {
  123. let document = CPDFDocument(url: documentModel.documentURL)
  124. if (document!.isLocked) {
  125. document?.unlock(withPassword: self.documentModel.password)
  126. }
  127. self.prePDFView?.document = document!
  128. let number: Int = Int((self.prePDFView?.document.pageCount)!)
  129. let string = "\(number)"
  130. totalNumberLabel.stringValue = string
  131. }
  132. if (self.prePDFView?.documentView() != nil) {
  133. self.prePDFView?.documentView().enclosingScrollView?.hasVerticalScroller = false
  134. self.prePDFView?.documentView().enclosingScrollView?.hasHorizontalScroller = false
  135. }
  136. self.prePDFView?.setDisplay(.singlePage)
  137. self.prePDFView?.layoutDocumentView()
  138. self.prePDFView?.autoScales = true
  139. self.prePDFView?.delegate = self
  140. backBox.fillColor = .clear
  141. backBox.contentView = backButtonVC.view
  142. backButtonVC.target = self
  143. backButtonVC.action = #selector(backButtonAction)
  144. backButtonVC.image = NSImage(named: "KMImageNameLeftButtonImage")!
  145. backButtonVC.image_disabled = NSImage(named: "KMImageNameLeftButtonImageDis")!
  146. backButtonVC.pagination()
  147. numberBox.wantsLayer = true
  148. self.numberBox.layer?.backgroundColor = NSColor.white.cgColor
  149. numberBox.layer?.borderWidth = 1
  150. numberBox.layer?.cornerRadius = 4
  151. numberTextField.focusRingType = .none
  152. numberTextField.delegate = self
  153. self.numberTextField.isBordered = false
  154. nextBox.fillColor = .clear
  155. nextBox.contentView = nextButtonVC.view
  156. nextButtonVC.target = self
  157. nextButtonVC.action = #selector(nextButtonAction)
  158. nextButtonVC.image = NSImage(named: "KMImageNameRightButtonImage")!
  159. nextButtonVC.image_disabled = NSImage(named: "KMImageNameRightButtonImageDis")!
  160. nextButtonVC.pagination()
  161. batchButton.title = NSLocalizedString("Batch", comment: "")
  162. batchButton.isBordered = false
  163. batchButton.wantsLayer = true
  164. batchButton.layer?.borderWidth = 1
  165. batchButton.layer?.cornerRadius = 4
  166. batchButton.target = self
  167. batchButton.action = #selector(batchButtonAction)
  168. batchButton.isHidden = true
  169. canelBox.fillColor = .clear
  170. canelBox.contentView = canelButtonVC.view
  171. canelButtonVC.target = self
  172. canelButtonVC.action = #selector(cancelButtonAction)
  173. canelButtonVC.stringValue = NSLocalizedString("Cancel", comment: "")
  174. canelButtonVC.button(type: .Sec, size: .m)
  175. canelButtonVC.button.keyEquivalent = KMKeyEquivalent.esc.string()
  176. convertBox.fillColor = .clear
  177. convertBox.contentView = convertButtonVC.view
  178. convertButtonVC.target = self
  179. convertButtonVC.action = #selector(convertButtonAction)
  180. convertButtonVC.stringValue = NSLocalizedString("Convert", comment: "")
  181. convertButtonVC.button(type: .Cta, size: .m)
  182. // convertButtonVC.button.keyEquivalent = KMKeyEquivalent.enter
  183. self.rightScrollView.hasVerticalScroller = false
  184. self.rightScrollView.hasHorizontalScroller = false
  185. self.initUIProperty()
  186. self.updateBackAndNextButtonState()
  187. }
  188. private func initUIProperty() {
  189. self.titleLabel.textColor = NSColor.titleColor()
  190. self.titleLabel.font = .SFProTextSemibold(16)
  191. self.leftBox.fillColor = NSColor(hex: "#F7F8FA")
  192. self.prePDFView?.backgroundColor = NSColor(hex: "#F7F8FA")
  193. self.numberBox.layer?.borderColor = NSColor(hex: "#DFE1E5").cgColor
  194. self.perLabel.textColor = NSColor.titleColor()
  195. self.perLabel.font = .SFProTextRegular(12)
  196. self.totalNumberLabel.textColor = NSColor.titleColor()
  197. self.totalNumberLabel.font = .SFProTextRegular(12)
  198. self.batchButton.setTitleColor(NSColor.buttonTitleColor())
  199. self.batchButton.layer?.borderColor = NSColor.buttonBorderColor().cgColor
  200. self.batchButton.font = .SFProTextRegular(14)
  201. }
  202. @objc func backButtonAction() {
  203. if let _ = self.prePDFView?.canGoToPreviousPage() {
  204. self.currentPageIndex -= 1
  205. self.prePDFView?.goToPreviousPage(nil)
  206. self.numberTextField.stringValue = "\(self.prePDFView!.currentPageIndex+1)"
  207. self.updateBackAndNextButtonState()
  208. }
  209. }
  210. @objc func nextButtonAction() {
  211. if let _ = self.prePDFView?.canGoToNextPage() {
  212. self.currentPageIndex += 1
  213. self.prePDFView?.goToNextPage(nil)
  214. self.numberTextField.stringValue = "\(self.prePDFView!.currentPageIndex+1)"
  215. self.updateBackAndNextButtonState()
  216. }
  217. }
  218. @objc func batchButtonAction() {
  219. guard let callback = itemClick else {
  220. return
  221. }
  222. callback(1)
  223. }
  224. @objc func cancelButtonAction() {
  225. guard let callback = itemClick else {
  226. return
  227. }
  228. callback(2)
  229. }
  230. @objc func convertButtonAction() {
  231. self.window?.makeFirstResponder(nil)
  232. if (self.pageRange == .custom) {
  233. let array = self.findSelectPage(pageRangeString: self.pageRangeString)
  234. if (array.count <= 0) {
  235. self.window?.makeFirstResponder(self.settingView.pageRangeItemView?.pageSizeVC.textField)
  236. return
  237. }
  238. }
  239. var pages: [Int] = []
  240. let filePath = self.getConvertFileSavePath()
  241. if (FileManager.default.fileExists(atPath: filePath)) {
  242. try?FileManager.default.removeItem(atPath: filePath)
  243. }
  244. if ((self.prePDFView?.document?.writeDecrypt(to: URL(fileURLWithPath: filePath)))!) {
  245. for i in 0 ..< self.prePDFView!.document.pageCount {
  246. pages.append(Int(i)+1)
  247. }
  248. }
  249. if (pages.count <= 0) {
  250. let alert = NSAlert()
  251. alert.alertStyle = .warning
  252. alert.messageText = NSLocalizedString("Invalid page range or the page number is out of range. Please try again.", comment: "")
  253. alert.runModal()
  254. return
  255. }
  256. NSPanel.savePanel(self.window!) { panel in
  257. let name = self.prePDFView!.document.documentURL.deletingPathExtension().lastPathComponent
  258. panel.nameFieldStringValue = name
  259. panel.allowedFileTypes = [self.fileExtension]
  260. } completion: { response, url in
  261. if (response == .cancel) {
  262. return
  263. }
  264. let outputFolderPath = url!.deletingLastPathComponent().path
  265. let convert = KMPDFConvert()
  266. self.convert = convert
  267. convert.password = self.documentModel.password
  268. convert.outputFolderPath = outputFolderPath
  269. convert.filePath = filePath
  270. convert.pages = pages
  271. var fileName = url!.deletingPathExtension().lastPathComponent
  272. if (self.fileExtension.isEmpty) {
  273. fileName = url!.lastPathComponent
  274. }
  275. // if (fileName.isEmpty) {
  276. // fileName = NSLocalizedString("Untitled", comment: "")
  277. // }
  278. convert.outputFileName = fileName
  279. self.convertModelAppendParams(convert: convert)
  280. DispatchQueue.main.async {
  281. self.showProgressWindow()
  282. self.progressController?.maxValue = Double(convert.pages.count)
  283. }
  284. DispatchQueue.global().async {
  285. KMPDFConvertManager.defaultManager.convert(convert: convert, progress: { index in
  286. DispatchQueue.main.async {
  287. self.progressController?.increment(by: 1.0)
  288. }
  289. }) { [unowned self] finished, error in
  290. self.hiddenProgressWindow()
  291. // 清除临时文件
  292. if (FileManager.default.fileExists(atPath: filePath)) {
  293. try?FileManager.default.removeItem(atPath: filePath)
  294. }
  295. if finished {
  296. cancelButtonAction()
  297. if FileManager.default.fileExists(atPath: convert.outputFilePath) {
  298. NSWorkspace.shared.activateFileViewerSelecting([URL(fileURLWithPath: convert.outputFilePath)])
  299. }
  300. } else {
  301. var errorString = ""
  302. let myError: NSError? = error as? NSError
  303. if myError?.code == 1 {
  304. errorString = NSLocalizedString("Password required or incorrect password. Please re-enter your password and try again", comment: "")
  305. } else if myError?.code == 2 {
  306. errorString = NSLocalizedString("The license doesn't allow the permission", comment: "")
  307. } else if myError?.code == 3 {
  308. errorString = NSLocalizedString("Malloc failure", comment: "")
  309. } else if myError?.code == 4 {
  310. errorString = NSLocalizedString("Unknown error in processing conversion. Please try again later", comment: "")
  311. } else if myError?.code == 5 {
  312. errorString = NSLocalizedString("Unknown error in processing PDF. Please try again later", comment: "")
  313. } else if myError?.code == 6 {
  314. errorString = NSLocalizedString("File not found or could not be opened. Check if your file exists or choose another file to convert", comment: "")
  315. } else if myError?.code == 7 {
  316. errorString = NSLocalizedString("File not in PDF format or corruptead. Change a PDF file and try again", comment: "")
  317. } else if myError?.code == 8 {
  318. errorString = NSLocalizedString("Unsupported security scheme", comment: "")
  319. } else if myError?.code == 9 {
  320. errorString = NSLocalizedString("Page not found or content error", comment: "")
  321. } else {
  322. errorString = NSLocalizedString("Table not found", comment: "")
  323. }
  324. let alert = NSAlert()
  325. alert.alertStyle = .critical
  326. alert.messageText = NSLocalizedString("Conversion Failed", comment: "")
  327. alert.informativeText = errorString
  328. alert.addButton(withTitle: NSLocalizedString("OK", comment: ""))
  329. alert.runModal()
  330. }
  331. }
  332. }
  333. }
  334. }
  335. func convertModelAppendParams(convert: KMPDFConvert) -> () {
  336. }
  337. func getConvertFileSavePath() -> String {
  338. var path = NSSearchPathForDirectoriesInDomains(FileManager.SearchPathDirectory.applicationSupportDirectory, FileManager.SearchPathDomainMask.userDomainMask, true).last
  339. path?.append("/")
  340. path?.append(Bundle.main.bundleIdentifier!)
  341. if (FileManager.default.fileExists(atPath: path!) == false) {
  342. try?FileManager.default.createDirectory(atPath: path!, withIntermediateDirectories: false)
  343. }
  344. path?.append("/")
  345. path?.append("convert.pdf")
  346. return path!
  347. }
  348. func isValidPagesString(pagesString: String)-> Bool {
  349. var valid = false
  350. for ch in pagesString {
  351. if ch != "0" && ch != "1" && ch != "2" && ch != "3" && ch != "4" && ch != "5" && ch != "6" && ch != "7" && ch != "8" && ch != "9" && ch != "," && ch != "-" {
  352. valid = false
  353. break
  354. } else {
  355. valid = true
  356. }
  357. }
  358. return valid
  359. }
  360. func findSelectPage(pageRangeString: String) -> ([Int]) {
  361. if !isValidPagesString(pagesString: pageRangeString) {
  362. return []
  363. }
  364. var result: [Int] = []
  365. let array = pageRangeString.components(separatedBy: ",")
  366. for string in array {
  367. if string.isEmpty {
  368. return []
  369. } else {
  370. let pages = string .components(separatedBy: "-")
  371. if pages.count > 2 {
  372. return []
  373. } else if pages.count == 1 {
  374. let page = pages[0]
  375. if page.isEmpty || Int(page)! > documentModel.pageCount || Int(page)! == 0 {
  376. return []
  377. } else {
  378. var hasSame: Bool = false
  379. for i in result {
  380. if i == Int(page)! {
  381. hasSame = true
  382. return []
  383. }
  384. }
  385. if !hasSame {
  386. result.append(Int(page)!)
  387. }
  388. }
  389. } else if pages.count == 2 {
  390. let page1 = pages[0]
  391. let page2 = pages[1]
  392. if page1.isEmpty || page2.isEmpty || Int(page1)! >= Int(page2)! || Int(page2)! > documentModel.pageCount || Int(page1)! == 0 {
  393. return []
  394. } else {
  395. var hasSame: Bool = false
  396. for i in Int(page1)! ... Int(page2)! {
  397. for j in result {
  398. if j == i {
  399. hasSame = true
  400. return []
  401. }
  402. }
  403. }
  404. if !hasSame {
  405. for i in Int(page1)! ... Int(page2)! {
  406. result.append(i)
  407. }
  408. }
  409. }
  410. }
  411. }
  412. }
  413. return result
  414. }
  415. /// 开始加载loading
  416. func beginLoading() {
  417. DispatchQueue.main.async { [self] in
  418. self.maskView = NSButton()
  419. self.maskView.isBordered = false
  420. self.maskView.title = ""
  421. self.maskView.wantsLayer = true
  422. self.maskView.layer?.backgroundColor = NSColor(white: 1, alpha: 0.3 ).cgColor
  423. self.maskView.target = self
  424. self.maskView.action = #selector(maskViewClick)
  425. self.maskView.frame = (self.window?.contentView!.bounds)!
  426. self.maskView.autoresizingMask = NSView.AutoresizingMask(rawValue: 18)
  427. self.window?.contentView?.addSubview(self.maskView)
  428. self.indicator = NSProgressIndicator()
  429. self.indicator.style = .spinning
  430. self.indicator.controlSize = .regular
  431. self.indicator.isIndeterminate = true
  432. self.window?.contentView?.addSubview(self.indicator)
  433. let indicatorSize: CGFloat = 32
  434. self.indicator.frame = NSMakeRect(0.5*(NSWidth((self.window?.contentView!.frame)!)-indicatorSize), 0.5*(NSHeight((self.window?.contentView!.frame)!)-indicatorSize), indicatorSize, indicatorSize)
  435. self.maskView.autoresizingMask = NSView.AutoresizingMask(rawValue: 45)
  436. self.indicator.startAnimation(nil)
  437. }
  438. }
  439. /// 结束加载loading
  440. func endLoading() {
  441. DispatchQueue.main.async {
  442. self.indicator.stopAnimation(nil)
  443. self.indicator.removeFromSuperview()
  444. self.indicator = nil
  445. self.maskView.removeFromSuperview()
  446. self.maskView = nil
  447. }
  448. }
  449. @objc func maskViewClick() {
  450. }
  451. // MARK: -
  452. // MARK: Private Methods
  453. private func updateBackAndNextButtonState() {
  454. var nextEnabled = false
  455. if let result = self.prePDFView?.canGoToNextPage() {
  456. nextEnabled = result
  457. }
  458. var backEnabled = false
  459. if let result = self.prePDFView?.canGoToPreviousPage() {
  460. backEnabled = result
  461. }
  462. if (Thread.isMainThread) {
  463. self.backButtonVC.state = backEnabled ? .Norm : .Disabled
  464. self.nextButtonVC.state = nextEnabled ? .Norm : .Disabled
  465. } else {
  466. DispatchQueue.main.async {
  467. self.backButtonVC.state = backEnabled ? .Norm : .Disabled
  468. self.nextButtonVC.state = nextEnabled ? .Norm : .Disabled
  469. }
  470. }
  471. }
  472. internal func updatePreView(pageRange: KMPageRange, pageString: String = "") {
  473. let pages = self.getPages(pageRange, pageString: pageString)
  474. if (pages == nil || pages!.isEmpty) {
  475. let alert = NSAlert()
  476. alert.alertStyle = .warning
  477. alert.messageText = NSLocalizedString("Invalid page range or the page number is out of range. Please try again.", comment: "")
  478. alert.runModal()
  479. return
  480. }
  481. var indexs = IndexSet()
  482. for i in 0 ..< self.prePDFView!.document.pageCount {
  483. indexs.insert(IndexSet.Element(i))
  484. }
  485. self.prePDFView?.document.removePage(at: indexs)
  486. #if DEBUG
  487. let pageIndexs = self.getPageIndexs(pageRange, pageString: pageString)
  488. if (pageIndexs != nil) {
  489. for i in pageIndexs! {
  490. print(i)
  491. }
  492. }
  493. #endif
  494. // self.prePDFView.document?.importPages(pageIndexs!, from: self.documentModel.document, at: 0)
  495. for page in pages! {
  496. self.prePDFView?.document.insertPageObject(page, at: self.prePDFView!.document.pageCount)
  497. }
  498. DispatchQueue.main.async {
  499. self.prePDFView?.layoutDocumentView()
  500. self.prePDFView?.setNeedsDisplayForVisiblePages()
  501. self.prePDFView?.goToFirstPage(nil)
  502. let number: Int = Int(self.prePDFView!.document!.pageCount)
  503. self.totalNumberLabel.stringValue = "\(number)"
  504. self.currentPageIndex = 1
  505. self.numberTextField.stringValue = "1"
  506. self.updateBackAndNextButtonState()
  507. }
  508. }
  509. private func getPageIndexs(_ pageRange: KMPageRange, pageString: String = "") -> IndexSet? {
  510. if (self.documentModel.pageCount <= 0) {
  511. return nil
  512. }
  513. let pageCount = self.documentModel.pageCount
  514. var indexs = IndexSet()
  515. if (pageRange == .all) {
  516. for i in 0 ..< pageCount {
  517. indexs.insert(i)
  518. }
  519. } else if (pageRange == .odd) {
  520. for i in 0 ..< pageCount {
  521. if ((i % 2) == 1) {
  522. continue
  523. }
  524. indexs.insert(i)
  525. }
  526. } else if (pageRange == .even) {
  527. for i in 0 ..< pageCount {
  528. if ((i % 2) == 0) {
  529. continue
  530. }
  531. indexs.insert(i)
  532. }
  533. } else if (pageRange == .current) {
  534. if (self.currentPageIndex >= 1 && self.currentPageIndex <= pageCount) {
  535. indexs.insert(self.currentPageIndex-1)
  536. }
  537. } else if (pageRange == .custom) {
  538. let array = self.findSelectPage(pageRangeString: pageString)
  539. if array.count == 0 {
  540. } else {
  541. for i in 0 ..< array.count {
  542. let index = array[i]
  543. if (index > pageCount || index < 1) {
  544. continue
  545. }
  546. indexs.insert(index-1)
  547. }
  548. }
  549. }
  550. return indexs
  551. }
  552. private func getPages(_ pageRange: KMPageRange, pageString: String = "") -> [CPDFPage]? {
  553. if (self.documentModel.pageCount <= 0) {
  554. return nil
  555. }
  556. let pageCount = self.documentModel.pageCount
  557. var pages: [CPDFPage] = []
  558. if (pageRange == .all) {
  559. for i in 0 ..< pageCount {
  560. let page = self.documentModel.document?.page(at: UInt(i))
  561. if (page == nil) {
  562. continue
  563. }
  564. pages.append(page!)
  565. }
  566. } else if (pageRange == .odd) {
  567. for i in 0 ..< pageCount {
  568. if ((i % 2) == 1) {
  569. continue
  570. }
  571. let page = self.documentModel.document?.page(at: UInt(i))
  572. if (page == nil) {
  573. continue
  574. }
  575. pages.append(page!)
  576. }
  577. } else if (pageRange == .even) {
  578. for i in 0 ..< pageCount {
  579. if ((i % 2) == 0) {
  580. continue
  581. }
  582. let page = self.documentModel.document?.page(at: UInt(i))
  583. if (page == nil) {
  584. continue
  585. }
  586. pages.append(page!)
  587. }
  588. } else if (pageRange == .current) {
  589. let page = self.prePDFView?.currentPage()
  590. if (page != nil) {
  591. pages.append(page!)
  592. }
  593. } else if (pageRange == .custom) {
  594. let array = self.findSelectPage(pageRangeString: pageString)
  595. if array.count == 0 {
  596. } else {
  597. for i in 0 ..< array.count {
  598. let index = array[i]
  599. if (index > pageCount || index < 1) {
  600. continue
  601. }
  602. let page = self.documentModel.document?.page(at: UInt(index-1))
  603. if (page == nil) {
  604. continue
  605. }
  606. pages.append(page!)
  607. }
  608. }
  609. }
  610. return pages
  611. }
  612. // MARK: Progress
  613. func showProgressWindow() {
  614. let progress = SKProgressController()
  615. progress.message = NSLocalizedString("Converting...", comment: "")
  616. progress.window?.backgroundColor = NSColor(hex: "#36383B")
  617. progress.window?.contentView?.wantsLayer = true
  618. progress.window?.contentView?.layer?.backgroundColor = NSColor(hex: "#36383B").cgColor
  619. progress.progressField.textColor = NSColor.white
  620. progress.closeBlock = { [weak self] in
  621. if (self!.convert != nil) {
  622. KMPDFConvertManager.defaultManager.cancel(convert: self!.convert!)
  623. }
  624. }
  625. self.progressController = progress
  626. self.window?.beginSheet(progress.window!)
  627. }
  628. func hiddenProgressWindow() {
  629. if (self.progressController != nil) {
  630. self.window?.endSheet((self.progressController?.window)!)
  631. self.progressController = nil
  632. }
  633. }
  634. func getCurrentLanguage() -> String {
  635. // return Bundle.main.preferredLocalizations.first!
  636. let array: [String] = UserDefaults.standard.object(forKey: "AppleLanguages") as! [String]
  637. return array.first!
  638. }
  639. /// 存储用户的选择的语言
  640. func saveLanugageSelectedIndex(index: Int) {
  641. UserDefaults.standard.setValue("\(index)", forKey: kKMConvertLanugageSelectedIndex)
  642. UserDefaults.standard.synchronize()
  643. }
  644. }
  645. extension KMConvertBaseWindowController: NSTextFieldDelegate {
  646. func controlTextDidChange(_ obj: Notification) {
  647. if numberTextField.isEqual(to: obj.object) {
  648. let textField: NSTextField = obj.object as! NSTextField
  649. var value: String = ""
  650. for ch in textField.stringValue {
  651. if ch != "0" && ch != "1" && ch != "2" && ch != "3" && ch != "4" && ch != "5" && ch != "6" && ch != "7" && ch != "8" && ch != "9" {
  652. } else {
  653. if value.isEmpty && ch == "0" {
  654. } else {
  655. value.append(ch)
  656. }
  657. }
  658. }
  659. if value.isEmpty {
  660. value.append("1")
  661. } else {
  662. if Int(value)! <= 0 {
  663. value = "1"
  664. } else if Int(value)! > self.prePDFView!.document!.pageCount {
  665. let number: Int = Int(self.prePDFView!.document!.pageCount)
  666. value = "\(number)"
  667. }
  668. }
  669. numberTextField.stringValue = value
  670. }
  671. }
  672. func controlTextDidEndEditing(_ obj: Notification) {
  673. if numberTextField.isEqual(to: obj.object) {
  674. if (numberTextField.stringValue.isEmpty) {
  675. numberTextField.stringValue = "1"
  676. self.prePDFView?.go(toPageIndex: 0, animated: false)
  677. self.currentPageIndex = 1
  678. self.updateBackAndNextButtonState()
  679. return
  680. }
  681. let number: Int = Int(numberTextField.stringValue)!
  682. if number > 0 && number <= self.prePDFView!.document!.pageCount {
  683. guard let page = self.prePDFView?.document?.page(at: UInt(number-1)) else {
  684. return
  685. }
  686. self.prePDFView?.go(to: page)
  687. currentPageIndex = number
  688. self.updateBackAndNextButtonState()
  689. }
  690. }
  691. }
  692. }
  693. extension KMConvertBaseWindowController: CPDFViewDelegate {
  694. func pdfViewCurrentPageDidChanged(_ pdfView: CPDFView!) {
  695. let currentIndex = pdfView.currentPageIndex
  696. currentPageIndex = currentIndex + 1
  697. numberTextField.stringValue = "\(currentIndex+1)"
  698. self.updateBackAndNextButtonState()
  699. }
  700. }