# whimsical feature to zoom in search of interesting items

import random
import operator

import gtk

import dialog

def show_autozoom(parent,f):
    AutozoomDialog.show(parent,f)
    
class AutozoomDialog(dialog.T):
    def __init__(self,main_window,f):
        dialog.T.__init__(
            self,
            _("Autozoom"),
            main_window,
            gtk.DIALOG_DESTROY_WITH_PARENT,
            (gtk.STOCK_CLOSE, gtk.RESPONSE_CLOSE))

        self.f = f
        self.tips = gtk.Tooltips()
        
        self.table = gtk.Table(2,2)
        self.vbox.add(self.table)
        
        self.zoombutton = gtk.ToggleButton(_("Start _Zooming"))
        self.tips.set_tip(self.zoombutton,_("Zoom into interesting areas automatically"))
        self.zoombutton.set_use_underline(True)
        self.zoombutton.connect('toggled',self.onZoomToggle)
        f.connect('status-changed',self.onStatusChanged)

        self.table.attach(self.zoombutton,0,2,0,1,gtk.EXPAND | gtk.FILL, 0, 2, 2)

        self.minsize = 1.0E-13 # FIXME, should calculate this better

        self.minsize_entry = gtk.Entry()
        self.tips.set_tip(self.minsize_entry,_("Stop zooming when size of fractal is this small"))
        minlabel = gtk.Label(_("_Min Size"))
        self.table.attach(minlabel,0,1,1,2,0,0,2,2)
        minlabel.set_use_underline(True)
        minlabel.set_mnemonic_widget(self.minsize_entry)

        def set_entry(*args):
            self.minsize_entry.set_text("%g" % self.minsize)

        def change_entry(*args):
            m = float(self.minsize_entry.get_text())
            if m != 0.0 and m != self.minsize:
                self.minsize = m
                set_entry()
            return False
        
        self.connect('focus-out-event',change_entry)
        set_entry()

        self.table.attach(self.minsize_entry,
                          1,2,1,2,
                          gtk.EXPAND | gtk.FILL, 0, 2, 2)

    def show(parent, f):
        dialog.T.reveal(AutozoomDialog, True, parent, None, f)

    show = staticmethod(show)


    def onZoomToggle(self,*args):
        if self.zoombutton.get_active():
            self.zoombutton.child.set_text_with_mnemonic("Stop _Zooming")
            self.select_quadrant_and_zoom()
        else:
            self.zoombutton.child.set_text_with_mnemonic("Start _Zooming")
            
    def select_quadrant_and_zoom(self,*args):
        (wby2,hby2) = (self.f.width/2,self.f.height/2)
        (w,h) = (self.f.width,self.f.height)
        regions = [ (0,   0,   wby2,hby2),# topleft
                    (wby2,0,   w,   hby2),# topright
                    (0,   hby2,wby2,h),   # botleft
                    (wby2,hby2,w,   h)]   # botright   

        counts = [self.f.count_colors(r) for r in regions]
        m = max(counts)
        i = counts.index(m)

        # some level of randomness
        j = random.randrange(0,4)
        if float(counts[j]) / counts[i] > 0.75:
            i = j
            
        #print "counts: %s max %d i %d" % (counts,m,i)
        
        # centers of each quadrant
        coords = [(1,1),(3,1),(1,3),(3,3)]

        (x,y) = coords[i]
        self.f.recenter(x * self.f.width/4, y * self.f.height/4, 0.75)
            
    def onStatusChanged(self,f,status_val):
        if status_val == 0:
            # done drawing current fractal.
            if self.zoombutton.get_active():
                if self.f.get_param(self.f.MAGNITUDE) > self.minsize:
                    self.select_quadrant_and_zoom()
                else:
                    self.zoombutton.set_active(False)