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 Datas ------------------------------
  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)