decompose_LU_Cmplx.h

Go to the documentation of this file.

#ifndef DECOMPOSE_LU_CMPLX_INCLUDED
#define DECOMPOSE_LU_CMPLX_INCLUDED

#include <cfloat>
#include <cmath>
#include <valarray>

#include "bridge_complex.h"

#include "IO/bridgeIO.h"

class Decompose_LU_Cmplx
{
 public:
  Decompose_LU_Cmplx(size_t N) : N(N), N2(2 * N), size(N * N2), m_lu(size) {}

  void set_matrix(const double *mat);

  // solve Ax = b: vec -> A^{-1} vec
  void solve(double *vec);

  // M -> A^{-1}
  void get_inverse(double *mat_inv);

  // M -> A^{-1} * M
  void mult_inverse(double *mat);

  // return det(A)
  dcomplex determinant();

 private:
  int N;
  int N2;
  int size;
  std::valarray<double> m_lu;

  inline size_t re(int i, int j)
  {
    return N2 * i + 2 * j;
  }

  inline size_t im(int i, int j)
  {
    return N2 * i + 2 * j + 1;
  }

  inline size_t re(int i)
  {
    return 2 * i;
  }

  inline size_t im(int i)
  {
    return 2 * i + 1;
  }
};
#endif