PHG4BlockSubsystem.cc

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

#include "PHG4BlockSubsystem.h"

#include "PHG4BlockDetector.h"
#include "PHG4BlockDisplayAction.h"
#include "PHG4BlockGeomContainer.h"
#include "PHG4BlockGeomv1.h"
#include "PHG4BlockSteppingAction.h"

#include <phparameter/PHParameters.h>

#include <g4main/PHG4DisplayAction.h>  // for PHG4DisplayAction
#include <g4main/PHG4HitContainer.h>
#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 <cmath>  // for NAN
#include <sstream>

class PHG4BlockGeom;
class PHG4Detector;

using namespace std;

//_______________________________________________________________________
PHG4BlockSubsystem::PHG4BlockSubsystem(const std::string &name, const int lyr)
  : PHG4DetectorSubsystem(name, lyr)
  , m_Detector(nullptr)
  , m_SteppingAction(nullptr)
  , m_DisplayAction(nullptr)
{
  m_ColorArray.fill(NAN);
  InitializeParameters();
}

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

//_______________________________________________________________________
int PHG4BlockSubsystem::InitRunSubsystem(PHCompositeNode *topNode)
{
  PHNodeIterator iter(topNode);
  PHCompositeNode *dstNode = dynamic_cast<PHCompositeNode *>(iter.findFirst("PHCompositeNode", "DST"));

  // create display settings before detector (detector adds its volumes to it)
  PHG4BlockDisplayAction *disp_action = new PHG4BlockDisplayAction(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;
  // create detector
  m_Detector = new PHG4BlockDetector(this, topNode, GetParams(), Name(), GetLayer());
  m_Detector->SuperDetector(SuperDetector());
  m_Detector->OverlapCheck(CheckOverlap());
  if (GetParams()->get_int_param("active"))
  {
    ostringstream nodename;
    ostringstream geonode;
    if (SuperDetector() != "NONE")
    {
      nodename << "G4HIT_" << SuperDetector();
      geonode << "BLOCKGEOM_" << SuperDetector();
    }
    else
    {
      nodename << "G4HIT_" << Name();
      geonode << "BLOCKGEOM_" << Name();
    }

    // create hit list
    PHG4HitContainer *block_hits = findNode::getClass<PHG4HitContainer>(topNode, nodename.str());
    if (!block_hits)
    {
      dstNode->addNode(new PHIODataNode<PHObject>(block_hits = new PHG4HitContainer(nodename.str()), nodename.str(), "PHObject"));
    }

    block_hits->AddLayer(GetLayer());
    PHG4BlockGeomContainer *geocont = findNode::getClass<PHG4BlockGeomContainer>(topNode,
                                                                                 geonode.str());
    if (!geocont)
    {
      geocont = new PHG4BlockGeomContainer();
      PHCompositeNode *runNode = dynamic_cast<PHCompositeNode *>(iter.findFirst("PHCompositeNode", "RUN"));
      PHIODataNode<PHObject> *newNode = new PHIODataNode<PHObject>(geocont, geonode.str(), "PHObject");
      runNode->addNode(newNode);
    }

    PHG4BlockGeom *geom = new PHG4BlockGeomv1(GetLayer(),
                                              GetParams()->get_double_param("size_x"),
                                              GetParams()->get_double_param("size_y"),
                                              GetParams()->get_double_param("size_z"),
                                              GetParams()->get_double_param("place_x"),
                                              GetParams()->get_double_param("place_y"),
                                              GetParams()->get_double_param("place_z"),
                                              GetParams()->get_double_param("rot_z"));
    geocont->AddLayerGeom(GetLayer(), geom);

    m_SteppingAction = new PHG4BlockSteppingAction(m_Detector, GetParams());
  }
  else if (GetParams()->get_int_param("blackhole"))
  {
    m_SteppingAction = new PHG4BlockSteppingAction(m_Detector, GetParams());
  }

  return 0;
}

//_______________________________________________________________________
int PHG4BlockSubsystem::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;
}

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

void PHG4BlockSubsystem::SetDefaultParameters()
{
  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("steplimits", NAN);
  set_default_double_param("size_x", 10.);
  set_default_double_param("size_y", 10.);
  set_default_double_param("size_z", 10.);

  set_default_int_param("use_g4steps", 0);

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