example_met_file.cc

Go to the documentation of this file.

#include "example_met_file.h"
#include "observation_id.h"

using namespace FullPhysics;
using namespace blitz;

#ifdef HAVE_LUA
#include "register_lua.h"
REGISTER_LUA_DERIVED_CLASS(ExampleMetFile, Meteorology)
.def(luabind::constructor<const boost::shared_ptr<HdfFile>&, const std::string&>())
REGISTER_LUA_END()
#endif

//-----------------------------------------------------------------------
//-----------------------------------------------------------------------

ExampleMetFile::ExampleMetFile(const boost::shared_ptr<HdfFile>& input_file, const std::string& observation_id)
: input(input_file)
{
    ObservationId<std::string> obs_id(input_file, "observation_ids", observation_id);
    data_index = obs_id.data_index();
}

ExampleMetFile::ExampleMetFile(const std::string& input_filename, const std::string& observation_id)
: input(new HdfFile(input_filename))
{
    ObservationId<std::string> obs_id(input, "observation_ids", observation_id);
    data_index = obs_id.data_index();
}

//-----------------------------------------------------------------------
//-----------------------------------------------------------------------

double ExampleMetFile::read_scalar(const std::string& field) const
{
    return input->read_field<double, 1>(field, TinyVector<int, 1>(data_index), TinyVector<int, 1>(1))(0);
}

//-----------------------------------------------------------------------
//-----------------------------------------------------------------------

Array<double, 1> ExampleMetFile::read_array(const std::string& Field) const
{
    TinyVector<int, 2> sz = input->read_shape<2>(Field);
    Array<double, 2> raw = input->read_field<double, 2>(Field, TinyVector<int, 2>(data_index, 0), TinyVector<int, 2>(1, sz[1]));
    Array<double, 1> value(raw.extent(secondDim));

    value = raw(0, Range::all());

    return value;
}