/[PAMELA software]/DarthVader/TrackerLevel2/src/TrkProcess.cpp
ViewVC logotype

Diff of /DarthVader/TrackerLevel2/src/TrkProcess.cpp

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 1.1 by mocchiut, Fri May 19 13:15:54 2006 UTC revision 1.17 by pam-ts, Wed Jun 4 07:57:04 2014 UTC
# Line 5  Line 5 
5  #include <TrkProcess.h>  #include <TrkProcess.h>
6  #include <iostream>  #include <iostream>
7  using namespace std;  using namespace std;
8    /**
9     * Create TrkProcess object, initializing the tracker processing variables at default values
10     */
11  TrkProcess::TrkProcess(){  TrkProcess::TrkProcess(){
12            
13      idrun = 0;      idrun = 0;
14      DEBUG = false;  //    dbg_mode.SetNone();
15      get1 = 0;      get1  = 0;
16      get2 = 1;      full1 = 0;
17        geth  = 0;
18        get2  = 1;
19        gete  = 1;
20        getn  = 1;
21        extAlgID = 202;
22      standalone = false;      standalone = false;
23      frame2 = "root";      frame2 = "root";
24      frame1 = "hbook";      frame1 = "root";
25        frameh = "root";
26        framee = "root";
27      outdir =  gSystem->WorkingDirectory();      outdir =  gSystem->WorkingDirectory();
28      pfolder = "/TrackerFolder";      pfolder = "/TrackerFolder";
29      if      (!frame2.CompareTo("root", TString::kIgnoreCase))  ifroot2 = true;                if      (!frame2.CompareTo("root", TString::kIgnoreCase))  ifroot2 = true;          
30      else if (!frame2.CompareTo("hbook", TString::kIgnoreCase)) ifroot2 =false;      else if (!frame2.CompareTo("hbook", TString::kIgnoreCase)) ifroot2 = false;
31            
32      if      (!frame1.CompareTo("root", TString::kIgnoreCase))  ifroot1 = true;                if      (!frame1.CompareTo("root", TString::kIgnoreCase))  ifroot1 = true;          
33      else if (!frame1.CompareTo("hbook", TString::kIgnoreCase)) ifroot1 =false;      else if (!frame1.CompareTo("hbook", TString::kIgnoreCase)) ifroot1 = false;
34    
35        if      (!frameh.CompareTo("root", TString::kIgnoreCase))  ifrooth = true;          
36        else if (!frameh.CompareTo("hbook", TString::kIgnoreCase)) ifrooth = false;
37    
38      file1 = "";      file1 = "";
39      file2 = "";      file2 = "";
40            
41        ostatus = 0;
42    
43        npar = 0;
44    
45        nskip = 0;
46    
47        watch = false;
48    
49    //     TrkParams::SetTrackingMode();
50    //     TrkParams::SetPrecisionFactor();
51    //     TrkParams::SetStepMin();  
52        TrkParams::SetMiniDefault();
53        TrkParams::SetPFA();
54  };  };
55                    
56    /**
57     * Create TrkProcess object, initializing the tracker processing variables according to input parameters.
58     * @param run Run ID
59     * @param f2 Pointer to output level2 file
60     */
61  TrkProcess::TrkProcess(ULong64_t run, TFile *f2){  TrkProcess::TrkProcess(ULong64_t run, TFile *f2){
62                    
63      idrun = run;      idrun = run;
64      DEBUG = false;  //    DEBUG = false;
65      get1 = 0;  //    VERBOSE = false;
66    //    dbg_mode.SetWarning();
67        get1  = 0;
68        full1 = 0;
69      get2 = 1;      get2 = 1;
70        geth = 0;
71        gete = 1;
72        getn = 1;
73        extAlgID = 202;
74      standalone = false;      standalone = false;
75      frame2 = "root";      frame2 = "root";
76      frame1 = "hbook";      frame1 = "root";
77      ifroot1 = false;      frameh = "root";
78      ifroot2 = true;      framee = "root";
79        pfolder = "/TrackerFolder";
80      file1 = "";      file1 = "";
81      if(f2->IsOpen()){      if(f2->IsOpen()){
82          file2  = f2->GetPath();          file2  = f2->GetPath();
 //      TString temp = file2(0,file2.Last(':'));  
 //      outdir = temp(0,temp.Last('/'));  
83          outdir = gSystem->DirName(gSystem->DirName(file2));          outdir = gSystem->DirName(gSystem->DirName(file2));
84  // check if the indicated output directory exists          // check if the indicated output directory exists
85          FileStat_t t;          FileStat_t t;
86          if( gSystem->GetPathInfo(outdir.Data(),t) )throw -12;          if( gSystem->GetPathInfo(outdir.Data(),t) )throw -12;
87      }else{      }else{
88          file2 = "";              file2 = "";    
89          outdir =  gSystem->WorkingDirectory();          outdir =  gSystem->WorkingDirectory();
90      };      };
     pfolder = "/TrackerFolder";  
91    
92      if      (!frame2.CompareTo("root", TString::kIgnoreCase))  ifroot2 = true;                if      (!frame2.CompareTo("root", TString::kIgnoreCase))  ifroot2 = true;          
93      else if (!frame2.CompareTo("hbook", TString::kIgnoreCase)) ifroot2 =false;      else if (!frame2.CompareTo("hbook", TString::kIgnoreCase)) ifroot2 =false;
# Line 60  TrkProcess::TrkProcess(ULong64_t run, TF Line 95  TrkProcess::TrkProcess(ULong64_t run, TF
95      if      (!frame1.CompareTo("root", TString::kIgnoreCase))  ifroot1 = true;                if      (!frame1.CompareTo("root", TString::kIgnoreCase))  ifroot1 = true;          
96      else if (!frame1.CompareTo("hbook", TString::kIgnoreCase)) ifroot1 =false;      else if (!frame1.CompareTo("hbook", TString::kIgnoreCase)) ifroot1 =false;
97    
98        if      (!frameh.CompareTo("root", TString::kIgnoreCase))  ifrooth = true;          
99        else if (!frameh.CompareTo("hbook", TString::kIgnoreCase)) ifrooth =false;
100    
101        ostatus = 0;
102        
103        npar=0;
104                    
105        nskip = 0;
106    
107        watch = false;
108    
109    //     TrkParams::SetTrackingMode();
110    //     TrkParams::SetPrecisionFactor();
111    //     TrkParams::SetStepMin();  
112        TrkParams::SetMiniDefault();
113        TrkParams::SetPFA();
114    
115  };  };
116            /**
117     * Set processing variables according to input custom parameters
118     */    
119  int TrkProcess::HandleCustomPar(int ncustom, char *vcustom[]){  int TrkProcess::HandleCustomPar(int ncustom, char *vcustom[]){
120    
121      for (int i=0; i< ncustom; i++){      for (int i=0; i< ncustom; i++){
122                    
123          cout << "Handling Parameter " << i+1  << " of " << ncustom << ": "  << vcustom[i] << endl;
124    
125            // -----------------------------------------------------//
126            if (!strcmp(vcustom[i], "--simu")){
127                cout << "Setting DarthVader to use Simulated Data\n";
128                TrkParams::SetSimuFlag(kTRUE);
129                continue;
130            }  
131          // -----------------------------------------------------//          // -----------------------------------------------------//
132          if (!strcmp(vcustom[i], "-processFolder")){          if (!strcmp(vcustom[i], "-processFolder")){
133              if (++i >= ncustom) throw -3;              if (++i >= ncustom) throw -3;
# Line 74  int TrkProcess::HandleCustomPar(int ncus Line 135  int TrkProcess::HandleCustomPar(int ncus
135              continue;              continue;
136          }            }  
137          // -----------------------------------------------------//              // -----------------------------------------------------//    
138          if (!strcmp(vcustom[i], "--get1")){          if (!strcmp(vcustom[i], "--get1") || !strcmp(vcustom[i], "--get-level1") ){
139              get1 = 1;              get1  = 1;  
140                full1 = 0;
141                continue;
142            }  
143            // -----------------------------------------------------//    
144            if (!strcmp(vcustom[i], "--get1-full") || !strcmp(vcustom[i], "--get-level1-full") ){
145                get1  = 1;      
146                full1 = 1;
147                continue;
148            }  
149            // -----------------------------------------------------//    
150            if (!strcmp(vcustom[i], "--geth") || !strcmp(vcustom[i], "--get-hough") ){
151                geth = 1;
152                continue;
153            }  
154            // -----------------------------------------------------//    
155    //      if (!strcmp(vcustom[i], "--dontget2") ||!strcmp(vcustom[i], "--dontget-level2") ){
156    //          get2 = 0;
157    //          gete = 0;
158    //          getn = 0;
159    //          continue;
160    //      }  
161            // -----------------------------------------------------//    
162            if (!strcmp(vcustom[i], "--dontgete") ||!strcmp(vcustom[i], "--dontget-exttrack") ){
163                gete = 0;
164                continue;
165            }  
166            // -----------------------------------------------------//    
167            if (!strcmp(vcustom[i], "-extalg")){
168                if (++i >= ncustom) throw -3;
169                gete = 1;      
170                get2 = 1;      
171                extAlgID = atoi(vcustom[i]);
172              continue;              continue;
173          }            }  
174          // -----------------------------------------------------//              // -----------------------------------------------------//    
175          if (!strcmp(vcustom[i], "--dontget2")){          if (!strcmp(vcustom[i], "--dontgetn") ||!strcmp(vcustom[i], "--dontget-nuclei") ){
176              get2 = 0;              getn = 0;
177              continue;              continue;
178          }            }  
179          // -----------------------------------------------------//              // -----------------------------------------------------//    
# Line 95  int TrkProcess::HandleCustomPar(int ncus Line 188  int TrkProcess::HandleCustomPar(int ncus
188              file1 = vcustom[i];              file1 = vcustom[i];
189              continue;              continue;
190          }            }  
191    //      // -----------------------------------------------------//    
192    //      if (!strcmp(vcustom[i], "-frame1")){
193    //          if (++i >= ncustom)throw -3;
194    //          get1 = 1;
195    //          frame1 = vcustom[i];                                                
196    //          continue;
197    //      }  
198    //      // -----------------------------------------------------//    
199    //      if (!strcmp(vcustom[i], "-frame2")){
200    //          if (++i >= ncustom)throw -3;
201    //          get2 = 1;
202    //          frame2 = vcustom[i];
203    //          continue;
204    //      }  
205    //      // -----------------------------------------------------//    
206    //      if (!strcmp(vcustom[i], "-framee")){
207    //          if (++i >= ncustom)throw -3;
208    //          gete = 1;
209    //          framee = vcustom[i];
210    //          continue;
211    //      }  
212          // -----------------------------------------------------//              // -----------------------------------------------------//    
213          if (!strcmp(vcustom[i], "-frame1")){          if (!strcmp(vcustom[i], "-pfa")){
214              if (++i >= ncustom)throw -3;              if (++i >= ncustom)throw -3;
215              get1 = 1;              TrkParams::SetPFA(atoi(vcustom[i]));
             frame1 = vcustom[i];                                                  
216              continue;              continue;
217          }            }  
218          // -----------------------------------------------------//              // -----------------------------------------------------//    
219          if (!strcmp(vcustom[i], "-frame2")){          if (!strcmp(vcustom[i], "-skip")){
220              if (++i >= ncustom)throw -3;              if (++i >= ncustom)throw -3;
221              get2 = 1;              nskip = atoi(vcustom[i]);
             frame2 = vcustom[i];  
222              continue;              continue;
223          }            }  
224          // -----------------------------------------------------//              // -----------------------------------------------------//    
225          else if (!strcmp(vcustom[i], "--verbose") || !strcmp(vcustom[i], "-v")){          if ( atoi(vcustom[i]) < 0){
226              DEBUG = true;              if (++i >= ncustom)throw -3;
227                partype[npar]=-1*atoi(vcustom[i-1]);
228                parpath[npar]=vcustom[i];
229                npar++;
230              continue;              continue;
231          }            }  
232            // -----------------------------------------------------//    
233            if (!strcmp(vcustom[i], "--verbose") || !strcmp(vcustom[i], "-v")){
234                TrkParams::SetVerboseMode();
235                continue;
236            }
237            // -----------------------------------------------------//    
238            if (!strcmp(vcustom[i], "--warning") || !strcmp(vcustom[i], "-w")){
239                TrkParams::SetWarningMode();
240                continue;
241            }
242            // -----------------------------------------------------//    
243            if (!strcmp(vcustom[i], "--help") || !strcmp(vcustom[i], "-h")){
244                PrintHelp();
245                return 1;
246            }
247            // -----------------------------------------------------//    
248            if (!strcmp(vcustom[i], "--watch") ){
249                watch = true;
250                continue;
251            }
252            // -----------------------------------------------------//    
253            else if (!strcmp(vcustom[i], "--debug") || !strcmp(vcustom[i], "-d")){
254                TrkParams::SetDebugMode();
255                continue;
256            };
257      }        }  
258                        
259      if      (!frame2.CompareTo("root", TString::kIgnoreCase))  ifroot2 = true;                if      (!frame2.CompareTo("root", TString::kIgnoreCase))  ifroot2 = true;          
260      else if (!frame2.CompareTo("hbook", TString::kIgnoreCase)) ifroot2 =false;      else if (!frame2.CompareTo("hbook", TString::kIgnoreCase)) ifroot2 =false;
261      else    throw -201;      else    throw -201;
# Line 123  int TrkProcess::HandleCustomPar(int ncus Line 263  int TrkProcess::HandleCustomPar(int ncus
263      if      (!frame1.CompareTo("root", TString::kIgnoreCase))  ifroot1 = true;                if      (!frame1.CompareTo("root", TString::kIgnoreCase))  ifroot1 = true;          
264      else if (!frame1.CompareTo("hbook", TString::kIgnoreCase)) ifroot1 =false;      else if (!frame1.CompareTo("hbook", TString::kIgnoreCase)) ifroot1 =false;
265      else    throw -201;      else    throw -201;
266                
267      if(get1){      if      (!frameh.CompareTo("root", TString::kIgnoreCase))  ifrooth = true;          
268          TString filety;      else if (!frameh.CompareTo("hbook", TString::kIgnoreCase)) ifrooth =false;
269          if      ( ifroot1)  filety=".root";                  else    throw -201;
         else if (!ifroot1)  filety=".rz";  
         else    throw -200;  
 //      file1 = outdir + "/"+ pfolder + "/" + file2(file2.Last('/')+1,file2.Last('.')) + ".Level1" + filety;  
 //      cout << p->file1 << endl;  
 //      TString path = gSystem->DirName(p->file1);  
 //      FileStat_t t;  
 //      // check if processFolder directory exists, and possibly create it  
 //      if( gSystem->GetPathInfo(path.Data(),t) ) {  
 //          if( gSystem->MakeDirectory(path.Data()) == -1)throw -13;  
 //      };  
 //      // if it does, open/create teh putput file  
270    
271      };      if      (!framee.CompareTo("root", TString::kIgnoreCase))  ifroote = true;          
272        else if (!framee.CompareTo("hbook", TString::kIgnoreCase)) ifroote =false;
273        else    throw -201;
274    
275        if(get1 && !get2)full1=true;
276    
277        ostatus = 0;
278    
279      return 0;      return 0;
280    
281    };
282    /**
283     * \brief Process Level0 event
284     * @param l0_event Pointer to an object of the tracker level0 class
285     * Starting from a Level0 object, this routine fills the level0 common, then calls
286     * the fortran routines, which process the event and fill level1 and level2 commons.
287     */
288    void TrkProcess::ProcessEvent(TrkLevel0 *l0_event){
289    
290        // fill Level0 common from Level0 object
291        l0_event->GetLevel0Struct();
292        // process the event Level0->Level1->Level2
293        int F77err = 0;
294        //
295        reductionflight_(&F77err);
296        if(F77err < 0)throw F77err;
297        //
298        if(get2) analysisflight_();
299        //
300    
301        ostatus = F77err;
302        
303  };  };
304                                    
305  void TrkProcess::Dump(){  void TrkProcess::Dump(){
306        cout << endl;
307        cout << " |-------------------------------------| "<<endl;
308        cout << "   tracker-processing input-parameters   "<< endl;
309        cout << " |-------------------------------------| "<<endl;
310        cout << "   idRun            " << idrun << endl;
311        cout << "                    (skip " << nskip <<" events)"<< endl;    
312        cout << "   get level1       " << get1 << endl;
313        cout << "   full level1      " << full1 << endl;
314        cout << "   get level2       " << get2 << endl;
315        cout << "   get hough        " << geth << endl;
316        cout << "   get ext.tracking " << gete << endl;
317        if(gete)
318        cout << "   ext.tracking alg " << extAlgID << endl;
319        cout << "   get nuclei       " << getn << endl;
320    //    cout << " frame1           " << frame1 << endl;
321    //    cout << " frame2           " << frame2 << endl;
322    //    cout << " frameh           " << frame2 << endl;
323    //    cout << " file1            " << file1 << endl;
324        cout << "   output file      " << file2 << endl;
325    //    cout << " outdir           " << outdir << endl;
326    //    cout << " process folder   " << pfolder << endl;
327    //    cout << " standalone mode  " << standalone << endl;
328        for(int i=0; i<npar; i++)
329            cout << "   force loading parameters of type "<<partype[i]<<" from <--- "<<parpath[i]<<endl;
330        cout<<endl;
331    
332    //     cout << " debug mode      warning " << dbg_mode.warning << endl;
333    //     cout << " debug mode      verbose " << dbg_mode.verbose << endl;
334    //     cout << " debug mode      debug   " << dbg_mode.debug << endl << endl;
335        
336    }
337    
338      cout <<endl<< "Tracker process parameters: "<< endl;  void TrkProcess::PrintHelp(){
339      cout << " idRun           " << idrun << endl;  
340      cout << " get1            " << get1 << endl;      printf( "\n\n +TRK [ options ] \n \n");
341      cout << " get2            " << get2 << endl;  //    printf( "\n --version          : Print tracker software version and exit ");      
342      cout << " frame1          " << frame1 << endl;      printf( "Options:");
343      cout << " frame2          " << frame2 << endl;      printf( "\n --help, -h                     : Print this help and exit ");  
344      cout << " file1           " << file1 << endl;  //    printf( "\n -idRun RUN         : ID number of the run to be processed (for reprocessing RUN=0) \n");
345      cout << " file2           " << file2 << endl;  //    printf( "\n -outDir OUTDIR     : Path where to put the LEVEL2 output                    [default ./ ] ");
346      cout << " outdir          " << outdir << endl;  //    printf( "\n -processFolder DIR   : Directory (relative to OUTDIR) for other output files  [default TrackerFolder/ ] ");
347      cout << " process folder  " << pfolder << endl;      //    printf( "\n -processFile FILE  : Name of the LEVEL2 output file                         [default RUN.Level2.root]");
348      cout << " standalone mode " << standalone << endl;  //    printf( "\n -processFile1 FILE   : Name of the LEVEL1 output file                         [default RUN.Level1.rz]");
349      cout << " debug mode      " << DEBUG << endl<< endl;  //    printf( "\n -frame1 FRAME        : type of output for LEVEL1, root/hbook                  [default hbook ] ");
350    //    printf( "\n -frame2 FRAME        : type of output for LEVEL2, root/hbook                  [default root  ] ");            
351        printf( "\n --get1, --get-level1           : get LEVEL1 output (store only cluster associated to tracks) [default (no LEVEL1 output) ]  ");
352        printf( "\n --get1-full, --get-level1-full : get LEVEL1 output (store all clusters)                      [default (no LEVEL1 output) ]  ");
353        printf( "\n --dontget2, --dontget-level2   : do not get LEVEL2 output                                    [default (get LEVEL2 output)]");
354        printf( "\n --dontgete, --dontget-exttrack : do not get ext.tracking output                              [default (ext.algorythm 202)]");
355        printf( "\n --dontgetn, --dontget-nuclei   : do not get nuclei cleaned tracks                            [default (get nuclei output)]");
356        printf( "\n --extalg ID                    : apply ext.tracking algorythm n.ID                           [default (ID=202) ]  ");
357        printf( "\n --geth,, --get-hough           : get Hough-transform output                                  [default (no Hough-t. output) ]  ");
358        printf( "\n -pfa PFAID                     : type of output for LEVEL2, root/hbook                       [default 14 (COG4)  ] ");          
359        printf( "\n -N PATH                        : load parameter of type N=1,2... from PATH                   [default (from DB)  ] ");  
360        printf( "\n --verbose, -v                  : verbose mode ");      
361        printf( "\n --debug, -d                    : debug mode ");
362        printf( "\n --warning, -w                  : warning mode ");      
363        printf( "\n -skip NEV                      : skip first NEV events from each run ");        
364    //    printf( "\n --standalone, -s   : standalone mode (without RunInfo) ... but it might not work \n");        
365    //    printf( "\n -host HOST         : Name for the host                                      [default mysql://localhost/pamelaprod ]");
366    //    printf( "\n -user USER         : Username for the DB                                    [default anonymous] ");
367    //    printf( "\n -psw  PSW          : Password for the DB                                    [default (none)]\n \n \n");
368            
369    
370  }  }
Legend:
Removed from v.1.1  
changed lines
  Added in v.1.17
  ViewVC Help
Powered by ViewVC 1.1.23