TrackerLevel2/src/TrkProcess.cpp

/**
 * \file TrkProcess.cpp
 * \author Elena Vannuccini
 */
#include <TrkProcess.h>
#include <iostream>
using namespace std;
/**
 * Create TrkProcess object, initializing the tracker processing variables at default values
 */
TrkProcess::TrkProcess(){
    
    idrun = 0;
//    dbg_mode.SetNone();
    get1  = 0;
    full1 = 0;
    geth  = 0;
    get2  = 1;
    standalone = false;
    frame2 = "root";
    frame1 = "root";
    frameh = "root";
    outdir =  gSystem->WorkingDirectory(); 
    pfolder = "/TrackerFolder"; 
    if      (!frame2.CompareTo("root", TString::kIgnoreCase))  ifroot2 = true;          
    else if (!frame2.CompareTo("hbook", TString::kIgnoreCase)) ifroot2 = false;
    
    if      (!frame1.CompareTo("root", TString::kIgnoreCase))  ifroot1 = true;          
    else if (!frame1.CompareTo("hbook", TString::kIgnoreCase)) ifroot1 = false;

    if      (!frameh.CompareTo("root", TString::kIgnoreCase))  ifrooth = true;          
    else if (!frameh.CompareTo("hbook", TString::kIgnoreCase)) ifrooth = false;

    file1 = "";
    file2 = "";
    
    ostatus = 0;

    npar = 0;

    nskip = 0;

//     TrkParams::SetTrackingMode();
//     TrkParams::SetPrecisionFactor(); 
//     TrkParams::SetStepMin();   
    TrkParams::SetMiniDefault();
    TrkParams::SetPFA();
};
        
/**
 * Create TrkProcess object, initializing the tracker processing variables according to input parameters.
 * @param run Run ID
 * @param f2 Pointer to output level2 file
 */
TrkProcess::TrkProcess(ULong64_t run, TFile *f2){
        
    idrun = run;
//    DEBUG = false;
//    VERBOSE = false;
//    dbg_mode.SetWarning();
    get1  = 0;
    full1 = 0;
    get2 = 1;
    geth = 0;
    standalone = false;
    frame2 = "root";
    frame1 = "root";
    frameh = "root";
    pfolder = "/TrackerFolder"; 
    file1 = "";
    if(f2->IsOpen()){
        file2  = f2->GetPath();
        outdir = gSystem->DirName(gSystem->DirName(file2));
        // check if the indicated output directory exists
        FileStat_t t;
        if( gSystem->GetPathInfo(outdir.Data(),t) )throw -12;
    }else{ 
        file2 = "";    
        outdir =  gSystem->WorkingDirectory();
    }; 

    if      (!frame2.CompareTo("root", TString::kIgnoreCase))  ifroot2 = true;          
    else if (!frame2.CompareTo("hbook", TString::kIgnoreCase)) ifroot2 =false;
    
    if      (!frame1.CompareTo("root", TString::kIgnoreCase))  ifroot1 = true;          
    else if (!frame1.CompareTo("hbook", TString::kIgnoreCase)) ifroot1 =false;

    if      (!frameh.CompareTo("root", TString::kIgnoreCase))  ifrooth = true;          
    else if (!frameh.CompareTo("hbook", TString::kIgnoreCase)) ifrooth =false;

    ostatus = 0;
    
    npar=0;
        
    nskip = 0;

//     TrkParams::SetTrackingMode();
//     TrkParams::SetPrecisionFactor(); 
//     TrkParams::SetStepMin();   
    TrkParams::SetMiniDefault();
    TrkParams::SetPFA();

};
/**
 * Set processing variables according to input custom parameters
 */     
