DATCONSVDClusterCandidate.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 <vector>
#include <numeric>
 
namespace Belle2 {
  struct DATCONSVDClusterCandidate {
    VxdID vxdID = 0; 
    std::vector<unsigned short> strips; 
    std::vector<unsigned short> charges; 
    float maxSNRinClusterCandidate = 0; 
    int charge = 0; 
    int seedStripIndex = 0; 
    int seedStrip = 0; 
    int seedCharge = 0; 
    float clusterPosition = 0; 
    bool add(VxdID nextID, int nextCharge, unsigned short nextCellID, float nextStripSNR, unsigned short maxClusterSize)
    {
      bool added = false;
 
      // do not add if you are on the wrong sensor or side
      if (vxdID != nextID) return false;
 
      // add if it's the first strip
      if (strips.size() == 0) added = true;
 
      // add if it adjacent to the last strip added (we assume that SVDRecoDigits are ordered)
      // and if cluster size would still be <= 4 strips
      if (strips.size() > 0 and nextCellID == strips.back() + 1 and strips.size() < maxClusterSize) {
        added  = true;
      }
 
      //add it to the vector od strips, update the seed nextCharge and index:
      if (added) {
        strips.push_back(nextCellID);
        charges.push_back(nextCharge);
 
        if (nextCharge > seedCharge) seedCharge = nextCharge;
 
        if (nextStripSNR > maxSNRinClusterCandidate) {
          maxSNRinClusterCandidate = nextStripSNR;
        }
      }
      return added;
 
    };
 
    void finalizeCluster(const double pitch, const int stripsInSensor)
    {
      charge = std::accumulate(charges.begin(), charges.end(), 0);
      seedStripIndex  = strips.size() / 2 + 1;
      seedStrip       = (strips.at(0) + seedStripIndex - 1);
      seedCharge      = charges.at(seedStripIndex - 1);
      clusterPosition = pitch * (seedStrip - stripsInSensor / 2);
    };
  };
 
} // end namespace Belle2