# encoding: utf-8 # -*- test-case-name: IPython.frontend.cocoa.tests.test_cocoa_frontend -*- """PyObjC classes to provide a Cocoa frontend to the IPython.kernel.engineservice.IEngineBase. To add an IPython interpreter to a cocoa app, instantiate an IPythonCocoaController in a XIB and connect its textView outlet to an NSTextView instance in your UI. That's it. Author: Barry Wark """ __docformat__ = "restructuredtext en" #----------------------------------------------------------------------------- # Copyright (C) 2008 The IPython Development Team # # Distributed under the terms of the BSD License. The full license is in # the file COPYING, distributed as part of this software. #----------------------------------------------------------------------------- #----------------------------------------------------------------------------- # Imports #----------------------------------------------------------------------------- import objc import uuid from Foundation import NSObject, NSMutableArray, NSMutableDictionary,\ NSLog, NSNotificationCenter, NSMakeRange,\ NSLocalizedString, NSIntersectionRange from AppKit import NSApplicationWillTerminateNotification, NSBeep,\ NSTextView, NSRulerView, NSVerticalRuler from pprint import saferepr import IPython from IPython.kernel.engineservice import ThreadedEngineService from IPython.frontend.frontendbase import FrontEndBase from twisted.internet.threads import blockingCallFromThread from twisted.python.failure import Failure #------------------------------------------------------------------------------ # Classes to implement the Cocoa frontend #------------------------------------------------------------------------------ # TODO: # 1. use MultiEngineClient and out-of-process engine rather than # ThreadedEngineService? # 2. integrate Xgrid launching of engines class IPythonCocoaController(NSObject, FrontEndBase): userNS = objc.ivar() #mirror of engine.user_ns (key=>str(value)) waitingForEngine = objc.ivar().bool() textView = objc.IBOutlet() def init(self): self = super(IPythonCocoaController, self).init() FrontEndBase.__init__(self, engine=ThreadedEngineService()) if(self != None): self._common_init() return self def _common_init(self): """_common_init""" self.userNS = NSMutableDictionary.dictionary() self.waitingForEngine = False self.lines = {} self.tabSpaces = 4 self.tabUsesSpaces = True self.currentBlockID = self.next_block_ID() self.blockRanges = {} # blockID=>NSRange def awakeFromNib(self): """awakeFromNib""" self._common_init() # Start the IPython engine self.engine.startService() NSLog('IPython engine started') # Register for app termination nc = NSNotificationCenter.defaultCenter() nc.addObserver_selector_name_object_( self, 'appWillTerminate:', NSApplicationWillTerminateNotification, None) self.textView.setDelegate_(self) self.textView.enclosingScrollView().setHasVerticalRuler_(True) r = NSRulerView.alloc().initWithScrollView_orientation_( self.textView.enclosingScrollView(), NSVerticalRuler) self.verticalRulerView = r self.verticalRulerView.setClientView_(self.textView) self._start_cli_banner() def appWillTerminate_(self, notification): """appWillTerminate""" self.engine.stopService() def complete(self, token): """Complete token in engine's user_ns Parameters ---------- token : string Result ------ Deferred result of IPython.kernel.engineservice.IEngineBase.complete """ return self.engine.complete(token) def execute(self, block, blockID=None): self.waitingForEngine = True self.willChangeValueForKey_('commandHistory') d = super(IPythonCocoaController, self).execute(block, blockID) d.addBoth(self._engine_done) d.addCallback(self._update_user_ns) return d def _engine_done(self, x): self.waitingForEngine = False self.didChangeValueForKey_('commandHistory') return x def _update_user_ns(self, result): """Update self.userNS from self.engine's namespace""" d = self.engine.keys() d.addCallback(self._get_engine_namespace_values_for_keys) return result def _get_engine_namespace_values_for_keys(self, keys): d = self.engine.pull(keys) d.addCallback(self._store_engine_namespace_values, keys=keys) def _store_engine_namespace_values(self, values, keys=[]): assert(len(values) == len(keys)) self.willChangeValueForKey_('userNS') for (k,v) in zip(keys,values): self.userNS[k] = saferepr(v) self.didChangeValueForKey_('userNS') def update_cell_prompt(self, result): if(isinstance(result, Failure)): blockID = result.blockID else: blockID = result['blockID'] self.insert_text(self.input_prompt(result=result), textRange=NSMakeRange(self.blockRanges[blockID].location,0), scrollToVisible=False ) return result def render_result(self, result): blockID = result['blockID'] inputRange = self.blockRanges[blockID] del self.blockRanges[blockID] #print inputRange,self.current_block_range() self.insert_text('\n' + self.output_prompt(result) + result.get('display',{}).get('pprint','') + '\n\n', textRange=NSMakeRange(inputRange.location+inputRange.length, 0)) return result def render_error(self, failure): self.insert_text('\n\n'+str(failure)+'\n\n') self.start_new_block() return failure def _start_cli_banner(self): """Print banner""" banner = """IPython1 %s -- An enhanced Interactive Python.""" % \ IPython.__version__ self.insert_text(banner + '\n\n') def start_new_block(self): """""" self.currentBlockID = self.next_block_ID() def next_block_ID(self): return uuid.uuid4() def current_block_range(self): return self.blockRanges.get(self.currentBlockID, NSMakeRange(self.textView.textStorage().length(), 0)) def currentBlock(self): """The current block's text""" return self.textForRange(self.current_block_range()) def textForRange(self, textRange): """textForRange""" ts = self.textView.textStorage() return ts.string().substringWithRange_(textRange) def currentLine(self): block = self.textForRange(self.current_block_range()) block = block.split('\n') return block[-1] def insert_text(self, string=None, textRange=None, scrollToVisible=True): """Insert text into textView at textRange, updating blockRanges as necessary """ if(textRange == None): #range for end of text textRange = NSMakeRange(self.textView.textStorage().length(), 0) for r in self.blockRanges.itervalues(): intersection = NSIntersectionRange(r,textRange) if(intersection.length == 0): #ranges don't intersect if r.location >= textRange.location: r.location += len(string) else: #ranges intersect if(r.location <= textRange.location): assert(intersection.length == textRange.length) r.length += textRange.length else: r.location += intersection.length self.textView.replaceCharactersInRange_withString_( textRange, string) self.textView.setSelectedRange_( NSMakeRange(textRange.location+len(string), 0)) if(scrollToVisible): self.textView.scrollRangeToVisible_(textRange) def replace_current_block_with_string(self, textView, string): textView.replaceCharactersInRange_withString_( self.current_block_range(), string) self.current_block_range().length = len(string) r = NSMakeRange(textView.textStorage().length(), 0) textView.scrollRangeToVisible_(r) textView.setSelectedRange_(r) def current_indent_string(self): """returns string for indent or None if no indent""" if(len(self.currentBlock()) > 0): lines = self.currentBlock().split('\n') currentIndent = len(lines[-1]) - len(lines[-1]) if(currentIndent == 0): currentIndent = self.tabSpaces if(self.tabUsesSpaces): result = ' ' * currentIndent else: result = '\t' * (currentIndent/self.tabSpaces) else: result = None return result # NSTextView delegate methods... def textView_doCommandBySelector_(self, textView, selector): assert(textView == self.textView) NSLog("textView_doCommandBySelector_: "+selector) if(selector == 'insertNewline:'): indent = self.current_indent_string() if(indent): line = indent + self.currentLine() else: line = self.currentLine() if(self.is_complete(self.currentBlock())): self.execute(self.currentBlock(), blockID=self.currentBlockID) self.start_new_block() return True return False elif(selector == 'moveUp:'): prevBlock = self.get_history_previous(self.currentBlock()) if(prevBlock != None): self.replace_current_block_with_string(textView, prevBlock) else: NSBeep() return True elif(selector == 'moveDown:'): nextBlock = self.get_history_next() if(nextBlock != None): self.replace_current_block_with_string(textView, nextBlock) else: NSBeep() return True elif(selector == 'moveToBeginningOfParagraph:'): textView.setSelectedRange_(NSMakeRange( self.current_block_range().location, 0)) return True elif(selector == 'moveToEndOfParagraph:'): textView.setSelectedRange_(NSMakeRange( self.current_block_range().location + \ self.current_block_range().length, 0)) return True elif(selector == 'deleteToEndOfParagraph:'): if(textView.selectedRange().location <= \ self.current_block_range().location): # Intersect the selected range with the current line range if(self.current_block_range().length < 0): self.blockRanges[self.currentBlockID].length = 0 r = NSIntersectionRange(textView.rangesForUserTextChange()[0], self.current_block_range()) if(r.length > 0): #no intersection textView.setSelectedRange_(r) return False # don't actually handle the delete elif(selector == 'insertTab:'): if(len(self.currentLine().strip()) == 0): #only white space return False else: self.textView.complete_(self) return True elif(selector == 'deleteBackward:'): #if we're at the beginning of the current block, ignore if(textView.selectedRange().location == \ self.current_block_range().location): return True else: self.current_block_range().length-=1 return False return False def textView_shouldChangeTextInRanges_replacementStrings_(self, textView, ranges, replacementStrings): """ Delegate method for NSTextView. Refuse change text in ranges not at end, but make those changes at end. """ assert(len(ranges) == len(replacementStrings)) allow = True for r,s in zip(ranges, replacementStrings): r = r.rangeValue() if(textView.textStorage().length() > 0 and r.location < self.current_block_range().location): self.insert_text(s) allow = False self.blockRanges.setdefault(self.currentBlockID, self.current_block_range()).length +=\ len(s) return allow def textView_completions_forPartialWordRange_indexOfSelectedItem_(self, textView, words, charRange, index): try: ts = textView.textStorage() token = ts.string().substringWithRange_(charRange) completions = blockingCallFromThread(self.complete, token) except: completions = objc.nil NSBeep() return (completions,0)