MultiTreeAccessor.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#include "LArSamplesMon/MultiTreeAccessor.h"
 
#include "LArSamplesMon/TreeAccessor.h"
#include "LArSamplesMon/Data.h"
#include "LArCafJobs/ShapeInfo.h"
#include "LArCafJobs/EventData.h"
#include "TSystem.h"
#include "TChain.h"
#include <iostream>
#include <iomanip>
#include <fstream>
#include <string>
 
using std::string; 
using std::cout;
using std::endl;
using namespace std;
 
using namespace LArSamples;
 
 
MultiTreeAccessor::MultiTreeAccessor(std::vector<std::unique_ptr<const TreeAccessor> >&& accessors)
  : m_accessors(std::move(accessors))
{
}
 
 
std::unique_ptr<MultiTreeAccessor> MultiTreeAccessor::open(const std::vector<TString>& files)
{
  std::vector<std::unique_ptr<const TreeAccessor> > accessors;
  for (const TString& fileName : files) {
    std::unique_ptr<const TreeAccessor> accessor = TreeAccessor::open(fileName);
    if (!accessor) {
      cout << "Skipping invalid file " << fileName << endl;
      continue;
    }
    accessors.push_back(std::move(accessor));
  }
  
  if (accessors.empty()) return nullptr;
  return std::make_unique<MultiTreeAccessor>(std::move(accessors));
}
 
 
std::unique_ptr<MultiTreeAccessor> MultiTreeAccessor::openList(const TString& fileList)
{
  std::ifstream f(fileList);
  if (!f) {
    cout << "file " << fileList << " not accessible" << endl;
    return nullptr;
  }
 
  std::string fileName;  
  unsigned int i = 0;
 
  std::vector<std::unique_ptr<const TreeAccessor> > accessors;
  
  while (f >> fileName) {
    std::unique_ptr<const TreeAccessor> accessor = TreeAccessor::open(fileName.c_str());
    if (!accessor) {
      cout << "Skipping invalid file " << fileName << endl;
      continue;
    }
    cout << std::setw(2) << ++i << " - " << fileName << endl;
    accessors.push_back(std::move(accessor));
  }
  
  if (accessors.empty()) return nullptr;
  return std::make_unique<MultiTreeAccessor>(std::move(accessors));
}
 
 
std::unique_ptr<MultiTreeAccessor> MultiTreeAccessor::openWild(const TString& wcName)
{
  // Piggyback on TChain wildcarding feature...
  TChain chain("");
  chain.Add(wcName);
  
  std::vector<std::unique_ptr<const TreeAccessor> > accessors;
  
  for (int i = 0; i < chain.GetListOfFiles()->GetEntries(); i++) {
    std::string fileName = chain.GetListOfFiles()->At(i)->GetTitle();
    std::unique_ptr<const TreeAccessor> accessor = TreeAccessor::open(fileName.c_str());
    if (!accessor) {
      cout << "Skipping invalid file " << fileName << endl;
      continue;
    }
    cout << std::setw(2) << i+1 << " - " << fileName <<  " , nEvents = " << accessor->nEvents() << ", nRuns = " << accessor->nRuns() << endl;
    accessors.push_back(std::move(accessor));
  }
  
  if (accessors.empty()) return nullptr;
  return std::make_unique<MultiTreeAccessor>(std::move(accessors));
}
 
 
MultiTreeAccessor::~MultiTreeAccessor()
{
}
 
 
const EventData* MultiTreeAccessor::eventData(unsigned int i) const
{
  unsigned int nEventsSoFar = 0;
  for (const std::unique_ptr<const TreeAccessor>& accessor : m_accessors) {
    unsigned int n = accessor->nEvents();
    if (i < nEventsSoFar + n) return accessor->eventData(i - nEventsSoFar);
    nEventsSoFar += n;
  }
  return nullptr;
}
 
 
const RunData* MultiTreeAccessor::runData(unsigned int i) const 
{ 
  unsigned int nRunsSoFar = 0;
  for (const std::unique_ptr<const TreeAccessor>& accessor : m_accessors) {
    unsigned int n = accessor->nRuns();
    if (i < nRunsSoFar + n) return accessor->runData(i - nRunsSoFar);
    nRunsSoFar += n;
  }
  return nullptr;
}
 
 
unsigned int MultiTreeAccessor::nEvents() const
{
  unsigned int n = 0;
  for (const std::unique_ptr<const TreeAccessor>& accessor : m_accessors)
     n += accessor->nEvents();
  return n;
}
 
 
unsigned int MultiTreeAccessor::nRuns() const
{
  unsigned int n = 0;
  for (const std::unique_ptr<const TreeAccessor>& accessor : m_accessors)
     n += accessor->nRuns();
  return n;
}
 
 
unsigned int MultiTreeAccessor::historySize(unsigned int i) const
{
  resetCache();
  unsigned int size = 0;
  for (const std::unique_ptr<const TreeAccessor>& accessor : m_accessors) {
    const HistoryContainer* cont = accessor->historyContainer(i);
    if (cont) size += cont->nDataContainers();
  }
  return size;
}
 
