Created
January 8, 2012 21:05
-
-
Save fperez/1579699 to your computer and use it in GitHub Desktop.
IPython 0.0.1, a simple script to be loaded as $PYTHONSTARTUP: of historical interest only...
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/usr/bin/env python | |
""" | |
Interactive execution with automatic history, tries to mimic Mathematica's | |
prompt system. This environment's main features are: | |
- Numbered prompts (In/Out) similar to Mathematica. Only actions that produce | |
output (NOT assingments, for example) affect the counter and cache. | |
- The following GLOBAL variables always exist (so don't overwrite them!): | |
_p: stores previous result which generated printable output. | |
_pp: next previous | |
_ppp: next-next previous | |
_cache[]: cache of all previous output, indexed by prompt counter. | |
_pc: short alias for _cache | |
- Global variables named _p<n> are dynamically created (<n> being the | |
prompt counter), such that the following is always true: | |
_p<n> == _pc[<n>] | |
E.g. the 4th result is always available as either _pc[4] or as _p4. | |
Use help() to open Python's builtin help system, which can also be | |
called with optional keywords as arguments. | |
If you are running inside an Emacs buffer, use doc() as dumb-terminal | |
simpler version of help() [doesn't do all that help() does, though]. | |
Current configuration (set _load.. variables to 0 to turn off): | |
Control loading of Numerical Python modules: _load_Numeric = %(_load_Numeric)s | |
Gnuplot Globals GP and gp (module/instance): _load_Gnuplot = %(_load_Gnuplot)s | |
Load gracePlot (2D-plotting only) : _load_gracePlot = %(_load_gracePlot)s | |
Globals for SI units (including g=9.8) : _load_units = %(_load_units)s | |
Starting number for prompt counter : _prompt_ini = %(_prompt_ini)s | |
Number of history items to store in cache : _cache_size = %(_cache_size)s """ | |
#**************************************************************************** | |
# Configure here | |
_load_Numeric = 1 | |
_load_Gnuplot = 1 | |
_load_gracePlot = 1 | |
_load_units = 1 | |
_cache_size = 1000 | |
_prompt_ini = 1 | |
# *** Don't modify below unless you know what you're doing. *** | |
# Crude first version, with minimal object structure. This could be done much | |
# better, by defining a Cache class (probably using weak references or | |
# generators). But it seems to work ok. Haven't checked for memory circularity | |
# problems, though. | |
#**************************************************************************** | |
# Copyright (C) 2001 Fernando Pérez. <fperez@pizero.colorado.edu> | |
# | |
# Distributed under the terms of the GNU General Public License. | |
# | |
# The full text of the GPL is available at: | |
# | |
# http://www.gnu.org/copyleft/gpl.html | |
#**************************************************************************** | |
__author__ = 'Fernando Pérez. <fperez@pizero.colorado.edu>' | |
__version__= '0.1' | |
#**************************************************************************** | |
# Class definitions | |
class _HistPrompt1: | |
"""Simple interactive prompt like Mathematica's.""" | |
def __str__(self): | |
return '\nIn['+`_prompt_count`+']:= ' | |
class _HistPrompt2: | |
"""Simple interactive continuation prompt.""" | |
def __str__(self): | |
return '...'+' '*(len('In['+`_prompt_count`+']:= ')-3) | |
#**************************************************************************** | |
# Function definitions | |
def _history_print(arg): | |
"""Printing with history cache management. | |
This is invoked everytime the interpreter needs to print, and is activated | |
by setting the variable sys.displayhook to it.""" | |
global _p,_pp,_ppp,_cache,_prompt_count | |
# cleanup cache if needed | |
if _prompt_count > _cache_size: | |
print '\nWARNING: History cache size limit (currently '+\ | |
`_cache_size`+') hit.' | |
print 'Flushing cache and resetting history counter...' | |
print 'The only history variable available will be _p with current result.' | |
print 'Increase the variable _cache_size if you need a larger cache.' | |
_prompt_count = _prompt_ini | |
_cache = [] | |
_p,_pp,_ppp = '','','' | |
# fill bottom of cache if initial offset is not zero | |
for n in range(_prompt_ini): _cache.append(None) | |
# delete auto-generated vars from global namespace | |
for n in range(_prompt_ini,_cache_size): | |
exec 'del _p'+`n` in globals() | |
# update cache | |
_cache.append(arg) | |
_ppp = _pp | |
_pp = _p | |
_p = arg | |
exec '_p'+`_prompt_count`+'=_cache[-1]' in globals() | |
print '\nOut['+`_prompt_count`+']=',arg | |
_prompt_count += 1 | |
# end of _history_print() | |
#---------------------------------------------------------------------------- | |
def intro(): | |
"""Show global docstring and config info.""" | |
print __doc__ % globals() | |
#---------------------------------------------------------------------------- | |
def dgrep(pat,*opts): | |
"""Return grep() on dir()+dir(__builtins__). | |
A very common use of grep() when working interactively.""" | |
exec 'data = dir()+dir(__builtins__)' in globals() | |
return grep(pat,data,*opts) | |
# end of dgrep() | |
#---------------------------------------------------------------------------- | |
def idgrep(pat): | |
"""Case-insensitive dgrep()""" | |
exec 'data = dir()+dir(__builtins__)' in globals() | |
return grep(pat,data,case=0) | |
# end of dgrep() | |
#----------------------------------------------------------------------------- | |
def Gnuplot_setup(): | |
"""Setup some convenient globals for Gnuplot plotting, including a global | |
gp Gnuplot object, ready for use.""" | |
global GP,gp | |
GP = Gnuplot | |
gp = GP.Gnuplot() # global plotting process | |
gp.logx = 'set logscale x' | |
gp.logy = 'set logscale y' | |
gp.nolog = 'set nologscale' | |
gp.points = 'set data style points' | |
gp.lines = 'set data style lines' | |
gp.linesp = 'set data style linespoints' | |
gp.errorbars = 'set data style errorbars' | |
# end of Gnuplot_setup() | |
#----------------------------------------------------------------------------- | |
def SI_Units(): | |
"""Define some globals for physical units and constants. | |
The following become available: | |
g,mm,cm,km,gm,s,sec,hr,day,year,deg,ft,lb,inch | |
These can be overwritten at any time. Calling SI_Units() again will reset | |
their values if needed. """ | |
global g,mm,cm,km,gm,s,sec,hr,day,year,deg,ft,lb,inch | |
# physical constants | |
g = 9.8 | |
# units | |
mm = 0.001 # length | |
cm = 0.01 | |
km = 1000. | |
gm = 0.001 # mass | |
s,sec = 1.,1. # time | |
hr = 3600. | |
day = 24*hr | |
year = 365*day | |
deg = pi/180. # angles | |
inch = 0.0254 # british | |
ft = 12*inch | |
lb = 0.454 | |
# end of SI_Units() | |
#----------------------------------------------------------------------------- | |
def import_fail_info(mod_name): | |
"""Inform load failure for a module.""" | |
print """ | |
WARNING: Loading of %(mod_name)s-related modules failed. | |
Fix your configuration or set _load_%(mod_name)s to 0 to prevent | |
seeing this message every time. | |
""" % vars() | |
#**************************************************************************** | |
# Setup everything at global level for history system to work | |
# Python standard modules | |
import sys | |
from math import * | |
# Other modules | |
from Itpl import Itpl # available at http://web.lfw.org/python/Itpl15.py | |
from genutils import * | |
# User-controlled modules and globals-affecting functions | |
if _load_Numeric: | |
try: | |
from Numeric import * # Numeric *must* come after math. | |
from numutils import * | |
# useful for testing infinities in results of array divisions | |
# (which don't raise an exception) | |
infty = Infinity = (array([1])/0.0)[0] | |
except: | |
import_fail_info('Numeric') | |
if _load_Gnuplot: | |
try: | |
import Gnuplot | |
Gnuplot_setup() | |
except: | |
import_fail_info('Gnuplot') | |
if _load_gracePlot: | |
try: | |
from gracePlot import gracePlot | |
except: | |
import_fail_info('gracePlot') | |
if _load_units: | |
SI_Units() | |
# pydoc doesn't exist in Python 2.0 (I think): | |
try: | |
from pydoc import help | |
except: | |
print "Module pydoc not found. help() function unavailable." | |
# Initialize cache | |
_p,_pp,_ppp = '','','' | |
_prompt_count = _prompt_ini | |
_pc = _cache = [] # _pc is just an alias | |
# fill bottom of cache if initial offset is not zero | |
for n in range(_prompt_ini): _cache.append(None) | |
# cleanup global namespace a bit | |
del import_fail_info,n | |
# Set in/out prompts and printing system | |
sys.displayhook = _history_print | |
sys.ps1 = _HistPrompt1() | |
sys.ps2 = _HistPrompt2() | |
# Startup info | |
print '\nPython',sys.version.split('\n')[0] | |
print '\nInteractive Python -- Type intro() for a brief explanation.' | |
#************************* end of file <ipython.py> *********************** |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment