File: covarianceMatrix2Correlation.cpp

package info (click to toggle)
groops 0%2Bgit20240830%2Bds-1
  • links: PTS, VCS
  • area: non-free
  • in suites: trixie
  • size: 11,052 kB
  • sloc: cpp: 134,939; fortran: 1,569; makefile: 20
file content (72 lines) | stat: -rw-r--r-- 2,403 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
/***********************************************/
/**
* @file covarianceMatrix2Correlation.cpp
*
* @brief Compute the correlation matrix from a covariance matrix
*
* @author Andreas Kvas
* @date 2018-01-10
*/
/***********************************************/

// Latex documentation
#define DOCSTRING docstring
static const char *docstring = R"(
This program computes the pearson correlation coefficient
\begin{equation}
  \rho_{ij} = \frac{\sigma_{ij}}{\sigma_i \sigma_j}
\end{equation}
from a given covariance matrix stored in \configFile{inputfileCovarianceMatrix}{matrix}.
The result is stored in \configFile{outputfileCorrelationMatrix}{matrix}.
)";

/***********************************************/

#include "programs/program.h"
#include "files/fileMatrix.h"

/***** CLASS ***********************************/

/** @brief Compute the correlation matrix from a covariance matrix.
* @ingroup programsGroup */
class CovarianceMatrix2Correlation
{
public:
  void run(Config &config, Parallel::CommunicatorPtr comm);
};

GROOPS_REGISTER_PROGRAM(CovarianceMatrix2Correlation, PARALLEL, "compute the correlation matrix from a covariance matrix", Covariance)

/***********************************************/

void CovarianceMatrix2Correlation::run(Config &config, Parallel::CommunicatorPtr /*comm*/)
{
  try
  {
    FileName fileNameOut, fileNameIn;

    readConfig(config, "outputfileCorrelationMatrix", fileNameOut, Config::MUSTSET, "",  "correlation matrix");
    readConfig(config, "inputfileCovarianceMatrix",   fileNameIn,  Config::MUSTSET, "",  "covariance matrix");
    if(isCreateSchema(config)) return;

    Matrix covarianceMatrix;
    readFileMatrix(fileNameIn, covarianceMatrix);
    if(covarianceMatrix.getType() != Matrix::SYMMETRIC)
      throw(Exception("Covariance matrix must be symmetric."));
    fillSymmetric(covarianceMatrix);

    Matrix correlationMatrix(covarianceMatrix.rows(), Matrix::SYMMETRIC, Matrix::UPPER);
    for(UInt r=0; r<covarianceMatrix.rows(); r++)
      for(UInt c=r; c<covarianceMatrix.columns(); c++)
        correlationMatrix(r,c) = covarianceMatrix(r,c)/std::sqrt(covarianceMatrix(r,r)*covarianceMatrix(c,c));

    logStatus<<"write correlation matrix to <"<<fileNameOut<<">"<<Log::endl;
    writeFileMatrix(fileNameOut, correlationMatrix);
  }
  catch(std::exception &e)
  {
    GROOPS_RETHROW(e)
  }
}

/***********************************************/