TPZPardisoControl.h

Go to the documentation of this file.

//
//  TPZPardisoControl.hpp
//  PZ
//
//  Created by Philippe Devloo on 5/5/16.
//
//

#ifndef TPZPardisoControl_hpp
#define TPZPardisoControl_hpp

#include "pz_config.h"

#ifdef USING_MKL

#include <stdio.h>

//#include "mkl_pardiso.h"
#include "pzmanvector.h"
#include "tpzautopointer.h"
#include "pzfmatrix.h"
template<class TVar>
class TPZSYsmpMatrix;

template<class TVar>
class TPZFYsmpMatrix;

// inspired by PardisoSolver by Armando Duarte
template<class TVar>
class TPZPardisoControl
{
public:
    enum MSystemType {ESymmetric, EHermitian, ENonSymmetric};
    
    enum MStructure {EStructureSymmetric, EStructureNonSymmetric};
    
    enum MProperty {EPositiveDefinite, EIndefinite};
    
    TPZPardisoControl();
    
public:
    TPZPardisoControl(MSystemType systemtype, MProperty prop);
    
    TPZPardisoControl(const TPZPardisoControl &copy);
    
    TPZPardisoControl &operator=(const TPZPardisoControl &copy);
    
    virtual ~TPZPardisoControl();
    
    // this method should only be called if the pardiso control is zero (non initialized)
    void SetMatrixType(MSystemType systemtype, MProperty prop);
    
    void SetMatrix(TPZFYsmpMatrix<TVar> *matrix)
    {
        fNonSymmetricSystem = matrix;
    }
    
    void SetMatrix(TPZSYsmpMatrix<TVar> *matrix)
    {
        fSymmetricSystem = matrix;
    }
    
    void Decompose();
    
    void Solve(TPZFMatrix<TVar> &rhs, TPZFMatrix<TVar> &sol) const;
    
protected:
    
    MSystemType fSystemType;
    
    MStructure fStructure;
    
    MProperty fProperty;
    
    // Solver internal data address pointers
    // 32-bit: int pt[64]; 64-bit: long long pt[64]
    // or void *pt[64] should be OK on both architectures
    // this datastructure should not be copied or duplicated, therefore the "autopointer" protection
    
    TPZAutoPointer<TPZManVector<long long, 64> > fPardisoControl;
    
    // adress of the first element of pt;
    long long *fHandle;
    
    // Array used to pass parameters to Pardiso
    TPZManVector<long long, 64> fParam;
    
    // Maximum number of factors we will pass to the solver
    long long fMax_num_factors;
    
    // Factor number we are using
    long long fMatrix_num;
    
    // Message level information
    long long fMessageLevel;
    
    // error flag from Pardiso
    long long fError;
    
    TPZVec<long long> fPermutation;
    
    // matrix type, computed based on the structural information and TVar
    long long fMatrixType;
    
    long long MatrixType();
    
    TPZFYsmpMatrix<TVar> *fNonSymmetricSystem;
    
    TPZSYsmpMatrix<TVar> *fSymmetricSystem;
    
    void Error_check(int error) const;
    
};

#endif
#endif /* TPZPardisoControl_hpp */