splitter.cpp

#include <stdlib.h>
#include <stdio.h>
#include <errno.h>

#include <getopt.h>
#include <unistd.h>

#define OSMIUM_MAIN
#define OSMIUM_WITH_PBF_INPUT
#define OSMIUM_WITH_XML_INPUT
#define OSMIUM_WITH_PBF_OUTPUT
#define OSMIUM_WITH_XML_OUTPUT
#include <osmium.hpp>
#include <osmium/output/pbf.hpp>
#include <osmium/output/xml.hpp>

#include <geos/geom/MultiPolygon.h>
#include <geos/algorithm/locate/IndexedPointInAreaLocator.h>

#include "softcut.hpp"
#include "hardcut.hpp"

template <class TExtractInfo> bool readConfig(char *conffile, CutInfo<TExtractInfo> &info);

int main(int argc, char *argv[]) {
    bool softcut = true;
    bool debug = false;
    char *filename, *conffile;

    static struct option long_options[] = {
        {"debug",               no_argument, 0, 'd'},
        {"softcut",             no_argument, 0, 's'},
        {"hardcut",             no_argument, 0, 'h'},
        {0, 0, 0, 0}
    };

    while (1) {
        int c = getopt_long(argc, argv, "dsh", long_options, 0);
        if (c == -1)
            break;

        switch (c) {
            case 'd':
                debug = true;
                break;
            case 's':
                softcut = true;
                break;
            case 'h':
                softcut = false;
                break;
        }
    }

    if (optind > argc-2) {
        fprintf(stderr, "Usage: %s [OPTIONS] OSMFILE CONFIGFILE\n", argv[0]);
        return 1;
    }

    filename = argv[optind];
    conffile = argv[optind+1];

    if(softcut & !strcmp(filename, "-")) {
        fprintf(stderr, "Can't read from stdin when in softcut\n");
        return 1;
    }

    Osmium::OSMFile infile(filename);

    if(softcut) {
        SoftcutInfo info;
        if(!readConfig(conffile, info))
        {
            fprintf(stderr, "error reading config\n");
            return 1;
        }

        SoftcutPassOne one(&info);
        one.debug = debug;
        Osmium::Input::read(infile, one);

        SoftcutPassTwo two(&info);
        two.debug = debug;
        Osmium::Input::read(infile, two);
    } else {
        HardcutInfo info;
        if(!readConfig(conffile, info))
        {
            fprintf(stderr, "error reading config\n");
            return 1;
        }

        Hardcut cutter(&info);
        cutter.debug = debug;
        Osmium::Input::read(infile, cutter);
    }

    return 0;
}

template <class TExtractInfo> bool readConfig(char *conffile, CutInfo<TExtractInfo> &info) {
    const int linelen = 4096;

    FILE *fp = fopen(conffile, "r");
    if(!fp) {
        fprintf(stderr, "unable to open config file %s\n", conffile);
        return false;
    }

    char line[linelen];
    while(fgets(line, linelen-1, fp)) {
        line[linelen-1] = '\0';
        if(line[0] == '#' || line[0] == '\r' || line[0] == '\n' || line[0] == '\0')
            continue;

        int n = 0;
        char *tok = strtok(line, "\t ");

        const char *name = NULL;
        double minlon = 0, minlat = 0, maxlon = 0, maxlat = 0;
        char type = '\0';
        char file[linelen];

        while(tok) {
            switch(n) {
                case 0:
                    name = tok;
                    break;

                case 1:
                    if(0 == strcmp("BBOX", tok))
                        type = 'b';
                    else if(0 == strcmp("POLY", tok))
                        type = 'p';
                    else if(0 == strcmp("OSM", tok))
                        type = 'o';
                    else {
                        type = '\0';
                        fprintf(stderr, "output %s of type %s: unknown output type\n", name, tok);
                        return false;
                    }
                    break;

                case 2:
                    switch(type) {
                        case 'b':
                            if(4 == sscanf(tok, "%lf,%lf,%lf,%lf", &minlon, &minlat, &maxlon, &maxlat)) {
                                info.addExtract(name, minlat, minlon, maxlat, maxlon);
                            } else {
                                fprintf(stderr, "error reading BBOX %s for %s\n", tok, name);
                                return false;
                            }
                            break;
                        case 'p':
                            if(1 == sscanf(tok, "%s", file)) {
                                geos::geom::Geometry *geom = OsmiumExtension::GeometryReader::fromPolyFile(file);
                                if(!geom) {
                                    fprintf(stderr, "error creating geometry from poly-file %s for %s\n", file, name);
                                    break;
                                }
                                info.addExtract(name, geom);
                            }
                            break;
                        case 'o':
                            if(1 == sscanf(tok, "%s", file)) {
                                geos::geom::Geometry *geom = OsmiumExtension::GeometryReader::fromOsmFile(file);
                                if(!geom) {
                                    fprintf(stderr, "error creating geometry from poly-file %s for %s\n", file, name);
                                    break;
                                }
                                info.addExtract(name, geom);
                            }
                            break;
                    }
                    break;
            }

            tok = strtok(NULL, "\t ");
            n++;
        }
    }
    fclose(fp);
    return true;
}