matrix.hpp

Go to the documentation of this file.

#ifndef EXPLICIT_MAPMAKING_MATRIX_HPP_
#define EXPLICIT_MAPMAKING_MATRIX_HPP_

#include "grid.hpp"
#include "io.hpp"
#include <stdio.h>
#include <string>
#include <boost/filesystem.hpp>

namespace fs = boost::filesystem;

namespace mapmaking {
// Squared matrix
// It maximizes the size of the blocks given a processor grid
class Matrix {
 public:
  Matrix() {}
  Matrix(Grid* grid, int n_row, int n_col);
  Matrix Dot(Matrix& right, char trans_left='N', char trans_right='N');
  std::vector<double>* local_part() { return &local_part_; }
  int n_row() const { return n_row_; }
  int n_col() const { return n_col_; }
  Grid* grid() { return grid_; }
  std::vector<int>* desc() { return &desc_; }
  void PrintLocalPart(); // Debugging tools
  void PrintMatrix();  //
  void TestAllocation();  //
  void TestAllocationVector();  //
  void SaveMatrix(fs::path folder);
  std::vector<double> GetColumn(int i_col);
  void LoadMatrix(std::vector<double>& matrix, int root_proc);
  Matrix& operator+= (const Matrix& matrix);
  Matrix& operator-= (const Matrix& matrix);
  Matrix& operator*= (double factor);
  Matrix& operator/= (double factor);
  friend Matrix LoadMatrix(fs::path folder, Grid* grid);  // TODO turn this into a constructor

 protected:
  std::vector<double> local_part_;
  int block_first_row_ = 0;
  int n_local_rows_ = 0;
  int n_local_cols_ = 0;
  int n_row_;
  int n_col_;
  int n_row_block_;  // Even by construction
  int n_col_block_;  // Even by construction
  Grid* grid_ = NULL;
  std::vector<int> desc_;  // Descriptor of the matrix
  std::vector<double> GatherSubmatrix(int proc,  int i_row, int i_col=0, int n_rows=1, int n_cols=1);
  void ScatterSubmatrix(std::vector<double>& submatrix, int proc,  int i_row, int i_col=0, int n_rows=1);
};

  Matrix LoadMatrix(fs::path folder, Grid* grid);
  Matrix operator+(Matrix m_left, const Matrix& m_right);
  Matrix operator-(Matrix m_left, const Matrix& m_right);
  Matrix operator*(Matrix& m_left, Matrix& m_right);
  Matrix operator*(Matrix m_left, double factor);
  Matrix operator/(Matrix m_left, double factor);
}

#endif  // EXPLICIT_MAPMAKING_MATRIX_HPP_