splitter.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import os
import configparser
import shutil


WORK_DIR = os.path.expanduser('~') + "/map_build/"


def info(msg):
    print("II: " + msg)


def warn(msg):
    print("WW: " + msg)


def write_config():
    with open('pygmap3.cfg', 'w') as configfile:
        config.write(configfile)


config = configparser.ConfigParser()


# split raw-data
def split():
    os.chdir(WORK_DIR)
    config.read('pygmap3.cfg')
    region = config['runtime']['region']
    splitter = config['splitter']['rev'] + "/splitter.jar "

    # Java HEAP, RAM oder Mode
    if config['java']['agh'] == "1":
        heap = " -XX:+AggressiveHeap "
    else:
        heap = (config['java']['xmx'] + ' ' + config['java']['xms'])

    java_opts = "java -ea " + heap + " -jar " + WORK_DIR + splitter
    log_opts = " > splitter.log "

    # splitter-options
    BUILD_O5M = WORK_DIR + "o5m/" + region + ".o5m"

    sea = " "

    if config.has_option('runtime', 'no_sea'):
        pass
    elif config.has_option('precomp', 'sea'):
        sea_zip = WORK_DIR + "precomp/" + config['precomp']['sea']
        if os.path.exists(sea_zip):
            sea = " --precomp-sea=" + sea_zip

    cities15000 = WORK_DIR + "cities15000.zip"

    if os.path.exists(cities15000):
        geonames = " --geonames-file=" + cities15000
    else:
        geonames = " "

    splitter_opts = (geonames +
                     " --mapid=" + config[region]['mapid'] + "0001 "
                     + " --output=o5m "
                     + sea
                     + " --write-kml=" + region + ".kml "
                     + " --keep-complete "
                     + " --overlap=0 ")

    # maxnodes
    if config.has_option('maxnodes', region):
        maxnodes = config['maxnodes'][region]
    else:
        maxnodes = config['maxnodes']['default']

    # splitter.jar command_line
    command_line = (java_opts
                    + log_opts
                    + splitter_opts
                    + " --max-nodes=" + maxnodes + " "
                    + BUILD_O5M)

    if config.has_option('runtime', 'verbose'):
        print()
        info(command_line)
        print()

    os.chdir("tiles")
    os.system(command_line)

    if config.has_option('runtime', 'logging'):
        log_dir = WORK_DIR + "log/splitter/" + region

        if os.path.exists(log_dir):
            path = log_dir
            for file in os.listdir(path):
                if os.path.isfile(os.path.join(path, file)):
                    os.remove(os.path.join(path, file))
        else:
            os.makedirs(log_dir)

        for i in ['densities-out.txt',
                  'template.args',
                  'areas.poly',
                  'splitter.log',
                  'areas.list']:
            if os.path.exists(i):
                shutil.copy2(i, log_dir)

    file = open(region + "_split.ready", "w")
    file.close()