1import numpy as num
2import logging
4from pyrocko import gf, util
5from pyrocko.guts import String, Float, Dict, Int
7from grond.meta import expand_template, Parameter, has_get_plot_classes
9from ..base import Problem, ProblemConfig
11guts_prefix = 'grond'
12logger = logging.getLogger('grond.problems.rectangular.problem')
13km = 1e3
14as_km = dict(scale_factor=km, scale_unit='km')
17class RectangularProblemConfig(ProblemConfig):
19 ranges = Dict.T(String.T(), gf.Range.T())
20 decimation_factor = Int.T(default=1)
21 distance_min = Float.T(default=0.)
22 nthreads = Int.T(default=4)
24 def get_problem(self, event, target_groups, targets):
25 if self.decimation_factor != 1:
26 logger.warning(
27 'Decimation factor for rectangular source set to %i. Results '
28 'may be inaccurate.' % self.decimation_factor)
30 base_source = gf.RectangularSource.from_pyrocko_event(
31 event,
32 anchor='top',
33 decimation_factor=self.decimation_factor)
35 subs = dict(
36 event_name=event.name,
37 event_time=util.time_to_str(event.time))
39 problem = RectangularProblem(
40 name=expand_template(self.name_template, subs),
41 base_source=base_source,
42 distance_min=self.distance_min,
43 target_groups=target_groups,
44 targets=targets,
45 ranges=self.ranges,
46 norm_exponent=self.norm_exponent,
47 nthreads=self.nthreads)
49 return problem
52@has_get_plot_classes
53class RectangularProblem(Problem):
55 problem_parameters = [
56 Parameter('east_shift', 'm', label='Easting', **as_km),
57 Parameter('north_shift', 'm', label='Northing', **as_km),
58 Parameter('depth', 'm', label='Depth', **as_km),
59 Parameter('length', 'm', label='Length', optional=False, **as_km),
60 Parameter('width', 'm', label='Width', optional=False, **as_km),
61 Parameter('slip', 'm', label='Slip', optional=False),
62 Parameter('strike', 'deg', label='Strike'),
63 Parameter('dip', 'deg', label='Dip'),
64 Parameter('rake', 'deg', label='Rake')
65 ]
67 problem_waveform_parameters = [
68 Parameter('nucleation_x', 'offset', label='Nucleation X'),
69 Parameter('nucleation_y', 'offset', label='Nucleation Y'),
70 Parameter('time', 's', label='Time'),
71 Parameter('velocity', 'm/s', label='Rupture Velocity')
72 ]
74 dependants = []
76 distance_min = Float.T(default=0.0)
78 def pack(self, source):
79 arr = self.get_parameter_array(source)
80 for ip, p in enumerate(self.parameters):
81 if p.name == 'time':
82 arr[ip] -= self.base_source.time
83 return arr
85 def get_source(self, x):
86 d = self.get_parameter_dict(x)
87 p = {}
88 for k in self.base_source.keys():
89 if k in d:
90 p[k] = float(
91 self.ranges[k].make_relative(self.base_source[k], d[k]))
93 source = self.base_source.clone(**p)
95 return source
97 def random_uniform(self, xbounds, rstate, fixed_magnitude=None):
98 x = num.zeros(self.nparameters)
99 for i in range(self.nparameters):
100 x[i] = rstate.uniform(xbounds[i, 0], xbounds[i, 1])
102 return x
104 def preconstrain(self, x):
105 # source = self.get_source(x)
106 # if any(self.distance_min > source.distance_to(t)
107 # for t in self.targets):
108 # raise Forbidden()
109 return x
111 @classmethod
112 def get_plot_classes(cls):
113 plots = super(RectangularProblem, cls).get_plot_classes()
114 return plots
117__all__ = '''
118 RectangularProblem
119 RectangularProblemConfig
120'''.split()