ColorTransform functions added to repo

libsrc/hyperion/CMakeLists.txt

@@ -13,7 +13,9 @@ add_library(hyperion
 		${CURRENT_SOURCE_DIR}/LedDeviceWs2801.cpp
 		${CURRENT_SOURCE_DIR}/LedString.cpp
 		${CURRENT_SOURCE_DIR}/Hyperion.cpp
-		${CURRENT_SOURCE_DIR}/ImageToLedsMap.cpp
+                ${CURRENT_SOURCE_DIR}/ImageToLedsMap.cpp
+                ${CURRENT_SOURCE_DIR}/ColorTransform.h
+                ${CURRENT_SOURCE_DIR}/ColorTransform.cpp
 )
 
 target_link_libraries(hyperion

libsrc/hyperion/ColorTransform.cpp

@@ -0,0 +1,102 @@
+#include <cmath>
+
+#include "ColorTransform.h"
+
+ColorTransform::ColorTransform() :
+    _threshold(0),
+    _gamma(1.0),
+    _blacklevel(0.0),
+    _whitelevel(1.0)
+{
+    initializeMapping();
+}
+
+ColorTransform::ColorTransform(double threshold, double gamma, double blacklevel, double whitelevel) :
+    _threshold(threshold),
+    _gamma(gamma),
+    _blacklevel(blacklevel),
+    _whitelevel(whitelevel)
+{
+    initializeMapping();
+}
+
+ColorTransform::~ColorTransform()
+{
+}
+
+double ColorTransform::getThreshold() const
+{
+    return _threshold;
+}
+
+void ColorTransform::setThreshold(double threshold)
+{
+    _threshold = threshold;
+    initializeMapping();
+}
+
+double ColorTransform::getGamma() const
+{
+    return _gamma;
+}
+
+void ColorTransform::setGamma(double gamma)
+{
+    _gamma = gamma;
+    initializeMapping();
+}
+
+double ColorTransform::getBlacklevel() const
+{
+    return _blacklevel;
+}
+
+void ColorTransform::setBlacklevel(double blacklevel)
+{
+    _blacklevel = blacklevel;
+    initializeMapping();
+}
+
+double ColorTransform::getWhitelevel() const
+{
+    return _whitelevel;
+}
+
+void ColorTransform::setWhitelevel(double whitelevel)
+{
+    _whitelevel = whitelevel;
+    initializeMapping();
+}
+
+void ColorTransform::initializeMapping()
+{
+    // initialize the mapping as a linear array
+    for (int i = 0; i < 256; ++i)
+    {
+        double output = i / 255.0;
+
+        // apply linear transform
+        if (output < _threshold)
+        {
+            output = 0.0;
+        }
+
+        // apply gamma correction
+        output = std::pow(output, _gamma);
+
+        // apply blacklevel and whitelevel
+        output = _blacklevel + (_whitelevel - _blacklevel) * output;
+
+        // calc mapping
+        int mappingValue = output * 255;
+        if (mappingValue < 0)
+        {
+            mappingValue = 0;
+        }
+        else if (mappingValue > 255)
+        {
+            mappingValue = 255;
+        }
+        _mapping[i] = mappingValue;
+    }
+}

libsrc/hyperion/ColorTransform.h

@@ -0,0 +1,49 @@
+#pragma once
+
+#include "stdint.h"
+
+/// Transform for a single color byte value
+///
+/// Transforms are applied in the following order:
+/// 1) a threshold is applied. All values below threshold will be set to zero
+/// 2) gamma color correction is applied
+/// 3) the output value is scaled from the [0:1] to [blacklevel:whitelevel]
+/// 4) finally, in case of a weird choice of parameters, the output is clamped between [0:1]
+///
+/// All configuration values are doubles and assume the color value to be between 0 and 1
+class ColorTransform
+{
+public:
+    ColorTransform();
+    ColorTransform(double threshold, double gamma, double whitelevel, double blacklevel);
+    ~ColorTransform();
+
+    double getThreshold() const;
+    void setThreshold(double threshold);
+
+    double getGamma() const;
+    void setGamma(double gamma);
+
+    double getBlacklevel() const;
+    void setBlacklevel(double blacklevel);
+
+    double getWhitelevel() const;
+    void setWhitelevel(double whitelevel);
+
+    /// get the transformed value for the given byte value
+    uint8_t transform(uint8_t input) const
+    {
+        return _mapping[input];
+    }
+
+private:
+    void initializeMapping();
+
+private:
+    double _threshold;
+    double _gamma;
+    double _blacklevel;
+    double _whitelevel;
+
+    uint8_t _mapping[256];
+};