axes.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__AXES_H
#define PLASK__AXES_H
 
#include <string>
#include <map>
#include "geometry/primitives.hpp"
 
#include "vector/3d.hpp"
 
namespace plask {
 
struct PLASK_API AxisNames {
 
    struct PLASK_API Register {
        std::map<std::string, AxisNames> axisNames;
 
        Register() {}
 
        template<typename... Params>
        Register(const std::string& c0_name, const std::string& c1_name, const std::string& c2_name, const Params&... names) {
            this->operator()(c0_name, c1_name, c2_name, names...);
        }
 
        void addname(const std::string& c0_name, const std::string& c1_name, const std::string& c2_name, const std::string& name) {
            axisNames[name] = AxisNames(c0_name, c1_name, c2_name);
        }
 
        Register& operator()(const std::string& c0_name, const std::string& c1_name, const std::string& c2_name) {
            addname(c0_name, c1_name, c2_name, c0_name + c1_name + c2_name);
            return *this;
        }
 
        template<typename Param1>
        Register& operator()(const std::string& c0_name, const std::string& c1_name, const std::string& c2_name, const Param1& name) {
            addname(c0_name, c1_name, c2_name, name);
            return this->operator()(c0_name, c1_name, c2_name);
        }
 
        template<typename Param1, typename... Params>
        Register& operator()(const std::string& c0_name, const std::string& c1_name, const std::string& c2_name, const Param1& firstName, const Params&... names) {
            addname(c0_name, c1_name, c2_name, firstName);
            return this->operator()(c0_name, c1_name, c2_name, names...);
        }
 
        const AxisNames& get(const std::string& name) const;
    };
 
    static Register axisNamesRegister;
 
    static const AxisNames& getAbsoluteNames();
 
    std::string byIndex[3];
 
    AxisNames() {}
 
    AxisNames(const std::string& c0_name, const std::string& c1_name, const std::string& c2_name);
 
    const std::string& operator[](const std::size_t i) const { return byIndex[i]; }
 
    std::size_t operator[](const std::string& name) const;
 
    Primitive<3>::Direction get3D(const std::string& name) const;
 
    Primitive<2>::Direction get2D(const std::string& name) const;
 
    template <int DIMS>
    typename Primitive<DIMS>::Direction get(const std::string& name) const;
 
    std::string str() const;
 
    std::string getNameForLong() const { return operator [](axis::lon_index); }
 
    std::string getNameForTran() const { return operator [](axis::tran_index); }
 
    std::string getNameForVert() const { return operator [](axis::up_index); }
 
    bool operator==(const AxisNames& to_compare) const;
 
    bool operator!=(const AxisNames& to_compare) const { return !(*this == to_compare); }
};
 
template <> inline Primitive<2>::Direction AxisNames::get<2>(const std::string& name) const { return this->get2D(name); }
template <> inline Primitive<3>::Direction AxisNames::get<3>(const std::string& name) const { return this->get3D(name); }
 
 
 
} // namespace plask
 
#endif // PLASK__AXES_H