snl_fei_tester.hpp

/*--------------------------------------------------------------------*/
/*    Copyright 2005 Sandia Corporation.                              */
/*    Under the terms of Contract DE-AC04-94AL85000, there is a       */
/*    non-exclusive license for use of this work by or on behalf      */
/*    of the U.S. Government.  Export of this program may require     */
/*    a license from the United States Government.                    */
/*--------------------------------------------------------------------*/

#ifndef _snl_fei_tester_h_
#define _snl_fei_tester_h_

#include <fei_macros.hpp>
#include <fei_mpi.h>
#include <fei_SharedPtr.hpp>

#include <test_utils/feitester.hpp>
#include <test_utils/DataReader.hpp>

#include <fei_fwd.hpp>

class snl_fei_tester : public feitester {
 public:
  snl_fei_tester(fei::SharedPtr<DataReader> data_reader,
     MPI_Comm comm, int localProc, int numProcs);
  ~snl_fei_tester();

  const char* getName()
    {
      static const char name[] = "snl_fei_tester";
      return((const char*)name);
    }

  int testInitialization();

  int testLoading();

  int testSolve();

  int testCheckResult();

  void dumpMatrixFiles();

  void setParameter(const char* param);

 private:
  void defineFieldsAndIDTypes();
  int initElemBlocks();
  int loadElemBlocks();
  int initConstraints();
  int loadConstraints();
  void definePattern(ElemBlock& eb, int& patternID);
  int createLibraryInstance(const char* solverName);

  int save_block_node_soln(DataReader& data, fei::Vector* vec,
         const char* solnFileName, int numProcs,
         int localProc, int solveCounter);

  int save_block_elem_soln(DataReader& data, fei::Vector* vec,
         const char* solnFileName,
         int numProcs, int localProc, int solveCounter);

  int save_multiplier_soln(DataReader& data, fei::Vector* vec,
         const char* solnFileName,
         int numProcs, int localProc, int solveCounter);

  int checkSolution(int localProc, int numProcs,
      const char* solnFileName, const char* checkFileName,
      const char* extension, int solveCounter);

  MPI_Comm comm_;

  fei::SharedPtr<fei::Factory> factory_;

  fei::SharedPtr<fei::VectorSpace> vecSpace_;
  fei::SharedPtr<fei::MatrixGraph> matrixGraph_;

  fei::SharedPtr<fei::Matrix> A_;
  fei::SharedPtr<fei::Vector> x_;
  fei::SharedPtr<fei::Vector> b_;

  fei::SharedPtr<fei::LinearSystem> linSys_;

  LinearSystemCore* linSysCore_;
  FiniteElementData* feData_;

  fei::SharedPtr<DataReader> data_;

  std::vector<int> idTypes_;
  int numPatterns_;
  int nodeTypeOffset_, elemTypeOffset_, constraintTypeOffset_;

  int localProc_, numProcs_;
};

#endif // _snl_fei_tester_h_