misc/piphoto
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Lut base class, ScoreImage -> ScoreLut

Browse Source
This commit is contained in:
Ian Gulliver
2017-08-12 12:01:21 -07:00
parent ed6ae66171
commit 1838d8c62f
5 changed files with 41 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
lut.h
View File

@@ -3,17 +3,41 @@
#include "array.h"
#include "color.h"
#include "coord.h"
#include "image.h"
class Lut {
public:
Lut() = default;
Lut(const Lut&) = default;
virtual ~Lut();
virtual Color<3> MapColor(const Color<3>& in) const = 0;
template <int32_t X, int32_t Y, int32_t C>
std::unique_ptr<Image<X, Y, C>> MapImage(const Image<X, Y, C>& in) const;
};
template <int32_t X, int32_t Y, int32_t C>
std::unique_ptr<Image<X, Y, C>> Lut::MapImage(const Image<X, Y, C>& in) const {
auto out = std::make_unique<Image<X, Y, C>>();
for (int32_t y = 0; y < Y; ++y) {
for (int32_t x = 0; x < X; ++x) {
Coord<2> coord = {{{{x, y}}}};
out->SetPixel(coord, MapColor(in.GetPixel(coord)));
}
}
return out;
}
// Hardcoded to Color<3>, so color dimensions == LUT dimensions
template <int32_t X, int32_t Y, int32_t Z>
class Lut3d : public Array<Array<Array<Color<3>, X>, Y>, Z> {
class Lut3d : public Array<Array<Array<Color<3>, X>, Y>, Z>, public Lut {
public:
static Lut3d<X, Y, Z> Identity();
Color<3> MapColor(const Color<3>& in) const;
template <int32_t IMG_X, int32_t IMG_Y, int32_t C>
std::unique_ptr<Image<IMG_X, IMG_Y, C>> MapImage(const Image<IMG_X, IMG_Y, C>& in) const;
Color<3> MapColor(const Color<3>& in) const override;
private:
// Return value is (root_indices, remainders)
@@ -82,21 +106,6 @@ Color<3> Lut3d<X, Y, Z>::MapColor(const Color<3>& in) const {
return inter0.Interpolate(inter1, rem.at(2), BlockSize(Z)).Crop();
}
template <int32_t X, int32_t Y, int32_t Z>
template <int32_t IMG_X, int32_t IMG_Y, int32_t C>
std::unique_ptr<Image<IMG_X, IMG_Y, C>> Lut3d<X, Y, Z>::MapImage(const Image<IMG_X, IMG_Y, C>& in) const {
auto out = std::make_unique<Image<IMG_X, IMG_Y, C>>();
for (int32_t y = 0; y < IMG_Y; ++y) {
for (int32_t x = 0; x < IMG_X; ++x) {
Coord<2> coord = {{{{x, y}}}};
out->SetPixel(coord, MapColor(in.GetPixel(coord)));
}
}
return out;
}
template <int32_t X, int32_t Y, int32_t Z>
constexpr std::pair<Coord<3>, Coord<3>> Lut3d<X, Y, Z>::FindRoot(const Color<3>& in) {
auto root_x = FindChannelRoot(in.at(0), X);