HitCountNtuple.cc

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

#include "HitCountNtuple.h"

#include <g4main/PHG4HitContainer.h>
#include <g4main/PHG4Hit.h>

#include <fun4all/Fun4AllHistoManager.h>

#include <phool/getClass.h>

#include <TFile.h>
#include <TH1.h>
#include <TH2.h>
#include <TNtuple.h>

#include <boost/foreach.hpp>

#include<sstream>

using namespace std;

HitCountNtuple::HitCountNtuple(const std::string &name, const std::string &filename):
  SubsysReco( name ),
  nblocks(0),
  hm(NULL),
  _filename(filename),
  ntup(NULL),
  outfile(NULL)
{}

HitCountNtuple::~HitCountNtuple()
{
  //  delete ntup;
  delete hm;
} 


int
HitCountNtuple::Init( PHCompositeNode* )
{
  ostringstream hname, htit;
  hm = new  Fun4AllHistoManager(Name());
  outfile = new TFile(_filename.c_str(), "RECREATE");
  //ntup = new TNtuple("hitntup", "G4Hits", "detid:layer:x0:y0:z0:x1:y1:z1:edep");
  ntupe = new TNtuple("hitcnt", "NHits", "cemc:cemc_abs:ihcal:ihcal_abs:ohcal:ohcal_abs");
 //  ntup->SetDirectory(0);
  return 0;
}

int
HitCountNtuple::process_event( PHCompositeNode* topNode )
{
  set<string>::const_iterator iter;
  vector<TH1 *>::const_iterator eiter;
  float ntvar[6]={0};
  PHG4HitContainer *hits = nullptr;
  string nodename[6] = {"G4HIT_CEMC", "G4HIT_ABSORBER_CEMC",
                        "G4HIT_HCALIN", "G4HIT_ABSORBER_HCALIN",
                        "G4HIT_HCALOUT", "G4HIT_ABSORBER_HCALOUT"};

  for (int i=0; i<6; i++)
  {
  hits = findNode::getClass<PHG4HitContainer>(topNode,nodename[i]);

  if (hits)
    {
      //          double numhits = hits->size();
      //          nhits[i]->Fill(numhits);
          ntvar[i] = hits->size();
    }
  }
  ntupe->Fill(ntvar);
  return 0;
}

int
HitCountNtuple::End(PHCompositeNode * topNode)
{
  outfile->cd();
  //  ntup->Write();
  ntupe->Write();
  outfile->Write();
  outfile->Close();
  delete outfile;
  hm->dumpHistos(_filename, "UPDATE");
  return 0;
}

void
HitCountNtuple::AddNode(const std::string &name, const int detid)
{
 _node_postfix.insert(name);
 _detid[name] = detid;
 return;
}