Home Research Papers Posters Manuals Talks Download People Help Internal

libs/ck-libs/fem/fem_interpolate.h

Go to the documentation of this file.

/* File: fem_interpolate.h
 * Authors: Terry Wilmarth, Nilesh Choudhury
 * 
 */

// This is an interface for specifying solution transfer operations
// for small mesh modification operations. It provides overridable
// defaults for node and element interpolations.  These functions are passed to
// FEM_add_element and FEM_add_node operations, along with a simple argument
// struct - TLW

#ifndef __CHARM_FEM_INTERPOLATE_H
#define __CHARM_FEM_INTERPOLATE_H

#include "charm-api.h"
#include "ckvector3d.h"
#include "fem.h"
#include "fem_mesh.h"

class femMeshModify;

class FEM_Interpolate {
  FEM_Mesh *theMesh;
  femMeshModify *theMod;
 public:
  typedef struct {
    int n;
    int nodes[8];
    int dim;
    int nNbrs;
    double coord[3];
    double frac;
    bool addNode;
  } NodalArgs;

  typedef struct {
    int e;
    int oldElement;
    int elType;
  } ElementArgs;

  typedef void (* FEM_InterpolateNodeFn)(NodalArgs, FEM_Mesh *);
  typedef void (* FEM_InterpolateElementFn)(ElementArgs);

  FEM_InterpolateNodeFn nodeEdgeFnPtr, nodeFaceFnPtr, nodeElementFnPtr, nodeCopyFnPtr;
  FEM_InterpolateElementFn elemCopyFnPtr, elemNodeFnPtr;

  FEM_Interpolate() {
    nodeEdgeFnPtr = nodeFaceFnPtr = nodeElementFnPtr = nodeCopyFnPtr = NULL;
    elemCopyFnPtr = elemNodeFnPtr = NULL;
  }

  FEM_Interpolate(FEM_Mesh *m) {
    nodeEdgeFnPtr = nodeFaceFnPtr = nodeElementFnPtr = nodeCopyFnPtr = NULL;
    elemCopyFnPtr = elemNodeFnPtr = NULL;
    theMesh = m;
  }

  FEM_Interpolate(FEM_Mesh *m, femMeshModify *fm) {
    nodeEdgeFnPtr = nodeFaceFnPtr = nodeElementFnPtr = nodeCopyFnPtr = NULL;
    elemCopyFnPtr = elemNodeFnPtr = NULL;
    theMesh = m;
    theMod = fm;
  }

  void FEM_InterpolateSetMesh(FEM_Mesh *m) { theMesh = m; }
  
  // Methods to set and reset interpolate functions on the fly; these will be
  // used to override defaults by Fortan code, and can also be set and reset
  // for using temporary special-purpose interpolation functions
  void FEM_SetInterpolateNodeEdgeFnPtr(FEM_InterpolateNodeFn fnPtr) {
    nodeEdgeFnPtr = fnPtr;
  }
  void FEM_SetInterpolateNodeFaceFnPtr(FEM_InterpolateNodeFn fnPtr) {
    nodeFaceFnPtr = fnPtr;
  }
  void FEM_SetInterpolateNodeElementFnPtr(FEM_InterpolateNodeFn fnPtr) {
    nodeElementFnPtr = fnPtr;
  }
  void FEM_SetInterpolateElementCopyFnPtr(FEM_InterpolateElementFn fnPtr) {
    elemCopyFnPtr = fnPtr;
  }
  void FEM_SetInterpolateElementNodeFnPtr(FEM_InterpolateElementFn fnPtr) {
    elemNodeFnPtr = fnPtr;
  }
  void FEM_SetInterpolateCopyAttributesFnPtr(FEM_InterpolateNodeFn fnPtr) {
    nodeCopyFnPtr = fnPtr;
  }
  void FEM_ResetInterpolateNodeEdgeFnPtr() { nodeEdgeFnPtr = NULL; }
  void FEM_ResetInterpolateNodeFaceFnPtr() { nodeFaceFnPtr = NULL; }
  void FEM_ResetInterpolateNodeElementFnPtr() { nodeElementFnPtr = NULL; }
  void FEM_ResetInterpolateElementCopyFnPtr() { elemCopyFnPtr = NULL; }  
  void FEM_ResetInterpolateElementNodeFnPtr() { elemNodeFnPtr = NULL; }
  void FEM_ReetInterpolateCopyAttributesFnPtr() { nodeCopyFnPtr = NULL; }

  // Nodal data

  virtual void FEM_InterpolateNodeOnEdge(NodalArgs args);

  virtual void FEM_InterpolateNodeOnFace(NodalArgs args);

  virtual void FEM_InterpolateNodeInElement(NodalArgs args);

  // Element data

  virtual void FEM_InterpolateElementCopy(ElementArgs args);

  virtual void FEM_InterpolateElementFromNodes(ElementArgs args);

  // Utility

  virtual void FEM_InterpolateElementToNodes(int e);

  //node data
  virtual void FEM_InterpolateCopyAttributes(int oldnode, int newnode);
};

#endif

---