Coverage for /usr/local/lib/python3.11/dist-packages/grond/problems/vlvd/problem.py: 54%

1# https://pyrocko.org/grond - GPLv3 

2# 

3# The Grond Developers, 21st Century 

4import logging 

5 

6from pyrocko import gf, util 

7from pyrocko.guts import String, Float, Dict 

8 

9from grond.meta import Forbidden, expand_template, Parameter, \ 

10 has_get_plot_classes 

11 

12from ..base import Problem, ProblemConfig 

13 

14guts_prefix = 'grond' 

15logger = logging.getLogger('grond.problems.volume_point') 

16km = 1e3 

17as_km = dict(scale_factor=km, scale_unit='km') 

18as_km3 = dict(scale_factor=km**3, scale_unit='km^3') 

19 

20 

21class VLVDProblemConfig(ProblemConfig): 

22 ranges = Dict.T(String.T(), gf.Range.T()) 

23 distance_min = Float.T(default=0.0) 

24 

25 def get_problem(self, event, target_groups, targets): 

26 self.check_deprecations() 

27 if event.depth is None: 

28 event.depth = 0. 

29 

30 base_source = gf.VLVDSource.from_pyrocko_event(event) 

31 base_source.stf = gf.HalfSinusoidSTF(duration=event.duration or 0.0) 

32 

33 subs = dict( 

34 event_name=event.name, 

35 event_time=util.time_to_str(event.time)) 

36 

37 problem = VLVDProblem( 

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) 

45 

46 return problem 

47 

48 

49@has_get_plot_classes 

50class VLVDProblem(Problem): 

51 

52 problem_parameters = [ 

53 Parameter('north_shift', 'm', label='Northing', **as_km), 

54 Parameter('east_shift', 'm', label='Easting', **as_km), 

55 Parameter('depth', 'm', label='Depth', **as_km), 

56 

57 Parameter('dip', 'deg', label='Dip'), 

58 Parameter('azimuth', 'deg', label='Azimuth'), 

59 Parameter('clvd_moment', 'Nm', label='CLVD Moment'), 

60 Parameter('volume_change', 'm^3', label='Volume Change', **as_km3), 

61 

62 ] 

63 

64 problem_waveform_parameters = [ 

65 Parameter('time', 's', label='Time'), 

66 Parameter('duration', 's', label='Duration') 

67 ] 

68 

69 distance_min = Float.T(default=0.0) 

70 

71 def pack(self, source): 

72 arr = self.get_parameter_array(source) 

73 for ip, p in enumerate(self.parameters): 

74 if p.name == 'time': 

75 arr[ip] -= self.base_source.time 

76 return arr 

77 

78 def get_source(self, x): 

79 d = self.get_parameter_dict(x) 

80 

81 p = {k: float(self.ranges[k].make_relative(self.base_source[k], d[k])) 

82 for k in self.base_source.keys() if k in d} 

83 

84 stf = None 

85 if self.has_waveforms: 

86 stf = gf.HalfSinusoidSTF(duration=float(d.duration)) 

87 

88 source = self.base_source.clone(stf=stf, **p) 

89 return source 

90 

91 def preconstrain(self, x): 

92 source = self.get_source(x) 

93 if any(self.distance_min > source.distance_to(t) 

94 for t in self.waveform_targets + self.phase_pick_targets): 

95 raise Forbidden() 

96 

97 return x 

98 

99 @classmethod 

100 def get_plot_classes(cls): 

101 from . import plot 

102 plots = super(VLVDProblem, cls).get_plot_classes() 

103 plots.extend([plot.VLVDLocationPlot]) 

104 return plots