/
ipython_startup.py
132 lines (98 loc) · 3.66 KB
/
ipython_startup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
import matplotlib.pyplot as plt
from collections import OrderedDict
import warnings
import pickle
import numpy as np
from numpy import pi, sin, cos
import numpy.ma as ma
import matplotlib
from matplotlib.pyplot import figure
from matplotlib.animation import FuncAnimation
from shutil import copyfile
import os
from numpy.core.numeric import array_repr
from seawater import f as coriolis_f
from IPython.core.oinspect import Inspector
from docrepr import sphinxify
from subprocess import Popen, PIPE
from MyFigureUtils import splay_figures
sf = splay_figures
from MyPlotFunctions import flipy, flipx
from MyNumpyTools import minmax, midpoint
from matplotlib.pyplot import clf
from MyInteractive import disp_latlon, click_dist, click_km
from MyMITgcmUtils import open_simulation, get_xgrid
from MyColormaps import red_yellow_grey_cyan_blue as rygcb
from MyXarray import xa_masked_equal as xme
# Work-around for warnings that keeps coming up but aren't problematic
warnings.filterwarnings('ignore', '.*assigned to before global declaration*.')
warnings.filterwarnings('ignore', '.*get_axis_bgcolor function was deprecated*.')
warnings.filterwarnings('ignore', '.*ishold function was deprecated*.')
warnings.filterwarnings('ignore', '.*axes.hold is deprecated*.')
warnings.filterwarnings('ignore','elementwise comparison')
warnings.filterwarnings('ignore','.*requires a non-empty pattern match*.')
warnings.filterwarnings('ignore',".*Couldn't find available_diagnostics.log*.")
warnings.filterwarnings('ignore',".*Not sure what to do with rlev = L*.")
warnings.filterwarnings('ignore',".*matplotlib.backends*.")
matplotlib.font_manager.X11FontDirectories.append('/home/hugke729/.local/share/fonts/')
get_ipython().magic('autocall 2');
# Make background grey, not white
def grey():
plt.gca().set_facecolor('grey')
plt.draw()
def grey_all():
fig = plt.gcf()
for ax in fig.get_axes():
fig.sca(ax)
grey()
# Define functions to load or dump pickle files
def pil(filename):
"""alias for pickle.load(open(filename, 'rb'))"""
return pickle.load(open(filename, 'rb'))
def unpack_dict(D):
for key, value in D.items():
exec("globals()['" + key + "'] = value")
def pid(obj, filename):
"""alias for pickle.dump(obj, open(filename, 'wb'))"""
return pickle.dump(obj, open(filename, 'wb'))
def dbg(obj_name, obj):
"""Hacky function to evaluate obj in base workspace
Useful because then I can play around with it more
"""
exec("globals()['" + obj_name + "'] = obj")
class AD(dict):
"""Attribute dictionary"""
def __init__(self, *args, **kwargs):
super(AD, self).__init__(*args, **kwargs)
self.__dict__ = self
# ipshell = InteractiveShellEmbed()
# ipshell.magic('%load_ext autoreload')
# Add array shape to Numpy's repr
def my_array_repr(arr):
orig_str = array_repr(arr)
return orig_str + '\n\nShape: ' + str(arr.shape) + '\n\n\n'
np.set_string_function(my_array_repr)
def fromfile(filename):
return np.fromfile(filename, '>d')
def equal(ax=None):
ax = plt.gca() if ax is None else ax
ax.set_aspect('equal', 'box-forced')
def doc(fn):
oinfo = Inspector().info(fn)
url = sphinxify.rich_repr(oinfo)
try:
oinfo['name'] = fn.__name__
except AttributeError:
oinfo['name'] = repr(fn)
new_url = os.path.dirname(url) + '/' + oinfo['name'] + '.html'
url = copyfile(url, new_url)
url = 'file://' + url
url_str = url
print(url_str)
# Popen(['chromium-browser', '--app=' + url_str],
# stderr=PIPE, stdout=PIPE)
Popen(['/usr/bin/google-chrome', url_str],
stderr=PIPE, stdout=PIPE)
def runfile(filename):
exec(open(filename).read())
pi2 = 2*pi