UnitStepFunction.hh

// Copyright QED Software 2023.
 
#ifndef GRAIL_SINGLE_STEP_FUNCTION_H
#define GRAIL_SINGLE_STEP_FUNCTION_H
 
#include "Curve.hh"
 
namespace grail
{
namespace evaluator
{
 template <typename ContextType>
 class UnitStepFunction final : public Curve<ContextType>
 {
 public:
 UnitStepFunction(std::shared_ptr<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 data::EvaluatorType GetEvaluatorType() const override final { return data::EvaluatorType::CURVE_UNIT_STEP; }
 
 private:
 float threshold = 0.0f;
 float beforeThresholdValue = 0.0f;
 float afterThresholdValue = 0.0f;
 };
}
}
 
#endif // GRAIL_SINGLE_STEP_FUNCTION_H