CalibHitElementPos.cc

Go to the documentation of this file.

#include <iomanip>
#include <interface_main/SQHitVector.h>
#include <fun4all/Fun4AllReturnCodes.h>
#include <phool/PHNodeIterator.h>
#include <phool/PHIODataNode.h>
#include <phool/getClass.h>
#include <geom_svc/GeomSvc.h>
#include "CalibHitElementPos.h"
using namespace std;
 
CalibHitElementPos::CalibHitElementPos(const std::string& name)
  : SubsysReco(name)
  , m_cal_hit   (true)
  , m_cal_tr_hit(true)
  , m_vec_hit   (0)
  , m_vec_trhit (0)
{
  ;
}
 
CalibHitElementPos::~CalibHitElementPos()
{
  ;
}
 
int CalibHitElementPos::Init(PHCompositeNode* topNode)
{
  return Fun4AllReturnCodes::EVENT_OK;
}
 
int CalibHitElementPos::InitRun(PHCompositeNode* topNode)
{
  if (m_cal_hit) {
    m_vec_hit = findNode::getClass<SQHitVector>(topNode, "SQHitVector");
    if (!m_vec_hit) return Fun4AllReturnCodes::ABORTEVENT;
  }
  if (m_cal_tr_hit) {
    m_vec_trhit = findNode::getClass<SQHitVector>(topNode, "SQTriggerHitVector");
    if (!m_vec_trhit) return Fun4AllReturnCodes::ABORTEVENT;
  }
  return Fun4AllReturnCodes::EVENT_OK;
}
 
int CalibHitElementPos::process_event(PHCompositeNode* topNode)
{
  if (Verbosity() > 10) cout << "CalibHitElementPos::process_event()"<< endl;
  GeomSvc* geom = GeomSvc::instance();
  if (m_cal_hit) {
    for (SQHitVector::Iter it = m_vec_hit->begin(); it != m_vec_hit->end(); it++) {
      SQHit* hit = *it;
      if (Verbosity() > 10) cout << "  hit " << hit->get_detector_id() << " " << hit->get_element_id() << endl;
      if (hit->get_detector_id() <= 0 || hit->get_detector_id() >= 1000) continue; // temporary solution to avoid a seg fault on strange hits.
      hit->set_pos(geom->getMeasurement(hit->get_detector_id(), hit->get_element_id()));
    }
  }
  if (m_cal_tr_hit) {
    for (SQHitVector::Iter it = m_vec_trhit->begin(); it != m_vec_trhit->end(); it++) {
      SQHit* hit = *it;
      if (Verbosity() > 10) cout << "  trig hit " << hit->get_detector_id() << " " << hit->get_element_id() << endl;
      if (hit->get_detector_id() <= 0 || hit->get_detector_id() >= 1000) continue; // temporary solution to avoid a seg fault on strange hits.
      hit->set_pos(geom->getMeasurement(hit->get_detector_id(), hit->get_element_id()));
    }
  }
  return Fun4AllReturnCodes::EVENT_OK;
}
 
int CalibHitElementPos::End(PHCompositeNode* topNode)
{
  return Fun4AllReturnCodes::EVENT_OK;
}