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