blob: fa6d5753dd1d2deb53a49253f82edf354ed9b0c7 [file] [log] [blame]
#include "rapidjson/document.h"
#include "rapidjson/prettywriter.h"
#include "delaunator.h"
#include <cstdio>
#include <fstream>
#include <string>
#include <exception>
#include <vector>
#include <initializer_list>
// #include "prettyprint.hpp"
#include <iostream>
using namespace std;
namespace {
string read_file(const char* filename) {
ifstream input_file(filename);
if(input_file.good()) {
string json_str(
(istreambuf_iterator<char>(input_file)),
istreambuf_iterator<char>()
);
return json_str;
} else {
printf("Error reading file %s", filename);
throw exception();
}
}
vector<double> get_geo_json_points(const string& json) {
rapidjson::Document document;
if(document.Parse(json.c_str()).HasParseError()) {
fprintf(stderr, "Cannot parse JSON");
throw exception();
}
const rapidjson::Value& features = document["features"];
vector<double> coords;
// vector<double> y_vector;
for(rapidjson::SizeType i = 0; i < features.Size(); i++) {
const rapidjson::Value& coordinates = features[i]["geometry"]["coordinates"];
const double x = coordinates[0].GetDouble();
const double y = coordinates[1].GetDouble();
coords.push_back(x);
coords.push_back(y);
}
return coords;
}
const string serialize_to_json(Delaunator &delaunator) {
rapidjson::StringBuffer sb;
rapidjson::PrettyWriter<rapidjson::StringBuffer> writer(sb);
writer.StartObject();
writer.String("type"); writer.String("FeatureCollection");
writer.String("crs");
writer.StartObject();
writer.String("type"); writer.String("name");
writer.String("properties");
writer.StartObject();
writer.String("name"); writer.String("urn:ogc:def:crs:EPSG::900913");
writer.EndObject();
writer.EndObject();
writer.String("features");
writer.StartArray();
for(long int i = 0; i < delaunator.triangles.size(); i+=3) {
writer.StartObject();
writer.String("type"); writer.String("Feature");
writer.String("properties"); writer.StartObject(); writer.EndObject();
writer.String("geometry");
writer.StartObject();
writer.String("type"); writer.String("Polygon");
writer.String("coordinates");
writer.StartArray();
writer.StartArray();
writer.StartArray();
writer.Uint(delaunator.coords[2 * delaunator.triangles[i]]);
writer.Uint(delaunator.coords[2 * delaunator.triangles[i] + 1]);
writer.EndArray();
writer.StartArray();
writer.Uint(delaunator.coords[2 * delaunator.triangles[i + 1]]);
writer.Uint(delaunator.coords[2 * delaunator.triangles[i + 1] + 1]);
writer.EndArray();
writer.StartArray();
writer.Uint(delaunator.coords[2 * delaunator.triangles[i + 2]]);
writer.Uint(delaunator.coords[2 * delaunator.triangles[i + 2] + 1]);
writer.EndArray();
writer.StartArray();
writer.Uint(delaunator.coords[2 * delaunator.triangles[i]]);
writer.Uint(delaunator.coords[2 * delaunator.triangles[i] + 1]);
writer.EndArray();
writer.EndArray();
writer.EndArray();
writer.EndObject();
writer.EndObject();
}
writer.EndArray();
writer.EndObject();
return string(sb.GetString());
}
}
int main(int, char* argv[]) {
const char* filename = argv[1];
const char* output = argv[2];
string json = read_file(filename);
const vector<double> coords = get_geo_json_points(json);
Delaunator delaunator(move(coords));
const char* out_json = serialize_to_json(delaunator).c_str();
if (output) {
printf("Writing to file %s", output);
ofstream stream;
stream.open(output);
stream << out_json;
stream.close();
} else {
puts(out_json);
}
return 0;
}