#!/usr/bin/env python
################################################################
# 
#
# Project:  Sasha.py
# Purpose:   Datum Converter 
#	    
#	   
# Author:   Massimo Di Stefano , massimodisasha@yahoo.it
#
################################################################
#
# Copyright (c) 2009, Massimo Di Stefano <massimodisasha@yahoo.it>
# 
# Permission is hereby granted, free of charge, to any person obtaining a
# copy of this software and associated documentation files (the "Software"),
# to deal in the Software without restriction, including without limitation
# the rights to use, copy, modify, merge, publish, distribute, sublicense,
# and/or sell copies of the Software, and to permit persons to whom the
# Software is furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included
# in all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
# THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
# DEALINGS IN THE SOFTWARE.
###############################################################################
import os
from scipy import io, linalg
from numpy import *
from pylab import save
import sys
from PyQt4 import QtCore, QtGui
from math import pi
 
def affine(gcpD1, gcpD2, knowD1, output):
	L = io.read_array(str(gcpD1))
	A = zeros((2*L.shape[0],6),float)
	A[ ::2, 2] = 1.0
	A[1::2, 5] = 1.0
	A[ ::2, 0] = L[:,0]
	A[1::2, 4] = L[:,1]
	A[ ::2, 1] = L[:,1]
	A[1::2, 3] = L[:,0]
 
 
 
	G = io.read_array(str(gcpD2))
	Y = zeros((2*G.shape[0],1),float)
	Y[ ::2, 0] = G[:,0]
	Y[1::2, 0] = G[:,1]
 
 
	N = dot(A.T.conj(), A)
	T = dot(A.T.conj(), Y)
	C = dot(linalg.inv(N), T)
	E0 = C[2]
	N0 = C[5]
	LX = io.read_array(str(knowD1))
	ss = LX.shape[0]
	ENglob = zeros((ss,2),float)
	for i in arange(ss):
		E2 = E0+LX[i,0]*C[0]+LX[i,1]*C[1]
		N2 = N0+LX[i,0]*C[3]+LX[i,1]*C[4]
		ENglob[i,:] = hstack((E2,N2))
	save(output, ENglob, fmt='%5.12f')
 
	sq = LX.shape[0]
	sqx = zeros((G.shape[0],1),float)
	sqy = zeros((G.shape[0],1),float)
	sqx[ ::1, 0] = G[:,0]
	sqy[::1, 0] = G[:,1]
	sizeq = sqx.shape[0]
	scartix = zeros((sizeq,1),float)
	scartiy = zeros((sizeq,1),float)
	scartiqx = zeros((sizeq,1),float)
	scartiqy = zeros((sizeq,1),float)
	sqm = zeros((sizeq,1),float)
	for i in arange(sizeq):
		Vx = E0 + LX[i,0] * C[0,0] + LX[i,1] * C[1,0] - sqx[i,0]
		Vy = N0 + LX[i,0] * C[3,0] + LX[i,1] * C[4,0] - sqy[i,0]
		sqmr = sqrt(Vx**2 + Vy**2)
		sqm[i,:] = sqmr
		scartiqx[i,0] = Vx**2
		scartiqy[i,0] = Vy**2
		scartix[i,0] = Vx
		scartiy[i,0] = Vx
		scartiq = concatenate((scartiqx,scartiqy))
		scarti = concatenate((scartix,scartiy))
		scartixy = concatenate((scartiqx,scartiqy),1)
		scartiqxy = concatenate((scartix,scartiy),1)
	sumscarti = sum(scartiq)
	varqp = sumscarti / ((2 * ss) - 4)
	varp = sqrt(varqp)
	varianzaq = varqp * linalg.inv(N)
	varianza = varp * linalg.inv(N)
	varqunitp = diag(varianzaq)
	varunitp = diag(varianza)
 
 
	prec = zeros((sq,2),float)
	for i in arange(sq):
		xx = sqrt((LX[i,0]**2) * (varqunitp[0]) + (LX[i,1]**2) * (varqunitp[1]) + varqunitp[2])
		yy = sqrt((LX[i,1]**2) * (varqunitp[0]) + (LX[i,0]**2) * (varqunitp[1]) + varqunitp[5])
		prec[i,:] = hstack((xx,yy))
	errore = concatenate((scartixy,scartiqxy,sqm),1)
	save(output+str('_precisione'), prec, fmt='%5.12f')
	save(output+str('_errore'), errore, fmt='%5.12f')
	err_med = sum(sqm) / sq
	print err_med
 
 
 
