src/Helper/RotateReflect2D.hpp

• View this file on GitHub
• Last update: 2020-05-10 22:28:28+09:00
• Include: #include "src/Helper/RotateReflect2D.hpp"

Code

template <class T> std::vector<std::vector<T>> rotate_2D(const std::vector<std::vector<T>>& t, bool counterclockwise = true) {
    int H = t.size(), W = t[0].size();
    std::vector<std::vector<T>> ret(W, std::vector<T>(H));
    for (int i = 0; i < H; i++) {
        for (int j = 0; j < W; j++) {
            ret[j][i] = counterclockwise ? t[i][W - 1 - j] : t[H - 1 - i][j];
        }
    }
    return ret;
}

template <class T> std::vector<std::vector<T>> reflect_2D(const std::vector<std::vector<T>>& t) {
    int H = t.size(), W = t[0].size();
    std::vector<std::vector<T>> ret(W, std::vector<T>(H));
    for (int i = 0; i < H; i++) {
        for (int j = 0; j < W; j++) {
            ret[j][i] = t[i][j];
        }
    }
    return ret;
}

#line 1 "src/Helper/RotateReflect2D.hpp"
template <class T> std::vector<std::vector<T>> rotate_2D(const std::vector<std::vector<T>>& t, bool counterclockwise = true) {
    int H = t.size(), W = t[0].size();
    std::vector<std::vector<T>> ret(W, std::vector<T>(H));
    for (int i = 0; i < H; i++) {
        for (int j = 0; j < W; j++) {
            ret[j][i] = counterclockwise ? t[i][W - 1 - j] : t[H - 1 - i][j];
        }
    }
    return ret;
}

template <class T> std::vector<std::vector<T>> reflect_2D(const std::vector<std::vector<T>>& t) {
    int H = t.size(), W = t[0].size();
    std::vector<std::vector<T>> ret(W, std::vector<T>(H));
    for (int i = 0; i < H; i++) {
        for (int j = 0; j < W; j++) {
            ret[j][i] = t[i][j];
        }
    }
    return ret;
}

Back to top page