LabelToColorMap.h

Go to the documentation of this file.

#pragma once

#include <functional>

namespace CortidQCT {

template <class Scalar, class Label>
using LabelToColorMap = std::function<std::array<Scalar, 3>(Label)>;

namespace LabelToColorMaps {

template <class Scalar, class Label>
inline std::array<Scalar, 3> defaultMap(Label label) {
  auto const convertedLabel = static_cast<std::uint64_t>(label);

  // Ensure the upper 5 bytes are all zero
  if ((convertedLabel & 0xffffffffff000000) != 0) {
    throw std::invalid_argument("Given label cannot losslessly be converted to "
                                "24-bit unsigned integer");
  }

  auto const redScaled = convertedLabel & 0xff;
  auto const greenScaled = (convertedLabel & 0xff00) >> 8;
  auto const blueScaled = (convertedLabel & 0xff0000) >> 16;

  auto const red = static_cast<Scalar>(redScaled) / Scalar{255};
  auto const green = static_cast<Scalar>(greenScaled) / Scalar{255};
  auto const blue = static_cast<Scalar>(blueScaled) / Scalar{255};

  return std::array<Scalar, 3>{{red, green, blue}};
}

} // namespace LabelToColorMaps

} // namespace CortidQCT