add cli tool for rendering a map to an image

2 files changed, 163 insertions, 0 deletions

diff --git a/bin/render.cpp b/bin/render.cpp
new file mode 100644
index 0000000000..d545c09292
--- /dev/null
+++ b/bin/render.cpp
@@ -0,0 +1,109 @@
+#include <mbgl/map/map.hpp>
+#include <mbgl/util/image.hpp>
+#include <mbgl/util/std.hpp>
+#include <mbgl/util/io.hpp>
+
+#include <rapidjson/document.h>
+#include <rapidjson/writer.h>
+#include <rapidjson/stringbuffer.h>
+
+#include <mbgl/platform/default/headless_view.hpp>
+#include <mbgl/platform/default/headless_display.hpp>
+#include <mbgl/storage/caching_http_file_source.hpp>
+
+#if __APPLE__
+#include <mbgl/platform/darwin/log_nslog.hpp>
+#else
+#include <mbgl/platform/default/log_stderr.hpp>
+#endif
+
+#include <boost/program_options.hpp>
+namespace po = boost::program_options;
+
+#include <cassert>
+#include <cstdlib>
+#include <iostream>
+
+int main(int argc, char *argv[]) {
+
+    std::string style_path;
+    double lat = 0, lon = 0;
+    double zoom = 0;
+    double bearing = 0;
+
+    int width = 256;
+    int height = 256;
+    double pixelRatio = 1.0;
+    const char *output = "out.png";
+    std::vector<std::string> classes;
+    std::string token;
+
+    po::options_description desc("Allowed options");
+    desc.add_options()
+        ("style,s", po::value(&style_path)->required()->value_name("json"),"Map stylesheet")
+        ("lon,x", po::value(&lon)->value_name("degrees"), "Longitude")
+        ("lat,y", po::value(&lat)->value_name("degrees"), "Latitude in degrees")
+        ("zoom,z", po::value(&zoom)->value_name("number"), "Zoom level")
+        ("bearing,b", po::value(&bearing)->value_name("degrees"), "Bearing")
+        ("width,w", po::value(&width)->value_name("pixels"), "Image width")
+        ("height,h", po::value(&height)->value_name("pixels"), "Image height")
+        ("class,c", po::value(&classes)->value_name("name"), "Class name")
+        ("token,t", po::value(&token)->value_name("key"), "Mapbox access token")
+    ;
+
+    try {
+        po::variables_map vm;
+        po::store(po::parse_command_line(argc, argv, desc), vm);
+        po::notify(vm);
+    } catch(std::exception& e) {
+        std::cout << "Error: " << e.what() << std::endl << desc;
+        exit(1);
+    }
+
+    std::string style = mbgl::util::read_file(style_path);
+
+    using namespace mbgl;
+
+
+#if __APPLE__
+    Log::Set<NSLogBackend>();
+#else
+    Log::Set<StderrLogBackend>();
+#endif
+
+    CachingHTTPFileSource fileSource("");
+
+    // Try to load the token from the environment.
+    if (!token.size()) {
+        token = getenv("MAPBOX_ACCESS_TOKEN");
+    }
+
+    // Set access token if present
+    if (token.size()) {
+        fileSource.setAccessToken(std::string(token));
+    }
+
+    HeadlessView view;
+    Map map(view, fileSource);
+
+    map.setStyleJSON(style, ".");
+    map.setAppliedClasses(classes);
+
+    view.resize(width, height, pixelRatio);
+    map.resize(width, height, pixelRatio);
+    map.setLonLatZoom(lon, lat, zoom);
+    map.setBearing(bearing);
+
+    std::unique_ptr<uint32_t[]> pixels;
+
+    // Run the loop. It will terminate when we don't have any further listeners.
+    map.run();
+
+    // Get the data from the GPU.
+    pixels = view.readPixels();
+
+    const unsigned int w = width * pixelRatio;
+    const unsigned int h = height * pixelRatio;
+    const std::string image = util::compress_png(w, h, pixels.get());
+    util::write_file(output, image);
+}
diff --git a/bin/render.gyp b/bin/render.gyp
new file mode 100644
index 0000000000..9fb6066eb5
--- /dev/null
+++ b/bin/render.gyp
@@ -0,0 +1,54 @@
+{
+  'includes': [
+    '../gyp/common.gypi',
+    '../gyp/version.gypi',
+    '../gyp/mbgl-platform.gypi',
+  ],
+  'targets': [
+    {
+      'target_name': 'mbgl-render',
+      'product_name': 'mbgl-render',
+      'type': 'executable',
+      'sources': [
+        './render.cpp',
+      ],
+      'variables' : {
+        'cflags': [
+          '<@(uv_cflags)',
+          '<@(png_cflags)',
+          '-I<(boost_root)/include',
+        ],
+        'ldflags': [
+          '<@(glfw3_ldflags)',
+          '<@(uv_ldflags)',
+          '<@(sqlite3_ldflags)',
+          '<@(curl_ldflags)',
+          '<@(png_ldflags)',
+          '<@(uv_static_libs)',
+          '-L<(boost_root)/lib',
+          '-lboost_program_options'
+        ],
+      },
+      'conditions': [
+        # add libuv include path and OpenGL libs
+        ['OS == "mac"',
+          {
+            'xcode_settings': {
+              'OTHER_CPLUSPLUSFLAGS': ['<@(cflags)'],
+              'OTHER_LDFLAGS': ['<@(ldflags)'],
+            },
+          },
+          {
+            'cflags': ['<@(cflags)'],
+            'libraries': ['<@(ldflags)'],
+          }],
+      ],
+      'include_dirs': [ '../src' ],
+      'dependencies': [
+        '../mapboxgl.gyp:mbgl-core',
+        '../mapboxgl.gyp:mbgl-headless',
+        '<(platform_library)',
+      ],
+    },
+  ],
+}