server_test.py

# Copyright 2019, The Jelly Bean World Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may not
# use this file except in compliance with the License. You may obtain a copy of
# the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations under
# the License.

import jbw
from timeit import default_timer
from simulator_test import SimpleAgent, make_config
from threading import Lock, Condition

cv = Condition(Lock())
waiting = False

def on_step():
	global waiting
	cv.acquire()
	waiting = False
	cv.notify()
	cv.release()

# create new simulator server
config = make_config()
sim = jbw.Simulator(sim_config=config, is_server=True, on_step_callback=on_step, default_client_permissions=jbw.GRANT_ALL_PERMISSIONS)

# add agents to simulator
agents = []
while len(agents) < 1:
	print("adding agent " + str(len(agents)))
	try:
		agent = SimpleAgent(sim)
		agents.append(agent)
	except jbw.AddAgentError:
		pass

	# move agents to avoid collision at (0,0)
	waiting = True
	for agent in agents:
		agent.do_next_action()

# start main loop
start_time = default_timer()
elapsed = 0.0
sim_start_time = sim.time()
while True:
	if not waiting:
		waiting = True
		for agent in agents:
			agent.do_next_action()
	if default_timer() - start_time > 1.0:
		elapsed += default_timer() - start_time
		print(str((sim.time() - sim_start_time) / elapsed) + " simulation steps per second.")
		start_time = default_timer()

	# wait for simulation to advance or for 1 second
	cv.acquire()
	if waiting:
		cv.wait(timeout=1.0)
	cv.release()