Fixed an bug in head angle

Author: yeshaokai

amadeusgpt/analysis_objects/relationship.py

@@ -71,6 +71,7 @@ def calc_angle_in_egocentric_animal(mouse_cs_inv, p):
         p_in_mouse[:, 1], p_in_mouse[:, 0]
     )  # relative angle between the object and the mouse body axis
     theta = np.rad2deg(theta % (2 * np.pi))
+ 
     return theta
 
 
@@ -91,6 +92,7 @@ def get_name(self):
         return self.__name__
 
     def query_relationship(self, query_name: str) -> ndarray:
+
         ret = self.data[query_name]
         return ret
 
@@ -239,7 +241,7 @@ def _animal_animal_relationship(
             mouse_cs_inv[valid] = np.linalg.inv(mouse_cs[valid])
             head_cs_inv.append(mouse_cs_inv)
             head_angles = calc_angle_in_egocentric_animal(
-                head_cs_inv, sender_animal.get_center()
+                head_cs_inv, receiver_animal.get_center()
             )
 
         relative_velocity = (
@@ -261,7 +263,7 @@ def _animal_animal_relationship(
 
         if head_angles is not None:
             ret["relative_head_angle"] = head_angles
-            # print ('head angle', head_angles)
+
         if angles is not None:
             ret["relative_angle"] = angles
         if orientation is not None:

amadeusgpt/managers/event_manager.py

@@ -320,6 +320,7 @@ def get_animals_animals_events(
                 _comparison = comparison_operator + "".join(
                     query.split(comparison_operator)[1:]
                 )
+               
                 events = self.get_events_from_relationship(
                     relationship, _query, _comparison, smooth_window_size
                 )

amadeusgpt/programs/sandbox.py

@@ -543,10 +543,9 @@ def render_temp_message(query, sandbox):
     from amadeusgpt.analysis_objects.object import ROIObject
     from amadeusgpt.main import create_amadeus
 
-    config = Config("amadeusgpt/configs/Horse_template.yaml")
+    config = Config("amadeusgpt/configs/mabe_template.yaml")
+
     amadeus = create_amadeus(config)
     sandbox = amadeus.sandbox
-    analysis = sandbox.exec_namespace["behavior_analysis"]
-    analysis.add_roi_object = ""
 
-    res = sandbox.step("plot the trajectory of the bodypart body_mouth")
+    render_temp_message('random', sandbox)