SparseZCholeskySolver.h

Go to the documentation of this file.

// ========================================================================= //
// This file is part of MyraMath, copyright (c) 2014-2019 by Ryan A Chilton  //
// and distributed by MyraCore, LLC. See LICENSE.txt for license terms.      //
// ========================================================================= //

#ifndef MYRAMATH_MULTIFRONTAL_SPARSEZCHOLESKYSOLVER_H
#define MYRAMATH_MULTIFRONTAL_SPARSEZCHOLESKYSOLVER_H

#include <myramath/MYRAMATH_EXPORT.h>

// For ReflectNumber<>
#include <myramath/utility/Number.h>

// Return type for all _jobgraph() methods.
#include <myramath/jobgraph/JobGraph.h>

// Options pack.
#include <myramath/multifrontal/Options.h>

// Underlying numeric storage.
#include <myramath/multifrontal/zcholesky/Kernel.h>
#include <myramath/multifrontal/detail/llt/LContainer.h>

// Permutation type.
#include <vector>

namespace myra {

// Forward declarations, components.
class AssemblyTree;
template <class Precision> class ZCholeskyKernel;

// Forward declarations, A type.
class Permutation;
template<class Number> class SparseMatrix;
template<class Number> class SparseMatrixRange;
template<class Number> class CSparseMatrixRange;

// Forward declarations, X/B types.
class intCRange;
template <class Number> class Matrix;
template <class Number> class MatrixRange;
template <class Number> class CMatrixRange;
template <class Number> class Vector;
template <class Number> class VectorRange;
template <class Number> class CVectorRange;
template <class Number> class LowerMatrix;
template <class Number> class LowerMatrixRange;
template <class Number> class CLowerMatrixRange;

// Forward declarations, serialization.
class InputStream;
class OutputStream;

template<class Precision> class MYRAMATH_EXPORT SparseZCholeskySolver
  {
  public:

    // Useful typedefs for various dense/sparse ranges.
    typedef std::complex<Precision> Number;
    typedef MatrixRange<Number> DRange;        // Dense range type (e.g. right hand sides)
    typedef LowerMatrixRange<Number> LRange;   // Lower range type (e.g. hermitian schur complement)
    typedef CSparseMatrixRange<Number> SRange; // Sparse range type (e.g. underlying A)
    typedef intCRange iRange;                  // Indices range type (e.g. stencils for partial solve etc)

    // Typedef for Options pack.
    typedef ::myra::multifrontal::Options Options;

    // ----------------------------------------- Construction, serialization, value semantics.

    SparseZCholeskySolver();

    SparseZCholeskySolver(const SparseZCholeskySolver& that);

    void swap(SparseZCholeskySolver& that);

#ifdef MYRAMATH_ENABLE_CPP11
    SparseZCholeskySolver(SparseZCholeskySolver&& that);
#endif

    SparseZCholeskySolver& operator = (SparseZCholeskySolver that);

    explicit SparseZCholeskySolver(InputStream& in);

    void write(OutputStream& out) const;

   ~SparseZCholeskySolver(); 

    // ----------------------------------------- Factorization and factorizing constructors.

    SparseZCholeskySolver(const SRange& A, Options options = defaults());
    void factor(const SRange& A, Options options = defaults());
    JobGraph factor_jobgraph(const SRange& A, Options options = defaults());

    SparseZCholeskySolver(const SRange& A, const Permutation& P, Options options = defaults());
    void factor(const SRange& A, const Permutation& P, Options options = defaults());
    JobGraph factor_jobgraph(const SRange& A, const Permutation& P, Options options = defaults());

    SparseZCholeskySolver(const SRange& A, const AssemblyTree& tree, Options options = defaults());
    void factor(const SRange& A, const AssemblyTree& tree, Options options = defaults());
    JobGraph factor_jobgraph(const SRange& A, const AssemblyTree& tree, Options options = defaults());

    // ----------------------------------------- Linear solution.

    //    side  = Solve by A from the 'L'eft or from the 'R'ight?
    //    op    = Apply an operation to A? ('T'ranspose, 'H'ermitian, 'C'onjugate or 'N'othing)
    void solve(const DRange& B, char side = 'L', char op = 'N', Options options = defaults().set_nthreads(1)) const;
    JobGraph solve_jobgraph(const DRange& B, char side = 'L', char op = 'N', Options options = defaults().set_nthreads(1)) const;    

    //    side  = Solve by L from the 'L'eft or from the 'R'ight?
    //    op    = Apply an operation to L? ('T'ranspose, 'H'ermitian, 'C'onjugate or 'N'othing)  
    void solveL(const DRange& B, char side = 'L', char op = 'N', Options options = defaults().set_nthreads(1)) const;
    JobGraph solveL_jobgraph(const DRange& B, char side = 'L', char op = 'N', Options options = defaults().set_nthreads(1)) const;    

    // ----------------------------------------- Calculating schur complements.

    LowerMatrix<Number> schur(const SRange& B, Options options = defaults()) const;
    void schur_inplace(const SRange& B, const LRange& S, Options options = defaults()) const;
    JobGraph schur_jobgraph(const SRange& B, const LRange& S, Options options = defaults()) const;

    LowerMatrix<Number> schur(const iRange& Bi, const DRange& Bv, Options options = defaults()) const;    
    void schur_inplace(const iRange& Bi, const DRange& Bv, const LRange& S, Options options = defaults()) const;
    JobGraph schur_jobgraph(const iRange& Bi, const DRange& Bv, const LRange& S, Options options = defaults()) const;

    Matrix<Number> schur(const SRange& B, const SRange& C, Options options = defaults()) const;
    void schur_inplace(const SRange& B, const SRange& C, const DRange& S, Options options = defaults()) const;
    JobGraph schur_jobgraph(const SRange& B, const SRange& C, const DRange& S, Options options = defaults()) const;

    Matrix<Number> schur(const iRange& Bi, const DRange& Bv, const iRange& Ci, const DRange& Cv, Options options = defaults()) const;    
    void schur_inplace(const iRange& Bi, const DRange& Bv, const iRange& Ci, const DRange& Cv, const DRange& S, Options options = defaults()) const;
    JobGraph schur_jobgraph(const iRange& Bi, const DRange& Bv, const iRange& Ci, const DRange& Cv, const DRange& S, Options options = defaults()) const;

    // ----------------------------------------- Sampling inv(A).

    Number inverse(int ij) const;

    Number inverse(int i, int j) const;

    Matrix<Number> inverse(const iRange& i, const iRange& j, Options options = defaults()) const;
    void inverse_inplace(const iRange& i, const iRange& j, const DRange& Z, Options options = defaults()) const;
    JobGraph inverse_jobgraph(const iRange& i, const iRange& j, const DRange& Z, Options options = defaults()) const;

    LowerMatrix<Number> inverse(const iRange& ij, Options options = defaults()) const;
    void inverse_inplace(const iRange& ij, const LRange& Z, Options options = defaults()) const;
    JobGraph inverse_jobgraph(const iRange& ij, const LRange& Z, Options options = defaults()) const;

    // ----------------------------------------- Partial linear solution.

    // Note X = partialsolve(linspace(0,N),linspace(0,N),B,side,op) yields the same result X = solve(B,side,op)
    // Note X = partialsolve(i,j,B,'L'eft, op) yields the same result as X =   op(this->inverse(i,j))*B
    // Note X = partialsolve(i,j,B,'R'ight,op) yields the same result as X = B*op(this->inverse(i,j))
    Matrix<Number> partialsolve(const iRange& i, const iRange& j, const CMatrixRange<Number>& B, char side = 'L', char op = 'N', Options options = defaults().set_nthreads(1)) const;
    void partialsolve_inplace(const iRange& i, const iRange& j, const CMatrixRange<Number>& B, const MatrixRange<Number>& X, char side = 'L', char op = 'N', Options options = defaults().set_nthreads(1)) const;
    JobGraph partialsolve_jobgraph(const iRange& i, const iRange& j, const CMatrixRange<Number>& B, const MatrixRange<Number>& X, char side = 'L', char op = 'N', Options options = defaults().set_nthreads(1)) const;

    // ----------------------------------------- Miscellany.

    int size() const;

    const AssemblyTree& tree() const;

    static Options defaults();
    
  private:

    // Numeric contents.    
    typedef ZCholeskyKernel<Precision> Kernel;
    typedef ::myra::multifrontal::detail::llt::LContainer<Kernel> LContainer;
    LContainer L;
    
  };

template<class Precision> class ReflectNumber <SparseZCholeskySolver<Precision> >
  { public: typedef std::complex<Precision> type; };

MYRAMATH_EXPORT Vector<NumberC> operator * (const SparseZCholeskySolver<float >& solver, const CVectorRange<NumberC>& x);
MYRAMATH_EXPORT Vector<NumberZ> operator * (const SparseZCholeskySolver<double>& solver, const CVectorRange<NumberZ>& x);

MYRAMATH_EXPORT Matrix<NumberC> operator * (const SparseZCholeskySolver<float >& solver, const CMatrixRange<NumberC>& X);
MYRAMATH_EXPORT Matrix<NumberZ> operator * (const SparseZCholeskySolver<double>& solver, const CMatrixRange<NumberZ>& X);

} // namespace

#endif