1 /*

     2  * Copyright (C) 2006, 2007, 2008 Apple Inc. All rights reserved.

     3  *

     4  * Redistribution and use in source and binary forms, with or without

     5  * modification, are permitted provided that the following conditions

     6  * are met:

     7  * 1. Redistributions of source code must retain the above copyright

     8  *    notice, this list of conditions and the following disclaimer.

     9  * 2. Redistributions in binary form must reproduce the above copyright

    10  *    notice, this list of conditions and the following disclaimer in the

    11  *    documentation and/or other materials provided with the distribution.

    12  *

    13  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY

    14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

    15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

    16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR

    17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,

    18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,

    19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR

    20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY

    21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

    22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE

    23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 

    24  */

    25 

    26 #include "config.h"

    27 #include "FrameWin.h"

    28 

    29 #include <windows.h>

    30 

    31 #include "BitmapInfo.h"

    32 #include "FrameView.h"

    33 #include "GraphicsContext.h"

    34 #include "Settings.h"

    35 

    36 #include <CoreGraphics/CoreGraphics.h>

    37 

    38 using std::min;

    39 

    40 namespace WebCore {

    41 

    42 static void drawRectIntoContext(IntRect rect, FrameView* view, GraphicsContext* gc)

    43 {

    44     IntSize offset = view->scrollOffset();

    45     rect.move(-offset.width(), -offset.height());

    46     rect = view->convertToContainingWindow(rect);

    47 

    48     gc->concatCTM(AffineTransform().translate(-rect.x(), -rect.y()));

    49 

    50     view->paint(gc, rect);

    51 }

    52 

    53 static HBITMAP imageFromRect(const Frame* frame, IntRect& ir)

    54 {

    55     PaintBehavior oldPaintBehavior = frame->view()->paintBehavior();

    56     frame->view()->setPaintBehavior(oldPaintBehavior | PaintBehaviorFlattenCompositingLayers);

    57 

    58     void* bits;

    59     HDC hdc = CreateCompatibleDC(0);

    60     int w = ir.width();

    61     int h = ir.height();

    62     BitmapInfo bmp = BitmapInfo::create(IntSize(w, h));

    63 

    64     HBITMAP hbmp = CreateDIBSection(0, &bmp, DIB_RGB_COLORS, static_cast<void**>(&bits), 0, 0);

    65     HBITMAP hbmpOld = static_cast<HBITMAP>(SelectObject(hdc, hbmp));

    66     CGColorSpaceRef deviceRGB = CGColorSpaceCreateDeviceRGB();

    67     CGContextRef context = CGBitmapContextCreate(static_cast<void*>(bits), w, h,

    68         8, w * sizeof(RGBQUAD), deviceRGB, kCGBitmapByteOrder32Little | kCGImageAlphaPremultipliedFirst);

    69     CGColorSpaceRelease(deviceRGB);

    70     CGContextSaveGState(context);

    71 

    72     GraphicsContext gc(context);

    73 

    74     drawRectIntoContext(ir, frame->view(), &gc);

    75 

    76     CGContextRelease(context);

    77     SelectObject(hdc, hbmpOld);

    78     DeleteDC(hdc);

    79 

    80     frame->view()->setPaintBehavior(oldPaintBehavior);

    81 

    82     return hbmp;

    83 }

    84 

    85 HBITMAP imageFromSelection(Frame* frame, bool forceBlackText)

    86 {

    87     frame->document()->updateLayout();

    88 

    89     frame->view()->setPaintBehavior(PaintBehaviorSelectionOnly | (forceBlackText ? PaintBehaviorForceBlackText : 0));

    90     FloatRect fr = frame->selectionBounds();

    91     IntRect ir(static_cast<int>(fr.x()), static_cast<int>(fr.y()),

    92                static_cast<int>(fr.width()), static_cast<int>(fr.height()));

    93     HBITMAP image = imageFromRect(frame, ir);

    94     frame->view()->setPaintBehavior(PaintBehaviorNormal);

    95     return image;

    96 }

    97 

    98 HBITMAP Frame::nodeImage(Node* node) const

    99 {

   100     RenderObject* renderer = node->renderer();

   101     if (!renderer)

   102         return 0;

   103 

   104     IntRect topLevelRect;

   105     IntRect paintingRect = renderer->paintingRootRect(topLevelRect);

   106 

   107     document()->updateLayout();

   108 

   109     m_view->setNodeToDraw(node); // invoke special sub-tree drawing mode

   110     HBITMAP result = imageFromRect(this, paintingRect);

   111     m_view->setNodeToDraw(0);

   112 

   113     return result;

   114 }

   115 

   116 } // namespace WebCore