UnitStepFunction.hh

#ifndef GRAIL_SINGLE_STEP_FUNCTION_H
#define GRAIL_SINGLE_STEP_FUNCTION_H
 
#include "Curve.hh"
 
namespace grail
{
    namespace curves
    {
        template <typename ContextType>
        class UnitStepFunction final : public Curve<ContextType>
        {
        public:
            UnitStepFunction(std::shared_ptr<utility::Evaluator<ContextType>> childEvaluator,
                             float threshold,
                             float beforeThresholdValue,
                             float afterThresholdValue)
                : Curve<ContextType>{childEvaluator},
                  threshold{threshold},
                  beforeThresholdValue{beforeThresholdValue},
                  afterThresholdValue{afterThresholdValue}
            {
            }
 
            virtual float Sample(float argument) const override final
            {
                return argument > threshold ? afterThresholdValue : beforeThresholdValue;
            }
 
            float GetThreshold() const { return threshold; }
            float GetBeforeThresholdValue() const { return beforeThresholdValue; }
            float GetAfterThresholdValue() const { return afterThresholdValue; }
 
            virtual EvaluatorType GetEvaluatorType() const override final { return EvaluatorType::CURVE_UNIT_STEP; }
 
        private:
            float threshold = 0.0f;
            float beforeThresholdValue = 0.0f;
            float afterThresholdValue = 0.0f;
        };
    }
}
 
#endif // GRAIL_SINGLE_STEP_FUNCTION_H