Source code for openquake.commands.plot_agg_curve

# -*- coding: utf-8 -*-
# vim: tabstop=4 shiftwidth=4 softtabstop=4
#
# Copyright (C) 2015-2018 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 openquake.baselib import sap
from openquake.commands import engine


[docs]def make_figure(curves): # NB: matplotlib is imported inside since it is a costly import import matplotlib.pyplot as plt loss_types = curves.dtype.names I, R = curves.shape num_lt = len(loss_types) fig = plt.figure() for j, lt in enumerate(loss_types, 1): for i in range(I): ax = fig.add_subplot(num_lt * I, I, i * num_lt + j) ax.grid(True) ax.set_ylabel('%s%s' % (lt, '_ins' if i else '')) ax.set_ylim([0, 1]) for rlzi, c in enumerate(curves[lt][i]): ax.plot(c['losses'], c['poes'], label='rlz-%03d' % rlzi) plt.legend() return plt
@sap.Script def plot_ac(calc_id): """ Aggregate loss curves plotter. """ # read the hazard data dstore = engine.read(calc_id) agg_curve = dstore['agg_curve-rlzs'] plt = make_figure(agg_curve) plt.show() plot_ac.arg('calc_id', 'a computation id', type=int)