MVAPID_module.cc

Go to the documentation of this file.

//
// \file MVAPID_module.cc
//
// m.haigh@warwick.ac.uk
//

#ifndef MVAPID_H
#define MVAPID_H

// ### Generic C++ includes ###
#include <iostream>

// ### Framework includes ###
#include "art/Framework/Core/ModuleMacros.h"
#include "art/Framework/Core/EDProducer.h" 
#include "art/Framework/Principal/Event.h" 
#include "fhiclcpp/ParameterSet.h" 
#include "messagefacility/MessageLogger/MessageLogger.h" 
#include "art/Framework/Services/Optional/TFileService.h" 
#include "TTree.h"

#include "MVAAlg.h"
#include "lardataobj/AnalysisBase/MVAPIDResult.h"
#include "lardataobj/RecoBase/Track.h"
#include "lardata/Utilities/AssociationUtil.h"

namespace mvapid {

  class MVAPID : public art::EDProducer {

  public:

    explicit MVAPID(fhicl::ParameterSet const& pset);
    virtual ~MVAPID();                               
    void beginJob();        
    void beginRun(art::Run& run); 
    void reconfigure(fhicl::ParameterSet const& pset);
    void produce(art::Event& evt);                   
   
   
 private:

    MVAAlg fAlg;
    std::vector<anab::MVAPIDResult>* fResult;
    unsigned int fRun,fSubrun,fEvent;
    TTree* fTree;
    
  }; // class MVAPID


 
//------------------------------------------------------------------------------
  MVAPID::MVAPID(fhicl::ParameterSet const& pset): fAlg(pset,this)
{
  this->reconfigure(pset);
  produces< std::vector<anab::MVAPIDResult> >();
  produces< art::Assns<recob::Track, anab::MVAPIDResult, void> >();
  produces< art::Assns<recob::Shower, anab::MVAPIDResult, void> >();
  fResult=new std::vector<anab::MVAPIDResult>;
}

//------------------------------------------------------------------------------
void MVAPID::reconfigure(fhicl::ParameterSet const& pset) 
{
  return;
}

//------------------------------------------------------------------------------
MVAPID::~MVAPID()
{
}

// ***************** //
void MVAPID::beginJob()
{
  art::ServiceHandle<art::TFileService> tfs;
  fTree =tfs->make<TTree>("MVAPID","Results");
  fTree->Branch("run",&fRun,"run/I");
  fTree->Branch("subrun",&fSubrun,"subrun/I");
  fTree->Branch("event",&fEvent,"event/I");
  fTree->Branch("MVAResult",&fResult);
  fAlg.GetDetectorEdges();
  fAlg.GetWireNormals();
}


void MVAPID::beginRun(art::Run&)
{
  
}

// ***************** //
void MVAPID::produce(art::Event& evt)
{ 
  std::unique_ptr<std::vector<anab::MVAPIDResult> > result(new std::vector<anab::MVAPIDResult>);
  std::unique_ptr< art::Assns<recob::Track, anab::MVAPIDResult> > trackAssns(new art::Assns<recob::Track, anab::MVAPIDResult>);
  std::unique_ptr< art::Assns<recob::Shower, anab::MVAPIDResult> > showerAssns(new art::Assns<recob::Shower, anab::MVAPIDResult>);
  fRun = evt.id().run();
  fSubrun = evt.id().subRun();
  fEvent = evt.id().event();
  fAlg.RunPID(evt,*result,*trackAssns,*showerAssns);
  *fResult=*result;
  fTree->Fill();
  evt.put(std::move(result));
  evt.put(std::move(trackAssns));
  evt.put(std::move(showerAssns));
}
  
  DEFINE_ART_MODULE(MVAPID)
  
  
} //namespace mvapid

#endif // MVAPID_H