from laspy.file import File
import numpy as np
import os,imageio
from scipy.ndimage.filters import gaussian_laplace
from osgeo import gdal

def log(archiv):
    a=imageio.imread('../DEMS/DEM_'+archiv[:-3]+'tif')
    b=gaussian_laplace(a,2)
    imageio.imwrite('../LOG/LOG_'+archiv[:-3]+'tif',b)

hechos=os.listdir('../HILLSH')


for arch in os.listdir('../LAZ'):
    print arch
    if 'laz' in arch and 'HILLSH_'+arch[:-3]+'tif' not in hechos:
         os.system('rm ../*.las')
         print 'laszip -i ../LAZ/'+arch+' -o ../LAZ/'+arch[:-3]+'las'
         os.system('laszip -i ../LAZ/'+arch+' -o ../LAZ/'+arch[:-3]+'las')
         inFile = File('../LAZ/'+arch[:-3]+'las', mode='r')
         I = inFile.Classification == 2
         outFile = File('tmp_'+arch[:-3]+'las', mode='w', header=inFile.header)
         outFile.points = inFile.points[I]
         outFile.close()
         os.system('points2grid --resolution 1 -r 5 --idw --input_format las --output_format arc -i '+'tmp_'+arch[:-3]+'las'+' -o tmp')
         os.system('saga_cmd grid_tools 7 -INPUT tmp.idw.asc -RESULT tmp')
         os.system('saga_cmd io_gdal 2 -GRIDS tmp.sdat -FILE ../DEMS/DEM_'+arch[:-3]+'tif')
         os.system('/usr/local/Cellar/gdal2-python/2.2.4/bin/gdal_edit.py -a_srs EPSG:25830 ../DEMS/DEM_'+arch[:-3]+'tif')
         os.system('gdaldem hillshade ../DEMS/DEM_'+arch[:-3]+'tif ../HILLSH/HILLSH_'+arch[:-3]+'tif')
         os.system('/usr/local/Cellar/gdal2-python/2.2.4/libexec/bin/gdalcopyproj.py ../DEMS/DEM_'+arch[:-3]+'tif ../HILLSH/HILLSH_'+arch[:-3]+'tif')
         log(arch)
         os.system('/usr/local/Cellar/gdal2-python/2.2.4/libexec/bin/gdalcopyproj.py ../DEMS/DEM_'+arch[:-3]+'tif ../LOG/LOG_'+arch[:-3]+'tif')
    else:
         print 'HECHO'
os.system('rm tmp*')