"""
     qtgui/ExponentialSlider.py: CCP4MG Molecular Graphics Program
     Copyright (C) 2009-2011 University of York

     This library is free software: you can redistribute it and/or
     modify it under the terms of the GNU Lesser General Public License
     version 3, modified in accordance with the provisions of the 
     license to address the requirements of UK law.
 
     You should have received a copy of the modified GNU Lesser General 
     Public License along with this library.  If not, copies may be 
     downloaded from http://www.ccp4.ac.uk/ccp4license.php
 
     This program is distributed in the hope that it will be useful,
     but WITHOUT ANY WARRANTY; without even the implied warranty of
     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
     GNU Lesser General Public License for more details.
"""
from PyQt4 import QtCore, QtGui
import sys
import math
import functools

"""
A couple of sliders with exponential ranges.
ExponentialSlider has a range of *small* to upper limit.
SignedExponentialSlider has a range of -upper limit to upper limit.
The range varies exponentially.
Useful methods are the constructors, value, setValue, setRange and the signals see below
and example usage at bottom of this file.
"""

class ExponentialSlider(QtGui.QWidget):
	expFactor = 0.1
	preFactor = 1.0

	def __init__(self,labels=(),reversed=False,parent=None):
		QtGui.QWidget.__init__(self,parent)
		layout = QtGui.QGridLayout()
                layout.setContentsMargins(0,0,0,0)
                layout.setSpacing(2)
		self.reversed = reversed

		self.slider = QtGui.QSlider(QtCore.Qt.Horizontal)
		if self.reversed:
			self.slider.setRange(-100,0)
			self.reverseFactor = -1
		else:
			self.slider.setRange(0,100)
			self.reverseFactor = 1

		self.connect(self.slider,QtCore.SIGNAL("valueChanged(int)"),self.valueChanged)
		self.connect(self.slider,QtCore.SIGNAL("sliderMoved(int)"),self.valueChanged)
		self.slider.setSizePolicy(QtGui.QSizePolicy.Expanding,QtGui.QSizePolicy.Fixed)

		if labels and len(labels)==2:
			if self.reversed:
				l1 = QtGui.QLabel(labels[1])
				l2 = QtGui.QLabel(labels[0])
			else:
				l1 = QtGui.QLabel(labels[0])
				l2 = QtGui.QLabel(labels[1])
			layout.addWidget(l1,0,0)
			layout.addWidget(l2,0,1,1,1,QtCore.Qt.AlignRight)

		rc = layout.rowCount()
		layout.addWidget(self.slider,rc,0,1,2)

		self.lineEdit = QtGui.QLineEdit()
		self.lineEdit.setMaximumWidth(50)
		layout.addWidget(self.lineEdit,rc,2,1,1)
		self.lineEdit.setSizePolicy(QtGui.QSizePolicy.Fixed,QtGui.QSizePolicy.Fixed)
		dv = QtGui.QDoubleValidator(self.lineEdit)
		self.lineEdit.setValidator(dv)
		self.connect(self.lineEdit,QtCore.SIGNAL("textChanged(const QString &)"),self.textChanged)

		self.connect(self.slider,QtCore.SIGNAL("valueChanged(int)"),self.sliderValueChangedSignal)
		self.connect(self.slider,QtCore.SIGNAL("sliderMoved(int)"),self.sliderMovedSignal)
		self.connect(self.slider,QtCore.SIGNAL("sliderReleased()"),self.sliderReleasedSignal)

		self.connect(self.lineEdit,QtCore.SIGNAL("editingFinished()"),self.textEditEditingFinishedSignal)
		self.connect(self.lineEdit,QtCore.SIGNAL("textChanged(const QString &)"),self.textEditTextChangedSignal)

		layout.setColumnStretch(0,2)
		layout.setColumnStretch(1,2)

		self.setLayout(layout)

		self.lineEdit.blockSignals(True)
		self.lineEdit.setText('-1')
		self.lineEdit.blockSignals(False)
		self.setRange(1e-3,20000)
		self.setValue(-1)

################################################################################
### The signals ###
	def textEditEditingFinishedSignal(self):
		self.emit(QtCore.SIGNAL('textEditingFinished'))
	
	def textEditTextChangedSignal(self,val):
		self.emit(QtCore.SIGNAL('textChanged'),(val))
	
	def sliderReleasedSignal(self):
		self.emit(QtCore.SIGNAL('sliderReleased'))
	
	def sliderMovedSignal(self,val):
		self.emit(QtCore.SIGNAL('sliderMoved'),(self.value()))
	
	def sliderValueChangedSignal(self,val):
		self.emit(QtCore.SIGNAL('sliderValueChanged'),(self.value()))
