--record as option for random agents

example/high_action_random_agent.py

@@ -23,7 +23,9 @@ def main():
                       help="Python randomization seed; uses python default if 0 or not given")
   parser.add_argument('--no-reorient', action='store_true',
                       help="Do not use the new Reorient action")
-  parser.add_argument('--record-dir', type=str, default='log/',
+  parser.add_argument('--record', action='store_true',
+                      help="Doing HFO --record")
+  parser.add_argument('--rdir', type=str, default='log/',
                       help="Set directory to use if doing HFO --record")
   args=parser.parse_args()
   if args.seed:
@@ -32,10 +34,15 @@ def main():
   hfo_env = hfo.HFOEnvironment()
   # Connect to the server with the specified
   # feature set. See feature sets in hfo.py/hfo.hpp.
-  hfo_env.connectToServer(hfo.LOW_LEVEL_FEATURE_SET,
-                          'bin/teams/base/config/formations-dt', args.port,
-                          'localhost', 'base_left', False,
-                          record_dir=args.record_dir)
+  if args.record:
+    hfo_env.connectToServer(hfo.LOW_LEVEL_FEATURE_SET,
+                            'bin/teams/base/config/formations-dt', args.port,
+                            'localhost', 'base_left', False,
+                            record_dir=args.rdir)
+  else:
+    hfo_env.connectToServer(hfo.LOW_LEVEL_FEATURE_SET,
+                            'bin/teams/base/config/formations-dt', args.port,
+                            'localhost', 'base_left', False)
   if args.seed:
     print("Python randomization seed: {0:d}".format(args.seed))
   for episode in itertools.count():

example/high_level_random_agent.py

@@ -21,7 +21,9 @@ def main():
                       help="Server port")
   parser.add_argument('--seed', type=int, default=None,
                       help="Python randomization seed; uses python default if 0 or not given")
-  parser.add_argument('--record-dir', type=str, default='log/',
+  parser.add_argument('--record', action='store_true',
+                      help="Doing HFO --record")
+  parser.add_argument('--rdir', type=str, default='log/',
                       help="Set directory to use if doing HFO --record")
   args=parser.parse_args()
   if args.seed:
@@ -30,10 +32,15 @@ def main():
   hfo_env = hfo.HFOEnvironment()
   # Connect to the server with the specified
   # feature set. See feature sets in hfo.py/hfo.hpp.
-  hfo_env.connectToServer(hfo.HIGH_LEVEL_FEATURE_SET,
-                          'bin/teams/base/config/formations-dt', args.port,
-                          'localhost', 'base_left', False,
-                          record_dir=args.record_dir)
+  if args.record:
+    hfo_env.connectToServer(hfo.HIGH_LEVEL_FEATURE_SET,
+                            'bin/teams/base/config/formations-dt', args.port,
+                            'localhost', 'base_left', False,
+                            record_dir=args.rdir)
+  else:
+    hfo_env.connectToServer(hfo.HIGH_LEVEL_FEATURE_SET,
+                            'bin/teams/base/config/formations-dt', args.port,
+                            'localhost', 'base_left', False)
   if args.seed:
     print("Python randomization seed: {0:d}".format(args.seed))
   for episode in itertools.count():