CentralityFiller.cpp

#include "CentralityFiller.hpp"

#include <memory>
#include "TFile.h"

namespace AnalysisTree{

CentralityFiller::CentralityFiller(const std::string& file, const std::string& getter) {
  auto centr_file = std::unique_ptr<TFile>(TFile::Open(file.c_str(), "read"));

  if((!centr_file) || (centr_file->IsZombie())) {
    throw std::runtime_error("No file or file is zombie");
  }
  getter_ = centr_file->Get<Centrality::Getter>(getter.c_str());
  if(getter_ == nullptr) {
    throw std::runtime_error("Centrality Getter is nullptr");
  }
}

void CentralityFiller::Init() {
  auto man = TaskManager::GetInstance();
  auto chain = man->GetChain();
  input_ = chain->GetBranch(input_name_);

  is_event_header_ = input_.GetBranchType() == DetType::kEventHeader;

  in_field_ = input_.GetField(input_field_name_);
  in_branches_.emplace(input_name_);

  auto conf = input_.GetConfig().Clone(output_name_, AnalysisTree::DetType::kEventHeader);
  conf.AddField<float>(output_field_name_, "centrality, %");
  output_ = Branch(conf);
  out_field_ = output_.GetField(output_field_name_);

  output_.SetMutable();
  output_.Freeze();

  input_.Freeze();

  man->AddBranch(&output_);
}

void CentralityFiller::Exec() {
  if(is_event_header_) {
    FillFromEventHeader();
  } else {
    FillFromChannels();
  }
}

void CentralityFiller::FillFromEventHeader() {

  const auto value = input_[0][in_field_];
  output_[0].CopyContent(input_[0]);

  const auto centrality = getter_->GetCentrality(value);
  output_[0].SetValue(out_field_, centrality);
}

void CentralityFiller::FillFromChannels() {
  const auto n = input_.size();
  int m{0};
  for(size_t i = 0; i < n; ++i) {
    if(!event_cuts_ || event_cuts_->Apply(input_[i])) {
      m++;
    }
  }
  const auto centrality = getter_->GetCentrality(m);
  output_[0].SetValue(out_field_, centrality);
}

}