source: branches/Mars_use_drstimefiles/star.cc@ 17989

Last change on this file since 17989 was 9482, checked in by tbretz, 15 years ago
*** empty log message ***
File size: 9.4 KB
Line 
1#include <TROOT.h>
2#include <TClass.h>
3#include <TSystem.h>
4#include <TGClient.h>
5#include <TApplication.h>
6#include <TObjectTable.h>
7
8#include "MLog.h"
9#include "MLogManip.h"
10
11#include "MEnv.h"
12#include "MArgs.h"
13#include "MArray.h"
14#include "MDirIter.h"
15
16#include "MStatusDisplay.h"
17
18#include "MSequence.h"
19#include "MJStar.h"
20
21using namespace std;
22
23static void StartUpMessage()
24{
25 gLog << all << endl;
26
27 // 1 2 3 4 5 6
28 // 123456789012345678901234567890123456789012345678901234567890
29 gLog << "========================================================" << endl;
30 gLog << " Star - MARS V" << MARSVER << endl;
31 gLog << " MARS -- STandard Analysis and Reconstruction" << endl;
32 gLog << " Compiled with ROOT v" << ROOT_RELEASE << " on <" << __DATE__ << ">" << endl;
33 gLog << "========================================================" << endl;
34 gLog << endl;
35}
36
37static void Usage()
38{
39 // 1 2 3 4 5 6 7 8
40 // 12345678901234567890123456789012345678901234567890123456789012345678901234567890
41 gLog << all << endl;
42 gLog << "Sorry the usage is:" << endl;
43 gLog << " star [options] sequence.txt|[tel:]number" << endl << endl;
44 gLog << " Arguments:" << endl;
45 gLog << " sequence.txt: Ascii file defining a sequence of runs" << endl;
46 gLog << " number: The sequence number (using file in the datacenter)" << endl;
47 gLog << " For more details see MSequence" << endl;
48 gLog << " Root Options:" << endl;
49 gLog << " -b Batch mode (no graphical output to screen)" << endl<<endl;
50 gLog << " Options:" << endl;
51 gLog.Usage();
52 gLog << " --debug-env=0 Disable debugging setting resources <default>" << endl;
53 gLog << " --debug-env[=1] Display untouched resources after program execution" << endl;
54 gLog << " --debug-env=2 Display untouched resources after eventloop setup" << endl;
55 gLog << " --debug-env=3 Debug setting resources from resource file and command line" << endl;
56 gLog << " --debug-mem Debug memory usage" << endl << endl;
57 gLog << " --rc=Name:option Set or overwrite a resource of the resource file." << endl;
58 gLog << " (Note, that this option can be used multiple times." << endl;
59 gLog << endl;
60 gLog << " Input Options:" << endl;
61 gLog << " -mc You must use this for MC files (PRELIMINARY)" << endl << endl;
62 gLog << " Output options:" << endl;
63 gLog << " -q Quit when job is finished" << endl;
64 gLog << " -f Force overwrite of existing files" << endl;
65// gLog << " -ff Force execution if not all files found" << endl;
66 gLog << " --ind=path Path where to search for the calibrated data (Y)" << endl;
67 gLog << " [default=standard path in datacenter]" << endl;
68 gLog << " --out=path Path to write the all results to [def=local path]" << endl;
69 gLog << " --dev-null Suppress output of I-files (for test purpose)" << endl;
70 gLog << " --no-muons Switch off Muon analysis (for fast tests)" << endl;
71 gLog << " --print-seq Print Sequence information" << endl;
72 gLog << " --print-files Print Files taken from Sequence" << endl;
73 gLog << " --print-found Print Files found from Sequence" << endl;
74 gLog << " --config=star.rc Resource file [default=star.rc]" << endl;
75 gLog << endl;
76 gLog << " --version, -V Show startup message with version number" << endl;
77 gLog << " -?, -h, --help This help" << endl << endl;
78}
79
80static void PrintFiles(const MSequence &seq, const TString &kInpathD, Bool_t allopt)
81{
82 const char *prep = allopt ? "Found" : "Scheduled";
83
84 MDirIter Next;
85 seq.GetRuns(Next, MSequence::kCalibrated, kInpathD);
86
87 gLog << all;
88 gLog.Separator(Form("%s Data Files", prep));
89 Next.Print(allopt?"all":"");
90 gLog << endl;
91}
92
93int main(int argc, char **argv)
94{
95 if (!MARS::CheckRootVer())
96 return 0xff;
97
98 MLog::RedirectErrorHandler(MLog::kColor);
99
100 //
101 // Evaluate arguments
102 //
103 MArgs arg(argc, argv);
104 gLog.Setup(arg);
105
106 StartUpMessage();
107
108 if (arg.HasOnly("-V") || arg.HasOnly("--version"))
109 return 0;
110
111 if (arg.HasOnly("-?") || arg.HasOnly("-h") || arg.HasOnly("--help"))
112 {
113 Usage();
114 return 2;
115 }
116
117 const TString kConfig = arg.GetStringAndRemove("--config=", "star.rc");
118
119 const Bool_t kPrintSeq = arg.HasOnlyAndRemove("--print-seq");
120 const Bool_t kPrintFiles = arg.HasOnlyAndRemove("--print-files");
121 const Bool_t kPrintFound = arg.HasOnlyAndRemove("--print-found");
122 const Bool_t kDebugMem = arg.HasOnlyAndRemove("--debug-mem");
123 const Bool_t kNullOut = arg.HasOnlyAndRemove("--dev-null");
124 Int_t kDebugEnv = arg.HasOnlyAndRemove("--debug-env") ? 1 : 0;
125 kDebugEnv = arg.GetIntAndRemove("--debug-env=", kDebugEnv);
126
127 const Bool_t kQuit = arg.HasOnlyAndRemove("-q");
128 const Bool_t kBatch = arg.HasOnlyAndRemove("-b");
129 const Bool_t kOverwrite = arg.HasOnlyAndRemove("-f");
130 //const Bool_t kForceExec = arg.HasOnlyAndRemove("-ff");
131 const Bool_t kIsMC = arg.HasOnlyAndRemove("-mc");
132 const Bool_t kNoMuons = arg.HasOnlyAndRemove("--no-muons");
133
134 const TString kInpath = arg.GetStringAndRemove("--ind=", "");
135 const TString kOutpath = arg.GetStringAndRemove("--out=", ".");
136
137 //
138 // check for the right usage of the program (number of arguments)
139 //
140 if (arg.GetNumArguments()!=1)
141 {
142 gLog << warn << "WARNING - Wrong number of arguments..." << endl;
143 Usage();
144 return 2;
145 }
146
147 //
148 // Now we access/read the resource file. This will remove all
149 // --rc= from the list of arguments.
150 //
151 MEnv env(kConfig, "star.rc");
152 if (!env.IsValid())
153 {
154 gLog << err << "ERROR - Reading resource file " << kConfig << "." << endl;
155 return 0xfe;
156 }
157
158 // And move the resource options from the command line to the MEnv
159 if (!env.TakeEnv(arg, kDebugEnv>2))
160 return 0xfd;
161
162 //
163 // check for the right usage of the program (number of options)
164 //
165 if (arg.GetNumOptions()>0)
166 {
167 gLog << warn << "WARNING - Unknown commandline options..." << endl;
168 arg.Print("options");
169 gLog << endl;
170 return 2;
171 }
172
173 //
174 // Setup sequence file and check for its existance
175 //
176 TString kSequence = arg.GetArgumentStr(0);
177
178 //
179 // Something special for datacenter access
180 //
181 if (!MSequence::InflateSeq(kSequence, kIsMC))
182 return 2;
183
184 //
185 // Setup sequence and check its validity
186 //
187 MSequence seq(kSequence, kInpath);
188 if (!seq.IsMonteCarlo())
189 seq.SetMonteCarlo(kIsMC);
190 if (kPrintSeq)
191 {
192 gLog << all;
193 gLog.Separator(kSequence);
194 seq.Print();
195 gLog << endl;
196 }
197 if (!seq.IsValid())
198 {
199 gLog << err << "Sequence read but not valid!" << endl << endl;
200 return 2;
201 }
202
203 //
204 // Process print options
205 //
206 if (kPrintFiles)
207 PrintFiles(seq, kInpath, kFALSE);
208 if (kPrintFound)
209 PrintFiles(seq, kInpath, kTRUE);
210
211 //
212 // Initialize root
213 //
214 MArray::Class()->IgnoreTObjectStreamer();
215 MParContainer::Class()->IgnoreTObjectStreamer();
216
217 TApplication app("star", &argc, argv);
218 if ((!gROOT->IsBatch() && !gClient) || (gROOT->IsBatch() && !kBatch))
219 {
220 gLog << err << "Bombing... maybe your DISPLAY variable is not set correctly!" << endl;
221 return 1;
222 }
223
224 //
225 // Update frequency by default = 1Hz
226 //
227 MStatusDisplay *d = new MStatusDisplay;
228
229 // From now on each 'Exit' means: Terminate the application
230 d->SetBit(MStatusDisplay::kExitLoopOnExit);
231 d->SetTitle(Form("-- Star: %s --", kSequence.Data()));
232
233 if (kDebugMem)
234 TObject::SetObjectStat(kTRUE);
235
236 //
237 // Do star in a block (debug mem)
238 //
239 {
240 MJStar job(Form("MJStar #%d", seq.GetSequence()));
241 job.SetSequence(seq);
242 job.SetEnv(&env);
243 job.SetEnvDebug(kDebugEnv);
244 job.SetDisplay(d);;
245 job.SetOverwrite(kOverwrite);
246 job.SetPathOut(kOutpath);
247 job.SetNullOut(kNullOut);
248 job.SetCommandLine(MArgs::GetCommandLine(argc, argv));
249 // job.SetPathIn(kInpath); // not yet needed
250 if (kNoMuons)
251 job.DisableMuonAnalysis();
252
253 if (!job.Process())
254 {
255 gLog << err << "Calculation of image parameters failed." << endl << endl;
256 return 2;
257 }
258
259 if (kDebugEnv>0)
260 env.PrintUntouched();
261
262 if (!job.GetDisplay())
263 {
264 gLog << warn << "Display closed by user... execution aborted." << endl << endl;
265 return 1;
266 }
267 }
268
269 if (kBatch || kQuit)
270 delete d;
271 else
272 {
273 // From now on each 'Close' means: Terminate the application
274 d->SetBit(MStatusDisplay::kExitLoopOnClose);
275
276 // Wait until the user decides to exit the application
277 app.Run(kFALSE);
278 }
279
280 if (TObject::GetObjectStat())
281 {
282 TObject::SetObjectStat(kFALSE);
283 gObjectTable->Print();
284 }
285
286 return 0;
287}
Note: See TracBrowser for help on using the repository browser.