ClusterProjection.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
namespace Belle2 {
  namespace PXD {
 
    class ClusterProjection {
    public:
      ClusterProjection(): m_pos(0), m_min(-1), m_max(0), m_minCharge(0),
        m_maxCharge(0), m_minPos(0), m_maxPos(0), m_error(0), m_charge(0),
        m_size(0) {}
 
      void add(unsigned int cell, float position, float charge);
      void finalize()
      {
        m_pos /= m_charge;
        m_size = m_max - m_min + 1;
      }
 
      double getPos() const { return m_pos; }
      double getError() const { return m_error; }
      unsigned int getSize() const { return m_size; }
      unsigned int getMinCell() const { return m_min; }
      unsigned int getMaxCell() const { return m_max; }
      double getCharge() const { return m_charge; }
      double getMinCharge() const { return m_minCharge; }
      double getMaxCharge() const { return m_maxCharge; }
      double getCenterCharge() const { return m_charge - m_minCharge - m_maxCharge; }
      double getMinPos() const { return m_minPos; }
      double getMaxPos() const { return m_maxPos; }
 
      void setPos(double pos) { m_pos = pos; }
      void setError(double error) { m_error = error; }
 
    private:
 
      double m_pos;
      unsigned int m_min;
      unsigned int m_max;
      double m_minCharge;
      double m_maxCharge;
      double m_minPos;
      double m_maxPos;
      double m_error;
      double m_charge;
      unsigned int m_size;
    };
 
    inline void ClusterProjection::add(unsigned int cell, float position, float charge)
    {
      //If the new cell is smaller than the last minimum cell than change that
      if (cell < m_min) {
        m_min = cell;
        m_minCharge = charge;
        m_minPos = position;
      } else if (cell == m_min) {
        //But if it is equal to the last known minimum cell sum up the charge
        m_minCharge += charge;
      }
      //Same as above for maximum cell.
      if (cell > m_max || m_max == 0) {
        m_max = cell;
        m_maxCharge = charge;
        m_maxPos = position;
      } else if (cell == m_max) {
        m_maxCharge += charge;
      }
      //Add weighted positions for center of gravity
      m_pos += charge * position;
      m_charge += charge;
    }
 
  } // PXD namespace
 
} // Belle2 namespace