ParticleCaloExtension.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef XAOD_PARTICLECALOEXTENSION_H
#define XAOD_PARTICLECALOEXTENSION_H
 
#include "xAODTracking/TrackingPrimitives.h"
#include "EventPrimitives/EventPrimitivesHelpers.h"
 
#ifndef XAOD_ANALYSIS
// Athena includes
#include "TrkParameters/TrackParameters.h"
#endif // not XAOD_ANALYSIS
 
#include <vector>
class CaloCell;
 
namespace xAOD {

  class ParticleCaloExtension {
  public:
    ParticleCaloExtension( float charge,
                           std::vector<std::vector<float> >&& parameters,
                           std::vector<std::vector<float> >&& parametersCovariance,
                           std::vector<int>&& identifiers );

    ~ParticleCaloExtension();

    ParticleCaloExtension(const ParticleCaloExtension&) = delete;

    ParticleCaloExtension& operator=(const ParticleCaloExtension&) = delete;

    size_t numberOfParameters() const;

    const CurvilinearParameters_t trackParameters(unsigned int index) const;

    int parameterIdentifier(unsigned int index) const;

    bool trackParameterCovarianceMatrix(ParametersCovMatrix_t& matrix, unsigned int index) const;
 
#ifndef XAOD_ANALYSIS
    const Trk::CurvilinearParameters curvilinearParameters(unsigned int index) const;
#endif // not XAOD_ANALYSIS

    bool cellsAreAssociated() const;

    const std::vector<CaloCell*>& caloCells() const;

    void setCaloCells(const std::vector<CaloCell*>& cells);
 
  private:
#ifdef __clang__
    [[maybe_unused]]
#endif
    float m_charge;

    std::vector<std::vector<float> > m_parameters;

    std::vector<std::vector<float> > m_parametersCovariance;

    std::vector<int> m_identifiers;

    bool                   m_cellsAreSet; // bool to store whether already set
    std::vector<CaloCell*> m_caloCells;   // vector of cells
  };

  inline size_t ParticleCaloExtension::numberOfParameters() const {
    return m_parameters.size();
  }

  inline const CurvilinearParameters_t ParticleCaloExtension::trackParameters(unsigned int index) const {
    CurvilinearParameters_t tmp;
    tmp << m_parameters[index][0],m_parameters[index][1],m_parameters[index][2],
      m_parameters[index][3],m_parameters[index][4],m_parameters[index][5];
    return tmp;
  }
 
  inline bool ParticleCaloExtension::trackParameterCovarianceMatrix(ParametersCovMatrix_t& cov, unsigned int index) const {
    const std::vector<float>& covVec = m_parametersCovariance[index];
    if( !covVec.empty() ) Amg::expand( covVec.begin(), covVec.end(),cov );
    else{
      cov.setIdentity();
      return false;
    }
    return true;
  }

  inline int ParticleCaloExtension::parameterIdentifier(unsigned int index) const {
    return m_identifiers[index];
  }
 
  inline bool ParticleCaloExtension::cellsAreAssociated() const {
    return m_cellsAreSet;
  }
 
  inline const std::vector<CaloCell*>& ParticleCaloExtension::caloCells() const {
    return m_caloCells;
  }
 
  inline void ParticleCaloExtension::setCaloCells(const std::vector<CaloCell*>& cells) {
    m_caloCells = cells;
    m_cellsAreSet=true;
  }
 
#ifndef XAOD_ANALYSIS
  inline const Trk::CurvilinearParameters ParticleCaloExtension::curvilinearParameters(unsigned int index) const {
 
    // copy the correct values into the temp matrix
    ParametersCovMatrix_t cov;
    if( !m_parametersCovariance[index].empty() ) {
      trackParameterCovarianceMatrix(cov,index);
    }
    // retrieve the parameters to build the curvilinear frame
    Amg::Vector3D pos( m_parameters[index][0],m_parameters[index][1],m_parameters[index][2]);
    Amg::Vector3D mom(m_parameters[index][3],m_parameters[index][4],m_parameters[index][5]);
    Trk::CurvilinearParameters param(pos,mom,m_charge,std::move(cov));
 
    return param;
  }
#endif // not XAOD_ANALYSIS
 
}
 
#include "xAODCore/CLASS_DEF.h"
#include "AthContainers/DataVector.h"
typedef DataVector<xAOD::ParticleCaloExtension> ParticleCaloExtensionCollection;
CLASS_DEF(ParticleCaloExtensionCollection, 1251279391 , 1 )
 
#endif