json_probes.h

Go to the documentation of this file.

#pragma once
 
#include "json_wrapper.h"
#include "probes_params.h"
namespace dg
{
namespace file
{
 
inline ProbesParams parse_probes( const dg::file::WrappedJsonValue& js, enum error
    probes_err = file::error::is_silent)
{
    ProbesParams out;
    int rank = 0;
#ifdef MPI_VERSION
    MPI_Comm_rank( MPI_COMM_WORLD, &rank);
#endif // MPI_VERSION
    if( (probes_err == file::error::is_silent) && !js.isMember( "probes"))
        return out;
    else if( (probes_err == file::error::is_warning) && !js.isMember( "probes"))
    {
        DG_RANK0 std::cerr << "WARNING: probes field not found.  No probes written to file!\n";
        return out;
    }
    else if ( !js.isMember("probes"))
        throw std::runtime_error( "\"probes\" field not found!");
 
    // test if parameters are file or direct
    auto jsprobes = js["probes"];
    std::string type = jsprobes["input"].asString();
    if( type == "file")
    {
        std::string path = jsprobes["file"].asString();
        // everyone reads the file
        jsprobes.asJson()["coords"] = dg::file::file2Json( path,
                dg::file::comments::are_discarded, dg::file::error::is_throw);
    }
    else if( type != "coords")
    {
        throw std::runtime_error( "Error: Unknown coordinates input '"
               + type + "'.");
    }
 
    auto coords = jsprobes["coords"];
 
    // read in parameters
 
    unsigned ndim = coords["coords-names"].size();
 
    std::string first = coords["coords-names"][0].asString();
    out.coords_names.resize(ndim);
    out.coords.resize(ndim);
    for( unsigned i=0; i<ndim; i++)
    {
        out.coords_names[i] = coords["coords-names"][i].asString();
    }
    unsigned num_pins = coords[out.coords_names[0]].size();
    out.probes = (num_pins > 0);
 
    if( rank == 0)
    {
    // only master thread reads probes
    for( unsigned i=0; i<ndim; i++)
    {
        unsigned num_pins = coords[out.coords_names[i]].size();
        out.coords[i].resize(num_pins);
        double scale = 1.;
        if( type == "file")
            scale = jsprobes["scale"][i].asDouble();
        for( unsigned k=0; k<num_pins; k++)
            out.coords[i][k] = coords.asJson()[out.coords_names[i]][k]
#ifdef DG_USE_JSONHPP
            .template get<double>()
#else
            .asDouble()
#endif
                *scale;
    }
    }
    // does not check that all probes have same size
    out.format = coords["format"].toStyledString();
    return out;
}
 
 
 
}//namespace file
} //namespace dg