_control.py


import os, random

from gi.repository import Gtk, GObject, Gdk

# app
import _appWindow
import _search
import _event

# data
from _config import Config
from _state import State
from _story import Story
from _preferences import Preferences

import _clipboard

# views
from _appBox import AppBox
from _indexView import IndexView
from _pageItemBox import PageItemBox
from _sceneItemBox import SceneItemBox
from _sequenceItemBox import SequenceItemBox
# from _scriptItemBox import ScriptItemBox
from _script import ScriptView

from _storyItemBox import StoryItemBox

# import _input

class Control(object):

    def __init__(self, app):

        # Data objects
        self.app = app
        self.stories = []
        self.index = 0
        self.historyEnabled = False
        self.printCount = 0
        self.category = 'sequence'
        self.saveDir = os.path.expanduser("~") + '/Stories/'
        self.verticalPanePosition = 150
        self.startingUpApp = True
        self.sequenceVisible = False
        self.testingTags = True
        self.updateNamesGlobally = True
        self.eventManager = _event.EventManager(self)
        self.windowPosition = (0,0)

        self.addWordPath = os.path.expanduser(('~')) + '/.config/diego/addedWords'
        self.removeWordPath = os.path.expanduser(('~')) + '/.config/diego/removeWords'

        self.historyDir = None

        self.config = Config(self)
        self.preferences = Preferences(self)
        self.state = State(self)
        self.copyClipboard = _clipboard.Clipboard(self)
        self.selectionClipboard = _clipboard.Clipboard(self)

        # View objects
        self.app.window = _appWindow.Window(self)
        self.appBox = AppBox(self)
        self.panelLabel = Gtk.Label()
        self.indexView = IndexView(self)
        self.pageItemBox = PageItemBox(self)
        self.sceneItemBox = SceneItemBox(self)
        self.sequenceItemBox = SequenceItemBox(self)
        self.storyItemBox = StoryItemBox(self)
        self.scriptView = ScriptView(self)
        self.appHeaderBar = Gtk.HeaderBar()
        self.headerBar = Gtk.HeaderBar()
        self.pathLabel = Gtk.Label()
        self.searchView = _search.View(self)

        self.doMarkSetIndexUpdate = True

        self.trie = None
        self.auxTries = []
        self.mispelledLine = None
        self.addTrie = None
        self.removeTrie = None

        self.clipboard = Gtk.Clipboard.get(Gdk.SELECTION_CLIPBOARD)

        self.scriptViewPanedPosition = 0
        self.settingPanedWithEsc = False

    def mispelledTimer(self):
        GObject.timeout_add(1000, self.removeMispelledTags)

    def removeMispelledTags(self):
        if self.mispelledLine:
            try:
                index = self.scriptView.lines.index(self.mispelledLine)
            except:
                self.mispelledLine = None
            else:
                if self.scriptView.textView.completion is None:
                    self.scriptView.textView.updateLineTag(index)
                self.mispelledLine = None

    def wordMispelled(self, word):

        word = unicode(word)

        # If it's a name, it's not mispelled.
        if word in self.currentStory().names:
            return False

        # If word comes in in all lower, then it must be in the dict as all lower or it's mispelled.
        allLower = True
        for c in word:
            if not c.islower():
                allLower = False

        if allLower:
            if word not in self.trie and word not in self.addTrie:
                return True

            if word in self.removeTrie:
                return True

            return False

        # The dict does not contain uppercase version of words, the capitalized version will be checked. All upper will be check as well for screenplay character names, locations and times.

        lower = word.lower()
        capitalized = word[0].upper()
        if len(word) > 1:
            capitalized += word[1:].lower()

        lower = unicode(lower)
        capitalized = unicode(capitalized)

        notInTrie = word not in self.trie and lower not in self.trie and capitalized not in self.trie
        notInAddTrie = word not in self.addTrie and lower not in self.addTrie and capitalized not in self.addTrie

        if notInTrie and notInAddTrie:
            return True

        if word in self.removeTrie or lower in self.removeTrie or capitalized in self.removeTrie:
            return True

        return False

    def notImplemented(self):
        print "not implemented"

    def p(self, *args):
        self.printCount += 1
        print self.printCount, args

        # if self.printCount == 8:
        #     print

        raiseException = 0
        if raiseException == self.printCount:
            raise Exception()

        return self.printCount

    def load(self, data=True):
        self.historyEnabled = False

        if data:
            self.config.load()
            self.preferences.load()

            # self.state.load() #move to app init

            for story in self.stories:
                story.loadId()

            for story in self.stories:
                story.load()

            self.scriptView.updateTitles()

        self.appBox.load()

        self.indexView.load()
        self.storyItemBox.load()
        self.sequenceItemBox.load()
        self.sceneItemBox.load()
        self.pageItemBox.load()
        self.scriptView.load()

        self.app.updateWindowTitle()

        for story in self.stories:
            self.scriptView.updateTitles(story)

        self.app.window.show_all()
        self.startingUpApp = False

        self.appBox.paned.set_position(self.verticalPanePosition)

        self.historyEnabled = True

    def reset(self, data=True):
        if data:
            self.config.reset()
            self.preferences.reset()
            self.stories = []
            self.index = 0
            self.historyEnabled = False
            self.state.reset()

        self.appBox.reset()

        self.indexView.reset()
        self.pageItemBox.reset()
        self.sceneItemBox.reset()
        self.sequenceItemBox.reset()
        self.storyItemBox.reset()

        self.scriptView.reset()

    def currentStory(self):
        if len(self.stories):
            return self.stories[self.index]
        else:
            return None

    def currentIssue(self):
        pass

    def currentSequence(self):
        return self.currentStory().currentSequence()

    def currentScene(self):
        return self.currentStory().currentScene()

    def currentPage(self):
        return self.currentStory().currentPage()

    def currentLine(self):
        return self.currentStory().currentLine()

    def currentPanel(self):
        cp = self.currentPage()
        cl = self.currentLine()
        panelNumber = 0
        for line in cp.lines:
            if line.tag == 'description':
                panelNumber += 1
            if line == cl:
                break
        return panelNumber

    def storyPaths(self):
        return [story.path for story in self.stories]

    def newStory(self):

        for s in self.stories:
            if not s.saved:
                s.save(pdf=False, rtf=False)

        story = Story(self)
        self.index += 1
        story.createId()

        story.makeHistoryDir()
        story.load()

        self.stories.insert(self.index, story)
        self.reset(data=False)
        self.scriptView.updateTitles()
        self.load(data=False)
        self.storyItemBox.listbox.get_row_at_index(self.index).grab_focus()
        self.storyItemBox.listbox.select_row(self.storyItemBox.listbox.get_row_at_index(self.index))

    def uniqueStoryId(self):
        chars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890"
        id = [list(chars), list(chars), list(chars), list(chars)]
        for item in id:
            random.shuffle(item)
        id = ["".join(item)[:6] for item in id]
        id = "-".join(id)
        return id

    def newSequence(self, prepend=False):
        self.currentStory().newSequence(prepend)
        self.reset(data=False)
        self.scriptView.updateTitles()
        self.load(data=False)
        index = self.currentStory().index.sequence
        self.sequenceItemBox.listbox.get_row_at_index(index).grab_focus()
        self.sequenceItemBox.listbox.select_row(self.sequenceItemBox.listbox.get_row_at_index(index))

    def newScene(self, prepend=False):
        self.currentStory().newScene(prepend)
        self.reset(data=False)
        self.scriptView.updateTitles()
        self.load(data=False)
        index = self.currentStory().index.scene
        self.sceneItemBox.listbox.get_row_at_index(index).grab_focus()
        self.sceneItemBox.listbox.select_row(self.sceneItemBox.listbox.get_row_at_index(index))

    def newPage(self, prepend=False):
        self.currentStory().newPage(prepend)
        self.reset(data=False)
        self.scriptView.updateTitles()
        self.load(data=False)
        index = self.currentStory().index.page
        self.pageItemBox.listbox.get_row_at_index(index).grab_focus()
        self.pageItemBox.listbox.select_row(self.pageItemBox.listbox.get_row_at_index(index))

    def newPanel(self):
        self.currentStory().newPanel()
        self.reset(data=False)
        self.load(data=False)

    def newDialog(self):
        self.currentStory().newDialog()
        self.reset(data=False)
        self.load(data=False)

    def moveCurrentSequenceUp(self):
        self.p('mcsu')

    def moveCurrentSequenceDown(self):
        self.p('mcsd')

    def moveCurrentSceneUp(self):
        item = self.sceneItemBox.getSelectedItem()
        itemIndex = self.sceneItemBox.scenes.index(item)
        scene = self.currentStory().sequences[0].scenes[itemIndex]

        if itemIndex == 0:
            return

        scene = self.currentStory().sequences[0].scenes.pop(itemIndex)

        self.currentStory().sequences[0].scenes.insert(itemIndex - 1, scene)

        self.sceneItemBox.reset()
        self.sceneItemBox.load()
        self.sceneItemBox.loadSceneAtIndex(itemIndex - 1)

    def moveCurrentSceneDown(self):
        item = self.sceneItemBox.getSelectedItem()
        itemIndex = self.sceneItemBox.scenes.index(item)
        scene = self.currentStory().sequences[0].scenes[itemIndex]

        if itemIndex == len(self.sceneItemBox.scenes) - 1:
            return

        scene = self.currentStory().sequences[0].scenes.pop(itemIndex)

        self.currentStory().sequences[0].scenes.insert(itemIndex + 1, scene)

        self.sceneItemBox.reset()
        self.sceneItemBox.load()
        self.sceneItemBox.loadSceneAtIndex(itemIndex + 1)

    def moveCurrentPageUp(self):
        item = self.pageItemBox.getSelectedItem()
        itemIndex = self.pageItemBox.pages.index(item)
        page = self.currentScene().pages[itemIndex]

        if itemIndex == 0:
            return

        page = self.currentScene().pages.pop(itemIndex)

        self.currentScene().pages.insert(itemIndex - 1, page)

        self.pageItemBox.updateNumberated()
        self.pageItemBox.loadPageAtIndex(itemIndex - 1)

    def moveCurrentPageDown(self):
        item = self.pageItemBox.getSelectedItem()
        itemIndex = self.pageItemBox.pages.index(item)
        page = self.currentScene().pages[itemIndex]

        if itemIndex == len(self.pageItemBox.pages) - 1:
            return

        page = self.currentScene().pages.pop(itemIndex)

        self.currentScene().pages.insert(itemIndex + 1, page)

        self.pageItemBox.updateNumberated()
        self.pageItemBox.loadPageAtIndex(itemIndex + 1)

    def updateHistoryColor(self):

        val = 0.94
        selectColor = Gdk.RGBA(0.75, 0.75, 0.85, 1.0)
        forground = Gdk.RGBA(0.0, 0.0, 0.0, 1.0)

        currentScene = self.currentScene()

        if currentScene.undoIndex > 0:

            if currentScene.saveIndex <= 0:
                color = Gdk.RGBA(0.90, 0.90, 1.0, 1.0)

            else:
                color = Gdk.RGBA(val, val, val, 1.0)

            self.scriptView.textView.modify_bg(Gtk.StateType.NORMAL, color.to_color())
            self.scriptView.textView.modify_bg(Gtk.StateType.SELECTED, selectColor.to_color())
            self.scriptView.textView.modify_fg(Gtk.StateType.SELECTED, forground.to_color())

        # New events.
        else:
            color = Gdk.RGBA(1.0, 1.0, 1.0, 1.0)
            self.scriptView.textView.modify_bg(Gtk.StateType.NORMAL, color.to_color())
            self.scriptView.textView.modify_bg(Gtk.StateType.SELECTED, selectColor.to_color())
            self.scriptView.textView.modify_fg(Gtk.StateType.SELECTED, forground.to_color())

        self.scriptView.textView.descriptionTag.props.background_rgba = color
        self.scriptView.textView.characterTag.props.background_rgba = color
        self.scriptView.textView.dialogTag.props.background_rgba = color
        self.scriptView.textView.parentheticTag.props.background_rgba = color
        self.scriptView.textView.sceneHeadingTag.props.background_rgba = color

        for he in self.scriptView.headingEntries:
            he.modify_bg(Gtk.StateType.NORMAL, color.to_color())

    def scroll(self, line, offset=0):
        if len(self.scriptView.lines) -1 < line:
            return
        lineIndex = self.scriptView.lines.index(line)
        lineIter = self.scriptView.textView.buffer.get_iter_at_line(lineIndex)
        lineIter.forward_chars(offset)
        self.scriptView.textView.scroll_to_iter(lineIter, 0.1, False, 0.0, 0.0)
        self.scriptView.textView.buffer.place_cursor(lineIter)
        self.scriptView.textView.grab_focus()

    def timedScroll(self, line, offset, time=250):
        GObject.timeout_add(time, self.scroll, line, offset)

    def selectionOffsets(self):

        bounds = self.scriptView.textView.buffer.get_selection_bounds()

        if len(bounds):

            startIter, endIter = bounds

            # Do not allow selection of zero space char at end of buffer.
            if self.scriptView.textView.endIter().get_offset() == endIter.get_offset():
                endIter.backward_char()

            return startIter.get_offset(), endIter.get_offset()

        return None, None

    def select(self, startOffset, endOffset):
        selectionStartIter = self.scriptView.textView.buffer.get_iter_at_offset(startOffset)
        selectionEndIter = self.scriptView.textView.buffer.get_iter_at_offset(endOffset)
        self.scriptView.textView.buffer.select_range(selectionStartIter, selectionEndIter)

    def timedSelect(self, startOffset, endOffset, time=250):
        GObject.timeout_add(time, self.select, startOffset, endOffset)