/*
 * Decompiled with CFR 0.152.
 * 
 * Could not load the following classes:
 *  org.joml.Matrix3f
 *  org.joml.Matrix3fc
 *  org.joml.Vector3f
 *  org.joml.Vector3i
 */
package com.mojang.math;

import java.util.Arrays;
import net.minecraft.core.Direction;
import net.minecraft.util.Util;
import org.joml.Matrix3f;
import org.joml.Matrix3fc;
import org.joml.Vector3f;
import org.joml.Vector3i;

public enum SymmetricGroup3 {
    P123(0, 1, 2),
    P213(1, 0, 2),
    P132(0, 2, 1),
    P312(2, 0, 1),
    P231(1, 2, 0),
    P321(2, 1, 0);

    private final int p0;
    private final int p1;
    private final int p2;
    private final Matrix3fc transformation;
    private static final SymmetricGroup3[][] CAYLEY_TABLE;
    private static final SymmetricGroup3[] INVERSE_TABLE;

    private SymmetricGroup3(int p0, int p1, int p2) {
        this.p0 = p0;
        this.p1 = p1;
        this.p2 = p2;
        this.transformation = new Matrix3f().zero().set(this.permute(0), 0, 1.0f).set(this.permute(1), 1, 1.0f).set(this.permute(2), 2, 1.0f);
    }

    public SymmetricGroup3 compose(SymmetricGroup3 that) {
        return CAYLEY_TABLE[this.ordinal()][that.ordinal()];
    }

    public SymmetricGroup3 inverse() {
        return INVERSE_TABLE[this.ordinal()];
    }

    public int permute(int i) {
        return switch (i) {
            case 0 -> this.p0;
            case 1 -> this.p1;
            case 2 -> this.p2;
            default -> throw new IllegalArgumentException("Must be 0, 1 or 2, but got " + i);
        };
    }

    public Direction.Axis permuteAxis(Direction.Axis axis) {
        return Direction.Axis.VALUES[this.permute(axis.ordinal())];
    }

    public Vector3f permuteVector(Vector3f v) {
        float v0 = v.get(this.p0);
        float v1 = v.get(this.p1);
        float v2 = v.get(this.p2);
        return v.set(v0, v1, v2);
    }

    public Vector3i permuteVector(Vector3i v) {
        int v0 = v.get(this.p0);
        int v1 = v.get(this.p1);
        int v2 = v.get(this.p2);
        return v.set(v0, v1, v2);
    }

    public Matrix3fc transformation() {
        return this.transformation;
    }

    static {
        CAYLEY_TABLE = Util.make(() -> {
            SymmetricGroup3[] values = SymmetricGroup3.values();
            SymmetricGroup3[][] table = new SymmetricGroup3[values.length][values.length];
            for (SymmetricGroup3 first : values) {
                for (SymmetricGroup3 second : values) {
                    SymmetricGroup3 result;
                    int p0 = first.permute(second.p0);
                    int p1 = first.permute(second.p1);
                    int p2 = first.permute(second.p2);
                    table[first.ordinal()][second.ordinal()] = result = Arrays.stream(values).filter(p -> p.p0 == p0 && p.p1 == p1 && p.p2 == p2).findFirst().get();
                }
            }
            return table;
        });
        INVERSE_TABLE = Util.make(() -> {
            SymmetricGroup3[] values = SymmetricGroup3.values();
            return (SymmetricGroup3[])Arrays.stream(values).map(f -> Arrays.stream(SymmetricGroup3.values()).filter(s -> f.compose((SymmetricGroup3)((Object)((Object)((Object)s)))) == P123).findAny().get()).toArray(SymmetricGroup3[]::new);
        });
    }
}