[Color] Make gradients work with ExtendedColors
[WebKit-https.git] / Source / WebCore / platform / graphics / win / GradientDirect2D.cpp
1 /*
2  * Copyright (C) 2016 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 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 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 "Gradient.h"
28
29 #include "FloatPoint.h"
30 #include "GraphicsContext.h"
31 #include <d2d1.h>
32 #include <wtf/RetainPtr.h>
33
34 #define GRADIENT_DRAWING 3
35
36 namespace WebCore {
37
38 void Gradient::platformDestroy()
39 {
40     if (m_gradient)
41         m_gradient->Release();
42     m_gradient = nullptr;
43 }
44
45 ID2D1Brush* Gradient::platformGradient()
46 {
47     ASSERT(m_gradient);
48     return m_gradient;
49 }
50
51 ID2D1Brush* Gradient::createPlatformGradientIfNecessary(ID2D1RenderTarget* context)
52 {
53     generateGradient(context);
54     return m_gradient;
55 }
56
57 void Gradient::generateGradient(ID2D1RenderTarget* renderTarget)
58 {
59     sortStopsIfNecessary();
60
61     Vector<D2D1_GRADIENT_STOP> gradientStops;
62     // FIXME: Add support for ExtendedColor.
63     for (auto stop : m_stops) {
64         float r;
65         float g;
66         float b;
67         float a;
68         stop.color.getRGBA(r, g, b, a);
69         gradientStops.append(D2D1::GradientStop(stop.offset, D2D1::ColorF(r, g, b, a)));
70     }
71
72     COMPtr<ID2D1GradientStopCollection> gradientStopCollection;
73     HRESULT hr = renderTarget->CreateGradientStopCollection(gradientStops.data(), gradientStops.size(), &gradientStopCollection);
74     RELEASE_ASSERT(SUCCEEDED(hr));
75
76     if (m_gradient) {
77         m_gradient->Release();
78         m_gradient = nullptr;
79     }
80
81     if (m_radial) {
82         FloatSize offset = p1() - p0();
83         ID2D1RadialGradientBrush* radialGradient = nullptr;
84         float radiusX = endRadius() + offset.width();
85         float radiusY = radiusX / m_aspectRatio;
86         hr = renderTarget->CreateRadialGradientBrush(
87             D2D1::RadialGradientBrushProperties(p0(), D2D1::Point2F(offset.width(), offset.height()), radiusX, radiusY),
88             D2D1::BrushProperties(), gradientStopCollection.get(),
89             &radialGradient);
90         RELEASE_ASSERT(SUCCEEDED(hr));
91         m_gradient = radialGradient;
92     } else {
93         ID2D1LinearGradientBrush* linearGradient = nullptr;
94         hr = renderTarget->CreateLinearGradientBrush(
95             D2D1::LinearGradientBrushProperties(p0(), p1()),
96             D2D1::BrushProperties(), gradientStopCollection.get(),
97             &linearGradient);
98         RELEASE_ASSERT(SUCCEEDED(hr));
99         m_gradient = linearGradient;
100     }
101
102     hash();
103 }
104
105 void Gradient::fill(GraphicsContext* context, const FloatRect& rect)
106 {
107     auto d2dContext = context->platformContext();
108
109     bool needScaling = aspectRatio() != 1;
110     if (needScaling) {
111         context->save();
112         // Scale from the center of the gradient. We only ever scale non-deprecated gradients,
113         // for which m_p0 == m_p1.
114         ASSERT(m_p0 == m_p1);
115
116         D2D1_MATRIX_3X2_F ctm = { };
117         d2dContext->GetTransform(&ctm);
118
119         AffineTransform transform(ctm);
120         transform.translate(m_p0);
121         transform.scaleNonUniform(1.0, 1.0 / aspectRatio());
122         transform.translate(-m_p0);
123
124         d2dContext->SetTransform(ctm);
125     }
126
127     if (!m_cachedHash || !m_gradient)
128         generateGradient(d2dContext);
129
130     d2dContext->SetTags(GRADIENT_DRAWING, __LINE__);
131
132     const D2D1_RECT_F d2dRect = rect;
133     d2dContext->FillRectangle(&d2dRect, m_gradient);
134
135     if (needScaling)
136         context->restore();
137 }
138
139 }