SimpleCut.hpp

/* Copyright (C) 2019-2021 GSI, Universität Tübingen
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Viktor Klochkov, Ilya Selyuzhenkov */
#ifndef ANALYSISTREE_SIMPLECUT_H
#define ANALYSISTREE_SIMPLECUT_H
 
#include <algorithm>
#include <functional>
#include <set>
#include <string>
#include <vector>
 
#include "Constants.hpp"
#include "Utils.hpp"
#include "Variable.hpp"
 
namespace AnalysisTree {
 
class SimpleCut {
 
 public:
  SimpleCut() = default;
  SimpleCut(const SimpleCut& cut) = default;
  SimpleCut(SimpleCut&& cut) = default;
  SimpleCut& operator=(SimpleCut&&) = default;
  SimpleCut& operator=(const SimpleCut& cut) = default;
  virtual ~SimpleCut() = default;
 
  SimpleCut(std::vector<std::string> vars, std::function<bool(std::vector<double>&)> lambda, std::string title = "") : title_(std::move(title)),
                                                                                                                       lambda_(std::move(lambda)) {
    std::transform(vars.begin(), vars.end(),
                   std::back_inserter(vars_),
                   [](const std::string& arg_name) { return Variable::FromString(arg_name); });
 
    FillBranchNames();
  }
 
  friend SimpleCut RangeCut(const std::string& variable_name, double lo, double hi, const std::string& title);
 
  friend SimpleCut EqualsCut(const std::string& variable_name, int value, const std::string& title);
 
  friend SimpleCut RangeCut(const Variable& var, double lo, double hi, const std::string& title);
 
  friend SimpleCut EqualsCut(const Variable& var, int value, const std::string& title);
 
  template<class T>
  bool Apply(const T& object) const {
    std::vector<double> variables;
    variables.reserve(vars_.size());
    for (const auto& var : vars_) {
      variables.emplace_back(var.GetValue(object));
    }
    return lambda_(variables);
  }
  bool Apply(const BranchChannel& object) const;
  bool Apply(std::vector<const BranchChannel*>& bch, std::vector<size_t>& id) const;
  [[deprecated]] bool Apply(const BranchChannel& a, size_t a_id, const BranchChannel& b, size_t b_id) const;
 
  void Print() const;
 
  std::vector<Variable>& Variables() { return vars_; }
  ANALYSISTREE_ATTR_NODISCARD const std::vector<Variable>& GetVariables() const { return vars_; }
  ANALYSISTREE_ATTR_NODISCARD const std::set<std::string>& GetBranches() const { return branch_names_; }
 
  friend bool operator==(const SimpleCut& that, const SimpleCut& other);
 
 protected:
  SimpleCut(const Variable& var, double min, double max, std::string title = "");
  SimpleCut(const Variable& var, int value, std::string title = "");
 
  void FillBranchNames();
 
  std::string title_;
  std::vector<Variable> vars_{};
  std::set<std::string> branch_names_{};
  std::function<bool(std::vector<double>&)> lambda_;
 
  ClassDef(SimpleCut, 1);
};
SimpleCut RangeCut(const std::string& variable_name, double lo, double hi, const std::string& title = "");
SimpleCut EqualsCut(const std::string& variable_name, int value, const std::string& title = "");
SimpleCut RangeCut(const Variable& var, double lo, double hi, const std::string& title = "");
SimpleCut EqualsCut(const Variable& var, int value, const std::string& title = "");
 
}// namespace AnalysisTree
#endif//ANALYSISTREE_SIMPLECUT_H