gmSparseMatrixLayoutBuilder.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_SPARSE_MATRIX_LAYOUT_BUILDER_H_
#define _GEMA_SPARSE_MATRIX_LAYOUT_BUILDER_H_
 
#include "gmCoreConfig.h"
 
#include "gmSparseMatrixLayout.h"
 
#include <QSet>
#include <QPair>
 
class GmElementMesh;
 
class GMC_API_EXPORT GmSparseMatrixLayoutBuilder
{
public:
  virtual ~GmSparseMatrixLayoutBuilder() {}
 
  virtual bool begin(int n, int expectedNnz) = 0;
 
  virtual bool addEntry(int row, int col) = 0;
 
  virtual bool end() = 0;
 
  static int nnzEstimative(const GmElementMesh* mesh, int maxDofsPerNode);
};
 
template <class IndexType, GmSparseMatrixLayoutTypes T> class GmHashCSxSparseMatrixLayoutBuilder: public GmSparseMatrixLayoutBuilder
{
public:
  GmHashCSxSparseMatrixLayoutBuilder(GmCSxSparseMatrixLayout<IndexType, T>* layout);
 
  virtual bool begin(int n, int expectedNnz);
  virtual bool addEntry(int row, int col);
  virtual bool end();
 
private:
  Q_DISABLE_COPY(GmHashCSxSparseMatrixLayoutBuilder);
 
  GmCSxSparseMatrixLayout<IndexType, T>*  _layout;       
  QSet<QPair<int, int> >                  _nodeNodeMap;  
  int                                     _expectedNnz;  
};
 
template <class IndexType, GmSparseMatrixLayoutTypes T> class GmVectorCSxSparseMatrixLayoutBuilder: public GmSparseMatrixLayoutBuilder
{
public:
  GmVectorCSxSparseMatrixLayoutBuilder(GmCSxSparseMatrixLayout<IndexType, T>* layout);
 
  virtual ~GmVectorCSxSparseMatrixLayoutBuilder();
 
  virtual bool begin(int n, int expectedNnz);
  virtual bool addEntry(int row, int col);
  virtual bool end();
 
private:
  Q_DISABLE_COPY(GmVectorCSxSparseMatrixLayoutBuilder);
 
  void clear();
  bool insertIfNew(int* p, int size, int value, int ptrIndex);
 
  struct ExtData
  {
    int  _nalloc; 
    int* _data;   
 
    bool allocData(int size, int* initData, int initSize);
  };
 
  GmCSxSparseMatrixLayout<IndexType, T>*  _layout;    
  int       _nnz;         
  int       _n;           
  bool      _hasExtData;  
  const int _extDataSize; 
  
  int* _dataVec;    
 
#ifdef ENABLE_SPARSE_MEMORY_DEBUG  
  void entryAdded(int addedSize);
 
  int  _memAloc;    
  int  _numEntries; 
#endif
};
 
 
#endif