G4EicDircSubsystem.cc

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

#include "G4EicDircSubsystem.h"

#include "G4EicDircDetector.h"
#include "G4EicDircDisplayAction.h"
#include "G4EicDircOpBoundaryProcess.h"
#include "G4EicDircStackingAction.h"
#include "G4EicDircSteppingAction.h"

#include <phparameter/PHParameters.h>

#include <g4main/PHG4HitContainer.h>
#include <g4main/PHG4StackingAction.h>  // for PHG4StackingAction
#include <g4main/PHG4SteppingAction.h>  // for PHG4SteppingAction

#include <phool/PHCompositeNode.h>
#include <phool/PHIODataNode.h>    // for PHIODataNode
#include <phool/PHNode.h>          // for PHNode
#include <phool/PHNodeIterator.h>  // for PHNodeIterator
#include <phool/PHObject.h>        // for PHObject
#include <phool/getClass.h>

#include <Geant4/G4ParticleTable.hh>
#include <Geant4/G4ProcessManager.hh>
#include <Geant4/G4SystemOfUnits.hh>
#include <Geant4/G4ios.hh>

#include <cmath>  // for isfinite

//_______________________________________________________________________
G4EicDircSubsystem::G4EicDircSubsystem(const std::string &name)
  : PHG4DetectorSubsystem(name)
  , m_Detector(nullptr)
  , m_SteppingAction(nullptr)
  , m_DisplayAction(nullptr)
{
  // call base class method which will set up parameter infrastructure
  // and call our SetDefaultParameters() method
  InitializeParameters();
}

//_______________________________________________________________________
G4EicDircSubsystem::~G4EicDircSubsystem()
{
  delete m_DisplayAction;
}

//_______________________________________________________________________
int G4EicDircSubsystem::InitRunSubsystem(PHCompositeNode *topNode)
{
  PHNodeIterator iter(topNode);
  PHCompositeNode *dstNode = dynamic_cast<PHCompositeNode *>(iter.findFirst("PHCompositeNode", "DST"));
  //DircBoundary = new G4EicDircOpBoundaryProcess();
  // G4EicDircDisplayAction *disp_action = new G4EicDircDisplayAction(Name(), GetParams());
  // if (isfinite(m_ColorArray[0]) &&
  //     isfinite(m_ColorArray[1]) &&
  //     isfinite(m_ColorArray[2]) &&
  //     isfinite(m_ColorArray[3]))
  // {
  //   disp_action->SetColor(m_ColorArray[0], m_ColorArray[1], m_ColorArray[2], m_ColorArray[3]);
  // }
  // m_DisplayAction = disp_action;
  m_DisplayAction = new G4EicDircDisplayAction(Name(), GetParams());
  // create detector
  m_Detector = new G4EicDircDetector(this, topNode, GetParams(), Name());
  m_Detector->SuperDetector(SuperDetector());
  m_Detector->OverlapCheck(CheckOverlap());
  m_Detector->Verbosity(Verbosity());

  std::string detector_suffix = SuperDetector();
  if (detector_suffix == "NONE")
  {
    detector_suffix = SuperDetector();
  }

  std::set<std::string> nodes;
  if (GetParams()->get_int_param("active"))
  {
    PHNodeIterator dstIter(dstNode);
    PHCompositeNode *DetNode = dstNode;
    if (SuperDetector() != "NONE")
    {
      DetNode = dynamic_cast<PHCompositeNode *>(dstIter.findFirst("PHCompositeNode", SuperDetector()));
      if (!DetNode)
      {
        DetNode = new PHCompositeNode(SuperDetector());
        dstNode->addNode(DetNode);
      }
    }
    m_HitNodeName = "G4HIT_" + detector_suffix;
    nodes.insert(m_HitNodeName);
    m_AbsorberNodeName = "G4HIT_ABSORBER_" + detector_suffix;
    if (GetParams()->get_int_param("absorberactive"))
    {
      nodes.insert(m_AbsorberNodeName);
    }
    m_SupportNodeName = "G4HIT_SUPPORT_" + detector_suffix;
    if (GetParams()->get_int_param("supportactive"))
    {
      nodes.insert(m_SupportNodeName);
    }
    for (auto thisnode : nodes)
    {
      PHG4HitContainer *g4_hits = findNode::getClass<PHG4HitContainer>(topNode, thisnode);
      if (!g4_hits)
      {
        g4_hits = new PHG4HitContainer(thisnode);
        DetNode->addNode(new PHIODataNode<PHObject>(g4_hits, thisnode, "PHObject"));
      }
    }
    G4EicDircSteppingAction *tmp = new G4EicDircSteppingAction(m_Detector, GetParams());
    tmp->SetHitNodeName(m_HitNodeName);
    tmp->SetAbsorberNodeName(m_AbsorberNodeName);
    tmp->SetSupportNodeName(m_SupportNodeName);
    m_SteppingAction = tmp;
  }
  m_StackingAction = new G4EicDircStackingAction(m_Detector);
  m_StackingAction->Verbosity(Verbosity());

  return 0;
}

