LHEReader.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Torben Ferber                                            *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <framework/core/FrameworkExceptions.h>
#include <mdst/dataobjects/MCParticleGraph.h>
#include <boost/tokenizer.hpp>
 
#include <string>
#include <fstream>
 
#include <TLorentzRotation.h>
 
namespace Belle2 {
  class LHEReader {
 
  public:
 
    //Define exceptions
    BELLE2_DEFINE_EXCEPTION(LHECouldNotOpenFileError, "Could not open file %1% !");
    BELLE2_DEFINE_EXCEPTION(LHEInvalidDaughterIndicesError,
                            "Line %1%: Invalid daughter indices d1=%2%, d2=%3%, N=%4% (0<=d1<=d2<=N required)");
    BELLE2_DEFINE_EXCEPTION(LHEHeaderNotValidError, "Line %1%: Event header not understood: %2%");
    BELLE2_DEFINE_EXCEPTION(LHEConvertFieldError, "Line %1%: Could not convert field %2%: %3%");
    BELLE2_DEFINE_EXCEPTION(LHEParticleFormatError, "Line %1%: Particle format not understood, got %2% fields !");
    BELLE2_DEFINE_EXCEPTION(LHEEmptyEventError, "Line %1%: Number of particles in event is %2% ! (This could mean EOF is reached.) ");
    LHEReader():
      m_wrongSignPz(false),
      m_lineNr(0),
      m_indexInitial(0),
      m_indexVirtual(0)
    {}
 
    ~LHEReader() { 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);
 
    void setInitialIndex(int index) { m_indexInitial = index; }
 
    void setVirtualIndex(int index) { m_indexVirtual = index; }
 
    bool m_wrongSignPz;    
    TLorentzRotation m_labboost;     
    double m_meanDecayLength = 0.;        
    double m_Rmin = 0.; 
    double m_Rmax = 0.; 
    int m_pdgDisplaced = 0; 
  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 LHEfile else -1.  */ */
 
 
    std::string getLine();
 
//     int readEventHeader(int& eventID, double& eventWeight);
    int readEventHeader(double& eventWeight);
 
    int readParticle(MCParticleGraph::GraphParticle& particle);
 
    int m_indexInitial;  
    int m_indexVirtual;  
  };
 
}