Class Reference for E1039 Core & Analysis Software
AnaDimuonLikeSign.cc
Go to the documentation of this file.
1 #include <fstream>
2 #include <iomanip>
3 #include <TSystem.h>
4 #include <TFile.h>
5 #include <TTree.h>
6 #include <TChain.h>
7 #include <TH1D.h>
8 #include <TCanvas.h>
9 #include <interface_main/SQRun.h>
10 #include <interface_main/SQEvent.h>
11 #include <interface_main/SQHitVector.h>
12 #include <interface_main/SQTrackVector.h>
13 #include <interface_main/SQDimuonVector.h>
14 #include <ktracker/SRecEvent.h>
15 #include <fun4all/Fun4AllReturnCodes.h>
16 #include <phool/PHNodeIterator.h>
17 #include <phool/PHIODataNode.h>
18 #include <phool/getClass.h>
19 #include <geom_svc/GeomSvc.h>
20 //#include <UtilAna/UtilHist.h>
21 #include "AnaDimuonLikeSign.h"
22 using namespace std;
23 
24 AnaDimuonLikeSign::AnaDimuonLikeSign(const std::string& name, const std::string& label)
25  : SubsysReco (name)
26  , m_sq_evt (0)
27  , m_sq_hit_vec(0)
28  , m_sq_trk_vec(0)
29  , m_sq_dim_vec(0)
30  , m_node_prefix("SQRecDimuonVector_")
31  , m_label (label)
32  , m_file_name ("")
33  , m_file (0)
34  , m_tree (0)
35 {
36  ;
37 }
38 
39 AnaDimuonLikeSign::~AnaDimuonLikeSign()
40 {
41  ;
42 }
43 
44 int AnaDimuonLikeSign::Init(PHCompositeNode* topNode)
45 {
46  if (m_label != "PP" && m_label != "MM") {
47  cout << "AnaDimuonLikeSign::Init(): ERROR Label is not 'PP' nor 'MM'. Abort." << endl;
48  return Fun4AllReturnCodes::ABORTEVENT;
49  }
50  m_file_name = "output_" + m_label + ".root";
51 
52  return Fun4AllReturnCodes::EVENT_OK;
53 }
54 
55 int AnaDimuonLikeSign::InitRun(PHCompositeNode* topNode)
56 {
57  //GeomSvc* geom = GeomSvc::instance();
58 
59  string node_name = m_node_prefix + m_label;
60 
61  m_sq_evt = findNode::getClass<SQEvent >(topNode, "SQEvent");
62  m_sq_hit_vec = findNode::getClass<SQHitVector >(topNode, "SQHitVector");
63  m_sq_trk_vec = findNode::getClass<SQTrackVector >(topNode, "SQRecTrackVector");
64  m_sq_dim_vec = findNode::getClass<SQDimuonVector>(topNode, node_name.c_str());
65  //m_srec = findNode::getClass<SRecEvent >(topNode, "SRecEvent");
66  if (!m_sq_evt || !m_sq_hit_vec || !m_sq_trk_vec || !m_sq_dim_vec) return Fun4AllReturnCodes::ABORTEVENT;
67 
68  m_file = new TFile(m_file_name.c_str(), "RECREATE");
69  m_tree = new TTree("tree", "Created by AnaDimuonLikeSign");
70  m_tree->Branch("event" , &m_evt);
71  m_tree->Branch("dimuon_list", &m_dim_list);
72 
73  SQRun* sq_run = findNode::getClass<SQRun>(topNode, "SQRun");
74  if (!sq_run) return Fun4AllReturnCodes::ABORTEVENT;
75  int LBtop = sq_run->get_v1495_id(2);
76  int LBbot = sq_run->get_v1495_id(3);
77  int ret = m_rs.LoadConfig(LBtop, LBbot);
78  if (ret != 0) {
79  cout << "!!WARNING!! OnlMonTrigEP::InitRunOnlMon(): roadset.LoadConfig returned " << ret << ".\n";
80  }
81  cout <<"Roadset " << m_rs.str(1) << endl;
82 
83  return Fun4AllReturnCodes::EVENT_OK;
84 }
85 
86 int AnaDimuonLikeSign::process_event(PHCompositeNode* topNode)
87 {
88  //if (! m_sq_evt->get_trigger(SQEvent::MATRIX1)) {
89  // return Fun4AllReturnCodes::EVENT_OK;
90  //}
91 
92  m_evt.run_id = m_sq_evt->get_run_id();
93  m_evt.spill_id = m_sq_evt->get_spill_id();
94  m_evt.event_id = m_sq_evt->get_event_id();
95  m_evt.fpga_bits = (m_sq_evt->get_trigger() >> SQEvent::MATRIX1) & 0x1f;
96  m_evt.nim_bits = (m_sq_evt->get_trigger() >> SQEvent::NIM1 ) & 0x1f;
97 
98  m_evt.D1 = m_evt.D2 = m_evt.D3p = m_evt.D3m = 0;
99  for (SQHitVector::Iter it = m_sq_hit_vec->begin(); it != m_sq_hit_vec->end(); it++) {
100  SQHit* hit = *it;
101  int det_id = hit->get_detector_id();
102  if ( 0 < det_id && det_id <= 6) m_evt.D1++;
103  else if (12 < det_id && det_id <= 18) m_evt.D2++;
104  else if (18 < det_id && det_id <= 24) m_evt.D3p++;
105  else if (24 < det_id && det_id <= 30) m_evt.D3m++;
106  }
107 
108  //int n_trk = m_srec->getNTracks();
109  //for (int i_trk = 0; i_trk < n_trk; i_trk++) {
110  // SRecTrack strk = m_srec->getTrack(i_trk);
111  // cout << m_evt.event_id
112  // << " " << i_trk
113  // << " " << strk.get_charge()
114  // << " " << strk.get_num_hits()
115  // << " " << strk.get_chisq()
116  // << " " << strk.get_pos_vtx().X()
117  // << " " << strk.get_pos_vtx().Y()
118  // << " " << strk.get_pos_vtx().Z()
119  // << " " << strk.get_mom_vtx().X()
120  // << " " << strk.get_mom_vtx().Y()
121  // << " " << strk.get_mom_vtx().Z()
122  // << endl;
123  //}
124 
125  m_dim_list.clear();
126  for (auto it = m_sq_dim_vec->begin(); it != m_sq_dim_vec->end(); it++) {
127  //SQDimuon* dim = *it;
128  SRecDimuon* dim = dynamic_cast<SRecDimuon*>(*it);
129  int trk_id_pos = dim->get_track_id_pos();
130  int trk_id_neg = dim->get_track_id_neg();
131  SRecTrack* trk_pos = dynamic_cast<SRecTrack*>(m_sq_trk_vec->at(trk_id_pos));
132  SRecTrack* trk_neg = dynamic_cast<SRecTrack*>(m_sq_trk_vec->at(trk_id_neg));
133 
134  int road_pos = trk_pos->getTriggerRoad();
135  int road_neg = trk_neg->getTriggerRoad();
136  bool pos_top = m_rs.PosTop()->FindRoad(road_pos);
137  bool pos_bot = m_rs.PosBot()->FindRoad(road_pos);
138  bool neg_top = m_rs.NegTop()->FindRoad(road_neg);
139  bool neg_bot = m_rs.NegBot()->FindRoad(road_neg);
140  //cout << "T " << road_pos << " " << road_neg << " " << pos_top << pos_bot << neg_top << neg_bot << endl;
141 
142  DimuonData dd;
143  dd.road_pos = road_pos;
144  dd.road_neg = road_neg;
145  dd.pos_top = pos_top;
146  dd.pos_bot = pos_bot;
147  dd.neg_top = neg_top;
148  dd.neg_bot = neg_bot;
149  dd.pos = dim->get_pos();
150  dd.mom = dim->get_mom();
151  dd.n_hits_pos = trk_pos->get_num_hits();
152  dd.chisq_pos = trk_pos->get_chisq();
153  dd.chisq_target_pos = trk_pos->getChisqTarget();//get_chisq_target();
154  dd.chisq_dump_pos = trk_pos->get_chisq_dump();
155  dd.chisq_upstream_pos = trk_pos->get_chsiq_upstream();
156  dd.pos_pos = trk_pos->get_pos_vtx();
157  dd.mom_pos = trk_pos->get_mom_vtx();
158  dd.pos_target_pos = trk_pos->get_pos_target();
159  dd.pos_dump_pos = trk_pos->get_pos_dump();
160  dd.n_hits_neg = trk_neg->get_num_hits();
161  dd.chisq_neg = trk_neg->get_chisq();
162  dd.chisq_target_neg = trk_neg->getChisqTarget();//get_chisq_target();
163  dd.chisq_dump_neg = trk_neg->get_chisq_dump();
164  dd.chisq_upstream_neg = trk_neg->get_chsiq_upstream(); // not chisq
165  dd.pos_neg = trk_neg->get_pos_vtx();
166  dd.mom_neg = trk_neg->get_mom_vtx();
167  dd.pos_target_neg = trk_neg->get_pos_target();
168  dd.pos_dump_neg = trk_neg->get_pos_dump();
169 
170  //sdim.calcVariables(1); // 1 = target
171  dd.mom_target = dim->p_pos_target + dim->p_neg_target; // sdim.get_mom();
172  //sdim.calcVariables(2); // 2 = dump
173  dd.mom_dump = dim->p_pos_dump + dim->p_neg_dump; // sdim.get_mom();
174 
175  m_dim_list.push_back(dd);
176  }
177 
178  m_tree->Fill();
179  return Fun4AllReturnCodes::EVENT_OK;
180 }
181 
182 int AnaDimuonLikeSign::End(PHCompositeNode* topNode)
183 {
184  m_file->cd();
185  m_file->Write();
186  m_file->Close();
187  return Fun4AllReturnCodes::EVENT_OK;
188 }
189 
190 void AnaDimuonLikeSign::AnalyzeTree(TChain* tree, const std::string label)
191 {
192  string dir_out = "result/" + label;
193  cout << "N of trees = " << tree->GetNtrees() << endl;
194  gSystem->mkdir(dir_out.c_str(), true);
195  ofstream ofs(dir_out + "/result.txt");
196 
197  TFile* file_out = new TFile((dir_out+"/result.root").c_str(), "RECREATE");
198 
199  TH1* h1_D1 = new TH1D("h1_D1" , ";D1 occupancy;N of events", 500, -0.5, 499.5);
200  TH1* h1_D2 = new TH1D("h1_D2" , ";D2 occupancy;N of events", 300, -0.5, 299.5);
201  TH1* h1_D3p = new TH1D("h1_D3p", ";D3p occupancy;N of events", 300, -0.5, 299.5);
202  TH1* h1_D3m = new TH1D("h1_D3m", ";D3m occupancy;N of events", 300, -0.5, 299.5);
203 
204  TH1* h1_nhit_pos = new TH1D("h1_nhit_pos", "#mu^{+};N of hits/track;", 6, 12.5, 18.5);
205  TH1* h1_chi2_pos = new TH1D("h1_chi2_pos", "#mu^{+};Track #chi^{2};", 100, 0, 2);
206  TH1* h1_z_pos = new TH1D("h1_z_pos" , "#mu^{+};Track z (cm);" , 100, -700, 300);
207  TH1* h1_pz_pos = new TH1D("h1_pz_pos" , "#mu^{+};Track p_{z} (GeV);", 100, 0, 100);
208  //TH1* h1_x_t_pos = new TH1D("h1_x_t_pos", "#mu^{+};Track x (cm) @ Target;", 100, -50, 50);
209  //TH1* h1_y_t_pos = new TH1D("h1_y_t_pos", "#mu^{+};Track y (cm) @ Target;", 100, -25, 25);
210  //TH1* h1_x_d_pos = new TH1D("h1_x_d_pos", "#mu^{+};Track x (cm) @ Dump;" , 100, -10, 10);
211  //TH1* h1_y_d_pos = new TH1D("h1_y_d_pos", "#mu^{+};Track y (cm) @ Dump;" , 100, -10, 10);
212 
213  TH1* h1_chi2_tgt_pos = new TH1D("h1_chi2_tgt_pos", "#mu^{+};Track #chi^{2} at target;" , 100, 0, 10);
214  TH1* h1_chi2_dum_pos = new TH1D("h1_chi2_dum_pos", "#mu^{+};Track #chi^{2} at dump;" , 100, 0, 10);
215  TH1* h1_chi2_ups_pos = new TH1D("h1_chi2_ups_pos", "#mu^{+};Track #chi^{2} at upstream;", 100, 0, 10);
216  TH1* h1_chi2_tmd_pos = new TH1D("h1_chi2_tmd_pos", "#mu^{+};#chi^{2}_{Target} - #chi^{2}_{Dump};" , 100, -10, 10);
217  TH1* h1_chi2_tmu_pos = new TH1D("h1_chi2_tmu_pos", "#mu^{+};#chi^{2}_{Target} - #chi^{2}_{Upstream};", 100, -10, 10);
218 
219  TH1* h1_nhit_neg = new TH1D("h1_nhit_neg", "#mu^{-};N of hits/track;", 6, 12.5, 18.5);
220  TH1* h1_chi2_neg = new TH1D("h1_chi2_neg", "#mu^{-};Track #chi^{2};", 100, 0, 2);
221  TH1* h1_z_neg = new TH1D("h1_z_neg" , "#mu^{-};Track z (cm);", 100, -700, 300);
222  TH1* h1_pz_neg = new TH1D("h1_pz_neg" , "#mu^{-};Track p_{z} (GeV);", 100, 0, 100);
223  //TH1* h1_x_t_neg = new TH1D("h1_x_t_neg", "#mu^{-};Track x (cm) @ Target;", 100, -50, 50);
224  //TH1* h1_y_t_neg = new TH1D("h1_y_t_neg", "#mu^{-};Track y (cm) @ Target;", 100, -25, 25);
225  //TH1* h1_x_d_neg = new TH1D("h1_x_d_neg", "#mu^{-};Track x (cm) @ Dump;" , 100, -10, 10);
226  //TH1* h1_y_d_neg = new TH1D("h1_y_d_neg", "#mu^{-};Track y (cm) @ Dump;" , 100, -10, 10);
227 
228  TH1* h1_chi2_tgt_neg = new TH1D("h1_chi2_tgt_neg", "#mu^{-};Track #chi^{2} at target;" , 100, 0, 10);
229  TH1* h1_chi2_dum_neg = new TH1D("h1_chi2_dum_neg", "#mu^{-};Track #chi^{2} at dump;" , 100, 0, 10);
230  TH1* h1_chi2_ups_neg = new TH1D("h1_chi2_ups_neg", "#mu^{-};Track #chi^{2} at upstream;", 100, 0, 10);
231  TH1* h1_chi2_tmd_neg = new TH1D("h1_chi2_tmd_neg", "#mu^{-};#chi^{2}_{Target} - #chi^{2}_{Dump};" , 100, -10, 10);
232  TH1* h1_chi2_tmu_neg = new TH1D("h1_chi2_tmu_neg", "#mu^{-};#chi^{2}_{Target} - #chi^{2}_{Upstream};", 100, -10, 10);
233 
234  TH1* h1_dx = new TH1D("h1_dx" , ";Dimuon x (cm);", 100, -1, 1);
235  TH1* h1_dy = new TH1D("h1_dy" , ";Dimuon y (cm);", 100, -1, 1);
236  TH1* h1_dz = new TH1D("h1_dz" , ";Dimuon z (cm);", 100, -700, 300);
237  TH1* h1_dpx = new TH1D("h1_dpx", ";Dimuon p_{x} (GeV);", 100, -5, 5);
238  TH1* h1_dpy = new TH1D("h1_dpy", ";Dimuon p_{y} (GeV);", 100, -5, 5);
239  TH1* h1_dpz = new TH1D("h1_dpz", ";Dimuon p_{z} (GeV);", 100, 30, 130);
240  TH1* h1_m = new TH1D("h1_m" , ";Dimuon mass (GeV);", 100, 0, 10);
241  TH1* h1_trk_sep = new TH1D("h1_trk_sep", ";Track separation: z_{#mu +} - z_{#mu -} (cm);", 100, -500, 500);
242 
243  TH1* h1_dz_sel = new TH1D("h1_dz_sel" , ";Dimuon z (cm);", 100, -700, 300);
244  TH1* h1_dpz_sel = new TH1D("h1_dpz_sel", ";Dimuon p_{z} (GeV);", 100, 30, 130);
245  TH1* h1_m_sel = new TH1D("h1_m_sel" , ";Dimuon mass (GeV);", 100, 0, 10);
246 
247  TH1* h1_dz_tgt = new TH1D("h1_dz_tgt" , ";Dimuon z (cm);" , 100, -700, 300);
248  TH1* h1_dpz_tgt = new TH1D("h1_dpz_tgt", ";Dimuon p_{z} (GeV);", 100, 30, 130);
249  TH1* h1_m_tgt = new TH1D("h1_m_tgt" , ";Dimuon mass (GeV);" , 100, 0, 10);
250 
251  //GeomSvc* geom = GeomSvc::instance();
252  ostringstream oss;
253 
254  EventData* evt = 0;
255  DimuonList* dim_list = 0;
256  tree->SetBranchAddress("event" , &evt);
257  tree->SetBranchAddress("dimuon_list", &dim_list);
258 
259  int n_ent = tree->GetEntries();
260  cout << "N of entries = " << n_ent << endl;
261  for (int i_ent = 0; i_ent < n_ent; i_ent++) {
262  if ((i_ent+1) % (n_ent/10) == 0) cout << " " << 10*(i_ent+1)/(n_ent/10) << "%" << flush;
263  tree->GetEntry(i_ent);
264  //ofs << evt->run_id << " " << evt->spill_id << " " << evt->event_id << " " << evt->D1 << " " << evt->D2 << " " << evt->D3p << " " << evt->D3m << endl;
265 
266  //if (evt->run_id != 6155 || evt->spill_id != 1941910) continue;
267  //if (! (evt->fpga_bits & 0x1)) continue;
268  if (! (evt->fpga_bits & 0x4)) continue;
269  //if (! (evt->nim_bits & 0x4)) continue;
270 
271  h1_D1 ->Fill(evt->D1 );
272  h1_D2 ->Fill(evt->D2 );
273  h1_D3p->Fill(evt->D3p);
274  h1_D3m->Fill(evt->D3m);
275  //if (evt->D1 > 120 || evt->D2 > 60 || evt->D3p > 50 || evt->D3m > 50) continue;
276 
277  for (auto it = dim_list->begin(); it != dim_list->end(); it++) {
278  DimuonData* dd = &(*it);
279 
280  double trk_sep = dd->pos_pos.Z() - dd->pos_neg.Z();
281  double chi2_tgt_pos = dd->chisq_target_pos;
282  double chi2_dum_pos = dd->chisq_dump_pos;
283  double chi2_ups_pos = dd->chisq_upstream_pos;
284  double chi2_tgt_neg = dd->chisq_target_neg;
285  double chi2_dum_neg = dd->chisq_dump_neg;
286  double chi2_ups_neg = dd->chisq_upstream_neg;
287 
288  ofs << evt->run_id << " " << evt->spill_id << " " << evt->event_id << " "
289  << evt->D1 << " " << evt->D2 << " " << evt->D3p << " " << evt->D3m << " "
290  << dd->pos.Z() << " " << dd->mom.M() << endl;
291  //ofs << chi2_tgt_pos << " " << chi2_dum_pos << " " << chi2_ups_pos << " " << chi2_tgt_neg << " " << chi2_dum_neg << " " << chi2_ups_neg << endl;
292 
293  //if (dd->n_hits_pos < 14 || dd->n_hits_neg < 14) continue;
294 
295  h1_nhit_pos->Fill(dd->n_hits_pos);
296  h1_chi2_pos->Fill(dd->chisq_pos);
297  h1_z_pos ->Fill(dd->pos_pos.Z());
298  h1_pz_pos ->Fill(dd->mom_pos.Z());
299 
300  h1_nhit_neg->Fill(dd->n_hits_neg);
301  h1_chi2_neg->Fill(dd->chisq_neg);
302  h1_z_neg ->Fill(dd->pos_neg.Z());
303  h1_pz_neg ->Fill(dd->mom_neg.Z());
304 
305  if (dd->pos_pos.Z() < -690 || dd->pos_neg.Z() < -690) continue;
306  //if (dd->n_hits_pos < 15 || dd->n_hits_neg < 15) continue;
307  //if (fabs(trk_sep) > 200) continue;
308 
309  //bool top_bot = dd->pos_top && dd->neg_bot;
310  //bool bot_top = dd->pos_bot && dd->neg_top;
311  //if (!top_bot && !bot_top) continue;
312 
313  h1_chi2_tgt_pos->Fill(chi2_tgt_pos);
314  h1_chi2_dum_pos->Fill(chi2_dum_pos);
315  h1_chi2_ups_pos->Fill(chi2_ups_pos);
316  h1_chi2_tmd_pos->Fill(chi2_tgt_pos - chi2_dum_pos);
317  h1_chi2_tmu_pos->Fill(chi2_tgt_pos - chi2_ups_pos);
318  //double x_t_pos = dd->pos_target_pos.X();
319  //double y_t_pos = dd->pos_target_pos.Y();
320  //double x_d_pos = dd->pos_dump_pos.X();
321  //double y_d_pos = dd->pos_dump_pos.Y();
322  //h1_x_t_pos->Fill(x_t_pos);
323  //h1_y_t_pos->Fill(y_t_pos);
324  //h1_x_d_pos->Fill(x_d_pos);
325  //h1_y_d_pos->Fill(y_d_pos);
326 
327  h1_chi2_tgt_neg->Fill(chi2_tgt_neg);
328  h1_chi2_dum_neg->Fill(chi2_dum_neg);
329  h1_chi2_ups_neg->Fill(chi2_ups_neg);
330  h1_chi2_tmd_neg->Fill(chi2_tgt_neg - chi2_dum_neg);
331  h1_chi2_tmu_neg->Fill(chi2_tgt_neg - chi2_ups_neg);
332  //double x_t_neg = dd->pos_target_neg.X();
333  //double y_t_neg = dd->pos_target_neg.Y();
334  //double x_d_neg = dd->pos_dump_neg.X();
335  //double y_d_neg = dd->pos_dump_neg.Y();
336  //h1_x_t_neg->Fill(x_t_neg);
337  //h1_y_t_neg->Fill(y_t_neg);
338  //h1_x_d_neg->Fill(x_d_neg);
339  //h1_y_d_neg->Fill(y_d_neg);
340 
341  h1_dx ->Fill(dd->pos.X());
342  h1_dy ->Fill(dd->pos.Y());
343  h1_dz ->Fill(dd->pos.Z());
344  h1_dpx ->Fill(dd->mom.X());
345  h1_dpy ->Fill(dd->mom.Y());
346  h1_dpz ->Fill(dd->mom.Z());
347  h1_m ->Fill(dd->mom.M());
348  h1_trk_sep->Fill(trk_sep);
349 
350  if (chi2_tgt_pos < 0 || chi2_dum_pos < 0 || chi2_ups_pos < 0 ||
351  chi2_tgt_pos - chi2_dum_pos > 0 || chi2_tgt_pos - chi2_ups_pos > 0) continue;
352  if (chi2_tgt_neg < 0 || chi2_dum_neg < 0 || chi2_ups_neg < 0 ||
353  chi2_tgt_neg - chi2_dum_neg > 0 || chi2_tgt_neg - chi2_ups_neg > 0) continue;
354 
355  //double r_t_pos = sqrt(x_t_pos*x_t_pos + y_t_pos*y_t_pos);
356  //double r_d_pos = sqrt(x_d_pos*x_d_pos + y_d_pos*y_d_pos);
357  //double r_t_neg = sqrt(x_t_neg*x_t_neg + y_t_neg*y_t_neg);
358  //double r_d_neg = sqrt(x_d_neg*x_d_neg + y_d_neg*y_d_neg);
359  //if (r_t_pos >= r_d_pos || r_t_neg >= r_d_neg) continue;
360 
361  h1_dz_sel ->Fill(dd->pos.Z());
362  h1_dpz_sel->Fill(dd->mom.Z());
363  h1_m_sel ->Fill(dd->mom.M());
364 
365  h1_dz_tgt ->Fill(dd->pos.Z());
366  h1_dpz_tgt->Fill(dd->mom_target.Z());
367  h1_m_tgt ->Fill(dd->mom_target.M());
368  }
369  }
370 
371  TCanvas* c1 = new TCanvas("c1", "");
372  c1->SetGrid();
373  //c1->SetLogy(true);
374 
375  h1_D1 ->Draw(); c1->SaveAs((dir_out+"/h1_D1.png").c_str());
376  h1_D2 ->Draw(); c1->SaveAs((dir_out+"/h1_D2.png").c_str());
377  h1_D3p->Draw(); c1->SaveAs((dir_out+"/h1_D3p.png").c_str());
378  h1_D3m->Draw(); c1->SaveAs((dir_out+"/h1_D3m.png").c_str());
379 
380  h1_nhit_pos->Draw(); c1->SaveAs((dir_out+"/h1_nhit_pos.png").c_str());
381  h1_chi2_pos->Draw(); c1->SaveAs((dir_out+"/h1_chi2_pos.png").c_str());
382  h1_z_pos ->Draw(); c1->SaveAs((dir_out+"/h1_z_pos.png").c_str());
383  h1_pz_pos ->Draw(); c1->SaveAs((dir_out+"/h1_pz_pos.png").c_str());
384  //h1_x_t_pos ->Draw(); c1->SaveAs(dir_out+"/h1_x_t_pos.png");
385  //h1_y_t_pos ->Draw(); c1->SaveAs(dir_out+"/h1_y_t_pos.png");
386  //h1_x_d_pos ->Draw(); c1->SaveAs(dir_out+"/h1_x_d_pos.png");
387  //h1_y_d_pos ->Draw(); c1->SaveAs(dir_out+"/h1_y_d_pos.png");
388 
389  h1_chi2_tgt_pos->Draw(); c1->SaveAs((dir_out+"/h1_chi2_tgt_pos.png").c_str());
390  h1_chi2_dum_pos->Draw(); c1->SaveAs((dir_out+"/h1_chi2_dum_pos.png").c_str());
391  h1_chi2_ups_pos->Draw(); c1->SaveAs((dir_out+"/h1_chi2_ups_pos.png").c_str());
392  h1_chi2_tmd_pos->Draw(); c1->SaveAs((dir_out+"/h1_chi2_tmd_pos.png").c_str());
393  h1_chi2_tmu_pos->Draw(); c1->SaveAs((dir_out+"/h1_chi2_tmu_pos.png").c_str());
394 
395  h1_nhit_neg->Draw(); c1->SaveAs((dir_out+"/h1_nhit_neg.png").c_str());
396  h1_chi2_neg->Draw(); c1->SaveAs((dir_out+"/h1_chi2_neg.png").c_str());
397  h1_z_neg ->Draw(); c1->SaveAs((dir_out+"/h1_z_neg.png").c_str());
398  h1_pz_neg ->Draw(); c1->SaveAs((dir_out+"/h1_pz_neg.png").c_str());
399  //h1_x_t_neg ->Draw(); c1->SaveAs(dir_out+"/h1_x_t_neg.png");
400  //h1_y_t_neg ->Draw(); c1->SaveAs(dir_out+"/h1_y_t_neg.png");
401  //h1_x_d_neg ->Draw(); c1->SaveAs(dir_out+"/h1_x_d_neg.png");
402  //h1_y_d_neg ->Draw(); c1->SaveAs(dir_out+"/h1_y_d_neg.png");
403 
404  h1_chi2_tgt_neg->Draw(); c1->SaveAs((dir_out+"/h1_chi2_tgt_neg.png").c_str());
405  h1_chi2_dum_neg->Draw(); c1->SaveAs((dir_out+"/h1_chi2_dum_neg.png").c_str());
406  h1_chi2_ups_neg->Draw(); c1->SaveAs((dir_out+"/h1_chi2_ups_neg.png").c_str());
407  h1_chi2_tmd_neg->Draw(); c1->SaveAs((dir_out+"/h1_chi2_tmd_neg.png").c_str());
408  h1_chi2_tmu_neg->Draw(); c1->SaveAs((dir_out+"/h1_chi2_tmu_neg.png").c_str());
409 
410  h1_dx ->Draw(); c1->SaveAs((dir_out+"/h1_dx.png" ).c_str());
411  h1_dy ->Draw(); c1->SaveAs((dir_out+"/h1_dy.png" ).c_str());
412  h1_dz ->Draw(); c1->SaveAs((dir_out+"/h1_dz.png" ).c_str());
413  h1_dpx ->Draw(); c1->SaveAs((dir_out+"/h1_dpx.png" ).c_str());
414  h1_dpy ->Draw(); c1->SaveAs((dir_out+"/h1_dpy.png" ).c_str());
415  h1_dpz ->Draw(); c1->SaveAs((dir_out+"/h1_dpz.png" ).c_str());
416  h1_m ->Draw(); c1->SaveAs((dir_out+"/h1_m.png" ).c_str());
417  h1_trk_sep->Draw(); c1->SaveAs((dir_out+"/h1_trk_sep.png").c_str());
418 
419  //c1->SetLogy(true);
420 
421  h1_dz_sel->SetLineColor(kRed);
422  h1_dz_sel->SetLineWidth(2);
423  h1_dz_sel->Draw();
424  c1->SaveAs((dir_out+"/h1_dz_sel.png").c_str());
425 
426  h1_dpz_sel->SetLineColor(kRed);
427  h1_dpz_sel->SetLineWidth(2);
428  h1_dpz_sel->Draw();
429  c1->SaveAs((dir_out+"/h1_dpz_sel.png").c_str());
430 
431  h1_m_sel ->SetLineColor(kRed);
432  h1_m_sel ->SetLineWidth(2);
433  h1_m_sel->Draw();
434  c1->SaveAs((dir_out+"/h1_m_sel.png").c_str());
435 
436  h1_dz_tgt->SetLineColor(kBlue);
437  h1_dz_tgt->SetLineWidth(2);
438  h1_dz_tgt->Draw();
439  c1->SaveAs((dir_out+"/h1_dz_tgt.png").c_str());
440 
441  h1_dpz_tgt->SetLineColor(kBlue);
442  h1_dpz_tgt->SetLineWidth(2);
443  h1_dpz_tgt->Draw();
444  c1->SaveAs((dir_out+"/h1_dpz_tgt.png").c_str());
445 
446  h1_m_tgt ->SetLineColor(kBlue);
447  h1_m_tgt ->SetLineWidth(2);
448  h1_m_tgt->Draw();
449  c1->SaveAs((dir_out+"/h1_m_tgt.png").c_str());
450 
451  delete c1;
452 
453  ofs.close();
454  file_out->Write();
455  file_out->Close();
456 }
DimuonList
std::vector< DimuonData > DimuonList
Definition: TreeData.h:54
AnaDimuonLikeSign::AnalyzeTree
static void AnalyzeTree(TChain *tree, const std::string label)
Definition: AnaDimuonLikeSign.cc:190
AnaDimuonLikeSign::AnaDimuonLikeSign
AnaDimuonLikeSign(const std::string &name="AnaDimuonLikeSign", const std::string &label="PP")
Definition: AnaDimuonLikeSign.cc:24
AnaDimuonLikeSign::InitRun
int InitRun(PHCompositeNode *topNode)
Definition: AnaDimuonLikeSign.cc:55
AnaDimuonLikeSign::~AnaDimuonLikeSign
virtual ~AnaDimuonLikeSign()
Definition: AnaDimuonLikeSign.cc:39
AnaDimuonLikeSign::Init
int Init(PHCompositeNode *topNode)
Definition: AnaDimuonLikeSign.cc:44
AnaDimuonLikeSign::End
int End(PHCompositeNode *topNode)
Called at the end of all processing.
Definition: AnaDimuonLikeSign.cc:182
AnaDimuonLikeSign::process_event
int process_event(PHCompositeNode *topNode)
Definition: AnaDimuonLikeSign.cc:86
SQDimuonVector::begin
virtual ConstIter begin() const =0
SQDimuonVector::end
virtual ConstIter end() const =0
SQEvent::get_run_id
virtual int get_run_id() const =0
Return the run ID.
SQEvent::MATRIX1
@ MATRIX1
Definition: SQEvent.h:27
SQEvent::NIM1
@ NIM1
Definition: SQEvent.h:22
SQEvent::get_trigger
virtual bool get_trigger(const SQEvent::TriggerMask i) const =0
Return the trigger bit (fired or not) of the selected trigger channel.
SQEvent::get_spill_id
virtual int get_spill_id() const =0
Return the spill ID.
SQEvent::get_event_id
virtual int get_event_id() const =0
Return the event ID, which is unique per run.
SQHitVector::end
virtual ConstIter end() const =0
SQHitVector::begin
virtual ConstIter begin() const =0
SQHitVector::Iter
std::vector< SQHit * >::iterator Iter
Definition: SQHitVector.h:38
SQHit
An SQ interface class to hold one detector hit.
Definition: SQHit.h:20
SQHit::get_detector_id
virtual short get_detector_id() const
Return the detector ID of this hit.
Definition: SQHit.h:42
SQRun
An SQ interface class to hold the run-level info.
Definition: SQRun.h:18
SQRun::get_v1495_id
virtual int get_v1495_id(const int chan) const
Return the ID of the chan-th V1495 boards.
Definition: SQRun.h:53
SQTrackVector::at
virtual const SQTrack * at(const size_t id) const =0
SRecDimuon::p_neg_target
TLorentzVector p_neg_target
Definition: SRecEvent.h:370
SRecDimuon::get_pos
virtual TVector3 get_pos() const
Return the dimuon position at vertex.
Definition: SRecEvent.h:326
SRecDimuon::p_pos_target
TLorentzVector p_pos_target
Track momentum projections at different location.
Definition: SRecEvent.h:369
SRecDimuon::get_mom
virtual TLorentzVector get_mom() const
Return the dimuon momentum at vertex.
Definition: SRecEvent.h:329
SRecDimuon::get_track_id_neg
virtual int get_track_id_neg() const
Return the track ID of the negative track.
Definition: SRecEvent.h:323
SRecDimuon::get_track_id_pos
virtual int get_track_id_pos() const
Return the track ID of the positive track.
Definition: SRecEvent.h:320
SRecDimuon::p_pos_dump
TLorentzVector p_pos_dump
Definition: SRecEvent.h:371
SRecDimuon::p_neg_dump
TLorentzVector p_neg_dump
Definition: SRecEvent.h:372
SRecTrack::getChisqTarget
Double_t getChisqTarget()
Definition: SRecEvent.h:199
SRecTrack::getTriggerRoad
Int_t getTriggerRoad()
Definition: SRecEvent.h:222
SRecTrack::get_chsiq_upstream
virtual double get_chsiq_upstream() const
Definition: SRecEvent.h:87
SRecTrack::get_pos_dump
virtual TVector3 get_pos_dump() const
Definition: SRecEvent.h:91
SRecTrack::get_num_hits
virtual int get_num_hits() const
Return the number of hits associated to this track.
Definition: SRecEvent.h:63
SRecTrack::get_pos_target
virtual TVector3 get_pos_target() const
Definition: SRecEvent.h:90
SRecTrack::get_chisq_dump
virtual double get_chisq_dump() const
Definition: SRecEvent.h:86
SRecTrack::get_mom_vtx
virtual TLorentzVector get_mom_vtx() const
Return the track momentum at vertex.
Definition: SRecEvent.h:75
SRecTrack::get_chisq
virtual double get_chisq() const
Definition: SRecEvent.h:84
SRecTrack::get_pos_vtx
virtual TVector3 get_pos_vtx() const
Return the track position at vertex.
Definition: SRecEvent.h:66
UtilTrigger::TrigRoads::FindRoad
TrigRoad * FindRoad(const int road_id)
Definition: TrigRoads.cc:24
UtilTrigger::TrigRoadset::LoadConfig
int LoadConfig(const std::string dir)
Definition: TrigRoadset.cc:23
UtilTrigger::TrigRoadset::NegBot
TrigRoads * NegBot()
Definition: TrigRoadset.h:55
UtilTrigger::TrigRoadset::str
std::string str(const int level=0) const
Definition: TrigRoadset.cc:79
UtilTrigger::TrigRoadset::NegTop
TrigRoads * NegTop()
Definition: TrigRoadset.h:54
UtilTrigger::TrigRoadset::PosTop
TrigRoads * PosTop()
Definition: TrigRoadset.h:52
UtilTrigger::TrigRoadset::PosBot
TrigRoads * PosBot()
Definition: TrigRoadset.h:53
DimuonData::mom_pos
TLorentzVector mom_pos
Definition: TreeData.h:37
DimuonData::chisq_upstream_neg
double chisq_upstream_neg
Definition: TreeData.h:45
DimuonData::mom
TLorentzVector mom
Definition: TreeData.h:36
DimuonData::pos_target_pos
TVector3 pos_target_pos
Definition: TreeData.h:39
DimuonData::chisq_dump_pos
double chisq_dump_pos
Definition: TreeData.h:35
DimuonData::neg_bot
bool neg_bot
Definition: TreeData.h:29
DimuonData::chisq_pos
double chisq_pos
Definition: TreeData.h:33
DimuonData::pos_top
bool pos_top
Definition: TreeData.h:26
DimuonData::mom_target
TLorentzVector mom_target
Dimuon momentum with choice = 1.
Definition: TreeData.h:50
DimuonData::chisq_upstream_pos
double chisq_upstream_pos
Definition: TreeData.h:36
DimuonData::mom_dump
TLorentzVector mom_dump
Dimuon momentum with choice = 2.
Definition: TreeData.h:51
DimuonData::n_hits_neg
short n_hits_neg
Definition: TreeData.h:41
DimuonData::mom_neg
TLorentzVector mom_neg
Definition: TreeData.h:38
DimuonData::pos_bot
bool pos_bot
Definition: TreeData.h:27
DimuonData::pos_target_neg
TVector3 pos_target_neg
Definition: TreeData.h:48
DimuonData::pos_dump_neg
TVector3 pos_dump_neg
Definition: TreeData.h:49
DimuonData::chisq_neg
double chisq_neg
Definition: TreeData.h:42
DimuonData::chisq_dump_neg
double chisq_dump_neg
Definition: TreeData.h:44
DimuonData::chisq_target_neg
double chisq_target_neg
Definition: TreeData.h:43
DimuonData::pos_dump_pos
TVector3 pos_dump_pos
Definition: TreeData.h:40
DimuonData::neg_top
bool neg_top
Definition: TreeData.h:28
DimuonData::n_hits_pos
short n_hits_pos
Definition: TreeData.h:32
DimuonData::pos
TVector3 pos
Definition: TreeData.h:35
DimuonData::pos_pos
TVector3 pos_pos
Definition: TreeData.h:37
DimuonData::pos_neg
TVector3 pos_neg
Definition: TreeData.h:46
DimuonData::road_pos
int road_pos
Definition: TreeData.h:24
DimuonData::chisq_target_pos
double chisq_target_pos
Definition: TreeData.h:34
DimuonData::road_neg
int road_neg
Definition: TreeData.h:25
EventData::spill_id
int spill_id
Definition: TreeData.h:9
EventData::event_id
int event_id
Definition: TreeData.h:10
EventData::nim_bits
short nim_bits
Definition: TreeData.h:12
EventData::D1
short D1
Definition: TreeData.h:13
EventData::D2
short D2
Definition: TreeData.h:14
EventData::fpga_bits
short fpga_bits
Definition: TreeData.h:11
EventData::run_id
short run_id
Definition: TreeData.h:8
EventData::D3p
short D3p
Definition: TreeData.h:15
EventData::D3m
short D3m
Definition: TreeData.h:16