NeuralNetTrainer.h

#ifndef NeuralNetTrainer_h
#define NeuralNetTrainer_h

//Marlin and LCIO includes
#include "marlin/Processor.h"
#include "lcio.h"
#include "EVENT/ReconstructedParticle.h"

//Neural Net includes
#include "nnet/inc/NeuralNet.h"
#include "nnet/inc/NeuralNetDataSet.h"
#include "nnet/inc/BackPropagationCGAlgorithm.h"

class NeuralNetTrainerProcessor : public marlin::Processor
{
public:
        //The usual Marlin processor methods
        virtual Processor* newProcessor() { return new NeuralNetTrainerProcessor; }
        NeuralNetTrainerProcessor();
        virtual ~NeuralNetTrainerProcessor();
        virtual void init();
        virtual void processRunHeader( LCRunHeader* pRun );
        virtual void processEvent( LCEvent* pEvent );
        //don't need this
        //virtual void check( LCEvent* pEvent );
        virtual void end();
protected:
        //variables for the steering file options
        std::string _JetCollectionName; //The name of the collection of ReconstructedParticles that is the jet (comes from the steering file)
        std::string _FlavourTagInputsCollectionName;
        std::string _TrueJetFlavourCollectionName;
        int _serialiseAsXML;
        nnet::NeuralNet::SerialisationMode _outputFormat;

        //These maps all use the same string keys to distinguish between the different nets.
        //The strings are of the form "c_net-2vtx", "bc_net-3vtx" etcetera.
        std::map<std::string,std::string> _filename;
        std::map<std::string,bool> _trainThisNet;       
        std::map<std::string,nnet::NeuralNetDataSet*> _dataSet;
        std::map<std::string,int> _numSignal;  
        std::map<std::string,int> _numBackground;  
        //List of strings for all the nets enabled for training
        //Strings are the same as the keys used in the maps above
        std::vector<std::string> _listOfSelectedNetNames; 
        //This map holds the position of the Inputs in the LCFloatVec
        std::map<std::string,unsigned int> _IndexOf; 
        int _nRun; 
        int _nEvent;
        int _nAcceptedEvents; 
        //useful constants
        static const int C_JET=4; 
        static const int B_JET=5; 
        //The following functions are just code that has been split off so that the code doesn't look quite so cluttered.
        void _displayCollectionNames( lcio::LCEvent* pEvent );
        void _trainNet( nnet::BackPropagationCGAlgorithm& pBackPropCGAlgo, nnet::NeuralNetDataSet& dataSet );
        bool _passesCuts( lcio::LCEvent* pEvent );
};

#endif //ifndef NeuralNetTrainer_h