DSTAIDAPlotProcessor.h

#ifndef DSTAIDAPlotProcessor_h
#define DSTAIDAPlotProcessor_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"


//AIDA includes...
#include <AIDA/IHistogram1D.h>
#include <AIDA/IDataPointSet.h>
#include <AIDA/ITuple.h>


class DSTAIDAPlotProcessor : public marlin::Processor
{
public:
        //The usual Marlin processor methods
        virtual Processor* newProcessor() { return new DSTAIDAPlotProcessor; }
        DSTAIDAPlotProcessor();
        virtual ~DSTAIDAPlotProcessor();
        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; 
        int _nRun; 
        std::map<std::string,AIDA::IHistogram1D*> _pLightJetBTag;
        std::map<std::string,AIDA::IHistogram1D*> _pLightJetCTag;
        std::map<std::string,AIDA::IHistogram1D*> _pBJetBTag;
        std::map<std::string,AIDA::IHistogram1D*> _pBJetCTag;
        std::map<std::string,AIDA::IHistogram1D*> _pCJetBTag;
        std::map<std::string,AIDA::IHistogram1D*> _pCJetCTag; 
        std::map<std::string,AIDA::IHistogram1D*> _pBJetBCTag;
        std::map<std::string,AIDA::IHistogram1D*> _pCJetBCTag;
        std::map<std::string,AIDA::IHistogram1D*> _pLightJetBCTag;
        std::map<std::string,AIDA::IHistogram1D*> _pBTagBackgroundValues;
        std::map<std::string,AIDA::IHistogram1D*> _pCTagBackgroundValues;
        std::map<std::string,AIDA::IHistogram1D*> _pBCTagBackgroundValues;
        
        std::map<std::string,AIDA::IHistogram1D*> _pBJetBTagIntegral;    
        std::map<std::string,AIDA::IHistogram1D*> _pCJetBTagIntegral;
        std::map<std::string,AIDA::IHistogram1D*> _pLightJetBTagIntegral;
        std::map<std::string,AIDA::IHistogram1D*> _pBJetCTagIntegral;  
        std::map<std::string,AIDA::IHistogram1D*> _pCJetCTagIntegral;
        std::map<std::string,AIDA::IHistogram1D*> _pLightJetCTagIntegral;
        std::map<std::string,AIDA::IHistogram1D*> _pBJetBCTagIntegral; 
        std::map<std::string,AIDA::IHistogram1D*> _pCJetBCTagIntegral;
        std::map<std::string,AIDA::IHistogram1D*> _pLightJetBCTagIntegral; 


        std::vector<std::string> _VertexCatNames;
        std::vector<std::string>  _NumVertexCatDir;
        
        std::map<std::string,AIDA::IHistogram1D*>  _inputsHistogramsBJets;
        std::map<std::string,AIDA::IHistogram1D*>  _inputsHistogramsCJets;
        std::map<std::string,AIDA::IHistogram1D*>  _inputsHistogramsUDSJets;



        static const unsigned int N_VERTEX_CATEGORIES=3;  

        int _numberOfPoints;

        AIDA::ITuple* _pMyTuple;

        //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 FillTagPlots( LCEvent* pEvent, unsigned int jetNumber );

        void CreateTagPlots();


        void CreateFlavourTagInputPlots(LCRunHeader* pRun );
        void CreateFlavourTagTuple();
        void CalculateIntegralAndBackgroundPlots();
        void CalculateEfficiencyPurityPlots();

        void FillInputsPlots( LCEvent* pEvent, unsigned int jetNumber );


        AIDA::IDataPointSet* CreateEfficiencyPlot(const AIDA::IHistogram1D* pSignal, AIDA::IDataPointSet* pDataPointSet);
        AIDA::IHistogram1D* CreateIntegralHistogram(const AIDA::IHistogram1D* pNN, AIDA::IHistogram1D* pIntegral);
        AIDA::IDataPointSet* CreatePurityPlot(const AIDA::IHistogram1D* pSignal, const AIDA::IHistogram1D* pBackground, AIDA::IDataPointSet* pDataPointSet);
        AIDA::IDataPointSet* CreateLeakageRatePlot(const AIDA::IHistogram1D* pBackground, AIDA::IDataPointSet* pDataPointSet);
        AIDA::IDataPointSet* CreateXYPlot(const AIDA::IDataPointSet* pDataPointSet0, const AIDA::IDataPointSet* pDataPointSet1, AIDA::IDataPointSet* xyPointSet, const int dim0=0, const int dim1=0 );





};

#endif //ifndef DSTAIDAPlotProcessor_h