PHG4mRICHSubsystem.cc

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

/*===============================================================*
 *                        March 2nd 2017                         *
 *         mRICH Subsystem created by Cheuk-Ping Wong @GSU       *
 *===============================================================*/
#include "PHG4mRICHSubsystem.h"
#include "PHG4mRICHDetector.h"
#include "PHG4mRICHSteppingAction.h"

#include <phparameter/PHParameters.h>

#include <g4detectors/PHG4EventActionClearZeroEdep.h>
#include <g4main/PHG4EventAction.h>  // for PHG4EventAction
#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 <boost/foreach.hpp>

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

class PHG4Detector;

//_______________________________________________________________________
PHG4mRICHSubsystem::PHG4mRICHSubsystem(const std::string& name, const int lyr)
  : PHG4DetectorSubsystem(name, lyr)
  , _detector(nullptr)
  , _detectorName(name)
  , _steppingAction(nullptr)
  , _eventAction(nullptr)
{
  InitializeParameters();
}

//_______________________________________________________________________
int PHG4mRICHSubsystem::InitSubsystem(PHCompositeNode* topNode)
{
  // kludge until the phg4parameters are sorted out (adding layers)
  GetParams()->set_name(Name());
  GetParams()->set_int_param("active", 1);
  GetParams()->set_int_param("absorberactive", 1);
  GetParams()->set_int_param("blackhole", 0);
  GetParams()->set_string_param("superdetector", "NONE");
  GetParams()->set_string_param("detectorname", "mRICH");
  return 0;
}

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

  //---------------------------------
  // create detector
  //---------------------------------
  _detector = new PHG4mRICHDetector(this, topNode, GetParams(), Name(), GetLayer());
  _detector->SuperDetector(SuperDetector());
  _detector->OverlapCheck(CheckOverlap());
  _detector->SetActive(GetParams()->get_int_param("active"));
  _detector->SetAbsorberActive(GetParams()->get_int_param("absorberactive"));

  //---------------------------------
  // create hit node and stepping action
  //---------------------------------
  if (GetParams()->get_int_param("active"))
  {
    std::set<std::string> nodes;

    // create hit output node
    std::ostringstream nodename;
    nodename << "G4HIT_" << GetParams()->get_string_param("detectorname");

    PHG4HitContainer* mRICH_hits = findNode::getClass<PHG4HitContainer>(topNode, nodename.str().c_str());
    if (!mRICH_hits)
    {
      mRICH_hits = new PHG4HitContainer(nodename.str());
      PHIODataNode<PHObject>* hitNode = new PHIODataNode<PHObject>(mRICH_hits, nodename.str().c_str(), "PHObject");
      dstNode->addNode(hitNode);
      nodes.insert(nodename.str());
    }

    std::ostringstream absnodename;
    absnodename << "G4HIT_ABSORBER_" << GetParams()->get_string_param("detectorname");

    PHG4HitContainer* absorber_hits = findNode::getClass<PHG4HitContainer>(topNode, absnodename.str().c_str());
    if (!absorber_hits)
    {
      absorber_hits = new PHG4HitContainer(absnodename.str());
      PHIODataNode<PHObject>* abshitNode = new PHIODataNode<PHObject>(absorber_hits, absnodename.str().c_str(), "PHObject");
      dstNode->addNode(abshitNode);
      nodes.insert(nodename.str());
    }

    // create stepping action
    _steppingAction = new PHG4mRICHSteppingAction(_detector, GetParams());

    // event actions
    BOOST_FOREACH (std::string node, nodes)
    {
      if (!_eventAction)
        _eventAction = new PHG4EventActionClearZeroEdep(topNode, node);
      else
      {
        PHG4EventActionClearZeroEdep* evtact = dynamic_cast<PHG4EventActionClearZeroEdep*>(_eventAction);
        evtact->AddNode(node);
      }
    }
  }

  return 0;
}

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

//_______________________________________________________________________
PHG4Detector*
PHG4mRICHSubsystem::GetDetector(void) const
{
  return _detector;
}
//_______________________________________________________________________
void PHG4mRICHSubsystem::SetDefaultParameters()
{
  set_default_int_param("detectorSetup", 1);  //1 for full setup
                                              //0 for skeleton setup: detector box, aerogel, and sensor for quick check

  set_default_int_param("subsystemSetup", 0);  //-1: single module
                                               //0 : build hemispheric wall
                                               //>0: mRICH wall as sector. "subsystemSetup" becomes num. of sector (max is 8)

  set_default_double_param("r_inner", 3);    //inner radius of mRICH wall, default=3m;
  set_default_double_param("eta_min", 1.1);  //limits of rapidity
  set_default_double_param("eta_max", 1.9);

  set_default_int_param("use_g4steps", 0);  //for stepping function
}