TrkReco.h

Go to the documentation of this file.

//-----------------------------------------------------------------------------
// $Id: TrkReco.h,v 1.30 2011/10/08 01:56:15 maoh Exp $
//-----------------------------------------------------------------------------
// Filename : TrkReco.h
// Section  : Tracking
// Owner    : Yoshi Iwasaki
// Email    : yoshihito.iwasaki@kek.jp
//-----------------------------------------------------------------------------
// Description : A tracking module.
//               See http://bsunsrv1.kek.jp/~yiwasaki/tracking/
//-----------------------------------------------------------------------------
 
#ifndef TrkReco_FLAG_
#define TrkReco_FLAG_
 
#ifdef TRKRECO_DEBUG_DETAIL
#  ifndef TRKRECO_DEBUG
#    define TRKRECO_DEBUG
#  endif
#endif
 
#include "GaudiKernel/Algorithm.h"
#include "GaudiKernel/NTuple.h"
#include <string>
 
// #define HEP_SHORT_NAMES
#include "CLHEP/Alist/AList.h"
#include "RawDataProviderSvc/IRawDataProviderSvc.h"
#include "TrkReco/TRungeFitter.h"
#include "TrkReco/TTrackManager.h"
class TMDC;
class TMDCWireHit;
class TFinderBase;
class TConformalFinder0;
class TCurlFinder;
class TTrack;
class TTrackMC;
 
#include "BesTimerSvc/BesTimer.h"
#include "BesTimerSvc/IBesTimerSvc.h"
#include "MdcCalibFunSvc/IMdcCalibFunSvc.h"
 
extern float TrkRecoHelixFitterChisqMax;
 
/// A tracking module.
class TrkReco : public Algorithm {
 
public:
  /// returns TrkReco.
  // static TrkReco * getTrkReco(void);
  TrkReco( const std::string& name, ISvcLocator* pSvcLocator );
  StatusCode initialize();
  StatusCode execute();
  StatusCode finalize();
  StatusCode beginRun();
 
public: // BASF interfaces
  /// initializes TrkReco.
  //    void init(int * status);
 
  /// terminates TrkReco.
  //    void term(void);
 
  /// processes an event.
  //    void event(BesEvent *, int *);
  //    void event(void);
 
  /// temporarily required
  //    void begin_run(BesEvent *, int *) {};
  //    void end_run(BesEvent *, int *) {};
  void disp_stat( const char* );
 
public: // TrkReco interfaces
  /// returns name.
  // zsl    std::string name(void) const;
 
  /// returns version.
  // liuqg    std::string version(void) const;
 
  /// dumps debug information.
  void dump( const std::string& message = std::string( "" ),
             const std::string& prefix  = std::string( "" ) ) const;
 
  /// returns a pointer to the rphi finder.
  const TFinderBase* confFinder( void ) const;
 
  /// returns a pointer to the curl finder.
  const TFinderBase* curlFinder( void ) const;
 
  /// returns a pointer to TTrackManager.
  const TTrackManager& trackManager( void ) const;
 
public: // to access information
  /// clears all TMDC information.
  void clear( void );
 
  /// clears TMDC information.
  void fastClear( void );
 
  /// returns a list of reconstructed tracks.
  const AList<TTrack>& tracks( void ) const;
 
private:
  /// initiate Parameters
  void initPara( void );
 
  /// returns true if an event is MC.
  bool mcEvent( void ) const;
 
  /// creates MC info. of reconstructed tracks.
  void mcInformation( void );
 
  /// cal maxdistance temporary in conformal plane
  double maxdDistance( TMLink* ) const;
 
  TMDC* cdcInit( void );
 
  /// initiate and fill Tuple
  void InitTuple( void );
  void FillTuple( void );
 
public: // public members for basf interface
  //...RawData/TUPLE...
  int  b_mcPar;
  int  b_mcHit;
  int  b_tuple;
  int  b_goodTrk;
  int  b_timeTest;
  bool m_dropHot;
  bool m_combineTracking;
  bool m_keepBadTdc;
  bool m_keepUnmatch;
  int  b_RungeKuttaCorrection;
  //...MDC...
  float       b_cdcVersion;
  std::string _cdcVersion;
  float       b_fudgeFactor;
 
  //...Bunch Time...
  int useESTime;
 
  //...TrkReco...
  int   b_debugLevel;
  int   b_useAllHits;
  int   b_doT0Reset;
  int   b_nT0ResetMax;
  int   b_doMCAnalysis;
  float b_helixFitterChisqMax;
 
  //...Perfect Finder...
  int b_doPerfectFinder;
  int b_perfectFitting;
 
  //...Conformal Finder...
  int   b_conformalFinder;
  int   b_doConformalFinder;
  int   b_doConformalFastFinder;
  int   b_doConformalSlowFinder;
  int   b_conformalPerfectSegmentFinding;
  int   b_conformalFittingFlag;
  float b_conformalMaxSigma;
  int   b_conformalMinNLinksForSegment;
  int   b_conformalMinNCores;
  int   b_conformalMinNSegments;
  float b_salvageLevel;
  int   b_conformalSalvageLoadWidth;
  int   b_conformalStereoMode;
  int   b_conformalStereoLoadWidth;
  float b_conformalStereoMaxSigma;
  float b_conformalStereoSzSegmentDistance;
  float b_conformalStereoSzLinkDistance;
 
  //...Old Conformal Finder...
  int   b_doConformalFinderStereo;
  int   b_doConformalFinderCosmic;
  float b_conformalFraction;
  float b_conformalStereoZ3;
  float b_conformalStereoZ4;
  float b_conformalStereoChisq3;
  float b_conformalStereoChisq4;
  int   b_conformalFittingCorrections;
  int   m_CalibFlag;
 
  //...Track Manager...
  double b_momentumCut;
  int    b_fittingFlag;
  int    b_doSalvage;
  int    b_doMerge;
  int    b_doT0Determination;
  int    b_nTracksForT0;
  int    b_sortMode;
  int    b_doAssociation;
  float  b_associateSigma;
  int    b_test;
 
  //...Curl Finder...
  int    b_doCurlFinder;
  int    min_segment;
  int    min_salvage;
  double bad_distance_for_salvage;
  double good_distance_for_salvage;
  int    min_sequence;
  int    min_fullwire;
  double range_for_axial_search;
  double range_for_stereo_search;
  int    superlayer_for_stereo_search;
  double range_for_axial_last2d_search;
  double range_for_stereo_last2d_search;
  double trace2d_distance;
  double trace2d_first_distance;
  double trace3d_distance;
  int    determine_one_track;
  double selector_max_impact;
  double selector_max_sigma;
  double selector_strange_pz;
  double selector_replace_dz;
  int    stereo_2dfind;
  int    merge_exe;
  double merge_ratio;
  double merge_z_diff;
  double mask_distance;
  double ratio_used_wire;
  double range_for_stereo1;
  double range_for_stereo2;
  double range_for_stereo3;
  double range_for_stereo4;
  double range_for_stereo5;
  double range_for_stereo6;
  double z_cut;
  double z_diff_for_last_attend;
  int    svd_reconstruction;
  double min_svd_electrons;
  int    on_correction;
  int    output_2dtracks;
  int    curl_version;
 
  // jialk
  double minimum_seedLength;
  double minimum_2DTrackLength;
  double minimum_3DTrackLength;
  double minimum_closeHitsLength;
  double MIN_RADIUS_OF_STRANGE_TRACK;
  double ULTIMATE_MIN_RADIUS_OF_STRANGE_TRACK;
 
private:
  bool m_beginRun{ false };
 
  //    static TrkReco * _TrkReco;
  TMDC*                _cdc;
  TFinderBase*         _perfectFinder;
  TFinderBase*         _confFinder;
  TCurlFinder*         _curlFinder;
  TTrackManager        _trackManager;
  TRungeFitter         _rkfitter;
  IRawDataProviderSvc* _rawDataProviderSvc;
  IMdcCalibFunSvc*     _mdcCalibFunSvc;
  IMdcGeomSvc*         _mdcGeomSvc;
 
  unsigned        _nEvents;
  AList<TTrackMC> _mcTracks;
 
  // Tuple related
  IBesTimerSvc* m_timersvc;
  BesTimer*     m_timer[2];
  double        t0_bes;
  double        t0Sta;
  int           havedigi[43][288];
  int           MC_DIGI_SIZE;
 
  NTuple::Tuple*      m_tuple; // rec3D
  NTuple::Item<float> t_mcphi, t_mctheta, t_mcptot, t_mcpt, t_mcpz, t_mct0, t_nDigi, t_pt,
      t_ptot, t_dr, t_dz, t_tanlmd, t_phi, t_radius, t_chi2, t_ndf, t_nHits, t_nCores, t_nSegs,
      t_length, t_length2, t_evtNo, t_dpt, t_dptot, t_dlmd, t_dphi, t_t0, t0_sta, t_good_theta,
      t_gdNLayers, t_mcNLayers, t_bestNLayers, t_bestMcNLayers;
 
  NTuple::Tuple*      m_tuple2; // rec2D
  NTuple::Item<float> t2_mctheta, t2_mcpt, t2_nDigi, t2_length, t2_ndf, t2_evtNo, t2_chi2,
      t2_nHits, t2_nSegs, t2_length2, t2_radius;
 
  NTuple::Tuple*      m_tuple3; // raw
  NTuple::Item<float> t3_mct0, t3_mcptot, t3_mcpt, t3_mctheta, t3_mcphi, t3_mcpid, t3_evtNo;
 
  NTuple::Tuple*      m_tuple31; // raw/event
  NTuple::Item<float> t3_nDigi, t3_goodLength, t3_t0Rec, t3_t0, t3_t0Sta, t3_finalLength,
      t3_length;
 
  NTuple::Tuple*      m_tuple4; // hits
  NTuple::Item<float> t4_Dist, t4_drift, t4_dDrift, t4_mcDrift, t4_mcLR, t4_pull, t4_lyrId,
      t4_localId;
  NTuple::Item<float> t4_LR, t4_tdc, t4_z, t4_bz, t4_fz, t4_fy, t4_phi, t4_nHits;
 
  NTuple::Tuple*      m_tuple5; // charge
  NTuple::Item<float> t5_ptotPos, t5_ptotNeg, t5_drPos, t5_drNeg, t5_dzPos, t5_dzNeg;
 
  NTuple::Tuple*      m_tuple6; // unreconstructed.
  NTuple::Item<float> u_length2, u_mcptot, u_mcpt, u_mctheta, u_nDigi, u_evtNo, u_mct0, ut_t0,
      ut0_sta;
 
  NTuple::Tuple*      m_tuple7; // time test.
  NTuple::Item<float> ti_eventTime, ti_recTrkNum, ti_evtNo, ti_nHits, ti_nDigi;
 
  NTuple::Tuple*      m_tuple9; // segments
  NTuple::Item<float> t9_times, t9_nLinks, t9_nUsed, t9_nSL, t9_mctheta;
 
  NTuple::Tuple*      m_tuple10; // raw hit
  NTuple::Item<float> t10_tdc, t10_adc, t10_drift, t10_dDrift, t10_lyrId, t10_localId;
};
 
//-----------------------------------------------------------------------------
 
#ifdef TRKRECO_NO_INLINE
#  define inline
#else
#  undef inline
#  define TrkReco_INLINE_DEFINE_HERE
#endif
 
#ifdef TrkReco_INLINE_DEFINE_HERE
 
inline const TFinderBase* TrkReco::confFinder( void ) const {
  return (TFinderBase*)_confFinder;
}
 
inline const TFinderBase* TrkReco::curlFinder( void ) const {
  return (TFinderBase*)_curlFinder;
}
 
inline const AList<TTrack>& TrkReco::tracks( void ) const {
  return _trackManager.tracksFinal();
}
 
/*
inline
std::string
TrkReco::name(void) const {
    return std::string("TrkReco");
}
*/
 
inline const TTrackManager& TrkReco::trackManager( void ) const { return _trackManager; }
 
#endif
 
#undef inline
 
#endif /* TrkReco_FLAG_ */