KMBotaTableView.swift 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384
  1. //
  2. // KMTableView.swift
  3. // PDF Master
  4. //
  5. // Created by tangchao on 2023/11/16.
  6. //
  7. import Cocoa
  8. /*
  9. @protocol SKTableViewDelegate <NSTableViewDelegate>
  10. - (id <SKImageToolTipContext>)tableView:(NSTableView *)aTableView imageContextForRow:(NSInteger)rowIndex;
  11. - (NSArray *)tableView:(NSTableView *)aTableView typeSelectHelperSelectionStrings:(SKTypeSelectHelper *)aTypeSelectHelper;
  12. - (void)tableView:(NSTableView *)aTableView typeSelectHelper:(SKTypeSelectHelper *)aTypeSelectHelper didFailToFindMatchForSearchString:(NSString *)searchString;
  13. - (void)tableView:(NSTableView *)aTableView typeSelectHelper:(SKTypeSelectHelper *)aTypeSelectHelper updateSearchString:(NSString *)searchString;
  14. @end
  15. */
  16. @objc protocol KMBotaTableViewDelegate: NSTableViewDelegate {
  17. @objc optional func tableView(_ aTableView: NSTableView, deleteRowsWithIndexes rowIndexes: IndexSet)
  18. @objc optional func tableView(_ aTableView: NSTableView, canDeleteRowsWithIndexes rowIndexes: IndexSet) -> Bool
  19. @objc optional func tableView(_ aTableView: NSTableView, copyRowsWithIndexes rowIndexes: IndexSet)
  20. @objc optional func tableView(_ aTableView: NSTableView, canCopyRowsWithIndexes rowIndexes: IndexSet) -> Bool
  21. @objc optional func tableView(_ aTableView: NSTableView, pasteFromPasteboard pboard: NSPasteboard)
  22. @objc optional func tableView(_ aTableView: NSTableView, canPasteFromPasteboard pboard: NSPasteboard) -> Bool
  23. @objc optional func tableViewMoveLeft(_ aTableView: NSTableView)
  24. @objc optional func tableViewMoveRight(_ aTableView: NSTableView)
  25. }
  26. class KMBotaTableView: NSTableView {
  27. /*
  28. @interface SKTableView : NSTableView <SKTypeSelectDelegate> {
  29. NSMutableSet *trackingAreas;
  30. SKTypeSelectHelper *typeSelectHelper;
  31. BOOL supportsQuickLook;
  32. }
  33. @property (nonatomic, readonly) BOOL canDelete, canCopy, canPaste;
  34. @property (nonatomic) BOOL hasImageToolTips, supportsQuickLook;
  35. @property (nonatomic, retain) SKTypeSelectHelper *typeSelectHelper;
  36. - (void)delete:(id)sender;
  37. - (void)copy:(id)sender;
  38. - (void)paste:(id)sender;
  39. - (void)scrollToBeginningOfDocument:(id)sender;
  40. - (void)scrollToEndOfDocument:(id)sender;
  41. - (void)moveLeft:(id)sender;
  42. - (void)moveRight:(id)sender;
  43. - (id <SKTableViewDelegate>)delegate;
  44. - (void)setDelegate:(id <SKTableViewDelegate>)newDelegate;
  45. @end
  46. */
  47. override func draw(_ dirtyRect: NSRect) {
  48. super.draw(dirtyRect)
  49. // Drawing code here.
  50. }
  51. // override var delegate: NSTableViewDelegate? {
  52. // get {
  53. //
  54. // }
  55. // set {
  56. //
  57. // }
  58. // }
  59. /*
  60. @interface SKTableView (SKPrivate)
  61. - (void)rebuildTrackingAreas;
  62. @end
  63. @implementation SKTableView
  64. @synthesize typeSelectHelper, supportsQuickLook;
  65. @dynamic canDelete, canCopy, canPaste, hasImageToolTips;
  66. - (void)dealloc {
  67. [typeSelectHelper setDelegate:nil];
  68. SKDESTROY(typeSelectHelper);
  69. SKDESTROY(trackingAreas);
  70. [super dealloc];
  71. }
  72. - (void)setTypeSelectHelper:(SKTypeSelectHelper *)newTypeSelectHelper {
  73. if (typeSelectHelper != newTypeSelectHelper) {
  74. if ([typeSelectHelper delegate] == self)
  75. [typeSelectHelper setDelegate:nil];
  76. [typeSelectHelper release];
  77. typeSelectHelper = [newTypeSelectHelper retain];
  78. [typeSelectHelper setDelegate:self];
  79. }
  80. }
  81. - (void)reloadData {
  82. [super reloadData];
  83. [self rebuildTrackingAreas];
  84. [typeSelectHelper rebuildTypeSelectSearchCache];
  85. }
  86. - (void)keyDown:(NSEvent *)theEvent {
  87. unichar eventChar = [theEvent firstCharacter];
  88. NSUInteger modifierFlags = [theEvent deviceIndependentModifierFlags];
  89. if ((eventChar == NSNewlineCharacter || eventChar == NSEnterCharacter || eventChar == NSCarriageReturnCharacter) && modifierFlags == 0) {
  90. if ([self doubleAction] == NULL || [self sendAction:[self doubleAction] to:[self target]] == NO)
  91. NSBeep();
  92. } else if ((eventChar == NSDeleteCharacter || eventChar == NSDeleteFunctionKey) && modifierFlags == 0 && [self canDelete]) {
  93. [self delete:self];
  94. } else if ((eventChar == SKSpaceCharacter) && modifierFlags == 0) {
  95. if (supportsQuickLook == NO)
  96. [[self enclosingScrollView] pageDown:nil];
  97. else if ([QLPreviewPanel sharedPreviewPanelExists] && [[QLPreviewPanel sharedPreviewPanel] isVisible])
  98. [[QLPreviewPanel sharedPreviewPanel] orderOut:nil];
  99. else
  100. [[QLPreviewPanel sharedPreviewPanel] makeKeyAndOrderFront:nil];
  101. } else if ((eventChar == SKSpaceCharacter) && modifierFlags == NSShiftKeyMask) {
  102. if (supportsQuickLook == NO)
  103. [[self enclosingScrollView] pageUp:nil];
  104. } else if (eventChar == NSHomeFunctionKey && (modifierFlags & ~NSFunctionKeyMask) == 0) {
  105. [self scrollToBeginningOfDocument:nil];
  106. } else if (eventChar == NSEndFunctionKey && (modifierFlags & ~NSFunctionKeyMask) == 0) {
  107. [self scrollToEndOfDocument:nil];
  108. } else if (eventChar == NSLeftArrowFunctionKey && modifierFlags == 0) {
  109. [self moveLeft:nil];
  110. } else if (eventChar == NSRightArrowFunctionKey && modifierFlags == 0) {
  111. [self moveRight:nil];
  112. } else if ([typeSelectHelper handleEvent:theEvent] == NO) {
  113. [super keyDown:theEvent];
  114. }
  115. }
  116. - (void)scrollToBeginningOfDocument:(id)sender {
  117. if ([self numberOfRows])
  118. [self scrollRowToVisible:0];
  119. }
  120. - (void)scrollToEndOfDocument:(id)sender {
  121. if ([self numberOfRows])
  122. [self scrollRowToVisible:[self numberOfRows] - 1];
  123. }
  124. - (void)moveLeft:(id)sender {
  125. if ([[self delegate] respondsToSelector:@selector(tableViewMoveLeft:)])
  126. [[self delegate] tableViewMoveLeft:self];
  127. }
  128. - (void)moveRight:(id)sender {
  129. if ([[self delegate] respondsToSelector:@selector(tableViewMoveRight:)])
  130. [[self delegate] tableViewMoveRight:self];
  131. }
  132. - (BOOL)canDelete {
  133. NSIndexSet *indexes = [self selectedRowIndexes];
  134. if ([indexes count] && [[self delegate] respondsToSelector:@selector(tableView:deleteRowsWithIndexes:)]) {
  135. if ([[self delegate] respondsToSelector:@selector(tableView:canDeleteRowsWithIndexes:)])
  136. return [[self delegate] tableView:self canDeleteRowsWithIndexes:indexes];
  137. else
  138. return YES;
  139. }
  140. return NO;
  141. }
  142. - (void)delete:(id)sender {
  143. if ([self canDelete])
  144. [[self delegate] tableView:self deleteRowsWithIndexes:[self selectedRowIndexes]];
  145. else
  146. NSBeep();
  147. }
  148. - (BOOL)canCopy {
  149. NSIndexSet *indexes = [self selectedRowIndexes];
  150. if ([indexes count] && [[self delegate] respondsToSelector:@selector(tableView:copyRowsWithIndexes:)]) {
  151. if ([[self delegate] respondsToSelector:@selector(tableView:canCopyRowsWithIndexes:)])
  152. return [[self delegate] tableView:self canCopyRowsWithIndexes:indexes];
  153. else
  154. return YES;
  155. }
  156. return NO;
  157. }
  158. - (void)copy:(id)sender {
  159. if ([self canCopy])
  160. [[self delegate] tableView:self copyRowsWithIndexes:[self selectedRowIndexes]];
  161. else
  162. NSBeep();
  163. }
  164. - (BOOL)canPaste {
  165. if ([[self delegate] respondsToSelector:@selector(tableView:pasteFromPasteboard:)]) {
  166. if ([[self delegate] respondsToSelector:@selector(tableView:canPasteFromPasteboard:)])
  167. return [[self delegate] tableView:self canPasteFromPasteboard:[NSPasteboard generalPasteboard]];
  168. else
  169. return YES;
  170. }
  171. return NO;
  172. }
  173. - (void)paste:(id)sender {
  174. if ([self canPaste])
  175. [[self delegate] tableView:self pasteFromPasteboard:[NSPasteboard generalPasteboard]];
  176. else
  177. NSBeep();
  178. }
  179. - (BOOL)validateMenuItem:(NSMenuItem *)menuItem {
  180. if ([menuItem action] == @selector(delete:))
  181. return [self canDelete];
  182. else if ([menuItem action] == @selector(copy:))
  183. return [self canCopy];
  184. else if ([menuItem action] == @selector(paste:))
  185. return [self canPaste];
  186. else if ([menuItem action] == @selector(selectAll:))
  187. return [self allowsMultipleSelection];
  188. else if ([menuItem action] == @selector(deselectAll:))
  189. return [self allowsEmptySelection];
  190. else if ([[SKTableView superclass] instancesRespondToSelector:@selector(validateMenuItem:)])
  191. return [super validateMenuItem:menuItem];
  192. return YES;
  193. }
  194. - (NSFont *)font {
  195. for (NSTableColumn *tc in [self tableColumns]) {
  196. NSCell *cell = [tc dataCell];
  197. if ([cell type] == NSTextCellType)
  198. return [cell font];
  199. }
  200. return nil;
  201. }
  202. - (void)setFont:(NSFont *)font {
  203. for (NSTableColumn *tc in [self tableColumns]) {
  204. NSCell *cell = [tc dataCell];
  205. if ([cell type] == NSTextCellType)
  206. [cell setFont:font];
  207. }
  208. [self setRowHeight:[font defaultViewLineHeight]];
  209. [self noteHeightOfRowsWithIndexesChanged:[NSIndexSet indexSetWithIndexesInRange:NSMakeRange(0, [self numberOfRows])]];
  210. }
  211. - (NSImage *)dragImageForRowsWithIndexes:(NSIndexSet *)dragRows tableColumns:(NSArray *)tableColumns event:(NSEvent *)dragEvent offset:(NSPointPointer)dragImageOffset{
  212. return [super dragImageForRowsWithIndexes:dragRows tableColumns:[[self tableColumns] subarrayWithRange:NSMakeRange(0, 1)] event:dragEvent offset:dragImageOffset];
  213. }
  214. #pragma mark Tracking
  215. - (void)removeTrackingAreas {
  216. if (trackingAreas == nil)
  217. return;
  218. for (NSTrackingArea *area in trackingAreas)
  219. [self removeTrackingArea:area];
  220. [trackingAreas removeAllObjects];
  221. }
  222. - (void)addTrackingAreaForRow:(NSInteger)row {
  223. if (trackingAreas == nil)
  224. return;
  225. NSDictionary *userInfo = [[NSDictionary alloc] initWithObjectsAndKeys:[NSNumber numberWithInteger:row], @"row", nil];
  226. NSTrackingArea *area = [[NSTrackingArea alloc] initWithRect:[self rectOfRow:row] options:NSTrackingMouseEnteredAndExited | NSTrackingActiveInActiveApp owner:self userInfo:userInfo];
  227. [self addTrackingArea:area];
  228. [trackingAreas addObject:area];
  229. [area release];
  230. [userInfo release];
  231. }
  232. - (void)rebuildTrackingAreas {
  233. if (trackingAreas == nil || [[self delegate] respondsToSelector:@selector(tableView:imageContextForRow:)] == NO)
  234. return;
  235. [self removeTrackingAreas];
  236. if ([self window]) {
  237. NSRect visibleRect = [self visibleRect];
  238. NSRange rowRange = [self rowsInRect:visibleRect];
  239. NSUInteger row;
  240. for (row = rowRange.location; row < NSMaxRange(rowRange); row++)
  241. [self addTrackingAreaForRow:row];
  242. }
  243. }
  244. - (void)updateTrackingAreas {
  245. [super updateTrackingAreas];
  246. [self rebuildTrackingAreas];
  247. }
  248. - (void)noteNumberOfRowsChanged {
  249. [super noteNumberOfRowsChanged];
  250. [self rebuildTrackingAreas];
  251. }
  252. - (void)mouseEntered:(NSEvent *)theEvent{
  253. if (trackingAreas == nil)
  254. return;
  255. NSDictionary *userInfo = [theEvent userData];
  256. NSNumber *rowNumber = [userInfo objectForKey:@"row"];
  257. if (rowNumber) {
  258. id <SKImageToolTipContext> context = [[self delegate] tableView:self imageContextForRow:[rowNumber integerValue]];
  259. if (context)
  260. [[SKImageToolTipWindow sharedToolTipWindow] showForImageContext:context atPoint:NSZeroPoint];
  261. }
  262. }
  263. - (void)mouseExited:(NSEvent *)theEvent{
  264. if (trackingAreas == nil)
  265. return;
  266. NSDictionary *userInfo = [theEvent userData];
  267. if ([userInfo objectForKey:@"row"])
  268. [[SKImageToolTipWindow sharedToolTipWindow] fadeOut];
  269. }
  270. - (BOOL)hasImageToolTips {
  271. return trackingAreas != nil;
  272. }
  273. - (void)setHasImageToolTips:(BOOL)flag {
  274. if (flag && trackingAreas == nil) {
  275. trackingAreas = [[NSMutableSet alloc] init];
  276. if ([self window])
  277. [self rebuildTrackingAreas];
  278. } else if (flag == NO && trackingAreas) {
  279. if ([self window])
  280. [self removeTrackingAreas];
  281. SKDESTROY(trackingAreas);
  282. }
  283. }
  284. #pragma mark SKTypeSelectHelper datasource protocol
  285. - (NSArray *)typeSelectHelperSelectionStrings:(SKTypeSelectHelper *)aTypeSelectHelper {
  286. if ([[self delegate] respondsToSelector:@selector(tableView:typeSelectHelperSelectionStrings:)])
  287. return [[self delegate] tableView:self typeSelectHelperSelectionStrings:aTypeSelectHelper];
  288. return nil;
  289. }
  290. - (NSUInteger)typeSelectHelperCurrentlySelectedIndex:(SKTypeSelectHelper *)aTypeSelectHelper {
  291. return [[self selectedRowIndexes] lastIndex];
  292. }
  293. - (void)typeSelectHelper:(SKTypeSelectHelper *)aTypeSelectHelper selectItemAtIndex:(NSUInteger)itemIndex {
  294. [self selectRowIndexes:[NSIndexSet indexSetWithIndex:itemIndex] byExtendingSelection:NO];
  295. [self scrollRowToVisible:itemIndex];
  296. }
  297. - (void)typeSelectHelper:(SKTypeSelectHelper *)aTypeSelectHelper didFailToFindMatchForSearchString:(NSString *)searchString {
  298. if ([[self delegate] respondsToSelector:@selector(tableView:typeSelectHelper:didFailToFindMatchForSearchString:)])
  299. [[self delegate] tableView:self typeSelectHelper:aTypeSelectHelper didFailToFindMatchForSearchString:searchString];
  300. }
  301. - (void)typeSelectHelper:(SKTypeSelectHelper *)aTypeSelectHelper updateSearchString:(NSString *)searchString {
  302. if ([[self delegate] respondsToSelector:@selector(tableView:typeSelectHelper:updateSearchString:)])
  303. [[self delegate] tableView:self typeSelectHelper:aTypeSelectHelper updateSearchString:searchString];
  304. }
  305. - (id <SKTableViewDelegate>)delegate {
  306. return (id <SKTableViewDelegate>)[super delegate];
  307. }
  308. - (void)setDelegate:(id <SKTableViewDelegate>)newDelegate {
  309. [super setDelegate:newDelegate];
  310. [self rebuildTrackingAreas];
  311. }
  312. @end
  313. */
  314. }