WdgFoto.py

# -*- coding: utf-8 -*-

"""
/***************************************************************************
Name                 : Omero RT
Description          : Omero plugin
Date                 : August 15, 2010 
copyright            : (C) 2010 by Giuseppe Sucameli (Faunalia)
email                : sucameli@faunalia.it
 ***************************************************************************/

Omero plugin
Works done from Faunalia (http://www.faunalia.it) with funding from Regione 
Toscana - S.I.T.A. (http://www.regione.toscana.it/territorio/cartografia/index.html)

/***************************************************************************
 *                                                                         *
 *   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 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/
"""

from PyQt4.QtCore import *
from PyQt4.QtGui import *

import qgis.core
import random
import string

from ui.wdgFoto_ui import Ui_Form
from AutomagicallyUpdater import *
from Utils import TemporaryFile, Porting

class WdgFoto(QWidget, MappingOne2One, Ui_Form):

	def __init__(self, parent=None):
		QWidget.__init__(self, parent)
		MappingOne2One.__init__(self, "FOTO_GEOREF")
		self.setupUi(self)

		self.SCHEDA_EDIFICIOID.hide()

		# imposta i validatori
		self.GEOREF_EPSG4326_X.setValidator( QDoubleValidator(self) )
		self.GEOREF_EPSG4326_Y.setValidator( QDoubleValidator(self) )
		self.GEOREF_PROIET_X.setValidator( QDoubleValidator(self) )
		self.GEOREF_PROIET_Y.setValidator( QDoubleValidator(self) )

		# carica i widget multivalore con i valori delle relative tabelle
		tablesDict = {
			self.ZZ_FRONTE_EDIFICIOID: AutomagicallyUpdater.ZZTable( "ZZ_FRONTE_EDIFICIO" )
		}
		self.setupTablesUpdater(tablesDict)
		self.loadTables()

		# mappa i widget con i campi delle tabelle
		childrenList = [
			self.SCHEDA_EDIFICIOID, 
			(self.GEOREF_PROIET_X, AutomagicallyUpdater.OPTIONAL), 
			(self.GEOREF_PROIET_Y, AutomagicallyUpdater.OPTIONAL), 
			(self.GEOREF_EPSG4326_X, AutomagicallyUpdater.OPTIONAL), 
			(self.GEOREF_EPSG4326_Y, AutomagicallyUpdater.OPTIONAL), 
			(self.FILENAME, AutomagicallyUpdater.OPTIONAL), 
			(self.ANNOTAZIONE, AutomagicallyUpdater.OPTIONAL), 
			(self.IMAGE, AutomagicallyUpdater.OPTIONAL), 
			self.ZZ_FRONTE_EDIFICIOID
		]
		self.setupValuesUpdater(childrenList)

		self.connect(self.IMAGE, SIGNAL( "openPicRequested" ), self.apriFoto)


	def caricaImmagine(self, filename):
		# TODO: recupera metadati tramite osgeo.gdal
		
		# set projections config to "useProject" to avoid asking SR loading foto
		settings = QSettings()
		oldProjectionBehavious = settings.value("/Projections/defaultBehaviour", "useProject", type=str)
		settings.setValue("/Projections/defaultBehaviour", "useProject")
		rl = qgis.core.QgsRasterLayer( filename )
		settings.setValue("/Projections/defaultBehaviour", oldProjectionBehavious)
		if not rl.isValid():
			return False

		# TODO: recupera georef info
		metadata = rl.metadata()

		GPSmetadata = { "EXIF_GPSLatitude" : None, "EXIF_GPSLongitude" : None, "EXIF_GPSAltitude" : None }
		for k in GPSmetadata.keys():
			# e.g. EXIF_GPSLatitude=(46) (42.5) (0)</p>
			rx = QRegExp( "%s=\((\d+(?:\.\d+)?)\) ?(?:\((\d+(?:\.\d+)?)\) ?(?:\((\d+(?:\.\d+)?)\)))</p>" % k )
			pos = rx.indexIn(metadata, 0)
			if pos != -1 and rx.pos(1) != -1:
				value = float(rx.cap(1))
				if rx.pos(2) != -1:
					value = value + float(rx.cap(2))/60
				if rx.pos(3) != -1:
					value = value + float(rx.cap(3))/60/60

				GPSmetadata[k] = value

		longitude = GPSmetadata["EXIF_GPSLongitude"]
		latitude = GPSmetadata["EXIF_GPSLatitude"]
		if latitude != None and longitude != None:
			point4326 = qgis.core.QgsPoint(longitude, latitude)

			self.setValue( self.GEOREF_EPSG4326_X, Porting.str(point4326.x()) )
			self.setValue( self.GEOREF_EPSG4326_Y, Porting.str(point4326.y()) )

		from ManagerWindow import ManagerWindow
		uvID = ManagerWindow.instance.scheda.UNITA_VOLUMETRICHE.firstTab.getUV()
		query = AutomagicallyUpdater.Query( "SELECT X(point), Y(point) FROM (SELECT PointOnSurface(geometria) AS point FROM GEOMETRIE_RILEVATE_NUOVE_O_MODIFICATE WHERE ID_UV_NEW = ?) AS sub", [uvID]).getQuery()
		if not query.exec_() or not query.next():
			AutomagicallyUpdater._onQueryError( query.lastQuery(), query.lastError().text(), self )
		else:
			self.setValue( self.GEOREF_PROIET_X, query.value(0) )
			self.setValue( self.GEOREF_PROIET_Y, query.value(1) )

		self.setValue(self.FILENAME, filename)
		self.setValue(self.IMAGE, filename)	# mostra la preview

		return True

	def getValue(self, widget):
		widget = self._getRealWidget(widget)

		if widget == self.IMAGE and self._ID != None:
			imagestr = AutomagicallyUpdater.Query( "SELECT IMAGE FROM FOTO_GEOREF WHERE ID = ?", [self._ID] ).getFirstResult()
			# control to allow cohesinstence of old 1.8 images db and new images db
			# new images are saved as QByteArray.toHex
			if isinstance(imagestr, unicode):
				imagestr = QByteArray.fromHex(imagestr)
			return QByteArray(imagestr)

		if widget == self.GEOREF_EPSG4326_X or widget == self.GEOREF_EPSG4326_Y or \
				widget == self.GEOREF_PROIET_X or widget == self.GEOREF_PROIET_Y:
			# il validatore QDoubleValidator permette di inserire valori 
			# non completi (es. "-" o "."), restitusci None in quei casi
			value = MappingOne2One.getValue(widget)
			try:
				value = str( float( str(value) ) )
			except ValueError:
				value = None
			return value

		return MappingOne2One.getValue(widget)

	def setValue(self, widget, value):
		MappingOne2One.setValue(widget, value)
		# disabilita il caching se l'immagine è già stata memorizzata sul db
		if self._getRealWidget(widget) == self.IMAGE and self._ID != None:
			self.IMAGE.clearCache()


	def apriFoto(self):
		ext = QFileInfo( self.getValue(self.FILENAME) ).suffix()
		image = self.getValue(self.IMAGE)
		filename = TemporaryFile.salvaDati( self.getValue(self.IMAGE), TemporaryFile.KEY_SCHEDAEDIFICIO, ext )
		if filename == None:
			return False
		url = QUrl.fromLocalFile( filename )
		QDesktopServices.openUrl( url )
		return True


	def _saveValue(self, name2valueDict, table, pk, ID=None):
		# non aggiornare le immagini per evitare overhead inutile: 
		# la GUI non permette di cambiare l'immagine salvata, l'unico modo è 
		# creare una nuova immagine ed eliminare quella vecchia
		if ID != None and name2valueDict.has_key( self.IMAGE.objectName() ):
			del name2valueDict[ self.IMAGE.objectName() ]
		return AutomagicallyUpdater._saveValue(name2valueDict, table, pk, ID)


	def toHtml(self, index, prefix=None):
		ext = QFileInfo( self.getValue(self.FILENAME) ).suffix()

		# save the image
		if prefix is None:
			filename = TemporaryFile.salvaDati( self.getValue(self.IMAGE), TemporaryFile.KEY_SCHEDAEDIFICIO2HTML, ext )
		else:
			suffix = self._ID
			print suffix
			if not suffix:
				suffix = ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(4))
				
			filename = u"%s/img_%s.%s" % (prefix, suffix, ext)
			with open( filename, "wb" ) as fout:
				fout.write( Porting.str( self.getValue(self.IMAGE) ) )

		filename = QUrl.fromLocalFile( filename ).toString()

		fronte_edificio = self.ZZ_FRONTE_EDIFICIOID.currentText() if self.getValue(self.ZZ_FRONTE_EDIFICIOID) >= 0 else ''
		annotazione = self.getValue(self.ANNOTAZIONE)

		georef_x = self.getValue(self.GEOREF_PROIET_X)
		georef_y = self.getValue(self.GEOREF_PROIET_Y)
		georef = '%s , %s' % ( Porting.str(georef_x), Porting.str(georef_y) ) if georef_x != None and georef_y != None else ''
		
		return u"""
<table class="border %s">
	<tr>
		<td>Foto #%d</td><td class="value">%s</td>
	</tr>
	<tr>
		<td colspan="2"><img class="foto border" src="%s" alt="foto"></td>
	</tr>
	<tr>
		<td>Coordinate EPSG:3003</td><td class="value">%s</td>
	</tr>
	<tr>
		<td>Annotazioni</td><td class="value">%s</td>
	</tr>
</table>
""" % ( 'newPage' if index%2 == 0 else '', index+1, fronte_edificio, filename, georef, annotazione if annotazione != None else '' )