gmpFemPhysicsCommonLuaMaterial.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_PHYSICS_COMMON_LUA_MATERIAL_H_
#define _GEMA_PLUGIN_FEM_PHYSICS_COMMON_LUA_MATERIAL_H_

#include "gmpFemPhysicsCommonMaterial.h"
#include "gmpFemPhysicsCommon.h"

#include <luaTable.h>

class GmPluginData;


class GMP_FEM_PROCESS_API_EXPORT GmpFemPhysicsCommonLuaMaterialImp
{
public:
  GmpFemPhysicsCommonLuaMaterialImp(const GmPluginData* pdata, GmpFemPhysicsCommonMaterial* mat)
    : _pdata(pdata), _mat(mat), _map(NULL)
  {
    assert(_pdata);
  }

  ~GmpFemPhysicsCommonLuaMaterialImp();

  // GmpFemPhysicsCommonMaterial methods
  // ----------------------------------------
  const QVariantMap* materialMetaDataMap();

  bool checkLoadedData(const GmElement* e) const;

  // Lua function calling methods and aux classes
  // ------------------------------------------------

  typedef QVariant (*GetFieldDataT)(QString field, const void* context, const GmElement* e, int ip, const GmVector* coord, const GmLogCategory& logger);

  typedef bool (*SetFieldDataT)(QString field, const void* context, const GmElement* e, int ip, const QVariant& v);

  enum CallDataType
  {
    CDT_PROPERTY,   
    CDT_NODEATTR,   
    CDT_GAUSSATTR,  
    CDT_GEOMETRY,   
    CDT_USER,       
    // ----  NO ADDS BELLOW -----
    CDT_NUMTYPES    
  };

  // The set of geometry fields
  enum GeometryFields
  {
    GF_ELEMENT_ID,  
    GF_ELEMENT,     
    GF_TYPE,        
    GF_IP,          
    GF_COORD,       
    GF_CLENGTH,     
    GF_CDIM,        
    GF_NODE_AC,     
    GF_NODE_COORD,  
    GF_N,           
  };

  struct CallDataFunctions
  {
    CallDataFunctions() : _subTableName(NULL), _getF(NULL), _setF(NULL), _context(NULL) {}

    const char*   _subTableName; 
    GetFieldDataT _getF;         
    SetFieldDataT _setF;         
    const void*   _context;      
  };

  class GMP_FEM_PROCESS_API_EXPORT CallData
  {
  public:
    CallData();

    LuaTable& inputTable()  { assert(_inTable.isValid());  return _inTable;  }
    LuaTable& outputTable() { assert(_outTable.isValid()); return _outTable; }
    bool call();
    void cleanup();
    static int indexMM(lua_State* L);

  private:
    friend class GmpFemPhysicsCommonLuaMaterialImp;

    QString          _fname;    
    LuaFunction*     _f;        
    LuaTable         _inTable;  
    LuaTable         _outTable; 
    const GmElement* _e;        
    int              _ip;       
    const GmVector*  _coord;    
    int              _luaStackTop;  

    const GmpFemPhysicsCommonLuaMaterialImp* _mat; 

    CallDataFunctions        _userF;    
    const CallDataFunctions* _subTabF[CDT_NUMTYPES]; //<! Pointers to the get/set data functions for each possible sub-table data types. NULL if the type is not available for this call
  };

  CallData* prepareUserMethodCall(QString fname, const GmElement* e, int ip, const GmVector* coord,
                                  const char* userSubTable = NULL, GetFieldDataT userGetF = NULL, 
                                  SetFieldDataT userSetF = NULL, const void* userContext = NULL) const;

  const GmPluginData* pluginData() const { return _pdata; }

  // Aux methods to make the life of derived classes easier
  // ------------------------------------------------
  static QVariant propertyAccessorValueToVariant  (GmCellAccessor*  ac, const GmElement* e, int ip, const GmVector* coord, const GmLogCategory& logger);
  static QVariant nodeAccessorValueToVariant      (GmValueAccessor* ac, GmValueAccessor* coordAc, const GmElement* e, int ip, const GmVector* coord, GmInterpolatorType interpType, QVariant interpParam, const GmLogCategory& logger);
  static QVariant gaussAccessorValueToVariant     (GmGaussAccessor* ac, const GmElement* e, int ip, const GmVector* coord, const GmLogCategory& logger);
  static void     setGaussAccessorValueFromVariant(GmGaussAccessor* ac, const GmElement* e, int ip, const QVariant& v);
  static bool     fillMatrixFromVariant           (GmMatrix& mat, int nlin, int ncol, const QVariant& v, QString& err);

private:
  void loadTableData(QString field, QMap<QString, int>& fieldMap, QVector<QPair<GmInterpolatorType, QVariant>>* interpData = NULL, bool savedGaussAttr = false);
  void loadEntryData(int index, const QVariantMap& data, GmpFemPhysicsCommon::ValueList& list, QMap<QString, int>& map);

