From 4e05c7c64054fdaa4513ae3fc5c384d5813a4a97 Mon Sep 17 00:00:00 2001
From: Antonio Andriella <aandriella@iri.upc.edu>
Date: Thu, 10 Sep 2020 13:44:24 +0200
Subject: [PATCH] Add function to dispach the result of the query
---
bn_functions.py | 27 ++++++++++++++++++++++++++-
1 file changed, 26 insertions(+), 1 deletion(-)
diff --git a/bn_functions.py b/bn_functions.py
index bd290e5..887def7 100644
--- a/bn_functions.py
+++ b/bn_functions.py
@@ -153,4 +153,29 @@ def get_stochastic_action(actions_distr_prob):
rnd_val = random.uniform(0, 1)
action_id = compute_distance(actions_distr_prob_scaled, rnd_val)
- return action_id
\ No newline at end of file
+ return action_id
+
+def interpret_user_output(action_id):
+ user_action = 0
+ user_react_time = 0
+
+ if action_id == 0:
+ user_action = 0; user_react_time = 0
+ elif action_id == 1:
+ user_action = 1; user_react_time = 0
+ elif action_id == 2:
+ user_action = 2; user_react_time = 0
+ elif action_id == 3:
+ user_action = 0; user_react_time = 1
+ elif action_id == 4:
+ user_action = 1; user_react_time = 1
+ elif action_id == 5:
+ user_action = 2; user_react_time = 1
+ elif action_id == 6:
+ user_action = 0; user_react_time = 2
+ elif action_id == 7:
+ user_action = 1; user_react_time = 2
+ elif action_id == 8:
+ user_action = 2; user_react_time = 2
+
+ return user_action, user_react_time
\ No newline at end of file
--
GitLab