#!/usr/bin/env python

'''

OrtoTex4XPL - Orto textures for X-Plane script

Author: Jan Bariencik (lktb@bari2.eu)

Version: 20160417

THIS SCRIPT IS FOR STUDYING PURPOSE ONLY!!! USE AT YOUR OWN RISK!!

Data are downloaded from Google Maps or Mapy.cz web service.

Plase, check Map source license conditions before use.

License:

This program is free software: you can redistribute it and/or modify

it under the terms of the GNU General Public License as published by

the Free Software Foundation, either version 3 of the License, or

(at your option) any later version.

This program is distributed in the hope that it will be useful,

but WITHOUT ANY WARRANTY; without even the implied warranty of

MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

GNU General Public License for more details.

You should have received a copy of the GNU General Public License

along with this program. If not, see <http://www.gnu.org/licenses/>.

'''

# TODO: follow ter folder

# get lat/lng coords from ter files inside folder

# TODO: add simple GUI

import argparse

import os

import platform

import sys

import time

import requests

import re

import cv2

import numpy as np

from PIL import Image

import thread

import time

import random

# https://raw.githubusercontent.com/hrldcpr/mercator.py/master/mercator.py

import mercator

# -------- Variables ----------------

# path to dsftool binary

DSFBIN="./DSFTool"

DDSBIN="./DDSTool"

# base output directory

OUTPUT="/tmp/orto4xpl/"

if platform.system() == "Windows":

DSFBIN="DSFTool.exe"

DSFBIN="DDSTool.exe"

OUTPUT="C:/orto4xpl"

# output sub directories

TEX="texture"

POL="pol"

DSF="Earth nav data"

TMP="tmp"

# -------- Classes ----------------

class ForkManager:

self.forks.append(pid)

class MapSource:

return 0

class GMaps(MapSource):

self.url_req='https://khms'+str(server)+'.google.com/kh/v='+str(self.c1)+'&src=app&x={1}&y={2}&z={0}&s=Gali'

class MapyCZ(MapSource):

self.url_req='http://m'+str(server)+'.mapserver.mapy.cz/ophoto-m/{0}-{1}-{2}'

# -------- Functions ----------------

def fixY(zl,y):

return y

def down_square(zl,x,y,pms,ams,cnt=8):

print "Downloading took", time.time()-start, "s"

def merge_square(zl,x,y,cnt=8):

print "Marging took", time.time()-start, "s"

def remove_logo(orig, child):

os.remove(child)

def create_pol(zl,x,y):

dsfin.close()

def convert_to_dds(tile_fname,dds_fname):

return newRef

def prepare_dsf():

print "-----------------------------------------------"

# -------- Main ----------------

parser = argparse.ArgumentParser()

parser.add_argument('--zl', metavar='<level>', type=int, default=18, help='required map zoom level (10-24) -- default: 18')

parser.add_argument('--src', choices=['gmaps', 'mapy'], default='gmaps', help='map source (mapy.cz, google maps) -- default: gmaps')

parser.add_argument('--base', metavar='<path>', default=OUTPUT, help='output directory base -- default: /tmp/orto4xpl')

#parser.add_argument('--coord1', nargs=2, type=float, metavar=('lat','lng'), default=[49.15296966,16.66625977], help='GPS lattitude and longtitude -- default: 49.15296966,16.66625977')

parser.add_argument('lat', type=float, help='GPS lattitude --- for ex. 49.15296966')

parser.add_argument('lng', type=float, help='GPS longtitude -- for ex. 16.66625977')

parser.add_argument('--coord2', nargs=2, type=float, metavar=('lat','lng'), help='Second GPS coordinates (lower right corner)')

parser.add_argument('--remove-logo', action='store_true', help='try to remove logo merging with higher zoom level')

parser.add_argument('--keep-downloaded', action='store_true', help='do not delete temporary img files after processing')

parser.add_argument('--dds-textures', action='store_true', help='enable converting textures to dds (instead of png)')

parser.add_argument('--dds-maxcpu', metavar='<num>', type=int, default=2, help='maximum CPU cores used for DDS converting')

parser.add_argument('--wed-import', action='store_true', help='enable creating DSF and POL files for import orto to WED')

parser.add_argument('--fix-missing', action='store_true', help='use alternative (google maps) source for missing tiles')

args = parser.parse_args()

Zl=args.zl

Lat1=args.lat

Lng1=args.lng

if args.coord2 is not None:

Lat2=args.coord2[0]

Lng2=args.coord2[1]

else:

Lat2=Lat1

Lng2=Lng1

if args.wed_import and not os.path.isfile(DSFBIN):

print "Cannot find DSF tool", DSFBIN

sys.exit(0)

if args.dds_textures and not os.path.isfile(DDSBIN):

print "Cannot find DDS tool", DDSBIN

sys.exit(0)

