| 1 | /** | 
| 2 | * \file TrkProcess.cpp | 
| 3 | * \author Elena Vannuccini | 
| 4 | */ | 
| 5 | #include <TrkProcess.h> | 
| 6 | #include <iostream> | 
| 7 | using namespace std; | 
| 8 | /** | 
| 9 | * Create TrkProcess object, initializing the tracker processing variables at default values | 
| 10 | */ | 
| 11 | TrkProcess::TrkProcess(){ | 
| 12 |  | 
| 13 | idrun = 0; | 
| 14 | //    DEBUG = false; | 
| 15 | //    VERBOSE = false; | 
| 16 | dbg_mode.SetNone(); | 
| 17 | get1 = 0; | 
| 18 | get2 = 1; | 
| 19 | standalone = false; | 
| 20 | frame2 = "root"; | 
| 21 | frame1 = "hbook"; | 
| 22 | outdir =  gSystem->WorkingDirectory(); | 
| 23 | pfolder = "/TrackerFolder"; | 
| 24 | if      (!frame2.CompareTo("root", TString::kIgnoreCase))  ifroot2 = true; | 
| 25 | else if (!frame2.CompareTo("hbook", TString::kIgnoreCase)) ifroot2 = false; | 
| 26 |  | 
| 27 | if      (!frame1.CompareTo("root", TString::kIgnoreCase))  ifroot1 = true; | 
| 28 | else if (!frame1.CompareTo("hbook", TString::kIgnoreCase)) ifroot1 = false; | 
| 29 |  | 
| 30 | file1 = ""; | 
| 31 | file2 = ""; | 
| 32 |  | 
| 33 | ostatus = 0; | 
| 34 | }; | 
| 35 |  | 
| 36 | /** | 
| 37 | * Create TrkProcess object, initializing the tracker processing variables according to input parameters. | 
| 38 | * @param run Run ID | 
| 39 | * @param f2 Pointer to output level2 file | 
| 40 | */ | 
| 41 | TrkProcess::TrkProcess(ULong64_t run, TFile *f2){ | 
| 42 |  | 
| 43 | idrun = run; | 
| 44 | //    DEBUG = false; | 
| 45 | //    VERBOSE = false; | 
| 46 | dbg_mode.SetWarning(); | 
| 47 | get1 = 0; | 
| 48 | get2 = 1; | 
| 49 | standalone = false; | 
| 50 | frame2 = "root"; | 
| 51 | frame1 = "root"; | 
| 52 | //    ifroot1 = false; | 
| 53 | //    ifroot2 = true; | 
| 54 | pfolder = "/TrackerFolder"; | 
| 55 | file1 = ""; | 
| 56 | if(f2->IsOpen()){ | 
| 57 | file2  = f2->GetPath(); | 
| 58 | outdir = gSystem->DirName(gSystem->DirName(file2)); | 
| 59 | // check if the indicated output directory exists | 
| 60 | FileStat_t t; | 
| 61 | if( gSystem->GetPathInfo(outdir.Data(),t) )throw -12; | 
| 62 | }else{ | 
| 63 | file2 = ""; | 
| 64 | outdir =  gSystem->WorkingDirectory(); | 
| 65 | }; | 
| 66 |  | 
| 67 | if      (!frame2.CompareTo("root", TString::kIgnoreCase))  ifroot2 = true; | 
| 68 | else if (!frame2.CompareTo("hbook", TString::kIgnoreCase)) ifroot2 =false; | 
| 69 |  | 
| 70 | if      (!frame1.CompareTo("root", TString::kIgnoreCase))  ifroot1 = true; | 
| 71 | else if (!frame1.CompareTo("hbook", TString::kIgnoreCase)) ifroot1 =false; | 
| 72 |  | 
| 73 | ostatus = 0; | 
| 74 |  | 
| 75 | }; | 
| 76 | /** | 
| 77 | * Set processing variables according to input custom parameters | 
| 78 | */ | 
| 79 | void TrkProcess::HandleCustomPar(int ncustom, char *vcustom[]){ | 
| 80 |  | 
| 81 | for (int i=0; i< ncustom; i++){ | 
| 82 |  | 
| 83 | // -----------------------------------------------------// | 
| 84 | if (!strcmp(vcustom[i], "-processFolder")){ | 
| 85 | if (++i >= ncustom) throw -3; | 
| 86 | pfolder = vcustom[i]; | 
| 87 | continue; | 
| 88 | } | 
| 89 | // -----------------------------------------------------// | 
| 90 | if (!strcmp(vcustom[i], "--get1")){ | 
| 91 | get1 = 1; | 
| 92 | continue; | 
| 93 | } | 
| 94 | // -----------------------------------------------------// | 
| 95 | if (!strcmp(vcustom[i], "--dontget2")){ | 
| 96 | get2 = 0; | 
| 97 | continue; | 
| 98 | } | 
| 99 | // -----------------------------------------------------// | 
| 100 | if (!strcmp(vcustom[i], "--standalone") || !strcmp(vcustom[i], "-s")){ | 
| 101 | standalone = true; | 
| 102 | continue; | 
| 103 | } | 
| 104 | // -----------------------------------------------------// | 
| 105 | if (!strcmp(vcustom[i], "-processFile1")){ | 
| 106 | if (++i >= ncustom) throw -3; | 
| 107 | get1 = 1; | 
| 108 | file1 = vcustom[i]; | 
| 109 | continue; | 
| 110 | } | 
| 111 | // -----------------------------------------------------// | 
| 112 | if (!strcmp(vcustom[i], "-frame1")){ | 
| 113 | if (++i >= ncustom)throw -3; | 
| 114 | get1 = 1; | 
| 115 | frame1 = vcustom[i]; | 
| 116 | continue; | 
| 117 | } | 
| 118 | // -----------------------------------------------------// | 
| 119 | if (!strcmp(vcustom[i], "-frame2")){ | 
| 120 | if (++i >= ncustom)throw -3; | 
| 121 | get2 = 1; | 
| 122 | frame2 = vcustom[i]; | 
| 123 | continue; | 
| 124 | } | 
| 125 | // -----------------------------------------------------// | 
| 126 | if (!strcmp(vcustom[i], "--verbose") || !strcmp(vcustom[i], "-v")){ | 
| 127 | //          VERBOSE = true; | 
| 128 | this->SetVerboseMode(); | 
| 129 | continue; | 
| 130 | } | 
| 131 | // -----------------------------------------------------// | 
| 132 | else if (!strcmp(vcustom[i], "--debug") || !strcmp(vcustom[i], "-d")){ | 
| 133 | //          DEBUG = true; | 
| 134 | this->SetDebugMode(); | 
| 135 | continue; | 
| 136 | }; | 
| 137 | } | 
| 138 |  | 
| 139 | if      (!frame2.CompareTo("root", TString::kIgnoreCase))  ifroot2 = true; | 
| 140 | else if (!frame2.CompareTo("hbook", TString::kIgnoreCase)) ifroot2 =false; | 
| 141 | else    throw -201; | 
| 142 |  | 
| 143 | if      (!frame1.CompareTo("root", TString::kIgnoreCase))  ifroot1 = true; | 
| 144 | else if (!frame1.CompareTo("hbook", TString::kIgnoreCase)) ifroot1 =false; | 
| 145 | else    throw -201; | 
| 146 |  | 
| 147 | if(get1){ | 
| 148 |  | 
| 149 | TString filety; | 
| 150 | if      ( ifroot1)  filety=".root"; | 
| 151 | else if (!ifroot1)  filety=".rz"; | 
| 152 | else    throw -200; | 
| 153 |  | 
| 154 | //              file1 = outdir + "/"+ gSystem->BaseName(pfolder) + "/" + file2(file2.Last('/')+1,file2.Last('.')) + ".Level1" + filety; | 
| 155 | TString base = file2; | 
| 156 | if(base.Contains(".Level2.root"))base = base(0,base.Index(".Level2.root")); | 
| 157 | else base = base(0,base.Index(".root")); | 
| 158 | base = gSystem->BaseName(base); | 
| 159 | file1 = outdir + "/"+ gSystem->BaseName(pfolder) + "/" + base +".Level1"+filety; | 
| 160 | //              cout << "Requested Level1 output. "<< endl << "Level1 File: "<< file1 << endl; | 
| 161 | TString path = gSystem->DirName(file1); | 
| 162 | FileStat_t t; | 
| 163 | // check if processFolder directory exists, and possibly create it | 
| 164 | if( gSystem->GetPathInfo(path.Data(),t) ) { | 
| 165 | if( gSystem->MakeDirectory(path.Data()) == -1)throw -13; | 
| 166 | }; | 
| 167 | // if it does, open/create teh putput file | 
| 168 |  | 
| 169 | }; | 
| 170 |  | 
| 171 | ostatus = 0; | 
| 172 |  | 
| 173 | }; | 
| 174 | /** | 
| 175 | * \brief Process Level0 event | 
| 176 | * @param l0_event Pointer to an object of the tracker level0 class | 
| 177 | * Starting from a Level0 object, this routine fills the level0 common, then calls | 
| 178 | * the fortran routines, which process the event and fill level1 and level2 commons. | 
| 179 | */ | 
| 180 | void TrkProcess::ProcessEvent(TrkLevel0 *l0_event){ | 
| 181 |  | 
| 182 | // fill Level0 common from Level0 object | 
| 183 | l0_event->GetCommonVar(&level0event_); | 
| 184 |  | 
| 185 | // process the event Level0->Level1->Level2 | 
| 186 | int F77err = 0; | 
| 187 | // | 
| 188 | reductionflight_(&F77err); | 
| 189 | if(F77err < 0)throw F77err; | 
| 190 | //    else if(F77err > 0 && WarningMode())cout<<" WARNING ("<<F77err<<") - TrkCore - Level1 reduction (CN computation failure, ...)"<<endl; | 
| 191 | // | 
| 192 | if(get2) analysisflight_(); | 
| 193 | //    if(F77err < 0)throw F77err; | 
| 194 | //    else if(F77err > 0 && WarningMode())cout<<" WARNING ("<<F77err<<") - TrkCore - @ Level2 reduction "<<endl; | 
| 195 | // | 
| 196 | ostatus = F77err; | 
| 197 |  | 
| 198 | }; | 
| 199 |  | 
| 200 | void TrkProcess::Dump(){ | 
| 201 |  | 
| 202 | cout <<endl<< "Tracker process parameters: "<< endl; | 
| 203 | cout << " idRun           " << idrun << endl; | 
| 204 | cout << " get1            " << get1 << endl; | 
| 205 | cout << " get2            " << get2 << endl; | 
| 206 | cout << " frame1          " << frame1 << endl; | 
| 207 | cout << " frame2          " << frame2 << endl; | 
| 208 | cout << " file1           " << file1 << endl; | 
| 209 | cout << " file2           " << file2 << endl; | 
| 210 | cout << " outdir          " << outdir << endl; | 
| 211 | cout << " process folder  " << pfolder << endl; | 
| 212 | cout << " standalone mode " << standalone << endl; | 
| 213 | cout << " debug mode      warning " << dbg_mode.warning << endl; | 
| 214 | cout << " debug mode      verbose " << dbg_mode.verbose << endl; | 
| 215 | cout << " debug mode      debug   " << dbg_mode.debug << endl << endl; | 
| 216 |  | 
| 217 | } |