MatrixSq.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_MATRIXSQ_H
#define ALICEO2_FWDALIGN_MATRIXSQ_H
 
#include <TMatrixDBase.h>
#include <TVectorD.h>
 
namespace o2
{
namespace fwdalign
{
 
class MatrixSq : public TMatrixDBase
{
 
 public:
  MatrixSq() : fSymmetric(kFALSE) {}
  MatrixSq(const MatrixSq& src);
  ~MatrixSq() override = default;
 
  MatrixSq& operator=(const MatrixSq& src);
 
  virtual Int_t GetSize() const { return fNcols; }
  virtual Float_t GetDensity() const = 0;
 
  void Clear(Option_t* option = "") override = 0;
 
  virtual Double_t Query(Int_t rown, Int_t coln) const { return operator()(rown, coln); }
  Double_t operator()(Int_t rown, Int_t coln) const override = 0;
  Double_t& operator()(Int_t rown, Int_t coln) override = 0;
 
  virtual Double_t QueryDiag(Int_t rc) const { return DiagElem(rc); }
  virtual Double_t DiagElem(Int_t r) const = 0;
  virtual Double_t& DiagElem(Int_t r) = 0;
  virtual void AddToRow(Int_t r, Double_t* valc, Int_t* indc, Int_t n) = 0;
 
  void Print(Option_t* option = "") const override = 0;
  virtual void Reset() = 0;
 
  virtual void PrintCOO() const;
 
  virtual void MultiplyByVec(const Double_t* vecIn, Double_t* vecOut) const;
 
  virtual void MultiplyByVec(const TVectorD& vecIn, TVectorD& vecOut) const;
 
  Bool_t IsSymmetric() const override { return fSymmetric; }
  void SetSymmetric(Bool_t v = kTRUE) { fSymmetric = v; }
 
  // ---------------------------------- Dummy methods of MatrixBase
  const Double_t* GetMatrixArray() const override
  {
    Error("GetMatrixArray", "Dummy");
    return nullptr;
  };
  Double_t* GetMatrixArray() override
  {
    Error("GetMatrixArray", "Dummy");
    return nullptr;
  };
  const Int_t* GetRowIndexArray() const override
  {
    Error("GetRowIndexArray", "Dummy");
    return nullptr;
  };
  Int_t* GetRowIndexArray() override
  {
    Error("GetRowIndexArray", "Dummy");
    return nullptr;
  };
  const Int_t* GetColIndexArray() const override
  {
    Error("GetColIndexArray", "Dummy");
    return nullptr;
  };
  Int_t* GetColIndexArray() override
  {
    Error("GetColIndexArray", "Dummy");
    return nullptr;
  };
  TMatrixDBase& SetRowIndexArray(Int_t*) override
  {
    Error("SetRowIndexArray", "Dummy");
    return *this;
  }
  TMatrixDBase& SetColIndexArray(Int_t*) override
  {
    Error("SetColIndexArray", "Dummy");
    return *this;
  }
  TMatrixDBase& GetSub(Int_t, Int_t, Int_t, Int_t, TMatrixDBase&, Option_t*) const override
  {
    Error("GetSub", "Dummy");
    return *((TMatrixDBase*)this);
  }
  TMatrixDBase& SetSub(Int_t, Int_t, const TMatrixDBase&) override
  {
    Error("GetSub", "Dummy");
    return *this;
  }
  TMatrixDBase& ResizeTo(Int_t, Int_t, Int_t) override
  {
    Error("ResizeTo", "Dummy");
    return *this;
  }
  TMatrixDBase& ResizeTo(Int_t, Int_t, Int_t, Int_t, Int_t) override
  {
    Error("ResizeTo", "Dummy");
    return *this;
  }
  virtual void Allocate(Int_t, Int_t, Int_t, Int_t, Int_t, Int_t)
  {
    Error("Allocate", "Dummy");
    return;
  }
 
  static Bool_t IsZero(Double_t x, Double_t thresh = 1e-64) { return x > 0 ? (x < thresh) : (x > -thresh); }
 
 protected:
  void Swap(int& r, int& c) const
  {
    int t = r;
    r = c;
    c = t;
  }
 
 protected:
  Bool_t fSymmetric; 
 
  ClassDefOverride(MatrixSq, 1);
};
 
//___________________________________________________________
inline void MatrixSq::MultiplyByVec(const TVectorD& vecIn, TVectorD& vecOut) const
{
  MultiplyByVec(vecIn.GetMatrixArray(), vecOut.GetMatrixArray());
}
 
} // namespace fwdalign
} // namespace o2
 
#endif