pztrnsform.h

Go to the documentation of this file.

#ifndef TTRANSFORMH
#define TTRANSFORMH

#include "pzfmatrix.h"

#ifdef _AUTODIFF
#include "fad.h"
#endif

template<class T>
class TPZVec;

template<class T=REAL>
class TPZTransform : public TPZSavable {
                int fRow;
                int fCol;
                TPZFNMatrix<9,T> fMult;
                TPZFNMatrix<3,T> fSum;
                // dynamic memory allocation
public:
                TPZTransform(int dim);
                TPZTransform();
                TPZTransform(int fRow,int fCol);
                TPZTransform(const TPZTransform<T> &tr);
                ~TPZTransform();
    
//    /** @brief Rows of transformation matrix */
//    int Rows();
//    
//    /** @brief Cols of transformation matrix */
//    int Cols();
                
    friend class TPZTransform<REAL>;
    
#ifdef _AUTODIFF
    friend class TPZTransform<Fad<REAL> >;
#endif
    
                TPZTransform<T> &operator=(const TPZTransform<T> &t);
                
    void CopyFrom(const TPZTransform<REAL> &cp);
    
                const TPZFMatrix<T>  & Mult() const {return fMult;}
                
                const TPZFMatrix<T>  & Sum() const {return fSum;}
                
                TPZFMatrix<T>  & Mult() {return fMult;}
                
                TPZFMatrix<T>  & Sum()  {return fSum;}
                
                void SetMatrix(TPZFMatrix<T> &mult,TPZFMatrix<T> &sum);
                
                TPZTransform<T> Multiply(TPZTransform<T> &right);
                
                void Apply(TPZVec<T> &vectorin,TPZVec<T> &vectorout);
                
                void PrintInputForm(std::ostream &out);
                
                int CompareTransform(TPZTransform<T> &t,REAL tol = 1.e-6);
                
        int ClassId() const override{
            return Hash("TPZTransform");
        }
        
        void Read(TPZStream& buf, void* context) override { //ok
            buf.Read(&fRow);
            buf.Read(&fCol);
            fMult.Read(buf,context);
            fSum.Read(buf,context);
        }
        
        void Write(TPZStream& buf, int withclassid) const override{ //ok
            buf.Write(&fRow);
            buf.Write(&fCol);
            fMult.Write(buf,withclassid);
            fSum.Write(buf,withclassid);
        }
                
};

template<class T>
inline std::ostream &operator<<(std::ostream &out, const TPZTransform<T> &tr)
{
                out << "mult = " << tr.Mult() << " sum = " << tr.Sum();
                return out;
}

#endif