BeastMagnetSubsystem.cc

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

//____________________________________________________________________________..
//
// This is the interface to the framework. You only need to define the parameters
// you use for your detector in the SetDefaultParameters() method here
// The place to do this is marked by //implement your own here//
// The parameters have no units, they need to be converted in the
// BeastMagnetDetector::ConstructMe() method
// but the convention is as mentioned cm and deg
//____________________________________________________________________________..
//
#include "BeastMagnetSubsystem.h"

#include "BeastMagnetDetector.h"
#include "BeastMagnetDisplayAction.h"
#include "BeastMagnetSteppingAction.h"

#include <phparameter/PHParameters.h>

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

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

using namespace std;

//_______________________________________________________________________
BeastMagnetSubsystem::BeastMagnetSubsystem(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();
}

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

//_______________________________________________________________________
int BeastMagnetSubsystem::InitRunSubsystem(PHCompositeNode *topNode)
{
  PHNodeIterator iter(topNode);
  PHCompositeNode *dstNode = dynamic_cast<PHCompositeNode *>(iter.findFirst("PHCompositeNode", "DST"));
  // create display settings before detector
  m_DisplayAction = new BeastMagnetDisplayAction(Name());

  // create detector
  m_Detector = new BeastMagnetDetector(this, topNode, GetParams(), Name());
  m_Detector->OverlapCheck(CheckOverlap());
  m_Detector->SuperDetector(SuperDetector());

  PHNodeIterator dstIter(dstNode);
  if (GetParams()->get_int_param("active"))
  {
    string name;
    if (SuperDetector() != "NONE")
    {
      name = SuperDetector();
    }
    else
    {
      name = Name();
    }
    PHCompositeNode *DetNode = dynamic_cast<PHCompositeNode *>(dstIter.findFirst("PHCompositeNode", name));
    if (!DetNode)
    {
      DetNode = new PHCompositeNode(name);
      dstNode->addNode(DetNode);
    }
    string g4hitnodename = "G4HIT_" + name;
    PHG4HitContainer *g4_hits = findNode::getClass<PHG4HitContainer>(DetNode, g4hitnodename);
    if (!g4_hits)
    {
      g4_hits = new PHG4HitContainer(g4hitnodename);
      DetNode->addNode(new PHIODataNode<PHObject>(g4_hits, g4hitnodename, "PHObject"));
    }
    m_SteppingAction = new BeastMagnetSteppingAction(m_Detector, GetParams());
  }
  return 0;
}
//_______________________________________________________________________
int BeastMagnetSubsystem::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 BeastMagnetSubsystem::Print(const string &what) const
{
  if (m_Detector)
  {
    m_Detector->Print(what);
  }
  return;
}

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

//_______________________________________________________________________
void BeastMagnetSubsystem::SetDefaultParameters()
{
  // sizes are in cm
  // angles are in deg
  // units should be converted to G4 units when used
  //implement your own here//
  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_string_param("GDMPath", "DefaultParameters-InvalidPath");
  set_default_string_param("TopVolName", "DefaultParameters-InvalidVol");
}