unsigned int MultiTreeAccessor::historySizeSC(unsigned int i) const
{
  resetCache();
  unsigned int size = 0;
  for (const std::unique_ptr<const TreeAccessor>& accessor : m_accessors) {
    const HistoryContainer* cont = accessor->historyContainerSC(i);
    if (cont) size += cont->nDataContainers();
  }
  return size;
}
 
 
std::unique_ptr<const History> MultiTreeAccessor::getCellHistory(unsigned int i) const
{ 
  std::unique_ptr<CellInfo> cellInfo;
  std::vector<std::unique_ptr<const Data> > allData;
  std::vector<std::unique_ptr<const EventData> > allEventData;
  for (const std::unique_ptr<const TreeAccessor>& accessor : m_accessors) {
    std::unique_ptr<const History> thisHistory = accessor->getCellHistory(i);
    if (!thisHistory) continue;
    if (!cellInfo) {
      cellInfo = std::make_unique<CellInfo>(*thisHistory->cellInfo());
    }
    const std::vector<std::unique_ptr<const EventData> >& thisEventData = thisHistory->eventData();
    std::map<const EventData*, const EventData*> eventMap;
    for (const std::unique_ptr<const EventData>& event : thisEventData) {
      if (eventMap.find(event.get()) != eventMap.end()) continue;
      auto newED = std::make_unique<EventData>(*event);
      eventMap[event.get()] = newED.get();
      allEventData.push_back(std::move(newED));
    }
   
    for (unsigned int j = 0; j < thisHistory->nData(); j++) {
      allData.push_back(std::make_unique<Data>(*thisHistory->data(j), eventMap[&thisHistory->data(j)->eventData()], nullptr, -1));
      if (!cellInfo->shape(thisHistory->data(j)->gain())) {
        const ShapeInfo* thisShape = thisHistory->cellInfo()->shape(thisHistory->data(j)->gain());
        cellInfo->setShape(thisHistory->data(j)->gain(), thisShape ? new ShapeInfo(*thisShape) : nullptr);
      }
    }
  }
  //data are copied from cellInfo into History member variable
  if (cellInfo)
    return std::make_unique<History>(std::move(allData), *cellInfo, std::move(allEventData), i);
  return nullptr;
}
      
std::unique_ptr<const History> MultiTreeAccessor::getSCHistory(unsigned int i) const
{ 
  std::unique_ptr<CellInfo> cellInfo;
  std::vector<std::unique_ptr<const Data> > allData;
  std::vector<std::unique_ptr<const EventData> > allEventData;
  for (const std::unique_ptr<const TreeAccessor>& accessor : m_accessors) {
    //cout << "---> Getting history for a treeAccessor..." << endl; 
    std::unique_ptr<const History> thisHistory (accessor->getSCHistory(i));
    //cout << "---> done Getting history for a treeAccessor..." << endl; 
    if (!thisHistory) continue;
    if (!cellInfo) {
      cellInfo = std::make_unique<CellInfo>(*thisHistory->cellInfo());
    //cout << "---> done new cell info" << endl; 
    }
    //cout << "---> Creating new event data N = " << thisHistory->eventData().size() << endl; 
    const std::vector<std::unique_ptr<const EventData> >& thisEventData = thisHistory->eventData();
    std::map<const EventData*, const EventData*> eventMap;
    for (const std::unique_ptr<const EventData>& event : thisEventData) {
      if (eventMap.find(event.get()) != eventMap.end()) continue;
      auto newED = std::make_unique<EventData>(*event);
      eventMap[event.get()] = newED.get();
      allEventData.push_back(std::move(newED));
    }
    //cout << "---> Creating new data N = " << thisHistory->nData() << endl; 
   
    for (unsigned int ii = 0; ii < thisHistory->nData(); ii++) {
     //cout << "------> Creating new data " << i << endl; 
      allData.push_back(std::make_unique<Data>(*thisHistory->data(ii), eventMap[&thisHistory->data(ii)->eventData()], nullptr, -1));
     //cout << "------> done Creating new data " << i << endl; 
      if (!cellInfo->shape(thisHistory->data(ii)->gain())) {
        const ShapeInfo* thisShape = thisHistory->cellInfo()->shape(thisHistory->data(ii)->gain());
        cellInfo->setShape(thisHistory->data(ii)->gain(), thisShape ? new ShapeInfo(*thisShape) : nullptr);
      }
     //cout << "------> done shape " << i << endl; 
    }
     //cout << "---> done Creating new data, deleting treeAcc history" << endl; 
  }
  //cout << "--->returning new history..." << endl;
  if (cellInfo)
    return std::make_unique<History>(std::move(allData), *cellInfo, std::move(allEventData), i);
  return nullptr;
}
      
        
std::unique_ptr<const CellInfo> MultiTreeAccessor::getCellInfo(unsigned int i) const
{
  resetCache();
  for (const std::unique_ptr<const TreeAccessor>& accessor : m_accessors) {
    const HistoryContainer* cont = accessor->historyContainer(i);
    if (cont && cont->cellInfo()) return std::make_unique<CellInfo>(*cont->cellInfo());
  }
  return nullptr;
}
 
 
bool MultiTreeAccessor::writeToFile(const TString& fileName) const
{
  std::vector<const Accessor*> accessors;
  for (unsigned int i = 0; i < m_accessors.size(); i++)
    accessors.push_back(m_accessors[i].get());
  cout << "Merging data..." << endl;
  std::unique_ptr<TreeAccessor> singleContainer = TreeAccessor::merge(accessors, fileName);
  return !!singleContainer;
}
 
 
std::vector<std::unique_ptr<MultiTreeAccessor> >
MultiTreeAccessor::filterComponents(const FilterList& filterList, const DataTweaker& tweaker) const
{
  std::vector< std::vector<std::unique_ptr<const TreeAccessor> > > filteredAccessors(filterList.size());
  
  std::vector<std::unique_ptr<MultiTreeAccessor> > result;
 
  for (unsigned int i = 0; i < nAccessors(); i++) {
    const TreeAccessor* treeAcc = dynamic_cast<const TreeAccessor*>(&accessor(i));
    cout << "Processing data " << i << " of " << nAccessors();
    if (treeAcc) cout << " (fileName = " << treeAcc->fileName() << ")";
    cout << endl;
    FilterList thisFilterList;
    //
    for (unsigned int f = 0; f < filterList.size(); f++) {
      std::string pathname = (string)filterList.fileName(f);
      if( pathname.find("eos/atlas/") < pathname.length() ){
    int nslashes = 0, slpos = 0;
    for( int k1 = 0; k1 < (int)pathname.length(); k1++ ){
      if( nslashes > 2 )break;
      if( pathname[k1] != '/' )continue;
      nslashes++;
      slpos = k1;
    }
    pathname.resize( slpos );
      }
      TString thisFN = Form("%s_filter%d.root", pathname.c_str(), i );//filterList.fileName(f).Data(), i);
      thisFilterList.add(filterList.filterParams(f), thisFN);
    }
    std::vector<std::unique_ptr<TreeAccessor> > filteredTreeAccs = TreeAccessor::filter(accessor(i), thisFilterList, tweaker);
    if (filteredTreeAccs.size() != filterList.size()) {
      cout << "Filtering failed, exiting" << endl;
      return result;
    }
    for (unsigned int f = 0; f < filteredTreeAccs.size(); f++) filteredAccessors[f].push_back(std::move(filteredTreeAccs[f]));
  }
 
  for (unsigned int f = 0; f < filteredAccessors.size(); f++)
    result.push_back(std::make_unique<MultiTreeAccessor>(std::move(filteredAccessors[f])));
  return result;
}