rectangular3d.hpp

Go to the documentation of this file.

/*
 * This file is part of PLaSK (https://plask.app) by Photonics Group at TUL
 * Copyright (c) 2022 Lodz University of Technology
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, version 3.
 *
 * This program 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 for more details.
 */
#ifndef PLASK__RECTANGULAR3D_H
#define PLASK__RECTANGULAR3D_H
 
#include "rectangular_common.hpp"
#include "rectilinear3d.hpp"
#include "../optional.hpp"
 
namespace plask {
 
class PLASK_API RectangularMesh3D: public RectilinearMesh3D {
 
  public:
 
    typedef RectilinearMesh3D::ElementMesh<RectangularMesh3D> ElementMesh;
 
    explicit RectangularMesh3D(IterationOrder iterationOrder = ORDER_012);
 
    RectangularMesh3D(shared_ptr<MeshAxis> mesh0, shared_ptr<MeshAxis> mesh1, shared_ptr<MeshAxis> mesh2, IterationOrder iterationOrder = ORDER_012);
 
    RectangularMesh3D(const RectangularMesh3D& src, bool clone_axes = false);
 
    const shared_ptr<MeshAxis>& lon() const { return axis[0]; }
 
    const shared_ptr<MeshAxis>& tran() const { return axis[1]; }
 
    const shared_ptr<MeshAxis>& vert() const { return axis[2]; }
 
    const shared_ptr<MeshAxis>& ee_z() const { return axis[0]; }
 
    const shared_ptr<MeshAxis>& ee_x() const { return axis[1]; }
 
    const shared_ptr<MeshAxis>& ee_y() const { return axis[2]; }
 
    const shared_ptr<MeshAxis>& rad_r() const { return axis[0]; }
 
    const shared_ptr<MeshAxis>& rad_phi() const { return axis[1]; }
 
    const shared_ptr<MeshAxis>& rad_z() const { return axis[2]; }
 
    void writeXML(XMLElement& object) const override;
 
    using RectilinearMesh3D::at;    // MSVC needs this
 
    Vec<3, double> at(std::size_t index0, std::size_t index1, std::size_t index2) const override {
        return Vec<3, double>(axis[0]->at(index0), axis[1]->at(index1), axis[2]->at(index2));
    }
 
    shared_ptr<RectangularMesh3D::ElementMesh> getElementMesh() const;
 
    double getElementArea(std::size_t index0, std::size_t index1, std::size_t index2) const {
        return (axis[0]->at(index0+1) - axis[0]->at(index0)) * (axis[1]->at(index1+1) - axis[1]->at(index1)) * (axis[2]->at(index2+1) - axis[2]->at(index2));
    }
 
    double getElementArea(std::size_t element_index) const {
        std::size_t bl_index = getElementMeshLowIndex(element_index);
        return getElementArea(index0(bl_index), index1(bl_index), index2(bl_index));
    }
 
    Vec<3, double> getElementMidpoint(std::size_t index0, std::size_t index1, std::size_t index2) const override {
        return vec(getElementMidpoint0(index0), getElementMidpoint1(index1), getElementMidpoint2(index2));
    }
 
    Vec<3, double> getElementMidpoint(std::size_t element_index) const {
        std::size_t bl_index = getElementMeshLowIndex(element_index);
        return getElementMidpoint(index0(bl_index), index1(bl_index), index2(bl_index));
    }
 
    Box3D getElementBox(std::size_t index0, std::size_t index1, std::size_t index2) const {
        return Box3D(axis[0]->at(index0), axis[1]->at(index1), axis[2]->at(index2), axis[0]->at(index0+1), axis[1]->at(index1+1), axis[2]->at(index2+1));
    }
 
    Box3D getElementBox(std::size_t element_index) const {
        std::size_t bl_index = getElementMeshLowIndex(element_index);
        return getElementBox(index0(bl_index), index1(bl_index), index2(bl_index));
    }
 
};
 
template <typename SrcT, typename DstT>
struct InterpolationAlgorithm<RectangularMesh3D, SrcT, DstT, INTERPOLATION_LINEAR> {
    static LazyData<DstT> interpolate(const shared_ptr<const RectangularMesh3D>& src_mesh, const DataVector<const SrcT>& src_vec,
                                      const shared_ptr<const MeshD<3>>& dst_mesh, const InterpolationFlags& flags) {
        if (src_mesh->axis[0]->size() == 0 || src_mesh->axis[1]->size() == 0 || src_mesh->axis[2]->size() == 0)
            throw BadMesh("interpolate", "source mesh empty");
        return new LinearInterpolatedLazyDataImpl<DstT, RectilinearMesh3D, SrcT>(src_mesh, src_vec, dst_mesh, flags);
    }
};
 
template <typename SrcT, typename DstT>
struct InterpolationAlgorithm<RectangularMesh3D, SrcT, DstT, INTERPOLATION_NEAREST> {
    static LazyData<DstT> interpolate(const shared_ptr<const RectangularMesh3D>& src_mesh, const DataVector<const SrcT>& src_vec,
                                      const shared_ptr<const MeshD<3>>& dst_mesh, const InterpolationFlags& flags) {
        if (src_mesh->axis[0]->size() == 0 || src_mesh->axis[1]->size() == 0 || src_mesh->axis[2]->size() == 0)
            throw BadMesh("interpolate", "source mesh empty");
        return new NearestNeighborInterpolatedLazyDataImpl<DstT, RectilinearMesh3D, SrcT>(src_mesh, src_vec, dst_mesh, flags);
    }
};
 
 
 
template <typename SrcT, typename DstT, InterpolationMethod method>
struct InterpolationAlgorithm<typename std::enable_if<method != INTERPOLATION_DEFAULT, RectangularMesh3D::ElementMesh>::type, SrcT, DstT, method> {
    static LazyData<DstT> interpolate(const shared_ptr<const RectangularMesh3D::ElementMesh>& src_mesh, const DataVector<const SrcT>& src_vec,
                                      const shared_ptr<const MeshD<3>>& dst_mesh, const InterpolationFlags& flags) {
        return InterpolationAlgorithm<RectangularMesh3D, SrcT, DstT, method>::interpolate(src_mesh, src_vec, dst_mesh, flags);
    }
};
 
template <typename SrcT, typename DstT>
struct InterpolationAlgorithm<RectangularMesh3D::ElementMesh, SrcT, DstT, INTERPOLATION_NEAREST> {
    static LazyData<DstT> interpolate(const shared_ptr<const RectangularMesh3D::ElementMesh>& src_mesh, const DataVector<const SrcT>& src_vec,
                                      const shared_ptr<const MeshD<3>>& dst_mesh, const InterpolationFlags& flags) {
        if (src_mesh->axis[0]->size() == 0 || src_mesh->axis[1]->size() == 0)
            throw BadMesh("interpolate", "source mesh empty");
        return new NearestNeighborInterpolatedLazyDataImpl<DstT, RectangularMesh3D::ElementMesh, SrcT>(src_mesh, src_vec, dst_mesh, flags);
    }
};
 
 
PLASK_API shared_ptr<RectangularMesh3D > make_rectangular_mesh(const RectangularMesh3D &to_copy);
inline shared_ptr<RectangularMesh3D> make_rectangular_mesh(shared_ptr<const RectangularMesh3D> to_copy) { return make_rectangular_mesh(*to_copy); }
 
}   // namespace plask
 
#endif // PLASK__RECTANGULAR3D_H