1/*
2 * Copyright 2015 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 SkPM_DEFINED
9#define SkPM_DEFINED
10
11#include "SkTypes.h"
12#include "SkColor.h"
13#include "SkColorPriv.h"
14#include "SkNx.h"
15
16// A pre-multiplied color storing each component in the same order as SkPMColor,
17// but as a float in the range [0, 255].
18class SkPMFloat : public Sk4f {
19public:
20    static SkPMFloat FromPMColor(SkPMColor c) { return SkPMFloat(c); }
21    static SkPMFloat FromARGB(float a, float r, float g, float b) { return SkPMFloat(a,r,g,b); }
22
23    // May be more efficient than one at a time.  No special alignment assumed for SkPMColors.
24    static void From4PMColors(const SkPMColor[4], SkPMFloat*, SkPMFloat*, SkPMFloat*, SkPMFloat*);
25
26    // Uninitialized.
27    SkPMFloat() {}
28    explicit SkPMFloat(SkPMColor);
29    SkPMFloat(float a, float r, float g, float b)
30    #ifdef SK_PMCOLOR_IS_RGBA
31        : INHERITED(r,g,b,a) {}
32    #else
33        : INHERITED(b,g,r,a) {}
34    #endif
35
36    SkPMFloat(const Sk4f& fs) : INHERITED(fs) {}
37
38    float a() const { return this->kth<SK_A32_SHIFT / 8>(); }
39    float r() const { return this->kth<SK_R32_SHIFT / 8>(); }
40    float g() const { return this->kth<SK_G32_SHIFT / 8>(); }
41    float b() const { return this->kth<SK_B32_SHIFT / 8>(); }
42
43    // N.B. All methods returning an SkPMColor call SkPMColorAssert on that result before returning.
44
45    // round() and roundClamp() round component values to the nearest integer.
46    SkPMColor round() const;  // Assumes all values in [0, 255].  Some implementations may clamp.
47    SkPMColor roundClamp() const;  // Will clamp all values to [0, 255].
48
49    // Like round(), but truncates instead of rounding.
50    // The domain of this function is (-1.0f, 256.0f).  Values in (-1.0f, 0.0f] trunc to a zero.
51    SkPMColor trunc() const;
52
53    // 4-at-a-time versions of round() and roundClamp(). Like From4PMColors(), no alignment assumed.
54    static void RoundTo4PMColors(
55            const SkPMFloat&, const SkPMFloat&, const SkPMFloat&, const SkPMFloat&, SkPMColor[4]);
56    static void RoundClampTo4PMColors(
57            const SkPMFloat&, const SkPMFloat&, const SkPMFloat&, const SkPMFloat&, SkPMColor[4]);
58
59    bool isValid() const {
60        return this->a() >= 0 && this->a() <= 255
61            && this->r() >= 0 && this->r() <= this->a()
62            && this->g() >= 0 && this->g() <= this->a()
63            && this->b() >= 0 && this->b() <= this->a();
64    }
65
66private:
67    typedef Sk4f INHERITED;
68};
69
70#ifdef SKNX_NO_SIMD
71    // Platform implementations of SkPMFloat assume Sk4f uses SSE or NEON.  _none is generic.
72    #include "../opts/SkPMFloat_none.h"
73#else
74    #if SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_SSSE3
75        #include "../opts/SkPMFloat_SSSE3.h"
76    #elif SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_SSE2
77        #include "../opts/SkPMFloat_SSE2.h"
78    #elif defined(SK_ARM_HAS_NEON)
79        #include "../opts/SkPMFloat_neon.h"
80    #else
81        #include "../opts/SkPMFloat_none.h"
82    #endif
83#endif
84
85#endif//SkPM_DEFINED
86