  static QVariant getPropertyValue (QString field, const void* context, const GmElement* e, int ip, const GmVector* coord, const GmLogCategory& logger);
  static QVariant getNodeAttrValue (QString field, const void* context, const GmElement* e, int ip, const GmVector* coord, const GmLogCategory& logger);
  static QVariant getGaussAttrValue(QString field, const void* context, const GmElement* e, int ip, const GmVector* coord, const GmLogCategory& logger);
  static bool     setGaussAttrValue(QString field, const void* context, const GmElement* e, int ip, const QVariant& v);
  static QVariant getGeometryValue (QString field, const void* context, const GmElement* e, int ip, const GmVector* coord, const GmLogCategory& logger);

  static QVariant variantListToOnStackLuaTable(LuaEnv* env, const QVariantList& v);

  QList<GmpFemPhysicsCommon::ValueType> parseDimField     (const QVariant& v) const;
  GmpFemPhysicsCommon::RequiredIf       parseRequiredField(const QVariant& v) const;

  const GmPluginData*          _pdata;   
  GmpFemPhysicsCommonMaterial* _mat;     

  QVariantMap*          _map;     
  QMap<QString, int>    _propertyIndexMap;   
  QMap<QString, int>    _nodeAttrIndexMap;   
  QMap<QString, int>    _gaussAttrIndexMap;  

  QVector<QPair<GmInterpolatorType, QVariant>> _nodeAttrInterpMode; 

  CallDataFunctions     _propertyCallFunctions;  
  CallDataFunctions     _nodeAttrCallFunctions;  
  CallDataFunctions     _gaussAttrCallFunctions; 
  CallDataFunctions     _geometryCallFunctions;  

  mutable GmTLS<CallData,      true> _callData;                     
  mutable GmTLS<QSet<QString>, true> _fieldCacheData[CDT_NUMTYPES]; 

  static QMap<QString, int> _geometryFieldsMap; 
};



template <class Base>
class GmpFemPhysicsCommonLuaMaterial : public Base
{
public:
  GmpFemPhysicsCommonLuaMaterial(const GmPluginData* pdata, int typeIndex, QString typeName, const GmLogCategory& logger)
         : Base(typeIndex, typeName, logger), _d(new GmpFemPhysicsCommonLuaMaterialImp(pdata, this)) {}

  virtual ~GmpFemPhysicsCommonLuaMaterial() { delete _d; }

  virtual const QVariantMap* materialMetaDataMap() { return _d->materialMetaDataMap(); }

  virtual bool checkLoadedData(const GmElement* e) const { return _d->checkLoadedData(e); }

  typedef GmpFemPhysicsCommonLuaMaterialImp::CallData CallData;

  CallData* prepareUserMethodCall(QString fname, const GmElement* e, int ip, const GmVector* coord,
                                  const char* userSubTable = NULL, 
                                  GmpFemPhysicsCommonLuaMaterialImp::GetFieldDataT userGetF = NULL, 
                                  GmpFemPhysicsCommonLuaMaterialImp::SetFieldDataT userSetF = NULL,
                                  const void* userContext = NULL) const
  {
    return _d->prepareUserMethodCall(fname, e, ip, coord, userSubTable, userGetF, userSetF, userContext);
  }

  // Aux methods to make the life of derived classes easier
  // ------------------------------------------------
  static QVariant propertyAccessorValueToVariant(GmCellAccessor*  ac, const GmElement* e, int ip, const GmVector* coord, const GmLogCategory& logger)
  {
    return GmpFemPhysicsCommonLuaMaterialImp::propertyAccessorValueToVariant(ac, e, ip, coord, logger);
  }

  static QVariant nodeAccessorValueToVariant(GmValueAccessor* ac, GmValueAccessor* coordAc, const GmElement* e, int ip, const GmVector* coord, GmInterpolatorType interpType, QVariant interpParam, const GmLogCategory& logger)
  {
    return GmpFemPhysicsCommonLuaMaterialImp::nodeAccessorValueToVariant(ac, coordAc, e, ip, coord, interpType, interpParam, logger);
  }

  static QVariant gaussAccessorValueToVariant(GmGaussAccessor* ac, const GmElement* e, int ip, const GmVector* coord, const GmLogCategory& logger)
  {
    return GmpFemPhysicsCommonLuaMaterialImp::gaussAccessorValueToVariant(ac, e, ip, coord, logger);
  }

  static void setGaussAccessorValueFromVariant(GmGaussAccessor* ac, const GmElement* e, int ip, const QVariant& v)
  {
    GmpFemPhysicsCommonLuaMaterialImp::setGaussAccessorValueFromVariant(ac, e, ip, v);
  }

  static bool fillMatrixFromVariant(GmMatrix& mat, int nlin, int ncol, const QVariant& v, QString& err)
  {
    return GmpFemPhysicsCommonLuaMaterialImp::fillMatrixFromVariant(mat, nlin, ncol, v, err);
  }

protected:
  const GmPluginData* pluginData() const { return _d->pluginData(); }

  GmpFemPhysicsCommonLuaMaterialImp* _d; 
};

#endif