ARICHReconstruction.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Luka Santelj, Rok Pestotnik                              *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#ifndef ARICHRECONSTRUCTION_H
#define ARICHRECONSTRUCTION_H
 
#include <arich/dbobjects/ARICHGeometryConfig.h>
#include <arich/dbobjects/ARICHReconstructionPar.h>
#include <arich/dbobjects/ARICHChannelMask.h>
#include <arich/dbobjects/ARICHChannelMapping.h>
#include <arich/dbobjects/ARICHGlobalAlignment.h>
#include <arich/dbobjects/ARICHMirrorAlignment.h>
#include <arich/dbobjects/ARICHAeroTilesAlignment.h>
#include "framework/datastore/StoreArray.h"
#include "arich/dataobjects/ARICHHit.h"
#include "arich/dataobjects/ARICHTrack.h"
#include "arich/dataobjects/ARICHLikelihood.h"
#include <framework/database/DBObjPtr.h>
 
 
#include <TVector3.h>
 
namespace Belle2 {
  class ARICHReconstruction {
 
  public:
 
    explicit ARICHReconstruction(int storePhotons = 0);
 
    ~ARICHReconstruction() {};
 
    void initialize();
 
    int smearTrack(ARICHTrack& arichTrack);
 
    void transformTrackToLocal(ARICHTrack& arichTrack, bool align);
 
    int likelihood2(ARICHTrack& arichTrack, const StoreArray<ARICHHit>& arichHits, ARICHLikelihood& arichLikelihood);
 
    void setTrackPositionResolution(double pRes);
    void setTrackAngleResolution(double aRes);
 
    void useMirrorAlignment(bool align)
    {
      m_alignMirrors = align;
    };
 
    void correctEmissionPoint(int tileID, double r);
 
  private:
 
    static const int c_noOfHypotheses = Const::ChargedStable::c_SetSize; 
    static const int c_noOfAerogels = 5; 
    double p_mass[c_noOfHypotheses];  
    DBObjPtr<ARICHGeometryConfig> m_arichgp; 
    DBObjPtr<ARICHReconstructionPar> m_recPars; 
    DBObjPtr<ARICHChannelMask> m_chnMask; 
    DBObjPtr<ARICHChannelMapping> m_chnMap; 
    DBObjPtr<ARICHGlobalAlignment> m_alignp; 
    DBObjPtr<ARICHMirrorAlignment> m_mirrAlign; 
    OptionalDBObjPtr<ARICHAeroTilesAlignment> m_tileAlign; 
    std::vector<TVector3> m_mirrorPoints; 
    std::vector<TVector3> m_mirrorNorms;  
    double m_trackPosRes; 
    double m_trackAngRes; 
    bool   m_alignMirrors; 
    unsigned int m_nAerogelLayers; 
    double  m_refractiveInd[c_noOfAerogels]; 
    double  m_zaero[c_noOfAerogels]; 
    double  m_thickness[c_noOfAerogels]; 
    double  m_transmissionLen[c_noOfAerogels]; 
    double  m_n0[c_noOfAerogels];  
    TVector3 m_anorm[c_noOfAerogels]; 
    int m_storePhot; 
    double m_tilePars[124][2] = {0};
 
    int InsideDetector(TVector3 a, int copyno);
 
    TVector3 HitVirtualPosition(const TVector3& hitpos, int mirrorID);
 
 
    TVector3 FastTracking(TVector3 dirf, TVector3 r,  double* refind, double* z, int n, int opt);
//    TVector3 FastTracking(TVector3 dirf, TVector3 r,  double* refind, double* z, int n);
 
 
    TVector3 FastTrackingSimple(TVector3 dirf, TVector3 r,  double* refind, double* z, int n);
 
 
    int  CherenkovPhoton(TVector3 r, TVector3 rh,
                         TVector3& rf, TVector3& dirf,
                         double* refind, double* z, int n, int mirrorID = 0);
 
    /*
      \param pos photon position
      \param dir photon direction
      \param mirrorID ID of mirrro plate
    */
    bool HitsMirror(const TVector3& pos, const TVector3& dir, int mirrorID);
 
    TVector3 getTrackMeanEmissionPosition(const ARICHTrack& track, int iAero);
 
    TVector3 getTrackPositionAtZ(const ARICHTrack& track, double zout);
 
    TVector3 getMirrorPoint(int mirrorID);
 
    TVector3 getMirrorNorm(int mirrorID);
 
  };
 
} // end of namespace Belle2
 
#endif // ARICHRECONSTRUCTION_H