SimpleDistanceColorMap.h

 
#pragma once

 
#if defined(SimpleDistanceColorMap_RECURSES)
#error Recursive header files inclusion detected in SimpleDistanceColorMap.h
#else // defined(SimpleDistanceColorMap_RECURSES)
#define SimpleDistanceColorMap_RECURSES
 
#if !defined SimpleDistanceColorMap_h
#define SimpleDistanceColorMap_h

// Inclusions
#include <iostream>
#include "DGtal/base/Common.h"
#include "DGtal/kernel/NumberTraits.h"
#include "DGtal/io/Color.h"
 
namespace DGtal
{

  // class SimpleDistanceColorMap
 
  template <typename TValue>
  class SimpleDistanceColorMap
  {
    // ----------------------- Standard services ------------------------------
  public:

    typedef TValue Value;

    SimpleDistanceColorMap ( const SimpleDistanceColorMap & other );

    SimpleDistanceColorMap & operator= ( const SimpleDistanceColorMap & anOther );
 

    SimpleDistanceColorMap(const Value & amin,
                           const Value & amax,
                           const bool withTick=true):
    myMin(amin), myMax(amax), myTick(withTick)
    {}
 
 

    Color operator()( const Value & aValue ) const
    {
      const double val = (NumberTraits<Value>::castToDouble(aValue) - NumberTraits<Value>::castToDouble(myMin))/
      (NumberTraits<Value>::castToDouble(myMax) - NumberTraits<Value>::castToDouble(myMin));
      const double d = 1. - val;
      double r = (1. - d*d) * .8;
      double g = (1. - (2. * (d - .5)) * (2. * (d - .5))) * .7;
      double b = (1. - (1. - d) * (1. - d));
 
      // lines
      if (myTick)
      {
        double h = val;
        h = h * 30.;
        h = h - floor( h );
        h = (1. / (1. + exp(-100.*(h - .55)))) + (1. / (1. + exp(-100.*(-h + .45))));
        h = 1. - h;
        r = h + (1. -h)*r;
        g = h + (1. - h)*g;
        b = h + (1. - h)*b;
      }
 
      return Color(static_cast<unsigned char>(r*255),
                   static_cast<unsigned char>(g*255),
                   static_cast<unsigned char>(b*255));
    }
 
 
 
    // ----------------------- Interface --------------------------------------
  public:

    void selfDisplay ( std::ostream & out ) const;

    bool isValid() const { return true;}

    const Value & min() const
    {
      return myMin;
    }

    const Value & max() const
    {
      return myMax;
    }
 
    // ------------------------- Protected Data ------------------------------
  protected:

    Value myMin;

    Value myMax;

    bool myTick;
 
    // ------------------------- Hidden services ------------------------------
  protected:

    SimpleDistanceColorMap();
 
 
  }; // end of class SimpleDistanceColorMap
} // namespace DGtal
 
 
//                                                                           //
 
#endif // !defined SimpleDistanceColorMap
 
#undef SimpleDistanceColorMap_RECURSES
#endif // else defined(SimpleDistanceColorMap_RECURSES)