SVDStateVarSet.cc

/**************************************************************************
 * 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.                  *
 **************************************************************************/
 
#include <tracking/ckf/svd/filters/states/SVDStateVarSet.h>
#include <tracking/trackFindingCDC/eventdata/trajectories/CDCTrajectory3D.h>
#include <tracking/trackFindingCDC/eventdata/trajectories/CDCTrajectory2D.h>
#include <tracking/trackFindingCDC/eventdata/trajectories/CDCTrajectorySZ.h>
#include <tracking/dataobjects/RecoTrack.h>
 
using namespace std;
using namespace Belle2;
using namespace TrackFindingCDC;
 
namespace {
  template<class APredicate>
  double meanOver(const std::vector<TrackFindingCDC::WithWeight<const CKFToSVDState*>>& states, const APredicate& t)
  {
    double sum = 0;
    unsigned int numberOfHits = 0;
 
    for (const CKFToSVDState* walkState : states) {
      const SpacePoint* spacePoint = walkState->getHit();
      if (not spacePoint) {
        continue;
      }
      for (auto& cluster : spacePoint->getRelationsTo<SVDCluster>()) {
        numberOfHits++;
        sum += t(cluster);
      }
    }
 
    return sum / numberOfHits;
  }
 
  template<class APredicate>
  double minOver(const std::vector<TrackFindingCDC::WithWeight<const CKFToSVDState*>>& states, const APredicate& t)
  {
    double minimalValue = NAN;
 
    for (const CKFToSVDState* walkState : states) {
      const SpacePoint* spacePoint = walkState->getHit();
      if (not spacePoint) {
        continue;
      }
      for (auto& cluster : spacePoint->getRelationsTo<SVDCluster>()) {
        double currentValue = t(cluster);
        if (std::isnan(minimalValue)) {
          minimalValue = currentValue;
        } else {
          minimalValue = std::min(currentValue, minimalValue);
        }
      }
    }
 
    return minimalValue;
  }
 
  template<class APredicate>
  double stdOver(const std::vector<TrackFindingCDC::WithWeight<const CKFToSVDState*>>& states, const APredicate& t)
  {
    double sum = 0;
    double sumSquared = 0;
    unsigned int numberOfHits = 0;
 
    for (const CKFToSVDState* walkState : states) {
      const SpacePoint* spacePoint = walkState->getHit();
      if (not spacePoint) {
        continue;
      }
      for (auto& cluster : spacePoint->getRelationsTo<SVDCluster>()) {
        numberOfHits++;
        const auto& currentValue =  t(cluster);
        sum += currentValue;
        sumSquared += currentValue * currentValue;
      }
    }
 
    return std::sqrt((sumSquared - sum * sum / numberOfHits) / (numberOfHits - 1));
  }
}
 
