size --> cellSize adjustment

chrome/browser/cocoa/bookmark_button_cell.mm

 // Copyright (c) 2009 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #import "chrome/browser/cocoa/bookmark_button_cell.h"
 
 #include "app/l10n_util_mac.h"
 #include "base/logging.h"
+#include "base/nsimage_cache_mac.h"
 #include "base/sys_string_conversions.h"
 #import "chrome/browser/bookmarks/bookmark_model.h"
 #import "chrome/browser/cocoa/bookmark_menu.h"
+#import "chrome/browser/cocoa/bookmark_button.h"
 #include "grit/generated_resources.h"
 
 
 @interface BookmarkButtonCell(Private)
 - (void)configureBookmarkButtonCell;
 @end
 
 
 @implementation BookmarkButtonCell
 
 @synthesize startingChildIndex = startingChildIndex_;
+@synthesize drawFolderArrow = drawFolderArrow_;
 
 + (id)buttonCellForNode:(const BookmarkNode*)node
             contextMenu:(NSMenu*)contextMenu
                cellText:(NSString*)cellText
               cellImage:(NSImage*)cellImage {
   id buttonCell =
       [[[BookmarkButtonCell alloc] initForNode:node
                                    contextMenu:contextMenu
                                       cellText:cellText
                                      cellImage:cellImage]
@@ skipping 13 matching lines @@
     if (node) {
       NSString* title = base::SysWideToNSString(node->GetTitle());
       [self setBookmarkCellText:title image:cellImage];
       [self setMenu:contextMenu];
     } else {
       [self setEmpty:YES];
       [self setBookmarkCellText:l10n_util::GetNSString(IDS_MENU_EMPTY_SUBMENU)
                           image:nil];
     }
   }
+
   return self;
 }
 
 - (id)initTextCell:(NSString*)string {
   return [self initForNode:nil contextMenu:nil cellText:string cellImage:nil];
 }
 
 // Used by the off-the-side menu, the only case where a
 // BookmarkButtonCell is loaded from a nib.
 - (void)awakeFromNib {
@@ skipping 103 matching lines @@
   [super mouseEntered:event];
   [[self controlView] mouseEntered:event];
 }
 
 // See comment above mouseEntered:, above.
 - (void)mouseExited:(NSEvent*)event {
   [super mouseExited:event];
   [[self controlView] mouseExited:event];
 }
 
+- (void)setDrawFolderArrow:(BOOL)draw {
+  drawFolderArrow_ = draw;
+  if (draw && !arrowImage_) {
+    arrowImage_.reset([nsimage_cache::ImageNamed(@"menu_hierarchy_arrow.pdf")
+                                    retain]);
+  }
+}
+
+// Add extra size for the arrow so it doesn't overlap the text.
+// Does not sanity check to be sure this is actually a folder node.
+- (NSSize)cellSize {
+  NSSize cellSize = [super cellSize];
+  if (drawFolderArrow_) {
+    cellSize.width += [arrowImage_ size].width;  // plus margin?
+  }
+  return cellSize;
+}
+
+// Override cell drawing to add a submenu arrow like a real menu.
+- (void)drawInteriorWithFrame:(NSRect)cellFrame inView:(NSView*)controlView {
+  // First draw "everything else".
+  [super drawInteriorWithFrame:cellFrame inView:controlView];
+
+  // If asked to do so, and if a folder, draw the arrow.
+  if (!drawFolderArrow_)
+    return;
+  BookmarkButton* button = static_cast<BookmarkButton*>([self controlView]);
+  DCHECK([button respondsToSelector:@selector(isFolder)]);
+  if ([button isFolder]) {
+    NSRect imageRect = NSZeroRect;
+    imageRect.size = [arrowImage_ size];
+    NSRect drawRect = NSOffsetRect(imageRect,
+                                   NSWidth(cellFrame) - NSWidth(imageRect),
+                                   (NSHeight(cellFrame) / 2.0) -
+                                   (NSHeight(imageRect) / 2.0));
+    [arrowImage_ setFlipped:[controlView isFlipped]];
+    [arrowImage_ drawInRect:drawRect
+                    fromRect:imageRect
+                   operation:NSCompositeSourceOver
+                    fraction:[self isEnabled] ? 1.0 : 0.5];
+  }
+}
+
 @end