topology.h

Go to the documentation of this file.

// This file is part of MARTY.
//
// MARTY 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, either version 3 of the License, or
// (at your option) any later version.
//
// MARTY 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.
//
// You should have received a copy of the GNU General Public License
// along with MARTY. If not, see <https://www.gnu.org/licenses/>.

#pragma once

#include <iostream>

namespace mty::Topology {

    enum Type {
        Tadpole  = (1 << 0), 
        Mass     = (1 << 1), 
        Triangle = (1 << 2), 
        Box      = (1 << 3), 
        Pentagon = (1 << 4), 
        Any = Tadpole | Mass | Triangle | Box | Pentagon
    };

    inline constexpr auto topologyList = {
        Tadpole, Mass, Triangle, Box, Pentagon
    };

    inline constexpr auto names = {
        "Tadpole", "Mass", "Triangle", "Box", "Pentagon"
    };

    inline Type legsToTopology(int nLegs) {
        return static_cast<Type>(
                ((nLegs == 1) * Tadpole)
                | ((nLegs == 2) * Mass)
                | ((nLegs == 3) * Triangle)
                | ((nLegs == 4) * Box)
                | ((nLegs == 5) * Pentagon)
                );
    }

    inline std::ostream &operator<<(std::ostream &out, Type t) {
        if (t == Any) out << "Any";
        else          out << *(names.begin() + static_cast<int>(t)/2);
        return out;
    }
}