small refactoring

.gitignore

@@ -5,4 +5,6 @@ test.ipynb
 */**/params.yaml
 /**/__pycache__
 /data/**
-!/data/.gitkeep
+!/data/.gitkeep
+/plots/**
+!/plots/.gitkeep

create_plots.py

@@ -5,19 +5,14 @@
 import numpy as np
 import argparse
 import matplotlib.pyplot as plt
-from matplotlib.path import Path
-import matplotlib.patches as patches
 import json
 import yaml
 
 from utils import Utils
 
 """
     TODO: 
-    - Create plot titles
-    - Check if all legends are added
     - Add collisions to path map
-    - Add support for declaration file to declare which data should be plotted and what plots should be used
 """
 
 

data_recorder_node.py → scripts/data_recorder_node.py

@@ -13,6 +13,7 @@
 from rosgraph_msgs.msg import Clock
 from rospy.core import traceback
 import rostopic
+import rospkg
 import subprocess
 from datetime import datetime
 import rosparam
@@ -107,7 +108,7 @@ class Recorder:
     def __init__(self):
         self.model = rospy.get_param(os.path.join(rospy.get_namespace(), "model"), "")
 
-        self.dir = os.path.dirname(os.path.abspath(__file__))
+        self.dir = rospkg.RosPack().get_path("arena-evaluation")
         self.result_dir = os.path.join(self.dir, "data", datetime.now().strftime("%d-%m-%Y_%H-%M-%S")) + "_" + rospy.get_namespace().replace("/", "")
 
         try:
@@ -232,7 +233,7 @@ def write_params(self):
 
 
 if __name__=="__main__":
-    rospy.init_node("data_recorder") 
+    rospy.init_node("data_recorder", anonymous=True) 
 
     time.sleep(5)