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 <QList>
#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 QList<int>& dofList = QList<int>());
  GmElementDof(int numNodes,    const QList<int>& dofList = QList<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];
  }

  QList<int> nodeDofList(int node) const;

  GmElementDofMap elementDofMap() const;

  void setNodeDof(int node, const QList<int>& dofList);
  void addNode(const QList<int>& dofList);

private:
  Q_DISABLE_COPY(GmElementDof)

  void init(int numNodes, const QList<int>& dofList);

  GmElementDofMap dofMapFromList(const QList<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