from pdal import Pipeline
import json
import traceback
import geopandas as gpd
import numpy as np
import sys
import os
import subprocess

global puntosIN
global buffer
global entwine

args=json.loads(sys.argv[1])
#args['path_entwine']
#args['disk']
#args['path_geojson']
#args['buffer']
#puntosIN=gpd.read_file("./in/"+args['path_geojson'])
buffer=gpd.read_file("./in/"+args['buffer'])
entwine="./in/"+args['path_entwine']

#{"path_entwine":"../DATA4/entwine_updated_Andalucia_Bloque1M/ept.json","disk":"","path_geojson":"./puntos.geojson","buffer":"2"}
#puntosIN=gpd.read_file('./puntos.geojson')
#buffer=2
#entwine="../DATA4/entwine_updated_Andalucia_Bloque1M/ept.json"
#"../DATA4/entwine_updated_Andalucia_Bloque1M/ept.json"
#"./puntos.geojson"

def pdal(i,geom):
	#bounds=puntosIN.total_bounds
	#buffered=puntosIN.buffer(buffer,1)
	#chull_multipoly=buffered.concave_hull()
	#chull_multipoly.to_file('./chull_multipoly.geojson',driver="GeoJSON")
	#quit()
#	print(buffered)
	#polygons=list(map(asigna,buffered))
	#buffered.to_file('./buffered.geojson',driver="GeoJSON")
	#quit()
	#print(buffered[0])
	#print(puntosIN)
	#print(np.column_stack((puntosIN['X'],puntosIN['Y'])))

	jsonZ=[]
	readers=   {
	      "type": "readers.ept",
	      "filename": entwine,
	      "polygon": str(geom)
	   }

	writers= {
		"type": "writers.las",
		"filename": "./in/"+args['buffer'].rsplit('/',1)[0]+'/'+args['identificador']+'/'+args['buffer'].rsplit('/',1)[1].rsplit('.',1)[0]+str(i)+'.laz'
	}
	jsonZ.append(readers)
	jsonZ.append(writers)
	peticion = json.dumps(jsonZ)
	pipeline = Pipeline(peticion)
	count = pipeline.execute()
	arrays = pipeline.arrays
	#print('arrays '+ str(arrays))
	metadata = pipeline.metadata
	#print('metadata '+ str(metadata))
	log = pipeline.log
	print(log)

def joinlaz():
	origen = "./in/"+args['buffer'].rsplit('/',1)[0]+'/'+args['identificador']+'/*.laz'
	destino = "./in/"+args['buffer'].rsplit('/',1)[0]+'/'+args['buffer'].rsplit('/',1)[1].rsplit('.',1)[0]+'.laz'
	json_to_save = [
		origen,
		destino
		]
	peticion = json.dumps(json_to_save)
	pipeline = Pipeline(peticion)
	count = pipeline.execute()
	arrays = pipeline.arrays
	print('arrays '+ str(arrays))
	metadata = pipeline.metadata
	print('metadata '+ str(metadata))
	log = pipeline.log
	print('log '+ str(log))

if __name__ == '__main__':
	try:
		subprocess.run('mkdir {}'.format("./in/"+args['buffer'].rsplit('/',1)[0]+'/'+args['identificador']),shell=True)
		for i,geom in enumerate(buffer['geometry']):
			pdal(i,geom)
			joinlaz()
	except Exception:
		traceback.print_exc()