G4Setup_EICDetector.C

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file G4Setup_EICDetector.C

#ifndef MACRO_G4SETUPEICDETECTOR_C
#define MACRO_G4SETUPEICDETECTOR_C

#include <G4_BMMG.C>
#include <G4_TRD.C>

#include <GlobalVariables.C>

#include <G4_BlackHole.C>

//#include <G4_AllSilicon.C>
//#include <G4_Mvtx_EIC.C>
#include <G4_BECAL.C>
#include <G4_Barrel_EIC.C>
#include <G4_CEmc_EIC.C>
#include <G4_DIRC.C>
#include <G4_DRCALO.C>
#include <G4_EEMC.C>
#include <G4_EEMC_hybrid.C>
#include <G4_EHCAL.C>
#include <G4_FEMC_EIC.C>
#include <G4_FHCAL.C>
#include <G4_B0ECAL.C> //for B0 ECAL
#include <G4_BWD.C> //for Far Backward Detectors
#include <G4_FST_EIC.C>
#include <G4_GEM_EIC.C>
#include <G4_HcalIn_ref.C>
#include <G4_HcalOut_ref.C>
#include <G4_Input.C>
#include <G4_LFHCAL.C>
#include <G4_Magnet.C>
#include <G4_Pipe_EIC.C>
#include <G4_PlugDoor_EIC.C>
#include <G4_TTL_EIC.C>
#include <G4_TrackingSupport.C>
#include <G4_Tracking_EIC.C>
#include <G4_B0Tracking_EIC.C> //for B0 Tracking
#include <G4_dRICH.C>
#include <G4_mRICH.C>
#include <G4_mRwell_EIC.C>
#include <G4_BToF.C>
#include <G4_HToF.C>
#include <G4_EToF.C>

// these two has to be ordered this way for now.
#include <G4_hFarFwdBeamLine_EIC.C>
#include <G4_hFarBwdBeamLine_EIC.C>

#include <G4_User.C>
#include <G4_World.C>

#include <g4detectors/PHG4CylinderSubsystem.h>
#include <eicg4b0/EICG4B0Subsystem.h>
#include <eicg4b0ecal/EICG4B0ECALSubsystem.h>
#include <eicg4bwd/EICG4BwdSubsystem.h>

#include <g4eval/PHG4DstCompressReco.h>

#include <g4main/PHG4Reco.h>
#include <g4main/PHG4TruthSubsystem.h>

#include <phfield/PHFieldConfig.h>

#include <g4decayer/EDecayType.hh>

#include <fun4all/Fun4AllDstOutputManager.h>
#include <fun4all/Fun4AllServer.h>

R__LOAD_LIBRARY(libg4decayer.so)
R__LOAD_LIBRARY(libg4detectors.so)