bool SVDStateVarSet::extract(const BaseSVDStateFilter::Object* pair)
{
  const std::vector<TrackFindingCDC::WithWeight<const CKFToSVDState*>>& previousStates = pair->first;
  CKFToSVDState* state = pair->second;
 
  const RecoTrack* cdcTrack = previousStates.front()->getSeed();
  B2ASSERT("Path without seed?", cdcTrack);
 
  const SpacePoint* spacePoint = state->getHit();
  B2ASSERT("Path without hit?", spacePoint);
 
  std::vector<TrackFindingCDC::WithWeight<const CKFToSVDState*>> allStates = previousStates;
  allStates.emplace_back(state, 0);
 
  const std::vector<CDCHit*>& cdcHits = cdcTrack->getSortedCDCHitList();
 
  var<named("seed_cdc_hits")>() = cdcHits.size();
  var<named("seed_lowest_cdc_layer")>() = cdcHits.front()->getICLayer();
 
  genfit::MeasuredStateOnPlane firstMeasurement;
  if (state->mSoPSet()) {
    firstMeasurement = state->getMeasuredStateOnPlane();
  } else {
    B2ASSERT("Previous state was not fitted?", previousStates.back()->mSoPSet());
    firstMeasurement = previousStates.back()->getMeasuredStateOnPlane();
  }
 
  Vector3D position = Vector3D(firstMeasurement.getPos());
  Vector3D momentum = Vector3D(firstMeasurement.getMom());
 
  const CDCTrajectory3D trajectory(position, 0, momentum, cdcTrack->getChargeSeed());
 
  const Vector3D& hitPosition = static_cast<Vector3D>(spacePoint->getPosition());
 
  const double arcLength = trajectory.calcArcLength2D(hitPosition);
  const Vector2D& trackPositionAtHit2D = trajectory.getTrajectory2D().getPos2DAtArcLength2D(arcLength);
  const double trackPositionAtHitZ = trajectory.getTrajectorySZ().mapSToZ(arcLength);
 
  Vector3D trackPositionAtHit(trackPositionAtHit2D, trackPositionAtHitZ);
 
  const auto calculateCharge = [](const auto & s) {
    return s.getCharge();
  };
  const auto calculateSeedCharge = [](const auto & s) {
    return s.getSeedCharge();
  };
  const auto calculateSize = [](const auto & s) {
    return s.getSize();
  };
  const auto calculateSNR = [](const auto & s) {
    return s.getSNR();
  };
  const auto calculateChargeSizeRatio = [](const auto & s) {
    return s.getCharge() / s.getSize();
  };
 
  if (spacePoint->getType() == VXD::SensorInfoBase::SensorType::SVD) {
    const auto& clusters = spacePoint->getRelationsTo<SVDCluster>();
 
    B2ASSERT("Must be related to exactly 2 clusters", clusters.size() == 2);
    const SVDCluster* firstCluster = clusters[0];
    const SVDCluster* secondCluster = clusters[1];
 
    var<named("cluster_1_charge")>() = calculateCharge(*firstCluster);
    var<named("cluster_2_charge")>() = calculateCharge(*secondCluster);
    var<named("mean_rest_cluster_charge")>() = meanOver(allStates, calculateCharge);
    var<named("min_rest_cluster_charge")>() = minOver(allStates, calculateCharge);
    var<named("std_rest_cluster_charge")>() = stdOver(allStates, calculateCharge);
 
    var<named("cluster_1_seed_charge")>() = calculateSeedCharge(*firstCluster);
    var<named("cluster_2_seed_charge")>() = calculateSeedCharge(*secondCluster);
    var<named("mean_rest_cluster_seed_charge")>() = meanOver(allStates, calculateSeedCharge);
    var<named("min_rest_cluster_seed_charge")>() = minOver(allStates, calculateSeedCharge);
    var<named("std_rest_cluster_seed_charge")>() = stdOver(allStates, calculateSeedCharge);
 
    var<named("cluster_1_size")>() = calculateSize(*firstCluster);
    var<named("cluster_2_size")>() = calculateSize(*secondCluster);
    var<named("mean_rest_cluster_size")>() = meanOver(allStates, calculateSize);
    var<named("min_rest_cluster_size")>() = minOver(allStates, calculateSize);
    var<named("std_rest_cluster_size")>() = stdOver(allStates, calculateSize);
 
    var<named("cluster_1_snr")>() = calculateSNR(*firstCluster);
    var<named("cluster_2_snr")>() = calculateSNR(*secondCluster);
    var<named("mean_rest_cluster_snr")>() = meanOver(allStates, calculateSNR);
    var<named("min_rest_cluster_snr")>() = minOver(allStates, calculateSNR);
    var<named("std_rest_cluster_snr")>() = stdOver(allStates, calculateSNR);
 
    var<named("cluster_1_charge_over_size")>() = calculateChargeSizeRatio(*firstCluster);
    var<named("cluster_2_charge_over_size")>() = calculateChargeSizeRatio(*secondCluster);
    var<named("mean_rest_cluster_charge_over_size")>() = meanOver(allStates, calculateChargeSizeRatio);
    var<named("min_rest_cluster_charge_over_size")>() = minOver(allStates, calculateChargeSizeRatio);
    var<named("std_rest_cluster_charge_over_size")>() = stdOver(allStates, calculateChargeSizeRatio);
  }
 
  std::vector<const SpacePoint*> spacePoints;
  for (const CKFToSVDState* walkState : allStates) {
    const SpacePoint* sp = walkState->getHit();
    if (sp) {
      spacePoints.push_back(sp);
    }
  }
 
  if (spacePoints.size() >= 3) {
    var<named("quality_index_triplet")>() = m_qualityTriplet.estimateQuality(spacePoints);
    var<named("quality_index_circle")>() = m_qualityCircle.estimateQuality(spacePoints);
    var<named("quality_index_helix")>() = m_qualityHelix.estimateQuality(spacePoints);
  } else {
    var<named("quality_index_triplet")>() = 0;
    var<named("quality_index_circle")>() = 0;
    var<named("quality_index_helix")>() = 0;
  }
 
  return true;
}