closed_loop.py

#!/usr/bin/env python
 
 
"""
script test to simulate a closed loop
 
Usage:
  closed_loop.py <parameters_filename> [options]
 
with 'parameters_filename' the path to the parameters file
 
Options:
  -h --help          Show this help message and exit
  --brahma           Distribute data with brahma
  --bench            For a timed call
  -i --interactive   keep the script interactive
  -d --devices devices Specify the devices
  -n --niter niter   Number of iterations
  -g --generic       Use generic controller
  -f --fast          Compute PSF only during monitoring
"""
from shesha.config import ParamConfig
from docopt import docopt
 
if __name__ == "__main__":
    arguments = docopt(__doc__)
 
    param_file = arguments["<parameters_filename>"]
    compute_tar_psf = not arguments["--fast"]
 
    config = ParamConfig(param_file)
 
    # Get parameters from file
    if arguments["--bench"]:
        from shesha.supervisor.benchSupervisor import BenchSupervisor as Supervisor
    elif arguments["--brahma"]:
        from shesha.supervisor.canapassSupervisor import CanapassSupervisor as Supervisor
    else:
        from shesha.supervisor.compassSupervisor import CompassSupervisor as Supervisor
 
    if arguments["--devices"]:
        config.p_loop.set_devices([
                int(device) for device in arguments["--devices"].split(",")
        ])
 
    if arguments["--generic"]:
        config.p_controllers[0].set_type("generic")
        print("Using GENERIC controller...")
 
    if arguments["--niter"]:
        config.p_loop.set_niter(int(arguments["--niter"]))
 
    supervisor = Supervisor(config)
 
    supervisor.loop(supervisor.config.p_loop.niter, compute_tar_psf=compute_tar_psf)
 
    if arguments["--interactive"]:
        from shesha.util.ipython_embed import embed
        from os.path import basename
        embed(basename(__file__), locals())