Homec4science
Diffusion Slides (master) Browse

R3683/latex_printing.py
86efe2e51d68master

Last Change
Raw File

latex_printing.py
View Options

from sympy import *
import sympy
import IPython
from IPython.display import display, Math, Latex, display_latex, display_png
from IPython.core.formatters import PNGFormatter
def _print_latex_png(o):
o = o._repr_latex_()
preamble = None
dvioptions = None
from io import BytesIO
exprbuffer = BytesIO()
preview(o, output='png', viewer='BytesIO',
outputbuffer=exprbuffer, preamble=preamble,
dvioptions=dvioptions)
return exprbuffer.getvalue()
def init_printing():
sympy.init_printing()
from IPython.core.interactiveshell import InteractiveShell
#f = InteractiveShell.instance().display_formatter
f = InteractiveShell.instance().display_formatter.formatters['image/png']
#print f
from IPython.core.interactiveshell import InteractiveShell
f.for_type(IPython.core.display.Math, func=_print_latex_png)
#print f.type_printers
from IPython.display import display, Math
def print_latex(sstr,*args):
args = [latex(a) for a in args]
sstr = sstr.format(*args)
display(Math(latex(sstr)))
from tempfile import NamedTemporaryFile
from tempfile import mkdtemp
VIDEO_TAG = """<video controls>
<source src="data:video/x-m4v;base64,{0}" type="video/mp4">
Your browser does not support the video tag.
</video>"""
def anim_to_html(anim):
if not hasattr(anim, '_encoded_video'):
with NamedTemporaryFile(suffix='.mp4') as f:
anim.save(f.name, fps=1./(anim._interval*1e-3), extra_args=['-vcodec', 'libx264','-pix_fmt', 'yuv420p'])
video = open(f.name, "rb").read()
toto = video.encode("base64")
#print anim._encoded_video
return VIDEO_TAG.format(toto)
from IPython.display import HTML
from matplotlib import animation
import matplotlib.pyplot as plt
import matplotlib.figure
def display_animation(anim):
plt.close(anim._fig)
return HTML(anim_to_html(anim))
from IPython.core.display import display
from IPython.core.interactiveshell import InteractiveShell
from wand.image import Image as WImage
from wand.color import Color
import base64
def display_image(obj, resolution=150,width=None,height=None):
if type(obj) == matplotlib.figure.Figure:
png = InteractiveShell.instance().display_formatter.format(obj)
if type(obj) == str:
fname = obj
img = WImage(filename=fname,resolution=resolution)
img.trim()
with Color('#FDFDFD') as white:
twenty_percent = int(65535 * 0.2)
img.transparent_color(white,alpha=0.0,fuzz=twenty_percent)
if (width is None) and (height is None):
display(img)
return img
png = InteractiveShell.instance().display_formatter.format(img)
size_tag = ""
if width is not None: size_tag += 'width={0} '.format(width)
if height is not None: size_tag += 'height={0} '.format(height)
data = png[0]['image/png']
html = r'<img src="data:image/png;base64,{0}" {1}/>'.format(base64.b64encode(data),size_tag)
display (HTML(html))
return html
import subprocess
import shutil
import os
def makeAnimationFromImageFiles(filenames,framerate):
temp_dir = mkdtemp()
ext = None
for idx, f in enumerate(filenames):
_dir = os.path.dirname(f)
base = os.path.basename(f)
base,_ext = os.path.splitext(base)
ext = _ext
if ext == '.pdf':
_png = os.path.join(_dir,base+".png")
subprocess.call('convert {0} {1}'.format(f,_png),shell=True)
ext = ".png"
f = _png
new_name = "file{0:05d}{1}".format(idx,ext)
new_name = os.path.join(temp_dir,new_name)
#print base, ext
#print new_name
#print "copy {0} to {1}".format(f,new_name)
shutil.copyfile(f,new_name)
pattern = os.path.join(temp_dir,'file%05d'+ext)
with NamedTemporaryFile(suffix='.mp4') as f:
cmd = 'ffmpeg -y -framerate {2} -i {0} -vf "scale=trunc(iw/2)*2:trunc(ih/2)*2" -c:v libx264 -r 30 -pix_fmt yuv420p {1}'.format(pattern,f.name,framerate)
#print cmd
subprocess.call(cmd,shell=True)
video = open(f.name, "rb").read()
toto = video.encode("base64")
shutil.rmtree(temp_dir)
return HTML(VIDEO_TAG.format(toto))
c4science · Help