ipython_startup.py

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