lcdg4.cc

Go to the documentation of this file.

/****************************************
Main Driver for NICADD simulation program
6/30/02, rmcintos@nicadd.niu.edu
****************************************/

#include "G4RunManager.hh"
#include "G4UImanager.hh"
#include "LHEP.hh"
#include "QGSP.hh"

#include "LCDXML2G4DetConst.hh"
#include "LCDG4_SIOglobal.hh"
#include "LCDG4readStdFile.hh"
#include "LCDG4PrimaryGeneratorAction.hh"
#include "LCDG4RunAction.hh"
#include "LCDG4EventAction.hh"
#include "LCDG4SteppingAction.hh"
#include "LCDG4TrackingAction.hh"
#include "LCDG4McPartManager.hh"

// include some pre-defined hadronic physics lists
#include "FTFP.hh"
#include "LHEP.hh"
#include "LHEP_BERT.hh"
#include "LHEP_BIC.hh"
#include "LHEP_GN.hh"
#include "LHEP_HP.hh"
#include "QGSC.hh"
#include "QGSP.hh"
#include "QGSP_BERT.hh"
#include "QGSP_BIC.hh"

#include <iostream>
#include <fstream>
using std::cout;
using std::endl;
using std::ifstream;

#ifdef G4VIS_USE
 #include "LCDG4VisManager.hh"
#endif