int TrkProcess::HandleCustomPar(int ncustom, char *vcustom[]){

    for (int i=0; i< ncustom; i++){

        // -----------------------------------------------------//
        if (!strcmp(vcustom[i], "-processFolder")){
            if (++i >= ncustom) throw -3;
            pfolder = vcustom[i];
            continue;
        }  
        // -----------------------------------------------------//    
        if (!strcmp(vcustom[i], "--get1") || !strcmp(vcustom[i], "--get-level1") ){
            get1  = 1;  
            full1 = 0;
            continue;
        }  
        // -----------------------------------------------------//    
        if (!strcmp(vcustom[i], "--get1-full") || !strcmp(vcustom[i], "--get-level1-full") ){
            get1  = 1;      
            full1 = 1;
            continue;
        }  
        // -----------------------------------------------------//    
        if (!strcmp(vcustom[i], "--geth") || !strcmp(vcustom[i], "--get-hough") ){
            geth = 1;
            continue;
        }  
        // -----------------------------------------------------//    
        if (!strcmp(vcustom[i], "--dontget2") ||!strcmp(vcustom[i], "--dontget-level2") ){
            get2 = 0;
            continue;
        }  
        // -----------------------------------------------------//    
        if (!strcmp(vcustom[i], "--standalone") || !strcmp(vcustom[i], "-s")){
            standalone = true;
            continue;
        }  
        // -----------------------------------------------------//    
        if (!strcmp(vcustom[i], "-processFile1")){
            if (++i >= ncustom) throw -3;
            get1 = 1;
            file1 = vcustom[i];
            continue;
        }  
        // -----------------------------------------------------//    
        if (!strcmp(vcustom[i], "-frame1")){
            if (++i >= ncustom)throw -3;
            get1 = 1;
            frame1 = vcustom[i];                                                
            continue;
        }  
        // -----------------------------------------------------//    
        if (!strcmp(vcustom[i], "-frame2")){
            if (++i >= ncustom)throw -3;
            get2 = 1;
            frame2 = vcustom[i];
            continue;
        }  
        // -----------------------------------------------------//    
        if (!strcmp(vcustom[i], "-pfa")){
            if (++i >= ncustom)throw -3;
            TrkParams::SetPFA(atoi(vcustom[i]));
            continue;
        }  
        // -----------------------------------------------------//    
        if (!strcmp(vcustom[i], "-skip")){
            if (++i >= ncustom)throw -3;
            nskip = atoi(vcustom[i]);
            continue;
        }  
        // -----------------------------------------------------//    
        if ( atoi(vcustom[i]) < 0){
            if (++i >= ncustom)throw -3;
            partype[npar]=-1*atoi(vcustom[i-1]);
            parpath[npar]=vcustom[i];
            npar++;
            continue;
        }  
        // -----------------------------------------------------//    
        if (!strcmp(vcustom[i], "--verbose") || !strcmp(vcustom[i], "-v")){
            TrkParams::SetVerboseMode();
            continue;
        }
        // -----------------------------------------------------//    
        if (!strcmp(vcustom[i], "--warning") || !strcmp(vcustom[i], "-w")){
            TrkParams::SetWarningMode();
            continue;
        }
        // -----------------------------------------------------//    
        if (!strcmp(vcustom[i], "--help") || !strcmp(vcustom[i], "-h")){
            PrintHelp();
            return 1;
        }
        // -----------------------------------------------------//    
        else if (!strcmp(vcustom[i], "--debug") || !strcmp(vcustom[i], "-d")){
            TrkParams::SetDebugMode();
            continue;
        };
    }   
                
    if      (!frame2.CompareTo("root", TString::kIgnoreCase))  ifroot2 = true;          
    else if (!frame2.CompareTo("hbook", TString::kIgnoreCase)) ifroot2 =false;
    else    throw -201;
    
    if      (!frame1.CompareTo("root", TString::kIgnoreCase))  ifroot1 = true;          
    else if (!frame1.CompareTo("hbook", TString::kIgnoreCase)) ifroot1 =false;
    else    throw -201;
        
    if      (!frameh.CompareTo("root", TString::kIgnoreCase))  ifrooth = true;          
    else if (!frameh.CompareTo("hbook", TString::kIgnoreCase)) ifrooth =false;
    else    throw -201;

    if(get1 && !get2)full1=true;

    ostatus = 0;

    return 0;

};
/**
 * \brief Process Level0 event
 * @param l0_event Pointer to an object of the tracker level0 class
 * Starting from a Level0 object, this routine fills the level0 common, then calls 
 * the fortran routines, which process the event and fill level1 and level2 commons.
 */
