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
14static inline Sk4f swizzle_rb(const Sk4f& x) {
15    return SkNx_shuffle<2, 1, 0, 3>(x);
16}
17
18static 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 */
29struct 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