lal_fakeligosrc.py

# Copyright (C) 2010  Leo Singer
# Copyright (C) 2009  Drew Keppel
#
# This program 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 2 of the License, or (at your
# option) any later version.
#
# This program 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 this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
"""Generate simulated initial LIGO h(t)"""
__author__ = "Drew Keppel <drew.keppel@ligo.org>"

#
# The filter design in this code is by Drew Keppel originally commited to
# the lloid_gui program in Jul 2009 (git hash d7e01fa5be6) and subsequently
# migrated into pipeparts.py.  It was converted into a stand-alone Python
# element by Leo Singer in 2010 (git hash c79657d297)
#


from gstlal import pipeutil
from gstlal.pipeutil import gobject, gst


class lal_fakeligosrc(gst.Bin):

    __gstdetails__ = (
        'Fake LIGO Source',
        'Source',
        __doc__,
        __author__
    )

    __gproperties__ = {
        'blocksize': (
            gobject.TYPE_UINT64,
            'blocksize',
            'Number of samples in each outgoing buffer',
            0, gobject.G_MAXULONG, 16384 * 8 * 1, # min, max, default
            gobject.PARAM_WRITABLE
        ),
        'instrument': (
            gobject.TYPE_STRING,
            'instrument',
            'Instrument name (e.g., "H1")',
            None,
            gobject.PARAM_WRITABLE
        ),
        'channel-name': (
            gobject.TYPE_STRING,
            'channel-name',
            'Channel name (e.g., "LSC-STRAIN")',
            None,
            gobject.PARAM_WRITABLE
        )
    }

    __gsttemplates__ = (
        gst.PadTemplate("src",
            gst.PAD_SRC, gst.PAD_ALWAYS,
            gst.caps_from_string("""
                audio/x-raw-float,
                channels = (int) 1,
                endianness = (int) BYTE_ORDER,
                width = (int) 64,
                rate = (int) 16384
            """)
        ),
    )

    def do_set_property(self, prop, val):
        if prop.name == 'blocksize':
            # Set property on all sources
            for elem in self.iterate_sources():
                elem.set_property('blocksize', val)
        elif prop.name in ('instrument', 'channel-name'):
            self.__tags[prop.name] = val
            tagstring = ','.join('%s="%s"' % kv for kv in self.__tags.iteritems())
            self.__taginject.set_property('tags', tagstring)


    def do_send_event(self, event):
        """
        Send SEEK and EOS events to the source elements, all others
        to the by default bins send all events to the sink
        elements.
        """
        # FIXME:  seeks should go to sink elements as well
        success = True
        if event.type in (gst.EVENT_SEEK, gst.EVENT_EOS):
            for elem in self.iterate_sources():
                success &= elem.send_event(event)
        else:
            for elem in self.iterate_sinks():
                success &= elem.send_event(event)
        return success


    def __init__(self):
        super(lal_fakeligosrc, self).__init__()

        self.__tags = {'units':'strain'}

        # Build first filter chain
        chains = (
            pipeutil.mkelems_in_bin(self,
                ('audiotestsrc', {'wave':'gaussian-noise', 'volume': 5.03407936516e-17, 'samplesperbuffer': 16384}),
                *((('audioiirfilter', {'a': (1.87140685e-05, 3.74281370e-05, 1.87140685e-05), 'b': (1., 1.98861643, -0.98869215)}),) * 14)
            ),
            pipeutil.mkelems_in_bin(self,
                ('audiotestsrc', {'wave': 'gaussian-noise', 'volume': 1.39238913312e-20, 'samplesperbuffer': 16384}),
                ('audioiirfilter', {'a': (9.17933667e-07, 1.83586733e-06, 9.17933667e-07), 'b': (1., 1.99728828, -0.99729195)})
            ),
            pipeutil.mkelems_in_bin(self,
                ('audiotestsrc', {'wave': 'gaussian-noise', 'volume': 2.16333076528e-23, 'samplesperbuffer': 16384})
            ),
            pipeutil.mkelems_in_bin(self,
                ('audiotestsrc', {'wave': 'gaussian-noise', 'volume': 1.61077910675e-20, 'samplesperbuffer': 16384}),
                ('audioiirfilter', {'a': (0.5591789, 0.5591789), 'b': (1., -0.1183578)}),
                ('audioiirfilter', {'a': (0.03780506, -0.03780506), 'b': (1.0, -0.9243905)})
            )
        )

        outputchain = pipeutil.mkelems_in_bin(self,
            ('lal_adder', {'sync': True}),
            ('audioamplify', {'clipping-method': 3, 'amplification': 16384.**.5}),
            ('taginject',)
        )

        for chain in chains:
            chain[-1].link(outputchain[0])

        self.__taginject = outputchain[-1]
        self.add_pad(gst.ghost_pad_new_from_template('src', outputchain[-1].get_static_pad('src'), self.__gsttemplates__[0]))



# Register element class
gobject.type_register(lal_fakeligosrc)

__gstelementfactory__ = (
    lal_fakeligosrc.__name__,
    gst.RANK_NONE,
    lal_fakeligosrc
)