From 11003147ac4ea947ec5017921019d668cb4953d1 Mon Sep 17 00:00:00 2001 From: Aleksey Lim Date: Tue, 28 Jun 2011 09:23:54 +0000 Subject: Switch to the singular sources tree and releasing scheme --- (limited to 'Jam/Picker.py') diff --git a/Jam/Picker.py b/Jam/Picker.py new file mode 100644 index 0000000..f1030bf --- /dev/null +++ b/Jam/Picker.py @@ -0,0 +1,398 @@ + +import pygtk +pygtk.require( '2.0' ) +import gtk + +import os + +import random #TEMP +import sets + +from common.Util.CSoundClient import new_csound_client +from common.port.scrolledbox import HScrolledBox +import common.Config as Config +from gettext import gettext as _ + +from sugar.graphics.palette import Palette, WidgetInvoker + +from common.Util import ControlStream +from common.Util import InstrumentDB + +from Jam import Block + +class Picker(HScrolledBox): + + def __init__( self, owner, filter = None ): + HScrolledBox.__init__(self) + + self.owner = owner + + # take drawing setup from owner + self.gc = owner.gc + self.colors = owner.colors + self.blockMask = owner.blockMask + + self.filter = filter + + self.desktop = owner.getDesktop() + + self.pickerBox = gtk.HBox() + self.set_viewport(self.pickerBox) + self.modify_bg( gtk.STATE_NORMAL, self.colors["Picker_Bg"] ) + + # spacers + self.pickerBox.pack_start( gtk.Label(" "), True, True ) + self.pickerBox.pack_end( gtk.Label(" "), True, True ) + + self.show_all() + self.scroll = {} + self.scroll[filter] = 0 + + self.blocks = [] + + def addBlock( self, data, name, block ): + # tooltip + invoker = WidgetInvoker(block) + invoker._position_hint = WidgetInvoker.AT_CURSOR + invoker.set_palette(Palette(name)) + + block.add_events( gtk.gdk.BUTTON_PRESS_MASK + | gtk.gdk.BUTTON_RELEASE_MASK + | gtk.gdk.ENTER_NOTIFY_MASK + | gtk.gdk.LEAVE_NOTIFY_MASK + | gtk.gdk.POINTER_MOTION_MASK + | gtk.gdk.POINTER_MOTION_HINT_MASK ) + block.connect( "button-press-event", self.on_button_press ) + block.connect( "button-release-event", self.on_button_release ) + block.connect( "motion-notify-event", self.on_motion_notify ) + block.data = data + + self.blocks.append( block ) + + if self._testAgainstFilter( block ): + self.pickerBox.pack_start( block, False, False, 3 ) + + block.show_all() + + return block + + def getFilter( self ): + return filter + + def setFilter( self, filter ): + if filter == self.filter: + return + + self.scroll[self.filter] = self.get_adjustment().get_value() + + self.filter = filter + + for block in self.pickerBox.get_children()[1:-1]: # outside children are spacers + self.pickerBox.remove( block ) + + for block in self.blocks: + if self._testAgainstFilter( block ): + self.pickerBox.pack_start( block, False, False, 3 ) + + if self.scroll.has_key( filter ): + self.get_adjustment().set_value( self.scroll[filter] ) + else: + self.get_adjustment().set_value( 0 ) + self.scroll[filter] = 0 + + def _testAgainstFilter( self, block ): + return True + + def on_button_press( self, widget, event ): + pass + + def on_button_release( self, widget, event ): + self.desktop.on_button_release( widget, event ) + + def on_motion_notify( self, widget, event ): + self.desktop.on_motion_notify( widget, event ) + + +class Instrument( Picker ): + + def __init__( self, owner, filter = ( "All" ) ): + Picker.__init__( self, owner, filter ) + + self.type = Instrument + + self.instrumentDB = InstrumentDB.getRef() + + all = [] + lab = [] + mic = [] + + for i in self.instrumentDB.getSet( "All" ): + if i.name.startswith('lab'): + lab.append(i) + elif i.name.startswith('mic'): + mic.append(i) + elif not i.kitStage and not i.kit: + all.append(i) + + all += sorted(lab, key=lambda i: i.name) + all += sorted(mic, key=lambda i: i.name) + + for inst in all: + self.addBlock( inst.instrumentId ) + + def addBlock( self, id ): + # match data structure of Block.Instrument + data = { "name": self.instrumentDB.instId[id].nameTooltip, + "id": id } + + win = gtk.gdk.get_default_root_window() + width = Block.Instrument.WIDTH + height = Block.Instrument.HEIGHT + pixmap = gtk.gdk.Pixmap( win, width, height ) + + self.gc.set_clip_rectangle( gtk.gdk.Rectangle( 0, 0, width, height ) ) + + # draw bg + self.gc.foreground = self.colors["Picker_Bg"] + pixmap.draw_rectangle( self.gc, True, 0, 0, width, height ) + + self.gc.set_clip_mask( self.blockMask ) + + # draw border + self.gc.foreground = self.colors["Border_Inactive"] + self.gc.set_clip_origin( -Block.Instrument.MASK_START, 0 ) + pixmap.draw_rectangle( self.gc, True, 0, 0, width, height ) + + # draw block + inst = self.owner.getInstrumentImage( data["id"] ) + self.gc.set_clip_origin( -Block.Instrument.MASK_START, -height ) + pixmap.draw_drawable( self.gc, inst, 0, 0, 0, 0, width, height ) + + image = gtk.Image() + image.set_from_pixmap( pixmap, None ) + + block = gtk.EventBox() + block.modify_bg( gtk.STATE_NORMAL, self.colors["Picker_Bg"] ) + block.add( image ) + + Picker.addBlock( self, data, data["name"], block ) + + def _testAgainstFilter( self, block ): + if "All" in self.filter: + return True + + if self.instrumentDB.getInstrument( block.data["id"] ).category in self.filter: + return True + + return False + + def on_button_press( self, widget, event ): + walloc = widget.get_allocation() + valloc = self.get_viewport_allocation() + loc = ( valloc.x + walloc.x + event.x, -1 ) + + block = self.desktop.addBlock( Block.Instrument, widget.data, loc, True ) + self.desktop.activateInstrument( block ) + + +class Drum( Picker ): + + def __init__( self, owner, filter = None ): + Picker.__init__( self, owner, filter ) + + self.type = Drum + + self.instrumentDB = InstrumentDB.getRef() + + for inst in self.instrumentDB.getSet( "percussions" ): + if self.instrumentDB.instNamed[inst.name].kit: + self.addBlock( inst.instrumentId ) + + def addBlock( self, id ): + # match data structure of Block.Drum + data = { "name": self.instrumentDB.instId[id].nameTooltip, + "id": id } + + win = gtk.gdk.get_default_root_window() + width = Block.Drum.WIDTH + height = Block.Drum.HEIGHT + pixmap = gtk.gdk.Pixmap( win, width, height ) + + self.gc.set_clip_rectangle( gtk.gdk.Rectangle( 0, 0, width, height ) ) + + # draw bg + self.gc.foreground = self.colors["Picker_Bg"] + pixmap.draw_rectangle( self.gc, True, 0, 0, width, height ) + + self.gc.set_clip_mask( self.blockMask ) + + # draw border + self.gc.foreground = self.colors["Border_Inactive"] + self.gc.set_clip_origin( -Block.Drum.MASK_START, 0 ) + pixmap.draw_rectangle( self.gc, True, 0, 0, width, height ) + + # draw block + inst = self.owner.getInstrumentImage( data["id"] ) + self.gc.set_clip_origin( -Block.Drum.MASK_START, -height ) + pixmap.draw_drawable( self.gc, inst, 0, 0, 0, 0, width, height ) + + image = gtk.Image() + image.set_from_pixmap( pixmap, None ) + + block = gtk.EventBox() + block.modify_bg( gtk.STATE_NORMAL, self.colors["Picker_Bg"] ) + block.add( image ) + + Picker.addBlock( self, data, data["name"], block ) + + def on_button_press( self, widget, event ): + walloc = widget.get_allocation() + valloc = self.get_viewport_allocation() + loc = ( valloc.x + walloc.x + event.x, -1 ) + self.desktop.addBlock( Block.Drum, widget.data, loc, True ) + + +class Loop( Picker ): + + def __init__( self, owner, filter = None ): + Picker.__init__( self, owner, filter ) + + self.type = Loop + + self.presetLoops = self._scanDirectory( Config.FILES_DIR+"/Loops" ) + + def _loadFile( self, fullpath, filename ): + if filename[-4:] != ".ttl": + if Config.DEBUG >= 3: print "WARNING: incorrect extension on loop file: " + filename + return -1 + try: + oldPages = sets.Set( self.owner.noteDB.getTune() ) + + ifile = open( fullpath, 'r' ) + ttt = ControlStream.TamTamTable ( self.owner.noteDB ) + ttt.parseFile( ifile ) + ifile.close() + + curPages = sets.Set( self.owner.noteDB.getTune() ) + newPages = curPages.difference( oldPages ) + + if len(newPages) != 1: + print "ERROR: bad loop file, contains more than one page (or none)" + return -1 + + id = newPages.pop() # new pageId + + self.owner.noteDB.getPage( id ).setLocal( False ) # flag as a global page + + self.addBlock( id, filename[:-4] ) + + return id + + except OSError,e: + print 'ERROR: failed to open file %s for reading\n' % ofilename + return -1 + + def _scanDirectory( self, path ): + dirlist = os.listdir( path ) + ids = [] + for fpath in dirlist: + id = self._loadFile( path+"/"+fpath, fpath ) + if id != -1: ids.append(id) + return ids + + def addBlock( self, id, name ): + # match data structure of Block.Loop + data = { "name": _('Loop'), + "id": id } + + self.owner.updateLoopImage( data["id"] ) + loop = self.owner.getLoopImage( data["id"] ) + + page = self.owner.noteDB.getPage( id ) + + win = gtk.gdk.get_default_root_window() + width = Block.Loop.WIDTH[page.beats] + height = Block.Loop.HEIGHT + pixmap = gtk.gdk.Pixmap( win, width, height ) + + self.gc.set_clip_rectangle( gtk.gdk.Rectangle( 0, 0, width, height ) ) + + # draw bg + self.gc.foreground = self.colors["Picker_Bg"] + pixmap.draw_rectangle( self.gc, True, 0, 0, width, height ) + + self.gc.set_clip_mask( self.blockMask ) + self.gc.foreground = self.owner.colors["Border_Inactive"] + + #-- draw head ----------------------------------------- + + # draw border + self.gc.set_clip_origin( -Block.Loop.MASK_START, 0 ) + pixmap.draw_rectangle( self.gc, True, 0, 0, Block.Loop.HEAD, height ) + + # draw block + self.gc.set_clip_origin( -Block.Loop.MASK_START, -height ) + pixmap.draw_drawable( self.gc, loop, 0, 0, 0, 0, Block.Loop.HEAD, height ) + + #-- draw beats ---------------------------------------- + + beats = page.beats - 1 # last beat is drawn with the tail + curx = Block.Loop.HEAD + while beats > 3: + # draw border + self.gc.set_clip_origin( curx-Block.Loop.MASK_BEAT, 0 ) + pixmap.draw_rectangle( self.gc, True, curx, 0, Block.Loop.BEAT_MUL3, height ) + + # draw block + self.gc.set_clip_origin( curx-Block.Loop.MASK_BEAT, -height ) + pixmap.draw_drawable( self.gc, loop, curx, 0, curx, 0, Block.Loop.BEAT_MUL3, height ) + + curx += Block.Loop.BEAT_MUL3 + beats -= 3 + + if beats: + w = Block.Loop.BEAT*beats + + # draw border + self.gc.set_clip_origin( curx-Block.Loop.MASK_BEAT, 0 ) + pixmap.draw_rectangle( self.gc, True, curx, 0, w, height ) + + # draw block + self.gc.set_clip_origin( curx-Block.Loop.MASK_BEAT, -height ) + pixmap.draw_drawable( self.gc, loop, curx, 0, curx, 0, w, height ) + + curx += w + + #-- draw tail ----------------------------------------- + + # draw border + self.gc.set_clip_origin( curx-Block.Loop.MASK_TAIL, 0 ) + pixmap.draw_rectangle( self.gc, True, curx, 0, Block.Loop.TAIL, height ) + + # draw block + self.gc.set_clip_origin( curx-Block.Loop.MASK_TAIL, -height ) + pixmap.draw_drawable( self.gc, loop, curx, 0, curx, 0, Block.Loop.TAIL, height ) + + image = gtk.Image() + image.set_from_pixmap( pixmap, None ) + + block = gtk.EventBox() + block.modify_bg( gtk.STATE_NORMAL, self.colors["Picker_Bg"] ) + block.add( image ) + + Picker.addBlock( self, data, data["name"], block ) + + def on_button_press( self, widget, event ): + walloc = widget.get_allocation() + valloc = self.get_viewport_allocation() + loc = ( valloc.x + walloc.x + event.x, -1 ) + + data = {} + for key in widget.data.keys(): + data[key] = widget.data[key] + + newid = self.owner.noteDB.duplicatePages( [ data["id"] ] )[data["id"]] + self.owner.updateLoopImage( newid ) + data["id"] = newid + + block = self.desktop.addBlock( Block.Loop, data, loc, True ) -- cgit v0.9.1