MeshFields.cpp

Go to the documentation of this file.

#include "MeshAdaptPUMI.h"
#include <apfMesh.h>
#include <apfShape.h>
#include <stdio.h>
#include <string.h>
 
void MeshAdaptPUMIDrvr::freeField(apf::Field*& f)
{
  if (f) {
    apf::destroyField(f);
    f = 0;
  }
}
 
void MeshAdaptPUMIDrvr::freeNumbering(apf::Numbering*& n)
{
  if (n) {
    apf::destroyNumbering(n);
    n = 0;
  }
}
 
int MeshAdaptPUMIDrvr::transferFieldToPUMI(const char* name, double const* inArray,
    int nVar, int nN)
{
  assert(nN == static_cast<int>(m->count(0)));
  apf::Field* f = m->findField(name);
  if (!strcmp(name, "coordinates")) {
    assert(!f); /* coordinates are special, not found by regular findField */
    f = m->getCoordinateField();
    assert(f);
  }
  if (!f) {
    assert(nVar == 1 || nVar == 3 || nVar == 9);
    int valueType;
    if (nVar == 1)
      valueType = apf::SCALAR;
    else if(nVar == 9)
      valueType = apf::MATRIX;
    else
      valueType = apf::VECTOR;
    f = apf::createFieldOn(m, name, valueType);
  }
  apf::NewArray<double> tmp(nVar);
  apf::MeshEntity* v;
  apf::MeshIterator* it = m->begin(0);
  while ((v = m->iterate(it))) {
    int i = localNumber(v);
    for(int j = 0; j < nVar; j++)
      tmp[j] = inArray[i * nVar + j];
    apf::setComponents(f, v, 0, &tmp[0]); 
  }
  m->end(it);
  return 0;
}
 
int MeshAdaptPUMIDrvr::transferFieldToProteus(const char* name, double* outArray,
    int nVar, int nN)
{
  assert(nN == static_cast<int>(m->count(0)));
  apf::Field* f = m->findField(name);
  if (!f)
    fprintf(stderr, "couldn't find field \"%s\"\n", name);
  assert(f);
  assert(apf::countComponents(f) == nVar);
  apf::NewArray<double> tmp(nVar);
  apf::MeshEntity* v;
  apf::MeshIterator* it = m->begin(0);
  while ((v = m->iterate(it))) {
    int i = localNumber(v);
    apf::getComponents(f, v, 0, &tmp[0]);
    for(int j = 0; j < nVar; j++)
      outArray[i * nVar + j] = tmp[j];
  }
  m->end(it);
  return 0;
}
 
int MeshAdaptPUMIDrvr::transferPropertiesToPUMI(double* rho_p, double* nu_p, double *g_p, double deltaT, double deltaT_next,double T_simulation,double interfaceBandSize)
{ 
  nsd = m->getDimension();
  //rho[0] = rho_p[0]; rho[1] = rho_p[1];
  //nu[0] = nu_p[0]; nu[1] = nu_p[1];
 
  rho = (double*) calloc(m->count(m->getDimension()),sizeof(double));
  nu = (double*) calloc(m->count(m->getDimension()),sizeof(double));
  for(int eID = 0; eID<m->count(m->getDimension()); eID++)
  {
    rho[eID] = rho_p[eID];
    nu[eID] = nu_p[eID];
  }  
 
  if(nsd==2){
    g[0] = g_p[0]; g[1] = g_p[1];
  }
  else if(nsd==3){
    g[0] = g_p[0]; g[1] = g_p[1]; g[2] = g_p[2];
  }
  N_interface_band = interfaceBandSize;
  delta_T = deltaT;
  delta_T_next = deltaT_next;
  T_current = T_simulation;
  return 0;
 
  return 0;
}
 
