#TODO: the line slicer should listen to all 2DREFRESH events, get the data and slice it
#      before pushing a new 1D data update.

#
#TODO: NEED MAJOR REFACTOR
#


# Debug printout
import math
import wx
from copy import deepcopy

from BaseInteractor import _BaseInteractor
from sans.guicomm.events import NewPlotEvent, StatusEvent,SlicerParameterEvent,EVT_SLICER_PARS
class BoxSum(_BaseInteractor):
    """
         Select an annulus through a 2D plot
    """
    def __init__(self,base,axes,color='black', zorder=3, x_min=0.0025, x_max=0.0025, y_min=0.0025, y_max=0.0025):
        
        _BaseInteractor.__init__(self, base, axes, color=color)
        self.markers = []
        self.axes = axes
        self.qmax = self.base.data2D.xmax
        self.connect = self.base.connect
        
        self.xmin= -1* x_min
        self.ymin= -1* y_min
        self.xmax= x_max
        self.ymax=  y_max
        # center of the figure
        self.center_x= 0.0
        self.center_y= 0.0
       
        ## Number of points on the plot
        self.nbins = 20
        self.count=0
        self.error=0
        
        self.left_line = VerticalLine(self, self.base.subplot,color='blue', zorder=zorder, 
                                        ymin= self.ymin, ymax= self.ymax,
                                        x= self.xmin,
                                        center_x= self.center_x,
                                        center_y= self.center_y)
        self.left_line.qmax = self.qmax
        
        self.right_line= VerticalLine(self, self.base.subplot,color='black', zorder=zorder,
                                     ymin= self.ymin, ymax= self.ymax,
                                     x=self.xmax,
                                     center_x= self.center_x,
                                     center_y= self.center_y)
        self.right_line.qmax = self.qmax
        
        self.top_line= HorizontalLine(self, self.base.subplot,color='green', zorder=zorder,
                                    y= self.ymax,
                                    xmin= self.xmin, xmax= self.xmax,
                                    center_x= self.center_x,
                                    center_y= self.center_y)
        self.top_line.qmax = self.qmax
        
        self.bottom_line= HorizontalLine(self, self.base.subplot,color='red', zorder=zorder,
                                    y =self.ymin,
                                    xmin= self.xmin, xmax= self.xmax,
                                    center_x= self.center_x,
                                    center_y= self.center_y)
        self.bottom_line.qmax = self.qmax
        #self.outer_circle.set_cursor(self.base.qmax/1.8, 0)
        
        self.connect_markers([])
                      
        self.update()
        self._post_data()
        
        # Bind to slice parameter events
        self.base.parent.Bind(EVT_SLICER_PARS, self._onEVT_SLICER_PARS)


    def _onEVT_SLICER_PARS(self, event):
        #printEVT("AnnulusSlicer._onEVT_SLICER_PARS")
        event.Skip()
        if event.type == self.__class__.__name__:
            self.set_params(event.params)
            self.base.update()

    def update_and_post(self):
        self.update()
        self._post_data()

    def save_data(self, path, image, x, y):
        output = open(path, 'w')
        
        data_x, data_y = self.get_data(image, x, y)
        
        output.write("<phi>  <average>\n")
        for i in range(len(data_x)):
            output.write("%g  %g\n" % (data_x[i], data_y[i]))
        output.close()

    def set_layer(self, n):
        self.layernum = n
        self.update()
        
    def clear(self):
        self.clear_markers()
        self.left_line.clear()
        self.right_line.clear()
        self.top_line.clear()
        self.bottom_line.clear()
        
        #self.base.connect.disconnect()
        self.base.parent.Unbind(EVT_SLICER_PARS)
        
    def update(self):
        """
        Respond to changes in the model by recalculating the profiles and
        resetting the widgets.
        """
        if self.left_line.has_move:
            print "left has moved"
            self.left_line.update(mline=[self.center_x, self.center_y],translation=True)
            
            #self.right_line.update(mline= [self.center_x, self.center_y],translation=True)
            self.top_line.update( xmin= self.left_line.x ,xmax= self.right_line.x,
                                  mline= [self.center_x, self.center_y],translation=True)
            self.bottom_line.update(xmin= self.left_line.x ,xmax= self.right_line.x,
                                    mline= [self.center_x, self.center_y],translation=True)
        if self.right_line.has_move:
            print "right has moved"
            self.right_line.update(mline= [self.center_x, self.center_y],translation=True)
            #self.left_line.update(mline=[self.center_x, self.center_y],translation=True)
            #self.left_line.update(xmin= self.right_line.x ,xmax=-1*self.right_line.x)
            self.top_line.update( xmin= self.left_line.x ,xmax= self.right_line.x,
                                  mline= [self.center_x, self.center_y],translation=True)
            self.bottom_line.update(xmin= self.left_line.x ,xmax= self.right_line.x,
                                    mline= [self.center_x, self.center_y],translation=True)
            
            
        if self.bottom_line.has_move:
            print "bottom has moved"
            self.bottom_line.update(mline= [self.center_x, self.center_y],translation=True)
            #self.top_line.update(y= -1*self.top_line.y,translation=True)
            self.left_line.update( ymin= self.bottom_line.y ,ymax= self.top_line.y,
                                   mline= [self.center_x, self.center_y],translation=True)
            self.right_line.update(ymin= self.bottom_line.y ,ymax= self.top_line.y,
                                   mline= [self.center_x, self.center_y],translation=True)
            
        if self.top_line.has_move:
            print "top has moved"
            self.top_line.update(mline= [self.center_x, self.center_y],translation=True)
            
            #self.bottom_line.update(y= -1*self.top_line.y,mline= [self.center_x, self.center_y],
            #                        translation=True )
            self.left_line.update(ymin= self.bottom_line.y ,ymax= self.top_line.y,
                                  mline= [self.center_x, self.center_y],translation=True)
            self.right_line.update(ymin= self.bottom_line.y ,ymax= self.top_line.y,
                                   mline= [self.center_x, self.center_y],translation=True)
            
    
    def save(self, ev):
        """
        Remember the roughness for this layer and the next so that we
        can restore on Esc.
        """
        self.base.freeze_axes()
        self.inner_circle.save(ev)
        self.outer_circle.save(ev)

    def _post_data(self):
        # Compute data
        data = self.base.data2D
        from DataLoader.manipulations import  Boxavg
        radius = math.sqrt(math.pow(self.qmax,2)+math.pow(self.qmax,2))
        x_min= self.left_line.x 
        x_max= self.right_line.x 
        y_min= self.bottom_line.y
        y_max= self.top_line.y
        box =  Boxavg (x_min=x_min, x_max=x_max, y_min=y_min, y_max=y_max)
        self.count, self.error = box(self.base.data2D)
        print "count, error",self.count, self.error
        
                                       
    def moveend(self, ev):
        self.base.thaw_axes()
        
        # Post paramters
        event = SlicerParameterEvent()
        event.type = self.__class__.__name__
        print "event type boxsum: ", event.type
        event.params = self.get_params()
        
        wx.PostEvent(self.base.parent, event)

        self._post_data()
            
    def restore(self):
        """
        Restore the roughness for this layer.
        """
        self.inner_circle.restore()
        self.outer_circle.restore()

    def move(self, x, y, ev):
        """
        Process move to a new position, making sure that the move is allowed.
        """
        print "in move"
        if self.xmin <= x and x <= self.xmax:
            print "has move whole", x
        
    def set_cursor(self, x, y):
        pass
        
    def get_params(self):
        params = {}
        params["x_min"] = self.left_line.x 
        params["x_max"] = self.right_line.x 
        params["y_min"] = self.bottom_line.y
        params["y_max"] = self.top_line.y
        params["count"] = self.count
        params["errors"]= self.error
        return params
    
    
    def get_result(self):
        """
            return the result of box summation
        """
        result={}
        result["count"] = self.count
        result["error"] = self.error
        return result
        
        
    def set_params(self, params):
        
        x_min = -math.fabs(params["x_min"] )
        x_max = math.fabs(params["x_max"] )
        y_min = -math.fabs(params["y_min"])
        y_max = math.fabs(params["y_max"]) 
        
        self.left_line.update(ymin= y_min ,ymax= y_max , x= x_min)
        self.right_line.update(ymin= y_min ,ymax= y_max, x= x_max)
        self.top_line.update( xmin= x_min ,xmax= x_max, y= y_max)
        self.bottom_line.update(xmin= x_min ,xmax= x_max, y=y_min)
        """
        self.left_line.update(mline= [center_x, center_y],ymin= y_min ,ymax= y_max)
        self.right_line.update(mline= [center_x, center_y],ymin= y_min ,ymax= y_max)
        self.top_line.update(mline= [center_x, center_y], xmin= x_min ,xmax= xmax)
        self.bottom_line.update(mline= [center_x, center_y],xmin= xmin ,xmax= xmax)
        """
        
        self._post_data()
    def freeze_axes(self):
        self.base.freeze_axes()
        
    def thaw_axes(self):
        self.base.thaw_axes()

    def draw(self):
        self.base.draw()