void G4Init()
{
  // First some check for subsystems which do not go together

  if (Enable::IP6 and Enable::IP8)
  {
    cout << "Can not enable Enable::IP6 and Enable::IP8 at the same time!" << endl;
    gSystem->Exit(1);
  }
  if (Enable::IP6 == false and Enable::IP8 == false)
  {
    cout << "None of the possible EIC IPs were selected: Enable::IP6 and Enable::IP8 !" << endl;
    gSystem->Exit(1);
  }

  if (Enable::EEMC and Enable::EEMCH)
  {
    cout << "Can not enable EEMC and EEMCH at the same time!" << endl;
    gSystem->Exit(1);
  }
  if (Enable::CEMC and Enable::BECAL)
  {
    cout << "Can not enable CEMC and BECAL at the same time!" << endl;
    gSystem->Exit(1);
  }
 if (Enable::BTOF and Enable::CTTL)
  {
    cout << "Can not enable BTOF and CTTL at the same time!" << endl;
    gSystem->Exit(1);
  }
  if (Enable::ETOF and Enable::ETTL)
  {
    cout << "Can not enable ETOF and ETTL at the same time!" << endl;
    gSystem->Exit(1);
  }
  if (Enable::HTOF and Enable::FTTL)
  {
    cout << "Can not enable HTOF and FTTL at the same time!" << endl;
    gSystem->Exit(1);
  }
  if (Enable::BMMG and Enable::DIRC)
  {
    cout << "Can not enable BMMG and DIRC at the same time!" << endl;
    gSystem->Exit(1);
  }
  if ((Enable::TRD||Enable::TRD_GAS) and Enable::RICH)
  {
    cout << "Can not enable TRD* and RICH at the same time!" << endl;
    gSystem->Exit(1);
  }
  if ((Enable::LFHCAL||Enable::EHCAL) and Enable::PLUGDOOR)
  {
    cout << "Can not enable *HCal and PLUGDOOR at the same time!" << endl;
    gSystem->Exit(1);
  }
  // load detector/material macros and execute Init() function
  if (Enable::PIPE) PipeInit();
  if (Enable::PLUGDOOR) PlugDoorInit();
  if (Enable::TRACKING) TrackingInit();
  if (Enable::B0TRACKING) B0TrackingInit();

  //Farforward/backward
  if (Enable::HFARFWD_MAGNETS) hFarBwdBeamLineInit();  //Shouldnt this be far backward enables
  if (Enable::HFARFWD_MAGNETS) hFarFwdBeamLineInit();

  //Barrel
  if (Enable::TrackingService) TrackingServiceInit();
  if (Enable::BARREL) BarrelInit();
  if (Enable::RWELL) RWellInit();
  if (Enable::CEMC) CEmcInit(72);  // make it 2*2*2*3*3 so we can try other combinations
  if (Enable::BECAL) BECALInit();
  if (Enable::HCALIN) HCalInnerInit(1);
  if (Enable::MAGNET) MagnetInit();
  MagnetFieldInit();  // We want the field - even if the magnet volume is disabled
  if (Enable::HCALOUT) HCalOuterInit();
  if (Enable::DIRC) DIRCInit();
  if (Enable::BTOF) BToFInit();
  if (Enable::BMMG) BMMGInit();
  
  //Forward
  if (Enable::FGEM) FGEM_Init();
  if (Enable::FEMC) FEMCInit();
  if (Enable::DRCALO) DRCALOInit();
  if (Enable::FHCAL) FHCALInit();
  if (Enable::LFHCAL) LFHCALInit();
  if (Enable::RICH) RICHInit();
  if (Enable::TRD) TRDInit();
  if (Enable::HTOF) HTOFInit();
  if (Enable::B0ECAL) B0ECALInit();

  //Backward
  if (Enable::EGEM) EGEM_Init();
  if (Enable::EEMC) EEMCInit();
  if (Enable::EEMCH) EEMCHInit();
  if (Enable::EHCAL) EHCALInit();
  if (Enable::mRICH) mRICHInit();
  if (Enable::ETOF) ETOFInit();
  if (Enable::BWD) BWDInit();

  //Combined
  if (Enable::FST) FST_Init();
  if (Enable::FTTL || Enable::ETTL || Enable::CTTL) TTL_Init();

  if (Enable::USER) UserInit();
  if (Enable::BLACKHOLE) BlackHoleInit();
}

