KFPEmcCluster.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file KFPEmcCluster.h

/*
 * This file is part of KFParticle package
 * Copyright (C) 2007-2019 FIAS Frankfurt Institute for Advanced Studies
 *               2007-2019 Goethe University of Frankfurt
 *               2007-2019 Ivan Kisel <I.Kisel@compeng.uni-frankfurt.de>
 *               2007-2019 Maksym Zyzak
 *
 * KFParticle is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * KFParticle is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

#ifndef KFPEmcCluster_H
#define KFPEmcCluster_H

#include "KFParticleDef.h"

class KFPEmcCluster
{
 public:
  KFPEmcCluster():fP(), fC(), fId() { }
  ~KFPEmcCluster() { }

  int Size() const { return fP[0].size(); }
  
  void Resize(const int n);
  void Set(KFPEmcCluster& v, int vSize, int offset);
  void SetTracks(const KFPEmcCluster& track, const kfvector_uint& trackIndex, const int nIndexes);
  
  const kfvector_float& X()  const { return fP[0]; } 
  const kfvector_float& Y()  const { return fP[1]; } 
  const kfvector_float& Z()  const { return fP[2]; } 
  const kfvector_float& E() const { return fP[3]; }  

  const kfvector_float& Parameter(const int i)  const { return fP[i]; }  
  const kfvector_float& Covariance(const int i)  const { return fC[i]; } 
  const kfvector_int& Id()    const { return fId; }                      

  //modifiers 
  void SetParameter (float value, int iP, int iTr) { fP[iP][iTr] = value; } 
  void SetCovariance(float value, int iC, int iTr) { fC[iC][iTr] = value; } 

  void SetParameter (const float_v& value, int iP, int iTr);  
  void SetCovariance(const float_v& value, int iC, int iTr);

  void SetId        (int value, int iTr)           { fId[iTr] = value; } 
  
  void PrintTrack(int n);
  void PrintTracks();
  
  KFPEmcCluster(const KFPEmcCluster& clusters): fId()
  {
    const int localSize = clusters.Size();
    
    for(int i=0; i<4; i++)
    {
      fP[i].resize(localSize);
      for(int n=0; n<localSize; n++)
        fP[i][n] = clusters.fP[i][n];
    }

    for(int i=0; i<10; i++)
    {
      fC[i].resize(localSize);
      for(int n=0; n<localSize; n++)
        fC[i][n] = clusters.fC[i][n];
    }
    
    fId.resize(localSize);
    for(int n=0; n<localSize; n++)
      fId[n] = clusters.fId[n];    
  }
  
  const KFPEmcCluster& operator = (const KFPEmcCluster& clusters)
  {
    const int localSize = clusters.Size();
    
    for(int i=0; i<4; i++)
    {
      fP[i].resize(localSize);
      for(int n=0; n<localSize; n++)
        fP[i][n] = clusters.fP[i][n];
    }

    for(int i=0; i<10; i++)
    {
      fC[i].resize(localSize);
      for(int n=0; n<localSize; n++)
        fC[i][n] = clusters.fC[i][n];
    }
    
    fId.resize(localSize);
    for(int n=0; n<localSize; n++)
      fId[n] = clusters.fId[n];
    
    return *this;
  }
  
 private:  
  kfvector_float fP[4];  
  kfvector_float fC[10]; 

  kfvector_int fId; 
};

#endif