def main(args=None): if args is None: args = sys.argv[1:] subcommand_descriptions = { 'print': 'get information on model/phase/material properties', 'arrivals': 'print list of phase arrivals', 'paths': 'print ray path details', 'plot-xt': 'plot traveltime vs distance curves', 'plot-xp': 'plot ray parameter vs distance curves', 'plot-rays': 'plot ray propagation paths', 'plot': 'plot combination of ray and traveltime curves', 'plot-model': 'plot velocity model', 'list-models': 'list builtin velocity models', 'list-phase-map': 'show translation table for classic phase names', 'simplify-model': 'create a simplified version of a layered model', 'scatter': 'show details about scattering at model interfaces' } usage = '''cake <subcommand> [options] Subcommands: print %(print)s arrivals %(arrivals)s paths %(paths)s plot-xt %(plot_xt)s plot-xp %(plot_xp)s plot-rays %(plot_rays)s plot %(plot)s plot-model %(plot_model)s list-models %(list_models)s list-phase-map %(list_phase_map)s simplify-model %(simplify_model)s scatter %(scatter)s To get further help and a list of available options for any subcommand run: cake <subcommand> --help '''.strip() % d2u(subcommand_descriptions) usage_sub = 'cake %s [options]' if len(args) < 1: sys.exit('Usage: %s' % usage) command = args[0] descr = subcommand_descriptions.get(command, None) subusage = usage_sub % command if command == 'print': c = optparse((), ('model', 'phases', 'material', 'output_format'), usage=subusage, descr=descr) if 'model' in c: if c.output_format == 'textual': print(c.model) print() elif c.output_format == 'nd': cake.write_nd_model_fh(c.model, sys.stdout) if 'phases' in c: for phase in c.phases: print(phase) print() if 'material' in c: print(c.material.describe()) print() elif command == 'arrivals': c = optparse(('model', 'phases', 'distances'), ('zstart', 'zstop', 'as_degrees'), usage=subusage, descr=descr) print_arrivals( c.model, **c.getn('zstart', 'zstop', 'phases', 'distances', 'as_degrees')) elif command == 'paths': c = optparse(('model', 'phases'), ('zstart', 'zstop', 'as_degrees'), usage=subusage, descr=descr) mod = c.model for path in mod.gather_paths(**c.getn('phases', 'zstart', 'zstop')): print(path.describe(path.endgaps(c.zstart, c.zstop), c.as_degrees)) elif command in ('plot-xt', 'plot-xp', 'plot-rays', 'plot'): if command in ('plot-xt', 'plot'): c = optparse(('model', 'phases'), ('zstart', 'zstop', 'distances', 'as_degrees', 'vred', 'phase_colors'), usage=subusage, descr=descr) else: c = optparse(('model', 'phases'), ('zstart', 'zstop', 'distances', 'as_degrees', 'aspect', 'shade_model', 'phase_colors'), usage=subusage, descr=descr) mod = c.model paths = mod.gather_paths(**c.getn('phases', 'zstart', 'zstop')) if c.distances is not None: arrivals = mod.arrivals( **c.getn('phases', 'zstart', 'zstop', 'distances')) else: arrivals = None if command == 'plot-xp': plot.my_xp_plot(paths, c.zstart, c.zstop, c.distances, c.as_degrees, phase_colors=c.phase_colors) elif command == 'plot-xt': plot.my_xt_plot(paths, c.zstart, c.zstop, c.distances, c.as_degrees, vred=c.vred, phase_colors=c.phase_colors) elif command == 'plot-rays': if c.as_degrees: plot.my_rays_plot_gcs(mod, paths, arrivals, c.zstart, c.zstop, c.distances, phase_colors=c.phase_colors) else: plot.my_rays_plot(mod, paths, arrivals, c.zstart, c.zstop, c.distances, aspect=c.aspect, shade_model=c.shade_model, phase_colors=c.phase_colors) elif command == 'plot': plot.my_combi_plot(mod, paths, arrivals, c.zstart, c.zstop, c.distances, c.as_degrees, vred=c.vred, phase_colors=c.phase_colors) elif command in ('plot-model', ): c = optparse(('model', ), (), usage=subusage, descr=descr) mod = c.model plot.my_model_plot(mod) elif command in ('simplify-model', ): c = optparse(('model', ), ('accuracy', ), usage=subusage, descr=descr) my_simplify_model(c.model, c.accuracy) elif command in ('list-models', ): c = optparse((), (), usage=subusage, descr=descr) for x in cake.builtin_models(): print(x) elif command in ('list-phase-map', ): c = optparse((), (), usage=subusage, descr=descr) defs = cake.PhaseDef.classic_definitions() for k in sorted(defs.keys()): print('%-15s: %s' % (k, ', '.join(defs[k]))) elif command in ('scatter', ): c = optparse(('model', ), ('slowness', 'interface', 'as_degrees'), usage=subusage, descr=descr) print_scatter(c.model, p=c.slowness, interface=c.interface) elif command in ('help-options', ): optparse((), ('model', 'accuracy', 'slowness', 'interface', 'phases', 'distances', 'zstart', 'zstop', 'distances', 'as_degrees', 'material', 'vred'), usage='cake help-options', descr='list all available options') elif command in ('--help', '-h', 'help'): sys.exit('Usage: %s' % usage) else: sys.exit('cake: no such subcommand: %s' % command)
def main(args=None): if args is None: args = sys.argv[1:] subcommand_descriptions = { 'print': 'get information on model/phase/material properties', 'arrivals': 'print list of phase arrivals', 'paths': 'print ray path details', 'plot-xt': 'plot traveltime vs distance curves', 'plot-xp': 'plot ray parameter vs distance curves', 'plot-rays': 'plot ray propagation paths', 'plot': 'plot combination of ray and traveltime curves', 'plot-model': 'plot velocity model', 'list-models': 'list builtin velocity models', 'list-phase-map': 'show translation table for classic phase names', 'simplify-model': 'create a simplified version of a layered model', 'scatter': 'show details about scattering at model interfaces'} usage = '''cake <subcommand> [options] Subcommands: print %(print)s arrivals %(arrivals)s paths %(paths)s plot-xt %(plot_xt)s plot-xp %(plot_xp)s plot-rays %(plot_rays)s plot %(plot)s plot-model %(plot_model)s list-models %(list_models)s list-phase-map %(list_phase_map)s simplify-model %(simplify_model)s scatter %(scatter)s To get further help and a list of available options for any subcommand run: cake <subcommand> --help '''.strip() % d2u(subcommand_descriptions) usage_sub = 'cake %s [options]' if len(args) < 1: sys.exit('Usage: %s' % usage) command = args[0] descr = subcommand_descriptions.get(command, None) subusage = usage_sub % command if command == 'print': c = optparse( (), ('model', 'phases', 'material', 'output_format'), usage=subusage, descr=descr) if 'model' in c: if c.output_format == 'textual': print(c.model) print() elif c.output_format == 'nd': cake.write_nd_model_fh(c.model, sys.stdout) if 'phases' in c: for phase in c.phases: print(phase) print() if 'material' in c: print(c.material.describe()) print() elif command == 'arrivals': c = optparse( ('model', 'phases', 'distances'), ('zstart', 'zstop', 'as_degrees'), usage=subusage, descr=descr) print_arrivals( c.model, **c.getn('zstart', 'zstop', 'phases', 'distances', 'as_degrees')) elif command == 'paths': c = optparse( ('model', 'phases'), ('zstart', 'zstop', 'as_degrees'), usage=subusage, descr=descr) mod = c.model for path in mod.gather_paths(**c.getn('phases', 'zstart', 'zstop')): print(path.describe(path.endgaps(c.zstart, c.zstop), c.as_degrees)) elif command in ('plot-xt', 'plot-xp', 'plot-rays', 'plot'): if command in ('plot-xt', 'plot'): c = optparse( ('model', 'phases'), ('zstart', 'zstop', 'distances', 'as_degrees', 'vred', 'phase_colors'), usage=subusage, descr=descr) else: c = optparse( ('model', 'phases'), ('zstart', 'zstop', 'distances', 'as_degrees', 'aspect', 'shade_model', 'phase_colors'), usage=subusage, descr=descr) mod = c.model paths = mod.gather_paths(**c.getn('phases', 'zstart', 'zstop')) if c.distances is not None: arrivals = mod.arrivals( **c.getn('phases', 'zstart', 'zstop', 'distances')) else: arrivals = None if command == 'plot-xp': plot.my_xp_plot( paths, c.zstart, c.zstop, c.distances, c.as_degrees, phase_colors=c.phase_colors) elif command == 'plot-xt': plot.my_xt_plot( paths, c.zstart, c.zstop, c.distances, c.as_degrees, vred=c.vred, phase_colors=c.phase_colors) elif command == 'plot-rays': if c.as_degrees: plot.my_rays_plot_gcs( mod, paths, arrivals, c.zstart, c.zstop, c.distances, phase_colors=c.phase_colors) else: plot.my_rays_plot( mod, paths, arrivals, c.zstart, c.zstop, c.distances, aspect=c.aspect, shade_model=c.shade_model, phase_colors=c.phase_colors) elif command == 'plot': plot.my_combi_plot( mod, paths, arrivals, c.zstart, c.zstop, c.distances, c.as_degrees, vred=c.vred, phase_colors=c.phase_colors) elif command in ('plot-model',): c = optparse(('model',), (), usage=subusage, descr=descr) mod = c.model plot.my_model_plot(mod) elif command in ('simplify-model',): c = optparse(('model',), ('accuracy',), usage=subusage, descr=descr) my_simplify_model(c.model, c.accuracy) elif command in ('list-models',): c = optparse((), (), usage=subusage, descr=descr) for x in cake.builtin_models(): print(x) elif command in ('list-phase-map',): c = optparse((), (), usage=subusage, descr=descr) defs = cake.PhaseDef.classic_definitions() for k in sorted(defs.keys()): print('%-15s: %s' % (k, ', '.join(defs[k]))) elif command in ('scatter',): c = optparse( ('model',), ('slowness', 'interface', 'as_degrees'), usage=subusage, descr=descr) print_scatter(c.model, p=c.slowness, interface=c.interface) elif command in ('help-options',): optparse( (), ('model', 'accuracy', 'slowness', 'interface', 'phases', 'distances', 'zstart', 'zstop', 'distances', 'as_degrees', 'material', 'vred'), usage='cake help-options', descr='list all available options') elif command in ('--help', '-h', 'help'): sys.exit('Usage: %s' % usage) else: sys.exit('cake: no such subcommand: %s' % command)
def my_simplify_model(mod, accuracy): mod_simple = mod.simplify(max_rel_error=accuracy) cake.write_nd_model_fh(mod_simple, sys.stdout)
def my_simplify_model(mod, accuracy): mod_simple = mod.simplify(max_rel_error=accuracy) cake.write_nd_model_fh(mod_simple, sys.stdout)