Skip to content

Acu agent updates #6

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
ssarkarbht wants to merge 3 commits into
base: main
Choose a base branch
from
Open

Acu agent updates #6

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
ce044f5
Test scripts, minor bug fixes
ssarkarbht Jul 24, 2025
4ba3442
Added TCS broadcasting switch
ssarkarbht Jul 24, 2025
eee5414
Updated module imports
ssarkarbht Jul 24, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 13 additions & 0 deletions pcs/agents/acu_interface/aculib.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -146,6 +146,19 @@ def abort(self):
self.log.info(response)
return response

def position_broadcast(self, udp_host: str, udp_port: int):
"""Switch ON the position broadcasting from the TCS
:param udp_host
:param udp_port
"""
data = {"destination_host": udp_host,
"destination_port": udp_port}
cmd = f"{self.url_prefix}/acu/position-broadcast"

print ("COMING HERE?")
response = self.post(cmd, data)
self.log.info(response.json())
return response

def move_to(self, azimuth: float, elevation: float):
"""send telescope to a given azimuth,elevation
Expand Down
49 changes: 45 additions & 4 deletions pcs/agents/acu_interface/agent.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -88,17 +88,22 @@ def __init__(self, agent, config, device='acu_sim', startup=False):
startup=False)

#register tasks
agent.register_task('tcs_broadcast',
self.tcs_broadcast,
blocking=True,
aborter=self._simple_task_abort)

agent.register_task('go_to',
self.go_to,
blocking=True,
aborter=self._simple_task_abort)
agent.register_task('az_scan',
self.az_scan,
blocking=False,
blocking=True,
aborter=self._simple_task_abort)
agent.register_task('fromfile_scan',
self.fromfile_scan,
blocking=False,
blocking=True,
aborter=self._simple_task_abort)


Expand Down Expand Up @@ -254,6 +259,39 @@ def datagramReceived(self, data, src_addr):
#self.agent.feeds['acu_udp_stream'].flush_buffer()
return True, 'Acquisition exited cleanly.'

@ocs_agent.param('udp_host', type=str)
@ocs_agent.param('udp_port', type=int)
def tcs_broadcast(self, session, params):
"""tcs_broadcast(udp_host, udp_port)

**Task** - Switch ON the position data stream broadcasting from
the TCS api. With the given UDP host address and port, the ACU
starts broadcasting the posotion data stream that the interface
agent can listen to and write the stream in the PCS HK.

Parameters:
udp_host (str): UDP host address
udp_port (int): udp port number
"""
udp_host = params['udp_host']
udp_port = params['udp_port']

self.log.info(f'Requested TCS broadcasting to switch ON at UDP host: {udp_host}, port: {udp_port}')
certs = self.acu_conf['certs']

tcs = aculib.observatory_control_system(
self.acu_conf['base_url'],
self.log,
server_cert=certs['server_cert'],
client_cert=certs['client_cert'],
client_key=certs['client_key'],
verify_cert=certs['verify']
)
msg = tcs.position_broadcast(udp_host, udp_port)
self.log.info(f"UDP stream broadcasting executed with response code: {msg.status_code}")

return True, msg.text

@ocs_agent.param('az', type=float)
@ocs_agent.param('el', type=float)
def go_to(self, session, params):
Expand Down Expand Up @@ -298,7 +336,7 @@ def go_to(self, session, params):
return True, msg.text

@ocs_agent.param('scan_params', type=dict)
def az_scan():
def az_scan(self, session, params):
"""az_scan(start_time, turnaround_time, elevation,
speed, num_scans, azimuth_range)

Expand Down Expand Up @@ -343,7 +381,7 @@ def az_scan():


@ocs_agent.param('scan_filename', type=str)
def fromfile_scan():
def fromfile_scan(self, session, params):
"""fromfile_scan(scan_filename)

**Task** - Send scan commands for a predefined arbitrary path which
Expand Down Expand Up @@ -413,3 +451,6 @@ def main(args=None):

if __name__=='__main__':
main()



41 changes: 41 additions & 0 deletions tests/acu-agent/test_azscan.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
#!/bin/python

from ocs.ocs_client import OCSClient
import time
from datetime import datetime

client = OCSClient('acu1')
print (client.tcs_broadcast(udp_host='172.17.0.1', udp_port=5601))
time.sleep(5)

#'''
print (client.broadcast.start())

print('\nMonitoring (ctrl-c to stop and exit)...\n')

#aximuth scan parameters
num_scans = 20
scan_param = {'start_time': float(datetime.utcnow().timestamp()),
'turnaround_time': 1.0,
'elevation': 20.0,
'speed': 1.0,
'num_scans': num_scans,
'azimuth_range': [10.0, 110.0]}

#execute the scan
print (f"Moving telescope for Azimuth Scan ...")
client.az_scan(scan_params=scan_param)

stay_alive = 3900 #seconds
it = 0
while it*10 < stay_alive:
print (client.az_scan.status())
time.sleep(10)
it += 1


print('Stopping data acquisition ...')
print (client.broadcast.stop())
#'''


43 changes: 43 additions & 0 deletions tests/acu-agent/test_move+acq.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,43 @@
#!/bin/python
'''
This script runs the test to command telescope to move to
a preset point. In addition it also turns on the udp stream capture
and writing into g3 files while the movement operation in place.

Test runs for 5 minute and exists by stopping the broadcast.
'''

from ocs.ocs_client import OCSClient
import time

client = OCSClient('acu1')

print (client.tcs_broadcast(udp_host='172.17.0.1', udp_port=5601))
time.sleep(5)

print (client.broadcast.start())

print('\nMonitoring (ctrl-c to stop and exit)...\n')

az_pos = 100.0
el_pos = 80.0

mv_cmd = False

try:
for i in range(50):
time.sleep(6)
print (client.broadcast.status())
r = client.broadcast.status()
print (r.session.get('data'))
if not(mv_cmd):
print (f"Moving telescope to Az: {az_pos}, El: {el_pos}")
client.go_to(az=az_pos, el=el_pos)
mv_cmd = True
print (client.go_to.status())
except KeyboardInterrupt:
print('Exiting on ctrl-c...')
print ()

print('Stopping data acquisition ...')
print (client.broadcast.stop())