VectorSparse.h

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
 
#ifndef ALICEO2_FWDALIGN_VECTORSPARSE_H
#define ALICEO2_FWDALIGN_VECTORSPARSE_H
 
#include <TObject.h>
#include <TMath.h>
 
namespace o2
{
namespace fwdalign
{
 
class VectorSparse : public TObject
{
 public:
  VectorSparse();
 
  VectorSparse(const VectorSparse& src);
 
  ~VectorSparse() override { Clear(); }
 
  void Print(Option_t* option = "") const override;
 
  Int_t GetNElems() const { return fNElems; }
  UShort_t* GetIndices() const { return fIndex; }
  Double_t* GetElems() const { return fElems; }
  UShort_t& GetIndex(Int_t i) { return fIndex[i]; }
  Double_t& GetElem(Int_t i) const { return fElems[i]; }
 
  void Clear(Option_t* option = "") override;
 
  void Reset() { memset(fElems, 0, fNElems * sizeof(Double_t)); }
 
  void ReSize(Int_t sz, Bool_t copy = kFALSE);
 
  void SortIndices(Bool_t valuesToo = kFALSE);
 
  void Add(Double_t* valc, Int_t* indc, Int_t n);
 
  VectorSparse& operator=(const VectorSparse& src);
 
  virtual Double_t operator()(Int_t ind) const;
  virtual Double_t& operator()(Int_t ind);
 
  virtual void SetToZero(Int_t ind);
 
  Double_t FindIndex(Int_t ind) const;
 
  Double_t& FindIndexAdd(Int_t ind);
 
  Int_t GetLastIndex() const { return fIndex[fNElems - 1]; }
  Double_t GetLastElem() const { return fElems[fNElems - 1]; }
  Double_t& GetLastElem() { return fElems[fNElems - 1]; }
 
 protected:
  Int_t fNElems;    
  UShort_t* fIndex; 
  Double_t* fElems; 
 
  ClassDefOverride(VectorSparse, 0);
};
 
//___________________________________________________
inline Double_t VectorSparse::operator()(Int_t ind) const
{
  return FindIndex(ind);
}
 
//___________________________________________________
inline Double_t& VectorSparse::operator()(Int_t ind)
{
  return FindIndexAdd(ind);
}
 
} // namespace fwdalign
} // namespace o2
 
#endif