Geant4  10.01.p03
CommandLineParser.cc
Go to the documentation of this file.
1 //
2 // ********************************************************************
3 // * License and Disclaimer *
4 // * *
5 // * The Geant4 software is copyright of the Copyright Holders of *
6 // * the Geant4 Collaboration. It is provided under the terms and *
7 // * conditions of the Geant4 Software License, included in the file *
8 // * LICENSE and available at http://cern.ch/geant4/license . These *
9 // * include a list of copyright holders. *
10 // * *
11 // * Neither the authors of this software system, nor their employing *
12 // * institutes,nor the agencies providing financial support for this *
13 // * work make any representation or warranty, express or implied, *
14 // * regarding this software system or assume any liability for its *
15 // * use. Please see the license in the file LICENSE and URL above *
16 // * for the full disclaimer and the limitation of liability. *
17 // * *
18 // * This code implementation is the result of the scientific and *
19 // * technical work of the GEANT4 collaboration. *
20 // * By using, copying, modifying or distributing the software (or *
21 // * any work based on the software) you agree to acknowledge its *
22 // * use in resulting scientific publications, and indicate your *
23 // * acceptance of all terms of the Geant4 Software license. *
24 // ********************************************************************
25 //
26 // This example is provided by the Geant4-DNA collaboration
27 // Any report or published results obtained using the Geant4-DNA software
28 // shall cite the following Geant4-DNA collaboration publication:
29 // Med. Phys. 37 (2010) 4692-4708
30 // J. Comput. Phys. 274 (2014) 841-882
31 // The Geant4-DNA web site is available at http://geant4-dna.org
32 //
33 // Author: Mathieu Karamitros
34 //
35 // $Id$
36 //
39 
40 #include <iomanip>
41 #include "CommandLineParser.hh"
42 
43 using namespace std;
44 using namespace G4DNAPARSER;
45 
46 CommandLineParser* CommandLineParser::fpInstance(0);
47 G4String Command::fNoOption = "NoOption";
48 
49 
50 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
51 
52 inline bool MATCH(const char *a, const char *b)
53 {
54  return strcmp(a, b) == 0;
55 }
56 
57 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
58 
59 CommandLineParser::CommandLineParser()
60 {
61  // G4cout << "############ NEW PARSE ##########" << G4endl;
62  fpInstance = this;
63  fOptionsWereSetup = false;
64  fMaxMarkerLength = 0;
65  fMaxOptionNameLength = 0;
66  AddCommand("--help", Command::WithoutOption, "Print this help");
67  AddCommand("-h", Command::WithoutOption, "Print this help");
68  AddCommand("&", Command::WithoutOption);
69 
70  fVerbose = 0;
71 }
72 
73 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
74 
75 CommandLineParser* CommandLineParser::GetParser()
76 {
77  if (!fpInstance) new CommandLineParser;
78  return fpInstance;
79 }
80 
81 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
82 
83 CommandLineParser::~CommandLineParser()
84 {
85  std::map<G4String, Command*>::iterator it = fCommandMap.begin();
86  for (; it != fCommandMap.end(); it++)
87  {
88  if (it->second) delete it->second;
89  }
90 }
91 
92 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
93 
94 void CommandLineParser::DeleteInstance()
95 {
96  if (fpInstance)
97  {
98  delete fpInstance;
99  fpInstance = 0;
100  }
101 }
102 
103 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
104 
105 Command::Command(Command::Type commandType,
106  const G4String& description)
107 {
108  fType = commandType;
109  fDescription = description;
110  fActive = false;
111 }
112 
113 CommandWithOption::CommandWithOption(Command::Type commandType,
114  const G4String& description,
115  const G4String& defaultOption,
116  const G4String& optionName) :
117 Command(commandType, description)
118 {
119  fDefaultOption = defaultOption;
120  fOptionName = optionName;
121  fOption = "";
122 }
123 
124 
125 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
126 
127 int CommandLineParser::Parse(int& argc, char **argv)
128 {
129  // G4cout << "Parse " << G4endl;
130  static char null[1] = { "" };
131  int firstArgc = argc;
132 
133  for (int i = 1; i < firstArgc; i++)
134  {
135  Command* command = FindCommand(argv[i]);
136  if (command == 0) continue;
137 
138  if (fVerbose) G4cout << "Command : " << argv[i] << G4endl;
139 
140  fOptionsWereSetup = true;
141  command->fActive = true;
142 
143  G4String marker(argv[i]);
144 
145  if (strcmp(argv[i], "-h") != 0 && strcmp(argv[i], "--help") != 0)
146  {
147  argv[i] = null;
148  }
149 
150  if (command->fType == Command::WithOption)
151  {
152  if (fVerbose) G4cout << "WithOption" << G4endl;
153 
154  if(i+1 > firstArgc || argv[i+1]==0 || argv[i+1][0]=='-')
155  {
156  G4cerr << "An command line option is missing for "
157  << marker << G4endl;
158  abort();
159  }
160 
161  command->SetOption( (const char*) strdup(argv[i+1]) );
162  argv[i+1] = null;
163  i++;
164  }
165  else if(command->fType == Command::OptionNotCompulsory)
166  {
167  if(fVerbose)
168  G4cout <<"OptionNotCompulsory"<<G4endl;
169 
170  if(i+1 < firstArgc)
171  {
172  G4String buffer = (const char*) strdup(argv[i+1]);
173 
174  if(buffer.empty() == false)
175  {
176  if(buffer.at(0) != '-'
177  && buffer.at(0) != '&'
178  && buffer.at(0) != '>'
179  && buffer.at(0) != '|')
180  {
181  if(fVerbose)
182  {
183  G4cout << "facultative option is : " << buffer << G4endl;
184  }
185 
186  command->SetOption( (const char*) strdup(argv[i+1]) );
187  argv[i+1] = null;
188  i++;
189  continue;
190  }
191  }
192  }
193 
194  if(fVerbose)
195  G4cout << "Option not set" << G4endl;
196 
197  command->SetOption("");
198  }
199  }
200  CorrectRemainingOptions(argc, argv);
201 
202  Command* commandLine(0);
203  if ((commandLine = GetCommandIfActive("--help")) || (commandLine =
204  GetCommandIfActive("-h")))
205  {
206  G4cout << "Usage : " << argv[0] << " [OPTIONS]" << G4endl;
207  PrintHelp();
208  return 1;
209  }
210 
211  return 0;
212 }
213 
214 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
215 
217 {
218  std::map<G4String, Command*>::iterator it;
219 
220  int maxFieldLength = fMaxMarkerLength + fMaxOptionNameLength + 4;
221 
222  G4cout << "Options: " << G4endl;
223 
224  for (it = fCommandMap.begin(); it != fCommandMap.end(); it++)
225  {
226  Command* command = it->second;
227  if (command)
228  {
229  G4cout << setw(maxFieldLength) << left;
230 
231  G4String toPrint = it->first;
232 
233  if (toPrint == "&")
234  {
235  continue;
236  }
237  else if (toPrint == "-h") continue;
238  else if (toPrint == "--help")
239  {
240  toPrint += ", -h";
241  }
242 
243  if (command->GetDefaultOption() != "")
244  {
245  toPrint += " \"" + command->GetDefaultOption() + "\"";
246  }
247 
248  G4cout << toPrint;
249 
250  G4cout << command->GetDescription() << G4endl;
251  }
252  }
253 }
254 
255 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
256 
257 void CommandLineParser::CorrectRemainingOptions(int& argc, char **argv)
258 {
259  // remove handled arguments from argument array
260  int j = 0;
261  for (int i = 0; i < argc; i++)
262  {
263  if (strcmp(argv[i], ""))
264  {
265  argv[j] = argv[i];
266  j++;
267  }
268  }
269  argc = j;
270 }
271 
272 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
273 
275  Command::Type type,
276  const G4String& description,
277  const G4String& defaultOption,
278  const G4String& optionName)
279 {
280  // G4cout << "Add command : "<< marker << G4endl;
281 
282  Command* command = 0;
283  switch(type)
284  {
286  command = new Command(type, description);
287  break;
288 
289  default:
290  command = new CommandWithOption(type,
291  description,
292  defaultOption,
293  optionName);
294  if ((int) defaultOption.length() > fMaxOptionNameLength)
295  fMaxOptionNameLength = defaultOption.length();
296  break;
297  }
298 
299  if ((int) marker.length() > fMaxMarkerLength) fMaxMarkerLength =
300  marker.length();
301  fCommandMap.insert(make_pair(marker, command));
302 }
303 
304 /*
305 // Add one command but multiple markers
306 void Parser::AddCommand(vector<G4String> markers,
307  CommandType type,
308  const G4String& description,
309  const G4String& optionName)
310 {
311  // G4cout << "Add command : "<< marker << G4endl;
312  Command* command = new Command(type, description, optionName);
313 
314  for (size_t i = 0; i < markers.size; i++)
315  {
316  G4String marker = markers[i];
317  if ((int) marker.length() > fMaxMarkerLength)
318  {
319  fMaxMarkerLength = marker.length();
320  }
321  if ((int) optionName.length() > fMaxOptionNameLength)
322  {
323  fMaxOptionNameLength = optionName.length();
324  }
325  fCommandMap.insert(make_pair(marker, command));
326  }
327 }
328 */
329 
330 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
331 
333 {
334  std::map<G4String, Command*>::iterator it = fCommandMap.find(marker);
335  if (it == fCommandMap.end())
336  {
337  // G4cerr << "command not found" << G4endl;
338  return 0;
339  }
340  return it->second;
341 }
342 
343 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
344 
346 {
347  Command* command = FindCommand(marker);
348  if (command)
349  {
350  // G4cout << "Command found : "<< marker << G4endl;
351 
352  if (command->fActive)
353  {
354  // G4cout << "Command Active" << G4endl;
355  return command;
356  }
357  // else
358  // G4cout <<"Command not active" << G4endl;
359  }
360  else
361  {
362  G4ExceptionDescription description;
363  description << "You try to retrieve a command that was not registered : "
364  << marker << G4endl;
365  G4Exception("CommandLineParser::GetCommandIfActive",
366  "COMMAND LINE NOT DEFINED", FatalException, description, "");
367  // If you are using this class outside of Geant4, use exit(-1) instead
368  //exit(-1);
369  }
370  return 0;
371 }
372 
373 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
374 
376 {
377  if (argc > 0)
378  {
379  G4bool kill = false;
380  for (G4int i = 1; i < argc; i++)
381  {
382  if (strcmp(argv[i], ""))
383  {
384  kill = true;
385  G4cerr << "Unknown argument : " << argv[i] << "\n";
386  }
387  }
388  if (kill)
389  {
390  G4cerr << "The option " << argv[0]
391  << " is not handled this programme." << G4endl;
392  G4cout << "Usage : " << argv[0] << " [OPTIONS]" << G4endl;
393  PrintHelp();
394  return true; // KILL APPLICATION
395  }
396  }
397  return false;
398 }
int Parse(int &argc, char **argv)
std::ostringstream G4ExceptionDescription
Definition: globals.hh:76
virtual const G4String & GetDefaultOption()
G4int first(char) const
void AddCommand(const G4String &marker, Command::Type, const G4String &description="", const G4String &defaultOption="", const G4String &optionName="")
Command * GetCommandIfActive(const G4String &marker)
const G4String & GetDescription()
virtual void SetOption(const G4String &)
bool CheckIfNotHandledOptionsExists(int &argc, char **argv)
#define buffer
Definition: xmlparse.cc:611
G4double a
Definition: TRTMaterials.hh:39
int G4int
Definition: G4Types.hh:78
std::map< G4String, Command * > fCommandMap
G4GLOB_DLL std::ostream G4cout
bool G4bool
Definition: G4Types.hh:79
void G4Exception(const char *originOfException, const char *exceptionCode, G4ExceptionSeverity severity, const char *comments)
Definition: G4Exception.cc:41
void CorrectRemainingOptions(int &argc, char **argv)
bool MATCH(const char *a, const char *b)
#define G4endl
Definition: G4ios.hh:61
Command * FindCommand(const G4String &marker)
G4GLOB_DLL std::ostream G4cerr