DofMapper.hpp

/*------------------------------------------------------------------------*/
/*                 Copyright 2010 Sandia Corporation.                     */
/*  Under 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 stk_linsys_DofMapper_hpp
#define stk_linsys_DofMapper_hpp

#include <map>

#include <stk_util/parallel/Parallel.hpp>
#include <stk_mesh/base/Field.hpp>
#include <stk_mesh/base/BulkData.hpp>

#include <stk_linsys/FieldIdMap.hpp>
#include <stk_linsys/FeiBaseIncludes.hpp>

namespace stk_classic {
namespace linsys {

class DofMapper {
 public:
  DofMapper(MPI_Comm comm, bool create_reverse_mappings=true);

//  DofMapper(fei::SharedPtr<fei::VectorSpace> vspace, bool create_reverse_mappings=true);

  virtual ~DofMapper();

  void add_dof_mappings( const stk_classic::mesh::BulkData& mesh_bulk,
                         const stk_classic::mesh::Selector& selector,
                         stk_classic::mesh::EntityRank ent_type,
                         const stk_classic::mesh::FieldBase& field );

  void finalize();

  bool reverse_mappings_enabled() const { return m_reverse_mappings_enabled; }

  int get_field_id(const stk_classic::mesh::FieldBase& field) const;

  int get_global_index(stk_classic::mesh::EntityRank ent_type,
                       stk_classic::mesh::EntityId ent_id,
                       stk_classic::mesh::FieldBase& field,
                       int offset_into_field=0);

  void get_dof(int global_index,
               stk_classic::mesh::EntityRank& ent_type,
               stk_classic::mesh::EntityId& ent_id,
               const stk_classic::mesh::FieldBase*& field,
               int& offset_into_field) const;

  const fei::SharedPtr<fei::VectorSpace> get_fei_VectorSpace() const { return m_fei_vecspace; }

  fei::SharedPtr<fei::VectorSpace> get_fei_VectorSpace() { return m_fei_vecspace; }

  const FieldIdMap& get_FieldIdMap() const { return m_field_id_map; }

  FieldIdMap& get_FieldIdMap() { return m_field_id_map; }

 private:

  FieldIdMap m_field_id_map;

  //we store the fei::VectorSpace in a fei::SharedPtr because other fei APIs expect it
  //to be that way. e.g., the constructor for fei::MatrixGraph requires a VectorSpace in a
  //fei::SharedPtr...
  fei::SharedPtr<fei::VectorSpace> m_fei_vecspace;
  bool m_reverse_mappings_enabled;
  fei::ReverseMapper* m_fei_reversemap;

  DofMapper(const DofMapper &);
  void operator = (const DofMapper &);
};//class DofMapper

}//namespace linsys
}//namespace stk_classic

#endif