ReaderSAD.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <framework/core/FrameworkExceptions.h>
 
#include <mdst/dataobjects/MCParticleGraph.h>
 
#include <TGeoMatrix.h>
#include <TFile.h>
#include <TTree.h>
 
#include <string>
 
 
namespace Belle2 {

  class ReaderSAD {
 
  public:
 
    //Define exceptions
    BELLE2_DEFINE_EXCEPTION(SADCouldNotOpenFileError, "Could not open file %1% !");
    BELLE2_DEFINE_EXCEPTION(SADEndOfFile, "End of the SAD file.");

    enum AcceleratorRings {
      c_HER = 0, 
      c_LER = 1  
    };

    ReaderSAD();

    ~ReaderSAD();

    void initialize(TGeoHMatrix* transMatrix, double sRange, AcceleratorRings accRing, double readoutTime);

    void open(const std::string& filename);

    void setMomentumRes(double pxRes, double pyRes) { m_pxRes = pxRes; m_pyRes = pyRes; }

    double getSADParticle(MCParticleGraph& graph);

    bool getRealParticle(MCParticleGraph& graph);

    void addAllSADParticles(MCParticleGraph& graph);

    TGeoHMatrix SADtoGeant(ReaderSAD::AcceleratorRings accRing, double s);
 
 
  protected:
 
    TFile* m_file;                 
    TTree* m_tree;                 
 
    TGeoHMatrix* m_transMatrix;    
    double m_sRange;               
    AcceleratorRings m_accRing;    
    double m_pxRes;                
    double m_pyRes;                
 
    double m_SADToRealFactor;      
    double m_readoutTime;          
 
    unsigned int m_realPartNum;    
    unsigned int m_realPartEntry;  
    int m_readEntry;               
 
    double m_lostX;             
    double m_lostY;             
    double m_lostS;             
    double m_lostPx;            
    double m_lostPy;            
    double m_lostRate;          
    double m_lostE;             
 
    double m_inputSAD_ssraw;     
    double m_inputSAD_sraw;      
    double m_inputSAD_ss;        
    double m_inputSAD_s;         
    double m_inputSAD_Lss;       
    int m_inputSAD_nturn;     
    double m_inputSAD_x;         
    double m_inputSAD_y;         
    double m_inputSAD_px;        
    double m_inputSAD_py;        
    double m_inputSAD_xraw;      
    double m_inputSAD_yraw;      
    double m_inputSAD_r;         
    double m_inputSAD_rr;        
    double m_inputSAD_dp_over_p0;
    double m_inputSAD_E;         
    double m_inputSAD_rate;      
    double m_inputSAD_watt;      
 
  private:

    void convertParamsToSADUnits();

    void addParticleToMCParticles(MCParticleGraph& graph, bool gaussSmearing = false);

    int calculateRealParticleNumber(double rate);

//    TGeoHMatrix SADtoGeant(ReaderSAD::AcceleratorRings accRing, double s);
 
    struct straightElement {
      double x0;        
      double z0;        
      double l;         
      double phi;       
    };

    struct bendingElement {
      double rt;        
      double x0;        
      double z0;        
      double sphi;      
      double dphi;      
    };
 
  };

}