HepRepOutputProcessor.h

#ifndef HEP_REP_OUTPUT_PROCESSOR_H
#define HEP_REP_OUTPUT_PROCESSOR_H 1

#include <marlin/Processor.h>
#include <lcio.h>
#include <string>

class HepRepOutputProcessor : public marlin::Processor
{

    public:

        HepRepOutputProcessor();


        ~HepRepOutputProcessor();


        Processor * newProcessor();


        void init();


        void processRunHeader (lcio::LCRunHeader *run);


        virtual void processEvent (lcio::LCEvent *evt);


        virtual void check (lcio::LCEvent *evt);


        virtual void end();

    private:

        enum ColorScheme
        {
            RAINBOW = 0,
                        REDBLUE = 1,
                        TOPOGRAPHIC = 2,
                        GREYSCALE = 3,
                        REDBLUESCALE = 4,
                        COLORBRIGHTNESS = 5,
                        COLORWHITE = 6
        };

                int _pulseColorScheme;

                std::vector<std::string> _namesOfSimTrackerHitCollections;
                std::vector<std::string> _namesOfTrackerDataCollections;
                std::vector<std::string> _namesOfTrackerHitCollections;
                std::vector<std::string> _namesOfTrackerPulseCollections;
                std::vector<std::string> _namesOfTrackCollections;
                std::string _nameOfOutputFile;

                virtual void drawHepRepPad (int moduleID, int padIndex); 
                virtual void setHepRepPadColor (double padCharge, float minCharge, float maxCharge, int colorScheme, int pulseQuality); 

                // the HepRepXML writer
                HepRepXMLWriter _hepRepWriter;

                std::vector<int> _eventsToDisplay;

                bool _drawPadPlane;
};

#endif // HEP_REP_OUTPUT_PROCESSOR_H