int G4Setup()
{
  //---------------
  // Fun4All server
  //---------------

  Fun4AllServer *se = Fun4AllServer::instance();

  PHG4Reco *g4Reco = new PHG4Reco();

  WorldInit(g4Reco);

  g4Reco->set_rapidity_coverage(1.1);  // according to drawings
                                       // uncomment to set QGSP_BERT_HP physics list for productions
                                       // (default is QGSP_BERT for speed)
  //  g4Reco->SetPhysicsList("QGSP_BERT_HP");

  if (G4P6DECAYER::decayType != EDecayType::kAll)
  {
    g4Reco->set_force_decay(G4P6DECAYER::decayType);
  }

  double fieldstrength;
  istringstream stringline(G4MAGNET::magfield);
  stringline >> fieldstrength;
  if (stringline.fail())
  {  // conversion to double fails -> we have a string

    if (G4MAGNET::magfield.find("sPHENIX.root") != string::npos)
//    if (G4MAGNET::magfield.find("B0MagField_all_v2.root") != string::npos) // for B0 Tracking
//    if (G4MAGNET::magfield.find("/cvmfs/eic.opensciencegrid.org/ecce/gcc-8.3/opt/fun4all/core/calibrations/Field/Map/sphenix3dbigmapxyz.root") != string::npos)
    {
//  std::cout <<"G4 Setup: Using B0MagField_all_v1.root as 3D fieldMap"<<std::endl;
      g4Reco->set_field_map(G4MAGNET::magfield, PHFieldConfig::Field3DCartesian);
    }
    else
    {
      g4Reco->set_field_map(G4MAGNET::magfield, PHFieldConfig::kField2D);
    }
  }
  else
  {
    g4Reco->set_field(fieldstrength);  // use const soleniodal field
  }
  g4Reco->set_field_rescale(G4MAGNET::magfield_rescale);

  // the radius is an older protection against overlaps, it is not
  // clear how well this works nowadays but it doesn't hurt either
  double radius = 0.;

  if (Enable::PIPE) radius = Pipe(g4Reco, radius);

  // Far Forward Region
  if (Enable::HFARFWD_MAGNETS_IP6 || Enable::HFARFWD_MAGNETS_IP8) hFarFwdDefineMagnets(g4Reco);
  if (Enable::HFARFWD_VIRTUAL_DETECTORS_IP6) hFarFwdDefineDetectorsIP6(g4Reco);
  if (Enable::HFARFWD_VIRTUAL_DETECTORS_IP8) hFarFwdDefineDetectorsIP8(g4Reco);

  // Far Backward Region
  if (Enable::HFARBWD_MAGNETS_IP6 || Enable::HFARBWD_MAGNETS_IP8) hFarBwdDefineMagnets(g4Reco);
  if (Enable::HFARBWD_VIRTUAL_DETECTORS_IP6) hFarBwdDefineDetectorsIP6(g4Reco);
  if (Enable::HFARBWD_VIRTUAL_DETECTORS_IP8) hFarBwdDefineDetectorsIP8(g4Reco);

  //Barrel
  if (Enable::TrackingService) TrackingService(g4Reco, radius);

  if (Enable::RWELL) RWellSetup(g4Reco);
  if (Enable::FST) FSTSetup(g4Reco);
  if (Enable::CTTL) CTTLSetup(g4Reco);
  if (Enable::BARREL) Barrel(g4Reco);
  if (Enable::CEMC) radius = CEmc(g4Reco, radius);
  if (Enable::BECAL) BECALSetup(g4Reco);
  if (Enable::HCALIN) radius = HCalInner(g4Reco, radius, 4);
  if (Enable::MAGNET) radius = Magnet(g4Reco, radius);
  if (Enable::HCALOUT) radius = HCalOuter(g4Reco, radius, 4);
  if (Enable::DIRC) DIRCSetup(g4Reco);
  if (Enable::BTOF) BToFSetup(g4Reco);
  if (Enable::BMMG) BMMGSetup(g4Reco);

  //Forward
  if (Enable::FGEM) FGEMSetup(g4Reco);
  if (Enable::FTTL) FTTLSetup(g4Reco);
  if (Enable::FEMC) FEMCSetup(g4Reco);
  if (Enable::DRCALO) DRCALOSetup(g4Reco);
  if (Enable::FHCAL) FHCALSetup(g4Reco);
  if (Enable::LFHCAL) LFHCALSetup(g4Reco);
  if (Enable::RICH) RICHSetup(g4Reco);
  if (Enable::TRD) TRDSetup(g4Reco);
  if (Enable::HTOF) HTOFSetup(g4Reco);
  
  //Backward
  if (Enable::ETTL) ETTLSetup(g4Reco);
  if (Enable::EGEM) EGEMSetup(g4Reco);
  if (Enable::EEMC) EEMCSetup(g4Reco);
  if (Enable::EEMCH) EEMCHSetup(g4Reco);
  if (Enable::EHCAL) EHCALSetup(g4Reco);
  if (Enable::mRICH) mRICHSetup(g4Reco);
  if (Enable::ETOF) ETOFSetup(g4Reco);

  //----------------------------------------
  // sPHENIX forward flux return door
  if (Enable::PLUGDOOR) PlugDoor(g4Reco);

  if (Enable::USER) UserDetector(g4Reco);

  //----------------------------------------
  // BLACKHOLE if enabled, needs info from all previous sub detectors for dimensions
  if (Enable::BLACKHOLE) BlackHole(g4Reco, radius);

  PHG4TruthSubsystem *truth = new PHG4TruthSubsystem();
  g4Reco->registerSubsystem(truth);
  // finally adjust the world size in case the default is too small
  WorldSize(g4Reco, radius);

  se->registerSubsystem(g4Reco);
  return 0;
}

void ShowerCompress()
{
  Fun4AllServer *se = Fun4AllServer::instance();

  PHG4DstCompressReco *compress = new PHG4DstCompressReco("PHG4DstCompressReco");
  compress->AddHitContainer("G4HIT_PIPE");

  //
  //  compress->AddHitContainer("G4HIT_ZDC");
  //  compress->AddHitContainer("G4HIT_RomanPots");
  compress->AddHitContainer("G4HIT_b0Truth");
  compress->AddHitContainer("G4HIT_FIELDCAGE");

  compress->AddHitContainer("G4HIT_CEMC_ELECTRONICS");
  compress->AddHitContainer("G4HIT_CEMC");
  compress->AddHitContainer("G4HIT_ABSORBER_CEMC");
  compress->AddHitContainer("G4HIT_CEMC_SPT");
  compress->AddCellContainer("G4CELL_CEMC");
  compress->AddTowerContainer("TOWER_SIM_CEMC");
  compress->AddTowerContainer("TOWER_RAW_CEMC");
  compress->AddTowerContainer("TOWER_CALIB_CEMC");

  compress->AddHitContainer("G4HIT_BECAL");
  compress->AddHitContainer("G4HIT_ABSORBER_BECAL");
  compress->AddCellContainer("G4CELL_BECAL");
  compress->AddTowerContainer("TOWER_SIM_BECAL");
  compress->AddTowerContainer("TOWER_RAW_BECAL");
  compress->AddTowerContainer("TOWER_CALIB_BECAL");

  compress->AddHitContainer("G4HIT_HCALIN");
  compress->AddHitContainer("G4HIT_ABSORBER_HCALIN");
  compress->AddHitContainer("G4HIT_HCALIN_SPT");
  compress->AddCellContainer("G4CELL_HCALIN");
  compress->AddTowerContainer("TOWER_SIM_HCALIN");
  compress->AddTowerContainer("TOWER_RAW_HCALIN");
  compress->AddTowerContainer("TOWER_CALIB_HCALIN");

  compress->AddHitContainer("G4HIT_MAGNET");

  compress->AddHitContainer("G4HIT_HCALOUT");
  compress->AddHitContainer("G4HIT_ABSORBER_HCALOUT");
  compress->AddCellContainer("G4CELL_HCALOUT");
  compress->AddTowerContainer("TOWER_SIM_HCALOUT");
  compress->AddTowerContainer("TOWER_RAW_HCALOUT");
  compress->AddTowerContainer("TOWER_CALIB_HCALOUT");

  compress->AddHitContainer("G4HIT_BH_1");
  compress->AddHitContainer("G4HIT_BH_FORWARD_PLUS");
  compress->AddHitContainer("G4HIT_BH_FORWARD_NEG");

  compress->AddHitContainer("G4HIT_FEMC");
  compress->AddHitContainer("G4HIT_ABSORBER_FEMC");
  compress->AddCellContainer("G4CELL_FEMC");
  compress->AddTowerContainer("TOWER_SIM_FEMC");
  compress->AddTowerContainer("TOWER_RAW_FEMC");
  compress->AddTowerContainer("TOWER_CALIB_FEMC");

  compress->AddHitContainer("G4HIT_DRCALO");
  compress->AddHitContainer("G4HIT_ABSORBER_DRCALO");
  compress->AddCellContainer("G4CELL_DRCALO");
  compress->AddTowerContainer("TOWER_SIM_DRCALO");
  compress->AddTowerContainer("TOWER_RAW_DRCALO");
  compress->AddTowerContainer("TOWER_CALIB_DRCALO");

  compress->AddHitContainer("G4HIT_FHCAL");
  compress->AddHitContainer("G4HIT_ABSORBER_FHCAL");
  compress->AddCellContainer("G4CELL_FHCAL");
  compress->AddTowerContainer("TOWER_SIM_FHCAL");
  compress->AddTowerContainer("TOWER_RAW_FHCAL");
  compress->AddTowerContainer("TOWER_CALIB_FHCAL");

  compress->AddHitContainer("G4HIT_LFHCAL");
  compress->AddHitContainer("G4HIT_ABSORBER_LFHCAL");
  compress->AddCellContainer("G4CELL_LFHCAL");
  compress->AddTowerContainer("TOWER_SIM_LFHCAL");
  compress->AddTowerContainer("TOWER_RAW_LFHCAL");
  compress->AddTowerContainer("TOWER_CALIB_LFHCAL");

  compress->AddHitContainer("G4HIT_EEMC");
  compress->AddHitContainer("G4HIT_EEMC_glass");
  compress->AddHitContainer("G4HIT_ABSORBER_EEMC");
  compress->AddCellContainer("G4CELL_EEMC");
  compress->AddTowerContainer("TOWER_SIM_EEMC");
  compress->AddTowerContainer("TOWER_RAW_EEMC");
  compress->AddTowerContainer("TOWER_CALIB_EEMC");

  compress->AddHitContainer("G4HIT_EHCAL");
  compress->AddHitContainer("G4HIT_ABSORBER_EHCAL");
  compress->AddTowerContainer("TOWER_SIM_EHCAL");
  compress->AddTowerContainer("TOWER_RAW_EHCAL");
  compress->AddTowerContainer("TOWER_CALIB_EHCAL");

  compress->AddHitContainer("G4HIT_B0ECAL");
  compress->AddHitContainer("G4HIT_ABSORBER_B0ECAL");
  compress->AddCellContainer("G4CELL_B0ECAL");
  compress->AddTowerContainer("TOWER_SIM_B0ECAL");
  compress->AddTowerContainer("TOWER_RAW_B0ECAL");
  compress->AddTowerContainer("TOWER_CALIB_B0ECAL");

  compress->AddHitContainer("G4HIT_BWD");
  compress->AddHitContainer("G4HIT_ABSORBER_BWD");
  compress->AddCellContainer("G4CELL_BWD");
  compress->AddTowerContainer("TOWER_SIM_BWD");
  compress->AddTowerContainer("TOWER_RAW_BWD");
  compress->AddTowerContainer("TOWER_CALIB_BWD");

  se->registerSubsystem(compress);

  return;
}

