blob: 527de91c5055153f0bd065db3c07bd12561c7e75 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
/*
* Copyright 2016 Google Inc.
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#ifndef GrGLSLColorSpaceXformHelper_DEFINED
#define GrGLSLColorSpaceXformHelper_DEFINED
#include "GrColorSpaceXform.h"
#include "GrGLSLUniformHandler.h"
/**
* Helper class to assist with using GrColorSpaceXform within an FP. This manages all of the
* uniforms needed, and can be passed to shader builder functions to automatically generate the
* correct color space transformation code.
*/
class GrGLSLColorSpaceXformHelper : public SkNoncopyable {
public:
GrGLSLColorSpaceXformHelper() : fFlags(0) {}
void emitCode(GrGLSLUniformHandler* uniformHandler, const GrColorSpaceXform* colorSpaceXform,
uint32_t visibility = kFragment_GrShaderFlag) {
SkASSERT(uniformHandler);
if (colorSpaceXform) {
fFlags = colorSpaceXform->fFlags;
if (this->applyGamutXform()) {
fGamutXformVar = uniformHandler->addUniform(visibility,
kHalf4x4_GrSLType,
"ColorXform");
}
if (this->applyTransferFn()) {
fTransferFnVar = uniformHandler->addUniformArray(visibility,
kHalf_GrSLType,
"TransferFn",
kNumTransferFnCoeffs);
}
}
}
void setData(const GrGLSLProgramDataManager& pdman, const GrColorSpaceXform* colorSpaceXform) {
if (this->applyGamutXform()) {
pdman.setSkMatrix44(fGamutXformVar, colorSpaceXform->gamutXform());
}
if (this->applyTransferFn()) {
pdman.set1fv(fTransferFnVar, kNumTransferFnCoeffs, colorSpaceXform->transferFnCoeffs());
}
}
bool isValid() const { return (0 != fFlags); }
bool applyInverseSRGB() const {
return SkToBool(fFlags & GrColorSpaceXform::kApplyInverseSRGB_Flag);
}
bool applyTransferFn() const {
return SkToBool(fFlags & GrColorSpaceXform::kApplyTransferFn_Flag);
}
bool applyGamutXform() const {
return SkToBool(fFlags & GrColorSpaceXform::kApplyGamutXform_Flag);
}
GrGLSLProgramDataManager::UniformHandle gamutXformUniform() const { return fGamutXformVar; }
GrGLSLProgramDataManager::UniformHandle transferFnUniform() const { return fTransferFnVar; }
private:
static const int kNumTransferFnCoeffs = 7;
GrGLSLProgramDataManager::UniformHandle fGamutXformVar;
GrGLSLProgramDataManager::UniformHandle fTransferFnVar;
uint32_t fFlags;
};
#endif
|