PBGitRevisionCell.m

   1 //
   2 //  PBGitRevisionCell.m
   3 //  GitX
   4 //
   5 //  Created by Pieter de Bie on 17-06-08.
   6 //  Copyright 2008 __MyCompanyName__. All rights reserved.
   7 //
   8
   9 #import "PBGitRevisionCell.h"
  10 #import "PBGitRef.h"
  11 #import "RoundedRectangle.h"
  12
  13 @implementation PBGitRevisionCell
  14
  15
  16 - (id) initWithCoder: (id) coder
  17 {
  18         self = [super initWithCoder:coder];
  19         textCell = [[NSTextFieldCell alloc] initWithCoder:coder];
  20         return self;
  21 }
  22
  23 - (NSArray*) colors
  24 {
  25         return  [NSArray arrayWithObjects:
  26                                 [NSColor colorWithCalibratedRed: 0X4e/256.0 green:0X9A/256.0 blue: 0X06/256.0 alpha: 1.0],
  27                                 [NSColor colorWithCalibratedRed: 0X20/256.0 green:0X4A/256.0 blue: 0X87/256.0 alpha: 1.0],
  28                                 [NSColor colorWithCalibratedRed: 0XC4/256.0 green:0XA0/256.0 blue: 0 alpha: 1.0],
  29                                 [NSColor colorWithCalibratedRed: 0X5C/256.0 green:0X35/256.0 blue: 0X66/256.0 alpha: 1.0],
  30                                 [NSColor colorWithCalibratedRed: 0XA4/256.0 green:0X00/256.0 blue: 0X00/256.0 alpha: 1.0],
  31                                 [NSColor colorWithCalibratedRed: 0XCE/256.0 green:0X5C/256.0 blue: 0 alpha: 1.0],
  32                                 nil];
  33 }
  34
  35 - (void) drawLineFromColumn: (int) from toColumn: (int) to inRect: (NSRect) r offset: (int) offset color: (int) c
  36 {
  37
  38         int columnWidth = 10;
  39         NSPoint origin = r.origin;
  40
  41         NSPoint source = NSMakePoint(origin.x + columnWidth* from, origin.y + offset);
  42         NSPoint center = NSMakePoint( origin.x + columnWidth * to, origin.y + r.size.height * 0.5 + 0.5);
  43
  44         // Just use red for now.
  45         NSArray* colors = [self colors];
  46         [[colors objectAtIndex: c % [colors count]] set];
  47
  48         NSBezierPath * path = [NSBezierPath bezierPath];
  49         [path setLineWidth:2];
  50
  51         [path moveToPoint: source];
  52         [path lineToPoint: center];
  53         [path stroke];
  54
  55 }
  56
  57 - (void) drawCircleInRect: (NSRect) r
  58 {
  59
  60         int c = cellInfo.position;
  61         int columnWidth = 10;
  62         NSPoint origin = r.origin;
  63         NSPoint columnOrigin = { origin.x + columnWidth * c, origin.y};
  64
  65         NSRect oval = { columnOrigin.x - 5, columnOrigin.y + r.size.height * 0.5 - 5, 10, 10};
  66
  67
  68         NSBezierPath * path = [NSBezierPath bezierPathWithOvalInRect:oval];
  69
  70         [[NSColor blackColor] set];
  71         [path fill];
  72
  73         NSRect smallOval = { columnOrigin.x - 3, columnOrigin.y + r.size.height * 0.5 - 3, 6, 6};
  74         [[NSColor whiteColor] set];
  75         path = [NSBezierPath bezierPathWithOvalInRect:smallOval];
  76         [path fill];
  77 }
  78
  79 - (void) drawTriangleInRect: (NSRect) r sign: (char) sign
  80 {
  81         int c = cellInfo.position;
  82         int columnHeight = 10;
  83         int columnWidth = 8;
  84
  85         NSPoint top;
  86         if (sign == '<')
  87                 top.x = round(r.origin.x) + 10 * c + 4;
  88         else {
  89                 top.x = round(r.origin.x) + 10 * c - 4;
  90                 columnWidth *= -1;
  91         }
  92         top.y = r.origin.y + (r.size.height - columnHeight) / 2;
  93
  94         NSBezierPath * path = [NSBezierPath bezierPath];
  95         // Start at top
  96         [path moveToPoint: NSMakePoint(top.x, top.y)];
  97         // Go down
  98         [path lineToPoint: NSMakePoint(top.x, top.y + columnHeight)];
  99         // Go left top
 100         [path lineToPoint: NSMakePoint(top.x - columnWidth, top.y + columnHeight / 2)];
 101         // Go to top again
 102         [path closePath];
 103
 104         [[NSColor whiteColor] set];
 105         [path fill];
 106         [[NSColor blackColor] set];
 107         [path setLineWidth: 2];
 108         [path stroke];
 109 }
 110
 111 - (NSMutableDictionary*) attributesForRefLabelSelected: (BOOL) selected
 112 {
 113         NSMutableDictionary *attributes = [[[NSMutableDictionary alloc] initWithCapacity:2] autorelease];
 114         NSMutableParagraphStyle* style = [[[NSParagraphStyle defaultParagraphStyle] mutableCopy] autorelease];
 115
 116         [style setAlignment:NSCenterTextAlignment];
 117         [attributes setObject:style forKey:NSParagraphStyleAttributeName];
 118         [attributes setObject:[NSFont fontWithName:@"Helvetica" size:9] forKey:NSFontAttributeName];
 119
 120         //if (selected)
 121         //      [attributes setObject:[NSColor alternateSelectedControlTextColor] forKey:NSForegroundColorAttributeName];
 122
 123         return attributes;
 124 }
 125
 126 - (NSColor*) colorForRef: (PBGitRef*) ref
 127 {
 128         BOOL isHEAD = [ref.ref isEqualToString:[[[controller repository] headRef] simpleRef]];
 129
 130         if (isHEAD)
 131                 return [NSColor colorWithCalibratedRed: 0Xfc/256.0 green:0Xa6/256.0 blue: 0X4f/256.0 alpha: 1.0];
 132
 133         NSString* type = [ref type];
 134         if ([type isEqualToString:@"head"])
 135                 return [NSColor colorWithCalibratedRed: 0Xaa/256.0 green:0Xf2/256.0 blue: 0X54/256.0 alpha: 1.0];
 136         else if ([type isEqualToString:@"remote"])
 137                 return [NSColor colorWithCalibratedRed: 0xb2/256.0 green:0Xdf/256.0 blue: 0Xff/256.0 alpha: 1.0];
 138         else if ([type isEqualToString:@"tag"])
 139                 return [NSColor colorWithCalibratedRed: 0Xfc/256.0 green:0Xed/256.0 blue: 0X4f/256.0 alpha: 1.0];
 140
 141         return [NSColor yellowColor];
 142 }
 143
 144 -(NSArray *)rectsForRefsinRect:(NSRect) rect;
 145 {
 146         NSMutableArray *array = [NSMutableArray array];
 147
 148         static const int ref_padding = 10;
 149         static const int ref_spacing = 2;
 150
 151         NSRect lastRect = rect;
 152         lastRect.origin.x = round(lastRect.origin.x) - 0.5;
 153         lastRect.origin.y = round(lastRect.origin.y) - 0.5;
 154
 155         for (PBGitRef *ref in self.objectValue.refs) {
 156                 NSMutableDictionary* attributes = [self attributesForRefLabelSelected:NO];
 157                 NSSize textSize = [[ref shortName] sizeWithAttributes:attributes];
 158
 159                 NSRect newRect = lastRect;
 160                 newRect.size.width = textSize.width + ref_padding;
 161                 newRect.size.height = textSize.height;
 162                 newRect.origin.y = rect.origin.y + (rect.size.height - newRect.size.height) / 2;
 163
 164                 [array addObject:[NSValue valueWithRect:newRect]];
 165                 lastRect = newRect;
 166                 lastRect.origin.x += (int)lastRect.size.width + ref_spacing;
 167         }
 168
 169         return array;
 170 }
 171
 172 - (void) drawLabelAtIndex:(int)index inRect:(NSRect)rect
 173 {
 174         NSArray *refs = self.objectValue.refs;
 175         PBGitRef *ref = [refs objectAtIndex:index];
 176
 177         NSMutableDictionary* attributes = [self attributesForRefLabelSelected:[self isHighlighted]];
 178         NSBezierPath *border = [NSBezierPath bezierPathWithRoundedRect:rect cornerRadius: 2.0];
 179         [[self colorForRef:ref] set];
 180         [border fill];
 181
 182         [[ref shortName] drawInRect:rect withAttributes:attributes];
 183         [border stroke];
 184 }
 185
 186 - (void) drawRefsInRect: (NSRect *)refRect
 187 {
 188         [[NSColor blackColor] setStroke];
 189
 190         NSRect lastRect;
 191         int index = 0;
 192         for (NSValue *rectValue in [self rectsForRefsinRect:*refRect])
 193         {
 194                 NSRect rect = [rectValue rectValue];
 195                 [self drawLabelAtIndex:index inRect:rect];
 196                 lastRect = rect;
 197                 ++index;
 198         }
 199         refRect->size.width -= lastRect.origin.x - refRect->origin.x + lastRect.size.width;
 200         refRect->origin.x    = lastRect.origin.x + lastRect.size.width;
 201 }
 202
 203 - (void) drawWithFrame: (NSRect) rect inView:(NSView *)view
 204 {
 205         cellInfo = [self.objectValue lineInfo];
 206
 207         if (cellInfo && ![controller hasNonlinearPath]) {
 208                 float pathWidth = 10 + 10 * cellInfo.numColumns;
 209
 210                 NSRect ownRect;
 211                 NSDivideRect(rect, &ownRect, &rect, pathWidth, NSMinXEdge);
 212
 213                 int i;
 214                 struct PBGitGraphLine *lines = cellInfo.lines;
 215                 for (i = 0; i < cellInfo.nLines; i++) {
 216                         if (lines[i].upper == 0)
 217                                 [self drawLineFromColumn: lines[i].from toColumn: lines[i].to inRect:ownRect offset: ownRect.size.height color: lines[i].colorIndex];
 218                         else
 219                                 [self drawLineFromColumn: lines[i].from toColumn: lines[i].to inRect:ownRect offset: 0 color:lines[i].colorIndex];
 220                 }
 221
 222                 if (cellInfo.sign == '<' || cellInfo.sign == '>')
 223                         [self drawTriangleInRect: ownRect sign: cellInfo.sign];
 224                 else
 225                         [self drawCircleInRect: ownRect];
 226         }
 227
 228
 229         if ([self.objectValue refs] && [[self.objectValue refs] count])
 230                 [self drawRefsInRect:&rect];
 231
 232         // Still use this superclass because of hilighting differences
 233         //_contents = [self.objectValue subject];
 234         //[super drawWithFrame:rect inView:view];
 235         [textCell setObjectValue: [self.objectValue subject]];
 236         [textCell setHighlighted: [self isHighlighted]];
 237         [textCell drawWithFrame:rect inView: view];
 238 }
 239
 240 - (void) setObjectValue: (PBGitCommit*)object {
 241         [super setObjectValue:[NSValue valueWithNonretainedObject:object]];
 242 }
 243
 244 - (PBGitCommit*) objectValue {
 245     return [[super objectValue] nonretainedObjectValue];
 246 }
 247
 248 - (int) indexAtX:(float)x
 249 {
 250         cellInfo = [self.objectValue lineInfo];
 251         float pathWidth = 0;
 252         if (cellInfo && ![controller hasNonlinearPath])
 253                 pathWidth = 10 + 10 * cellInfo.numColumns;
 254
 255         int index = 0;
 256         NSRect refRect = NSMakeRect(pathWidth, 0, 1000, 10000);
 257         for (NSValue *rectValue in [self rectsForRefsinRect:refRect])
 258         {
 259                 NSRect rect = [rectValue rectValue];
 260                 if (x >= rect.origin.x && x <= (rect.origin.x + rect.size.width))
 261                         return index;
 262                 ++index;
 263         }
 264
 265         return -1;
 266 }
 267
 268 - (NSRect) rectAtIndex:(int)index
 269 {
 270         cellInfo = [self.objectValue lineInfo];
 271         float pathWidth = 0;
 272         if (cellInfo && ![controller hasNonlinearPath])
 273                 pathWidth = 10 + 10 * cellInfo.numColumns;
 274         NSRect refRect = NSMakeRect(pathWidth, 0, 1000, 10000);
 275
 276         return [[[self rectsForRefsinRect:refRect] objectAtIndex:index] rectValue];
 277 }
 278
 279 # pragma mark context menu delegate methods
 280
 281 - (NSMenu *) menuForEvent:(NSEvent *)event inRect:(NSRect)rect ofView:(NSView *)view
 282 {
 283         if (!contextMenuDelegate)
 284                 return [self menu];
 285
 286         int i = [self indexAtX:[view convertPointFromBase:[event locationInWindow]].x];
 287         if (i < 0)
 288                 return [self menu];
 289
 290         id ref = [[[self objectValue] refs] objectAtIndex:i];
 291         if (!ref)
 292                 return [self menu];
 293
 294         NSArray *items = [contextMenuDelegate menuItemsForRef:ref commit:[self objectValue]];
 295         NSMenu *menu = [[NSMenu alloc] init];
 296         for (NSMenuItem *item in items)
 297                 [menu addItem:item];
 298         return menu;
 299
 300         return [self menu];
 301 }
 302 @end