int main(int argc,char *argv[])
{
  const char* env = getenv("PEGS4PATH");
  if(env == NULL) {
    cout << "Error! The PEGS4PATH environment variable is not set.\n";
    exit(0);
  }

  if(argc < 6) {
    cout << "Error! Not enough command line arguments. Usage is:\n";
    cout << "lcdg4 input_file.xml output_file.sio Detector_ID input_file.stdhep macro_file output_file.slcio\n\n";
    cout << "For example:\n";
    cout << "lcdg4 SDMar01.xml output.sio SDMar01 events.stdhep macro_file output.slcio\n";
    exit(0);
  }

  // command line arguments
  char* xml_file = argv[1];
  char* sio_file = argv[2];
  char* det_id = argv[3];
  char* stdhep_file = argv[4];
  char* macro_file = argv[5];
  char* lcio_file = argv[6];
  char* hadphys = (char*)"LHEP"; // default
  if(argc==8) hadphys = argv[7];

  cout << "The following inputs have been selected:"<< endl;
  cout << "XML File input: " << xml_file << endl;
  cout << "SIO File output: " << sio_file << endl;
  cout << "Detector ID: " << det_id << endl;
  cout << "StdHEP file: " << stdhep_file << endl;
  cout << "Macro file: " << macro_file << endl;
  cout << "LCIO File output: " << lcio_file << endl;
  cout << "PEGS4PATH = " << env << endl;

  // Open and initialize output SIO file
  LCDG4_SIOglobal sio(sio_file, det_id);
  SIO_stream* stream = sio.getStream();

  //*** Geant4 stuff ***

  // run manager
  G4RunManager* runManager = new G4RunManager;

  // detector construction
  LCDXML2G4DetConst* xmldet = new LCDXML2G4DetConst(xml_file);
  runManager->SetUserInitialization(xmldet);

  //*** LOOK HERE *** CHOOSE THE PHYSICS LIST ***
  /*-- SELECT YOUR G4 PHYSICS LIST -------------------
   * LHEP - parameterized (Gheisha)
   * LHEP-BERT - Gheisha + Bertini cascade (E<3GeV)
   * LHEP-BIC  - Gheisha + Binary cascade (E<3GeV)
   * LHEP-GN   - Gheisha + gamma-Nucleon processes
   * LHEP-HP   - Gheisha + evaluated x-sec data (for neutrons w/E<20MeV)
   * QGSP      - Gheisha + quark-gluon string model (E>25GeV): theory-driven
   * QGSP-BERT - Gheisha + Bertini (E<3GeV) + qg string model (E>25GeV)
   * QGSP-BIC  - Gheisha + Binary (E<3GeV)  + qg string model (E>25GeV)
   * FTFP      - Gheisha + qg string model (E>25GeV) with Fritjof fragmentation
   * QGSC      - Gheisha + quark-gluon string model (E>25GeV): theory-driven
   */
  G4cout << "Hadronic physics selected: "<< hadphys << G4endl;
  if(!strcmp(hadphys,"LHEP")) runManager->SetUserInitialization(new LHEP);
  if(!strcmp(hadphys,"LHEP-BERT")) runManager->SetUserInitialization(new LHEP_BERT);
  if(!strcmp(hadphys,"LHEP-BIC")) runManager->SetUserInitialization(new LHEP_BIC);
  if(!strcmp(hadphys,"LHEP-GN")) runManager->SetUserInitialization(new LHEP_GN);
  if(!strcmp(hadphys,"LHEP-HP")) runManager->SetUserInitialization(new LHEP_HP);
  if(!strcmp(hadphys,"QGSP")) runManager->SetUserInitialization(new QGSP);
  if(!strcmp(hadphys,"QGSP-BERT")) runManager->SetUserInitialization(new QGSP_BERT);
  if(!strcmp(hadphys,"QGSP-BIC")) runManager->SetUserInitialization(new QGSP_BIC);
  if(!strcmp(hadphys,"QGSC")) runManager->SetUserInitialization(new QGSC);
  if(!strcmp(hadphys,"FTFP")) runManager->SetUserInitialization(new FTFP);
  //***endLOOKHERE***

  // LCDG4 specific stuff
  int debflg=0x0; //  Check LCDG4McPartManager.cc for details
  LCDG4McPartManager* mcmgr = new LCDG4McPartManager(xmldet->GetDetTypeMap(),debflg);

  // Visualization system
#ifdef G4VIS_USE
  G4VisManager* visManager = new LCDG4VisManager;
  visManager->Initialize();
#endif

  //=========================

//   //*** set run with single particles ***
//   double energy=0;
//   G4String partType = "";
//   ifstream singfile("singfile");
//   singfile >> partType >> energy;
//   singfile.close();
//   energy *= GeV;

//   LCDG4PrimaryGeneratorAction* pgen = new LCDG4PrimaryGeneratorAction();
//   pgen->SetHEPEvtGenerator(false);  // false forces use of particle gun
//   pgen->SetConeRadius(0.15);  // false forces use of particle gun
//   pgen->SetPartGunEnergy(energy);
//   pgen->SetPartGunType(partType);

  //=======

  // StdHEP and event generation
  LCDG4readStdFile* stdfile = new LCDG4readStdFile(stdhep_file,mcmgr);
  LCDG4PrimaryGeneratorAction* pgen =
    new LCDG4PrimaryGeneratorAction(stdfile);
//   pgen->SetHEPEvtGenerator(false);  // false forces use of G4ParticleGun

  //========

  runManager->SetUserAction( pgen );

  // run action
  runManager->SetUserAction( new LCDG4RunAction(det_id) );

  // event action
  LCDG4EventAction* pEventAction = new LCDG4EventAction(stream,mcmgr);
  runManager->SetUserAction(pEventAction);

  // tracking and stepping
  runManager->SetUserAction(new LCDG4TrackingAction(mcmgr));
  runManager->SetUserAction(new LCDG4SteppingAction(mcmgr));

  // This sets up ASCII output format (NICADD studies)
  // The filename is set from TXTFILEPATH environment variable
#ifdef TXT_MODE
  pEventAction->SetupOutputTextFile();
#endif
  pEventAction->SetupOutputLcio(lcio_file);

  runManager->Initialize();

  G4UImanager* UI = G4UImanager::GetUIpointer();
  G4String command = "/control/execute ";
  G4String fileName = macro_file;
  UI->ApplyCommand(command+fileName);

  stream->close();

#ifdef G4VIS_USE
  delete visManager;
#endif

  delete runManager;
  exit(0);
}

---