################################################################################
	
	def value(self):
		try:
			val = float(self.lineEdit.text())
			return val
		except:
			return 0.0

	def setValue(self,val):
		self.textChanged(str(val))
		self.lineEdit.blockSignals(True)
		self.lineEdit.setText(str(val))
		self.lineEdit.blockSignals(False)

	def setRange(self,minval,maxval):
		t=self.lineEdit.text()
		minval = max(minval,1e-12)
		maxval = min(maxval,1e+12)
		try:
			minval =  math.log(minval)/self.expFactor
			maxval =  math.log(maxval)/self.expFactor
			if self.reversed:
				self.slider.setRange(-maxval,-minval)
				self.reverseFactor = -1
			else:
				self.slider.setRange(minval,maxval)
				self.reverseFactor = 1
		except:
			print "Range error in ExponentialSlider.setRange", minval,maxval
		self.lineEdit.blockSignals(True)
		self.lineEdit.setText(t)
		self.lineEdit.blockSignals(False)

	def textChanged(self,text):
		try:
			expval = float(text)
			val = self.reverseFactor*math.log(expval)/self.expFactor
		except:
			val = 1e-7
		self.slider.blockSignals(True)
		self.slider.setValue(int(math.floor(val+0.5)))
		self.slider.blockSignals(False)

	def valueChanged(self,val):
		expval = self.preFactor*math.exp(self.reverseFactor*self.slider.value()*self.expFactor)
		expvalstr = ("%10.2f"%expval).strip()
		self.lineEdit.blockSignals(True)
		self.lineEdit.setText(expvalstr)
		self.lineEdit.blockSignals(False)

class SignedExponentialSlider(ExponentialSlider):
	def __init__(self,labels=(),reversed=False,parent=None):
		ExponentialSlider.__init__(self,labels,reversed,parent)
		self.lineEdit.blockSignals(True)
		self.lineEdit.setText('-1')
		self.lineEdit.blockSignals(False)
		self.setRange(1e-3,20000)
		self.setValue(-1)

	def valueChanged(self,val):
		if val<0:
			sliderval = -val+self._minval
			expval = -self.preFactor*math.exp(self.reverseFactor*sliderval*self.expFactor)
		else:
			sliderval = val+self._minval
			expval = self.preFactor*math.exp(self.reverseFactor*sliderval*self.expFactor)
		expvalstr = ("%10.2f"%expval).strip()
		self.lineEdit.blockSignals(True)
		self.lineEdit.setText(expvalstr)
		self.lineEdit.blockSignals(False)

	def textChanged(self,text):
		expval = 1
		try:
			expval = float(text)
			if expval < 0:
				val = -self.reverseFactor*math.log(-expval)/self.expFactor+self._minval
			else:
				val = self.reverseFactor*math.log(expval)/self.expFactor-self._minval
		except:
			val = 1e-7
		self.slider.blockSignals(True)
		self.slider.setValue(int(math.floor(val+0.5)))
		self.slider.blockSignals(False)

	def setRange(self,minval,maxval):
		t=self.lineEdit.text()
		minval = max(minval,1e-12)
		maxval = min(maxval,1e+12)
		try:
			minval =  math.log(minval)/self.expFactor
			maxval =  math.log(maxval)/self.expFactor
			self._minval = minval
			self._maxval = maxval
			self.range = maxval-minval
			minval = -self.range
			maxval = self.range
			if self.reversed:
				self.slider.setRange(-maxval,-minval)
				self.reverseFactor = -1
			else:
				self.slider.setRange(minval,maxval)
				self.reverseFactor = 1
		except:
			print "Range error in ExponentialSlider.setRange", minval,maxval
		self.lineEdit.blockSignals(True)
		self.lineEdit.setText(t)
		self.lineEdit.blockSignals(False)

if __name__=="__main__":
	if "-quit" in sys.argv:
		sys.exit()

	def testEditFinish():
		print "edit finish"

	def testTextChange(val):
		print "text change",val

	def testValueChange(val):
		print "change",val

	def testMove(val):
		print "move",val

	def testRelease():
		print "release"

	app = QtGui.QApplication(sys.argv)

	window = QtGui.QWidget()
	layout = QtGui.QVBoxLayout()
        layout.setContentsMargins(0,0,0,0)
        layout.setSpacing(2)

	widget = ExponentialSlider(reversed=True,labels=(QtCore.QCoreApplication.translate("QtGui.QWidget",'Thicker'),QtCore.QCoreApplication.translate("QtGui.QWidget",'Thinner')))
	widget.setRange(1e-3,20000)
	widget.setValue(100)

	layout.addWidget(widget)

	widget = SignedExponentialSlider(reversed=False,labels=(QtCore.QCoreApplication.translate("QtGui.QWidget",'Closer'),QtCore.QCoreApplication.translate("QtGui.QWidget",'Further')))
	#widget.setRange(1e-3,20000)
	widget.setValue(-10)
	widget.setRange(1e-3,20)

	app.connect(widget,QtCore.SIGNAL('sliderReleased'),testRelease)
	app.connect(widget,QtCore.SIGNAL('sliderMoved'),testMove)
	app.connect(widget,QtCore.SIGNAL('sliderValueChanged'),testValueChange)
	app.connect(widget,QtCore.SIGNAL('textEditingFinished'),testEditFinish)
	app.connect(widget,QtCore.SIGNAL('textChanged'),testTextChange)

	layout.addWidget(widget)

	window.setLayout(layout)

	window.show()

	sys.exit(app.exec_())