void DstCompress(Fun4AllDstOutputManager *out)
{
  if (out)
  {
    out->StripNode("G4HIT_PIPE");

    //
    //    out->StripNode("G4HIT_ZDC");
    //    out->StripNode("G4HIT_RomanPots");
    out->StripNode("G4HIT_b0Truth");
    out->StripNode("G4HIT_BWD");
    out->StripNode("G4HIT_SVTXSUPPORT");
    out->StripNode("G4HIT_CEMC_ELECTRONICS");
    out->StripNode("G4HIT_CEMC");
    out->StripNode("G4HIT_ABSORBER_CEMC");
    out->StripNode("G4HIT_CEMC_SPT");
    out->StripNode("G4CELL_CEMC");
    out->StripNode("G4HIT_BECAL");
    out->StripNode("G4CELL_BECAL");
    out->StripNode("G4HIT_ABSORBER_BECAL");
    out->StripNode("G4HIT_ABSORBER_HCALIN");
    out->StripNode("G4HIT_HCALIN");
    out->StripNode("G4HIT_HCALIN_SPT");
    out->StripNode("G4CELL_HCALIN");
    out->StripNode("G4HIT_MAGNET");
    out->StripNode("G4HIT_HCALOUT");
    out->StripNode("G4HIT_ABSORBER_HCALOUT");
    out->StripNode("G4CELL_HCALOUT");
    out->StripNode("G4HIT_BH_1");
    out->StripNode("G4HIT_BH_FORWARD_PLUS");
    out->StripNode("G4HIT_BH_FORWARD_NEG");

    out->StripNode("G4HIT_FEMC");
    out->StripNode("G4HIT_ABSORBER_FEMC");
    out->StripNode("G4HIT_FHCAL");
    out->StripNode("G4HIT_ABSORBER_FHCAL");
    out->StripNode("G4CELL_FEMC");
    out->StripNode("G4HIT_DRCALO");
    out->StripNode("G4HIT_ABSORBER_DRCALO");
    out->StripNode("G4CELL_DRCALO");
    out->StripNode("G4CELL_FHCAL");
    out->StripNode("G4HIT_LFHCAL");
    out->StripNode("G4HIT_ABSORBER_LFHCAL");
    out->StripNode("G4CELL_LFHCAL");
    out->StripNode("G4HIT_EEMC");
    out->StripNode("G4HIT_EEMC_glass");
    out->StripNode("G4HIT_ABSORBER_EEMC");
    out->StripNode("G4CELL_EEMC");
    out->StripNode("G4HIT_EHCAL");
    out->StripNode("G4HIT_ABSORBER_EHCAL");
    out->StripNode("G4CELL_EHCAL");
    out->StripNode("G4CELL_B0ECAL");
    out->StripNode("G4CELL_BWD");

  }
}
#endif