void TrkProcess::ProcessEvent(TrkLevel0 *l0_event){

    // fill Level0 common from Level0 object
    l0_event->GetLevel0Struct();
    // process the event Level0->Level1->Level2
    int F77err = 0;
    //
    reductionflight_(&F77err);
    if(F77err < 0)throw F77err;
    //
    if(get2) analysisflight_();
    //
    ostatus = F77err;
    
};
                
void TrkProcess::Dump(){
    cout << endl;
    cout << " |-------------------------------------| "<<endl;
    cout << "   tracker-processing input-parameters   "<< endl;
    cout << " |-------------------------------------| "<<endl;
    cout << "   idRun           " << idrun << endl;
    cout << "                   (skip " << nskip <<" events)"<< endl;    
    cout << "   get level1      " << get1 << endl;
    cout << "   full level1     " << full1 << endl;
    cout << "   get level2      " << get2 << endl;
    cout << "   get hough       " << geth << endl;
//    cout << " frame1          " << frame1 << endl;
//    cout << " frame2          " << frame2 << endl;
//    cout << " frameh          " << frame2 << endl;
//    cout << " file1           " << file1 << endl;
    cout << "   output file     " << file2 << endl;
//    cout << " outdir          " << outdir << endl;
//    cout << " process folder  " << pfolder << endl;
//    cout << " standalone mode " << standalone << endl;
    for(int i=0; i<npar; i++)
        cout << "   force loading parameters of type "<<partype[i]<<" from <--- "<<parpath[i]<<endl;
    cout<<endl;

//     cout << " debug mode      warning " << dbg_mode.warning << endl;
//     cout << " debug mode      verbose " << dbg_mode.verbose << endl;
//     cout << " debug mode      debug   " << dbg_mode.debug << endl << endl;
    
}

