SVDSignal.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Peter Kvasnicka, Martin Ritter                           *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#ifndef SVDSIGNAL_H
#define SVDSIGNAL_H
 
#include <framework/dataobjects/RelationElement.h>
#include <string>
#include <sstream>
#include <deque>
#include <map>
#include <svd/simulation/SVDSimulationTools.h>
 
namespace Belle2 {
  namespace SVD {
 
    class SVDSignal {
 
    public:
 
      struct Wave {
        Wave(double initTime, double charge, double tau,
             RelationElement::index_type particle, RelationElement::index_type truehit):
          m_initTime(initTime), m_charge(charge), m_tau(tau), 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;
        RelationElement::index_type m_particle;
        RelationElement::index_type m_truehit;
      };
 
      typedef std::deque< Wave > function_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;
 
      SVDSignal(): m_charge(0) {}
 
      SVDSignal(const SVDSignal& other)
      {
        m_charge = other.getCharge();
        for (Wave wave : other.getFunctions())
          m_functions.push_back(wave);
      }
 
      bool isNoise() const { return m_charge < 0; }
 
      void add(double initTime, double charge, double tau, int particle = -1, int truehit = -1)
      {
        if (charge > 0) {
          m_charge += charge;
          m_functions.push_back(Wave(initTime, charge, tau, 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;
      }
 
      SVDSignal& operator=(const SVDSignal& other)
      {
        m_charge = other.getCharge();
        m_functions.clear();
        std::copy(other.getFunctions().begin(), other.getFunctions().end(), m_functions.begin());
        return *this;
      }
 
      double waveform(double t, double initTime, double charge, double tau,
                      waveFunction wfun = w_betaprime) const
      {
        double z = (t - initTime) / tau;
        return charge * wfun(z);
      }
 
      double waveform(double t, const Wave& wave) const
      { return waveform(t, wave.m_initTime, wave.m_charge, wave.m_tau); }
 
      double operator()(double t) const
      {
        double wave_sum = 0;
        for (SVDSignal::Wave wave : m_functions) {
          wave_sum += waveform(t, wave);
        }
        return wave_sum;
      }
 
      double getCharge() const { return m_charge; }
 
      const function_list& getFunctions() const { return m_functions; }
 
      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 w : m_functions)
          os << ++i << '\t' << w.toString();
        return os.str();
      }
 
    protected:
 
      double m_charge;
      function_list m_functions;
      relations_map m_particles;
      relations_map m_truehits;
 
    }; // class SVDSignal
 
  } // end namespace SVD
} // end namespace Belle2
 
#endif