KMBatesController.swift 7.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225
  1. //
  2. // KMBatesController.swift
  3. // PDF Reader Pro
  4. //
  5. // Created by Niehaoyu on 2024/11/8.
  6. //
  7. import Cocoa
  8. import KMComponentLibrary
  9. @objc protocol KMBatesControllerDelegate: AnyObject {
  10. @objc optional func kmBatesControllerDidUpdateMode(_ view: KMBatesController)
  11. //数据发生变化时处理
  12. @objc optional func kmBatesControllerDidModelDataUpdated(_ view: KMBatesController)
  13. }
  14. class KMBatesController: NSViewController {
  15. @IBOutlet var contendView: NSView!
  16. @IBOutlet var preInfoView: NSView!
  17. @IBOutlet var propertyInfoView: NSView!
  18. var documentPreview: KMEditDocumentController = KMEditDocumentController.init()
  19. var propertyController: KMBatesPropertyController = KMBatesPropertyController.init()
  20. var templatesController: KMBatesTemplateController = KMBatesTemplateController()
  21. var saveTemplateMessage: ComponentMessage = ComponentMessage()
  22. weak open var delegate: KMBatesControllerDelegate?
  23. var totalPDFCount: Int = 1
  24. var pdfDocument: CPDFDocument?
  25. var batesModel: KMBatesModel?
  26. var editSubType: KMPDFEditSubModeType = .none
  27. override func viewDidLoad() {
  28. super.viewDidLoad()
  29. // Do view setup here.
  30. documentPreview.view.frame = preInfoView.bounds
  31. documentPreview.view.autoresizingMask = [.width, .height]
  32. preInfoView.addSubview(documentPreview.view)
  33. propertyController.view.frame = propertyInfoView.bounds
  34. propertyController.view.autoresizingMask = [.width, .height]
  35. propertyController.delegate = self
  36. propertyController.view.isHidden = true
  37. propertyInfoView.addSubview(propertyController.view)
  38. templatesController.view.frame = propertyInfoView.bounds
  39. templatesController.view.autoresizingMask = [.width, .height]
  40. templatesController.delegate = self
  41. templatesController.view.isHidden = true
  42. propertyInfoView.addSubview(templatesController.view)
  43. setupProperty()
  44. }
  45. func setupProperty() {
  46. saveTemplateMessage.properties = ComponentMessageProperty(messageType: .success, title: KMLocalizedString("Saved Template"))
  47. }
  48. func resetUI() {
  49. propertyController.view.isHidden = true
  50. templatesController.view.isHidden = true
  51. if editSubType == .template {
  52. templatesController.view.isHidden = false
  53. templatesController.reloadData()
  54. } else if editSubType == .add {
  55. propertyController.view.isHidden = false
  56. propertyController.editSubType = editSubType
  57. propertyController.reloadData()
  58. } else if editSubType == .edit {
  59. propertyController.view.isHidden = false
  60. propertyController.editSubType = editSubType
  61. propertyController.reloadData()
  62. }
  63. }
  64. func reloadData() {
  65. resetUI()
  66. propertyController.totalPDFCount = totalPDFCount
  67. propertyController.reloadData()
  68. documentPreview.pdfDocument = pdfDocument
  69. documentPreview.reloadData()
  70. }
  71. func updatePDFDocumentBates() {
  72. if let data = batesModel {
  73. let cPDFBates = documentPreview.pdfDocument?.bates()
  74. cPDFBates?.pageString = "0"
  75. var fontSize = data.fontsize
  76. var fontName: String = data.fontName
  77. let font = NSFont.boldSystemFont(ofSize:fontSize)
  78. let style = NSMutableParagraphStyle()
  79. style.alignment = .center
  80. style.lineBreakMode = .byCharWrapping
  81. let size: NSSize = "text".boundingRect(with: NSSize(width: 1000, height: 1000), options: NSString.DrawingOptions(rawValue: 3), attributes: [NSAttributedString.Key.font : font, NSAttributedString.Key.paragraphStyle : style]).size
  82. cPDFBates?.margin = NSEdgeInsetsMake(max(CGFloat(data.topMargin)-size.height, 0), CGFloat(data.leftMargin), max(CGFloat(data.bottomMargin)-size.height, 0), CGFloat(data.rightMargin))
  83. let strings = KMBatesManager.parseModel(model: data, UInt(totalPDFCount))
  84. var count: Int = 0
  85. let color = data.color
  86. for text in strings {
  87. cPDFBates?.setText(text, at: UInt(count))
  88. cPDFBates?.setTextColor(color, at: UInt(count))
  89. cPDFBates?.setFontSize(fontSize, at: UInt(count))
  90. cPDFBates?.setFontName(fontName, at: UInt(count))
  91. count += 1
  92. }
  93. cPDFBates?.update()
  94. self.documentPreview.refreshUI()
  95. } else {
  96. pdfDocument?.bates().clear()
  97. documentPreview.refreshUI()
  98. }
  99. }
  100. }
  101. //MARK: - KMHeaderPropertyControllerDelegate
  102. extension KMBatesController: KMBatesPropertyControllerDelegate {
  103. func batesPropertyControllerDidUpdate(_ controller: KMBatesPropertyController) {
  104. batesModel = controller.batesModel
  105. updatePDFDocumentBates()
  106. }
  107. func batesPropertyControllerDidChangetoTemplate(_ controller: KMBatesPropertyController) {
  108. editSubType = .template
  109. resetUI()
  110. batesModel = nil
  111. updatePDFDocumentBates()
  112. reloadData()
  113. delegate?.kmBatesControllerDidUpdateMode?(self)
  114. }
  115. func batesPropertyControllerSaveTemplateSuccess(_ controller: KMBatesPropertyController, _ data: KMBatesModel) {
  116. editSubType = .template
  117. reloadData()
  118. delegate?.kmBatesControllerDidUpdateMode?(self)
  119. }
  120. func batesPropertyControllerCancelTemplateEdit(_ controller: KMBatesPropertyController) {
  121. editSubType = .template
  122. reloadData()
  123. delegate?.kmBatesControllerDidUpdateMode?(self)
  124. }
  125. func batesPropertyControllerFinishTemplateEdit(_ controller: KMBatesPropertyController) {
  126. editSubType = .template
  127. reloadData()
  128. delegate?.kmBatesControllerDidUpdateMode?(self)
  129. }
  130. }
  131. //MARK: - KMHFTemplateControllerDelegate
  132. extension KMBatesController: KMBatesTemplateControllerDelegate {
  133. func templateControllerDidSelectedChanged(_ controller: KMBatesTemplateController) {
  134. batesModel = controller.selectedDataModel
  135. updatePDFDocumentBates()
  136. delegate?.kmBatesControllerDidModelDataUpdated?(self)
  137. }
  138. func templateControllerDidAddData(_ controller: KMBatesTemplateController) {
  139. editSubType = .add
  140. resetUI()
  141. propertyController.clearData()
  142. propertyController.batesModel = KMBatesModel()
  143. reloadData()
  144. delegate?.kmBatesControllerDidUpdateMode?(self)
  145. delegate?.kmBatesControllerDidModelDataUpdated?(self)
  146. }
  147. func templateControllerDidEditData(_ controller: KMBatesTemplateController, _ data: KMBatesModel) {
  148. editSubType = .edit
  149. propertyController.batesModel = data
  150. propertyController.originalDataDict = KMBatesManager.defaultManager.parseModel(model: data)
  151. resetUI()
  152. delegate?.kmBatesControllerDidUpdateMode?(self)
  153. delegate?.kmBatesControllerDidModelDataUpdated?(self)
  154. }
  155. }