FlatMaker_module.cc

Go to the documentation of this file.

#include "art/Framework/Principal/Event.h"
#include "art/Framework/Principal/Handle.h"
#include "art/Framework/Principal/SubRun.h"
#include "art/Framework/Core/EDProducer.h"
#include "art/Framework/Core/ModuleMacros.h"
#include "art/Framework/Core/FileBlock.h"
#include "cetlib_except/exception.h"
#include "fhiclcpp/ParameterSet.h"
#include "messagefacility/MessageLogger/MessageLogger.h"

#include "FlatMaker/FlatRecord.h"

#include "StandardRecord/StandardRecord.h"
#include "StandardRecord/SRSpillTruthBranch.h"

#include "Metadata/MetadataManager.h"

#include "TFile.h"
#include "TH1.h"
#include "TTree.h"

namespace flat
{
  //.......................................................................
  class FlatMaker: public art::EDProducer
  {
  public:
    explicit FlatMaker(const fhicl::ParameterSet& pset);
    virtual ~FlatMaker() noexcept;

    void respondToOpenInputFile(const art::FileBlock& fb);
    void InitializeOutfile();

    void beginJob();
    void produce(art::Event& evt) noexcept;
    void endJob();

  protected:
    TFile* fFile;

    TTree* fTree;
    TTree* fSpillTree;
    TTree* fNuTree;

    TDirectory* fRecDir;
    TDirectory* fSpillDir;
    TDirectory* fNuDir;

    flat::Flat<caf::StandardRecord>* fRec;
    flat::Flat<caf::SRSpill>* fSpillRec;
    flat::Flat<caf::SRNeutrino>* fNuRec;

    double fTotalPOT;
    double fTotalLivetime;
    long fTotalEvents;

    std::string fStandardRecordLabel;
    std::string fOutputName;
    bool fFillNuTree;

    std::string fDataTier;
    std::string fFileExtension;
  };

  //.......................................................................
  FlatMaker::FlatMaker(const fhicl::ParameterSet& pset)
    : art::EDProducer(),
      fFile(0),
      fNuTree(0), fNuRec(0),
      fTotalPOT(0), fTotalLivetime(0), fTotalEvents(0),
      fStandardRecordLabel(pset.get<std::string>("StandardRecordLabel")),
      fOutputName(pset.get<std::string>("OutputName")),
      fFillNuTree(pset.get<bool>("FillNuTree")),
      fDataTier(pset.get<std::string>("DataTier")),
      fFileExtension(pset.get<std::string>("FileExtension"))
  {
  }

  //......................................................................
  FlatMaker::~FlatMaker() noexcept
  {
  }

  //......................................................................
  void FlatMaker::respondToOpenInputFile(const art::FileBlock& fb)
  {
    if(!fFile) {      
      if(fOutputName == "") {

        // Filename wasn't set in the FCL. Construct it based on the input
        // filename.
        char temp[fb.fileName().size() + 1];
        std::strcpy(temp, fb.fileName().c_str());
        fOutputName = basename(temp);

        // if input is PID file (most likely should be), be a little careful
        // Allows for dots in the file name than just the last bit (eg. prod5.1)
        if(fOutputName.find(".pid.root") != std::string::npos) {
          const size_t pidpos = fOutputName.find(".pid.root");
          fOutputName.resize(pidpos);
        }
        else{
          const size_t dotpos = fOutputName.find('.');
          assert(dotpos != std::string::npos); // Must have a dot, surely?
          fOutputName.resize(dotpos);
        }
      }

      fOutputName += fFileExtension;
      
      InitializeOutfile();

    }
  }


  //......................................................................
  void FlatMaker::InitializeOutfile()
  {
    assert(!fFile);
    assert(!fOutputName.empty());
    
    mf::LogInfo("FlatMaker") << "Output filename is " << fOutputName;

    consumes<std::vector<caf::StandardRecord>>(fStandardRecordLabel);
    consumes<caf::SRSpill>(fStandardRecordLabel);
    if(fFillNuTree) consumes<caf::SRSpillTruthBranch>(fStandardRecordLabel);

    fFile = new TFile(fOutputName.c_str(), "RECREATE");

    fTree = new TTree("recTree", "recTree");
    // Have trouble with memory usage (because several trees are open at
    // once?). Set the maximum buffer usage (per tree) to 3MB (10x less than
    // default)
    fTree->SetAutoFlush(-3*1000*1000);
    fRec = new flat::Flat<caf::StandardRecord>(fTree, "rec", "", 0);

    fSpillTree = new TTree("spillTree", "spillTree");
    fSpillTree->SetAutoFlush(-3*1000*1000);
    fSpillRec = new flat::Flat<caf::SRSpill>(fSpillTree, "spill", "", 0);

    if(fFillNuTree){
      fNuTree = new TTree("nuTree", "nuTree");
      fNuTree->SetAutoFlush(-3*1000*1000);
      fNuRec = new flat::Flat<caf::SRNeutrino>(fNuTree, "nu", "", 0);
    }
  }

  //......................................................................
  void FlatMaker::beginJob()
  {
  }

  //......................................................................
  void FlatMaker::produce(art::Event& evt) noexcept
  {
    ++fTotalEvents;

    art::Handle<std::vector<caf::StandardRecord>> srcol;
    evt.getByLabel(fStandardRecordLabel, srcol);

    for(const caf::StandardRecord& sr: *srcol){
      fRec->Clear();
      fRec->Fill(sr);
      fTree->Fill();
    }


    art::Handle<caf::SRSpill> spill;
    evt.getByLabel(fStandardRecordLabel, spill);

    fSpillRec->Clear();
    fSpillRec->Fill(*spill);
    fSpillTree->Fill();

    // Should we have checked isgoodspill?
    fTotalPOT += spill->spillpot;
    fTotalLivetime += spill->livetime;


    if(fFillNuTree){
      art::Handle<std::vector<caf::SRSpillTruthBranch>> srtruthcol;
      evt.getByLabel(fStandardRecordLabel, srtruthcol);

      for(const caf::SRSpillTruthBranch& sr: *srtruthcol){
        fNuRec->Clear();
        fNuRec->Fill(sr.nu);
        fNuTree->Fill();
      }
    }
  }

  //......................................................................
  void FlatMaker::endJob()
  {
    fFile->cd();

    fTree->Write();
    delete fRec;

    fSpillTree->Write();
    delete fSpillRec;

    if(fFillNuTree){
      fNuTree->Write();
      delete fNuRec;
    }

    fFile->cd();

    std::map<std::string, std::string> metadata = meta::MetadataManager::getInstance().GetMetadata();
    metadata["data_tier"] = fDataTier;
    fFile->mkdir("metadata")->cd();

    for(const auto& m: metadata){
      TObjString(m.second.c_str()).Write(m.first.c_str());
    }

    fFile->cd();

    TH1* hPOT = new TH1D("TotalPOT", "TotalPOT;; POT", 1, 0, 1);
    hPOT->Fill(.5, fTotalPOT);
    hPOT->Write();

    TH1* hEvents = new TH1D("TotalEvents", "TotalEvents;; Events", 1, 0, 1);
    hEvents->Fill(.5, fTotalEvents);
    hEvents->Write();

    TH1* hLivetime = new TH1D("TotalLivetime", "TotalLiveTime;;Seconds", 1, 0, 1);
    hLivetime->Fill(.5, fTotalLivetime);
    hLivetime->Write();

    fFile->Close();
  }

  DEFINE_ART_MODULE(FlatMaker)

} // end namespace flat
////////////////////////////////////////////////////////////////////////