#pragma once

#include <mfem.hpp>

#include "serif/discretization/domain/concepts.hpp"
#include "serif/discretization/domain/schema/concepts.hpp"

namespace serif::discretization::domain::schema {
    // Build the 0/1 marker array MFEM wants when an integrator or an essential
    // BC should apply only on part of the mesh.
    template <IsDomainOrSet DomainT, IsSchema SchemaT>
    [[nodiscard]] mfem::Array<int> make_attribute_marker(const mfem::Mesh &mesh) {
        static_assert(
            SchemaT::template contains_domain<DomainT>(),
            "Requested marker domain is not completely registered in the supplied DomainSchema."
        );

        mfem::Array<int> marker(mesh.attributes.Max());
        for (int domainID = 1; domainID <= marker.Size(); ++domainID) {
            marker[domainID - 1] = SchemaT::template domain_id_belongs_to<DomainT>(domainID) ? 1 : 0;
        }
        return marker;
    }
}