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;
//    DEBUG = false;
//    VERBOSE = false;
    dbg_mode.SetNone();
    get1 = 0;
    get2 = 1;
    standalone = false;
    frame2 = "root";
    frame1 = "hbook";
    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;

    file1 = "";
    file2 = "";
    
    ostatus = 0;
};
        
/**
 * 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;
    get2 = 1;
    standalone = false;
    frame2 = "root";
    frame1 = "root";
//    ifroot1 = false;
//    ifroot2 = true;
        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;

    ostatus = 0;
        
};
/**
 * Set processing variables according to input custom parameters
 */     
void 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")){
            get1 = 1;
            continue;
        }  
        // -----------------------------------------------------//    
        if (!strcmp(vcustom[i], "--dontget2")){
            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], "--verbose") || !strcmp(vcustom[i], "-v")){
//          VERBOSE = true;
            this->SetVerboseMode();
            continue;
        }
        // -----------------------------------------------------//    
        else if (!strcmp(vcustom[i], "--debug") || !strcmp(vcustom[i], "-d")){
//          DEBUG = true;
            this->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(get1){
                
                TString filety;
                if      ( ifroot1)  filety=".root";             
                else if (!ifroot1)  filety=".rz";
                else    throw -200;
                
//              file1 = outdir + "/"+ gSystem->BaseName(pfolder) + "/" + file2(file2.Last('/')+1,file2.Last('.')) + ".Level1" + filety;
                TString base = file2;           
                if(base.Contains(".Level2.root"))base = base(0,base.Index(".Level2.root"));
                else base = base(0,base.Index(".root"));
                base = gSystem->BaseName(base);
                file1 = outdir + "/"+ gSystem->BaseName(pfolder) + "/" + base +".Level1"+filety;
//              cout << "Requested Level1 output. "<< endl << "Level1 File: "<< file1 << endl;
                TString path = gSystem->DirName(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

    };

    ostatus = 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->GetCommonVar(&level0event_);

    // process the event Level0->Level1->Level2
    int F77err = 0;
    //
    reductionflight_(&F77err);
    if(F77err < 0)throw F77err;
    //    else if(F77err > 0 && WarningMode())cout<<" WARNING ("<<F77err<<") - TrkCore - Level1 reduction (CN computation failure, ...)"<<endl; 
    //
    if(get2) analysisflight_(); 
//    if(F77err < 0)throw F77err;
//    else if(F77err > 0 && WarningMode())cout<<" WARNING ("<<F77err<<") - TrkCore - @ Level2 reduction "<<endl; 
    //
    ostatus = F77err;
    
};
                
void TrkProcess::Dump(){

    cout <<endl<< "Tracker process parameters: "<< endl;
    cout << " idRun           " << idrun << endl;
    cout << " get1            " << get1 << endl;
    cout << " get2            " << get2 << endl;
    cout << " frame1          " << frame1 << endl;
    cout << " frame2          " << frame2 << endl;
    cout << " file1           " << file1 << endl;
    cout << " file2           " << file2 << endl;
    cout << " outdir          " << outdir << endl;
    cout << " process folder  " << pfolder << endl;
    cout << " standalone mode " << standalone << 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;
    
}
1	mocchiut	1.1	/**
2			* \file TrkProcess.cpp
3			* \author Elena Vannuccini
4			*/
5			#include <TrkProcess.h>
6			#include <iostream>
7			using namespace std;
8	pam-fi	1.2	/**
9			* Create TrkProcess object, initializing the tracker processing variables at default values
10			*/
11	mocchiut	1.1	TrkProcess::TrkProcess(){
12
13			idrun = 0;
14	pam-fi	1.2	// DEBUG = false;
15			// VERBOSE = false;
16			dbg_mode.SetNone();
17	mocchiut	1.1	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	pam-fi	1.2	else if (!frame2.CompareTo("hbook", TString::kIgnoreCase)) ifroot2 = false;
26	mocchiut	1.1
27			if (!frame1.CompareTo("root", TString::kIgnoreCase)) ifroot1 = true;
28	pam-fi	1.2	else if (!frame1.CompareTo("hbook", TString::kIgnoreCase)) ifroot1 = false;
29	mocchiut	1.1
30			file1 = "";
31			file2 = "";
32
33	pam-fi	1.2	ostatus = 0;
34	mocchiut	1.1	};
35
36	pam-fi	1.2	/**
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	mocchiut	1.1	TrkProcess::TrkProcess(ULong64_t run, TFile *f2){
42
43			idrun = run;
44	pam-fi	1.2	// DEBUG = false;
45			// VERBOSE = false;
46			dbg_mode.SetWarning();
47	mocchiut	1.1	get1 = 0;
48			get2 = 1;
49			standalone = false;
50			frame2 = "root";
51	pam-fi	1.4	frame1 = "root";
52			// ifroot1 = false;
53			// ifroot2 = true;
54			pfolder = "/TrackerFolder";
55			file1 = "";
56	mocchiut	1.1	if(f2->IsOpen()){
57	pam-fi	1.4	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	mocchiut	1.1	}else{
63	pam-fi	1.4	file2 = "";
64			outdir = gSystem->WorkingDirectory();
65	mocchiut	1.1	};
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	pam-fi	1.2	ostatus = 0;
74	mocchiut	1.1
75			};
76	pam-fi	1.2	/**
77			* Set processing variables according to input custom parameters
78			*/
79			void TrkProcess::HandleCustomPar(int ncustom, char *vcustom[]){
80	mocchiut	1.1
81			for (int i=0; i< ncustom; i++){
82	pam-fi	1.4
83	mocchiut	1.1	// -----------------------------------------------------//
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	pam-fi	1.2	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	mocchiut	1.1	continue;
136	pam-fi	1.2	};
137	mocchiut	1.1	}
138	pam-fi	1.4
139	mocchiut	1.1	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	pam-fi	1.4
147	mocchiut	1.1	if(get1){
148	pam-fi	1.4
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	mocchiut	1.1
169			};
170
171	pam-fi	1.2	ostatus = 0;
172	mocchiut	1.1
173	pam-fi	1.2	};
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	mocchiut	1.3	// else if(F77err > 0 && WarningMode())cout<<" WARNING ("<<F77err<<") - TrkCore - Level1 reduction (CN computation failure, ...)"<<endl;
191	pam-fi	1.2	//
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	mocchiut	1.1	};
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	pam-fi	1.2	cout << " process folder " << pfolder << endl;
212	mocchiut	1.1	cout << " standalone mode " << standalone << endl;
213	pam-fi	1.2	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	mocchiut	1.1
217			}