MouthView.py

# -*- coding: ISO-8859-1 -*-
# generated by wxGlade 0.3.5.1 on Fri Apr 15 17:02:14 2005

# Papagayo-NG, a lip-sync tool for use with several different animation suites
# Original Copyright (C) 2005 Mike Clifton
# Contact information at http://www.lostmarble.com
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

#import os
import wx
import re
from utilities import *


# begin wxGlade: dependencies
# end wxGlade


class MouthView(wx.Panel):
    def __init__(self, *args, **kwds):
        # begin wxGlade: MouthView.__init__
        kwds["style"] = wx.BORDER_SUNKEN | wx.TAB_TRAVERSAL
        wx.Panel.__init__(self, *args, **kwds)

        self.__set_properties()
        self.__do_layout()
        # end wxGlade

        # Other initialization
        self.doc = None
        self.curFrame = 0
        self.oldFrame = 0
        self.currentPhoneme = "rest"
        self.currentMouth = None
        self.mouths = {}
        self.LoadMouths()

        # Connect event handlers
        # window events
        wx.EVT_PAINT(self, self.OnPaint)

    def __set_properties(self):
        # begin wxGlade: MouthView.__set_properties
        self.SetMinSize((200, 200))
        self.SetBackgroundColour(wx.Colour(255, 255, 255))
        # end wxGlade

    def __do_layout(self):
        # begin wxGlade: MouthView.__do_layout
        self.Layout()
        # end wxGlade

    def OnPaint(self, event):
        dc = wx.PaintDC(self)
        dc.SetBackground(wx.Brush(self.GetBackgroundColour()))
        dc.Clear()
        self.DrawMe(dc)

    def DrawMe(self, dc=None):
        if (self.doc is not None) and (self.doc.sound is not None) and (self.doc.sound.IsPlaying()):
            if self.doc.currentVoice is not None:
                phoneme = self.doc.currentVoice.GetPhonemeAtFrame(self.curFrame)
            else:
                phoneme = "rest"
            if phoneme == self.currentPhoneme:
                return
            else:
                self.currentPhoneme = phoneme
        else:
            self.currentPhoneme = "rest"
        if dc is None:
            dc = wx.ClientDC(self)
            freeDC = True
        else:
            freeDC = False
        # dc.BeginDrawing()
        if 1:
            bitmap = self.mouths[self.currentMouth][self.currentPhoneme]
            width, height = self.GetClientSize()
            dc.SetBackground(wx.Brush(self.GetBackgroundColour()))
            dc.Clear()
            dc.DrawBitmap(bitmap, width / 2 - bitmap.GetWidth() / 2, height / 2 - bitmap.GetHeight() / 2)
        else:
            dc.SetBackground(wx.Brush(self.GetBackgroundColour()))
            dc.Clear()
        # dc.EndDrawing()
        if freeDC:
            del dc

    def SetFrame(self, frame):
        self.oldFrame = self.curFrame
        self.curFrame = frame
        self.DrawMe()

    def SetDocument(self, doc):
        self.doc = doc
        self.DrawMe()

    def ProcessMouthDir(self, dirname, names, supportedimagetypes):
        hasImages = False
        for files in names:
            files = files.lower()
            if files.split(".")[-1] in supportedimagetypes:
                hasImages = True
        if not hasImages:
            return
        print(os.path.normpath(dirname), names)
        self.AddMouth(os.path.normpath(dirname), names)

    def LoadMouths(self):
        print(os.path.join(get_main_dir(), "rsrc", "mouths"))
        # wx gives us a string instead of a list of filetypes
        full_pattern = re.compile('[^a-zA-Z0-9.\\\/]|_')
        supportedimagetypes = re.sub(full_pattern, '', wx.Image.GetImageExtWildcard()).split(".")
        for directory, dirnames, filenames in os.walk(os.path.join(get_main_dir(), "rsrc", "mouths")):
            self.ProcessMouthDir(directory, filenames, supportedimagetypes)

    def AddMouth(self, dirname, names):
        bitmaps = {}
        for files in names:
            if ".svn" in files:
                continue
            path = os.path.normpath(os.path.join(dirname, files))
            nolog = wx.LogNull()
            bitmaps[files.split('.')[0]] = wx.Bitmap(path, wx.BITMAP_TYPE_ANY)
            del nolog
        self.mouths[os.path.basename(dirname)] = bitmaps
        if self.currentMouth is None:
            self.currentMouth = os.path.basename(dirname)

# end of class MouthView