feorthosystem.cpp

Go to the documentation of this file.

#include "feorthosystem.h"


FEOrthoSystem::FEOrthoSystem(FEOrthoMesh &fe,unsigned multiplicity) 
  :_fes(multiplicity,&fe)
{
  assert(multiplicity > 0);
  this->_fe_dof_offsets.resize(_fes.size());


  //_local_dof_offset[i] means the offset where the dofs
  //of the  element i in the
  //vector of finite element _fes begin to be counted.
  //If _local_dof_offset[2]=7 means that the 3 element in _fes
  //has its first dof counted starting from 7
  _fe_dof_offsets[0]=0;
  for (unsigned i=1;i<_fes.size();i++)
  {
    _fe_dof_offsets[i]=_fe_dof_offsets[i-1]+_fes[i-1]->n_local_dofs();
  }

  //Now get the number of local dofs in the FE System
  //and the number of fields
  unsigned total_dofs=0;
  _n_fields=0;
  for (unsigned i=0;i<_fes.size();i++)
  {
    _n_fields+=_fes[i]->n_fields();
    total_dofs+=_fes[i]->n_local_dofs();
  }
  //Reset the global_map vector with the total dofs in
  //the system
  _global_map.resize(total_dofs);

  //Create the dof to field number map
  _dof_field_map.resize(total_dofs);
  unsigned field_off=0;
  for (unsigned i=0;i<_fes.size();i++)
  {
    unsigned n_local_dofs = _fes[i]->n_local_dofs();
    for (unsigned dofs=0;dofs<n_local_dofs;dofs++)
    {
      _dof_field_map[field_off+dofs]=_fes[i]->local_dof_field(dofs);
    }
    field_off+=_fes[i]->n_fields();
  }
}






 void FEOrthoSystem::reinit(OrthoMesh &mesh) 
{
  //Do reinit for all its finite elements
  std::vector<FEOrthoMesh*>::iterator it;
  for ( it= _fes.begin();it!=_fes.end();it++)
  {
    (*it)->reinit(mesh);
  }
}



const VecIndex& FEOrthoSystem::get_global_map(OrthoMesh::Cell_It &cell) 
{
  VecIndex::iterator it = _global_map.begin();
  for (unsigned feI=0;feI<_fes.size();feI++)
  {
    const VecIndex &fe_global_map=_fes[feI]->get_global_map(cell);
    unsigned offset = _fe_dof_offsets[feI];
    for (unsigned i=0;i<fe_global_map.size();i++)
    {
      *it++=offset+fe_global_map[i];
    }
  }
  assert(it==_global_map.end());
  return _global_map;

}




 Index FEOrthoSystem::local_dof_field(Index dof) 
{
  assert(dof < _dof_field_map.size());
  return _dof_field_map[dof];
  
}

---