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