Skip to content
Snippets Groups Projects
Commit 66b4a092 authored by Patrick Godwin's avatar Patrick Godwin
Browse files

matplotlibhelper.py: fix Gst errors when importing module

parent 73717893
No related branches found
No related tags found
No related merge requests found
...@@ -26,17 +26,18 @@ __all__ = ("padtemplate", "figure", "render", "BaseMatplotlibTransform") ...@@ -26,17 +26,18 @@ __all__ = ("padtemplate", "figure", "render", "BaseMatplotlibTransform")
import gi import gi
gi.require_version('Gst', '1.0') gi.require_version('Gst', '1.0')
from gi.repository import GObject gi.require_version('GstBase', '1.0')
from gi.repository import Gst from gi.repository import GObject, Gst, GstBase
GObject.threads_init() GObject.threads_init()
Gst.init(None) Gst.init(None)
from gstlal.pipeutil import * from gstlal.pipeutil import *
from gstlal import pipeio from gstlal import pipeio
"""Pad template suitable for producing video frames using Matplotlib. """Pad template suitable for producing video frames using Matplotlib.
The Agg backend supports rgba, argb, and bgra.""" The Agg backend supports rgba, argb, and bgra."""
padtemplate = Gst.PadTemplate( padtemplate = Gst.PadTemplate.new(
"src", "src",
Gst.PadDirection.SRC, Gst.PadPresence.ALWAYS, Gst.PadDirection.SRC, Gst.PadPresence.ALWAYS,
Gst.caps_from_string(""" Gst.caps_from_string("""
...@@ -93,7 +94,7 @@ def render(fig, buf, (width, height), fmt): ...@@ -93,7 +94,7 @@ def render(fig, buf, (width, height), fmt):
buf.datasize = datasize buf.datasize = datasize
class BaseMatplotlibTransform(Gst.BaseTransform): class BaseMatplotlibTransform(GstBase.BaseTransform):
"""Base class for transform elements that use Matplotlib to render video.""" """Base class for transform elements that use Matplotlib to render video."""
__gsttemplates__ = padtemplate __gsttemplates__ = padtemplate
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment