Source code for openquake.commands.plot_uhs
# -*- coding: utf-8 -*-
# vim: tabstop=4 shiftwidth=4 softtabstop=4
#
# Copyright (C) 2015-2017 GEM Foundation
#
# OpenQuake is free software: you can redistribute it and/or modify it
# under the terms of the GNU Affero General Public License as published
# by the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# OpenQuake 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 Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with OpenQuake. If not, see <http://www.gnu.org/licenses/>.
from __future__ import print_function
import numpy
from openquake.baselib import sap
from openquake.commonlib import datastore, calc
@sap.Script
def plot_uhs(calc_id, sites='0'):
"""
UHS plotter.
"""
# read the hazard data
dstore = datastore.read(calc_id)
getter = calc.PmapGetter(dstore)
oq = dstore['oqparam']
indices = list(map(int, sites.split(',')))
n_sites = len(dstore['sitecol'])
if not set(indices) <= set(range(n_sites)):
invalid = sorted(set(indices) - set(range(n_sites)))
print('The indices %s are invalid: no graph for them' % invalid)
valid = sorted(set(range(n_sites)) & set(indices))
print('Found %d site(s); plotting %d of them' % (n_sites, len(valid)))
pmaps = getter.get_pmaps(numpy.array(indices))
plt = make_figure(valid, n_sites, oq.imtls, oq.poes, pmaps)
plt.show()
plot_uhs.arg('calc_id', 'a computation id', type=int)
plot_uhs.opt('sites', 'comma-separated string with the site indices')