RawCluster.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
 
#include <vxd/dataobjects/VxdID.h>
#include <svd/dataobjects/SVDShaperDigit.h>
#include <vector>
 
namespace Belle2::SVD {
  struct StripInRawCluster {
    int shaperDigitIndex; 
    int cellID; 
    int maxSample; 
    float noise; 
    Belle2::SVDShaperDigit::APVFloatSamples samples; 
    double charge; 
    double time; 
  };
 
  class RawCluster {
 
  public:
 
    RawCluster() {};
 
    RawCluster(VxdID vxdID, bool isUside, double cutSeed, double cutAdjacent);
 
    RawCluster(VxdID vxdID, bool isUside, double cutSeed, double cutAdjacent, const std::string& storeShaperDigitsName);
 
    bool add(VxdID vxdID, bool isUside, struct  StripInRawCluster& aStrip);
 
    bool isGoodRawCluster();
 
    VxdID getSensorID() const {return m_vxdID;}
 
    bool isUSide() const {return m_isUside;}
 
    Belle2::SVDShaperDigit::APVFloatSamples getClsSamples(bool inElectrons) const;
 
    std::pair<int, std::vector<float>> getMaxSum3Samples(bool inElectrons = false) const;
 
    int getSize() const { return m_strips.size(); }
 
    const std::vector<StripInRawCluster> getStripsInRawCluster() const { return m_strips; };
 
    int getSeedMaxSample() const {return m_seedMaxSample;};
 
    int getSeedInternalIndex() const {return m_seedInternalIndex;};
 
    void setStripCharge(int index, double charge) {m_strips.at(index).charge = charge;}
    void setStripTime(int index, double time) {m_strips.at(index).time = time;}
 
  protected:
 
    VxdID m_vxdID;
 
    bool m_isUside;
 
    double m_cutSeed = 5;
 
    double m_cutAdjacent = 3;
 
    int m_seedMaxSample = -1;
 
    float m_seedSNR = -1;
 
    int m_seedIndex = -1;
 
    int m_seedInternalIndex = -1;
 
    std::vector<StripInRawCluster> m_strips;
 
    std::string m_storeShaperDigitsName;
 
  };
 
}