/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.31/xAODTracking/xAODTracking/ParticleCaloExtension.h

#ifndef XAOD_PARTICLECALOEXTENSION_H 
#define XAOD_PARTICLECALOEXTENSION_H

#include "xAODTracking/TrackingPrimitives.h" 
#include "EventPrimitives/EventPrimitivesHelpers.h"

#ifndef XAOD_STANDALONE
#ifndef XAOD_MANACORE
// Athena includes
#include "TrkParameters/TrackParameters.h"
#endif // not XAOD_MANACORE
#endif // not XAOD_STANDALONE

#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;

#if ( ! defined(XAOD_STANDALONE) ) && ( ! defined(XAOD_MANACORE) )
    const Trk::CurvilinearParameters curvilinearParameters(unsigned int index) const;          
#endif // not XAOD_STANDALONE and not XAOD_MANACORE

    bool cellsAreAssociated() const;

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

    void setCaloCells(const std::vector<CaloCell*>& cells);

  private:
    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;
  }

#if ( ! defined(XAOD_STANDALONE) ) && ( ! defined(XAOD_MANACORE) )
  inline const Trk::CurvilinearParameters ParticleCaloExtension::curvilinearParameters(unsigned int index) const {    

    // copy the correct values into the temp matrix
    ParametersCovMatrix_t* cov = 0;
    if( !m_parametersCovariance[index].empty() ) {
      cov = new ParametersCovMatrix_t(); 
      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,cov);

    return param;
  }
#endif // not XAOD_STANDALONE and not XAOD_MANACORE

}

#include "xAODCore/CLASS_DEF.h"
#include "AthContainers/DataVector.h"
typedef DataVector<xAOD::ParticleCaloExtension> ParticleCaloExtensionCollection;
CLASS_DEF(ParticleCaloExtensionCollection, 1251279391 , 1 )

#endif

---