irie 0.0.57__py3-none-any.whl → 0.0.59__py3-none-any.whl

irie/apps/evaluation/identification.py

@@ -21,14 +21,9 @@
                 bb[event] += 1/(len(aa)-1)
 
 """
-import os
-import json
-from collections import defaultdict
 
 import numpy as np
-from irie.apps.events.models import EventRecord
-from irie.apps.site.view_utils import raise404
-from irie.apps.inventory.models import Asset
+from irie.apps.evaluation.models import Evaluation
 
 def mkdd(asset, pid, key):
     evaluations = list(reversed(sorted(Evaluation.objects.filter(asset=asset),

irie/apps/inventory/migrations/0009_datum_angle_x_datum_angle_y.py

@@ -0,0 +1,23 @@
+# Generated by Django 5.1.2 on 2025-07-31 05:48
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('irie_apps_inventory', '0008_alter_sensor_dx_alter_sensor_dy_alter_sensor_dz_and_more'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='datum',
+            name='angle_x',
+            field=models.DecimalField(blank=True, decimal_places=2, default=0.0, max_digits=10, null=True),
+        ),
+        migrations.AddField(
+            model_name='datum',
+            name='angle_y',
+            field=models.DecimalField(blank=True, decimal_places=2, default=0.0, max_digits=10, null=True),
+        ),
+    ]

irie/apps/inventory/models.py

@@ -111,10 +111,54 @@ class Datum(models.Model):
     locate_y = models.CharField(max_length=240) # eg, center of deck
     orient_z = models.CharField(max_length=240) # eg, vertical upwards
     locate_z = models.CharField(max_length=240) # eg, deck surface
+
+    angle_x = models.DecimalField(decimal_places=2, max_digits=10, default=0.0, null=True, blank=True)
+    angle_y = models.DecimalField(decimal_places=2, max_digits=10, default=0.0, null=True, blank=True)
     asset    = models.ForeignKey(Asset, on_delete=models.RESTRICT)
 
     def __str__(self):
         return f"{self.name}"
+    
+    def to_cardinal(self):
+        """
+        Create rotation matrix from datum basis to cardinal basis.
+        The cardinal basis is defined as:
+        cx = North, 
+        cy = East, 
+        cz = Up
+
+        The datum basis is defined as:
+        ex = Exp(orient_x cz) cx
+        ey = Exp(orient_y cz) cy
+        ez = Exp(orient_z cz) cz
+        where orient_x, orient_y, and orient_z are the angles in radians and
+        Exp is the exponential map which encodes the Rodrigues' rotation formula.
+
+        We want to return the rotation Rec  such that:
+            Rec @ [ex, ey, ez] = [cx, cy, cz]
+        """
+        import numpy as np
+        from shps.rotor import exp
+
+        orient_x = self.angle_x
+        orient_y = self.angle_y
+
+        ex, ey, ez = np.eye(3)
+        cz = ez
+        cx = exp(-orient_x*cz) @ ex
+        cy = exp(-orient_y*cz) @ ey
+
+        rotation = np.column_stack((cx, cy, cz))
+
+        return rotation
+    
+    def to_other(self, other: "Datum"):
+        """
+        Create rotation matrix from this datum to another datum.
+        """
+        R_cs = self.to_cardinal()
+        R_cm = other.to_cardinal()
+        return R_cm.T @ R_cs
 
 class SensorGroup(models.Model):
     name    = models.CharField(max_length=100)
@@ -123,6 +167,7 @@ class SensorGroup(models.Model):
     datum   = models.ForeignKey(Datum, on_delete=models.RESTRICT)
 #   network = models.CharField(max_length=100)
 #   events  = None
+
     def __str__(self):
         return f"{self.asset.calid} - {self.name} ({self.datum})"
 
@@ -145,7 +190,7 @@ class Sensor(models.Model):
                                on_delete=models.RESTRICT)
 
     def __str__(self):
-        return f"{self.group.asset.calid} - {self.name} ({self.group.name})"
+        return f"{self.group.asset.cesmd} / {self.name}"
 
 
     def acceleration(self, event):