Albany_GenericSTKMeshStruct.hpp

Go to the documentation of this file.

//*****************************************************************//
//    Albany 2.0:  Copyright 2012 Sandia Corporation               //
//    This Software is released under the BSD license detailed     //
//    in the file "license.txt" in the top-level Albany directory  //
//*****************************************************************//

#ifndef ALBANY_GENERICSTKMESHSTRUCT_HPP
#define ALBANY_GENERICSTKMESHSTRUCT_HPP

#include "Albany_AbstractSTKMeshStruct.hpp"
#include "Teuchos_ParameterList.hpp"
#include "Epetra_Comm.h"

// Refinement
#ifdef ALBANY_STK_PERCEPT
#include <stk_percept/PerceptMesh.hpp>
#include <stk_adapt/UniformRefinerPattern.hpp>
#endif


namespace Albany {


  class GenericSTKMeshStruct : public AbstractSTKMeshStruct {

    public:
    virtual void setFieldAndBulkData(
                  const Teuchos::RCP<const Epetra_Comm>& comm,
                  const Teuchos::RCP<Teuchos::ParameterList>& params,
                  const unsigned int neq_,
                  const AbstractFieldContainer::FieldContainerRequirements& req,
                  const Teuchos::RCP<Albany::StateInfoStruct>& sis,
                  const unsigned int worksetSize) = 0;

    Teuchos::ArrayRCP<Teuchos::RCP<Albany::MeshSpecsStruct> >& getMeshSpecs();

#ifdef ALBANY_STK_PERCEPT
    Teuchos::RCP<stk::percept::PerceptMesh> getPerceptMesh(){ return eMesh; }
    Teuchos::RCP<stk::adapt::UniformRefinerPatternBase> getRefinerPattern(){ return refinerPattern; }
#endif

    void rebalanceAdaptedMesh(const Teuchos::RCP<Teuchos::ParameterList>& params,
                              const Teuchos::RCP<const Epetra_Comm>& comm);

    bool useCompositeTet(){ return compositeTet; }

    void setupMeshBlkInfo();

    const Albany::DynamicDataArray<Albany::CellSpecs>::type& getMeshDynamicData() const
        { return meshDynamicData; }

    protected: 
    GenericSTKMeshStruct(
                  const Teuchos::RCP<Teuchos::ParameterList>& params,
                  const Teuchos::RCP<Teuchos::ParameterList>& adaptParams,
                  const int numDim=-1);

    void SetupFieldData(
                  const Teuchos::RCP<const Epetra_Comm>& comm,
                  const int neq_,
                  const AbstractFieldContainer::FieldContainerRequirements& req,
                  const Teuchos::RCP<Albany::StateInfoStruct>& sis,
                  const int worksetSize_);

    bool buildUniformRefiner();

    bool buildLocalRefiner();

    void printParts(stk::mesh::fem::FEMMetaData *metaData);

    void cullSubsetParts(std::vector<std::string>& ssNames,
        std::map<std::string, stk::mesh::Part*>& partVec);

    int computeWorksetSize(const int worksetSizeMax, const int ebSizeMax) const;

    void rebalanceInitialMesh(const Teuchos::RCP<const Epetra_Comm>& comm);

    bool buildEMesh;
    bool buildPerceptEMesh();

    void uniformRefineMesh(const Teuchos::RCP<const Epetra_Comm>& comm);

    void checkInput(std::string option, std::string value, std::string allowed_values);

    void computeAddlConnectivity();

    ~GenericSTKMeshStruct();

    Teuchos::RCP<Teuchos::ParameterList> getValidGenericSTKParameters(
         std::string listname = "Discretization Param Names") const;

    Teuchos::RCP<Teuchos::ParameterList> params;

    Teuchos::RCP<Teuchos::ParameterList> adaptParams;

    Teuchos::ArrayRCP<Teuchos::RCP<Albany::MeshSpecsStruct> > meshSpecs;

    // Information that changes when the mesh adapts
    Albany::DynamicDataArray<CellSpecs>::type meshDynamicData;

#ifdef ALBANY_STK_PERCEPT
    Teuchos::RCP<stk::percept::PerceptMesh> eMesh;
    Teuchos::RCP<stk::adapt::UniformRefinerPatternBase> refinerPattern;
#endif

    bool uniformRefinementInitialized;

    bool compositeTet;

  };

}

#endif