EntityJSON.cpp

#include <EntityJSON.h>

#include <boost/property_tree/ptree.hpp>
#include <boost/property_tree/json_parser.hpp>

#define UNUSED(x) (void)(x)

void GenerateEntityJSON(MinecraftWorld &world, std::string filename)
{

    boost::property_tree::ptree tree;

    tree.put("type", "FeatureCollection");

    boost::property_tree::ptree features;

    int num = 0;
    int natural = 0;
    for (auto i1 : world.chunk_entities) {
        int chunkx = i1.first;
        UNUSED(chunkx);
        for(auto i2 : i1.second) {
            int chunkz = i2.first;
            UNUSED(chunkz);
            for(auto entity : i2.second) {
                num++;
                auto ns = entity->get_byte_property("NaturalSpawn");
                if (ns and *ns) {
                    natural++;
                }

                std::cout << "got entity of type: " << entity->get_type() << " at ";
                std::cout << entity->get_position().get_string() << " and rot: ";
                std::cout << entity->get_rotation().get_string();
                std::cout << " uid: " << entity->get_unique_id();

                for(auto def : entity->get_definitions().get_defs()) {
                    std::cout << " " << def;
                }
                std::cout << "\n";;

                boost::property_tree::ptree feature_node;

                feature_node.put("type", "Feature");
                feature_node.put("properties.type", entity->get_type());

                if (entity->get_type() == "minecraft:item") {
                    std::cout << "more info " << entity->get_string() << "\n";
                    feature_node.put("properties.item_type", entity->get_name());
                    if (entity->get_name() == "") {
                        std::cout << "no subtype\n";
                    }
                }

                // this seems like a goofy way to have a list.
                boost::property_tree::ptree coords;
                boost::property_tree::ptree coord;

                // the 0 coord is x, 1 coord is y (the height), 2 coord is z.

                // this first list is the full coordinates for reporting.
                for (int i : {0,1,2}) {
                    coord.put("", entity->get_position().get_value(i));
                    coords.push_back(std::make_pair("", coord));
                }
                feature_node.add_child("properties.coords", coords);

                feature_node.put("geometry.type", "Point");

                // this second list of coordinates is for the geojson format.
                // 0(x) and 2(z) are the normal "xy" coordinates
                coords.clear();
                for (int i : {0,2}) {
                    coord.put("", entity->get_position().get_value(i));
                    coords.push_back(std::make_pair("", coord));
                }

                feature_node.add_child("geometry.coordinates", coords);

                features.push_back(std::make_pair("", feature_node));
            }
        }
    }

    tree.add_child("features", features);

    boost::property_tree::write_json(filename, tree);

    std::cout << "wrote " << num << " entities\n";
    std::cout << "wrote " << natural << " natural spawns\n";

}