manager.py

import zmq
import random
import sys
import time

from ops.pyTLS import TLSaccess

# TLSHost = "lassi.ad.nrao.edu"
TLS_HOST = "galileo.gb.nrao.edu"

def runScan(config):

    a = TLSaccess(TLS_HOST)

    # wait till it's ready
    status = a.get_status()
    print ("scanner status:", status)
    while status.state != "Ready":
        time.sleep(1)
        status = a.get_status()

    # configure

    proj = config["proj"] #"test"
    res = "63mm@100m"
    sensitivity = "Normal"
    scan_mode = "Speed"
    cntr_az = 270
    cntr_el = 45
    az_fov = 35
    el_fov = 35
    a.configure_scanner(proj, res, sensitivity, scan_mode, cntr_az, cntr_el, az_fov, el_fov)

    simFile = "/home/scratch/pmargani/LASSI/scannerData/tmp/test.ptx"
    print ("setting sim file to", simFile)
    a.set_simulated_scan_data_file(simFile)

    # get the device scan started,
    a.start_scan()

    # now wait until it's NOT in ready
    while status.state == "Ready":
        print("waiting for scan to start")
        time.sleep(1)
        status = a.get_status()    

    # dissconnect! Because it only supports one control connection
    a.cntrl_exit()
    del a

    # then once it's started, get the processing going:
    # connect to the process server
    port = "5557"
    context = zmq.Context()
    socket = context.socket(zmq.REQ)
    socket.connect("tcp://localhost:%s" % port)

    # # see what state the server is in
    socket.send_string("get_state")
    msg = socket.recv()
    print ("process server state: ", msg)
    time.sleep(1)

    socket.send_string("set:proj=cat")
    msg = socket.recv()
    print ("server response: ", msg)
    time.sleep(1)

    # set the meta-data
    for k, v in config.items():
        cmd = "set:%s=%s" % (k, v)
        print(cmd)
        socket.send_string(cmd)
        msg = socket.recv()
        print ("server set response: ", msg)
        time.sleep(1)

    # start processing
    socket.send_string("process")
    msg = socket.recv()
    msgStr = "".join( chr(x) for x in msg)
    print ("process server response: ", msg)

    time.sleep(3)

    sec = 0
    while msgStr != "Ready":
    # while True:
        try:
            # status = a.get_status()
            # print ("scanner state: ", status.state)
            # see what state the server is in
            socket.send_string("get_state")
            msg = socket.recv()
            msgStr = "".join( chr(x) for x in msg)
            print ("process server state: ", msg)
            time.sleep(1)
            sec += 1
            print(sec)
        except KeyboardInterrupt:
            # a.cntrl_exit()
            break

    print ("done processing Scan")

def runFakeScan(config):

    port = "5557"
    context = zmq.Context()
    socket = context.socket(zmq.PAIR)
    socket.connect("tcp://localhost:%s" % port)

    # # see what state the server is in
    socket.send_string("get_state")
    msg = socket.recv()
    print ("process server state: ", msg)
    time.sleep(1)

    
    # set the meta-data
    for k, v in config.items():
        cmd = "set:%s=%s" % (k, v)
        print(cmd)
        socket.send_string(cmd)
        msg = socket.recv()
        print ("server set response: ", msg)
        time.sleep(1)

    # start processing
    socket.send_string("process")
    msg = socket.recv()
    msgStr = "".join( chr(x) for x in msg)
    print ("process server response: ", msg)

    time.sleep(3)

    sec = 0
    while msgStr != "Ready":
    # while True:
        try:
            # status = a.get_status()
            # print ("scanner state: ", status.state)
            # see what state the server is in
            socket.send_string("get_state")
            msg = socket.recv()
            msgStr = "".join( chr(x) for x in msg)
            print ("process server state: ", msg)
            time.sleep(1)
            sec += 1
            print(sec)
        except KeyboardInterrupt:
            # a.cntrl_exit()
            break

    print ("done processing Scan")

def main():
    c = {
        "proj": "TEST",
        "scanNum": 1,
        "refScan": "True",
        "refScanNum": 0,
        "filename": "1"
    }
    runScan(c)
    time.sleep(3)
    u = {
        "scanNum": 2,
        "refScan": "False",
        "refScanNum": 1,
        "filename": "2"
    }
    c.update(u)
    runScan(c)

if __name__ == '__main__':
    main()