CaTestModule.cc

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#include <calibration/example_caf_lib/modules/CaTest/CaTestModule.h>
 
#include <string>
 
#include <TTree.h>
#include <TH1F.h>
#include <TRandom.h>
 
#include <alignment/dataobjects/MilleData.h>
#include <framework/pcore/ProcHandler.h>
 
using namespace Belle2;
using namespace std;
 
//-----------------------------------------------------------------
//                 Register the Module
//-----------------------------------------------------------------
REG_MODULE(CaTest);
 
//-----------------------------------------------------------------
//                 Implementation
//-----------------------------------------------------------------
 
CaTestModule::CaTestModule() : CalibrationCollectorModule()
{
  // Set module properties
  setDescription("Test Module for saving big data in CAF");
  // Parameter definitions
  addParam("entriesPerEvent", m_entriesPerEvent,
           "Number of entries that we fill into the saved tree per event. As we increase this we start storing larger amonuts of data in a smaller number of events to test the limits.",
           int(10));
  addParam("spread", m_spread,
           "Spread of gaussian (mean=42) filling of test histogram (range=<0,100>) - probability of algo iterations depend on it", int(17.));
}
 
void CaTestModule::prepare()
{
  describeProcess("CaTest::prepare");
  std::string objectName = "MyTree";
  // Data object creation --------------------------------------------------
  TTree* tree = new TTree(objectName.c_str(), "");
  tree->Branch<int>("event", &m_evt);
  tree->Branch<int>("run", &m_run);
  tree->Branch<int>("exp", &m_exp);
  tree->Branch<double>("hitX", &m_hitX);
  tree->Branch<double>("hitY", &m_hitY);
  tree->Branch<double>("hitZ", &m_hitZ);
  tree->Branch<double>("trackX", &m_trackX);
  tree->Branch<double>("trackY", &m_trackY);
  tree->Branch<double>("trackZ", &m_trackZ);
  tree->Branch<double>("chisq", &m_chisq);
  tree->Branch<double>("pvalue", &m_pvalue);
  tree->Branch<double>("dof", &m_dof);
 
  // We register the objects so that our framework knows about them.
  // Don't try and hold onto the pointers or fill these objects directly
  // Use the getObjectPtr functions to access collector objects
  registerObject<TTree>(objectName, tree);
 
  auto hist = new TH1F("histo", "Test histogram, which mean value should be found by test calibration algo", 100, 0., 100.);
  registerObject<TH1F>("MyHisto", hist);
 
  auto mille = new MilleData();
  registerObject<MilleData>("test_mille", mille);
}
 
void CaTestModule::inDefineHisto()
{
  describeProcess("CaTest::inDefineHisto()");
}
 
void CaTestModule::startRun()
{
  describeProcess("CaTest::startRun()");
}
 
void CaTestModule::closeRun()
{
  describeProcess("CaTest::closeRun()");
  // We close the file at end of run, producing
  // one file per run (and process id) which is more
  // convenient than one large binary block.
  auto mille = getObjectPtr<MilleData>("test_mille");
  if (mille->isOpen()) {
    for (auto& fileName : mille->getFiles()) {
      B2DEBUG(100, "Stored Mille binary file: " << fileName);
    }
    mille->close();
  }
  //getObjectPtr<TT/ree>("MyTree")->GetDirectory()->ls();
}
 
void CaTestModule::collect()
{
  describeProcess("CaTest::collect()");
  m_evt = m_emdata->getEvent();
  m_run = m_emdata->getRun();
  m_exp = m_emdata->getExperiment();
 
  std::string objectName = "MyTree";
  auto tree = getObjectPtr<TTree>(objectName);
  auto hist = getObjectPtr<TH1F>("MyHisto");
 
  for (int i = 0; i < m_entriesPerEvent; ++i) {
    m_hitX = gRandom->Gaus();
    m_hitY = gRandom->Gaus();
    m_hitZ = gRandom->Gaus();
    m_trackX = gRandom->Gaus();
    m_trackY = gRandom->Gaus();
    m_trackZ = gRandom->Gaus();
    m_chisq = gRandom->Gaus();
    m_pvalue = gRandom->Gaus();
    m_dof = gRandom->Gaus();
    tree->Fill();
    hist->Fill(gRandom->Gaus(42., m_spread));
  }
 
  auto mille = getObjectPtr<MilleData>("test_mille");
  // Open new file on request (at start or after being closed)
  if (!mille->isOpen()) {
    string newFileName = to_string(m_exp) + "-" + to_string(m_run) + "-sevt-" + to_string(m_evt) + "-pid" + std::to_string(
                           ProcHandler::EvtProcID()) + ".mille";
    B2INFO("Opening new binary file " << newFileName);
    mille->open(newFileName);
  }
}
 
void CaTestModule::finish()
{
  describeProcess("CaTest::finish()");
}
 
void CaTestModule::describeProcess(const string& functionName)
{
  B2DEBUG(100, "Running " + functionName + " function from a Process of type " + ProcHandler::getProcessName()
          + "\nParallel Processing Used = " + to_string(ProcHandler::parallelProcessingUsed())
          + "\nThis EvtProcID Id = " + to_string(ProcHandler::EvtProcID())
          + "\nThe gDirectory is " + gDirectory->GetPath());
}