MillePedeRecord.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_MILLEPEDERECORD_H
#define ALICEO2_FWDALIGN_MILLEPEDERECORD_H
 
#include <TObject.h>
 
namespace o2
{
namespace fwdalign
{
class MillePedeRecord : public TObject
{
 public:
  MillePedeRecord();
 
  MillePedeRecord(const MillePedeRecord& src);
 
  MillePedeRecord& operator=(const MillePedeRecord& rhs);
 
  ~MillePedeRecord() override;
 
  void Reset();
 
  void Print(const Option_t* opt = "") const override;
 
  Int_t GetSize() const { return fSize; }
  Int_t* GetIndex() const { return fIndex; }
  Int_t GetIndex(int i) const { return fIndex[i]; }
 
  void GetIndexValue(Int_t i, Int_t& ind, Double_t& val) const
  {
    ind = fIndex[i];
    val = fValue[i];
  }
 
  void AddIndexValue(Int_t ind, Double_t val);
 
  void AddResidual(Double_t val) { AddIndexValue(-1, val); }
  void AddWeight(Double_t val) { AddIndexValue(-2, val); }
  void SetWeight(Double_t w = 1) { fWeight = w; }
  Bool_t IsResidual(Int_t i) const { return fIndex[i] == -1; }
  Bool_t IsWeight(Int_t i) const { return fIndex[i] == -2; }
 
  Double_t* GetValue() const { return fValue; }
  Double_t GetValue(Int_t i) const { return fValue[i]; }
  Double_t GetWeight() const { return fWeight; }
 
  void MarkGroup(Int_t id);
  Int_t GetNGroups() const { return fNGroups; }
  Int_t GetGroupID(Int_t i) const { return fGroupID[i] - 1; }
 
  Bool_t IsGroupPresent(Int_t id) const;
 
  UInt_t GetRunID() const { return fRunID; }
  void SetRunID(UInt_t run) { fRunID = run; }
 
  Double_t GetGlobalDeriv(Int_t pnt, Int_t indx) const;
 
  Double_t GetLocalDeriv(Int_t pnt, Int_t indx) const;
 
  Double_t GetResidual(Int_t pnt) const;
 
  Double_t GetGloResWProd(Int_t indx) const;
 
  Double_t GetWeight(Int_t indx) const;
 
 protected:
  Int_t GetDtBufferSize() const { return GetUniqueID() & 0x0000ffff; }
  Int_t GetGrBufferSize() const { return GetUniqueID() >> 16; }
  void SetDtBufferSize(Int_t sz) { SetUniqueID((GetGrBufferSize() << 16) + sz); }
  void SetGrBufferSize(Int_t sz) { SetUniqueID(GetDtBufferSize() + (sz << 16)); }
 
  void ExpandDtBuffer(Int_t bfsize);
 
  void ExpandGrBuffer(Int_t bfsize);
 
 protected:
  Int_t fSize;        
  Int_t fNGroups;     
  UInt_t fRunID;      
  UShort_t* fGroupID; 
  Int_t* fIndex;      
  Double32_t* fValue; 
  Double32_t fWeight; 
 
  ClassDefOverride(MillePedeRecord, 3);
};
 
//_____________________________________________________________________________
inline void MillePedeRecord::AddIndexValue(Int_t ind, Double_t val)
{
  if (fSize >= GetDtBufferSize()) {
    ExpandDtBuffer(2 * (fSize + 1));
  }
  fIndex[fSize] = ind;
  fValue[fSize++] = val;
}
 
//_____________________________________________________________________________
inline Bool_t MillePedeRecord::IsGroupPresent(Int_t id) const
{
  id++;
  for (int i = fNGroups; i--;) {
    if (fGroupID[i] == id) {
      return kTRUE;
    }
  }
  return kFALSE;
}
 
} // namespace fwdalign
} // namespace o2
 
#endif