Skip to content
Snippets Groups Projects
Commit 5a17d2d0 authored by Soichiro Kuwahara's avatar Soichiro Kuwahara
Browse files

gstlal_cherenkov_plot_rankingstat:

 - Change the plot range for x and y axis
 - add rankingstat.finish() for normalization and kde estimation because such function is added to rankingstat.py
parent acd7449b
No related branches found
No related tags found
No related merge requests found
......@@ -46,7 +46,6 @@ import numpy
from optparse import OptionParser
import sys
from lal import rate
from ligo.lw import utils as ligolw_utils
from gstlal.cherenkov import rankingstat as cherenkov_rankingstat
......@@ -70,6 +69,8 @@ def parse_command_line():
description = "GstLAL-based cherenkov burst ranking statistic plotting tool."
)
parser.add_option("--verbose", action = "store_true", help = "Be verbose.")
parser.add_option("--input-rank", metavar = "filename", help = "Name of the input rankingstat .")
parser.add_option("--output", metavar = "string", help = "Name of output png.")
options, filenames = parser.parse_args()
# save for the process_params table
......@@ -95,7 +96,7 @@ options, filenames = parse_command_line()
#
rankingstat = cherenkov_rankingstat.RankingStat.from_xml(ligolw_utils.load_filename(filenames[0], verbose = options.verbose, contenthandler = cherenkov_rankingstat.LIGOLWContentHandler), "rankingstat")
rankingstat = cherenkov_rankingstat.RankingStat.from_xml(ligolw_utils.load_filename(options.input_rank, verbose = options.verbose, contenthandler = cherenkov_rankingstat.LIGOLWContentHandler), "rankingstat")
#
......@@ -104,9 +105,6 @@ rankingstat = cherenkov_rankingstat.RankingStat.from_xml(ligolw_utils.load_filen
for instrument, pdf in rankingstat.denominator.lnpdfs.items():
# FIXME: this won't live here permanently, just doing it temporarily for now
pdf.normalize()
x,y = pdf.bins.centres()
lnpdf = pdf.at_centres()
......@@ -120,15 +118,16 @@ for instrument, pdf in rankingstat.denominator.lnpdfs.items():
axes.loglog()
axes.grid(which = "both")
axes.set_xlim((4, 500))
axes.set_ylim((.0001, 5))
axes.set_ylim((.00001, 5))
fig.colorbar(mesh, ax = axes)
axes.set_xlabel("$\mathrm{SNR}$")
axes.set_ylabel("$\chi^{2}/ \mathrm{SNR}^{2}$")
fig.savefig("test_pdf_%s_25.png" % instrument)
fig.savefig("%s_%s_pdf.png" % (options.output, instrument))
rankingstat.finish()
rate.filter_array(pdf.array, rate.gaussian_window(4., 4.))
pdf.normalize()
for instrument, pdf in rankingstat.denominator.lnpdfs.items():
x,y = pdf.bins.centres()
lnpdf = pdf.at_centres()
fig = figure.Figure()
......@@ -143,9 +142,9 @@ for instrument, pdf in rankingstat.denominator.lnpdfs.items():
axes.loglog()
axes.grid(which = "both")
axes.set_xlim((4, 500))
axes.set_ylim((.0001, 5))
axes.set_ylim((.00001, 5))
colorbar = fig.colorbar(mesh, ax = axes)
colorbar.set_clim(-30, lnpdf.max())
axes.set_xlabel("$\mathrm{SNR}$")
axes.set_ylabel("$\chi^{2}/ \mathrm{SNR}^{2}$")
fig.savefig("test_kde_pdf_%s_25.png" % instrument)
fig.savefig("%s_%s_kdepdf.png" % (options.output, instrument))
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