PlotProcessor.h

#ifndef PlotProcessor_h
#define PlotProcessor_h

#include <vector>
#include <algorithm>

#include "../vertex_lcfi/util/inc/util.h"

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

using vertex_lcfi::util::efficiency_purity;
using vertex_lcfi::util::histogram_data;

class PlotProcessor : public marlin::Processor
{
public:
        //The usual Marlin processor methods
        virtual Processor* newProcessor() { return new PlotProcessor; }
        PlotProcessor();
        virtual ~PlotProcessor();
        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:
        std::string _JetCollectionName; 
        std::vector<std::string> _FlavourTagCollectionNames;    
        std::string _TrueJetFlavourColName; 
        std::string _OutputFilename; 
        std::vector<std::map<std::string,unsigned int> > _IndexOfForEachTag;
        int _nRun; 
        histogram_data<double> _jetEnergy; 
        std::vector<efficiency_purity<double> > _BTagEfficiencyPurity;
        std::vector<efficiency_purity<double> > _CTagEfficiencyPurity;
        std::vector<efficiency_purity<double> > _BCTagEfficiencyPurity;
        //useful constants
        static const int C_JET=4;
        static const int B_JET=5;
        void _displayCollectionNames( lcio::LCEvent* pEvent );
        bool _passesEventCuts( lcio::LCEvent* pEvent ); 
        bool _passesJetCuts( lcio::ReconstructedParticle* pJet ); 

        void _fillPlots( LCEvent* pEvent, unsigned int jet);
        void _outputDataToFile( std::string filename );
        double _jetEMax;
};

#endif //ifndef PlotProcessor_h