1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
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()
|
