GLpApp_GLpYUEpetraDataPool.hpp

Go to the documentation of this file.

#ifndef GLPAPP_GLPYUEPETRADATAPOOL_H
#define GLPAPP_GLPYUEPETRADATAPOOL_H

//#include "Epetra_config.h"

#include <iostream>

#include "Epetra_Map.h"
#include "Epetra_MultiVector.h"
#include "Epetra_Vector.h"
#include "Epetra_Import.h"
#include "Epetra_Export.h"
#include "Epetra_CrsMatrix.h"
#include "Epetra_FECrsMatrix.h"
#include "Epetra_LinearProblem.h"
#include "Epetra_LAPACK.h"
#include "Epetra_FEVector.h"
#include "Epetra_IntSerialDenseVector.h"
#include "Epetra_SerialDenseMatrix.h"
#include "Epetra_SerialDenseVector.h"
#include "GenSQP_DataPool.hpp"
#include "GenSQP_YUEpetraVector.hpp"
#include "Epetra_SerialDenseMatrix.h"
#include "Epetra_SerialDenseVector.h"

#ifdef HAVE_MPI
#  include "Epetra_MpiComm.h"
#else
#  include "Epetra_SerialComm.h"
#endif

namespace GLpApp {

class GLpYUEpetraDataPool : public GenSQP::DataPool
{
public:

  GLpYUEpetraDataPool(
    Teuchos::RefCountPtr<const Epetra_Comm>    const& commptr
    ,const double                              beta
    ,const double                              len_x     // Ignored if myfile is *not* empty
    ,const double                              len_y     // Ignored if myfile is *not* empty
    ,const int                                 local_nx  // Ignored if myfile is *not* empty
    ,const int                                 local_ny  // Ignored if myfile is *not* empty
    ,const char                                myfile[]
    ,const bool                                trace
    );

  void computeAll( const GenSQP::Vector &x );

  int  solveAugsys( const Teuchos::RefCountPtr<const Epetra_MultiVector> & rhsy,
                    const Teuchos::RefCountPtr<const Epetra_MultiVector> & rhsu,
                    const Teuchos::RefCountPtr<const Epetra_MultiVector> & rhsp,
                    const Teuchos::RefCountPtr<Epetra_MultiVector> & y,
                    const Teuchos::RefCountPtr<Epetra_MultiVector> & u,
                    const Teuchos::RefCountPtr<Epetra_MultiVector> & p,
                    double tol );

  Teuchos::RefCountPtr<const Epetra_Comm> getCommPtr();

  Teuchos::RefCountPtr<Epetra_FECrsMatrix> getA();
  Teuchos::RefCountPtr<Epetra_FECrsMatrix> getB();
  Teuchos::RefCountPtr<Epetra_FECrsMatrix> getH();
  Teuchos::RefCountPtr<Epetra_FECrsMatrix> getR();
  Teuchos::RefCountPtr<Epetra_CrsMatrix> getAugmat();
  Teuchos::RefCountPtr<Epetra_FECrsMatrix> getNpy();

  Teuchos::RefCountPtr<Epetra_FEVector> getb();
  Teuchos::RefCountPtr<Epetra_FEVector> getq();
  Teuchos::RefCountPtr<Epetra_FEVector> getNy();

  void computeNy(const Teuchos::RefCountPtr<const Epetra_MultiVector> & y);

  void computeNpy(const Teuchos::RefCountPtr<const Epetra_MultiVector> & y);

  void computeAugmat();
  
  Teuchos::RefCountPtr<const Epetra_SerialDenseMatrix> getipcoords();
  Teuchos::RefCountPtr<const Epetra_IntSerialDenseVector> getipindx();
  Teuchos::RefCountPtr<const Epetra_SerialDenseMatrix> getpcoords();
  Teuchos::RefCountPtr<const Epetra_IntSerialDenseVector> getpindx();
  Teuchos::RefCountPtr<const Epetra_IntSerialDenseMatrix> gett();
  Teuchos::RefCountPtr<const Epetra_IntSerialDenseMatrix> gete();

  double getbeta();
  
  void PrintVec( const Teuchos::RefCountPtr<const Epetra_Vector> & x );

private:

  Teuchos::RefCountPtr<const Epetra_Comm> commptr_;
          
  Teuchos::RefCountPtr<Epetra_SerialDenseMatrix> ipcoords_;
  Teuchos::RefCountPtr<Epetra_IntSerialDenseVector> ipindx_;
  Teuchos::RefCountPtr<Epetra_SerialDenseMatrix> pcoords_;
  Teuchos::RefCountPtr<Epetra_IntSerialDenseVector> pindx_;
  Teuchos::RefCountPtr<Epetra_IntSerialDenseMatrix> t_;
  Teuchos::RefCountPtr<Epetra_IntSerialDenseMatrix> e_;

  Teuchos::RefCountPtr<Epetra_FECrsMatrix> A_;
  Teuchos::RefCountPtr<Epetra_FECrsMatrix> B_;
  Teuchos::RefCountPtr<Epetra_FECrsMatrix> H_;
  Teuchos::RefCountPtr<Epetra_FECrsMatrix> R_;

  Teuchos::RefCountPtr<Epetra_MultiVector> B_bar_;

  Teuchos::RefCountPtr<Epetra_CrsMatrix> Augmat_;

  Teuchos::RefCountPtr<Epetra_FECrsMatrix> Npy_;

  Teuchos::RefCountPtr<Epetra_FEVector> b_;
  Teuchos::RefCountPtr<Epetra_FEVector> q_;

  Teuchos::RefCountPtr<Epetra_FEVector> Ny_;

  double beta_;

};

class Usr_Par {
public:

  Usr_Par();

  Epetra_SerialDenseMatrix Nodes;
  Epetra_SerialDenseVector Weights;

  Epetra_SerialDenseMatrix N;

  Epetra_SerialDenseMatrix Nx1;

  Epetra_SerialDenseMatrix Nx2;

  Epetra_SerialDenseMatrix S1;
  Epetra_SerialDenseMatrix S2;
  Epetra_SerialDenseMatrix S3;

  Epetra_SerialDenseVector Nw;

  Epetra_SerialDenseMatrix NNw;

  Epetra_SerialDenseMatrix * NNNw;

  Epetra_SerialDenseMatrix * NdNdx1Nw;

  Epetra_SerialDenseMatrix * NdNdx2Nw;

  ~Usr_Par() {
    delete [] NNNw;
    delete [] NdNdx1Nw;
    delete [] NdNdx2Nw;
  }
  
  void Print(ostream& os) const;
};

} // namespace GLpApp

#endif // GLPAPP_GLPYUEPETRADATAPOOL_H