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,      
  //---------------------------------------
  // No changing bellow 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