GenericUnpackPRDF.cc

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

#include "GenericUnpackPRDF.h"

#include "PROTOTYPE3_FEM.h"
#include "RawTower_Prototype3.h"

#include <calobase/RawTower.h>  // for RawTower
#include <calobase/RawTowerContainer.h>
#include <calobase/RawTowerDefs.h>  // for NONE

#include <fun4all/Fun4AllBase.h>  // for Fun4AllBase::VERBOSITY_SOME
#include <fun4all/Fun4AllReturnCodes.h>
#include <fun4all/SubsysReco.h>  // for SubsysReco

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

#include <Event/Event.h>
#include <Event/packet.h>  // for Packet

#include <cassert>
#include <cmath>    // for NAN
#include <cstdlib>  // for exit
#include <iostream>
#include <string>

using namespace std;

//____________________________________
GenericUnpackPRDF::GenericUnpackPRDF(const string &detector)
  : SubsysReco("GenericUnpackPRDF_" + detector)
  ,  //
  _detector(detector)
  , _towers(nullptr)
{
}

//_____________________________________
int GenericUnpackPRDF::InitRun(PHCompositeNode *topNode)
{
  CreateNodeTree(topNode);
  return Fun4AllReturnCodes::EVENT_OK;
}

//____________________________________
int GenericUnpackPRDF::process_event(PHCompositeNode *topNode)
{
  Event *_event = findNode::getClass<Event>(topNode, "PRDF");
  if (!_event)
  {
    if (Verbosity() >= VERBOSITY_SOME)
      cout << "GenericUnpackPRDF::Process_Event - Event not found" << endl;
    return Fun4AllReturnCodes::DISCARDEVENT;
  }

  if (Verbosity() >= VERBOSITY_SOME)
    _event->identify();

  map<int, Packet *> packet_list;

  for (hbd_channel_map::const_iterator it = _hbd_channel_map.begin();
       it != _hbd_channel_map.end(); ++it)
  {
    const int packet_id = it->first.first;
    const int channel = it->first.second;
    const int tower_id = it->second;

    if (packet_list.find(packet_id) == packet_list.end())
    {
      packet_list[packet_id] = _event->getPacket(packet_id);
    }

    Packet *packet = packet_list[packet_id];

    if (!packet)
    {
      //          if (Verbosity() >= VERBOSITY_SOME)
      cout << "GenericUnpackPRDF::process_event - failed to locate packet "
           << packet_id << endl;
      continue;
    }
    assert(packet);

//    packet->setNumSamples(PROTOTYPE3_FEM::NSAMPLES);
    packet->setInternalParameter(PROTOTYPE3_FEM::NSAMPLES);

    RawTower_Prototype3 *tower =
        dynamic_cast<RawTower_Prototype3 *>(_towers->getTower(tower_id));
    if (!tower)
    {
      tower = new RawTower_Prototype3();
      tower->set_energy(NAN);
      _towers->AddTower(tower_id, tower);
    }
    tower->set_HBD_channel_number(channel);
    for (int isamp = 0; isamp < PROTOTYPE3_FEM::NSAMPLES; isamp++)
    {
      tower->set_signal_samples(isamp, packet->iValue(channel, isamp));
    }
  }

  for (map<int, Packet *>::iterator it = packet_list.begin();
       it != packet_list.end(); ++it)
  {
    if (it->second)
      delete it->second;
  }

  return Fun4AllReturnCodes::EVENT_OK;
}

//_______________________________________
void GenericUnpackPRDF::CreateNodeTree(PHCompositeNode *topNode)
{
  PHNodeIterator nodeItr(topNode);
  // DST node
  PHCompositeNode *dst_node = static_cast<PHCompositeNode *>(
      nodeItr.findFirst("PHCompositeNode", "DST"));
  if (!dst_node)
  {
    cout << "PHComposite node created: DST" << endl;
    dst_node = new PHCompositeNode("DST");
    topNode->addNode(dst_node);
  }

  // DATA nodes
  PHCompositeNode *data_node = static_cast<PHCompositeNode *>(
      nodeItr.findFirst("PHCompositeNode", "RAW_DATA"));
  if (!data_node)
  {
    if (Verbosity())
      cout << "PHComposite node created: RAW_DATA" << endl;
    data_node = new PHCompositeNode("RAW_DATA");
    dst_node->addNode(data_node);
  }

  // output as towers
  _towers = new RawTowerContainer(RawTowerDefs::NONE);
  PHIODataNode<PHObject> *towerNode =
      new PHIODataNode<PHObject>(_towers, "TOWER_RAW_" + _detector, "PHObject");
  data_node->addNode(towerNode);
}

void GenericUnpackPRDF::add_channel(const int packet_id,  
                                    const int channel,    
                                    const int tower_id    
)
{
  hbd_channel_typ hbd_channel(packet_id, channel);

  if (_hbd_channel_map.find(hbd_channel) != _hbd_channel_map.end())
  {
    cout << "GenericUnpackPRDF::add_channel - packet " << packet_id
         << ", channel " << channel << " is already registered as tower "
         << _hbd_channel_map.find(hbd_channel)->second << endl;
    exit(12);
  }
  _hbd_channel_map[hbd_channel] = tower_id;
}