MeshRecordComponent.cpp

/* Copyright 2017-2021 Fabian Koller
 *
 * This file is part of openPMD-api.
 *
 * openPMD-api is free software: you can redistribute it and/or modify
 * it under the terms of of either the GNU General Public License or
 * 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.
 *
 * openPMD-api 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 General Public License and the GNU Lesser General Public License
 * for more details.
 *
 * You should have received a copy of the GNU General Public License
 * and the GNU Lesser General Public License along with openPMD-api.
 * If not, see <http://www.gnu.org/licenses/>.
 */
#include "openPMD/backend/MeshRecordComponent.hpp"
#include "openPMD/backend/BaseRecord.hpp"

namespace openPMD
{
MeshRecordComponent::MeshRecordComponent() : RecordComponent()
{}

MeshRecordComponent::MeshRecordComponent(NoInit) : RecordComponent(NoInit())
{}

MeshRecordComponent::MeshRecordComponent(
    BaseRecord<MeshRecordComponent> const &baseRecord)
    : RecordComponent(NoInit())
{
    setData(baseRecord.m_recordComponentData);
}

void MeshRecordComponent::read()
{
    using DT = Datatype;
    Parameter<Operation::READ_ATT> aRead;

    aRead.name = "position";
    IOHandler()->enqueue(IOTask(this, aRead));
    IOHandler()->flush(internal::defaultFlushParams);
    Attribute a = Attribute(Attribute::from_any, *aRead.m_resource);
    if (*aRead.dtype == DT::VEC_FLOAT || *aRead.dtype == DT::FLOAT)
        setPosition(a.get<std::vector<float> >());
    else if (*aRead.dtype == DT::VEC_DOUBLE || *aRead.dtype == DT::DOUBLE)
        setPosition(a.get<std::vector<double> >());
    else if (
        *aRead.dtype == DT::VEC_LONG_DOUBLE || *aRead.dtype == DT::LONG_DOUBLE)
        setPosition(a.get<std::vector<long double> >());
    // conversion cast if a backend reports an integer type
    else if (auto val = a.getOptional<std::vector<double> >(); val.has_value())
        setPosition(val.value());
    else
        throw error::ReadError(
            error::AffectedObject::Attribute,
            error::Reason::UnexpectedContent,
            {},
            "Unexpected Attribute datatype for 'position' (expected a vector "
            "of any floating point type, found " +
                datatypeToString(
                    Attribute(Attribute::from_any, *aRead.m_resource).dtype) +
                ")");

    readBase(/* require_unit_si = */ true);
}

void MeshRecordComponent::flush(
    std::string const &name, internal::FlushParams const &params)
{
    if (!dirtyRecursive())
    {
        return;
    }
    if (access::write(IOHandler()->m_frontendAccess) &&
        !containsAttribute("position"))
    {
        setPosition(std::vector<double>{0});
    }
    RecordComponent::flush(name, params);
}

template <typename T>
MeshRecordComponent &MeshRecordComponent::setPosition(std::vector<T> pos)
{
    static_assert(
        std::is_floating_point<T>::value,
        "Type of attribute must be floating point");

    setAttribute("position", pos);
    return *this;
}

template MeshRecordComponent &
MeshRecordComponent::setPosition(std::vector<float> pos);
template MeshRecordComponent &
MeshRecordComponent::setPosition(std::vector<double> pos);
template MeshRecordComponent &
MeshRecordComponent::setPosition(std::vector<long double> pos);
} // namespace openPMD