class HorizontalLine(_BaseInteractor):
    """
         Select an annulus through a 2D plot
    """
    def __init__(self,base,axes,color='black', zorder=5, y=0.5,
                 xmin=0.0,xmax=0.5,
                 center_x= 0.0,
                 center_y= 0.0):
        
        _BaseInteractor.__init__(self, base, axes, color=color)
        self.markers = []
        self.axes = axes
        # center
        self.center_x = center_x
        self.center_y = center_y
        
        self.y= y - self.center_y
        self.save_y= y- self.center_y
        
        self.xmin = xmin - self.center_x
        self.save_xmin = xmin - self.center_x
        self.xmax = xmax - self.center_x
        self.save_xmax = xmax - self.center_x
        
       
        self.line = self.axes.plot([self.xmin,self.xmax],[self.y,self.y],
                                      linestyle='-', marker='',
                                      color=self.color,
                                      visible=True)[0]
 

        self.npts = 20
        self.has_move=False
        self.connect_markers([self.line])
        self.update()

    def set_layer(self, n):
        self.layernum = n
        self.update()
        
    def clear(self):
        self.clear_markers()
        try:
            
            self.line.remove()
        except:
            # Old version of matplotlib
            for item in range(len(self.axes.lines)):
                del self.axes.lines[0]
        
    def get_radius(self):
        
        return 0
   
    def update(self,xmin=None, xmax=None,y=None, mline=None,translation=False):
        """
        Draw the new roughness on the graph.
        """
        #print "update main line", self.has_move
        if xmin !=None:
            self.xmin = xmin # - self.center_x
        if xmax !=None:
            self.xmax = xmax #- self.center_x
        if y !=None:
            self.y = y #- self.center_y
        
        self.line.set(xdata=[self.xmin,self.xmax], ydata=[self.y,self.y])
     
        
        
    def save(self, ev):
        """
        Remember the roughness for this layer and the next so that we
        can restore on Esc.
        """
        self.save_xmin= self.xmin
        self.save_xmax= self.xmax
       
        self.save_y= self.y
        self.base.freeze_axes()

    def moveend(self, ev):
        
        self.has_move=False
        self.base.moveend(ev)
            
    def restore(self):
        """
        Restore the roughness for this layer.
        """
        self.y= self.save_y
       
    def move(self, x, y, ev):
        """
        Process move to a new position, making sure that the move is allowed.
        """
        self.y= y - self.center_y
        
        self.has_move=True
        self.base.base.update()
        
    def set_cursor(self, x, y):
        self.move(x, y, None)
        self.update()
        
        
    def get_params(self):
        params = {}
        params["radius"] = self.xmin
        params["theta"] = self.xmax
        
        return params
    
    def set_params(self, params):

        x = params["radius"] 
        self.set_cursor(x, self._inner_mouse_y)
        



