TecPlotTools.cpp File Reference

Detailed Description

Copyright

Copyright (c) 2012-2021, OpenGeoSys Community (http://www.opengeosys.org) Distributed under a Modified BSD License. See accompanying file LICENSE.txt or http://www.opengeosys.org/project/license

Definition in file TecPlotTools.cpp.

#include <tclap/CmdLine.h>
#include <iostream>
#include <memory>
#include <string>
#include <string_view>
#include <vector>
#include "BaseLib/StringTools.h"
#include "GeoLib/Point.h"
#include "InfoLib/GitInfo.h"
#include "MeshLib/IO/VtkIO/VtuInterface.h"
#include "MeshLib/Mesh.h"
#include "MeshLib/MeshGenerators/RasterToMesh.h"

Include dependency graph for TecPlotTools.cpp:

Go to the source code of this file.

Functions
std::string	getValue (std::string const &line, std::string const &val_name, bool is_string)
std::string	getName (std::string const &line)
	Returns the name/title from the "Zone"-description. More...
std::string_view	getZonetype (std::string const &line)
std::pair< std::size_t, std::size_t >	getDimensions (std::string const &line)
	Returns raster dimensions from the "Zone"-description. More...
std::string	trimVariable (std::string &var)
	Trims a substring containing a variable-name. More...
std::vector< std::string >	getVariables (std::string const &line)
	Returns an vector of variable names from the "Variables"-description. More...
bool	dataCountError (std::string const &name, std::size_t const &current, std::size_t const &total)
	Tests if the number of read values equals the number of expected values. More...
bool	dataCountError (std::ofstream &out, std::string const &name, std::size_t const &current, std::size_t const &total)
void	resetDataStructures (std::size_t const &n_scalars, std::vector< std::vector< double >> &scalars, std::size_t &val_count)
	Resets all data structures after a new "Variables"-description is found. More...
void	writeTecPlotSection (std::ofstream &out, std::string const &file_name, std::size_t &write_count, std::size_t &val_count, std::size_t &val_total)
	Writes one section/zone into a separate TecPlot file. More...
int	writeDataToMesh (std::string const &file_name, std::size_t &write_count, std::vector< std::string > const &vec_names, std::vector< std::vector< double >> const &scalars, std::pair< std::size_t, std::size_t > const &dims)
	Writes one section/zone into a separate OGS-mesh. More...
void	skipGeometrySection (std::ifstream &in, std::string &line)
	If a geometry-section is encountered, it is currently ignored. More...
int	splitFile (std::ifstream &in, std::string const &file_name)
	Splits a TecPlot file containing multiple sections/zones into separate files. More...
int	convertFile (std::ifstream &in, std::string const &file_name)
int	main (int argc, char *argv[])

Function Documentation

convertFile()

int convertFile	(	std::ifstream &	in,
		std::string const &	file_name
	)

Converts a TecPlot file into one or more OGS-meshes (one mesh per section/zone)

Definition at line 338 of file TecPlotTools.cpp.

{
    std::string line;
    std::string name;
    std::pair<std::size_t, std::size_t> dims(0, 0);
    std::vector<std::string> var_names;
    std::vector<std::vector<double>> scalars;
    std::size_t val_count(0);
    std::size_t val_total(0);
    std::size_t write_count(0);
    while (std::getline(in, line))
    {
        if (line.find("GEOMETRY") != std::string::npos)
        {
            skipGeometrySection(in, line);
        }
 
        if (line.empty())
        {
            continue;
        }
        if (line.find("TITLE") != std::string::npos)
        {
            if (dataCountError(name, val_count, val_total))
            {
                return -3;
            }
            if (val_count != 0)
            {
                writeDataToMesh(file_name, write_count, var_names, scalars,
                                dims);
                resetDataStructures(var_names.size(), scalars, val_count);
            }
            continue;
        }
        if (line.find("VARIABLES") != std::string::npos)
        {
            if (val_count != 0)
            {
                if (dataCountError(name, val_count, val_total))
                {
                    return -3;
                }
                writeDataToMesh(file_name, write_count, var_names, scalars,
                                dims);
            }
            var_names.clear();
            var_names = getVariables(line);
            resetDataStructures(var_names.size(), scalars, val_count);
            continue;
        }
        if (line.find("ZONE") != std::string::npos)
        {
            if (val_count != 0)
            {
                if (dataCountError(name, val_count, val_total))
                {
                    return -3;
                }
                writeDataToMesh(file_name, write_count, var_names, scalars,
                                dims);
                resetDataStructures(var_names.size(), scalars, val_count);
            }
            name = getName(line);
            if (auto const zonetype = getZonetype(line); zonetype != "ORDERED")
            {
                ERR("Given zonetype '{:s}' is not supported. Only 'ORDERED' "
                    "zonetype data can be converted.",
                    zonetype);
                return -4;
            }
            dims = getDimensions(line);
            val_total = dims.first * dims.second;
            val_count = 0;
            continue;
        }
 
        double x;
        std::stringstream iss(line);
        std::size_t i(0);
        std::size_t const n_scalars(scalars.size());
        while (iss >> x)
        {
            if (i > n_scalars - 1)
            {
                ERR("Too much data for existing scalar arrays");
                return -3;
            }
            scalars[i++].push_back(x);
        }
        if (i < n_scalars)
        {
            ERR("Not enough data for existing scalar arrays");
            return -3;
        }
        val_count++;
    }
    if (dataCountError(name, val_count, val_total))
    {
        return -3;
    }
    writeDataToMesh(file_name, write_count, var_names, scalars, dims);
    INFO("Finished conversion.");
    return 0;
}

References dataCountError(), ERR(), getDimensions(), getName(), getVariables(), getZonetype(), INFO(), MaterialPropertyLib::name, resetDataStructures(), skipGeometrySection(), and writeDataToMesh().

Referenced by main().

dataCountError() [1/2]

bool dataCountError	(	std::ofstream &	out,
		std::string const &	name,
		std::size_t const &	current,
		std::size_t const &	total
	)

Tests if the number of read values equals the number of expected values and closes the stream

Definition at line 153 of file TecPlotTools.cpp.

{
    if (dataCountError(name, current, total))
    {
        out.close();
        return true;
    }
    return false;
}

References dataCountError(), and MaterialPropertyLib::name.

dataCountError() [2/2]

bool dataCountError	(	std::string const &	name,
		std::size_t const &	current,
		std::size_t const &	total
	)

Tests if the number of read values equals the number of expected values.

Definition at line 137 of file TecPlotTools.cpp.

{
    if (current != total)
    {
        ERR("Data rows found do not fit specified dimensions for section "
            "'{:s}'.",
            name);
        return true;
    }
    return false;
}

References ERR(), and MaterialPropertyLib::name.

Referenced by convertFile(), dataCountError(), and splitFile().

getDimensions()

std::pair<std::size_t, std::size_t> getDimensions

(

std::string const &

line

)

Returns raster dimensions from the "Zone"-description.

Definition at line 92 of file TecPlotTools.cpp.

{
    std::pair<std::size_t, std::size_t> dims;
    std::stringstream start(getValue(line, "I=", false));
    start >> dims.first;
    std::stringstream end(getValue(line, "J=", false));
    end >> dims.second;
    return dims;
}

References getValue().

Referenced by convertFile(), and splitFile().

getName()

std::string getName

(

std::string const &

line

)

Returns the name/title from the "Zone"-description.

Definition at line 59 of file TecPlotTools.cpp.

{
    return getValue(line, "T=", true);
}

References getValue().

Referenced by StationTreeModel::addStationList(), GeoLib::GEOObjects::appendPolylineVec(), GeoLib::GEOObjects::appendSurfaceVec(), ProcessLib::ComponentTransport::LocalAssemblerData< ShapeFunction, IntegrationMethod, GlobalDim >::assembleBlockMatrices(), ProcessLib::ComponentTransport::LocalAssemblerData< ShapeFunction, IntegrationMethod, GlobalDim >::assembleComponentTransportEquation(), FileIO::Gocad::GocadAsciiReader::checkMeshNames(), convertFile(), GeoLib::GEOObjects::exists(), GeoLib::GEOObjects::getPolylineVec(), GeoLib::GEOObjects::getPolylineVecObj(), GeoLib::GEOObjects::getSurfaceVec(), GeoLib::GEOObjects::getSurfaceVecObj(), splitFile(), GeoLib::IO::XmlStnInterface::write(), FileIO::Legacy::writeAllDataToGLIFileV4(), FileIO::GMSH::GMSHPolygonTree::writeStations(), and StationTreeView::writeStratigraphiesAsImages().

getValue()

std::string getValue	(	std::string const &	line,
		std::string const &	val_name,
		bool	is_string
	)

Returns the value for the given parameter name (i.e. for "x = 3" it returns "3")

Definition at line 27 of file TecPlotTools.cpp.

{
    std::string value;
    std::size_t start(line.find(val_name));
    std::size_t end(std::string::npos);
    if (start == end)
    {
        ERR("Value not found.");
        return "";
    }
    value = line.substr(start + 1, std::string::npos);
    if (is_string)
    {
        start = value.find("\"");
        value = value.substr(start + 1, std::string::npos);
        end = value.find("\"");
    }
    else
    {
        start = value.find_first_not_of(" ");
        value = value.substr(start + 1, std::string::npos);
        BaseLib::trim(value);
        end = value.find_first_of(" ");
    }
    value = value.substr(0, end);
    BaseLib::trim(value);
    return value;
}

References ERR(), and BaseLib::trim().

Referenced by getDimensions(), getName(), MaterialLib::Fluid::FluidPropertiesWithDensityDependentModels::getValue(), and main().

getVariables()

std::vector<std::string> getVariables

(

std::string const &

line

)

Returns an vector of variable names from the "Variables"-description.

Definition at line 112 of file TecPlotTools.cpp.

{
    std::string const var_str("VARIABLES");
    std::size_t start(line.find(var_str));
    std::string all_vars =
        line.substr(start + var_str.length(), std::string::npos);
    start = all_vars.find("=");
    all_vars = all_vars.substr(start + 1, std::string::npos);
    BaseLib::trim(all_vars);
 
    std::vector<std::string> variables;
    std::size_t end = 0;
    while (end != std::string::npos)
    {
        end = all_vars.find_first_of(" ");
        std::string var = all_vars.substr(0, end);
        variables.push_back(trimVariable(var));
        all_vars = all_vars.substr(end + 1, std::string::npos);
        BaseLib::trim(all_vars);
    }
 
    return variables;
}

References BaseLib::trim(), and trimVariable().

Referenced by convertFile().

getZonetype()

std::string_view getZonetype

(

std::string const &

line

)

Definition at line 64 of file TecPlotTools.cpp.

{
    auto start = line.find("ZONETYPE=");
    if (start == std::string::npos)
    {
        OGS_FATAL(
            "A required 'ZONETYPE=' substring is not available in the ZONE "
            "description: '{:s}'.",
            line);
    }
    start += std::size("ZONETYPE=") - 1;
 
    auto end = line.find(',', start);
    if (end == std::string::npos)
    {
        OGS_FATAL(
            "Expected the 'ZONETYPE=type' to be followed by a comma in the "
            "ZONE description '{:s}'. The zone type starts at position {:d}.",
            line, start);
    }
    if (start == end)
    {
        ERR("ZONETYPE string is empty in ZONE description '{:s}'.", line);
    }
    return std::string_view(&line[start], end - start);
}

References ERR(), and OGS_FATAL.

Referenced by convertFile().

main()

int main	(	int	argc,
		char *	argv[]
	)

Small collection of scripts to handle TecPlot files.

Return codes: 0 : no errors -1 : missing arguments -2 : file I/O error -3 : index error -4 : error interpreting data -5 : error creating data structures

Definition at line 455 of file TecPlotTools.cpp.

{
    TCLAP::CmdLine cmd(
        "TecPlot Parser\n\n"
        "OpenGeoSys-6 software, version " +
            GitInfoLib::GitInfo::ogs_version +
            ".\n"
            "Copyright (c) 2012-2021, OpenGeoSys Community "
            "(http://www.opengeosys.org)",
        ' ', GitInfoLib::GitInfo::ogs_version);
    TCLAP::SwitchArg split_arg("s", "split",
                               "split time steps into separate files");
    cmd.add(split_arg);
    TCLAP::SwitchArg convert_arg("c", "convert",
                                 "convert TecPlot data into OGS meshes");
    cmd.add(convert_arg);
    TCLAP::ValueArg<std::string> output_arg(
        "o", "output-file", "output mesh file", false, "", "string");
    cmd.add(output_arg);
    TCLAP::ValueArg<std::string> input_arg(
        "i", "input-file", "TecPlot input file", true, "", "string");
    cmd.add(input_arg);
    cmd.parse(argc, argv);
 
    if (!input_arg.isSet())
    {
        ERR("No input file given. Please specify TecPlot (*.plt) file");
        return -1;
    }
 
    if (convert_arg.getValue() && !output_arg.isSet())
    {
        ERR("No output file given. Please specify OGS mesh (*.vtu) file");
        return -1;
    }
 
    std::ifstream in(input_arg.getValue().c_str());
    if (!in.is_open())
    {
        ERR("Could not open file {:s}.", input_arg.getValue());
        return -2;
    }
 
    if (!convert_arg.isSet() && !split_arg.isSet())
    {
        INFO("Nothing to do. Use -s to split or -c to convert.");
        return 0;
    }
 
    std::string const filename =
        (output_arg.isSet()) ? output_arg.getValue() : input_arg.getValue();
    int return_val(0);
    if (split_arg.getValue())
    {
        return_val = splitFile(in, filename);
    }
    else if (convert_arg.getValue())
    {
        return_val = convertFile(in, filename);
    }
 
    in.close();
    return return_val;
}

References convertFile(), ERR(), INFO(), GitInfoLib::GitInfo::ogs_version, and splitFile().

resetDataStructures()

void resetDataStructures	(	std::size_t const &	n_scalars,
		std::vector< std::vector< double >> &	scalars,
		std::size_t &	val_count
	)

Resets all data structures after a new "Variables"-description is found.

Definition at line 167 of file TecPlotTools.cpp.

{
    scalars.clear();
    scalars.reserve(n_scalars);
    for (std::size_t i = 0; i < n_scalars; ++i)
    {
        scalars.emplace_back(0);
    }
    val_count = 0;
}

Referenced by convertFile().

skipGeometrySection()

void skipGeometrySection	(	std::ifstream &	in,
		std::string &	line
	)

If a geometry-section is encountered, it is currently ignored.

Definition at line 261 of file TecPlotTools.cpp.

{
    while (std::getline(in, line))
    {
        if ((line.find("TITLE") != std::string::npos) ||
            (line.find("VARIABLES") != std::string::npos) ||
            (line.find("ZONE") != std::string::npos))
        {
            return;
        }
    }
}

Referenced by convertFile().

splitFile()

int splitFile	(	std::ifstream &	in,
		std::string const &	file_name
	)

Splits a TecPlot file containing multiple sections/zones into separate files.

Definition at line 275 of file TecPlotTools.cpp.

{
    std::ofstream out;
    std::string line;
    std::string name;
    std::size_t val_count(0);
    std::size_t val_total(0);
    std::size_t write_count(0);
    while (std::getline(in, line))
    {
        if (line.find("TITLE") != std::string::npos)
        {
            if (dataCountError(out, name, val_count, val_total))
            {
                return -3;
            }
            writeTecPlotSection(out, file_name, write_count, val_count,
                                val_total);
            out << line << "\n";
            continue;
        }
        if (line.find("VARIABLES") != std::string::npos)
        {
            if (dataCountError(out, name, val_count, val_total))
            {
                return -3;
            }
            writeTecPlotSection(out, file_name, write_count, val_count,
                                val_total);
            out << line << "\n";
            continue;
        }
        if (line.find("ZONE") != std::string::npos)
        {
            if (dataCountError(out, name, val_count, val_total))
            {
                return -3;
            }
            writeTecPlotSection(out, file_name, write_count, val_count,
                                val_total);
            out << line << "\n";
            name = getName(line);
            std::pair<std::size_t, std::size_t> dims = getDimensions(line);
            val_total = dims.first * dims.second;
            val_count = 0;
            continue;
        }
 
        out << line << "\n";
        val_count++;
    }
    if (dataCountError(out, name, val_count, val_total))
    {
        return -3;
    }
    INFO("Writing time step #{:i}", write_count);
    out.close();
    INFO("Finished split.");
    return 0;
}

References dataCountError(), getDimensions(), getName(), INFO(), MaterialPropertyLib::name, and writeTecPlotSection().

Referenced by main().

trimVariable()

std::string trimVariable

(

std::string &

var

)

Trims a substring containing a variable-name.

Definition at line 103 of file TecPlotTools.cpp.

{
    std::size_t const start = var.find_first_not_of("\"");
    var = var.substr(start, std::string::npos);
    std::size_t const end = var.find_first_of("\"");
    return var.substr(0, end);
}

Referenced by getVariables().

writeDataToMesh()

int writeDataToMesh	(	std::string const &	file_name,
		std::size_t &	write_count,
		std::vector< std::string > const &	vec_names,
		std::vector< std::vector< double >> const &	scalars,
		std::pair< std::size_t, std::size_t > const &	dims
	)

Writes one section/zone into a separate OGS-mesh.

Definition at line 203 of file TecPlotTools.cpp.

{
    double cellsize = 0;
    for (std::size_t i = 0; i < vec_names.size(); ++i)
    {
        if (vec_names[i] == "x" || vec_names[i] == "X")
        {
            cellsize = scalars[i][1] - scalars[i][0];
            break;
        }
    }
 
    if (cellsize == 0)
    {
        ERR("Cell size not found. Aborting...");
        return -4;
    }
 
    GeoLib::Point origin(0, 0, 0);
    GeoLib::RasterHeader header{dims.first, dims.second, 1,
                                origin,     cellsize,    -9999};
 
    std::unique_ptr<MeshLib::Mesh> mesh(
        MeshLib::RasterToMesh::convert(scalars[0].data(),
                                       header,
                                       MeshLib::MeshElemType::QUAD,
                                       MeshLib::UseIntensityAs::DATAVECTOR,
                                       vec_names[0]));
    MeshLib::Properties& properties = mesh->getProperties();
    for (std::size_t i = 1; i < vec_names.size(); ++i)
    {
        auto* const prop = properties.createNewPropertyVector<double>(
            vec_names[i], MeshLib::MeshItemType::Cell, 1);
        if (!prop)
        {
            ERR("Error creating array '{:s}'.", vec_names[i]);
            return -5;
        }
        prop->reserve(scalars[i].size());
        std::copy(scalars[i].cbegin(), scalars[i].cend(),
                  std::back_inserter(*prop));
    }
 
    std::size_t const delim_pos(file_name.find_last_of("."));
    std::string const base_name(file_name.substr(0, delim_pos + 1));
    std::string const extension(file_name.substr(delim_pos, std::string::npos));
 
    INFO("Writing section #{:i}", write_count);
    MeshLib::IO::VtuInterface vtu(mesh.get());
    vtu.writeToFile(base_name + std::to_string(write_count++) + extension);
    return 0;
}

References MeshLib::Cell, MeshLib::RasterToMesh::convert(), MathLib::LinAlg::copy(), MeshLib::Properties::createNewPropertyVector(), MeshLib::DATAVECTOR, ERR(), INFO(), MeshLib::QUAD, and MeshLib::IO::VtuInterface::writeToFile().

Referenced by convertFile().

writeTecPlotSection()

void writeTecPlotSection	(	std::ofstream &	out,
		std::string const &	file_name,
		std::size_t &	write_count,
		std::size_t &	val_count,
		std::size_t &	val_total
	)

Writes one section/zone into a separate TecPlot file.

Definition at line 181 of file TecPlotTools.cpp.

{
    if (write_count == 0 || val_total != 0)
    {
        std::size_t const delim_pos(file_name.find_last_of("."));
        std::string const base_name(file_name.substr(0, delim_pos + 1));
        std::string const extension(
            file_name.substr(delim_pos, std::string::npos));
 
        val_count = 0;
        val_total = 0;
        INFO("Writing section #{:i}", write_count);
        out.close();
        out.open(base_name + std::to_string(write_count++) + extension);
    }
}

References INFO().

Referenced by splitFile().