PHG4ConsistencyCheck.cc

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

#include "PHG4ConsistencyCheck.h"

#include "PHG4HitContainer.h"
#include "PHG4TruthInfoContainer.h"
#include "PHG4Hit.h"
#include "PHG4Particle.h"

#include <phool/getClass.h>

#include <iostream>                  // for operator<<, basic_ostream::opera...
#include <map>                       // for _Rb_tree_const_iterator, map<>::...
#include <set>                       // for set
#include <utility>                   // for pair

class PHCompositeNode;

using namespace std;

PHG4ConsistencyCheck::PHG4ConsistencyCheck(const std::string &name):
  SubsysReco(name),
  errorcnt(0)
{}

int
PHG4ConsistencyCheck::InitRun(PHCompositeNode */*topNode*/)
{

  return 0;
}

int
PHG4ConsistencyCheck::process_event(PHCompositeNode *topNode)
{
  PHG4TruthInfoContainer *truthcont = findNode::getClass<PHG4TruthInfoContainer>(topNode, "G4TruthInfo");
  if (!truthcont)
    {
      return 0;
    }
  PHG4TruthInfoContainer::ConstRange trange = truthcont->GetParticleRange();
  PHG4TruthInfoContainer::ConstIterator titer;
  int imax = 1000000;
  for (titer = trange.first; titer != trange.second; ++titer)
    {
      if (titer->first < imax)
        {
          imax = titer->first;
        }
    }
  cout << "min index: " << imax << endl;
  std::pair< std::map<int,int>::const_iterator, std::map<int,int>::const_iterator > embtrk_b_e = truthcont->GetEmbeddedTrkIds();
  std::map<int,int>::const_iterator embiter;
  for (embiter = embtrk_b_e.first; embiter != embtrk_b_e.second; ++embiter)
    {
      cout << "embedded trkid: " << embiter->first << endl;
    }
  PHG4HitContainer *ghit = findNode::getClass<PHG4HitContainer>(topNode, "G4HIT_CEMC_E");
  if (ghit)
    {
      PHG4HitContainer::ConstIterator hit;
      PHG4HitContainer::ConstRange hit_begin_end = ghit->getHits();
      set<int> printpart;
      for (hit = hit_begin_end.first; hit != hit_begin_end.second; ++hit)
        {

          int trkid = hit->second->get_trkid();
          PHG4Particle* part = truthcont->GetParticle(trkid);
          if (!part)
            {
              hit->second->identify();
              cout << "could not locate geant particle " << trkid << " in G4HIT_CEMC_E" << endl;
              errorcnt++;
            }
          else
            {
              int primary_id = part->get_primary_id();
              if (truthcont->isEmbeded(primary_id)>0)
                {
                  if (printpart.find(primary_id) == printpart.end())
                    {
                      cout << "primary id " << primary_id << " is embedded" << endl;
                      printpart.insert(primary_id);
                      PHG4Particle* parta = truthcont->GetParticle(primary_id);
                      parta->identify();
                    }
                }
            }
        }
    }
  ghit = findNode::getClass<PHG4HitContainer>(topNode, "G4HIT_SVTX");
  if (ghit)
    {
      PHG4HitContainer::ConstIterator hit;
      PHG4HitContainer::ConstRange hit_begin_end = ghit->getHits();
      for (hit = hit_begin_end.first; hit != hit_begin_end.second; ++hit)
        {
          int trkid = hit->second->get_trkid();
          PHG4Particle* part = truthcont->GetParticle(trkid);
          if (!part)
            {
              cout << "could not locate geant particle " << trkid << " in G4HIT_SVTX" << endl;
              errorcnt++;
            }
        }
    }

  return 0;
}