PHG4LFHcalSubsystem.cc

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

#include "PHG4LFHcalSubsystem.h"

#include "PHG4LFHcalDetector.h"
#include "PHG4LFHcalDisplayAction.h"
#include "PHG4LFHcalSteppingAction.h"

#include <phparameter/PHParameters.h>

#include <g4main/PHG4DisplayAction.h>  // for PHG4DisplayAction
#include <g4main/PHG4HitContainer.h>
#include <g4main/PHG4SteppingAction.h>  // for PHG4SteppingAction
#include <g4main/PHG4Subsystem.h>       // for PHG4Subsystem
#include <g4main/PHG4Utils.h>

#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 <TSystem.h>

#include <set>  // for set
#include <sstream>

class PHG4Detector;

//_______________________________________________________________________
PHG4LFHcalSubsystem::PHG4LFHcalSubsystem(const std::string& name, const int lyr)
  : PHG4DetectorSubsystem(name, lyr)
{
  InitializeParameters();
}

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

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

  // create display settings before detector
  m_DisplayAction = new PHG4LFHcalDisplayAction(Name(), showdetailed);

  // create detector
  m_Detector = new PHG4LFHcalDetector(this, topNode, GetParams(), Name());
  m_Detector->SuperDetector(SuperDetector());
  m_Detector->OverlapCheck(CheckOverlap());
  m_Detector->Verbosity(Verbosity());
  m_Detector->DoFullLightProp(_do_lightpropagation);

  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);
      }
    }
    // create hit output node
    std::string nodename;
    if (SuperDetector() != "NONE")
    {
      nodename = "G4HIT_" + SuperDetector();
    }
    else
    {
      nodename = "G4HIT_" + Name();
    }
    nodes.insert(nodename);
    if (GetParams()->get_int_param("absorberactive"))
    {
      if (SuperDetector() != "NONE")
      {
        nodename = "G4HIT_ABSORBER_" + SuperDetector();
      }
      else
      {
        nodename = "G4HIT_ABSORBER_" + Name();
      }
      nodes.insert(nodename);
    }
    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"));
      }
    }
    // create stepping action
    m_SteppingAction = new PHG4LFHcalSteppingAction(m_Detector, GetParams());
    // m_SteppingAction = new PHG4LFHcalSteppingAction(m_Detector, m_Detector->getParamsDet());
  }

  return 0;
}

//_______________________________________________________________________
int PHG4LFHcalSubsystem::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* PHG4LFHcalSubsystem::GetDetector() const
{
  return m_Detector;
}

void PHG4LFHcalSubsystem::SetDefaultParameters()
{
  set_default_double_param("place_x", 0.);
  set_default_double_param("place_y", 0.);
  set_default_double_param("place_z", 400.);
  set_default_double_param("tower_dx", 5.);
  set_default_double_param("tower_dy", 5.);
  set_default_double_param("tower_dz", 100.);
  set_default_double_param("dz", 100.);
  set_default_double_param("rMin1", 5.);
  set_default_double_param("rMax1", 262.);
  set_default_double_param("rMin2", 5.);
  set_default_double_param("rMax2", 336.9);
  set_default_double_param("wls_dw", 0.125);
  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("xoffset", 0.);
  set_default_double_param("yoffset", 0.);
  set_default_double_param("thickness_absorber", 1.6);
  set_default_double_param("thickness_scintillator", 0.4);
  set_default_int_param("nlayerspertowerseg", 10);
  set_default_int_param("usetailcatcher", 0);
  set_default_double_param("frame_width", 0.);
  set_default_double_param("width_coating", 0.);
  set_default_double_param("zdepthcatcheroffset", 1000000.);
  set_default_int_param("nLayerOffsetTailcatcher", 0);
  set_default_int_param("embed_fiber", 0);

  std::ostringstream mappingfilename;
  const char* calibroot = getenv("CALIBRATIONROOT");
  if (calibroot)
  {
    mappingfilename << calibroot;
  }
  else
  {
    std::cout << "no CALIBRATIONROOT environment variable" << std::endl;
    gSystem->Exit(1);
  }

  mappingfilename << "/LFHcal/mapping/towerMap_LFHCAL_2x.txt";
  set_default_string_param("mapping_file", mappingfilename.str());
  set_default_string_param("mapping_file_md5", PHG4Utils::md5sum(mappingfilename.str()));
  set_default_string_param("scintillator", "G4_POLYSTYRENE");
  set_default_string_param("absorber", "G4_Fe");
  set_default_string_param("absorber_W", "G4_W");

  return;
}

void PHG4LFHcalSubsystem::SetTowerMappingFile(const std::string& filename)
{
  set_string_param("mapping_file", filename);
  set_string_param("mapping_file_md5", PHG4Utils::md5sum(get_string_param("mapping_file")));
}