#!/home/raul/miniconda/bin/python

""" Creates a digital terrain model profile

Raul Valenzuela

May, 2015

"""

import pyqtgraph as pg

from pyqtgraph.Qt import QtCore, QtGui

import numpy as np

import gdal

from os.path import expanduser

import matplotlib.cm as cm # colormap

# Update profile

def updatePlot():

global img, roi, data, p2

selected = roi.getArrayRegion(data, img)

p2.plot(selected, clear=True)

def makeLut(ncolors):

lut_temp=cm.gist_earth(np.arange(ncolors))*256

return lut_temp[::,0:3].astype(int)

def makeColorbarGradient(levels,lut):

return grad

# create app

pg.mkQApp()

# pg.setConfigOption('background', 'w')

# create window

win = pg.GraphicsLayoutWidget()

win.setWindowTitle('pyqtgraph example: Image Analysis')

# Add a ViewBox for displaying the image

p1 = win.addViewBox()

p1.setBackgroundColor('w')

# print p1.viewRect()

# Custom ROI for selecting an image region

roi=pg.LineSegmentROI([[10,10],[30,30]])

p1.addItem(roi)

# Add plot area (ViewBox + axes) for displaying the profile

win.nextRow()

p2 = win.addPlot(colspan=2)

p2.setMaximumHeight(250)

win.resize(600, 700)

# show window with widgets

win.show()

# make lookup table (colormap)

lut=makeLut(256)

# handles geotiff dem (1 band only)

home = expanduser("~")

dem_file = home+'/Github/RadarQC/merged_dem_38-39_123-124_extended.tif'

layer = gdal.Open(dem_file)

gt =layer.GetGeoTransform()

# prepare dtm

clip=[-124.17, -122.65, 37.80, 39.30]

xmin = int((clip[0] - gt[0]) / gt[1])

ymin =int((clip[3] - gt[3]) / gt[5])

xmax = int((clip[1] - gt[0]) / gt[1])

ymax =int((clip[2] - gt[3]) / gt[5])

win_xsize=xmax-xmin

win_ysize=ymax-ymin

x_buff=250 #new resolution

y_buff=250 #new resolution

data = np.rot90(layer.GetRasterBand(1).ReadAsArray(xmin,ymin,

win_xsize,win_ysize,

x_buff,y_buff),3)

# Item for displaying image data

img = pg.ImageItem()

img.setImage(data,lut=lut,levels=(0,1000))

p1.addItem(img)

# print p1.itemBoundingRect(img)

# p1.setSceneRect(-180, -90, 360, 180)

# colorbar

cb=pg.GradientLegend((10,200), (30,30))

cb.setParentItem(img)

val=np.linspace(0,1000,20).astype(int)

foograd=makeColorbarGradient(val,lut)

cb.setGradient(foograd)

cblabels = {'0': 0,'200':0.2,'400':0.4,

'600':0.6,'800':0.8,'1000': 1}

cb.setLabels(cblabels)

cb.scale(1,-1)

# set position and scale of image

# img.scale(0.2, 0.2)

# img.translate(-50, 0)

roi.sigRegionChanged.connect(updatePlot)

updatePlot()

## Start Qt event loop unless running in interactive mode or using pyside.

if __name__ == '__main__':

import sys

if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):

QtGui.QApplication.instance().exec_()