/*void G4EicDircSubsystem::AddProcesses(G4ParticleDefinition *particle)
{
    G4ProcessManager* pmanager = particle->GetProcessManager();
    if (DircBoundary->IsApplicable(*particle))
    {
      pmanager->AddDiscreteProcess(DircBoundary);
      pmanager->SetProcessOrderingToFirst(DircBoundary, idxPostStep);
      
      G4cout << "dirc boundary process index = " << pmanager->GetProcessIndex(DircBoundary) << G4endl;
      G4cout << "dirc boundary process ordering = " << pmanager->GetProcessOrdering(DircBoundary, idxPostStep) << G4endl;
      }
}*/

//_______________________________________________________________________
int G4EicDircSubsystem::process_event(PHCompositeNode *topNode)
{
  // pass top node to stepping action so that it gets
  // relevant nodes needed internally
  if (m_SteppingAction)
  {
    m_SteppingAction->SetInterfacePointers(topNode);
  }
  return 0;
}

void G4EicDircSubsystem::Print(const std::string &what) const
{
  if (m_Detector)
  {
    m_Detector->Print(what);
  }
  return;
}

//_______________________________________________________________________
PHG4Detector *G4EicDircSubsystem::GetDetector(void) const
{
  return m_Detector;
}

void G4EicDircSubsystem::SetDefaultParameters()
{
  // sizes are in cm
  // angles are in deg
  // units will be converted to G4 units when used
  set_default_double_param("place_x", 0.);
  set_default_double_param("place_y", 0.);
  set_default_double_param("place_z", 0.);
  set_default_double_param("rot_x", 0.);
  set_default_double_param("rot_y", 0.);
  set_default_double_param("rot_z", 0.);
  set_default_double_param("size_x", 20.);
  set_default_double_param("size_y", 20.);
  set_default_double_param("size_z", 20.);

  set_default_double_param("rMin", 74.1);
  set_default_double_param("rMin_inner", 60.0);
  set_default_double_param("length", 287 + 168);

  set_default_double_param("NBars", 11);
  set_default_double_param("Radius", 75.0 * cm);
  set_default_double_param("Prizm_width", 38.65 * cm);
  set_default_double_param("Prizm_length", 30.0 * cm);
  set_default_double_param("Prizm_height_at_lens", 5.0 * cm);
  set_default_double_param("Bar_thickness", 1.725 * cm);
  set_default_double_param("Bar_width", 3.5 * cm);
  set_default_double_param("BarL_length", 122.5 * cm);
  set_default_double_param("BarS_length", 56.0 * cm);
  set_default_double_param("Mirror_height", 2.0 * cm);
  set_default_double_param("z_shift", -43.75 * cm);
  set_default_int_param("Geom_type", 0);  // 0-whole DIRC, 1-one bar box
  set_default_int_param("Lens_id", 3);    // 3- 3-layer spherical lens
  set_default_int_param("MCP_rows", 6);
  set_default_int_param("MCP_columns", 4);
  set_default_int_param("NBoxes", 12);
  set_default_int_param("Bar_pieces", 4);

  set_default_int_param("disable_photon_sim", 0);  // if true, disable photon simulations

  set_default_string_param("material", "G4_Galactic");
}