#include <iostream>

#include "colorchecker.h"
#include "lut.h"
#include "piraw.h"
#include "util.h"

int main() {
  auto image = PiRaw2::FromJpeg(ReadFile("test.jpg"));
  std::cout << "Initial score: " << ScoreImage(*image) << std::endl;

  auto lut = MinimalLut3d::Identity();
  uint32_t diff = 1;
  while (diff) {
    diff = OptimizeLut<4>(*image, &lut);
    std::cout << "diff=" << diff << " error=" << ScoreImage(*lut.MapImage(*image)) << std::endl;
  }

  auto image2 = lut.MapImage(*image);
  WriteFile("test.png", HighlightClosest(*image2)->ToPng());
}