Coverage for /usr/local/lib/python3.11/dist-packages/grond/problems/double_dc/problem.py: 45%
71 statements
« prev ^ index » next coverage.py v6.5.0, created at 2023-10-27 13:30 +0000
« prev ^ index » next coverage.py v6.5.0, created at 2023-10-27 13:30 +0000
1import numpy as num
2import logging
4from pyrocko import gf, util
5from pyrocko.guts import String, Float, Dict
7from grond.meta import Forbidden, expand_template, Parameter, \
8 has_get_plot_classes
10from ..base import Problem, ProblemConfig
13guts_prefix = 'grond'
14logger = logging.getLogger('grond.problems.double_dc.problem')
15km = 1e3
16as_km = dict(scale_factor=km, scale_unit='km')
19class DoubleDCProblemConfig(ProblemConfig):
21 ranges = Dict.T(String.T(), gf.Range.T())
22 distance_min = Float.T(default=0.0)
24 def get_problem(self, event, target_groups, targets):
25 self.check_deprecations()
27 if event.depth is None:
28 event.depth = 0.
30 base_source = gf.DoubleDCSource.from_pyrocko_event(event)
31 base_source.stf = gf.HalfSinusoidSTF(duration=event.duration or 0.0)
33 subs = dict(
34 event_name=event.name,
35 event_time=util.time_to_str(event.time))
37 problem = DoubleDCProblem(
38 name=expand_template(self.name_template, subs),
39 base_source=base_source,
40 target_groups=target_groups,
41 targets=targets,
42 ranges=self.ranges,
43 distance_min=self.distance_min,
44 norm_exponent=self.norm_exponent)
46 return problem
49@has_get_plot_classes
50class DoubleDCProblem(Problem):
52 problem_parameters = [
53 Parameter('time', 's', label='Time'),
54 Parameter('north_shift', 'm', label='Northing', **as_km),
55 Parameter('east_shift', 'm', label='Easting', **as_km),
56 Parameter('depth', 'm', label='Depth', **as_km),
57 Parameter('magnitude', label='Magnitude'),
58 Parameter('strike1', 'deg', label='Strike 1'),
59 Parameter('dip1', 'deg', label='Dip 1'),
60 Parameter('rake1', 'deg', label='Rake 1'),
61 Parameter('strike2', 'deg', label='Strike 2'),
62 Parameter('dip2', 'deg', label='Dip 2'),
63 Parameter('rake2', 'deg', label='Rake 2'),
64 Parameter('delta_time', 's', label='$\\Delta$ Time'),
65 Parameter('delta_depth', 'm', label='$\\Delta$ Depth'),
66 Parameter('azimuth', 'deg', label='Azimuth'),
67 Parameter('distance', 'm', label='Distance'),
68 Parameter('mix', label='Mix'),
69 Parameter('duration1', 's', label='Duration 1'),
70 Parameter('duration2', 's', label='Duration 2')]
72 dependants = []
73 distance_min = Float.T(default=0.0)
75 def get_source(self, x):
76 d = self.get_parameter_dict(x)
77 p = {}
78 for k in self.base_source.keys():
79 if k in d:
80 p[k] = float(
81 self.ranges[k].make_relative(self.base_source[k], d[k]))
83 stf1 = gf.HalfSinusoidSTF(duration=float(d.duration1))
84 stf2 = gf.HalfSinusoidSTF(duration=float(d.duration2))
86 source = self.base_source.clone(stf1=stf1, stf2=stf2, **p)
87 return source
89 def make_dependant(self, xs, pname):
90 if xs.ndim == 1:
91 return self.make_dependant(xs[num.newaxis, :], pname)[0]
93 raise KeyError(pname)
95 def pack(self, source):
96 arr = self.get_parameter_array(source)
97 for ip, p in enumerate(self.parameters):
98 if p.name == 'time':
99 arr[ip] -= self.base_source.time
100 if p.name == 'duration1':
101 arr[ip] = source.stf1.duration if source.stf1 else 0.0
102 if p.name == 'duration2':
103 arr[ip] = source.stf2.duration if source.stf2 else 0.0
104 return arr
106 def random_uniform(self, xbounds, rstate, fixed_magnitude=None):
107 x = num.zeros(self.nparameters)
108 for i in range(self.nparameters):
109 x[i] = rstate.uniform(xbounds[i, 0], xbounds[i, 1])
111 if fixed_magnitude is not None:
112 x[4] = fixed_magnitude
114 return x.tolist()
116 def preconstrain(self, x):
117 source = self.get_source(x)
118 if any(self.distance_min > source.distance_to(t)
119 for t in self.waveform_targets):
120 raise Forbidden()
122 return num.array(x, dtype=float)
124 @classmethod
125 def get_plot_classes(cls):
126 from . import plot as ddcplot
127 from .. import plot
129 plots = super(DoubleDCProblem, cls).get_plot_classes()
130 plots.extend([plot.HudsonPlot, plot.MTDecompositionPlot,
131 plot.MTLocationPlot, plot.MTFuzzyPlot,
132 ddcplot.DoubleDCDecompositionPlot])
133 return plots
136__all__ = '''
137 DoubleDCProblem
138 DoubleDCProblemConfig
139'''.split()