void TrkProcess::PrintHelp(){

    printf( "\n\n +TRK [ options ] \n \n");
//    printf( "\n --version          : Print tracker software version and exit ");      
    printf( "Options:");
    printf( "\n --help, -h                     : Print this help and exit ");   
//    printf( "\n -idRun RUN         : ID number of the run to be processed (for reprocessing RUN=0) \n");
//    printf( "\n -outDir OUTDIR     : Path where to put the LEVEL2 output                    [default ./ ] ");
//    printf( "\n -processFolder DIR   : Directory (relative to OUTDIR) for other output files  [default TrackerFolder/ ] ");
//    printf( "\n -processFile FILE  : Name of the LEVEL2 output file                         [default RUN.Level2.root]");
//    printf( "\n -processFile1 FILE   : Name of the LEVEL1 output file                         [default RUN.Level1.rz]");
//    printf( "\n -frame1 FRAME        : type of output for LEVEL1, root/hbook                  [default hbook ] ");
//    printf( "\n -frame2 FRAME        : type of output for LEVEL2, root/hbook                  [default root  ] ");            
    printf( "\n --get1, --get-level1           : get LEVEL1 output (store only cluster associated to tracks) [default (no LEVEL1 output) ]  ");
    printf( "\n --get1-full, --get-level1-full : get LEVEL1 output (store all clusters)                      [default (no LEVEL1 output) ]  ");
    printf( "\n --dontget2, --dontget-level2   : do not get LEVEL2 output                                    [default (get LEVEL2 output)]");
    printf( "\n --geth,, --get-hough           : get Hough-transform output                                  [default (no Hough-t. output) ]  ");
    printf( "\n -pfa PFAID                     : type of output for LEVEL2, root/hbook                       [default 14 (COG4)  ] ");          
    printf( "\n -N PATH                        : load parameter of type N=1,2... from PATH                   [default (from DB)  ] ");   
    printf( "\n --verbose, -v                  : verbose mode ");       
    printf( "\n --debug, -d                    : debug mode "); 
    printf( "\n --warning, -w                  : warning mode ");       
    printf( "\n -skip NEV                      : skip first NEV events from each run ");        
//    printf( "\n --standalone, -s   : standalone mode (without RunInfo) ... but it might not work \n");        
//    printf( "\n -host HOST         : Name for the host                                      [default mysql://localhost/pamelaprod ]");
//    printf( "\n -user USER         : Username for the DB                                    [default anonymous] ");
//    printf( "\n -psw  PSW          : Password for the DB                                    [default (none)]\n \n \n");
    

}
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	// dbg_mode.SetNone();
15	get1 = 0;
16	full1 = 0;
17	geth = 0;
18	get2 = 1;
19	standalone = false;
20	frame2 = "root";
21	frame1 = "root";
22	frameh = "root";
23	outdir = gSystem->WorkingDirectory();
24	pfolder = "/TrackerFolder";
25	if (!frame2.CompareTo("root", TString::kIgnoreCase)) ifroot2 = true;
26	else if (!frame2.CompareTo("hbook", TString::kIgnoreCase)) ifroot2 = false;
27
28	if (!frame1.CompareTo("root", TString::kIgnoreCase)) ifroot1 = true;
29	else if (!frame1.CompareTo("hbook", TString::kIgnoreCase)) ifroot1 = false;
30
31	if (!frameh.CompareTo("root", TString::kIgnoreCase)) ifrooth = true;
32	else if (!frameh.CompareTo("hbook", TString::kIgnoreCase)) ifrooth = false;
33
34	file1 = "";
35	file2 = "";
36
37	ostatus = 0;
38
39	npar = 0;
40
41	nskip = 0;
42
43	// TrkParams::SetTrackingMode();
44	// TrkParams::SetPrecisionFactor();
45	// TrkParams::SetStepMin();
46	TrkParams::SetMiniDefault();
47	TrkParams::SetPFA();
48	};
49
50	/**
51	* Create TrkProcess object, initializing the tracker processing variables according to input parameters.
52	* @param run Run ID
53	* @param f2 Pointer to output level2 file
54	*/
55	TrkProcess::TrkProcess(ULong64_t run, TFile *f2){
56
57	idrun = run;
58	// DEBUG = false;
59	// VERBOSE = false;
60	// dbg_mode.SetWarning();
61	get1 = 0;
62	full1 = 0;
63	get2 = 1;
64	geth = 0;
65	standalone = false;
66	frame2 = "root";
67	frame1 = "root";
68	frameh = "root";
69	pfolder = "/TrackerFolder";
70	file1 = "";
71	if(f2->IsOpen()){
72	file2 = f2->GetPath();
73	outdir = gSystem->DirName(gSystem->DirName(file2));
74	// check if the indicated output directory exists
75	FileStat_t t;
76	if( gSystem->GetPathInfo(outdir.Data(),t) )throw -12;
77	}else{
78	file2 = "";
79	outdir = gSystem->WorkingDirectory();
80	};
81
82	if (!frame2.CompareTo("root", TString::kIgnoreCase)) ifroot2 = true;
83	else if (!frame2.CompareTo("hbook", TString::kIgnoreCase)) ifroot2 =false;
84
85	if (!frame1.CompareTo("root", TString::kIgnoreCase)) ifroot1 = true;
86	else if (!frame1.CompareTo("hbook", TString::kIgnoreCase)) ifroot1 =false;
87
88	if (!frameh.CompareTo("root", TString::kIgnoreCase)) ifrooth = true;
89	else if (!frameh.CompareTo("hbook", TString::kIgnoreCase)) ifrooth =false;
90
91	ostatus = 0;
92
93	npar=0;
94
95	nskip = 0;
96
97	// TrkParams::SetTrackingMode();
98	// TrkParams::SetPrecisionFactor();
99	// TrkParams::SetStepMin();
100	TrkParams::SetMiniDefault();
101	TrkParams::SetPFA();
102
103	};
104	/**
105	* Set processing variables according to input custom parameters
106	*/
107	int TrkProcess::HandleCustomPar(int ncustom, char *vcustom[]){
108
109	for (int i=0; i< ncustom; i++){
110
111	// -----------------------------------------------------//
112	if (!strcmp(vcustom[i], "-processFolder")){
113	if (++i >= ncustom) throw -3;
114	pfolder = vcustom[i];
115	continue;
116	}
117	// -----------------------------------------------------//
118	if (!strcmp(vcustom[i], "--get1") \|\| !strcmp(vcustom[i], "--get-level1") ){
119	get1 = 1;
120	full1 = 0;
121	continue;
122	}
123	// -----------------------------------------------------//
124	if (!strcmp(vcustom[i], "--get1-full") \|\| !strcmp(vcustom[i], "--get-level1-full") ){
125	get1 = 1;
126	full1 = 1;
127	continue;
128	}
129	// -----------------------------------------------------//
130	if (!strcmp(vcustom[i], "--geth") \|\| !strcmp(vcustom[i], "--get-hough") ){
131	geth = 1;
132	continue;
133	}
134	// -----------------------------------------------------//
135	if (!strcmp(vcustom[i], "--dontget2") \|\|!strcmp(vcustom[i], "--dontget-level2") ){
136	get2 = 0;
137	continue;
138	}
139	// -----------------------------------------------------//
140	if (!strcmp(vcustom[i], "--standalone") \|\| !strcmp(vcustom[i], "-s")){
141	standalone = true;
142	continue;
143	}
144	// -----------------------------------------------------//
145	if (!strcmp(vcustom[i], "-processFile1")){
146	if (++i >= ncustom) throw -3;
147	get1 = 1;
148	file1 = vcustom[i];
149	continue;
150	}
151	// -----------------------------------------------------//
152	if (!strcmp(vcustom[i], "-frame1")){
153	if (++i >= ncustom)throw -3;
154	get1 = 1;
155	frame1 = vcustom[i];
156	continue;
157	}
158	// -----------------------------------------------------//
159	if (!strcmp(vcustom[i], "-frame2")){
160	if (++i >= ncustom)throw -3;
161	get2 = 1;
162	frame2 = vcustom[i];
163	continue;
164	}
165	// -----------------------------------------------------//
166	if (!strcmp(vcustom[i], "-pfa")){
167	if (++i >= ncustom)throw -3;
168	TrkParams::SetPFA(atoi(vcustom[i]));
169	continue;
170	}
171	// -----------------------------------------------------//
172	if (!strcmp(vcustom[i], "-skip")){
173	if (++i >= ncustom)throw -3;
174	nskip = atoi(vcustom[i]);
175	continue;
176	}
177	// -----------------------------------------------------//
178	if ( atoi(vcustom[i]) < 0){
179	if (++i >= ncustom)throw -3;
180	partype[npar]=-1*atoi(vcustom[i-1]);
181	parpath[npar]=vcustom[i];
182	npar++;
183	continue;
184	}
185	// -----------------------------------------------------//
186	if (!strcmp(vcustom[i], "--verbose") \|\| !strcmp(vcustom[i], "-v")){
187	TrkParams::SetVerboseMode();
188	continue;
189	}
190	// -----------------------------------------------------//
191	if (!strcmp(vcustom[i], "--warning") \|\| !strcmp(vcustom[i], "-w")){
192	TrkParams::SetWarningMode();
193	continue;
194	}
195	// -----------------------------------------------------//
196	if (!strcmp(vcustom[i], "--help") \|\| !strcmp(vcustom[i], "-h")){
197	PrintHelp();
198	return 1;
199	}
200	// -----------------------------------------------------//
201	else if (!strcmp(vcustom[i], "--debug") \|\| !strcmp(vcustom[i], "-d")){
202	TrkParams::SetDebugMode();
203	continue;
204	};
205	}
206
207	if (!frame2.CompareTo("root", TString::kIgnoreCase)) ifroot2 = true;
208	else if (!frame2.CompareTo("hbook", TString::kIgnoreCase)) ifroot2 =false;
209	else throw -201;
210
211	if (!frame1.CompareTo("root", TString::kIgnoreCase)) ifroot1 = true;
212	else if (!frame1.CompareTo("hbook", TString::kIgnoreCase)) ifroot1 =false;
213	else throw -201;
214
215	if (!frameh.CompareTo("root", TString::kIgnoreCase)) ifrooth = true;
216	else if (!frameh.CompareTo("hbook", TString::kIgnoreCase)) ifrooth =false;
217	else throw -201;
218
219	if(get1 && !get2)full1=true;
220
221	ostatus = 0;
222
223	return 0;
224
225	};
226	/**
227	* \brief Process Level0 event
228	* @param l0_event Pointer to an object of the tracker level0 class
229	* Starting from a Level0 object, this routine fills the level0 common, then calls
230	* the fortran routines, which process the event and fill level1 and level2 commons.
231	*/
232	void TrkProcess::ProcessEvent(TrkLevel0 *l0_event){
233
234	// fill Level0 common from Level0 object
235	l0_event->GetLevel0Struct();
236	// process the event Level0->Level1->Level2
237	int F77err = 0;
238	//
239	reductionflight_(&F77err);
240	if(F77err < 0)throw F77err;
241	//
242	if(get2) analysisflight_();
243	//
244	ostatus = F77err;
245
246	};
247
248	void TrkProcess::Dump(){
249	cout << endl;
250	cout << " \|-------------------------------------\| "<<endl;
251	cout << " tracker-processing input-parameters "<< endl;
252	cout << " \|-------------------------------------\| "<<endl;
253	cout << " idRun " << idrun << endl;
254	cout << " (skip " << nskip <<" events)"<< endl;
255	cout << " get level1 " << get1 << endl;
256	cout << " full level1 " << full1 << endl;
257	cout << " get level2 " << get2 << endl;
258	cout << " get hough " << geth << endl;
259	// cout << " frame1 " << frame1 << endl;
260	// cout << " frame2 " << frame2 << endl;
261	// cout << " frameh " << frame2 << endl;
262	// cout << " file1 " << file1 << endl;
263	cout << " output file " << file2 << endl;
264	// cout << " outdir " << outdir << endl;
265	// cout << " process folder " << pfolder << endl;
266	// cout << " standalone mode " << standalone << endl;
267	for(int i=0; i<npar; i++)
268	cout << " force loading parameters of type "<<partype[i]<<" from <--- "<<parpath[i]<<endl;
269	cout<<endl;
270
271	// cout << " debug mode warning " << dbg_mode.warning << endl;
272	// cout << " debug mode verbose " << dbg_mode.verbose << endl;
273	// cout << " debug mode debug " << dbg_mode.debug << endl << endl;
274
275	}
276
277	void TrkProcess::PrintHelp(){
278
279	printf( "\n\n +TRK [ options ] \n \n");
280	// printf( "\n --version : Print tracker software version and exit ");
281	printf( "Options:");
282	printf( "\n --help, -h : Print this help and exit ");
283	// printf( "\n -idRun RUN : ID number of the run to be processed (for reprocessing RUN=0) \n");
284	// printf( "\n -outDir OUTDIR : Path where to put the LEVEL2 output [default ./ ] ");
285	// printf( "\n -processFolder DIR : Directory (relative to OUTDIR) for other output files [default TrackerFolder/ ] ");
286	// printf( "\n -processFile FILE : Name of the LEVEL2 output file [default RUN.Level2.root]");
287	// printf( "\n -processFile1 FILE : Name of the LEVEL1 output file [default RUN.Level1.rz]");
288	// printf( "\n -frame1 FRAME : type of output for LEVEL1, root/hbook [default hbook ] ");
289	// printf( "\n -frame2 FRAME : type of output for LEVEL2, root/hbook [default root ] ");
290	printf( "\n --get1, --get-level1 : get LEVEL1 output (store only cluster associated to tracks) [default (no LEVEL1 output) ] ");
291	printf( "\n --get1-full, --get-level1-full : get LEVEL1 output (store all clusters) [default (no LEVEL1 output) ] ");
292	printf( "\n --dontget2, --dontget-level2 : do not get LEVEL2 output [default (get LEVEL2 output)]");
293	printf( "\n --geth,, --get-hough : get Hough-transform output [default (no Hough-t. output) ] ");
294	printf( "\n -pfa PFAID : type of output for LEVEL2, root/hbook [default 14 (COG4) ] ");
295	printf( "\n -N PATH : load parameter of type N=1,2... from PATH [default (from DB) ] ");
296	printf( "\n --verbose, -v : verbose mode ");
297	printf( "\n --debug, -d : debug mode ");
298	printf( "\n --warning, -w : warning mode ");
299	printf( "\n -skip NEV : skip first NEV events from each run ");
300	// printf( "\n --standalone, -s : standalone mode (without RunInfo) ... but it might not work \n");
301	// printf( "\n -host HOST : Name for the host [default mysql://localhost/pamelaprod ]");
302	// printf( "\n -user USER : Username for the DB [default anonymous] ");
303	// printf( "\n -psw PSW : Password for the DB [default (none)]\n \n \n");
304
305
306	}