ParticleFlowElementv1.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file ParticleFlowElementv1.h

#ifndef PARTICLEFLOW_PARTICLEFLOWELEMENTV1_H
#define PARTICLEFLOW_PARTICLEFLOWELEMENTV1_H

//===========================================================
//===========================================================

#include "TObject.h"

#include <iostream>
#include <utility>   // for pair, make_pair

#include "ParticleFlowElement.h"

class ParticleFlowElementv1 : public ParticleFlowElement
{
 public:
  ParticleFlowElementv1();
  ~ParticleFlowElementv1() override {}
  
  // PHObject virtual overloads
  
  void identify(std::ostream& os = std::cout) const override;
  void Reset() override;
  int isValid() const override;
  
  // pflow element info

  ParticleFlowElement::PFLOWTYPE get_type() const override {return _type; }
  void set_type( ParticleFlowElement::PFLOWTYPE type ) override { _type = type; }
  
  unsigned int get_id() const override { return _id; }
  void set_id(unsigned int id) override { _id = id; }
  
  float get_px() const override { return _mom[0]; }
  void set_px(float px) override { _mom[0] = px; }
  
  float get_py() const override { return _mom[1]; }
  void set_py(float py) override { _mom[1] = py; }
  
  float get_pz() const override { return _mom[2]; }
  void set_pz(float pz) override { _mom[2] = pz; }
  
  float get_e() const override { return _e; }
  void set_e(float e) override { _e = e; }
  
  float get_p() const override;
  float get_pt() const override;
  float get_et() const override;
  float get_eta() const override;
  float get_phi() const override;
  float get_mass() const override;
  
 private:
  unsigned int _id;

  // particle flow type 
  ParticleFlowElement::PFLOWTYPE _type;
  
  float _mom[3];
  
  float _e;
  
  ClassDefOverride(ParticleFlowElementv1, 1);
};

#endif