gmModelData.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_MODEL_DATA_H_
#define _GEMA_MODEL_DATA_H_

#include "gmCoreConfig.h"

#include <QString>
#include <QMap>

class GmStateVar;
class GmPropertySet;
class GmMesh;
class GmUserFunction;
class GmBoundaryCondition;
class GmContactBoundaryCondition;
class GmSpatialIndex;
class GmDiscontinuitySet;
class GmPluginData;
class GmSimulationData;

class GmLogCategory;


class GMC_API_EXPORT GmModelData
{
public:
  GmModelData();
  ~GmModelData();

  void clear();

  const QMap<QString, GmStateVar*>& stateVars() const { return _stateVarMap;  }

  const QMap<QString, int>& stateVarGroupNames() const { return _stateVarGroupNames; }

  const QMap<QString, GmPropertySet*>& propertySets() const { return _propertySetMap; }

  const QMap<QString, GmMesh*>& meshes() const { return _meshMap; }

  const QMap<QString, GmUserFunction*>& userFunctions() const { return _userFunctionMap; }

  const QMap<QString, GmBoundaryCondition*>& boundaryConditions() const { return _bcMap; }

  const QMap<QString, GmContactBoundaryCondition*>& contactBoundaryConditions() const { return _cbcMap; }

  const QMap<QString, GmSpatialIndex*>& spatialIndices() const { return _spatialIndexMap; }

  const QMap<QString, GmDiscontinuitySet*>& discontinuitySets() const { return _discontinuitySetMap; }

  const QMap<QString, QMap<QString, GmPluginData*>>& pluginData() const { return _pluginDataMap; }

  const QMap<QString, GmPluginData*> pluginData(QString typeName) const { return _pluginDataMap.value(typeName); }

  bool addStateVar(GmStateVar* stateVar);

  int numStateVarDofs() const { return _nextDof; }

  bool addPropertySet(GmPropertySet* propSet) { return addObjectToMap<GmPropertySet>(propSet, _propertySetMap); }

  bool addMesh(GmMesh* mesh) { return addObjectToMap<GmMesh>(mesh, _meshMap); }

  bool addUserFunction(GmUserFunction* ufunc) { return addObjectToMap<GmUserFunction>(ufunc, _userFunctionMap); }

  bool addBoundaryCondition(GmBoundaryCondition* bc) { return addObjectToMap<GmBoundaryCondition>(bc, _bcMap); }

  bool addContactBoundaryCondition(GmContactBoundaryCondition* bc) { return addObjectToMap<GmContactBoundaryCondition>(bc, _cbcMap); }

  bool addSpatialIndex(GmSpatialIndex* index) { return addObjectToMap<GmSpatialIndex>(index, _spatialIndexMap); }

  bool addDiscontinuitySet(GmDiscontinuitySet* dset) { return addObjectToMap<GmDiscontinuitySet>(dset, _discontinuitySetMap); }

  bool addPluginData(GmPluginData* pdata);

  void setStateVarGroupName(GmStateVar* stateVar, QString name);

  void printInfo(const GmLogCategory& logger, const GmSimulationData* simData);

private:
  template <class T> bool addObjectToMap   (T* obj, QMap<QString, T*>& map);

  QMap<QString, GmStateVar*>          _stateVarMap;           
  QMap<QString, GmPropertySet*>       _propertySetMap;        
  QMap<QString, GmMesh*>              _meshMap;               
  QMap<QString, GmUserFunction*>      _userFunctionMap;       
  QMap<QString, GmBoundaryCondition*> _bcMap;                 
  QMap<QString, GmContactBoundaryCondition*> _cbcMap;         
  QMap<QString, GmSpatialIndex*>      _spatialIndexMap;       
  QMap<QString, GmDiscontinuitySet*>  _discontinuitySetMap;   
  QMap<QString, QMap<QString, GmPluginData*>> _pluginDataMap; 

  QMap<QString, int> _stateVarGroupNames; 

  int _nextDof; 
};

#endif