def __init__(self, port):
self.stateInfo = State()
transport = TSocket.TSocket('127.0.0.1', port)
# Buffering is critical. Raw sockets are very slow
transport = TTransport.TBufferedTransport(transport)
# Wrap in a protocol
protocol = TBinaryProtocol.TBinaryProtocol(transport)
protocol = TMultiplexedProtocol.TMultiplexedProtocol(
protocol, "standard")
self.client = Standard.Client(protocol)
self.transport = transport
transport.open()
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
class SimpleSwitchAPI:
client = None
transport = None
stateInfo = None
def __init__(self, port):
self.stateInfo = State()
transport = TSocket.TSocket('127.0.0.1', port)
# Buffering is critical. Raw sockets are very slow
transport = TTransport.TBufferedTransport(transport)
# Wrap in a protocol
protocol = TBinaryProtocol.TBinaryProtocol(transport)
protocol = TMultiplexedProtocol.TMultiplexedProtocol(
protocol, "standard")
self.client = Standard.Client(protocol)
self.transport = transport
transport.open()
def __del__(self):
self.transport.close()
def set_counter(self, counter_name, idx, pkts, byts):
counter = self.stateInfo.get_counter_info(counter_name)
if counter.is_direct:
table_name = counter.binding
self.client.bm_mt_write_counter(
0, table_name, idx,
Standard.BmCounterValue(bytes=byts, packets=pkts))
else:
self.client.bm_counter_write(
0, counter_name, idx,
Standard.BmCounterValue(bytes=byts, packets=pkts))
def set_register(self, reg_name, idx, value):
register = self.stateInfo.get_register_info(reg_name)
self.client.bm_register_write(0, register.name, idx, value)
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)