1 /*
2  * Copyright 2016 Google Inc.
3  *
4  * Use of this source code is governed by a BSD-style license that can be
5  * found in the LICENSE file.
6  */
7 
8 #ifndef SkPM4f_DEFINED
9 #define SkPM4f_DEFINED
10 
11 #include "SkColorData.h"
12 #include "SkNx.h"
13 
14 static inline Sk4f swizzle_rb(const Sk4f& x) {
15     return SkNx_shuffle<2, 1, 0, 3>(x);
16 }
17 
18 static inline Sk4f swizzle_rb_if_bgra(const Sk4f& x) {
19 #ifdef SK_PMCOLOR_IS_BGRA
20     return swizzle_rb(x);
21 #else
22     return x;
23 #endif
24 }
25 
26 /*
27  *  The float values are 0...1 premultiplied in RGBA order (regardless of SkPMColor order)
28  */
29 struct SK_API SkPM4f {
30     enum {
31         R, G, B, A,
32     };
33     float fVec[4];
34 
35     float r() const { return fVec[R]; }
36     float g() const { return fVec[G]; }
37     float b() const { return fVec[B]; }
38     float a() const { return fVec[A]; }
39 
40     static SkPM4f FromPremulRGBA(float r, float g, float b, float a) {
41         SkPM4f p;
42         p.fVec[R] = r;
43         p.fVec[G] = g;
44         p.fVec[B] = b;
45         p.fVec[A] = a;
46         return p;
47     }
48 
49     static SkPM4f From4f(const Sk4f& x) {
50         SkPM4f pm;
51         x.store(pm.fVec);
52         return pm;
53     }
54     static SkPM4f FromF16(const uint16_t[4]);
55     static SkPM4f FromPMColor(SkPMColor);
56 
57     Sk4f to4f() const { return Sk4f::Load(fVec); }
58     Sk4f to4f_rgba() const { return this->to4f(); }
59     Sk4f to4f_bgra() const { return swizzle_rb(this->to4f()); }
60     Sk4f to4f_pmorder() const { return swizzle_rb_if_bgra(this->to4f()); }
61 
62     SkPMColor toPMColor() const {
63         Sk4f value = swizzle_rb_if_bgra(this->to4f());
64         SkPMColor result;
65         SkNx_cast<uint8_t>(value * Sk4f(255) + Sk4f(0.5f)).store(&result);
66         return result;
67     }
68 
69     void toF16(uint16_t[4]) const;
70     uint64_t toF16() const; // 4 float16 values packed into uint64_t
71 
72     SkColor4f unpremul() const;
73 
74 #ifdef SK_DEBUG
75     void assertIsUnit() const;
76 #else
77     void assertIsUnit() const {}
78 #endif
79 };
80 
81 #endif
82