ClusterAlgoResult.h

Go to the documentation of this file.

 
#ifndef EVENT_CLUSTERALGORESULT_H_
#define EVENT_CLUSTERALGORESULT_H_
 
// ROOT
#include "TArrayD.h"
#include "TString.h"
 
// STL
#include <iostream>
 
namespace ldmx {
 
class ClusterAlgoResult {
 public:
  ClusterAlgoResult() = default;
 
  virtual ~ClusterAlgoResult();
 
  friend std::ostream& operator<<(std::ostream& o, const ClusterAlgoResult& d);
 
  void clear();
 
  const TString& getName() const { return name_; }
 
  double getAlgoVar(int element) const { return variables_[element]; }
 
  double getAlgoVar0() const {
    return (variables_.GetSize() < 1) ? (0) : (variables_[0]);
  }
 
  double getAlgoVar1() const {
    return (variables_.GetSize() < 2) ? (0) : (variables_[1]);
  }
 
  double getAlgoVar2() const {
    return (variables_.GetSize() < 3) ? (0) : (variables_[2]);
  }
 
  double getAlgoVar3() const {
    return (variables_.GetSize() < 4) ? (0) : (variables_[3]);
  }
 
  double getAlgoVar4() const {
    return (variables_.GetSize() < 5) ? (0) : (variables_[4]);
  }
 
  double getWeight(int nCluster) const {
    return (nCluster < weights_.GetSize()) ? (0) : (weights_[nCluster]);
  }
 
  void set(const TString& name, int nvar);
 
  void set(const TString& name, int nvar, int nweights);
 
  void setAlgoVar(int element, double value);
 
  void setWeight(int nClusters, double weight);
 
 private:
  TString name_{};
 
  /* Algorithm variable results from the cluster algo. */
  TArrayD variables_;
 
  /* Array of weights when a certain number of clusters is reached. */
  TArrayD weights_;
 
  ClassDef(ClusterAlgoResult, 2);
};
}  // namespace ldmx
 
#endif