int MeshAdaptPUMIDrvr::transferElementFieldToProteus(const char* name, double* outArray,
    int nVar, int nN)
{
  assert(nN == static_cast<int>(m->count(2)));
  apf::Field* f = m->findField(name);
  if (!f)
    fprintf(stderr, "couldn't find field \"%s\"\n", name);
  assert(f);
  assert(apf::countComponents(f) == nVar);
  apf::NewArray<double> tmp(nVar);
  apf::MeshEntity* v;
  apf::MeshIterator* it = m->begin(2);
  while ((v = m->iterate(it))) {
    int i = localNumber(v);
    apf::getComponents(f, v, 0, &tmp[0]);
    for(int j = 0; j < nVar; j++)
      outArray[i * nVar + j] = tmp[j];
  }
  m->end(it);
  return 0;
}
 
 
/*
int MeshAdaptPUMIDrvr::transferBCtagsToProteus(int* tagArray,int idx, int* ebN, int*eN_global,double* fluxBC)
{
  //Suppose I have a list of identifiers from Proteus that classifies each boundary element
  apf::MeshIterator* it= m->begin(2);
  apf::MeshEntity* f;
  apf::ModelEntity* me;
  apf::ModelEntity* boundary_face;
  int tag = 0;
  int fID,type,boundary_ID;
  int numqpt;
  int count = 0;
 
  char label[9],labelflux[6],type_flag;
 
  //assign a label to the BC type tag
  if(idx == 0) sprintf(&type_flag,"p");
  else if(idx == 1) sprintf(&type_flag,"u");
  else if(idx == 2) sprintf(&type_flag,"v");
  else if(idx == 3) sprintf(&type_flag,"w");
  sprintf(&label[0],"BCtype_%c",type_flag);
  BCtag[idx] = m->createIntTag(label,1);
  std::cout<<"Boundary label "<<label<<std::endl;
 
  //assign a label to the flux tag
  if(idx>0) sprintf(labelflux,"%c_flux",type_flag);
 
  while(f=m->iterate(it)){
    if(count==0){ //happens only once
      apf::MeshElement* sample_elem = apf::createMeshElement(m,f);
      numqpt = apf::countIntPoints(sample_elem,integration_order);
      apf::destroyMeshElement(sample_elem);
      count++;
      if(idx>0)fluxtag[idx]= m->createDoubleTag(labelflux,numqpt);
    }
    me=m->toModel(f);
    tag = m->getModelTag(me);
    boundary_face = m->findModelEntity(2,tag); //faces
    if(me==boundary_face){ //is on model entity
      //Assign a tag to the face for the given type of boundary condition
      fID=localNumber(f);
      boundary_ID = exteriorGlobaltoLocalElementBoundariesArray[fID];
      type = tagArray[numqpt*boundary_ID + 0 ];
      m->setIntTag(f,BCtag[idx],&type);
 
      if(idx>0){
        m->setDoubleTag(f,fluxtag[idx],&fluxBC[numqpt*boundary_ID]); //set the quadrature points
      }
 
    } //end if boundary face
  } //end face loop
  m->end(it);
 
  std::cout<<"Finished Transfer of BC Tags "<<std::endl;
  return 0;
}
*/
 
/*
int MeshAdaptPUMIDrvr::transferBCsToProteus()
{
  //Want to use some sort of Hierarchic projection
  apf::FieldShape* BC_shape = apf::getHierarchic(2);
  apf::MeshEntity* v;
  int nVar = 4; //pressure + 3 velocity components
  //DBC = apf::createPackedField(m, "proteus_DBC", nVar);
  fluxBC = apf::createField(m, "proteus_fluxBC",apf::VECTOR, BC_shape);
 
  //iterate through faces, find adjacent vertices and edges and then construct the linear system to solve for the coefficients
  apf::MeshIterator* it= m->begin(2);
  apf::MeshEntity* f;
  apf::ModelEntity* me;
  apf::ModelEntity* boundary_face;
  int tag = 0;
  int count = 0;
  int numqpt;
  apf::Adjacent adjvtx, adjedg;
  while(f=m->iterate(it)){
 
    if(count==0){ //happens only once
      apf::MeshElement* sample_elem = apf::createMeshElement(m,f);
      numqpt = apf::countIntPoints(sample_elem,integration_order);
      fluxtag[1]= m->createDoubleTag("u_flux",numqpt);
      fluxtag[2]= m->createDoubleTag("v_flux",numqpt);
      fluxtag[3]= m->createDoubleTag("w_flux",numqpt);
      apf::destroyMeshElement(sample_elem);
      count++;
    }
 
    me=m->toModel(f);
    tag=m->getModelTag(me);
    boundary_face = m->findModelEntity(2,tag); //faces
    if(me==boundary_face){
      double test[numqpt];
      for(int i=0;i<numqpt;i++){test[i]=i;}
      m->setDoubleTag(f,fluxtag[1],test);
      double data[numqpt];
      m->getDoubleTag(f,fluxtag[1],&data[0]);
//std::cout<<"What is the data? "<<data[0]<<" "<<data[5]<<" numqpt "<<numqpt<<std::endl;
      //get adjacent vertices
      m->getAdjacent(f,0,adjvtx);
      //evaluate boundary condition and set value
 
      //get adjacent edges
      m->getAdjacent(f,1,adjedg);
    }
  }
  std::cout<<"Finished Transferring BC "<<std::endl;
  m->end(it);
  return 0;
}
*/