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