def conforme(gcpD1, gcpD2, knowD1, output):
	L = io.read_array(str(gcpD1))
	A = zeros((2*L.shape[0],4),float)
	A[ ::2, 0] = 1.0
	A[1::2, 1] = 1.0
	A[ ::2, 2] = L[:,0]
	A[1::2, 2] = L[:,1]
	A[ ::2, 3] = L[:,1]
	A[1::2, 3] = -L[:,0]
	G = io.read_array(str(gcpD2))
	Y = zeros((2*G.shape[0],1),float)
	Y[ ::2, 0] = G[:,0]
	Y[1::2, 0] = G[:,1]
	N = dot(A.T.conj(), A)
	T = dot(A.T.conj(), Y)
	C = dot(linalg.inv(N), T)
	Lambda = abs(C[2]+C[3]*1j)
	Alpha = angle(C[2]+C[3]*1j)
	E0 = C[0]
	N0 = C[1]
	LX = io.read_array(str(knowD1))
	ss = LX.shape[0]
	pq = G.shape[0]
	ENglob = zeros((ss,2),float)
	lam = sqrt(C[2]** + C[3]**2)
	alp = arctan(C[3] / C[2]) / (pi / 180.)
	for i in arange(ss):
		E2 = E0+LX[i,0]*C[2]+LX[i,1]*C[3]
		N2 = N0+LX[i,1]*C[2]-LX[i,0]*C[3]
		ENglob[i,:] = hstack((E2,N2))
	save(output, ENglob, fmt='%5.12f')
	print ENglob	
	sq = LX.shape[0]
	sqx = zeros((G.shape[0],1),float)
	sqy = zeros((G.shape[0],1),float)
	sqx[ ::1, 0] = G[:,0]
	sqy[::1, 0] = G[:,1]
	sizeq = sqx.shape[0]
	scartix = zeros((sizeq,1),float)
	scartiy = zeros((sizeq,1),float)
	scartiqx = zeros((sizeq,1),float)
	scartiqy = zeros((sizeq,1),float)
	sqm = zeros((sizeq,1),float)
	for i in arange(sizeq):
		Vx = E0 + LX[i,0] * C[2,0] + LX[i,1] * C[3,0] - sqx[i,0]
		Vy = N0 + LX[i,1] * C[2,0] - LX[i,0] * C[3,0] - sqy[i,0]
		print Vx , Vy
		sqmr = sqrt(Vx**2 + Vy**2)
		sqm[i,:] = sqmr
		scartiqx[i,0] = Vx**2
		scartiqy[i,0] = Vy**2
		scartix[i,0] = Vx
		scartiy[i,0] = Vy
	scartiq = concatenate((scartiqx,scartiqy))
	scarti = concatenate((scartix,scartiy))
	scartixy = concatenate((scartiqx,scartiqy),1)
	scartiqxy = concatenate((scartix,scartiy),1)
	print sqm
	sumscarti = sum(scartiq)
	varqp = sumscarti / ((2 * sizeq) - 4)
	varp = sqrt(varqp)
	varianzaq = varqp * linalg.inv(N)
	varianza = varp * linalg.inv(N)
	varqunitp = diag(varianzaq)
	varunitp = diag(varianza)
 
 
	prec = zeros((sq,2),float)
	for i in arange(sq):
		xx = sqrt((LX[i,0]**2) * (varqunitp[2]) + (LX[i,1]**2) * (varqunitp[3]) + varqunitp[0])
		yy = sqrt((LX[i,1]**2) * (varqunitp[2]) + (LX[i,0]**2) * (varqunitp[3]) + varqunitp[1])
		prec[i,:] = hstack((xx,yy))
	errore = concatenate((scartixy,scartiqxy,sqm),1)
	save(output+str('_precisione'), prec, fmt='%5.12f')
	save(output+str('_errore'), errore, fmt='%5.12f')
	err_med = sum(sqm) / pq
 
	a = C[2,0]
	b = C[3,0]
	lamvar = sqrt(((((2 * a) / sqrt(a**2 + b**2))**2) * varqunitp[2]) + ((((2 * b) / sqrt(a**2 + b**2))**2) * varqunitp[3]))
	alphavar = sqrt(((((-b / a**2) / (1 + (b / a)**2))**2) * varqunitp[2]) + (((1 / a) / (1 + (b / a)**2))**2 * varqunitp[3])) / (pi / 180.)
 
 
 
	print str('Parametri di trasformazione :')
	print str('Tx : ') , C[0,0]
	print str('Ty : ') , C[1,0]
	print str('Rotazione rigida :') , alp
	print str('Fattore di scala:') , lam
	print str(" ")
	print str('Varianza dei parametri geometrici : ') 
	print str('Sigma quadro rotazione rigida :') , alphavar
	print str('Sigma quadro varazione di scala :') , lamvar
	print str(" ")
	print str('Scarto quadratico medio :') , err_med
 
 
