summaryrefslogtreecommitdiff
path: root/bin/chroma-cam
blob: c75f28f62ab28b700d52e6ade27be1619a3cf1d2 (plain)
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
#!/usr/bin/env python
#--*-python-*-

if __name__ == '__main__':
    import optparse
    import sys
    import os

    from chroma import view, build, EventViewer
    from chroma import mesh_from_stl
    from chroma.tools import enable_debug_on_crash
    from chroma.log import logger, logging
    logger.setLevel(logging.INFO)

    parser = optparse.OptionParser('%prog')
    parser.add_option('--debug', dest='debug', action='store_true', 
                      default=False, help='Start python debugger on exception')
    parser.add_option('-r', '--resolution', dest='resolution',
                      help='specify window resolution', default='1024,576')
    parser.add_option('-i', dest='io_file', default=None)
    options, args = parser.parse_args()

    if len(args) < 1:
        sys.exit(parser.format_help())

    if options.debug:
        enable_debug_on_crash()

    size = [int(s) for s in options.resolution.split(',')]

    if os.path.exists(args[0]) and args[0].lower().endswith(('.stl', '.bz2')):
        obj = mesh_from_stl(args[0])
    else:
        module_name, function_name = args[0].rsplit('.', 1)
        orig_sys_path = list(sys.path)
        try:
            sys.path.append('.')
            module = __import__(module_name, fromlist=[function_name])
            sys.path = orig_sys_path
        except ImportError:
            sys.path = orig_sys_path
            raise

        obj = getattr(module, function_name)

    if options.io_file is None:
        view(obj, size)
    else:
        geometry = build(obj)
        viewer = EventViewer(geometry, options.io_file, size=size)
        viewer.start()