HepevtReader.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Martin Ritter, Susanne Koblitz                           *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#ifndef HEPEVTREADER_H
#define HEPEVTREADER_H
 
#include <framework/core/FrameworkExceptions.h>
#include <mdst/dataobjects/MCParticleGraph.h>
#include <boost/tokenizer.hpp>
 
#include <string>
#include <fstream>
 
#include <TLorentzRotation.h>
 
namespace Belle2 {
  class HepevtReader {
 
  public:
 
    //Define exceptions
    BELLE2_DEFINE_EXCEPTION(HepEvtCouldNotOpenFileError, "Could not open file %1% !");
    BELLE2_DEFINE_EXCEPTION(HepEvtInvalidDaughterIndicesError,
                            "Line %1%: Invalid daughter indices d1=%2%, d2=%3%, N=%4% (0<=d1<=d2<=N required)");
    BELLE2_DEFINE_EXCEPTION(HepEvtHeaderNotValidError, "Line %1%: Event header not understood: %2%");
    BELLE2_DEFINE_EXCEPTION(HepEvtConvertFieldError, "Line %1%: Could not convert field %2%: %3%");
    BELLE2_DEFINE_EXCEPTION(HepEvtParticleFormatError, "Line %1%: Particle format not understood, got %2% fields !");
    BELLE2_DEFINE_EXCEPTION(HepEvtEmptyEventError,
                            "Line %1%: Number of particles in event is %2% ! (This could mean EOF is reached.) ");
    HepevtReader(): m_nVirtual(0), m_wrongSignPz(false), m_lineNr(0) {}
 
    ~HepevtReader() { if (m_input) m_input.close(); }
 
    void open(const std::string& filename);
 
    void closeCurrentInputFile() {m_input.close();}
 
    int getEvent(MCParticleGraph& graph, double& weight);
 
    bool skipEvents(int n);
 
    int m_nVirtual;        
    bool m_wrongSignPz;    
    TLorentzRotation m_labboost;     
  protected:
    typedef boost::tokenizer<boost::char_separator<char> > tokenizer;
    static const boost::char_separator<char> sep;
 
    int m_lineNr;          
    std::ifstream m_input; 
    //  /*   int eventID; /**< The event ID number if provided in HepEvtfile else -1.  */ */
 
 
    std::string getLine();
 
    int readEventHeader(int& eventID, double& eventWeight);
 
    void readParticle(MCParticleGraph::GraphParticle& particle);
  };
 
}
 
#endif //HEPEVTREADER_H