print "--- Input:"

print "Zl="+str(Zl)

print "Lat1="+str(Lat1)

print "Lng1="+str(Lng1)

print "Lat2="+str(Lat2)

print "Lng2="+str(Lng2)

if Lat2>Lat1 or Lng2<Lng1:

print "Info: Right corrner must be lower or equal to left"

parser.help();

sys.exit(0)

ams = None

if args.src == "gmaps":

pms = GMaps()

else:

pms = MapyCZ()

if args.fix_missing:

ams = GMaps()

OUTPUT=args.base+'/'

# output dirs

OUTPOL=OUTPUT+POL

OUTTEX=OUTPUT+TEX

OUTTMP=OUTPUT+TMP

OUTDSF=OUTPUT+DSF

# output filenames

FILEBASE="g2xpl_8_{0}_{1}_{2}"

FILEIMGTMP=OUTTMP+"/"+FILEBASE+"-{3}-{4}.jpg"

FILEIMG=OUTTEX+"/"+FILEBASE+".png"

FILEPOL=OUTPOL+"/"+FILEBASE+".pol"

print "--- Mercators"

xy1=mercator.get_lat_lng_tile(Lat1,Lng1,Zl)

print 'x1=',xy1[0],'y1=',xy1[1]

xyr1=((int(xy1[0])/8)*8, (int(xy1[1])/8)*8)

print 'xr1=',xyr1[0],'yr1=',xyr1[1]

xy2=mercator.get_lat_lng_tile(Lat2,Lng2,Zl)

print 'x2=',xy2[0],'y2=',xy2[1]

xyr2=((int(xy2[0])/8)*8, (int(xy2[1])/8)*8)

print 'xr2=',xyr2[0],'yr2=',xyr2[1]

print "--- Min/Max Lat/Lng"

latlng1=mercator.get_tile_box(Zl,xyr1[0],xyr1[1])

latlng2=mercator.get_tile_box(Zl,xyr2[0],xyr2[1])

print latlng1

print latlng2

print "--- Processing"

if not os.path.isdir(OUTPUT):os.mkdir(OUTPUT)

if not os.path.isdir(OUTTEX):os.mkdir(OUTTEX)

if not os.path.isdir(OUTTMP):os.mkdir(OUTTMP)

if args.wed_import:

if not os.path.isdir(OUTDSF):os.mkdir(OUTDSF)

if not os.path.isdir(OUTPOL):os.mkdir(OUTPOL)

fman = ForkManager(args.dds_maxcpu)

squares_start = time.time()

squares_cur = 0

squares_all = int((xyr2[1]+8-xyr1[1])*(xyr2[0]+8-xyr1[0])/64)

print "Total number of files to download: " + str((xyr2[1]+8-xyr1[1])*(xyr2[0]+8-xyr1[0])) + " (" + str(xyr2[0]+8-xyr1[0]) + "*" + str(xyr2[1]+8-xyr1[1]) + ")"

print "Total number of squares to create: " + str(squares_all)

for y in range(xyr1[1],xyr2[1]+8,8):

for x in range(xyr1[0],xyr2[0]+8,8):

tile_fname = FILEIMG.format(int(Zl),int(x),int(fixY(Zl,y)))

dds_fname = os.path.splitext(tile_fname)[0] + ".dds"

pol_fname = FILEPOL.format(int(Zl),int(x),int(fixY(Zl,y)));

child_fname= FILEIMG.format(int(Zl+1),int(x*2),int(fixY(Zl+1,y*2)))

if args.dds_textures and os.path.isfile(dds_fname):

print "- skipped file", dds_fname, "(already exits)"

squares_cur += 1

continue

# preparing child tile (zl+1) for removing logo

if args.remove_logo:

if (not os.path.isfile(child_fname)) and (not os.path.isfile(tile_fname)):

down_square(int(Zl+1),x*2,y*2,pms,ams,16)

merge_square(int(Zl+1),x*2,y*2,16)

else:

print "- skipped file", child_fname, "(already exits or merged)"

# preparing tile

if not os.path.isfile(tile_fname):

down_square(int(Zl),x,y,pms,ams)

merge_square(int(Zl),x,y)

else:

print "- skipped file", tile_fname, "(already exits)"

if args.wed_import and not os.path.isfile(pol_fname):

create_pol(int(Zl),x,y)

# removing logo

if args.remove_logo and os.path.isfile(child_fname):

remove_logo(tile_fname,child_fname)

# convert img to dds

if args.dds_textures:

fman.wait_slot()

fman.add_fork(convert_to_dds(tile_fname,dds_fname))

squares_cur += 1

print "*** Finished " + str(squares_cur/(squares_all/100)) + "% (" + str(squares_cur) + " of " + str(squares_all) + ") squares in time", time.time()-squares_start

if args.wed_import: prepare_dsf()

fman.wait_finish()