tpzdohrsubstruct.h

Go to the documentation of this file.

#ifndef TPZDOHRSUBSTRUCT_H
#define TPZDOHRSUBSTRUCT_H

#include "pzfmatrix.h"
#include "tpzautopointer.h"
#include "pzstepsolver.h"

#ifndef MAKEINTERNAL
#define MAKEINTERNAL
#endif

template<class TVar>
class TPZDohrSubstruct : public TPZSavable {
                // @TODO implement the interface to make the substruct class actually saveable
public:
    TPZDohrSubstruct();
                
    ~TPZDohrSubstruct();
    
    enum EWeightType {CorrectWeight, Uniform};
    
    int ClassId() const override  {
        return Hash("TPZDohrSubstruct") ^ ClassIdOrHash<TVar>() << 1;
    }
    
    static EWeightType fWeightType;
                
                void ReallocMatRed() {};

    void ContributeResidual(TPZFMatrix<TVar> &u, TPZFMatrix<TVar> &r);
                
    void ContributeTestV1(TPZFMatrix<TVar> &testV1, int NumCoarse);
    void LoadWeightedResidual(const TPZFMatrix<TVar> &r_global);
                
    void AdjustResidual(TPZFMatrix<TVar> &r_global);
    void Contribute_rc(TPZFMatrix<TVar> &rc);
    void Contribute_rc_local(TPZFMatrix<TVar> &residual_local, TPZFMatrix<TVar> &rc_local) const;
    void Contribute_Kc(TPZMatrix<TVar> &Kc, TPZVec<int> &coarseindex);
    void Contribute_v1(TPZFMatrix<TVar> &v1, TPZFMatrix<TVar> &invKc_rc);
    void Contribute_v1_local(TPZFMatrix<TVar> &v1_local, TPZFMatrix<TVar> &invKc_rc) const;
    void Contribute_v2(TPZFMatrix<TVar> &v2);
    void Contribute_v2_local(TPZFMatrix<TVar> &residual_local, TPZFMatrix<TVar> &v2_local);
    void Contribute_v3(TPZFMatrix<TVar> &v3, const TPZFMatrix<TVar> &r, TPZFMatrix<TVar> &v1Plusv2) const;
    void Contribute_v3_local(TPZFMatrix<TVar> &v3, TPZFMatrix<TVar> &v1Plusv2) const;
                
    void Print(std::ostream &out) const;
    void ContributeKU(const TVar alpha, const TPZFMatrix<TVar> &uglobal, TPZFMatrix<TVar> &z) const;
                
                void ContributeKULocal(const TVar alpha, const TPZFMatrix<TVar> &u, TPZFMatrix<TVar> &z) const;
    void ContributeGlobalDiagonal(TPZFMatrix<TVar> &StiffnessDiag);
    void ContributeDiagonalLocal(TPZFMatrix<TVar> &StiffnessDiag);
    void ComputeWeights(TPZFMatrix<TVar> &StiffnessDiag);
    void ComputeWeightsLocal(TPZFMatrix<TVar> &StiffnessDiagLocal);
    void Initialize();
    
    //Daqui pra baixo são funcões cujos nomes foram dados pelo Phil
    void GetCoarseResidual(TPZFMatrix<TVar> &rc, TPZVec<int> &indices);
    
    void GetCoarseStiffness(TPZMatrix<TVar> &stiff, TPZVec<int> &indices);
public:
    void PrepareSystems();
    void SolveSystemPhi();
    void SolveSystemZi();
    void ComputeCoarseStiffness();
                void AddInternalSolution(TPZFMatrix<TVar> &sol);
                
                TPZFMatrix<TVar> fC_star;
    TPZFMatrix<TVar> fKeC_star; //K_star_inv*C_star_trans
    TPZStepSolver<TVar> finv;
    TPZFMatrix<TVar> fNullPivots;

                int fNEquations; 
                
                int NumEquations() const
                {
                                return fNEquations;
                }
    TPZVec<int> fCoarseNodes;
                
    TPZFMatrix<TVar> fC; //C(i)
    TPZFMatrix<TVar> fPhiC; //Phí(i)
                TPZFMatrix<TVar> fPhiC_Weighted_Condensed;

    TPZVec<int> fCoarseIndex; //R(ci)
    TPZManVector<TVar> fWeights; //W(i) = diagonal(fWeights)
    TPZAutoPointer<TPZMatrix<TVar> > fStiffness; //K(i)
    TPZFMatrix<TVar> fKCi; //K(ci)
    TPZVec<int> fGlobalIndex; //R(i)
                
                
                TPZVec<std::pair<int,int> > fGlobalEqs;
    TPZStepSolver<TVar> fInvertedStiffness; //Ke, or K_star
    mutable TPZStepSolver<TVar> fInvertedInternalStiffness; //R(Ii)*K(i)*R(Ii)invertida
    TPZStepSolver<TVar> fKCInvert; //K(c) invertida
    TPZVec<int> fInternalEqs; //R(Ii)
                TPZVec<int> fBoundaryEqs;
    TPZFMatrix<TVar> fLocalLoad; //F(i) - Actually it's a vector
    TPZFMatrix<TVar> fLocalWeightedResidual; //W(i)*R(i)*r - Actually it's a vector
    TPZFMatrix<TVar> fzi; //z(i)
                
                TPZFMatrix<TVar> fAdjustSolution;
                
};

#endif