1 /**********************************************************************************************/
3 /* Abstract class for matrix used for */
4 /* millepede2 operation. */
5 /* Works for expandable square matrices */
6 /* of arbitrary dimension */
7 /* Author: ruben.shahoyan@cern.ch */
10 /**********************************************************************************************/
18 #include "AliMatrixSq.h"
23 AliMatrixSq & AliMatrixSq::operator=(const AliMatrixSq &src)
26 if (this == &src) return *this;
27 TMatrixDBase::operator=(src);
28 fSymmetric = src.fSymmetric;
32 //___________________________________________________________
33 void AliMatrixSq::MultiplyByVec(const Double_t *vecIn,Double_t *vecOut) const
35 // fill vecOut by matrix*vecIn
36 // vector should be of the same size as the matrix
37 for (int i=GetSize();i--;) {
39 for (int j=GetSize();j--;) vecOut[i] += vecIn[j]*(*this)(i,j);
44 //___________________________________________________________
45 void AliMatrixSq::PrintCOO() const
47 // print matrix in COO sparse format
49 // get number of non-zero elements
52 for (int ir=0;ir<sz;ir++) for (int ic=0;ic<sz;ic++) if (Query(ir,ic)!=0) nnz++;
54 printf("%d %d %d\n",sz,sz,nnz);
56 for (int ir=0;ir<sz;ir++) for (int ic=0;ic<sz;ic++) if ((vl=Query(ir,ic))!=0) printf("%d %d %f\n",ir,ic,vl);