simulation.py

#
# Copyright (C) 2010
# Chad Hanna <chad.hanna@ligo.org>
#
# 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.
#


from ligo import segments
from glue.ligolw import ligolw
from glue.ligolw import lsctables
from glue.ligolw import utils as ligolw_utils
from lal import LIGOTimeGPS
import math

__doc__ = """

**Review status**

+---------------------------------------+------------------------------------------+------------+
| Names                                 | Hash                                     | Date       |
+=======================================+==========================================+============+
| Sathya, Duncan Me, Jolien, Kipp, Chad | 7536db9d496be9a014559f4e273e1e856047bf71 | 2014-05-02 |
+---------------------------------------+------------------------------------------+------------+

"""




class ContentHandler(ligolw.LIGOLWContentHandler):
    pass
lsctables.use_in(ContentHandler)


#
# open ligolw_xml file containing sim_inspiral and create a segment list
#


def sim_inspiral_to_segment_list(fname, pad=1, verbose=False):
    """
    Given an xml file create a segment list that marks the time of an
    injection with padding

    - fname: the xml file name
    - pad: duration in seconds to pad the coalescence time when producing a segment, e.g., [tc-pad, tc+pad)
    """

    # initialization

    seglist = segments.segmentlist()

    # Parse the XML file

    xmldoc = ligolw_utils.load_filename(fname, contenthandler=ContentHandler, verbose=verbose)

    # extract the padded geocentric end times into segment lists

    for row in lsctables.SimInspiralTable.get_table(xmldoc):
        t = LIGOTimeGPS(row.get_time_geocent())
        seglist.append(segments.segment(LIGOTimeGPS(int(math.floor(t-pad))), LIGOTimeGPS(int(math.ceil(t+pad)))))

    # help the garbage collector

    xmldoc.unlink()

    return seglist.coalesce()