LCOV - code coverage report
Current view: top level - ves - CoeffsMatrix.cpp (source / functions) Hit Total Coverage
Test: plumed test coverage Lines: 105 356 29.5 %
Date: 2024-10-11 08:09:47 Functions: 20 79 25.3 %

          Line data    Source code
       1             : /* +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
       2             :    Copyright (c) 2016-2021 The VES code team
       3             :    (see the PEOPLE-VES file at the root of this folder for a list of names)
       4             : 
       5             :    See http://www.ves-code.org for more information.
       6             : 
       7             :    This file is part of VES code module.
       8             : 
       9             :    The VES code module is free software: you can redistribute it and/or modify
      10             :    it under the terms of the GNU Lesser General Public License as published by
      11             :    the Free Software Foundation, either version 3 of the License, or
      12             :    (at your option) any later version.
      13             : 
      14             :    The VES code module is distributed in the hope that it will be useful,
      15             :    but WITHOUT ANY WARRANTY; without even the implied warranty of
      16             :    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
      17             :    GNU Lesser General Public License for more details.
      18             : 
      19             :    You should have received a copy of the GNU Lesser General Public License
      20             :    along with the VES code module.  If not, see <http://www.gnu.org/licenses/>.
      21             : +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ */
      22             : 
      23             : #include "CoeffsMatrix.h"
      24             : #include "CoeffsVector.h"
      25             : #include "BasisFunctions.h"
      26             : 
      27             : #include "tools/Tools.h"
      28             : #include "core/Value.h"
      29             : #include "tools/File.h"
      30             : #include "tools/Exception.h"
      31             : #include "tools/Random.h"
      32             : #include "tools/Communicator.h"
      33             : 
      34             : #include <vector>
      35             : #include <cmath>
      36             : #include <iostream>
      37             : #include <sstream>
      38             : #include <cstdio>
      39             : #include <cfloat>
      40             : 
      41             : 
      42             : namespace PLMD {
      43             : namespace ves {
      44             : 
      45           0 : CoeffsMatrix::CoeffsMatrix(
      46             :   const std::string& label,
      47             :   const std::vector<std::string>& dimension_labels,
      48             :   const std::vector<unsigned int>& indices_shape,
      49             :   Communicator& cc,
      50             :   const bool diagonal,
      51           0 :   const bool use_iteration_counter):
      52             :   CoeffsBase(label,dimension_labels,indices_shape,use_iteration_counter),
      53           0 :   data(0),
      54           0 :   size_(0),
      55           0 :   nrows_(0),
      56           0 :   ncolumns_(0),
      57           0 :   diagonal_(diagonal),
      58           0 :   averaging_counter(0),
      59           0 :   averaging_exp_decay_(0),
      60           0 :   mycomm(cc)
      61             : {
      62           0 :   setupMatrix();
      63           0 : }
      64             : 
      65             : 
      66           0 : CoeffsMatrix::CoeffsMatrix(
      67             :   const std::string& label,
      68             :   std::vector<Value*>& args,
      69             :   std::vector<BasisFunctions*>& basisf,
      70             :   Communicator& cc,
      71             :   const bool diagonal,
      72           0 :   const bool use_iteration_counter):
      73             :   CoeffsBase(label,args,basisf,use_iteration_counter),
      74           0 :   data(0),
      75           0 :   size_(0),
      76           0 :   nrows_(0),
      77           0 :   ncolumns_(0),
      78           0 :   diagonal_(diagonal),
      79           0 :   averaging_counter(0),
      80           0 :   averaging_exp_decay_(0),
      81           0 :   mycomm(cc)
      82             : {
      83           0 :   setupMatrix();
      84           0 : }
      85             : 
      86             : 
      87           0 : CoeffsMatrix::CoeffsMatrix(
      88             :   const std::string& label,
      89             :   std::vector<std::vector<Value*> >& argsv,
      90             :   std::vector<std::vector<BasisFunctions*> >& basisfv,
      91             :   Communicator& cc,
      92             :   const bool diagonal,
      93             :   const bool use_iteration_counter,
      94           0 :   const std::string& multicoeffs_label):
      95             :   CoeffsBase(label,argsv,basisfv,use_iteration_counter,multicoeffs_label),
      96           0 :   data(0),
      97           0 :   size_(0),
      98           0 :   nrows_(0),
      99           0 :   ncolumns_(0),
     100           0 :   diagonal_(diagonal),
     101           0 :   averaging_counter(0),
     102           0 :   averaging_exp_decay_(0),
     103           0 :   mycomm(cc)
     104             : {
     105           0 :   setupMatrix();
     106           0 : }
     107             : 
     108             : 
     109         172 : CoeffsMatrix::CoeffsMatrix(
     110             :   const std::string& label,
     111             :   CoeffsVector* coeffsVec,
     112             :   Communicator& cc,
     113         172 :   const bool diagonal):
     114             :   CoeffsBase( *(static_cast<CoeffsBase*>(coeffsVec)) ),
     115         172 :   data(0),
     116         172 :   size_(0),
     117         172 :   nrows_(0),
     118         172 :   ncolumns_(0),
     119         172 :   diagonal_(diagonal),
     120         172 :   averaging_counter(0),
     121         172 :   averaging_exp_decay_(0),
     122         172 :   mycomm(cc)
     123             : {
     124         172 :   setLabels(label);
     125         172 :   setupMatrix();
     126         172 : }
     127             : 
     128             : 
     129         344 : CoeffsMatrix::~CoeffsMatrix() {}
     130             : 
     131             : 
     132         172 : void CoeffsMatrix::setupMatrix() {
     133         172 :   nrows_=numberOfCoeffs();
     134         172 :   ncolumns_=nrows_;
     135         172 :   if(diagonal_) {
     136         172 :     size_=nrows_;
     137             :   }
     138             :   else {
     139           0 :     size_=(nrows_*nrows_-nrows_)/2+nrows_;
     140             :   }
     141         172 :   clear();
     142         172 : }
     143             : 
     144             : 
     145         439 : size_t CoeffsMatrix::getSize() const {
     146         439 :   return size_;
     147             : }
     148             : 
     149             : 
     150       23395 : bool CoeffsMatrix::isDiagonal() const {
     151       23395 :   return diagonal_;
     152             : }
     153             : 
     154             : 
     155           0 : bool CoeffsMatrix::sameShape(CoeffsVector& coeffsvector_in) const {
     156           0 :   return CoeffsBase::sameShape( *(static_cast<CoeffsBase*>(&coeffsvector_in)) );
     157             : }
     158             : 
     159             : 
     160           0 : bool CoeffsMatrix::sameShape(CoeffsMatrix& coeffsmat_in) const {
     161           0 :   return CoeffsBase::sameShape( *(static_cast<CoeffsBase*>(&coeffsmat_in)) );
     162             : }
     163             : 
     164             : 
     165           0 : bool CoeffsMatrix::sameShape(CoeffsMatrix& coeffsmat0, CoeffsMatrix& coeffsmat1) {
     166           0 :   return coeffsmat0.sameShape(coeffsmat1);
     167             : }
     168             : 
     169             : 
     170           0 : bool CoeffsMatrix::sameShape(CoeffsVector& coeffsvec, CoeffsMatrix& coeffsmat) {
     171           0 :   return coeffsmat.sameShape(coeffsvec);
     172             : }
     173             : 
     174             : 
     175           0 : bool CoeffsMatrix::sameShape(CoeffsMatrix& coeffsmat, CoeffsVector& coeffsvec) {
     176           0 :   return coeffsmat.sameShape(coeffsvec);
     177             : }
     178             : 
     179             : 
     180           0 : void CoeffsMatrix::sumCommMPI() {
     181           0 :   mycomm.Sum(data);
     182           0 : }
     183             : 
     184             : 
     185           0 : void CoeffsMatrix::sumCommMPI(Communicator& cc) {
     186           0 :   cc.Sum(data);
     187           0 : }
     188             : 
     189             : 
     190           0 : void CoeffsMatrix::sumMultiSimCommMPI(Communicator& multi_sim_cc) {
     191           0 :   if(mycomm.Get_rank()==0) {
     192           0 :     double nwalkers = static_cast<double>(multi_sim_cc.Get_size());
     193           0 :     multi_sim_cc.Sum(data);
     194           0 :     scaleAllValues(1.0/nwalkers);
     195             :   }
     196           0 :   mycomm.Bcast(data,0);
     197           0 : }
     198             : 
     199             : 
     200     5425906 : size_t CoeffsMatrix::getMatrixIndex(const size_t index1, const size_t index2) const {
     201             :   size_t matrix_idx;
     202             :   plumed_dbg_assert(index1<nrows_);
     203             :   plumed_dbg_assert(index2<ncolumns_);
     204     5425906 :   if(diagonal_) {
     205             :     // plumed_massert(index1==index2,"CoeffsMatrix: you trying to access a off-diagonal element of a diagonal coeffs matrix");
     206             :     matrix_idx=index1;
     207             :   }
     208           0 :   else if (index1<=index2) {
     209           0 :     matrix_idx=index2+index1*(nrows_-1)-index1*(index1-1)/2;
     210             :   }
     211             :   else {
     212           0 :     matrix_idx=index1+index2*(nrows_-1)-index2*(index2-1)/2;
     213             :   }
     214     5425906 :   return matrix_idx;
     215             : }
     216             : 
     217             : 
     218         267 : void CoeffsMatrix::clear() {
     219         267 :   data.resize(getSize());
     220       19388 :   for(size_t i=0; i<data.size(); i++) {
     221       19121 :     data[i]=0.0;
     222             :   }
     223         267 : }
     224             : 
     225             : 
     226           0 : void CoeffsMatrix::setAllValuesToZero() {
     227           0 :   for(size_t i=0; i<data.size(); i++) {
     228           0 :     data[i]=0.0;
     229             :   }
     230           0 : }
     231             : 
     232             : 
     233       27630 : double CoeffsMatrix::getValue(const size_t index1, const size_t index2) const {
     234       27630 :   return data[getMatrixIndex(index1,index2)];
     235             : }
     236             : 
     237             : 
     238           0 : double CoeffsMatrix::getValue(const std::vector<unsigned int>& indices1, const std::vector<unsigned int>& indices2) const {
     239           0 :   return getValue(getIndex(indices1),getIndex(indices2));
     240             : }
     241             : 
     242             : 
     243         219 : void CoeffsMatrix::setValue(const size_t index1, const size_t index2, const double value) {
     244         219 :   data[getMatrixIndex(index1,index2)]=value;
     245         219 : }
     246             : 
     247             : 
     248           0 : void CoeffsMatrix::setValue(const std::vector<unsigned int>& indices1, const std::vector<unsigned int>& indices2, const double value) {
     249           0 :   setValue(getIndex(indices1),getIndex(indices2),value);
     250           0 : }
     251             : 
     252             : 
     253           0 : double& CoeffsMatrix::operator()(const size_t index1, const size_t index2) {
     254           0 :   return data[getMatrixIndex(index1,index2)];
     255             : }
     256             : 
     257             : 
     258     1790345 : const double& CoeffsMatrix::operator()(const size_t index1, const size_t index2) const {
     259     1790345 :   return data[getMatrixIndex(index1,index2)];
     260             : }
     261             : 
     262             : 
     263           0 : double& CoeffsMatrix::operator()(const std::vector<unsigned int>& indices1, const std::vector<unsigned int>& indices2) {
     264           0 :   return data[getMatrixIndex(getIndex(indices1),getIndex(indices2))];
     265             : }
     266             : 
     267             : 
     268           0 : const double& CoeffsMatrix::operator()(const std::vector<unsigned int>& indices1, const std::vector<unsigned int>& indices2) const {
     269           0 :   return data[getMatrixIndex(getIndex(indices1),getIndex(indices2))];
     270             : }
     271             : 
     272             : 
     273       22735 : CoeffsVector operator*(const CoeffsMatrix& coeffs_matrix, const CoeffsVector& coeffs_vector) {
     274       22735 :   CoeffsVector new_coeffs_vector(coeffs_vector);
     275       22735 :   new_coeffs_vector.clear();
     276       22735 :   plumed_massert(coeffs_vector.numberOfCoeffs()==coeffs_matrix.numberOfCoeffs(),"CoeffsMatrix and CoeffsVector are of the wrong size");
     277             :   size_t numcoeffs = coeffs_vector.numberOfCoeffs();
     278       22735 :   if(coeffs_matrix.isDiagonal()) {
     279     1813080 :     for(size_t i=0; i<numcoeffs; i++) {
     280     1790345 :       new_coeffs_vector(i) = coeffs_matrix(i,i)*coeffs_vector(i);
     281             :     }
     282             :   }
     283             :   else {
     284           0 :     for(size_t i=0; i<numcoeffs; i++) {
     285           0 :       for(size_t j=0; j<numcoeffs; j++) {
     286           0 :         new_coeffs_vector(i) += coeffs_matrix(i,j)*coeffs_vector(j);
     287             :       }
     288             :     }
     289             :   }
     290       22735 :   return new_coeffs_vector;
     291           0 : }
     292             : 
     293             : 
     294           0 : void CoeffsMatrix::addToValue(const size_t index1, const size_t index2, const double value) {
     295           0 :   data[getMatrixIndex(index1,index2)]+=value;
     296           0 : }
     297             : 
     298             : 
     299           0 : void CoeffsMatrix::addToValue(const std::vector<unsigned int>& indices1, const std::vector<unsigned int>& indices2, const double value) {
     300           0 :   addToValue(getIndex(indices1),getIndex(indices2),value);
     301           0 : }
     302             : 
     303             : 
     304       22810 : void CoeffsMatrix::scaleAllValues(const double scalef) {
     305     1823850 :   for(size_t i=0; i<data.size(); i++) {
     306     1801040 :     data[i]*=scalef;
     307             :   }
     308       22810 : }
     309             : 
     310             : 
     311       22810 : CoeffsMatrix& CoeffsMatrix::operator*=(const double scalef) {
     312       22810 :   scaleAllValues(scalef);
     313       22810 :   return *this;
     314             : }
     315             : 
     316             : 
     317           0 : CoeffsMatrix operator*(const double scalef, const CoeffsMatrix& coeffsmatrix) {
     318           0 :   return CoeffsMatrix(coeffsmatrix)*=scalef;
     319             : }
     320             : 
     321             : 
     322           0 : CoeffsMatrix operator*(const CoeffsMatrix& coeffsmatrix, const double scalef) {
     323           0 :   return scalef*coeffsmatrix;
     324             : }
     325             : 
     326             : 
     327           0 : CoeffsMatrix& CoeffsMatrix::operator*=(const CoeffsMatrix& other_coeffsmatrix) {
     328           0 :   plumed_massert(data.size()==other_coeffsmatrix.getSize(),"Coeffs matrices do not have the same size");
     329           0 :   for(size_t i=0; i<data.size(); i++) {
     330           0 :     data[i]*=other_coeffsmatrix.data[i];
     331             :   }
     332           0 :   return *this;
     333             : }
     334             : 
     335             : 
     336           0 : CoeffsMatrix CoeffsMatrix::operator*(const CoeffsMatrix& other_coeffsmatrix) const {
     337           0 :   return CoeffsMatrix(*this)*=other_coeffsmatrix;
     338             : }
     339             : 
     340             : 
     341           0 : void CoeffsMatrix::setValues(const double value) {
     342           0 :   for(size_t i=0; i<data.size(); i++) {
     343           0 :     data[i]=value;
     344             :   }
     345           0 : }
     346             : 
     347             : 
     348       22810 : void CoeffsMatrix::setValues(const std::vector<double>& values) {
     349       22810 :   plumed_massert( data.size()==values.size(), "Incorrect size");
     350     1823850 :   for(size_t i=0; i<data.size(); i++) {
     351     1801040 :     data[i]=values[i];
     352             :   }
     353       22810 : }
     354             : 
     355             : 
     356           0 : void CoeffsMatrix::setValues(const CoeffsMatrix& other_coeffsmatrix) {
     357           0 :   plumed_massert( data.size()==other_coeffsmatrix.getSize(), "Incorrect size");
     358           0 :   for(size_t i=0; i<data.size(); i++) {
     359           0 :     data[i]=other_coeffsmatrix.data[i];
     360             :   }
     361           0 : }
     362             : 
     363             : 
     364           0 : CoeffsMatrix& CoeffsMatrix::operator=(const double value) {
     365           0 :   setValues(value);
     366           0 :   return *this;
     367             : }
     368             : 
     369             : 
     370       22810 : CoeffsMatrix& CoeffsMatrix::operator=(const std::vector<double>& values) {
     371       22810 :   setValues(values);
     372       22810 :   return *this;
     373             : }
     374             : 
     375             : 
     376             : // CoeffsMatrix& CoeffsMatrix::operator=(const CoeffsMatrix& other_coeffsmatrix) {
     377             : //   setValues(other_coeffsmatrix);
     378             : //   return *this;
     379             : // }
     380             : 
     381             : 
     382           0 : CoeffsMatrix CoeffsMatrix::operator+() const {
     383           0 :   return *this;
     384             : }
     385             : 
     386             : 
     387           0 : CoeffsMatrix CoeffsMatrix::operator-() const {
     388           0 :   return CoeffsMatrix(*this)*=-1.0;
     389             : }
     390             : 
     391             : 
     392           0 : void CoeffsMatrix::addToValues(const double value) {
     393           0 :   for(size_t i=0; i<data.size(); i++) {
     394           0 :     data[i]+=value;
     395             :   }
     396           0 : }
     397             : 
     398             : 
     399           0 : void CoeffsMatrix::addToValues(const std::vector<double>& values) {
     400           0 :   plumed_massert( data.size()==values.size(), "Incorrect size");
     401           0 :   for(size_t i=0; i<data.size(); i++) {
     402           0 :     data[i]+=values[i];
     403             :   }
     404           0 : }
     405             : 
     406             : 
     407           0 : void CoeffsMatrix::addToValues(const CoeffsMatrix& other_coeffsmatrix) {
     408           0 :   plumed_massert( data.size()==other_coeffsmatrix.getSize(), "Incorrect size");
     409           0 :   for(size_t i=0; i<data.size(); i++) {
     410           0 :     data[i]+=other_coeffsmatrix.data[i];
     411             :   }
     412           0 : }
     413             : 
     414             : 
     415           0 : void CoeffsMatrix::subtractFromValues(const double value) {
     416           0 :   for(size_t i=0; i<data.size(); i++) {
     417           0 :     data[i]-=value;
     418             :   }
     419           0 : }
     420             : 
     421             : 
     422           0 : void CoeffsMatrix::subtractFromValues(const std::vector<double>& values) {
     423           0 :   plumed_massert( data.size()==values.size(), "Incorrect size");
     424           0 :   for(size_t i=0; i<data.size(); i++) {
     425           0 :     data[i]-=values[i];
     426             :   }
     427           0 : }
     428             : 
     429             : 
     430           0 : void CoeffsMatrix::subtractFromValues(const CoeffsMatrix& other_coeffsmatrix) {
     431           0 :   plumed_massert( data.size()==other_coeffsmatrix.getSize(), "Incorrect size");
     432           0 :   for(size_t i=0; i<data.size(); i++) {
     433           0 :     data[i]-=other_coeffsmatrix.data[i];
     434             :   }
     435           0 : }
     436             : 
     437             : 
     438           0 : CoeffsMatrix& CoeffsMatrix::operator+=(const double value) {
     439           0 :   addToValues(value);
     440           0 :   return *this;
     441             : }
     442             : 
     443             : 
     444           0 : CoeffsMatrix operator+(const double value, const CoeffsMatrix& coeffsmatrix) {
     445           0 :   return coeffsmatrix+value;
     446             : }
     447             : 
     448             : 
     449           0 : CoeffsMatrix operator+(const CoeffsMatrix& coeffsmatrix, const double value) {
     450           0 :   return CoeffsMatrix(coeffsmatrix)+=value;
     451             : }
     452             : 
     453             : 
     454           0 : CoeffsMatrix& CoeffsMatrix::operator+=(const std::vector<double>& values) {
     455           0 :   addToValues(values);
     456           0 :   return *this;
     457             : }
     458             : 
     459             : 
     460           0 : CoeffsMatrix operator+(const std::vector<double>& values, const CoeffsMatrix& coeffsmatrix) {
     461           0 :   return coeffsmatrix+values;
     462             : }
     463             : 
     464             : 
     465           0 : CoeffsMatrix operator+(const CoeffsMatrix& coeffsmatrix, const std::vector<double>& values) {
     466           0 :   return CoeffsMatrix(coeffsmatrix)+=values;
     467             : }
     468             : 
     469             : 
     470           0 : CoeffsMatrix& CoeffsMatrix::operator-=(const double value) {
     471           0 :   subtractFromValues(value);
     472           0 :   return *this;
     473             : }
     474             : 
     475             : 
     476           0 : CoeffsMatrix operator-(const double value, const CoeffsMatrix& coeffsmatrix) {
     477           0 :   return -1.0*coeffsmatrix+value;
     478             : }
     479             : 
     480             : 
     481           0 : CoeffsMatrix operator-(const CoeffsMatrix& coeffsmatrix, const double value) {
     482           0 :   return CoeffsMatrix(coeffsmatrix)-=value;
     483             : }
     484             : 
     485             : 
     486           0 : CoeffsMatrix& CoeffsMatrix::operator-=(const std::vector<double>& values) {
     487           0 :   subtractFromValues(values);
     488           0 :   return *this;
     489             : }
     490             : 
     491             : 
     492           0 : CoeffsMatrix operator-(const std::vector<double>& values, const CoeffsMatrix& coeffsmatrix) {
     493           0 :   return -1.0*coeffsmatrix+values;
     494             : }
     495             : 
     496             : 
     497           0 : CoeffsMatrix operator-(const CoeffsMatrix& coeffsmatrix, const std::vector<double>& values) {
     498           0 :   return CoeffsMatrix(coeffsmatrix)-=values;
     499             : }
     500             : 
     501             : 
     502           0 : CoeffsMatrix& CoeffsMatrix::operator+=(const CoeffsMatrix& other_coeffsmatrix) {
     503           0 :   addToValues(other_coeffsmatrix);
     504           0 :   return *this;
     505             : }
     506             : 
     507             : 
     508           0 : CoeffsMatrix CoeffsMatrix::operator+(const CoeffsMatrix& other_coeffsmatrix) const {
     509           0 :   return CoeffsMatrix(*this)+=other_coeffsmatrix;
     510             : }
     511             : 
     512             : 
     513           0 : CoeffsMatrix& CoeffsMatrix::operator-=(const CoeffsMatrix& other_coeffsmatrix) {
     514           0 :   subtractFromValues(other_coeffsmatrix);
     515           0 :   return *this;
     516             : }
     517             : 
     518             : 
     519           0 : CoeffsMatrix CoeffsMatrix::operator-(const CoeffsMatrix& other_coeffsmatrix) const {
     520           0 :   return CoeffsMatrix(*this)-=other_coeffsmatrix;
     521             : }
     522             : 
     523             : 
     524           0 : void CoeffsMatrix::averageMatrices(CoeffsMatrix& coeffsmat0, CoeffsMatrix& coeffsmat1) {
     525           0 :   plumed_massert(sameShape(coeffsmat0,coeffsmat1),"both CoeffsMatrix objects need to have the same shape");
     526           0 :   for(size_t i=0; i<coeffsmat0.getSize(); i++) {
     527           0 :     coeffsmat0.data[i] = coeffsmat1.data[i] = 0.5 * (coeffsmat0.data[i]+coeffsmat1.data[i]);
     528             :   }
     529           0 : }
     530             : 
     531             : 
     532           0 : void CoeffsMatrix::averageMatrices(const std::vector<CoeffsMatrix*>& coeffsmatSet) {
     533           0 :   double norm_factor = 1.0/static_cast<double>(coeffsmatSet.size());
     534           0 :   for(unsigned int k=1; k<coeffsmatSet.size(); k++) {
     535           0 :     plumed_massert(coeffsmatSet[0]->sameShape(*coeffsmatSet[k]),"All CoeffsMatrix objects need to have the same shape");
     536             :   }
     537           0 :   for(size_t i=0; i<coeffsmatSet[0]->getSize(); i++) {
     538             :     double value = 0.0;
     539           0 :     for(unsigned int k=0; k<coeffsmatSet.size(); k++) {
     540           0 :       value += coeffsmatSet[k]->data[i];
     541             :     }
     542           0 :     value *= norm_factor;
     543           0 :     for(unsigned int k=0; k<coeffsmatSet.size(); k++) {
     544           0 :       coeffsmatSet[k]->data[i] = value;
     545             :     }
     546             :   }
     547           0 : }
     548             : 
     549             : 
     550             : 
     551           0 : double CoeffsMatrix::getMinValue() const {
     552             :   double min_value=DBL_MAX;
     553           0 :   for(size_t i=0; i<data.size(); i++) {
     554           0 :     if(data[i]<min_value) {
     555             :       min_value=data[i];
     556             :     }
     557             :   }
     558           0 :   return min_value;
     559             : }
     560             : 
     561             : 
     562           0 : double CoeffsMatrix::getMaxValue() const {
     563             :   double max_value=DBL_MIN;
     564           0 :   for(size_t i=0; i<data.size(); i++) {
     565           0 :     if(data[i]>max_value) {
     566             :       max_value=data[i];
     567             :     }
     568             :   }
     569           0 :   return max_value;
     570             : }
     571             : 
     572             : 
     573           0 : void CoeffsMatrix::randomizeValuesGaussian(int randomSeed) {
     574           0 :   Random rnd;
     575             :   if (randomSeed<0) {randomSeed = -randomSeed;}
     576           0 :   rnd.setSeed(-randomSeed);
     577           0 :   for(size_t i=0; i<data.size(); i++) {
     578           0 :     data[i]=rnd.Gaussian();
     579             :   }
     580           0 : }
     581             : 
     582             : 
     583           0 : void CoeffsMatrix::resetAveraging() {
     584           0 :   clear();
     585             :   resetAveragingCounter();
     586           0 : }
     587             : 
     588             : 
     589           0 : void CoeffsMatrix::addToAverage(const CoeffsMatrix& coeffsmat) {
     590           0 :   plumed_massert( data.size()==coeffsmat.getSize(), "Incorrect size");
     591             :   //
     592           0 :   double aver_decay = 1.0 / ( static_cast<double>(averaging_counter) + 1.0 );
     593           0 :   if(averaging_exp_decay_>0 &&  (averaging_counter+1 > averaging_exp_decay_) ) {
     594           0 :     aver_decay = 1.0 / static_cast<double>(averaging_exp_decay_);
     595             :   }
     596             :   //
     597           0 :   for(size_t i=0; i<data.size(); i++) {
     598           0 :     data[i]+=(coeffsmat.data[i]-data[i])*aver_decay;
     599             :   }
     600           0 :   averaging_counter++;
     601           0 : }
     602             : 
     603             : 
     604         660 : void CoeffsMatrix::writeToFile(OFile& ofile) {
     605         660 :   writeHeaderToFile(ofile);
     606         660 :   writeDataToFile(ofile);
     607         660 : }
     608             : 
     609             : 
     610           0 : void CoeffsMatrix::writeToFile(const std::string& filepath, const bool append_file, Action* action_pntr) {
     611           0 :   OFile file;
     612           0 :   if(action_pntr!=NULL) {
     613           0 :     file.link(*action_pntr);
     614             :   }
     615             :   else {
     616           0 :     file.link(mycomm);
     617             :   }
     618           0 :   if(append_file) { file.enforceRestart(); }
     619           0 :   file.open(filepath);
     620           0 :   writeToFile(file);
     621           0 :   file.close();
     622           0 : }
     623             : 
     624             : 
     625         660 : void CoeffsMatrix::writeMatrixInfoToFile(OFile& ofile) {
     626         660 :   std::string field_diagonal = "diagonal_matrix";
     627         660 :   ofile.addConstantField(field_diagonal).printField(field_diagonal,isDiagonal());
     628         660 : }
     629             : 
     630             : 
     631         660 : void CoeffsMatrix::writeHeaderToFile(OFile& ofile) {
     632         660 :   ofile.clearFields();
     633         660 :   if(isIterationCounterActive()) {
     634         660 :     writeIterationCounterAndTimeToFile(ofile);
     635             :   }
     636         660 :   writeCoeffsInfoToFile(ofile);
     637         660 :   writeMatrixInfoToFile(ofile);
     638         660 : }
     639             : 
     640             : 
     641         660 : void CoeffsMatrix::writeDataToFile(OFile& ofile) {
     642         660 :   if(diagonal_) {
     643         660 :     writeDataDiagonalToFile(ofile);
     644             :   }
     645             :   else {
     646           0 :     writeDataFullToFile(ofile);
     647             :   }
     648         660 : }
     649             : 
     650             : 
     651         660 : void CoeffsMatrix::writeDataDiagonalToFile(OFile& ofile) {
     652             :   //
     653         660 :   std::string field_indices_prefix = "idx_";
     654             :   std::string field_coeffs = getDataLabel();
     655         660 :   std::string field_index = "index";
     656             :   //
     657         660 :   std::string int_fmt = "%8d";
     658         660 :   std::string str_separate = "#!-------------------";
     659             :   //
     660         660 :   std::vector<char> s1(20);
     661         660 :   std::vector<unsigned int> indices(numberOfDimensions());
     662         660 :   std::vector<std::string> ilabels(numberOfDimensions());
     663        1520 :   for(unsigned int k=0; k<numberOfDimensions(); k++) {
     664        1720 :     ilabels[k]=field_indices_prefix+getDimensionLabel(k);
     665             :   }
     666             :   //
     667       28290 :   for(size_t i=0; i<numberOfCoeffs(); i++) {
     668       27630 :     indices=getIndices(i);
     669       77540 :     for(unsigned int k=0; k<numberOfDimensions(); k++) {
     670       49910 :       std::sprintf(s1.data(),int_fmt.c_str(),indices[k]);
     671       99820 :       ofile.printField(ilabels[k],s1.data());
     672             :     }
     673       55260 :     ofile.fmtField(" "+getOutputFmt()).printField(field_coeffs,getValue(i,i));
     674       27630 :     std::sprintf(s1.data(),int_fmt.c_str(),i); ofile.printField(field_index,s1.data());
     675       27630 :     ofile.printField();
     676             :   }
     677         660 :   ofile.fmtField();
     678             :   // blank line between iterations to allow proper plotting with gnuplot
     679         660 :   ofile.printf("%s\n",str_separate.c_str());
     680         660 :   ofile.printf("\n");
     681         660 :   ofile.printf("\n");
     682        1320 : }
     683             : 
     684             : 
     685           0 : void CoeffsMatrix::writeDataFullToFile(OFile& ofile) {
     686             :   //
     687           0 :   std::string field_index_row = "idx_row";
     688           0 :   std::string field_index_column = "idx_column";
     689             :   std::string field_coeffs = getDataLabel();
     690             :   //
     691           0 :   std::string int_fmt = "%8d";
     692           0 :   std::string str_separate = "#!-------------------";
     693             :   //
     694           0 :   std::vector<char> s1(20);
     695             :   //
     696           0 :   for(size_t i=0; i<nrows_; i++) {
     697           0 :     for(size_t j=0; j<ncolumns_; j++) {
     698             :       std::sprintf(s1.data(),int_fmt.c_str(),i);
     699           0 :       ofile.printField(field_index_row,s1.data());
     700             :       std::sprintf(s1.data(),int_fmt.c_str(),j);
     701           0 :       ofile.printField(field_index_column,s1.data());
     702           0 :       ofile.fmtField(" "+getOutputFmt()).printField(field_coeffs,getValue(i,j));
     703           0 :       ofile.printField();
     704             :     }
     705             :   }
     706           0 :   ofile.fmtField();
     707             :   // blank line between iterations to allow proper plotting with gnuplot
     708           0 :   ofile.printf("%s\n",str_separate.c_str());
     709           0 :   ofile.printf("\n");
     710           0 :   ofile.printf("\n");
     711           0 : }
     712             : 
     713             : 
     714             : }
     715             : }

Generated by: LCOV version 1.15