NoiseMap.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 SVD_NOISEMAP_H
#define SVD_NOISEMAP_H
 
#include <vxd/dataobjects/VxdID.h>
#include <svd/geometry/SensorInfo.h>
#include <svd/reconstruction/Sample.h>
#include <vxd/geometry/GeoCache.h>
 
namespace Belle2 {
  namespace SVD {
 
    class NoiseMap {
    public:
      NoiseMap(): m_eInADU(1.0), m_sensorID(0), m_isU(false), m_noiseLevel(0) {}
 
      virtual ~NoiseMap() {}
 
      void setADU(double eInADU) { m_eInADU = eInADU; }
 
      void setNoiseLevel(float noise) { m_noiseLevel = noise; }
 
      virtual void setSensorID(VxdID sensorID, bool side)
      {
        const SensorInfo& info = dynamic_cast<const SensorInfo&>(VXD::GeoCache::getInstance().getSensorInfo(sensorID));
        m_sensorID = sensorID;
        if (side)
          m_noiseLevel = info.getElectronicNoiseU() / m_eInADU;
        else
          m_noiseLevel = info.getElectronicNoiseV() / m_eInADU;
        m_isU = side;
      }
 
      float getSignificance(const Sample& sample) const { return sample.getCharge() / getNoise(sample); }
      virtual float getNoise(short) const { return m_noiseLevel; }
 
      virtual float getNoise(const Sample& sample) const { return m_noiseLevel; }
 
      bool operator()(const Sample& sample, float significance) const
      { return getSignificance(sample) >= significance; }
 
      bool operator()(float signal, float significance) const { return signal / m_noiseLevel >= significance; }
 
    protected:
      double m_eInADU;
      VxdID m_sensorID;
      bool m_isU;
      float m_noiseLevel;
    };
 
  }
 
}
 
#endif //SVD_NOISEMAP_H