PView *GMSH_Scal2VecPlugin::execute(PView *v) { int iViewX = (int)Scal2VecOptions_Number[0].def; int iViewY = (int)Scal2VecOptions_Number[1].def; int iViewZ = (int)Scal2VecOptions_Number[2].def; std::string nameNewView = Scal2VecOptions_String[0].def; PView *vRef = 0, *vX = 0, *vY = 0, *vZ = 0; PViewData *dataRef = 0, *dataX = 0, *dataY = 0, *dataZ = 0; // Load data if(iViewX >= 0){ vX = getView(iViewX, v); if(!vX){ Msg::Error("Scal2Vec plugin could not find View X: %i", iViewX); return v; } dataX = vX->getData(); if(!vRef){ vRef = vX; dataRef = dataX; } } if(iViewY >= 0){ vY = getView(iViewY, v); if(!vY){ Msg::Error("Scal2Vec plugin could not find View Y: %i", iViewY); return v; } dataY = vY->getData(); if(!vRef){ vRef = vY; dataRef = dataY; } } if(iViewZ >= 0){ vZ = getView(iViewZ, v); if(!vZ){ Msg::Error("Scal2Vec plugin could not find View Z: %i", iViewZ); return v; } dataZ = vZ->getData(); if(!vRef){ vRef = vZ; dataRef = dataZ; } } if(!vRef){ Msg::Error("Scal2Vec plugin could not find any view.", iViewZ); return v; } // Initialize the new view PView *vNew = new PView(); PViewDataList *dataNew = getDataList(vNew); for(int ent = 0; ent < dataRef->getNumEntities(0); ent++){ for(int ele = 0; ele < dataRef->getNumElements(0, ent); ele++){ if(dataRef->skipElement(0, ent, ele)) continue; int numComp = 3; // The 3 components of the new view: x,y,z int type = dataRef->getType(0, ent, ele); int numNodes = dataRef->getNumNodes(0, ent, ele); std::vector<double> *out = dataNew->incrementList(numComp, type, numNodes); // Pointer in data of the new view if(!out) continue; double x[8], y[8], z[8], valX, valY, valZ; for(int nod = 0; nod < numNodes; nod++) dataRef->getNode(0, ent, ele, nod, x[nod], y[nod], z[nod]); int dim = dataRef->getDimension(0, ent, ele); elementFactory factory; element *element = factory.create(numNodes, dim, x, y, z); if(!element) continue; for(int nod = 0; nod < numNodes; nod++) out->push_back(x[nod]); // Save coordinates (x,y,z) for(int nod = 0; nod < numNodes; nod++) out->push_back(y[nod]); for(int nod = 0; nod < numNodes; nod++) out->push_back(z[nod]); for(int step = 0; step < dataRef->getNumTimeSteps(); step++){ for(int nod = 0; nod < numNodes; nod++){ if(vX) dataX->getValue(step, ent, ele, nod, 0, valX); else valX = 0; if(vY) dataY->getValue(step, ent, ele, nod, 0, valY); else valY = 0; if(vZ) dataZ->getValue(step, ent, ele, nod, 0, valZ); else valZ = 0; out->push_back(valX); // Save values (fx,fy,fz) out->push_back(valY); out->push_back(valZ); } } delete element; } } for(int step = 0; step < dataRef->getNumTimeSteps(); step++){ if(dataRef->hasTimeStep(step)){ double time = dataRef->getTime(step); dataNew->Time.push_back(time); } } dataNew->setName(nameNewView); dataNew->setFileName(nameNewView + ".pos"); dataNew->finalize(); return vNew; }
PView *GMSH_Scal2VecPlugin::execute(PView *v) { // Load options int iView[3]; for (int comp=0; comp<3; comp++) iView[comp] = (int)Scal2VecOptions_Number[comp].def; // Load data PView *vRef=0, *vComp[3]; for (int comp=0; comp<3; comp++) { if (iView[comp]<0) vComp[comp] = 0; else { vComp[comp] = getView(iView[comp], v); if (!vComp[comp]) { Msg::Error("Scal2Vec plugin could not find View '%i'", iView[comp]); return v; } if (!vRef) vRef = vComp[comp]; } } if (!vRef) { Msg::Error("Scal2Vec plugin could not find any view."); return v; } PViewData *dataRef = vRef->getData(); // Initialize the new view PView *vNew = new PView(); PViewDataList *dataNew = getDataList(vNew); int step0 = dataRef->getFirstNonEmptyTimeStep(); for (int ent=0; ent < dataRef->getNumEntities(step0); ent++) { for (int ele=0; ele < dataRef->getNumElements(step0, ent); ele++) { if (dataRef->skipElement(step0, ent, ele)) continue; int type = dataRef->getType(step0, ent, ele); int numNodes = dataRef->getNumNodes(step0, ent, ele); std::vector<double> *out = dataNew->incrementList(3, type, numNodes); // Pointer in data of the new view if (!out) continue; double x[8], y[8], z[8]; for (int nod=0; nod<numNodes; nod++) dataRef->getNode(step0, ent, ele, nod, x[nod], y[nod], z[nod]); int dim = dataRef->getDimension(step0, ent, ele); elementFactory factory; element *element = factory.create(numNodes, dim, x, y, z); if (!element) continue; for (int nod=0; nod<numNodes; nod++) out->push_back(x[nod]); // Save coordinates (x,y,z) for (int nod=0; nod<numNodes; nod++) out->push_back(y[nod]); for (int nod=0; nod<numNodes; nod++) out->push_back(z[nod]); for (int step=step0; step < dataRef->getNumTimeSteps(); step++) { if (!dataRef->hasTimeStep(step)) continue; for (int nod=0; nod<numNodes; nod++) { for (int comp=0; comp<3; comp++) { double val=0.; if(vComp[comp]) vComp[comp]->getData()->getValue(step, ent, ele, nod, 0, val); out->push_back(val); // Save value } } } delete element; } } for (int step=step0; step < dataRef->getNumTimeSteps(); step++) { if (!dataRef->hasTimeStep(step)) continue; dataNew->Time.push_back(dataRef->getTime(step)); } std::string nameNewView = Scal2VecOptions_String[0].def; dataNew->setName(nameNewView); dataNew->setFileName(nameNewView + ".pos"); dataNew->finalize(); return vNew; }