TreatmentFilter.hpp

#pragma once
#include <creativity/data/Treatment.hpp>
#include <memory>

namespace creativity { namespace data {

class TreatmentFilter : public Treatment {
    public:
        class StageProperties final {
            public:
                double value(const std::string &field) const;

            private:
                const Treatment &source_;
                const unsigned row_;
                StageProperties(const Treatment &source, unsigned rownum)
                    : source_(source), row_(rownum) {}
                friend class TreatmentFilter;
        };

        struct Properties final {
            public:
                std::unique_ptr<const StageProperties> pre;
                std::unique_ptr<const StageProperties> piracy;
                std::unique_ptr<const StageProperties> policy;
                std::unique_ptr<const StageProperties> piracy_SR;
                std::unique_ptr<const StageProperties> policy_SR;
                std::string source;
        };

        TreatmentFilter(const Treatment &source,
                std::function<bool(const Properties&)> filter,
                std::function<bool(bool pre, bool piracy, bool policy, bool short_run)> stage_filter = [](bool,bool,bool,bool) { return true; }
                );
};

}}