"""
     qtgui/MGGLWidgetOSMesa.py: CCP4MG Molecular Graphics Program
     Copyright (C) 2001-2008 University of York, CCLRC
     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.
"""
import sys
import mgOSMesa as opengl
from PyQt4 import QtCore, QtGui
from MGGLWidgetCore import MGGLWidgetGL

class MGGLWidgetOSMesa(MGGLWidgetGL,QtCore.QObject):
    antialias_samples = 4
    def setDoDraw(self,doit):
        self._doDraw = doit
        print "self.setDoDraw",self._doDraw

    def setBackground(self,r,g,b,a=255):
        if self.transparent:
           MGGLWidgetGL.setBackground(self,r,g,b,0)
        else:
           MGGLWidgetGL.setBackground(self,r,g,b,a)

    def updateGL(self):
        if not self._doDraw: return
        import time
        t1 = time.time()
        self.makeCurrent()
        self.paintGL()
        opengl.glFlush()
        opengl.glFinish()
        t2 = time.time()
        print "Time to draw", t2-t1

    def makeCurrent(self):
        self.osMesaBuffer.makeCurrent()

    def bindTexture(self,texture):
        # To be implemented ?
        pass

    def write(self,f):
        self.osMesaBuffer.write(f)
        if self.antialias:
           print "antialias"
           im = QtGui.QImage(f)
           im = im.scaled(self.width(),self.height(),QtCore.Qt.IgnoreAspectRatio,QtCore.Qt.SmoothTransformation)
           im.save(f)

    def __init__(self,parent=None,share=None,format=None,use_pbuffer=False,pbuffer_size=QtCore.QSize(600,600)):
	QtCore.QObject.__init__(self,parent)
        MGGLWidgetGL.__init__(self,parent,share,format)
	self._size = QtCore.QSize(640,480)
        options = ''
        if sys.argv.count('-OSMesaO'):
          i = sys.argv.index("-OSMesaO")
	  if i+1<len(sys.argv):
            if sys.argv[i+1] != '-quit':
              options = sys.argv[i+1]
              sys.argv.pop(i)
          sys.argv.pop(i)
        print "OSMesa options",options
        self.transparent = False
        self.antialias = False
        self.useShaders = False
        self.renderQuality = 0
        if options:
          exec "options="+options
          if hasattr(options,"has_key"):
            if options.has_key("size"):
              try:
                w,h = options['size'].split('x')
                self._size = QtCore.QSize(int(w),int(h))
              except:
                print "Bad render option",options['size']
            if options.has_key("transparent"):
              try:
                print options['transparent']
                self.transparent = bool(int(options['transparent']))
              except:
                print "Bad render option",options['transparent']
            if options.has_key("antialias"):
              try:
                self.antialias = bool(int(options['antialias']))
              except:
                print "Bad render option",options['antialias']
            if options.has_key("useShaders"):
              try:
                self.useShaders = bool(int(options['useShaders']))
              except:
                print "Bad render option",options['useShaders']
            if options.has_key("renderQuality"):
              try:
                self.renderQuality = int(options['renderQuality'])
              except:
                print "Bad render option",options['renderQuality']
	print "transparent",self.transparent
	print "antialias",self.antialias
	print "useShaders",self.useShaders
	print "renderQuality",self.renderQuality
	self._lights = []
        self.atom_font = QtGui.QFont("times",12)
        self.font = QtGui.QFont("times",18)
	self.fm	 = QtGui.QFontMetrics(self.font)
        self._doDraw = False
        if self.antialias:
          self.osMesaBuffer = opengl.OSMesaBuffer(self.width()*self.antialias_samples,self.height()*self.antialias_samples)
        else:
          self.osMesaBuffer = opengl.OSMesaBuffer(self.width(),self.height())
        self.osMesaBuffer.makeCurrent()
        if self.antialias:
          self.resizeGL(self.width()*self.antialias_samples,self.height()*self.antialias_samples)
        else:
          self.resizeGL(self.width(),self.height())
        self.initializeGL()
        self.drawText = False
        self.drawImages = False
        if self.renderQuality:
           import cprim
           cprim.RenderQuality.SetRenderQuality(1)
        if self.useShaders:
           print "Shaders not yet used my MG's OSMesa mode"
           """
           import os
           import glShaders
           shaderVersion = glShaders.MGGLShader.getShadingLanguageVersion()
           print "Shader version:",shaderVersion

           useClip = True
           try:
             shaderVersion_f = float(shaderVersion.strip().split()[0])
             if shaderVersion_f >= 1.3:
               useClip = True
           except:
             pass

           if useClip:
             #vertexShader_f = open(os.path.join(os.environ["CCP4MG"],"shaders","mg_smooth_light_fog_trans_clip_vertex.glsl"))
             vertexShader_f = open(os.path.join(os.environ["CCP4MG"],"shaders","osmesa_vertex.glsl"))
           else:
             vertexShader_f = open(os.path.join(os.environ["CCP4MG"],"shaders","mg_smooth_light_fog_trans_vertex.glsl"))
           vertexShader = vertexShader_f.read()
           vertexShader_f.close()
           #fragmentShader_f = open(os.path.join(os.environ["CCP4MG"],"shaders","mg_smooth_light_fog_trans_fragment.glsl"))
           fragmentShader_f = open(os.path.join(os.environ["CCP4MG"],"shaders","osmesa_fragment.glsl"))
           fragmentShader = fragmentShader_f.read()
           fragmentShader_f.close()
           self.setShader(vertexShader,fragmentShader)
           self.setShaderState(1)
           """




    def DrawAxes(self):
        return # We cannot do textures, yet
        self.makeCurrent()
        if not self.show_axes: return
	MGGLWidgetGL.DrawAxes(self)

    def SetGO(self, go_in=None,redraw=0):
        self.go=go_in

    def setMouseBindings(self,bindings={}):
	pass