StepInfo.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef LARG4GENSHOWERLIB_STEPINFO_H
#define LARG4GENSHOWERLIB_STEPINFO_H
 
// STL includes
#include <iostream>
 
// CLHEP include for Hep3Vector
#include "CLHEP/Vector/ThreeVector.h"
 
 
namespace ShowerLib
{
 
  class StepInfo
  {
 
  public:
 
    StepInfo(): m_dep(0.), m_time(0.), m_valid(true) {}
 
    StepInfo(const StepInfo& first, const StepInfo& second);
 
    /* access functions */
 
    inline void setP(const CLHEP::Hep3Vector& p) { m_pos = p; }
    inline void setX(const double x) { return m_pos.setX(x); }
    inline void setY(const double y) { return m_pos.setY(y); }
    inline void setZ(const double z) { return m_pos.setZ(z); }
    inline void setE(const double t) { m_dep = t; }
    inline void setTime(const double t) { m_time = t; }
    inline void setValid(const bool flag) { m_valid = flag; }
 
    inline CLHEP::Hep3Vector position() const { return m_pos; }
    inline double x() const { return m_pos.x(); }
    inline double y() const { return m_pos.y(); }
    inline double z() const { return m_pos.z(); }
    inline double dep() const { return m_dep; }
    inline double time() const { return m_time; }
    inline bool valid() const { return m_valid; }
 
    /* helper functions */
 
    double diff2(const StepInfo& other) const;
 
    StepInfo& operator+=(const StepInfo& other );
 
  private:
 
    // data members
    CLHEP::Hep3Vector m_pos;    
    double m_dep;        
    double m_time;       
    bool m_valid;        
 
  };
 
 
  // INLINE DEFINITIONS
  inline StepInfo::StepInfo(const StepInfo& first, const StepInfo& second)
  {
    double esum = first.m_dep + second.m_dep;
    double w1 = 0;
    double w2 = 0;
 
    if (esum > 0) {
      w1 =  first.m_dep/esum;
      w2 =  second.m_dep/esum;
    }
 
    m_pos = w1*first.m_pos + w2*second.m_pos;
    m_time = w1*first.m_time + w2*second.m_time;
    m_dep = esum;
    m_valid = true;
  }
 
  inline double StepInfo::diff2(const StepInfo& other) const
  {
    return (this->position().diff2(other.position()));
  }
 
  inline StepInfo& StepInfo::operator+=(const StepInfo& other )
  {
    double esum = m_dep + other.m_dep;
    double w1 = 0;
    double w2 = 0;
 
    if (esum > 0) {
      w1 =  m_dep/esum;
      w2 =  other.m_dep/esum;
    } else {
      std::cout << "Error: hit combination: sum of deposited energy is zero!" << std::endl;
    }
 
    m_pos = w1*m_pos + w2*other.m_pos;
    m_dep += other.m_dep;
 
    return *this;
  }
 
} // namespace ShowerLib
 
#endif // LARG4GENSHOWERLIB_STEPINFO_H