class VerticalLine(_BaseInteractor):
    """
         Select an annulus through a 2D plot
    """
    def __init__(self,base,axes,color='black', zorder=5, ymin=0.0, 
                 ymax=0.5,x= 0.5,
                   center_x= 0,
                 center_y= 0):
        
        _BaseInteractor.__init__(self, base, axes, color=color)
        self.markers = []
        self.axes = axes
        # x coordinate of the vertical line
        self.center_x= center_x
        self.center_y= center_y
        self.x = x - center_x
        self.save_x = x - center_x
        # minimum value of y coordinate of the vertical line 
        self.ymin = ymin - center_y
        self.save_ymin = ymin - center_y
        # maximum value of y coordinate of the vertical line 
        self.ymax= ymax - center_y
        self.save_ymax= ymax - center_y
        
        # Draw vertical line
        self.line = self.axes.plot([self.x,self.x],[self.ymin,self.ymax],
                                      linestyle='-', marker='',
                                      color=self.color,
                                      visible=True)[0]
      
        self.npts = 20
        # Check vertical line motion
        self.has_move=False
        self.connect_markers([self.line])
        self.update()

    def set_layer(self, n):
        self.layernum = n
        self.update()
        
    def clear(self):
        self.clear_markers()
        try:
            
            self.line.remove()
        except:
            # Old version of matplotlib
            for item in range(len(self.axes.lines)):
                del self.axes.lines[0]
        
    def get_radius(self):
        return 0
    
    def update(self,x=None,ymin=None, ymax=None, mline=None,translation=False):
        """
        Draw the new roughness on the graph.
        """
        if x!=None:
            self.x = x #-self.center_x
        if ymin !=None:
            self.ymin = ymin #- self.center_y
        if ymax !=None:
            self.ymax = ymax #- self.center_y
        
        self.line.set(xdata=[self.x,self.x], ydata=[self.ymin,self.ymax])
     
        
        
    def save(self, ev):
        """
        Remember the roughness for this layer and the next so that we
        can restore on Esc.
        """
        self.save_x= self.x
        self.save_ymin= self.ymin
        self.save_ymax= self.ymax
        
        self.base.freeze_axes()

    def moveend(self, ev):
        
        self.has_move=False
        self.base.moveend(ev)
            
    def restore(self):
        """
        Restore the roughness for this layer.
        """
        self.x = self.save_x
        
        self.ymin=self.save_ymin
        self.ymax=self.save_ymax
    def move(self, x, y, ev):
        """
        Process move to a new position, making sure that the move is allowed.
        """
        self.x = x - self.center_x
       
        self.has_move=True
        self.base.base.update()
        
        
    def set_cursor(self, x, y):
        self.move(x, y, None)
        self.update()
        
        
    def get_params(self):
        params = {}
        params["x"] = self.xmin
        params["ymin"] = self.ymin
        params["ymax"] = self.ymax
        return params
    
    def set_params(self, params):
        """
            Draw a vertical line given some value of params
            @param params: a dictionary containing value for x, ymin , ymax to draw 
            a vertical line
        """
        x = params["x"] 
        ymin = params["ymin"] 
        ymax = params["ymax"] 
        #self.set_cursor(x, self._inner_mouse_y)
        self.update(self,x =x,ymin =ymin, ymax =ymax)