#pragma once

#include <sstream>
#include <stddef.h>
#include <iostream>

template <size_t R, size_t C, typename T = float>
struct Matrix {
public:
    Matrix<R, C, T>() : elements{} {};

    template<typename ...Args>
    explicit Matrix<R, C, T>(Args... args): elements{ args... } {};

    Vector<C, T> row(size_t index) {
        return { &elements[index * C] };
    }

    Vector<R, T> col(size_t index) {
        Vector<R, T> result{};
        for (int i = 0; i < R; i++) {
            result[i] = this->operator()(index, i);
        }
        return result;
    }

    template<size_t N>
    Matrix<R, N, T> operator*(Matrix<C, N, T> other) {
        Matrix<R, N, T> result{};
        for (int y = 0; y < R; y++) {
            for (int x = 0; x < N; x++) {
                auto r = row(y);
                auto c = other.col(x);

                auto dot = r * c;

                std::cout << x << ", " << y << ": "
                    << "(" << r.string() << ", " << c.string() << ")"
                    << dot << std::endl;

                result(x, y) = dot;
            }
        }
        return result;
    }

    auto& operator()(size_t x, size_t y) {
        return elements[y * C + x];
    }

    std::string string() {
        std::stringstream str{};
        for (int x = 0; x < R; x++) {
            for (int y = 0; y < C; y++) {
                str << this->operator()(x, y) << " ";
            }

            str << "\n";
        }

        return str.str();
    }

    T elements[R * C];
};