"""
     qtgui/mgUpperLowerSlider.py: CCP4MG Molecular Graphics Program
     Copyright (C) 2011 University of York
     Copyright (C) 2012 STFC

     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.
"""
import sys
import os
import math
import functools
from PyQt4 import QtGui, QtCore
from MGSimpleWidgets import SVGLabel

class UpperLowerSlider(QtGui.QWidget):
  def toggleLock(self,e=None):
     self._lock = 1-self._lock
     icon_path = os.path.join(os.environ['CCP4MG'],'qticons')
     if self._lock == 0:
       self.padlock.loadFile(os.path.join(icon_path,"padlock_open.svg"))
     else:
       self.padlock.loadFile(os.path.join(icon_path,"padlock.svg"))
     self.repaint()

  def resizeEvent(self,e):
     if self._orientation == QtCore.Qt.Horizontal:
       self.padlock._bounds = QtCore.QRectF(self.s1.width()/2-10,-3,20,20)
     else:
       self.padlock._bounds = QtCore.QRectF(self.lowerSpin.width()/2-25,self.s1.height()/2-10,20,20)
     e.accept()

  def __init__(self,orientation=None,parent=None):
     QtGui.QWidget.__init__(self,parent)
     self._orientation = orientation
     self.lowerSpin = QtGui.QDoubleSpinBox()
     self.upperSpin = QtGui.QDoubleSpinBox()
     layout = QtGui.QGridLayout()
     if orientation == QtCore.Qt.Horizontal:
        self.s1 = QtGui.QSlider(QtCore.Qt.Horizontal,parent)
        self.s2 = QtGui.QSlider(QtCore.Qt.Horizontal,parent)
     else:
        self.s1 = QtGui.QSlider(parent)
        self.s2 = QtGui.QSlider(parent)

     icon_path = os.path.join(os.environ['CCP4MG'],'qticons')
     self.lowerSpin.setRange(0,100)
     self.upperSpin.setRange(0,100)
     self.s1.setRange(0,100)
     self.s2.setRange(0,100)
     self.padlock = SVGLabel()
     self.padlock.loadFile(os.path.join(icon_path,"padlock_open.svg"))
     self._lock = 0
     self.connect(self.padlock,QtCore.SIGNAL("MousePressed"),self.toggleLock)

     if orientation == QtCore.Qt.Horizontal:
       layout.addWidget(self.s1,0,1,1,1)
       layout.addWidget(self.padlock,1,1,1,1)
       layout.addWidget(self.s2,2,1,1,1)
       layout.addWidget(self.lowerSpin,1,0,2,1)
       layout.addWidget(self.upperSpin,0,2,2,1)
       lowerLabel = QtGui.QLabel(self.tr("Lower value"))
       upperLabel = QtGui.QLabel(self.tr("Upper value"))
       layout.addWidget(lowerLabel,0,0)
       layout.addWidget(upperLabel,2,2)
       layout.setRowStretch(3,2)
       layout.setVerticalSpacing(0)
     else:
       layout.addWidget(self.s1,1,0,1,1)
       layout.addWidget(self.padlock,1,1,1,1)
       layout.addWidget(self.s2,1,2,1,1)
       layout.addWidget(self.lowerSpin,2,0,1,3)
       layout.addWidget(self.upperSpin,0,0,1,3)
       layout.setColumnStretch(3,2)
       layout.setHorizontalSpacing(0)

     self.setLayout(layout)
     self.sliderScale = 1.
     self.oldS1Val = self.s1.value()
     self.oldS2Val = self.s2.value()
     self.connect(self.s1,QtCore.SIGNAL("valueChanged(int)"),functools.partial(self.valChanged,self.s1))
     self.connect(self.s2,QtCore.SIGNAL("valueChanged(int)"),functools.partial(self.valChanged,self.s2))
     self.connect(self.upperSpin,QtCore.SIGNAL("valueChanged(double)"),functools.partial(self.valChanged,self.upperSpin))
     self.connect(self.lowerSpin,QtCore.SIGNAL("valueChanged(double)"),functools.partial(self.valChanged,self.lowerSpin))

  def maximum(self):
      return self.lowerSpin.maximum()

  def minimum(self):
      return self.lowerSpin.minimum()

  def decimals(self):
      return self.lowerSpin.decimals()

  def setRange(self,lower,upper,minDecimals=1):
     if (upper-lower)<1e-8:
        return
     realRange = upper-lower
     self.sliderScale = 100000.0/realRange
     self.sliderMin = int(math.floor(self.sliderScale*lower))
     self.sliderMax = int(math.floor(self.sliderScale*upper))
     self.s1.setRange(self.sliderMin,self.sliderMax)
     self.s2.setRange(self.sliderMin,self.sliderMax)
     self.s1.setValue(self.sliderMin)
     self.s2.setValue(self.sliderMax)
     self.lowerSpin.setRange(lower,upper)
     self.upperSpin.setRange(lower,upper)
     if (upper-lower)<1e-2:
       decimals = int(math.ceil(math.log(1./(upper-lower),10))+1)
     else:
       decimals = 2
     self.lowerSpin.setDecimals(max(decimals,minDecimals))
     self.upperSpin.setDecimals(max(decimals,minDecimals))

  def isClamped(self):
     #return self.clampCheck.isChecked()
     return self._lock

  def getValues(self):
     return (self.lowerSpin.value(), self.upperSpin.value())

  def valChanged(self,emitter,val):
     self.s1.blockSignals(True)
     self.s2.blockSignals(True)
     self.lowerSpin.blockSignals(True)
     self.upperSpin.blockSignals(True)
     if emitter is self.lowerSpin:
        self.s1.setValue(int(val*self.sliderScale))
        if self.s1.value()>self.s2.value():
          self.s2.setValue(int(val*self.sliderScale))
     if emitter is self.upperSpin:
        self.s2.setValue(int(val*self.sliderScale))
        if self.s2.value()<self.s1.value():
          self.s1.setValue(int(val*self.sliderScale))
     if self.s2.value() <  self.s1.value() and emitter is self.s2:
       self.s1.setValue(self.s2.value())
     #"""
     elif emitter is self.s2 and self.isClamped():
       diff = self.s2.value() - self.oldS2Val
       self.s1.setValue(self.s1.value()+diff)
     #"""
     if self.s1.value() >  self.s2.value() and emitter is self.s1:
       self.s2.setValue(self.s1.value())
     #"""
     elif emitter is self.s1 and self.isClamped():
       diff = self.s1.value() - self.oldS1Val
       self.s2.setValue(self.s2.value()+diff)
     #"""
     self.lowerSpin.setValue(self.s1.value()/self.sliderScale)
     self.upperSpin.setValue(self.s2.value()/self.sliderScale)
     self.s1.blockSignals(False)
     self.s2.blockSignals(False)
     self.lowerSpin.blockSignals(False)
     self.upperSpin.blockSignals(False)
     self.oldS1Val = self.s1.value()
     self.oldS2Val = self.s2.value()
     self.emit(QtCore.SIGNAL("rangeChanged"),(self.lowerSpin.value(), self.upperSpin.value()))

if __name__ == "__main__":

  def rangeChanged(vals):
     min,max = vals
     print min,max, max-min

  app = QtGui.QApplication(sys.argv)

  if sys.argv.count('-v'):
    slider = UpperLowerSlider(QtCore.Qt.Vertical)
  else:
    slider = UpperLowerSlider(QtCore.Qt.Horizontal)

  slider.setRange(-10,120)

  app.connect(slider,QtCore.SIGNAL("rangeChanged"),rangeChanged)
  slider.show()
  slider.raise_()

  sys.exit(app.exec_())


