ECLShowerVariables.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.                  *
 **************************************************************************/
 
// analysis
#include <analysis/VariableManager/Manager.h>
 
// framework
#include <framework/core/Module.h>
#include <framework/datastore/StoreObjPtr.h>
#include <framework/datastore/StoreArray.h>
 
// dataobjects
#include <mdst/dataobjects/ECLCluster.h>
 
#include <analysis/dataobjects/Particle.h>
#include <ecl/dataobjects/ECLShower.h>
 
using namespace std;
 
namespace Belle2 {
  namespace Variable {
 
    double getShowerNumberOfCrystalsForEnergy(const Particle* particle)
    {
      const ECLCluster* cluster = particle->getECLCluster();
      if (cluster) {
        double showerNumberOfCrystalsForEnergy = -1.0;
 
        auto clusterShowerRelations = cluster->getRelationsWith<ECLShower>();
        for (unsigned int ir = 0; ir < clusterShowerRelations.size(); ++ir) {
          const auto shower = clusterShowerRelations.object(ir);
 
          showerNumberOfCrystalsForEnergy = shower->getNumberOfCrystalsForEnergy();
        }
 
        return showerNumberOfCrystalsForEnergy;
      }
 
      return std::numeric_limits<float>::quiet_NaN();
    }
 
    double getShowerNominalNumberOfCrystalsForEnergy(const Particle* particle)
    {
      const ECLCluster* cluster = particle->getECLCluster();
      if (cluster) {
        double showerNominalNumberOfCrystalsForEnergy = -1.0;
 
        auto clusterShowerRelations = cluster->getRelationsWith<ECLShower>();
        for (unsigned int ir = 0; ir < clusterShowerRelations.size(); ++ir) {
          const auto shower = clusterShowerRelations.object(ir);
 
          showerNominalNumberOfCrystalsForEnergy = shower->getNominalNumberOfCrystalsForEnergy();
        }
 
        return showerNominalNumberOfCrystalsForEnergy;
      }
 
      return std::numeric_limits<float>::quiet_NaN();
    }
 
 
    VARIABLE_GROUP("ECL Shower Debugging (cDST)");
 
    REGISTER_VARIABLE("eclShowerNumberOfCrystalsForEnergy", getShowerNumberOfCrystalsForEnergy,
                      "[debugging] Returns the number of crystals ued to calculate the shower energy (optimized to minimize the energy resolution). This should not be confused with the number of crystals contained in the cluster. ");
    REGISTER_VARIABLE("eclShowerNominalNumberOfCrystalsForEnergy", getShowerNominalNumberOfCrystalsForEnergy,
                      "[debugging] Returns the nominal number of crystals ued to calculate the shower energy (optimized to minimize the energy resolution). This should not be confused with the number of crystals contained in the cluster. ");
 
  }
}