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