telegram-mathbot/mathbot/rendering.py

40 lines
1.6 KiB
Python
Raw Normal View History

2018-01-18 09:50:14 +00:00
# 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/>. #
# #
2018-01-18 08:09:39 +00:00
from io import BytesIO
2018-01-18 10:22:18 +00:00
import matplotlib
matplotlib.use("Agg")
2018-01-18 09:47:45 +00:00
import matplotlib.pyplot as plt
2018-01-18 08:09:39 +00:00
def get_png(tex):
buf = BytesIO()
2018-01-18 09:47:45 +00:00
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)
2018-01-18 08:09:39 +00:00
2018-01-18 09:47:45 +00:00
plt.savefig(buf)
2018-01-18 08:09:39 +00:00
buf.seek(0)
return buf
2018-01-18 09:47:45 +00:00