Commit 109a04a9 authored by Daniel Brötzmann's avatar Daniel Brötzmann

[latex] Fix plugin, cleanup code

parent afe3f827
# This file is part of Gajim.
#
# Gajim 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; version 3 only.
#
# Gajim 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 Gajim. If not, see <http://www.gnu.org/licenses/>.
from gi.repository import GObject
from gi.repository import Gtk
from gajim.gtk.settings import SettingsDialog
from gajim.gtk.settings import SpinSetting
from gajim.gtk.const import Setting
from gajim.gtk.const import SettingType
from gajim.plugins.plugins_i18n import _
class LatexPluginConfiguration(SettingsDialog):
def __init__(self, plugin, parent):
self.plugin = plugin
settings = [
Setting('LatexDPISpinSetting', _('PNG DPI'),
SettingType.VALUE, int(self.plugin.config['png_dpi']),
callback=self.on_setting, data='png_dpi',
desc=_('Scale of the rendered PNG file'),
props={'range_': (72, 300)}),
]
SettingsDialog.__init__(self, parent, _('Latex Configuration'),
Gtk.DialogFlags.MODAL, settings, None,
extend=[('LatexDPISpinSetting',
DPISpinSetting)])
def on_setting(self, value, data):
self.plugin.config[data] = value
class DPISpinSetting(SpinSetting):
__gproperties__ = {
"setting-value": (int, 'Size', '', 72, 300, 108,
GObject.ParamFlags.READWRITE), }
def __init__(self, *args, **kwargs):
SpinSetting.__init__(self, *args, **kwargs)
<?xml version="1.0"?>
<interface>
<requires lib="gtk+" version="2.16"/>
<!-- interface-naming-policy toplevel-contextual -->
<object class="GtkTextBuffer" id="textbuffer1">
<property name="text">Plug-in description should be displayed here. This text will be erased during PluginsWindow initialization.</property>
</object>
<object class="GtkWindow" id="window1">
<child>
<object class="GtkVBox" id="vbox1">
<property name="visible">True</property>
<property name="orientation">vertical</property>
<property name="spacing">6</property>
<child>
<object class="GtkHBox" id="hbox111">
<property name="visible">True</property>
<property name="border_width">3</property>
<property name="spacing">6</property>
<child>
<object class="GtkLabel" id="label1">
<property name="visible">True</property>
<property name="xalign">0</property>
<property name="label" translatable="yes">PNG dpi:</property>
</object>
<packing>
<property name="expand">False</property>
<property name="fill">False</property>
<property name="position">0</property>
</packing>
</child>
<child>
<object class="GtkEntry" id="png_dpi_label">
<property name="visible">True</property>
<property name="can_focus">True</property>
<property name="invisible_char">&#x25CF;</property>
<signal name="changed" handler="on_png_dpi_label_changed"/>
</object>
<packing>
<property name="position">1</property>
</packing>
</child>
</object>
<packing>
<property name="expand">False</property>
<property name="position">0</property>
</packing>
</child>
<child>
<object class="GtkHBox" id="hbox1">
<property name="visible">True</property>
<child>
<placeholder/>
</child>
<child>
<object class="GtkButton" id="test_button">
<property name="label" translatable="yes">Test Latex Configuration</property>
<property name="visible">True</property>
<property name="can_focus">True</property>
<property name="receives_default">True</property>
<signal name="clicked" handler="on_test_button_clicked"/>
</object>
<packing>
<property name="expand">False</property>
<property name="position">1</property>
</packing>
</child>
<child>
<placeholder/>
</child>
</object>
<packing>
<property name="expand">False</property>
<property name="position">1</property>
</packing>
</child>
<child>
<object class="GtkLabel" id="result_label">
<property name="visible">True</property>
<property name="label" translatable="yes">Result:
</property>
</object>
<packing>
<property name="position">2</property>
</packing>
</child>
</object>
</child>
</object>
</interface>
This diff is collapsed.
# Copyright (C) 2010-2011 Yves Fischer <yvesf AT xapek.org>
# Copyright (C) 2011 Yann Leboulanger <asterix AT lagaule.org>
#
# This file is part of Gajim.
#
# Gajim 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; version 3 only.
#
# Gajim 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 Gajim. If not, see <http://www.gnu.org/licenses/>.
import random
import os
import logging
from tempfile import mkstemp
from tempfile import mkdtemp
from gi.repository import Gtk
from gi.repository import GdkPixbuf
from gi.repository import GLib
from gajim.common import app
from gajim.plugins.plugins_i18n import _
from latex.util import try_run
from latex.util import write_latex
from latex.util import BLACKLIST
log = logging.getLogger('gajim.p.latex')
class LatexRenderer:
def __init__(self, iter_start, iter_end, widget, png_dpi):
self.widget = widget
self.buffer = widget.get_buffer()
self.code = iter_start.get_text(iter_end)
self.mark_name = 'LatexRendererMark%s' % str(random.randint(0, 1000))
self.mark = self.buffer.create_mark(self.mark_name, iter_start, True)
self.png_dpi = str(png_dpi)
# delete code and show message 'processing'
self.buffer.delete(iter_start, iter_end)
self.buffer.insert(iter_start, _('Processing LaTeX'))
self._start_processing()
def _start_processing(self):
try:
if self._check_code():
self._show_image()
else:
self._show_error(_('There are bad commands!'))
except Exception as err:
self._show_error(_('Error processing LaTeX: %s' % err))
finally:
self.buffer.delete_mark(self.mark)
def _check_code(self):
for bad_cmd in BLACKLIST:
if self.code.find(bad_cmd) != -1:
# Found bad command
return False
return True
def _show_error(self, message):
"""
String -> TextBuffer
"""
iter_mark = self.buffer.get_iter_at_mark(self.mark)
iter_end = iter_mark.copy().forward_search(
_('Processing LaTeX'),
Gtk.TextSearchFlags.TEXT_ONLY, None)[1]
self.buffer.delete(iter_mark, iter_end)
self.buffer.insert(iter_end, message)
def _show_image(self):
"""
Latex -> PNG -> TextBuffer
"""
def _fg_str(fmt):
try:
return [
{'hex' : '+level-colors', 'tex' : '-fg'}[fmt],
app.interface.get_fg_color(fmt)]
except KeyError:
# interface may not be available when we test latex at startup
return []
except AttributeError:
# interface may not be available when we test latext at startup
return {'hex': ['+level-colors', '0x000000'],
'tex': ['-fg', 'rgb 0.0 0.0 0.0']}[fmt]
try:
tmpdir = mkdtemp(prefix='gajim_tex')
tmpfd, tmppng = mkstemp(prefix='gajim_tex', suffix='.png')
os.close(tmpfd)
except Exception:
msg = 'Could not create temporary files for Latex plugin'
log.debug(msg)
self._show_error(
_('latex error: %(error)s\n===ORIGINAL CODE====\n'
'%(code)s') % {
'error': msg,
'code': self.code[2:len(self.code)-2]})
return False
tmpfile = os.path.join(tmpdir, 'gajim_tex')
# build latex string
write_latex(tmpfile + '.tex', self.code[2:len(self.code)-2])
# convert TeX to dvi
exitcode = try_run(
['latex', '--interaction=nonstopmode', tmpfile + '.tex'], tmpdir)
if exitcode == 0:
# convert dvi to png
log.debug('DVI OK')
exitcode = try_run(
['dvipng', '-bg', 'Transparent'] + _fg_str('tex') + \
['-T', 'tight', '-D', self.png_dpi, tmpfile + '.dvi', '-o',
tmpfile + '.png'], tmpdir)
if exitcode:
# dvipng failed, try convert
log.debug('dvipng failed, try convert')
exitcode = try_run(
['convert'] + _fg_str('hex') + \
['-trim', '-density', self.png_dpi,
tmpfile + '.dvi', tmpfile + '.png'], tmpdir)
# remove temp files created by us and TeX
extensions = ['.tex', '.log', '.aux', '.dvi']
for ext in extensions:
try:
os.remove(tmpfile + ext)
except Exception:
pass
if exitcode == 0:
log.debug('PNG OK')
os.rename(tmpfile + '.png', tmppng)
else:
log.debug('PNG FAILED')
os.remove(tmppng)
os.rmdir(tmpdir)
self._show_error(
_('Convertion to image failed\n===ORIGINAL CODE===='
'\n%s') % self.code[2:len(self.code)-2])
return False
log.debug('Loading PNG %s', tmppng)
try:
pixbuf = GdkPixbuf.Pixbuf.new_from_file(tmppng)
log.debug('png loaded')
iter_mark = self.buffer.get_iter_at_mark(self.mark)
iter_end = iter_mark.copy().forward_search(
_('Processing LaTeX'),
Gtk.TextSearchFlags.TEXT_ONLY, None)[1]
log.debug('Delete old Text')
self.buffer.delete(iter_mark, iter_end)
log.debug('Insert pixbuf')
self.buffer.insert_pixbuf(iter_end, pixbuf)
except GLib.GError:
self._show_error(_('Cannot open %s for reading') % tmppng)
log.debug('Cant open %s for reading', tmppng)
finally:
os.remove(tmppng)
# This file is part of Gajim.
#
# Gajim 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; version 3 only.
#
# Gajim 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 Gajim. If not, see <http://www.gnu.org/licenses/>.
import logging
import os
from subprocess import Popen
from subprocess import PIPE
from gajim.plugins.plugins_i18n import _
log = logging.getLogger('gajim.p.latex')
BLACKLIST = [
'\def',
'\\let',
'\\futurelet',
'\\newcommand',
'\\renewcomment',
'\\else',
'\\fi',
'\\write',
'\\input',
'\\include',
'\\chardef',
'\\catcode',
'\\makeatletter',
'\\noexpand',
'\\toksdef',
'\\every',
'\\errhelp',
'\\errorstopmode',
'\\scrollmode',
'\\nonstopmode',
'\\batchmode',
'\\read',
'\\csname',
'\\newhelp',
'\\relax',
'\\afterground',
'\\afterassignment',
'\\expandafter',
'\\noexpand',
'\\special',
'\\command',
'\\loop',
'\\repeat',
'\\toks',
'\\output',
'\\line',
'\\mathcode',
'\\name',
'\\item',
'\\section',
'\\mbox',
'\\DeclareRobustCommand',
'\\[',
'\\]',
]
def try_run(argv, directory):
try:
proc = popen_nt_friendly(argv, directory)
out = proc.communicate()[0]
log.info(out)
return proc.wait()
except Exception as err:
return _('Error executing "%(command)s": %(error)s') % {
'command': " ".join(argv),
'error': str(err)}
def popen_nt_friendly(command, directory):
if os.name == 'nt':
# CREATE_NO_WINDOW
return Popen(command, creationflags=0x08000000, cwd=directory,
stdout=PIPE)
return Popen(command, cwd=directory, stdout=PIPE)
def write_latex(filename, string):
texstr = _get_latex_template(string)
file_ = open(filename, 'w+')
file_.write(texstr)
file_.flush()
file_.close()
def _get_latex_template(code):
template = '''
\\documentclass[12pt]{article}
\\usepackage[dvips]{graphicx}
\\usepackage{amsmath}
\\usepackage{amssymb}
\\pagestyle{empty}
\\begin{document}
\\begin{large}
\\begin{gather*}
%s
\\end{gather*}
\\end{large}
\\end{document}''' % (code)
return template
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment