riplet/LxTrackAna.h

Go to the documentation of this file.

#ifndef LXTRACKANA_INCLUDED
#define LXTRACKANA_INCLUDED
 
#pragma GCC diagnostic ignored "-Weffc++"
 
#include "CbmMuchPixelHit.h"
#include "FairTask.h"
#include "LxSettings.h"
#include "LxTrackAnaSegments.h"
#include "TClonesArray.h"
#include <list>
#include <vector>
 
struct LxSimplePoint {
  Double_t x;
  Double_t y;
  Double_t z;
  Double_t tx;
  Double_t ty;
  LxSimplePoint() : x(0), y(0), z(0), tx(0), ty(0) {}
  LxSimplePoint(Double_t X, Double_t Y, Double_t Z, Double_t Tx, Double_t Ty)
    : x(X), y(Y), z(Z), tx(Tx), ty(Ty) {}
};
 
struct LxSimpleTrack {
  Int_t pdgCode;
  Int_t motherId;
  Double_t p;
  Double_t pt;
  Double_t px;
  Double_t py;
  Double_t pz;
  Double_t e;
  Double_t charge;
  LxSimpleTrack(Int_t pdgc,
                Int_t mid,
                Double_t P,
                Double_t Pt,
                Double_t Px,
                Double_t Py,
                Double_t Pz,
                Double_t E)
    : pdgCode(pdgc)
    , motherId(mid)
    , p(P)
    , pt(Pt)
    , px(Px)
    , py(Py)
    , pz(Pz)
    , e(E)
    , charge(0)
    , linkedMuchTrack(0, 0)
    , linkedStsTrack(0)
    , parent(0) {}
  std::list<LxSimplePoint> stsPoints[LXSTSSTATIONS];
  std::list<LxSimplePoint> muchPoints[LXSTATIONS][LXLAYERS];
  std::list<LxSimplePoint> muchMCPts
    [LXSTATIONS]
    [LXLAYERS];  // These array is used for storing MUCH MC points when the 'main' array contains hits.
  std::pair<LxSimpleTrack*, Double_t> linkedMuchTrack;
  std::list<std::pair<LxSimpleTrack*, Double_t>>
    linkedStsTracks;  // The front() contains STS track with the minimal chi2.
  LxSimpleTrack* linkedStsTrack;
  LxSimpleTrack* parent;
  void RebindMuchTrack();
};
 
class LxTrackAnaTriplet : public FairTask {
public:
  LxTrackAnaTriplet();
  ~LxTrackAnaTriplet();
  InitStatus Init();         // Inherited virtual.
  void Exec(Option_t* opt);  // Inherited virtual.
  void FinishTask();         // Inherited virtual.
  bool GetUseHitsInStat() const { return useHitsInStat; }
  void SetUseHitsInStat(bool v) { useHitsInStat = v; }
  bool GetAveragePoints() const { return averagePoints; }
  void SetAveragePoints(bool v) { averagePoints = v; }
  bool GetDontTouchNonPrimary() const { return dontTouchNonPrimary; }
  void SetDontTouchNonPrimary(bool v) { dontTouchNonPrimary = v; }
  bool GetUseChargeSignInCuts() const { return useChargeSignInCuts; }
  void SetUseChargeSignInCuts(bool v) { useChargeSignInCuts = v; }
  bool GetBuildConnectStat() const { return buildConnectStat; }
  void SetBuildConnectStat(bool v) { buildConnectStat = v; }
  bool GetBuildBgrInvMass() const { return buildBgrInvMass; }
  void SetBuildBgrInvMass(bool v) { buildBgrInvMass = v; }
  bool GetBuildSigInvMass() const { return buildSigInvMass; }
  void SetBuildSigInvMass(bool v) { buildSigInvMass = v; }
  bool GetJoinData() const { return joinData; }
  void SetJoinData(bool v) { joinData = v; }
  bool GetBuildNearestHitDist() const { return buildNearestHitDist; }
  void SetBuildNearestHitDist(bool v) { buildNearestHitDist = v; }
  bool GetCropHits() const { return cropHits; }
  void SetCropHits(bool v) { cropHits = v; }
  bool GetBuildSegmentsStat() const { return buildSegmentsStat; }
  void SetBuildSegmentsStat(bool v) { buildSegmentsStat = v; }
  void SetParticleType(TString v) {
    particleType = v;
    segmentsAnalyzer.SetParticleType(v);
  }
  void SetUseBgr(bool v) { segmentsAnalyzer.SetUseBgr(v); }
 
private:
  void Clean();
  void AveragePoints();
  void BuildStatistics();
  void Connect(bool useCuts);
  void Connect(LxSimpleTrack* muchTrack,
               LxSimplePoint muchPt0,
               Double_t txMuch,
               Double_t tyMuch,
               bool useCuts);
 
  TClonesArray* listMCTracks;
  TClonesArray* listStsPts;
  TClonesArray* listMuchPts;
  TClonesArray* listMuchPixelHits;
  TClonesArray* listMuchClusters;
  TClonesArray* listMuchPixelDigiMatches;
  std::vector<LxSimpleTrack*> allTracks;
  std::list<LxSimpleTrack*> posTracks;
  std::list<LxSimpleTrack*> negTracks;
  TTree* superEventTracks;
  LxSimpleTrack superEventBrachTrack;
  bool useHitsInStat;
  bool averagePoints;
  bool dontTouchNonPrimary;
  bool useChargeSignInCuts;
  bool buildConnectStat;
  bool buildBgrInvMass;
  bool buildSigInvMass;
  bool joinData;
  bool buildNearestHitDist;
  bool cropHits;
  bool buildSegmentsStat;
  TString particleType;
  LxTrackAnaSegments segmentsAnalyzer;
 
  friend class LxTrackAnaSegments;
 
  ClassDef(LxTrackAnaTriplet, 1);
};
 
#endif  //LXTRACKANA_INCLUDED