gmpFemVectorSet.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_PLUGIN_FEM_VECTOR_SET_H_
#define _GEMA_PLUGIN_FEM_VECTOR_SET_H_
 
#include "gmpFemProcessConfig.h"
 
#include <gmVectorSet.h>
 
enum GmpFemVectorTypes
{
  GmpFemVector_Eq = 0,  
  GmpFemVector_Fe = 0,  
  GmpFemVector_Fi,      
  GmpFemVector_RHS,         
 
  //---------------------------------------
  // No changing  below this line
  //---------------------------------------
  GmpFemVector_NumTypes
};
 
class GMP_FEM_PROCESS_API_EXPORT GmpFemVectorSet : public GmVectorSet
{
public:
  GmpFemVectorSet();
  ~GmpFemVectorSet();
 
  enum GmpFemVectorMode
  {
    VECTOR_MATCH_MODE,  
    VECTOR_SINGLE_MODE, 
    VECTOR_SINGLE_DEBUG_MODE,  
  };
 
  void clear();
 
  bool initTypes  (GmpFemVectorMode mode, int numVectors, ...);
  bool adjustSizes(int globalNumDofs, int globalNumFixedDofs, int maxElemDofs);
 
  void addTransposedDofVector(GmpFemVectorTypes type);
  void addSavedDofVector     (GmpFemVectorTypes type);
 
  void clearGlobalVectors();
 
  GmVector& globalVector(GmpFemVectorTypes type) const
  {
    assert((_mode == VECTOR_MATCH_MODE && index(type) >= 0) || (_mode != VECTOR_MATCH_MODE && type == GmpFemVector_Eq));
    return _globalVectors[tindex(type)]._data;
  }
 
  GmVector& globalVectorFromIndex(int index) const
  {
    assert(_mode == VECTOR_MATCH_MODE && index >= 0 && index < _nVectors); 
    return _globalVectors[index]._data;
  }
 
  GmVector& globalTransposedDofVector(GmpFemVectorTypes type) const
  {
    assert((_mode == VECTOR_MATCH_MODE && index(type) >= 0) || (_mode != VECTOR_MATCH_MODE && type == GmpFemVector_Eq));
    return *(_globalVectors[tindex(type)]._transposed);
  }
 
  GmVector& globalTransposedDofVectorFromIndex(int index) const
  {
    assert(_mode == VECTOR_MATCH_MODE && index >= 0 && index < _nVectors);
    return *(_globalVectors[index]._transposed);
  }
 
  GmVector& globalSavedDofVector(GmpFemVectorTypes type) const
  {
    assert((_mode == VECTOR_MATCH_MODE && index(type) >= 0) || (_mode != VECTOR_MATCH_MODE && type == GmpFemVector_Eq));
    return _globalVectors[tindex(type)]._saved;
  }
 
  GmVector& globalSavedDofVectorFromIndex(int index) const
  {
    assert(_mode == VECTOR_MATCH_MODE && index >= 0 && index < _nVectors);
    return _globalVectors[index]._saved;
  }
 
  GmVector& globalDebugPrintVector(GmpFemVectorTypes type) const
  {
    assert(_mode != VECTOR_SINGLE_MODE && index(type) >= 0);
    return (_mode == VECTOR_MATCH_MODE) ? _globalVectors[index(type)]._data : _printVectors[index(type)];
  }
 
  GmVector& globalDebugPrintVectorFromIndex(int index) const
  {
    assert(_mode != VECTOR_SINGLE_MODE && index >= 0 && index < _nVectors);
    return (_mode == VECTOR_MATCH_MODE) ? _globalVectors[index]._data : _printVectors[index];
  }
 
  GmpFemVectorMode mode() const { return _mode; }
 
  GmpFemVectorTypes femTypeFromIndex(int index) const { return (GmpFemVectorTypes)typeFromIndex(index); }
 
 
protected:
 
  int tindex(GmpFemVectorTypes type) const { return (_mode == VECTOR_MATCH_MODE) ? index(type) : 0; }
 
  GmpFemVectorMode _mode; 
 
  struct GmpFemVectorTypeData
  {
    GmVector  _data;       
    GmVector* _transposed; 
    GmVector  _saved;      
    bool      _hasSaved;   
 
    GmpFemVectorTypeData() { _transposed = &_data; _hasSaved = false; }
 
    ~GmpFemVectorTypeData() { if(_transposed != &_data) delete _transposed; }
 
    void zeros() 
    { 
      _data.zeros(); 
      _saved.zeros();  
      if(_transposed != &_data)
        _transposed->zeros();
    }
  };
 
  GmpFemVectorTypeData* _globalVectors;
 
  GmVector* _printVectors;
};
 
 
#endif