PythiaEventReader.h

Go to the documentation of this file.

//PythiaEventReader.h
//M.L. Miller (Yale Software)
//07/02

#ifndef PythiaEventReader_HH
#define PythiaEventReader_HH

#include "TObject.h"
#include <iostream>

#include <map>
using std::map;
#include <vector>
using std::vector;
#include <string>
using std::string;

class pythiaEvent;
class TTree;
class TFile;

//helper structs for book-keeping:
struct PythiaEventSummary
{
    PythiaEventSummary(pythiaEvent*);
    int subprocessType;
    int nParticle;
    int nFinalState;
    double shat;
    double that;
    double Q2;
    double costheta;
};

struct SummaryValue
{
    string fileName; //the file that it came from
    int eventNumber; //the event number
};

struct PythiaEventSummaryLessThan
{
    bool operator()(const PythiaEventSummary& lhs, const PythiaEventSummary& rhs) const;
};

class PythiaEventReader
{
public:
    PythiaEventReader();
    virtual ~PythiaEventReader();

    //close current file (if any) and open a new one
    void openFile(const char*);
    void setTree(TTree* chain); //someone's created a chain for us, we just set the pointer + branch
    void closeFile();

    //event logging
    void setLog(bool v) {mDoLog=v;}
    bool log() const {return mDoLog;}
    void checkLog(bool verbose=false);
    
    //Event access
    int numberOfEvents() const;
    pythiaEvent* getEvent(int event_number);

protected:
    TFile* mFile; 
    pythiaEvent* mEvent; 
    TTree* mTree; 
    
    bool mDoLog;
    
    typedef vector<SummaryValue> SummaryVec;
    typedef map<PythiaEventSummary, SummaryVec, PythiaEventSummaryLessThan> EventSumMap; 
    EventSumMap* mEventSumMap; 
    
private:
    ClassDef(PythiaEventReader,1)
};

//inlines
inline ostream& operator<<(ostream& os, const PythiaEventSummary& s)
{
    return os<<"spt: "<<s.subprocessType<<"\tnP: "<<s.nParticle<<"\tnFS: "<<s.nFinalState
             <<"\ts: "<<s.shat<<"\tt: "<<s.that<<"\tQ2: "<<s.Q2<<"\tcos(t*): "<<s.costheta;
}
#endif