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