src/third_party/WebKit/Source/web/mac/WebSubstringUtil.mm

   1 /*
   2  * Copyright (C) 2005, 2007, 2008 Apple Inc. All rights reserved.
   3  * Copyright (C) 2011 Google Inc. All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions are
   7  * met:
   8  *
   9  *     * Redistributions of source code must retain the above copyright
  10  * notice, this list of conditions and the following disclaimer.
  11  *     * Redistributions in binary form must reproduce the above
  12  * copyright notice, this list of conditions and the following disclaimer
  13  * in the documentation and/or other materials provided with the
  14  * distribution.
  15  *     * Neither the name of Google Inc. nor the names of its
  16  * contributors may be used to endorse or promote products derived from
  17  * this software without specific prior written permission.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  20  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  21  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  22  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  23  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  24  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  25  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  26  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  27  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  28  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  29  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  30  */
  31
  32 #include "config.h"
  33 #include "public/web/mac/WebSubstringUtil.h"
  34
  35 #import <Cocoa/Cocoa.h>
  36
  37 #include "bindings/v8/ExceptionStatePlaceholder.h"
  38 #include "core/dom/Document.h"
  39 #include "core/dom/Element.h"
  40 #include "core/dom/Node.h"
  41 #include "core/dom/Range.h"
  42 #include "core/editing/FrameSelection.h"
  43 #include "core/editing/PlainTextRange.h"
  44 #include "core/editing/TextIterator.h"
  45 #include "core/frame/FrameView.h"
  46 #include "core/frame/LocalFrame.h"
  47 #include "core/html/HTMLElement.h"
  48 #include "core/rendering/HitTestResult.h"
  49 #include "core/rendering/RenderObject.h"
  50 #include "core/rendering/style/RenderStyle.h"
  51 #include "platform/fonts/Font.h"
  52 #include "platform/mac/ColorMac.h"
  53 #include "public/platform/WebRect.h"
  54 #include "public/web/WebHitTestResult.h"
  55 #include "public/web/WebRange.h"
  56 #include "public/web/WebView.h"
  57 #include "web/WebLocalFrameImpl.h"
  58
  59 using namespace WebCore;
  60
  61 static NSAttributedString* attributedSubstringFromRange(const Range* range)
  62 {
  63     NSMutableAttributedString* string = [[NSMutableAttributedString alloc] init];
  64     NSMutableDictionary* attrs = [NSMutableDictionary dictionary];
  65     size_t length = range->endOffset() - range->startOffset();
  66
  67     unsigned position = 0;
  68     for (TextIterator it(range); !it.atEnd() && [string length] < length; it.advance()) {
  69         unsigned numCharacters = it.length();
  70         if (!numCharacters)
  71             continue;
  72
  73         Node* container = it.range()->startContainer();
  74         RenderObject* renderer = container->renderer();
  75         ASSERT(renderer);
  76         if (!renderer)
  77             continue;
  78
  79         RenderStyle* style = renderer->style();
  80         NSFont* font = style->font().primaryFont()->getNSFont();
  81         // If the platform font can't be loaded, it's likely that the site is
  82         // using a web font. For now, just use the default font instead.
  83         // TODO(rsesek): Change the font activation flags to allow other processes
  84         // to use the font.
  85         if (!font)
  86           font = [NSFont systemFontOfSize:style->font().fontDescription().computedSize()];
  87         [attrs setObject:font forKey:NSFontAttributeName];
  88
  89         if (style->visitedDependentColor(CSSPropertyColor).alpha())
  90             [attrs setObject:nsColor(style->visitedDependentColor(CSSPropertyColor)) forKey:NSForegroundColorAttributeName];
  91         else
  92             [attrs removeObjectForKey:NSForegroundColorAttributeName];
  93         if (style->visitedDependentColor(CSSPropertyBackgroundColor).alpha())
  94             [attrs setObject:nsColor(style->visitedDependentColor(CSSPropertyBackgroundColor)) forKey:NSBackgroundColorAttributeName];
  95         else
  96             [attrs removeObjectForKey:NSBackgroundColorAttributeName];
  97
  98         Vector<UChar> characters;
  99         it.appendTextTo(characters);
 100         NSString* substring =
 101             [[[NSString alloc] initWithCharacters:characters.data()
 102                                            length:characters.size()] autorelease];
 103         [string replaceCharactersInRange:NSMakeRange(position, 0)
 104                               withString:substring];
 105         [string setAttributes:attrs range:NSMakeRange(position, numCharacters)];
 106         position += numCharacters;
 107     }
 108     return [string autorelease];
 109 }
 110
 111 namespace blink {
 112
 113 NSAttributedString* WebSubstringUtil::attributedWordAtPoint(WebView* view, WebPoint point, WebPoint& baselinePoint)
 114 {
 115     HitTestResult result = view->hitTestResultAt(point);
 116     if (!result.targetNode())
 117       return nil;
 118     LocalFrame* frame = result.targetNode()->document().frame();
 119     FrameView* frameView = frame->view();
 120
 121     RefPtrWillBeRawPtr<Range> range = frame->rangeForPoint(result.roundedPointInInnerNodeFrame());
 122     if (!range)
 123         return nil;
 124
 125     // Expand to word under point.
 126     VisibleSelection selection(range.get());
 127     selection.expandUsingGranularity(WordGranularity);
 128     RefPtrWillBeRawPtr<Range> wordRange = selection.toNormalizedRange();
 129
 130     // Convert to NSAttributedString.
 131     NSAttributedString* string = attributedSubstringFromRange(wordRange.get());
 132
 133     // Compute bottom left corner and convert to AppKit coordinates.
 134     IntRect stringRect = enclosingIntRect(wordRange->boundingRect());
 135     IntPoint stringPoint = stringRect.minXMaxYCorner();
 136     stringPoint.setY(frameView->height() - stringPoint.y());
 137
 138     // Adjust for the font's descender. AppKit wants the baseline point.
 139     if ([string length]) {
 140         NSDictionary* attributes = [string attributesAtIndex:0 effectiveRange:NULL];
 141         NSFont* font = [attributes objectForKey:NSFontAttributeName];
 142         if (font)
 143             stringPoint.move(0, ceil(-[font descender]));
 144     }
 145
 146     baselinePoint = stringPoint;
 147     return string;
 148 }
 149
 150 NSAttributedString* WebSubstringUtil::attributedSubstringInRange(WebLocalFrame* webFrame, size_t location, size_t length)
 151 {
 152     LocalFrame* frame = toWebLocalFrameImpl(webFrame)->frame();
 153     if (frame->view()->needsLayout())
 154         frame->view()->layout();
 155
 156     Element* editable = frame->selection().rootEditableElementOrDocumentElement();
 157     ASSERT(editable);
 158     RefPtrWillBeRawPtr<Range> range(PlainTextRange(location, location + length).createRange(*editable));
 159     if (!range)
 160         return nil;
 161
 162     return attributedSubstringFromRange(range.get());
 163 }
 164
 165 } // namespace blink