ParticleFlowJetInput.cc

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

#include "ParticleFlowJetInput.h"

#include "ParticleFlowElement.h"
#include "ParticleFlowElementContainer.h"

#include <jetbase/Jet.h>
#include <jetbase/Jetv2.h>

#include <phool/getClass.h>

// standard includes
#include <cassert>
#include <iostream>
#include <map>                               // for _Rb_tree_const_iterator
#include <utility>                           // for pair
#include <vector>

void ParticleFlowJetInput::identify(std::ostream &os)
{
  os << "   ParticleFlowJetInput: ";
  os << std::endl;
}

std::vector<Jet *> ParticleFlowJetInput::get_input(PHCompositeNode *topNode)
{
  if (_verbosity > 0) std::cout << "ParticleFlowJetInput::process_event -- entered" << std::endl;

  ParticleFlowElementContainer *pflowContainer = findNode::getClass<ParticleFlowElementContainer>(topNode, "ParticleFlowElements");
  if (!pflowContainer)
    {
      return std::vector<Jet *>();
    }

  std::vector<Jet *> pseudojets;
  ParticleFlowElementContainer::ConstRange begin_end = pflowContainer->getParticleFlowElements();
  ParticleFlowElementContainer::ConstIterator rtiter;


  for (rtiter = begin_end.first; rtiter != begin_end.second; ++rtiter)
  {
    ParticleFlowElement *pflow = rtiter->second;

    Jet *jet = new Jetv2();
    jet->set_px( pflow->get_px() );
    jet->set_py( pflow->get_py() );
    jet->set_pz( pflow->get_pz() );
    jet->set_e( pflow->get_e() );
    jet->insert_comp( Jet::SRC::PARTICLE , pflow->get_id() );
    pseudojets.push_back( jet );
    if (_verbosity > 15) for (auto c : pseudojets[pseudojets.size()-1]->get_comp_vec()) std::cout << " GOT " << ((int)c.first) << " and " << ((int)c.second) << std::endl;
  }

  if (_verbosity > 0) std::cout << "ParticleFlowJetInput::process_event -- exited" << std::endl;

  return pseudojets;
}