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 (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.clear();
        std::copy(other.getElementaryWaveforms().begin(), other.getElementaryWaveforms().end(), m_elementaryWaveforms.begin());
        return *this;
      }
 
      double waveform(double t, double initTime, double charge, double tau,
                      WaveformShape wfun = w_betaprime) const
      {
        double z = (t - initTime) / tau;
        return charge * wfun(z);
      }
 
      double waveform(double t, const ElementaryWaveform& elemWaveform) const
      { 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 (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 (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