54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
# Copyright (c) 2018 Daniel Knüttel #
|
|
# #
|
|
# This file is part of mathbot. #
|
|
# #
|
|
# mathbot is free software: you can redistribute it and/or modify #
|
|
# it under the terms of the GNU General Public License as published by #
|
|
# the Free Software Foundation, either version 3 of the License, or #
|
|
# (at your option) any later version. #
|
|
# #
|
|
# mathbot is distributed in the hope that it will be useful, #
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of #
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the #
|
|
# GNU General Public License for more details. #
|
|
# #
|
|
# You should have received a copy of the GNU General Public License #
|
|
# along with mathbot. If not, see <http://www.gnu.org/licenses/>. #
|
|
# #
|
|
|
|
from io import BytesIO
|
|
import matplotlib
|
|
matplotlib.use("Agg")
|
|
import matplotlib.pyplot as plt
|
|
import importlib
|
|
|
|
|
|
def get_png(tex):
|
|
importlib.reload(plt)
|
|
buf = BytesIO()
|
|
|
|
exc = None
|
|
|
|
try:
|
|
plt.subplot()
|
|
|
|
plt.text(0, 0.6, r"$%s$" % tex, fontsize = 50)
|
|
fig = plt.gca()
|
|
fig.axes.get_xaxis().set_visible(False)
|
|
fig.axes.get_yaxis().set_visible(False)
|
|
|
|
plt.savefig(buf)
|
|
|
|
buf.seek(0)
|
|
|
|
except Exception as e:
|
|
exc = e
|
|
finally:
|
|
plt.clf()
|
|
|
|
if(exc):
|
|
raise exc
|
|
|
|
return buf
|
|
|