/data/refman/pasoursint/CMSSW_5_3_0/src/PhysicsTools/MVATrainer/interface/Source.h

Go to the documentation of this file.

#ifndef PhysicsTools_MVATrainer_Source_h
#define PhysicsTools_MVATrainer_Source_h

#include "PhysicsTools/MVAComputer/interface/AtomicId.h"

#include "PhysicsTools/MVATrainer/interface/SourceVariable.h"
#include "PhysicsTools/MVATrainer/interface/SourceVariableSet.h"

namespace PhysicsTools {

class MVATrainer;

class Source {
    public:
        Source(AtomicId name, bool trained = false) :
                trained(trained), name(name) {}
        virtual ~Source() {}

        inline AtomicId getName() const { return name; }

        inline SourceVariable *getOutput(AtomicId name) const
        { return outputs.find(name); }

        inline bool isTrained() const { return trained; }

        inline const SourceVariableSet &getInputs() const { return inputs; }
        inline const SourceVariableSet &getOutputs() const { return outputs; }

    protected:
        friend class MVATrainer;

        inline SourceVariableSet &getInputs() { return inputs; }
        inline SourceVariableSet &getOutputs() { return outputs; }

        bool                    trained;

    private:
        AtomicId                name;
        SourceVariableSet       inputs;
        SourceVariableSet       outputs;
};

} // namespace PhysicsTools

#endif // PhysicsTools_MVATrainer_Source_h