SVDWaveform.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.                  *
 **************************************************************************/
 
#ifndef SVDWaveform_H
#define SVDWaveform_H
 
#include <framework/dataobjects/RelationElement.h>
#include <string>
#include <sstream>
#include <deque>
#include <map>
#include <svd/simulation/SVDSimulationTools.h>
 
namespace Belle2 {
  namespace SVD {

 
    class SVDWaveform {
 
    public:

      struct ElementaryWaveform {
        ElementaryWaveform(double initTime, double charge, double tau, WaveformShape wfun,
                           RelationElement::index_type particle, RelationElement::index_type truehit):
          m_initTime(initTime), m_charge(charge), m_tau(tau), m_wfun(wfun),
          m_particle(particle), m_truehit(truehit)
        {}

        std::string toString() const
        {
          std::ostringstream os;
          os << m_initTime << '\t' << m_charge << '\t' << m_tau << std::endl;
          return os.str();
        }

        double m_initTime;
        double m_charge;
        double m_tau;
        WaveformShape m_wfun;
        RelationElement::index_type m_particle;
        RelationElement::index_type m_truehit;
      };

      typedef std::deque< ElementaryWaveform > elementary_waveform_list;

      typedef std::map<RelationElement::index_type, RelationElement::weight_type> relations_map;
      typedef std::map<RelationElement::index_type, RelationElement::weight_type>::value_type relation_value_type;

      SVDWaveform(): m_charge(0) {}

      SVDWaveform(const SVDWaveform& other)
      {
        m_charge = other.getCharge();
        for (const ElementaryWaveform& elementary_waveform : other.getElementaryWaveforms())
          m_elementaryWaveforms.push_back(elementary_waveform);
      }

      void add(double initTime, double charge, double tau, int particle = -1, int truehit = -1, WaveformShape wfun = w_betaprime)
      {
        if (charge > 0) {
          m_charge += charge;
          m_elementaryWaveforms.push_back(ElementaryWaveform(initTime, charge, tau, wfun, particle, truehit));
          if (particle > -1) m_particles[particle] += static_cast<float>(charge);
          if (truehit > -1) m_truehits[truehit] += static_cast<float>(charge);
        } else if (m_charge == 0)
          m_charge += charge;
      }

      SVDWaveform& operator=(const SVDWaveform& other)
      {
        m_charge = other.getCharge();
        m_elementaryWaveforms = other.getElementaryWaveforms();
        return *this;
      }

      static double waveform(double t, double initTime, double charge, double tau,
                             WaveformShape wfun = w_betaprime)
      {
        double z = (t - initTime) / tau;
        return charge * wfun(z);
      }

      static double waveform(double t, const ElementaryWaveform& elemWaveform)
      { return waveform(t, elemWaveform.m_initTime, elemWaveform.m_charge, elemWaveform.m_tau, elemWaveform.m_wfun); }

      double operator()(double t) const
      {
        double total_waveform = 0;
        for (const SVDWaveform::ElementaryWaveform& elementary_waveform : m_elementaryWaveforms) {
          total_waveform += waveform(t, elementary_waveform);
        }
        return total_waveform;
      }

      double getCharge() const { return m_charge; }

      const elementary_waveform_list& getElementaryWaveforms() const { return m_elementaryWaveforms; }

      const relations_map& getMCParticleRelations() const { return m_particles; }

      const relations_map& getTrueHitRelations() const { return m_truehits; }
      std::string toString() const
      {
        std::ostringstream os;
        size_t i = 0;
        for (const auto& elementary_waveform : m_elementaryWaveforms)
          os << ++i << '\t' << elementary_waveform.toString();
        return os.str();
      }
 
    protected:

      double m_charge;
      elementary_waveform_list m_elementaryWaveforms;
      relations_map m_particles;
      relations_map m_truehits;
 
    }; // class SVDWaveform
 
  } // end namespace SVD
} // end namespace Belle2
 
#endif