DSTPlotProcessor.h

#ifndef DSTPlotProcessor_h
#define DSTPlotProcessor_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 DSTPlotProcessor : public marlin::Processor
{
public:
        //The usual Marlin processor methods
        virtual Processor* newProcessor() { return new DSTPlotProcessor; }
        DSTPlotProcessor();
        virtual ~DSTPlotProcessor();
        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::string _TrueJetFlavourColName; 
        std::string _OutputFilename; 
        int _nRun; 
        std::string _FlavourTagCollectionName;
        std::string _FlavourTagInputsCollectionName;

        std::map<std::string,unsigned int> _IndexOfForEachTag;
        std::map<std::string,unsigned int>  _FlavourIndex;
        std::map<std::string,unsigned int>_InputsIndex;


        histogram_data<double> _jetEnergy; 
        efficiency_purity<double>_BTagEfficiencyPurity;
        efficiency_purity<double>_CTagEfficiencyPurity;
        efficiency_purity<double>_BCTagEfficiencyPurity;

        int _checkDST; 
        //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;
        void _checkDSTParameters(LCEvent* pEvent);

};

#endif //ifndef DSTPlotProcessor_h