AnalysisTreeCuts/HadesCuts.h at master · HeavyIonAnalysis/AnalysisTreeCuts · 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
66
67
68
69
70
71
72
73
74
75
76
77
#include "AnalysisTree/Cuts.h"
#include "AnalysisTree/SimpleCut.h"

namespace AnalysisTree{

Cuts* GetHadesEventCuts(std::string branch, std::string name="HadesGoodEvent") {
  Cuts* event_cuts = new Cuts(std::move(name), std::move(branch));
  SimpleCut vtx_xy_cut{{"vtx_x", "vtx_y"}, [](std::vector<double> r){
                         return sqrt(r.at(0)*r.at(0) + r.at(1)*r.at(1)) < 3.0;
                       }};
  SimpleCut vtx_z_cut("vtx_z", -60.0, 0.0);
  SimpleCut vtx_chi2_cut("vtx_chi2", 0.5, 40);
  SimpleCut good_vertex_cut("good_vertex_cluster", 0.5, 1.5);
  SimpleCut good_vertex_candidate_cut("good_vertex_candidate", 0.5, 1.5);
  SimpleCut good_start_cut("good_start", 0.5, 1.5);
  SimpleCut no_pile_up_start_cut("no_pile_up_start", 0.5, 1.5);
  SimpleCut no_veto_cut("no_veto", 0.5, 1.5);
  SimpleCut good_start_veto_cut("good_start_veto", 0.5, 1.5);
  SimpleCut good_start_meta_cut("good_start_meta", 0.5, 1.5);
  event_cuts->AddCuts({
      vtx_xy_cut,
      vtx_z_cut,
      vtx_chi2_cut,
      good_vertex_cut,
      good_vertex_candidate_cut,
      good_start_cut,
      no_pile_up_start_cut,
      no_veto_cut,
      good_start_veto_cut,
      good_start_meta_cut
  } );
  return event_cuts;
}

Cuts* GetHadesTrackCuts(std::string branch, std::string name="HadesGoodVertexTrack") {
  Cuts* vertex_tracks_cuts = new Cuts(std::move(name), std::move(branch));
  SimpleCut dca_xy_cut("dca_xy", 0, 10);
  SimpleCut dca_z_cut("dca_z", 0, 10);
  SimpleCut chi2_rk_cut("chi2", 0, 100);

  vertex_tracks_cuts->AddCuts({
      dca_xy_cut,
      dca_z_cut,
      chi2_rk_cut,
  });
  return vertex_tracks_cuts;
};

Cuts* GetHadesMetaHitsCuts(std::string branch, std::string name="HadesGoodMetaHit") {
  Cuts* tof_cuts = new Cuts(std::move(name), std::move(branch));
  SimpleCut meta_match_chi2_cut("match_quality", 0, 3.0);

  tof_cuts->AddCuts({meta_match_chi2_cut});
  return tof_cuts;
};
Cuts* GetHadesWallHitsCuts(std::string branch, std::string name="HadesGoodWallHit") {
  Cuts* wall_cuts = new Cuts(std::move(name), std::move(branch));
  SimpleCut beta_cut("beta", 0, 1.0);
  SimpleCut ring_by_ring_cuts({"ring", "beta", "signal"},
                              []( std::vector<double> vars ){
                                if( vars.at(0) <= 5.0 )
                                  return vars.at(1) > 0.84
                                      && vars.at(2) > 80.0;
                                if( vars.at(0) >= 6.0 && vars.at(0) <= 7.0 )
                                  return vars.at(1) > 0.85
                                      && vars.at(2) > 85.0;
                                if( vars.at(0) >= 8.0)
                                  return vars.at(1) > 0.80
                                      && vars.at(2) > 88.0;
                                return false;
                              });

  wall_cuts->AddCuts({beta_cut, ring_by_ring_cuts});
  return wall_cuts;
};

}