#include <gtest/gtest.h>

#include <common/fs.hh>
#include <common/decompile.hh>
#include <common/bsputils.hh>
#include <qbsp/map.hh>
#include <bsputil/bsputil.hh>

#include <fstream>

#include "testmaps.hh"
#include "test_qbsp.hh"


TEST(bsputil, q1DecompilerTest)
{
    const auto [bsp, bspx, prt] = LoadTestmapQ1("q1_decompiler_test.map");

    auto path = std::filesystem::path(testmaps_dir) / "q1_decompiler_test-decompile.map";
    std::ofstream f(path);

    decomp_options options;
    DecompileBSP(&bsp, options, f);

    f.close();

    // checks on the .map file
    auto &entity = LoadMapPath(path);
    EXPECT_EQ(entity.mapbrushes.size(), 7); // two floor brushes

    // qbsp the decompiled map
    const auto [bsp2, bspx2, prt2] = LoadTestmapQ1("q1_decompiler_test-decompile.map");

    EXPECT_EQ(bsp2.dmodels.size(), bsp.dmodels.size());
    EXPECT_EQ(bsp2.dleafs.size(), bsp.dleafs.size());
    EXPECT_EQ(bsp2.dnodes.size(), bsp.dnodes.size());

    for (int i = 0; i < bsp.dmodels[0].numfaces; ++i) {
        auto *face = &bsp.dfaces[bsp.dmodels[0].firstface + i];
        auto *face_texinfo = Face_Texinfo(&bsp, face);
        const qvec3d face_centroid = Face_Centroid(&bsp, face);
        const qvec3d face_normal = Face_Normal(&bsp, face);

        auto *face2 = BSP_FindFaceAtPoint(&bsp2, &bsp2.dmodels[0], face_centroid, face_normal);
        ASSERT_TRUE(face2);

        auto *face2_texinfo = Face_Texinfo(&bsp2, face2);
        EXPECT_EQ(face2_texinfo->vecs, face_texinfo->vecs);
    }
}

TEST(bsputil, extractTextures)
{
    const auto [bsp, bspx, prt] = LoadTestmapQ1("q1_extract_textures.map");

    // extract .bsp textures to test.wad
    std::ofstream wadfile("test.wad", std::ios::binary);
    ExportWad(wadfile, &bsp);

    // reload .wad
    fs::clear();
    img::clear();

    auto ar = fs::addArchive("test.wad");
    ASSERT_TRUE(ar);

    for (std::string texname : {"*swater4", "bolt14", "sky3", "brownlight"}) {
        fs::data data = ar->load(texname);
        ASSERT_TRUE(data);
        auto loaded_tex = img::load_mip(texname, data, false, bspver_q1.game);
        EXPECT_TRUE(loaded_tex);
    }
}