Skip to content
Snippets Groups Projects
Commit 4fdd4d7a authored by Christophe Geuzaine's avatar Christophe Geuzaine
Browse files

adding partition and saveInterpMat option to PVIewData::writeMSH

parent cb538422
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Post/PViewData.h
+ 2
− 1
View file @ 4fdd4d7a
......@@ -270,7 +270,8 @@ class PViewData {
virtual bool writePOS(const std::string &fileName, bool binary=false,
bool parsed=true, bool append=false);
virtual bool writeMSH(const std::string &fileName, double version=2.2, bool binary=false,
bool savemesh=true, bool multipleView=false);
bool savemesh=true, bool multipleView=false,
int partitionNum=0, bool saveInterpolationMatrices=true);
virtual bool writeMED(const std::string &fileName);
virtual bool toVector(std::vector<std::vector<double> > &vec);
virtual bool fromVector(const std::vector<std::vector<double> > &vec);
......
Post/PViewDataGModel.h
+ 2
− 1
View file @ 4fdd4d7a
......@@ -250,7 +250,8 @@ class PViewDataGModel : public PViewData {
double time, int partition, int numComp, int numNodes,
const std::string &interpolationScheme);
virtual bool writeMSH(const std::string &fileName, double version=2.2, bool binary=false,
bool savemesh=true, bool multipleView=false);
bool savemesh=true, bool multipleView=false,
int partitionNum=0, bool saveInterpolationMatrices=true);
bool readMED(const std::string &fileName, int fileIndex);
bool writeMED(const std::string &fileName);
};
......
Post/PViewDataGModelIO.cpp
+ 12
− 5
View file @ 4fdd4d7a
......@@ -133,7 +133,8 @@ bool PViewDataGModel::readMSH(const std::string &viewName, const std::string &fi
}
bool PViewDataGModel::writeMSH(const std::string &fileName, double version, bool binary,
bool savemesh, bool multipleView)
bool savemesh, bool multipleView, int partitionNum,
bool saveInterpolationMatrices)
{
if(_steps.empty()) return true;
......@@ -181,7 +182,7 @@ bool PViewDataGModel::writeMSH(const std::string &fileName, double version, bool
}
}
if(haveInterpolationMatrices()){
if(saveInterpolationMatrices && haveInterpolationMatrices()){
fprintf(fp, "$InterpolationScheme\n");
fprintf(fp, "\"INTERPOLATION_SCHEME\"\n");
fprintf(fp, "%d\n", (int)_interpolation.size());
......@@ -212,7 +213,10 @@ bool PViewDataGModel::writeMSH(const std::string &fileName, double version, bool
fprintf(fp, "$NodeData\n");
fprintf(fp, "1\n\"%s\"\n", getName().c_str());
fprintf(fp, "1\n%.16g\n", _steps[step]->getTime());
fprintf(fp, "3\n%d\n%d\n%d\n", step, numComp, numEnt);
if(partitionNum)
fprintf(fp, "4\n%d\n%d\n%d\n%d\n", step, numComp, numEnt, partitionNum);
else
fprintf(fp, "3\n%d\n%d\n%d\n", step, numComp, numEnt);
for(int i = 0; i < _steps[step]->getNumData(); i++){
if(_steps[step]->getData(i)){
MVertex *v = _steps[step]->getModel()->getMeshVertexByTag(i);
......@@ -242,13 +246,16 @@ bool PViewDataGModel::writeMSH(const std::string &fileName, double version, bool
fprintf(fp, "$ElementNodeData\n");
else
fprintf(fp, "$ElementData\n");
if(haveInterpolationMatrices())
if(saveInterpolationMatrices && haveInterpolationMatrices())
fprintf(fp, "2\n\"%s\"\n\"INTERPOLATION_SCHEME\"\n", getName().c_str());
else
fprintf(fp, "1\n\"%s\"\n", getName().c_str());
fprintf(fp, "1\n%.16g\n", _steps[step]->getTime());
fprintf(fp, "3\n%d\n%d\n%d\n", step, numComp, numEnt);
if(partitionNum)
fprintf(fp, "4\n%d\n%d\n%d\n%d\n", step, numComp, numEnt, partitionNum);
else
fprintf(fp, "3\n%d\n%d\n%d\n", step, numComp, numEnt);
for(int i = 0; i < _steps[step]->getNumData(); i++){
if(_steps[step]->getData(i)){
MElement *e = model->getMeshElementByTag(i);
......
Post/PViewDataIO.cpp
+ 2
− 1
View file @ 4fdd4d7a
......@@ -179,7 +179,8 @@ bool PViewData::writePOS(const std::string &fileName, bool binary, bool parsed,
}
bool PViewData::writeMSH(const std::string &fileName, double version, bool binary,
bool savemesh, bool multipleView)
bool savemesh, bool multipleView, int partitionNum,
bool saveInterpolationMatrices)
{
Msg::Error("MSH export not implemented for this view type");
return false;
......
Post/PViewDataList.h
+ 2
− 1
View file @ 4fdd4d7a
......@@ -127,7 +127,8 @@ class PViewDataList : public PViewData {
bool writePOS(const std::string &fileName, bool binary=false, bool parsed=true,
bool append=false);
virtual bool writeMSH(const std::string &fileName, double version=2.2, bool binary=false,
bool savemesh=true, bool multipleView=false);
bool savemesh=true, bool multipleView=false,
int partitionNum=0, bool saveInterpolationMatrices=true);
virtual void importLists(int N[24], std::vector<double> *V[24]);
virtual void getListPointers(int N[24], std::vector<double> *V[24]);
};
......
Post/PViewDataListIO.cpp
+ 9
− 4
View file @ 4fdd4d7a
......@@ -527,7 +527,8 @@ static void createElements(std::vector<double> &list, int nbelm, int nbnod,
}
bool PViewDataList::writeMSH(const std::string &fileName, double version, bool binary,
bool savemesh, bool multipleView)
bool savemesh, bool multipleView,
int partitionNum, bool saveInterpolationMatrices)
{
if(_adaptive){
Msg::Warning("Writing adapted dataset (will only export current time step)");
......@@ -588,7 +589,7 @@ bool PViewDataList::writeMSH(const std::string &fileName, double version, bool b
}
fprintf(fp, "$EndElements\n");
if(haveInterpolationMatrices()){
if(saveInterpolationMatrices && haveInterpolationMatrices()){
fprintf(fp, "$InterpolationScheme\n");
fprintf(fp, "\"INTERPOLATION_SCHEME\"\n");
fprintf(fp, "%d\n", (int)_interpolation.size());
......@@ -612,12 +613,16 @@ bool PViewDataList::writeMSH(const std::string &fileName, double version, bool b
for(int ts = 0; ts < NbTimeStep; ts++){
fprintf(fp, "$ElementNodeData\n");
if(haveInterpolationMatrices())
if(saveInterpolationMatrices && haveInterpolationMatrices())
fprintf(fp, "2\n\"%s\"\n\"INTERPOLATION_SCHEME\"\n", getName().c_str());
else
fprintf(fp, "1\n\"%s\"\n", getName().c_str());
fprintf(fp, "1\n%.16g\n", getTime(ts));
fprintf(fp, "3\n%d\n%d\n%d\n", ts, numComponents, (int)elements.size());
if(partitionNum)
fprintf(fp, "4\n%d\n%d\n%d\n%d\n", ts, numComponents, (int)elements.size(),
partitionNum);
else
fprintf(fp, "3\n%d\n%d\n%d\n", ts, numComponents, (int)elements.size());
num = 0;
for(int i = 0; i < 24; i++){
std::vector<double> *list = 0;
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment