jet.h

Go to the documentation of this file.

#ifndef TATOOINE_COLOR_SCALES_JET_H
#define TATOOINE_COLOR_SCALES_JET_H
//==============================================================================
#include <tatooine/concepts.h>
#include <tatooine/vec.h>
 
#include <memory>
//==============================================================================
namespace tatooine::color_scales {
//==============================================================================
template <floating_point Real>
struct jet  {
  using real_type  = Real;
  using this_type  = jet<Real>;
  using color_type = vec<Real, 3>;
  //==============================================================================
  static constexpr auto num_samples() -> std::size_t{return 7;}
  //==============================================================================
  std::unique_ptr<color_type[]> m_data;
  //==============================================================================
  jet()
      : m_data{new color_type[]{{0, 0, 0.5625},
                                {0, 0, 1},
                                {0, 1, 1},
                                {0.5, 1, 0.5},
                                {1, 1, 0},
                                {1, 0, 0},
                                {0.5, 0, 0}}} {}
  //----------------------------------------------------------------------------
  auto data() -> auto& { return m_data; }
  auto data() const -> auto const& { return m_data; }
  //----------------------------------------------------------------------------
  auto sample(real_type t) const {
    if (t <= 0) {
      return m_data[0];
    }
    if (t >= 1) {
      return m_data[num_samples() - 1];
    }
    t *= num_samples() - 1;
    auto const i = static_cast<size_t>(std::floor(t));
    t            = t - i;
    return m_data[i] * (1 - t) + m_data[i + 1] * t;
  }
  //----------------------------------------------------------------------------
  auto operator()(real_type const t) const { return sample(t); }
};
//==============================================================================
jet()->jet<double>;
//==============================================================================
}  // namespace tatooine::color_scales
//==============================================================================
#endif