Compressed Alignment (#59)

* getting direct widths for compressed segments * remove console
Kitware · Feb 23, 2024 · 14a54e1 · 14a54e1
1 parent 7a6fb01
commit 14a54e1
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 12 deletions.
diff --git a/bats_ai/core/models/spectrogram.py b/bats_ai/core/models/spectrogram.py
@@ -238,12 +238,14 @@ def compressed(self):
             starts_ = []
             stops_ = []
             domain = img.shape[1]
+            widths = []
             for start, stop in ranges:
                 segment = img[:, start:stop]
                 segments.append(segment)
 
                 starts_.append(int(round(self.duration * (start / domain))))
                 stops_.append(int(round(self.duration * (stop / domain))))
+                widths.append(stop - start)
 
                 # buffer = np.zeros((len(img), 20, 3), dtype=img.dtype)
                 # segments.append(buffer)
@@ -271,7 +273,7 @@ def compressed(self):
         buf.seek(0)
         img_base64 = base64.b64encode(buf.getvalue()).decode('utf-8')
 
-        return img_base64, starts_, stops_
+        return img_base64, starts_, stops_, widths
 
     @property
     def image_np(self):

diff --git a/bats_ai/core/views/recording.py b/bats_ai/core/views/recording.py
@@ -268,7 +268,7 @@ def get_spectrogram_compressed(request: HttpRequest, id: int):
         return {'error': 'Recording not found'}
 
     spectrogram = recording.spectrogram
-    compressed, starts, ends = spectrogram.compressed
+    compressed, starts, ends, widths = spectrogram.compressed
 
     spectro_data = {
         'base64_spectrogram': compressed,
@@ -281,6 +281,7 @@ def get_spectrogram_compressed(request: HttpRequest, id: int):
             'end_times': ends,
             'low_freq': spectrogram.frequency_min,
             'high_freq': spectrogram.frequency_max,
+            'widths': widths,
         },
     }
 

diff --git a/client/src/components/geoJS/geoJSUtils.ts b/client/src/components/geoJS/geoJSUtils.ts
@@ -178,6 +178,7 @@ export interface SpectroInfo {
   end_time: number;
   start_times?: number[];
   end_times?: number[];
+  widths?: number[], //widths of segements
   low_freq: number;
   high_freq: number;
 }

diff --git a/client/src/components/geoJS/layers/legendLayer.ts b/client/src/components/geoJS/layers/legendLayer.ts
@@ -144,14 +144,15 @@ export default class LegendLayer {
     const time = this.spectroInfo.end_time - this.spectroInfo.start_time;
     const timeToPixels = this.spectroInfo.width / time;
 
-    const { start_times, end_times } = this.spectroInfo;
-    if (start_times && end_times) {
+    const { start_times, end_times, widths } = this.spectroInfo;
+    if (start_times && end_times && widths) {
       // We need a pixel time to map to the 0 position
       let pixelOffset = 0;
       for (let i = 0; i < start_times.length; i += 1) {
         const length = yBuffer * 4;
         const start_time = start_times[i];
         const end_time = end_times[i];
+        const width = widths[i];
         this.lineDataX.push({
           line: {
             type: "LineString",
@@ -167,11 +168,11 @@ export default class LegendLayer {
             type: "LineString",
             coordinates: [
               [
-                (end_time - start_time) * timeToPixels + pixelOffset,
+                width + pixelOffset,
                 baseYPos + yBuffer,
               ],
               [
-                (end_time - start_time) * timeToPixels + pixelOffset,
+                width + pixelOffset,
                 baseYPos + topBuffer,
               ],
             ],
@@ -182,8 +183,8 @@ export default class LegendLayer {
           line: {
             type: "LineString",
             coordinates: [
-              [(end_time - start_time) * timeToPixels + pixelOffset, baseTopPos],
-              [(end_time - start_time) * timeToPixels + pixelOffset, baseTopPos - topBuffer],
+              [width + pixelOffset, baseTopPos],
+              [width + pixelOffset, baseTopPos - topBuffer],
             ],
           },
           thicker: true,
@@ -193,10 +194,10 @@ export default class LegendLayer {
             type: "LineString",
             coordinates: [
               [
-                (end_time - start_time) * timeToPixels + pixelOffset,
+                width + pixelOffset,
                 baseYPos + yBuffer,
               ],
-              [(end_time - start_time) * timeToPixels + pixelOffset, baseTopPos],
+              [width + pixelOffset, baseTopPos],
             ],
           },
           grid: true,
@@ -212,12 +213,12 @@ export default class LegendLayer {
         });
         this.textDataX.push({
           text: `${end_time}ms`,
-          x: (end_time - start_time) * timeToPixels + pixelOffset,
+          x: width + pixelOffset,
           y: baseTopPos,
           offsetX: 3,
           offsetY: baseTopPos === 0 ? -16 : 16,
         });
-        pixelOffset += (end_time - start_time) * timeToPixels;
+        pixelOffset += width;
         // Need to add the current
       }
     }