#ifndef OERSTED_FINITEELEMENTMESH_H
#define OERSTED_FINITEELEMENTMESH_H

#include <vector>

#include "MatrixGroup.h"
#include "Node.h"
#include "Triangle.h"

template<size_t D, size_t P>
class FiniteElementMesh {};

template<size_t P>
class FiniteElementMesh<2,P> {
public:
    FiniteElementMesh() {};

    FiniteElementMesh(std::vector<XY> nodes, std::vector<Triangle<P>> tris) : Nodes(nodes), Triangles(tris) {};

    std::vector<XY> const nodes() const { return Nodes; };

    std::vector<Triangle<P>> const triangles() const { return Triangles; };

    template<size_t Q>
    DiagonalMatrixGroup<TriangleQuadratureRule<Q>::size> determinant() const {
        DiagonalMatrixGroup<TriangleQuadratureRule<Q>::size> mat(Triangles.size());

        for (size_t i = 0; i != Triangles.size(); ++i) {
            Triangles[i].determinant<Q>(mat, Nodes);
        }

        return mat;
    }

    template<size_t Q>
    SparseMatrixGroup<TriangleQuadratureRule<Q>::size> basis() const {
        SparseMatrixGroup<TriangleQuadratureRule<Q>::size> mat(Nodes.size(), Triangles.size(), Triangle<P>::NumNodes);

        for (size_t i = 0; i != Triangles.size(); ++i) {
            Triangles[i].basis<Q>(mat, Nodes);
        }

        return mat;
    }

    template<size_t Q>
    DerivativeMatrixGroup<TriangleQuadratureRule<Q>::size> derivative() const {
        DerivativeMatrixGroup<TriangleQuadratureRule<Q>::size> df(Nodes.size(), Triangles.size(), Triangle<P>::NumNodes);

        for (size_t i = 0; i!= Triangles.size();++i) {
            Triangles[i].derivative<Q>(df, Nodes);
        }

        return df;
    }

protected:
    std::vector<XY> Nodes;
    std::vector<Triangle<P>> Triangles;
};

#endif //OERSTED_FINITEELEMENTMESH_H