SimAna_module.cc

Go to the documentation of this file.

////////////////////////////////////////////////////////////////////////
// Class:       SimAna
// Module Type: analyzer
// File:        SimAna_module.cc
//
// Generated at Mon Jan  9 16:10:25 2017 by Andrey Sheshukov using artmod
// from cetpkgsupport v1_10_02.
////////////////////////////////////////////////////////////////////////
#include "TTree.h"
#include "TH1D.h"
#include "Simulation/FLSHit.h"
#include "Simulation/FLSHitList.h"
#include "RawData/RawDigit.h"
#include "RecoBase/CellHit.h"
#include "art/Framework/Core/EDAnalyzer.h"
#include "art/Framework/Core/ModuleMacros.h"
#include "art/Framework/Principal/Event.h"
#include "art/Framework/Principal/Handle.h"
#include "art/Framework/Principal/Run.h"
#include "art/Framework/Principal/SubRun.h"
#include "art_root_io/TFileService.h"
#include "art/Utilities/InputTag.h"
#include "fhiclcpp/ParameterSet.h"
#include "messagefacility/MessageLogger/MessageLogger.h"
#include "MCCheater/BackTracker.h"


namespace supernova {
  class SimAna;
}

class supernova::SimAna : public art::EDAnalyzer {
public:
  explicit SimAna(fhicl::ParameterSet const & p);
  // The destructor generated by the compiler is fine for classes
  // without bare pointers or other resource use.

  // Plugins should not be copied or assigned.
  SimAna(SimAna const &) = delete;
  SimAna(SimAna &&) = delete;
  SimAna & operator = (SimAna const &) = delete;
  SimAna & operator = (SimAna &&) = delete;

  // Required functions.
  void analyze(art::Event const & e) override;
  void beginJob() override;


private:

  // Declare member data here.
  TTree *fTree;
  std::vector<long> fPdg;
  std::vector<double> fP;
  std::vector<double> fX,fY,fZ,fT;

//  struct { long adc; int nx, ny; } fCluster;

};


supernova::SimAna::SimAna(fhicl::ParameterSet const & p)
  :
  EDAnalyzer(p)  // ,
 // More initializers here.
{

}

void supernova::SimAna::beginJob(){
    art::ServiceHandle<art::TFileService> tfs;
    fTree = tfs->make<TTree>("particles", "Particles tree");
    fTree->Branch("pdg",&fPdg);
    fTree->Branch("p",&fP);
    fTree->Branch("x",&fX);
    fTree->Branch("y",&fY);
    fTree->Branch("z",&fZ);
    fTree->Branch("t",&fT);
}

void supernova::SimAna::analyze(art::Event const & e)
{

  auto truth=e.getValidHandle<std::vector<simb::MCTruth> >("generator");
//  auto hits=e.getValidHandle<std::vector<novaddt::DAQHit> >("");
//  fCluster=CalcCluster(*hits);
    //
    for(const auto& mc: *truth){
        fPdg.clear();
        fP.clear();
        fX.clear();
        fY.clear();
        fZ.clear();
        fT.clear();
        for(int i=0; i<mc.NParticles();++i){
            auto p=mc.GetParticle(i);
            fPdg.push_back(p.PdgCode());
            fP.push_back(p.P(0));
            fX.push_back(p.Vx());
            fY.push_back(p.Vy());
            fZ.push_back(p.Vz());
            fT.push_back(p.T());
        }
        fTree->Fill();
    }

}

DEFINE_ART_MODULE(supernova::SimAna)