TrackML_CombinatorialTracker/TrackletConstructor.h at master · sgorbuno/TrackML_CombinatorialTracker · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
#ifndef TRACKLETCONSTRUCTOR_H
#define TRACKLETCONSTRUCTOR_H

#include "util.h"
#include "TString.h"
#include "TFile.h"
#include "TH1.h"
#include "TNtuple.h"
#include <iostream>
#include "TrackModelPhysical.h"

struct Tracker;
struct FitLayerHit;

#include "Cuts.h"

using namespace std;

struct Tracklet
{
  vector<int> vHits;
  int nLayers;
  double chi2;
  int isStopped;
  double pt;
  bool vtxTrack;
};


struct TrackletConstructor
{

  void ReconstructMC( const CreationCuts &crCuts, const ProlongationCuts &prCuts );
  void Reconstruct( const CreationCuts &crCuts, const ProlongationCuts &prCuts );

  void ReconstructPrim  ( const CreationPrimCuts &crCuts, const ProlongationCuts &prCuts );
  void ReconstructPrimMC( const CreationPrimCuts &crCuts, const ProlongationCuts &prCuts );

  void ProlongateTracklet( int layer1, FitLayerHit h01,
			   int layer2, FitLayerHit h02,
			   int layer3, FitLayerHit h03,
			   TrackModelPhysical t0,
			   vector<Tracklet> &tracklets );

  void CleanTracklets( int minTrackNLayers );


  void TrackletFitTest( const CreationCuts &crCuts );
  void TrackletFitTest( const CreationPrimCuts &crCuts );
  void TrackletFitTest( int layer1, int layer2, int layer3 );


  void TrackletEfficiency( int useBaseLayers, const double ptCutMin, const double ptCutMax );

  Tracker *mTracker;
  CreationCuts mCreationCuts;
  CreationPrimCuts mCreationPrimCuts;
  ProlongationCuts mProlongationCuts;

  vector<Tracklet> vTracklets;
  vector<Tracklet> vTracks;
};

#endif