ScalarColorMap.h

/*
Copyright 2007-2024. Algoryx Simulation AB.
 
All AGX source code, intellectual property, documentation, sample code,
tutorials, scene files and technical white papers, are copyrighted, proprietary
and confidential material of Algoryx Simulation AB. You may not download, read,
store, distribute, publish, copy or otherwise disseminate, use or expose this
material without having a written signed agreement with Algoryx Simulation AB.
 
Algoryx Simulation AB disclaims all responsibilities for loss or damage caused
from using this software, unless otherwise stated in written agreements with
Algoryx Simulation AB.
*/
 
#pragma once
 
#include "momentum_export.h"
#include "momentum_namespace.h"
 
#include "macros.h"
#include "Vec4.h"
 
#include <vector>
 
#include <memory>
#include <agx/ref_ptr.h>
 
#ifndef SWIG
#include <agxOSG/ScalarColorMap.h>
 
#endif
 
namespace MOMENTUM_NAMESPACE
{
  class MOMENTUM_EXPORT ScalarColorMap
  {
  public:
#ifndef SWIG
 
    ScalarColorMap(agxOSG::ScalarColorMap* scalarColorMap);
 
    ScalarColorMap();
#endif
 
    enum ColorRangePreset
    {
      RAINBOW,
      COOLWARM,
      VIRIDIS,
      INFERNO,
      PLASMA,
      GRAYSCALE,
      BLUE_WHITE_RED,
      CIVIDIS,
      BONE,
      COPPER,
      AFMHOT
    };
 
#ifndef AGX_MOMENTUM_ANALYSIS_API
 
    ScalarColorMap(
      double minScalar,
      double maxScalar);
 
    void setColorsFromPreset(ColorRangePreset colorRangePreset);
#endif
 
    Vec4 getColor(double scalar, bool modifyAlpha = false) const;
 
#ifndef AGX_MOMENTUM_ANALYSIS_API
    void setMinScalar(double minScalar);
 
    void setMaxScalar(double maxScalar);
 
    void setScalarInterval(double minScalar, double maxScalar);
 
    /*
    Manually set the colors used in the interpolation range.
    \param colorVector - a vector containing the specified colors of the interval.
    */
    void setColors(const std::vector<Vec4>& colorVector);
#endif
 
    double getMinScalar() const;
 
    double getMaxScalar() const;
 
    std::vector<Vec4> getColors() const;
 
  private:
    // static agxRender::ColorVector* toRenderColorVec(const std::vector<Vec4>& colorVector);
 
#ifndef SWIG
    agx::ref_ptr<agxOSG::ScalarColorMap> m_scalarColorMap;
    agxOSG::ScalarColorMap* getAGXScalarColorMap() const;
#endif
  };
 
  V1_DECLARE_CONTAINER_TYPE(ScalarColorMap);
} // namespace momentum