RunInfo/src/RunInfoLevel2.cpp

//
// C/C++ headers
//
#include <fstream>
#include <sstream>
#include <iostream>
//
// ROOT headers
//
#include <TSystem.h>
#include <TString.h>
#include <TSQLServer.h>
//
// This package headers
//
#include <RunInfo.h>
#include <RunInfoVerl2.h>
//
using namespace std;
//
// Usage subroutine
//
void usage(){
  printf("\nUsage:\n");
  printf("\n RunInfo [-v] [-h] [--version] -idRun ID_RUN [-processFile filename] [-processFolder folder]\n");
  printf("\n --version       print informations about compilation and exit\n");
  printf("\n -h | --help     print this help and exit \n");
  printf("\n -v | --verbose  be verbose [default: print nothing on STDOUT]\n");
  printf("\n -idRun          ID_RUN: ID number of the run to be processed \n");
  printf("\n -processFile    filename : output filename [default ID_RUN.Level2.root]\n");
  printf("\n -processFolder  where to store temporary data [default \"runinfoFolder\"]\n");
  printf("\n Notice that parameter order does not matter. \n");
  printf("\nExample: \n\nRunInfo -idRun 1085 -processFile nomefile.root\n\n");
};
//
Bool_t existfile(TString filename){
    ifstream myfile;
    myfile.open(filename.Data());
    if ( !myfile ){
      return(false);
    };
    myfile.close();
    return(true);
}
//
Bool_t debug = false;
//
//
// Here the main
//
int main(int numinp, char *inps[]){
  //
  // Variables booking
  //
  int nul = 0;
  Int_t error = 0;
  Bool_t beverbose = false;
  UInt_t run = 0;
  TString filename;
  TString processFolder = "runinfoFolder";
  TSQLServer *dbc = 0;
  Bool_t givenid = false;
  //
  // Checking input parameters
  //
  if ( numinp > 1 ){
    for ( int i = 0; i < numinp; i++ ){
      if ( !strcmp(inps[i],"--version") ){
        RunInfoInfo(true);
        exit(0);
      };
      if ( !strcmp(inps[i],"-h") || !strcmp(inps[i],"--help") ){
        usage();
        exit(0);
      };
      if ( !strcmp(inps[i],"-idRun") ) {
        if ( numinp-1 < i+1 ){
          usage();
          exit(-3);
        };
        givenid = true;
        run = (UInt_t)atoll(inps[i+1]); 
      };
      if ( !strcmp(inps[i],"-processFile") ) {
        if ( numinp-1 < i+1 ){
          usage();
          exit(-3);
        };
        filename = (TString)inps[i+1];  
      };
      if ( !strcmp(inps[i],"-processFolder") ) {
        if ( numinp-1 < i+1 ){
          usage();
          exit(-3);
        };
        processFolder = (TString)inps[i+1];     
      };
      if ( !strcmp(inps[i],"-v") || !strcmp(inps[i],"--verbose") ) beverbose = true;
    };
  } else {
    //
    // no input parameters exit with error, we need at least the run id.
    //
    printf(" RunInfo - ERROR: you must provide a run number (at least -1)\n");
    exit(-1);
  };
  //
  // If not in verbose mode redirect to /dev/null the stdout and stderr
  //
  if ( !beverbose ){
    nul = open("/dev/null", O_CREAT | O_RDWR,S_IREAD | S_IWRITE);
    dup2(nul,1);
    dup2(nul,2);
  };
  //
  // Check that an input run number has been given
  //
  if ( !givenid ) {
    printf(" RunInfo - ERROR: you must provide a run number (at least -1)\n");
    exit(-1);
  };
  //
  char *version = RunInfoInfo(false);
  //
  // Start:
  //
  printf("\n Welcome to the RunInfo software, version %s \n",version);
  //
  // Connect to the DB
  //
  printf("\nConnecting to database... \n"); 
  //
  //  dbc = TSQLServer::Connect("mysql://localhost/","anonymous","");
  dbc = TSQLServer::Connect("mysql://srv-g2-01.ts.infn.it/pamelaflightnew","root","Calo0%Ts");
  //
  if( !dbc ) {
    printf(" RunInfo - ERROR: problems connecting to the DB (check username and passwd), exiting...\n\n");
    exit(-2);    
  };
  bool connect = dbc->IsConnected();
  //
  if( !connect ) {
    printf(" RunInfo - ERROR: problems connecting to the DB (check username and passwd), exiting...\n\n");
    exit(-2);    
  };
  printf("...connected! \n\n");
  //
  //
  //
  stringstream name;
  name.str("");
  if ( run != 0 ){
    //
    // Run is not 0, we must process only one run .
    //
    //
    // If no filename is given, use the deafult one run_id.Level2.root.
    //
    if ( !strcmp(filename.Data(),"") ){
      name << run << ".Level2.root";
    } else {
      name << filename.Data();
    };
    //
  } else {
    //
    // Run is 0, we must process all entries in the given file.
    //
    //
    // OUCH! no filename has been given, run out!
    //
    if ( !strcmp(filename.Data(),"") ) {
      printf(" RunInfo - ERROR: processing all runs but no filename is given\n");
      exit(-4);
    };
    name << filename.Data();
    //
  };
  TString outputfile;
  ItoRunInfo *runinfo = 0;
  TFile *file = 0;
  //
  stringstream temprname;
  const char* routdir = gSystem->WorkingDirectory();
  temprname.str("");
  temprname << routdir;
  temprname << "/" << processFolder.Data();  
  //
  // Output file is "outputfile"
  //
  outputfile = name.str().c_str(); 
  printf("\n Output filename is: \n %s \n\n",outputfile.Data());
  //
  // Check if file exists, if not exit with error (calorimeter needs tracker data).
  //
  if ( !existfile(outputfile.Data()) ) {
    printf(" RunInfo: creating LEVEL2 file \n");
  };
  // 
  // OK, file exists, open it in "update" mode. 
  //
  file = new TFile(outputfile.Data(),"UPDATE");
  if ( !file->IsOpen() ){
    printf(" RunInfo - ERROR: cannot open file for writing\n");
    error = -802;
    goto theend;
  };
  //
  runinfo = new ItoRunInfo(dbc,file,processFolder);
  error = runinfo->Update(run,"NONE","");
  gSystem->Unlink(temprname.str().c_str());
  //  if ( error < 0 ) goto theend;
  runinfo->Close();
  if ( file ){
    file->Write();
    file->Close();
  };
  //
 theend:
  //
  // Close the DB connection
  //
  printf("\nClose the connection to the database... \n");
  dbc->Close();
  printf("...connection terminated!\n\n");   
  //
  // Close redirection if the case.
  //
  if ( !beverbose ) close(nul);
  //
  // Return "error"
  //
  exit(error);
}
1	mocchiut	1.1	//
2			// C/C++ headers
3			//
4			#include <fstream>
5			#include <sstream>
6			#include <iostream>
7			//
8			// ROOT headers
9			//
10			#include <TSystem.h>
11			#include <TString.h>
12			#include <TSQLServer.h>
13			//
14			// This package headers
15			//
16			#include <RunInfo.h>
17			#include <RunInfoVerl2.h>
18			//
19			using namespace std;
20			//
21			// Usage subroutine
22			//
23			void usage(){
24			printf("\nUsage:\n");
25			printf("\n RunInfo [-v] [-h] [--version] -idRun ID_RUN [-processFile filename] [-processFolder folder]\n");
26			printf("\n --version print informations about compilation and exit\n");
27			printf("\n -h \| --help print this help and exit \n");
28			printf("\n -v \| --verbose be verbose [default: print nothing on STDOUT]\n");
29			printf("\n -idRun ID_RUN: ID number of the run to be processed \n");
30			printf("\n -processFile filename : output filename [default ID_RUN.Level2.root]\n");
31			printf("\n -processFolder where to store temporary data [default \"runinfoFolder\"]\n");
32			printf("\n Notice that parameter order does not matter. \n");
33			printf("\nExample: \n\nRunInfo -idRun 1085 -processFile nomefile.root\n\n");
34			};
35			//
36			Bool_t existfile(TString filename){
37			ifstream myfile;
38			myfile.open(filename.Data());
39			if ( !myfile ){
40			return(false);
41			};
42			myfile.close();
43			return(true);
44			}
45			//
46			Bool_t debug = false;
47			//
48			//
49			// Here the main
50			//
51			int main(int numinp, char *inps[]){
52			//
53			// Variables booking
54			//
55			int nul = 0;
56			Int_t error = 0;
57			Bool_t beverbose = false;
58	mocchiut	1.2	UInt_t run = 0;
59	mocchiut	1.1	TString filename;
60			TString processFolder = "runinfoFolder";
61			TSQLServer *dbc = 0;
62			Bool_t givenid = false;
63			//
64			// Checking input parameters
65			//
66			if ( numinp > 1 ){
67			for ( int i = 0; i < numinp; i++ ){
68			if ( !strcmp(inps[i],"--version") ){
69			RunInfoInfo(true);
70			exit(0);
71			};
72			if ( !strcmp(inps[i],"-h") \|\| !strcmp(inps[i],"--help") ){
73			usage();
74			exit(0);
75			};
76			if ( !strcmp(inps[i],"-idRun") ) {
77			if ( numinp-1 < i+1 ){
78			usage();
79			exit(-3);
80			};
81			givenid = true;
82	mocchiut	1.2	run = (UInt_t)atoll(inps[i+1]);
83	mocchiut	1.1	};
84			if ( !strcmp(inps[i],"-processFile") ) {
85			if ( numinp-1 < i+1 ){
86			usage();
87			exit(-3);
88			};
89			filename = (TString)inps[i+1];
90			};
91			if ( !strcmp(inps[i],"-processFolder") ) {
92			if ( numinp-1 < i+1 ){
93			usage();
94			exit(-3);
95			};
96			processFolder = (TString)inps[i+1];
97			};
98			if ( !strcmp(inps[i],"-v") \|\| !strcmp(inps[i],"--verbose") ) beverbose = true;
99			};
100			} else {
101			//
102			// no input parameters exit with error, we need at least the run id.
103			//
104			printf(" RunInfo - ERROR: you must provide a run number (at least -1)\n");
105			exit(-1);
106			};
107			//
108			// If not in verbose mode redirect to /dev/null the stdout and stderr
109			//
110			if ( !beverbose ){
111			nul = open("/dev/null", O_CREAT \| O_RDWR,S_IREAD \| S_IWRITE);
112			dup2(nul,1);
113			dup2(nul,2);
114			};
115			//
116			// Check that an input run number has been given
117			//
118			if ( !givenid ) {
119			printf(" RunInfo - ERROR: you must provide a run number (at least -1)\n");
120			exit(-1);
121			};
122			//
123			char *version = RunInfoInfo(false);
124			//
125			// Start:
126			//
127			printf("\n Welcome to the RunInfo software, version %s \n",version);
128			//
129			// Connect to the DB
130			//
131			printf("\nConnecting to database... \n");
132			//
133	mocchiut	1.2	// dbc = TSQLServer::Connect("mysql://localhost/","anonymous","");
134			dbc = TSQLServer::Connect("mysql://srv-g2-01.ts.infn.it/pamelaflightnew","root","Calo0%Ts");
135			//
136	mocchiut	1.1	if( !dbc ) {
137			printf(" RunInfo - ERROR: problems connecting to the DB (check username and passwd), exiting...\n\n");
138			exit(-2);
139			};
140			bool connect = dbc->IsConnected();
141			//
142			if( !connect ) {
143			printf(" RunInfo - ERROR: problems connecting to the DB (check username and passwd), exiting...\n\n");
144			exit(-2);
145			};
146			printf("...connected! \n\n");
147			//
148			//
149			//
150			stringstream name;
151			name.str("");
152	mocchiut	1.2	if ( run != 0 ){
153	mocchiut	1.1	//
154			// Run is not 0, we must process only one run .
155			//
156			//
157			// If no filename is given, use the deafult one run_id.Level2.root.
158			//
159			if ( !strcmp(filename.Data(),"") ){
160			name << run << ".Level2.root";
161			} else {
162			name << filename.Data();
163			};
164			//
165			} else {
166			//
167			// Run is 0, we must process all entries in the given file.
168			//
169			//
170			// OUCH! no filename has been given, run out!
171			//
172			if ( !strcmp(filename.Data(),"") ) {
173			printf(" RunInfo - ERROR: processing all runs but no filename is given\n");
174			exit(-4);
175			};
176			name << filename.Data();
177			//
178			};
179			TString outputfile;
180			ItoRunInfo *runinfo = 0;
181			TFile *file = 0;
182			//
183			stringstream temprname;
184			const char* routdir = gSystem->WorkingDirectory();
185			temprname.str("");
186			temprname << routdir;
187			temprname << "/" << processFolder.Data();
188			//
189			// Output file is "outputfile"
190			//
191			outputfile = name.str().c_str();
192			printf("\n Output filename is: \n %s \n\n",outputfile.Data());
193			//
194			// Check if file exists, if not exit with error (calorimeter needs tracker data).
195			//
196			if ( !existfile(outputfile.Data()) ) {
197			printf(" RunInfo: creating LEVEL2 file \n");
198			};
199			//
200			// OK, file exists, open it in "update" mode.
201			//
202			file = new TFile(outputfile.Data(),"UPDATE");
203			if ( !file->IsOpen() ){
204			printf(" RunInfo - ERROR: cannot open file for writing\n");
205			error = -802;
206			goto theend;
207			};
208			//
209			runinfo = new ItoRunInfo(dbc,file,processFolder);
210			error = runinfo->Update(run,"NONE","");
211			gSystem->Unlink(temprname.str().c_str());
212			// if ( error < 0 ) goto theend;
213			runinfo->Close();
214			if ( file ){
215			file->Write();
216			file->Close();
217			};
218			//
219			theend:
220			//
221			// Close the DB connection
222			//
223			printf("\nClose the connection to the database... \n");
224			dbc->Close();
225			printf("...connection terminated!\n\n");
226			//
227			// Close redirection if the case.
228			//
229			if ( !beverbose ) close(nul);
230			//
231			// Return "error"
232			//
233			exit(error);
234			}