commonstateevaluators.h

Go to the documentation of this file.

#ifndef COMMONSTATEEVALUATORS_H
#define COMMONSTATEEVALUATORS_H

#include <dtree/stateevaluator.h>

#include <memory>

namespace dtree {

template <typename T>
class VariableReferenceStateEvaluator: public StateEvaluator {
private:
    T& mVariable;
public:

    VariableReferenceStateEvaluator(T &var)
        : StateEvaluator(), mVariable(var) {
    }

    double evaluate(int, Vessel *) const {
        return static_cast<double>(mVariable);
    }


};

template <typename T>
class ConstStateEvaluator : public StateEvaluator {
private:
    const T mConst;
public:
    ConstStateEvaluator(const T &var)
        : StateEvaluator(), mConst(var) {
    }

    double evaluate(int, Vessel *) const {
        return static_cast<double>(mConst);
    }


};




template <typename Operator>
class TwoArgumentsComparatorStateEvaluator : public StateEvaluator {
private:
    std::shared_ptr<StateEvaluator> mOp1, mOp2;
    Operator mOperator;

public:
    TwoArgumentsComparatorStateEvaluator (std::shared_ptr<StateEvaluator> op1, std::shared_ptr<StateEvaluator> op2, Operator oper)
        : StateEvaluator(), mOp1(op1), mOp2(op2), mOperator(oper) {
    }

    double evaluate(int tstep, Vessel *v) const {
        if (mOperator(mOp1->evaluate(tstep, v), mOp2->evaluate(tstep, v))) {
            return 1.0;
        } else {
            return 0.0;
        }
    }
};

} // ns

#endif // COMMONSTATEEVALUATORS_H