class Dialog(QtGui.QDialog):
 
 
    def __init__(self, parent=None):
        global layout
        QtGui.QDialog.__init__(self, parent)
        self.openFilesPath = QtCore.QString()
        self.errorMessageDialog = QtGui.QErrorMessage(self)
 
        frameStyle = QtGui.QFrame.Sunken | QtGui.QFrame.Panel
 
 
        self.push = QtGui.QPushButton(self.tr("Toggle Push Button"))
 
 
 
        self.exit = QtGui.QPushButton("Exit", self)
 
 
        self.connect(self.exit, QtCore.SIGNAL("clicked()"),
                     QtGui.qApp, QtCore.SLOT("quit()"))
 
 
        self.convert = QtGui.QPushButton("Convert", self)
 
 
        self.connect(self.convert, QtCore.SIGNAL("clicked()"),
        self.saveFunction)
 
 
        self.affineRadioButton = QtGui.QRadioButton(self.tr("Affine"))
        self.conformeRadioButton = QtGui.QRadioButton(self.tr("Conforme"))
        self.affineRadioButton.setChecked(True)
        self.shapeRadioButton = QtGui.QRadioButton(self.tr("shape"))
        self.grassRadioButton = QtGui.QRadioButton(self.tr("grass"))
 
        self.openFileNameLabel = QtGui.QLineEdit()
 
        self.openFileNameButton = QtGui.QPushButton(self.tr("G.C.P.1"))
        self.connect(self.openFileNameButton, QtCore.SIGNAL("clicked()"), self.setOpenFileName)
 
        self.openFileNameLabel_2 = QtGui.QLineEdit()
 
        self.openFileNameButton_2 = QtGui.QPushButton(self.tr("G.C.P.2"))
        self.connect(self.openFileNameButton_2, QtCore.SIGNAL("clicked()"), self.setOpenFileName_2)
 
        self.openFileNameLabel_3 = QtGui.QLineEdit()
 
        self.openFileNameButton_3 = QtGui.QPushButton(self.tr("input"))
        self.connect(self.openFileNameButton_3, QtCore.SIGNAL("clicked()"), self.setOpenFileName_3)
 
        self.saveFileNameLabel = QtGui.QLineEdit()
 
        self.saveFileNameButton = QtGui.QPushButton(self.tr("output"))
        self.connect(self.saveFileNameButton, QtCore.SIGNAL("clicked()"), self.setSaveFileName)
 
        self.statusLabel = QtGui.QLabel(self.tr("Sasha.py"))
 
        layout = QtGui.QGridLayout()
        layout.setColumnStretch(1, 1)
        layout.setColumnMinimumWidth(1, 25)
 
        layout.addWidget(self.openFileNameButton, 3, 0)
        layout.addWidget(self.openFileNameLabel, 3, 1)
        layout.addWidget(self.openFileNameButton_2, 4, 0)
        layout.addWidget(self.openFileNameLabel_2, 4, 1)
        layout.addWidget(self.openFileNameButton_3, 1, 0)
        layout.addWidget(self.openFileNameLabel_3, 1, 1)
        layout.addWidget(self.saveFileNameButton, 2, 0)
        layout.addWidget(self.saveFileNameLabel, 2, 1)
        layout.addWidget(self.exit, 7, 0)
        layout.addWidget(self.convert, 6, 0)
        layout.addWidget(self.affineRadioButton, 5, 0)
        layout.addWidget(self.conformeRadioButton, 5, 1)
 
 
 
        self.setLayout(layout)
        self.setWindowTitle(self.tr("DATUM CONVERTER"))
 
    def changeTitle(self, value):
        if self.check.isChecked():
            layout.addWidget(self.push,9,1)
 
        else :
            layout.removeWidget(self.push)
 
 
 
    def saveFunction(self):
	if self.affineRadioButton.isChecked():
		print affine(unicode(self.openFileNameLabel.text()),unicode(self.openFileNameLabel_2.text()), unicode(self.openFileNameLabel_3.text()), unicode(self.saveFileNameLabel.text()))
	else:
		print conforme(unicode(self.openFileNameLabel.text()),unicode(self.openFileNameLabel_2.text()), unicode(self.openFileNameLabel_3.text()), unicode(self.saveFileNameLabel.text()))	
 
 
    def setOpenFileName_3(self):
	fileName_3 = QtGui.QFileDialog.getOpenFileName(self,
	self.tr("file_datum_3"),
	self.openFileNameLabel_3.text(),
	self.tr("ALL Files (*);;Text Files (*.txt)"))
	if not fileName_3.isEmpty():
		self.openFileNameLabel_3.setText(fileName_3)
 
 
 
 
    def setOpenFileName(self):
	fileName = QtGui.QFileDialog.getOpenFileName(self,
	self.tr("file_datum_1"),
	self.openFileNameLabel.text(),
	self.tr("All Files (*);;Text Files (*.txt)"))
	if not fileName.isEmpty():
		self.openFileNameLabel.setText(fileName)
 
 
    def setOpenFileName_2(self):
	fileName_2 = QtGui.QFileDialog.getOpenFileName(self,
	self.tr("file_datum_2"),
	self.openFileNameLabel_2.text(),
	self.tr("ALL Files (*);;Text Files (*.txt)"))
	if not fileName_2.isEmpty():
		self.openFileNameLabel_2.setText(fileName_2)
 
    def setSaveFileName(self):
	fileName = QtGui.QFileDialog.getSaveFileName(self,
	self.tr("file_to_save"),
	self.saveFileNameLabel.text(),
	self.tr("All Files (*);;Text Files (*.txt)"))
	if not fileName.isEmpty():
		self.saveFileNameLabel.setText(fileName)
 
if __name__ == '__main__':
    app = QtGui.QApplication(sys.argv)
    dialog = Dialog()
    sys.exit(dialog.exec_())