diff options
| -rwxr-xr-x | bin/chroma-geo | 61 | ||||
| -rw-r--r-- | setup.py | 2 |
2 files changed, 62 insertions, 1 deletions
diff --git a/bin/chroma-geo b/bin/chroma-geo new file mode 100755 index 0000000..5113609 --- /dev/null +++ b/bin/chroma-geo @@ -0,0 +1,61 @@ +#!/usr/bin/env python +#--*-python-*- +import optparse +import sys +import time + +from chroma.cache import Cache +from chroma.loader import load_geometry_from_string +from chroma.log import logger, logging + +def list_geo(cache, args): + geometries = cache.list_geometry() + print 'List of cached geometries:' + print '\n'.join(geometries) + +def save(cache, args): + start = time.time() + geometry = load_geometry_from_string(args[0]) + print 'Geometry generated in %1.1f seconds.' % (time.time() - start) + + start = time.time() + cache.save_geometry(args[1], geometry) + print 'Geometry saved to cache in %1.1f seconds.' % (time.time() - start) + +def remove(cache, args): + cache.remove_geometry(args[0]) + print 'Geometry "%s" removed.' % args[0] + +commands = { 'list' : list_geo, + 'save' : save, + 'remove' : remove } + + +def main(): + logger.setLevel(logging.INFO) + + parser = optparse.OptionParser('%prog <cmd> <options>') + parser.add_option('-c', '--cache', dest='cache', + default=None, help='Chroma cache directory') + options, args = parser.parse_args() + + if len(args) < 1: + sys.exit(parser.format_help()) + + if options.cache is None: + cache = Cache() + else: + cache = Cache(options.cache) + + cmd = args[0] + cmd_args = args[1:] + + if cmd in commands: + commands[cmd](cache, cmd_args) + else: + print 'error: unknown cmd %s' % cmd + sys.exit(1) + +if __name__ == '__main__': + main() + @@ -25,7 +25,7 @@ setup( packages = find_packages(), include_package_data=True, - scripts = ['bin/chroma-sim', 'bin/chroma-cam'], + scripts = ['bin/chroma-sim', 'bin/chroma-cam', 'bin/chroma-geo'], ext_modules = [ Extension('chroma.generator._g4chroma', ['src/G4chroma.cc'], |