Homec4science
Diffusion iohelper (master) Browse

iohelper/src/dumper_paraview.cc
9c0b4ce82ee8master

Last Change
Raw File

dumper_paraview.cc
View Options

/**
* @file dumper_paraview.cc
*
* @author Guillaume Anciaux <guillaume.anciaux@epfl.ch>
*
* @date Thu Mar 11 12:45:06 2010
*
* @brief implementation of the paraview dumper
*
* @section LICENSE
*
* Copyright (©) 2010-2011 EPFL (Ecole Polytechnique Fédérale de Lausanne)
* Laboratory (LSMS - Laboratoire de Simulation en Mécanique des Solides)
*
* Akantu is free software: you can redistribute it and/or modify it under the
* terms of the GNU Lesser General Public License as published by the Free
* Software Foundation, either version 3 of the License, or (at your option) any
* later version.
*
* Akantu is distributed in the hope that it will be useful, but WITHOUT ANY
* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
* A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
* details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with Akantu. If not, see <http://www.gnu.org/licenses/>.
*
*/
/* -------------------------------------------------------------------------- */
#include <sys/stat.h>
#include <sys/types.h>
#include <string>
#include <iomanip>
#include "dumper_paraview.hh"
#include "file_manager.hh"
/* -------------------------------------------------------------------------- */
__BEGIN_IOHELPER__
/* -------------------------------------------------------------------------- */
#if defined(__INTEL_COMPILER)
/// remark #981: operands are evaluated in unspecified order
#pragma warning ( disable : 981 )
/// remark #383: value copied to temporary, reference to temporary used
#pragma warning ( disable : 383 )
#endif //defined(__INTEL_COMPILER)
DumperParaview::DumperParaview(const std::string & prefix) :
Dumper(prefix),flag_compressed(false){
mode_offset = 0;
root_rank = 0;
this->mode = BASE64;
vtu_subdirectory = std::string(".") + directory_separator;
}
/* -------------------------------------------------------------------------- */
DumperParaview::~DumperParaview(){
}
/* -------------------------------------------------------------------------- */
void DumperParaview::setConnectivity(int * connectivity,ElemType element_type,
UInt nb_elem, int mode){
Dumper::setConnectivity(connectivity,element_type,nb_elem,mode);
if (connectivity_mode == FORTRAN_MODE) mode_offset = -1;
else mode_offset = 0;
}
/* -------------------------------------------------------------------------- */
void DumperParaview::setMode(int mode){
Dumper::setMode(mode);
}
/* -------------------------------------------------------------------------- */
void DumperParaview::dump(const std::string & name){
// int offset=0,count=0;
File file;
//create directory to store the per proc vtus
mkdir(std::string(getPrefix()).c_str(),0755);
std::string basename = this->getBaseName();
if (name != "") basename = name;
std::string full_name = this->getFullName(basename);
if (my_rank == root_rank){
/* ouverture du fichier parallele vtk */
std::stringstream temp;
temp << full_name << ".pvtu";
file.open(temp.str());
if (!file.is_open()) IOHELPER_THROW("could not open file " << temp.str(),
_et_file_error);
ParaviewHelper paraHelper(file,this->mode);
std::string fullname;
fullname = vtu_subdirectory + basename;
paraHelper.writePVTU(per_node_data,per_element_data,
world_size,fullname);
}
//create directory to store the per proc vtus
mkdir(std::string(getPrefix() + vtu_subdirectory).c_str(),0755);
//open current file
std::string vtuname = ParaviewHelper::getVTUName(basename, my_rank);
std::string fullvtuname = getFullName(vtu_subdirectory + "/" + vtuname);
file.open(fullvtuname, std::fstream::out, flag_compressed);
if (!file.is_open()) IOHELPER_THROW("could not open file " << fullvtuname,
_et_file_error);
ParaviewHelper paraHelper(file, this->mode);
field_map::iterator pos_it = per_node_data.find("positions");
if (pos_it == per_node_data.end())
IOHELPER_THROW("positions field was not specified",
_et_missing_field);
FieldInterface & positions = *(pos_it->second);
UInt nb_nodes = positions.size();
field_map::iterator elty_it = per_element_data.find("element_type");
if (elty_it == per_element_data.end())
IOHELPER_THROW("element type field was not specified",
_et_missing_field);
FieldInterface & element_types = *(elty_it->second);
UInt nb_elems = element_types.size();
paraHelper.writeHeader(nb_nodes, nb_elems);
// if (positions.getDim() != 3) FATAL("there is a problem with the position iterator");
paraHelper.startDofList(3);
paraHelper.pushPosition(positions);
paraHelper.endDofList();
//connectivity dump
field_map::iterator conn_it = per_element_data.find("connectivities");
if (conn_it == per_element_data.end())
IOHELPER_THROW("connectivities were not specified",
_et_missing_field);
FieldInterface & connectivities = *(conn_it->second);
// start to push info on cells
paraHelper.startCells();
// push connectivities
paraHelper.startCellsConnectivityList();
if (connectivities.size() == 0){
for (UInt i = 0 ; i < positions.size() ; ++i) {
paraHelper.pushDatum(i);
}
}
else
paraHelper.pushField(connectivities);
paraHelper.endCellsConnectivityList();
// build offsets
paraHelper.startCellsoffsetsList();
if (connectivities.size() == 0){
paraHelper.pushDatum(positions.size());
}
else
paraHelper.buildOffsets(connectivities);
paraHelper.endCellsoffsetsList();
// push cell types
paraHelper.startCellstypesList();
if (connectivities.size() == 0){
paraHelper.pushDatum(paraview_code_type[POINT_SET]);
}
else
paraHelper.pushField(element_types);
paraHelper.endCellstypesList();
//close cell section
paraHelper.endCells();
//! add data fields
paraHelper.pushDataFields(per_node_data, per_element_data);
paraHelper.write_conclusion();
file.close();
this->incDumpStep();
}
/* -------------------------------------------------------------------------- */
void DumperParaview::setVTUSubDirectory(const std::string & sub){
vtu_subdirectory = sub;
if (vtu_subdirectory[vtu_subdirectory.size()-1] != directory_separator) {
vtu_subdirectory += directory_separator;
}
}
/* -------------------------------------------------------------------------- */
__END_IOHELPER__
c4science · Help