gmElementDof.h

Go to the documentation of this file.

/************************************************************************
**
** Copyright (C) 2014 by Carlos Augusto Teixera Mendes
** All rights reserved.
**
** This file is part of the "GeMA" software. It's use should respect
** the terms in the license agreement that can be found together 
** with this source code.
** It is provided AS IS, with NO WARRANTY OF ANY KIND, 
** INCLUDING THE WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR 
** A PARTICULAR PURPOSE.
**
************************************************************************/
 
#ifndef _GEMA_ELEMENT_DOF_H_
#define _GEMA_ELEMENT_DOF_H_
 
#include "gmCell.h"
 
typedef quint64 GmElementDofMap;
 
#define GM_MAX_DOF (8*sizeof(GmElementDofMap))
 
 
class GMC_API_EXPORT GmDofMap
{
public:
  virtual int numDof() const = 0;
 
  virtual int dof(int columnIndex) const = 0;
};
 
 
class GMC_API_EXPORT GmElementDof : public GmDofMap
{
public:
  GmElementDof(GmCellType type, const QVector<int>& dofList = QVector<int>());
  GmElementDof(int numNodes,    const QVector<int>& dofList = QVector<int>());
 
  ~GmElementDof();
 
  virtual int numDof() const { return _ndof; }
 
  int numNodes() const { return _nodeMaps.size(); }
 
  int node(int columnIndex) const { return _columnNodeList.at(columnIndex); }
 
  virtual int dof(int columnIndex) const { return _columnDofList.at(columnIndex); }
 
  GmElementDofMap nodeDofMap(int node) const
  {
    assert(node >= 0 && node < numNodes());
    return _nodeMaps[node];
  }
 
  QVector<int> nodeDofList(int node) const;
 
  GmElementDofMap elementDofMap() const;
 
  void setNodeDofUnsafe(int node, const QVector<int>& dofList);
  void setNodeDof(int node, const QVector<int>& dofList);
  void addNode(const QVector<int>& dofList);
  void addNodeUnsafe(const QVector<int>& dofList);
 
private:
  Q_DISABLE_COPY(GmElementDof)
 
  void init(int numNodes, const QVector<int>& dofList);
  GmElementDofMap dofMapFromList(const QVector<int>& dofList) const;
  
  int                      _ndof;           
  QVector<GmElementDofMap> _nodeMaps;       
  QVector<int>             _columnNodeList; 
  QVector<int>             _columnDofList;  
};
 
class GMC_API_EXPORT GmMatrixDof : public GmDofMap
{
public:
  GmMatrixDof() {}
 
  void clear() { _dofList.clear(); _nodeList.clear(); _globalDofMap.clear(); }
 
  void setDofList(const QVector<int>& dofList) { _dofList = dofList; }
 
  void addNode(int node) { _nodeList.append(node); }
 
  virtual int numDof() const { return _nodeList.size() * _dofList.size(); }
 
  int node(int columnIndex) const 
  { 
    assert(columnIndex >= 0 && columnIndex < numDof());
    return _nodeList[columnIndex / _dofList.size()];
  }
 
  virtual int dof(int columnIndex) const 
  { 
    assert(columnIndex >= 0 && columnIndex < numDof());
    return _dofList[columnIndex % _dofList.size()]; 
  }
 
  QVector<int>& globalDofMap() { return _globalDofMap; }
 
private:
  Q_DISABLE_COPY(GmMatrixDof)
 
  QVector<int> _nodeList; 
  QVector<int> _dofList;  
 
  QVector<int> _globalDofMap; 
};
 
 
#endif