reframe-video 0.1.1 → 0.1.3

package/dist/analyze.js

@@ -155,6 +155,43 @@ async function* frameStream(source, opts) {
   if (pending.length !== 0) throw new Error(`trailing partial frame (${pending.length} bytes)`);
 }
 
+// ../../benchmark/harness/motion/grid.ts
+var DEFAULT_GRID = { cols: 8, rows: 5 };
+var cellOf = (x, y, w, h, spec) => {
+  const c = Math.min(spec.cols - 1, Math.floor(x * spec.cols / w));
+  const r = Math.min(spec.rows - 1, Math.floor(y * spec.rows / h));
+  return r * spec.cols + c;
+};
+function cellDiff(prev, next, w, h, spec) {
+  const n = spec.cols * spec.rows;
+  const sum = new Array(n).fill(0);
+  const cnt = new Array(n).fill(0);
+  for (let y = 0; y < h; y++) {
+    for (let x = 0; x < w; x++) {
+      const idx = cellOf(x, y, w, h, spec);
+      sum[idx] += Math.abs(next[y * w + x] - prev[y * w + x]);
+      cnt[idx]++;
+    }
+  }
+  return sum.map((s, i) => cnt[i] ? s / cnt[i] : 0);
+}
+function frameOccupancy(frame, w, h, spec) {
+  const n = spec.cols * spec.rows;
+  const sum = new Array(n).fill(0);
+  const cnt = new Array(n).fill(0);
+  for (let y = 0; y < h; y++) {
+    for (let x = 0; x < w; x++) {
+      const here = frame[y * w + x];
+      const gx = x + 1 < w ? Math.abs(frame[y * w + x + 1] - here) : 0;
+      const gy = y + 1 < h ? Math.abs(frame[(y + 1) * w + x] - here) : 0;
+      const idx = cellOf(x, y, w, h, spec);
+      sum[idx] += gx + gy;
+      cnt[idx]++;
+    }
+  }
+  return sum.map((s, i) => cnt[i] ? s / cnt[i] : 0);
+}
+
 // ../../benchmark/harness/motion/analyze.ts
 var ANALYSIS_WIDTH = 480;
 var ANALYSIS_HEIGHT = 270;
@@ -243,6 +280,7 @@ async function analyzeMotion(inputPath, opts = {}) {
     height: ANALYSIS_HEIGHT,
     ...opts.changedThreshold !== void 0 && { changedThreshold: opts.changedThreshold }
   };
+  const gridSpec = opts.grid ? typeof opts.grid === "object" ? opts.grid : DEFAULT_GRID : null;
   const source = await resolveSource(resolve(inputPath));
   const keys = [
     "blockSpeedMean",
@@ -256,13 +294,17 @@ async function analyzeMotion(inputPath, opts = {}) {
     "activeBlocks"
   ];
   const series = Object.fromEntries([...keys.map((k) => [k, []]), ["t", []]]);
+  const gridDiff = [];
+  const gridOcc = [];
   let prev = null;
   let pair = 0;
   for await (const frame of frameStream(source, blockOpts)) {
+    if (gridSpec) gridOcc.push(frameOccupancy(frame, ANALYSIS_WIDTH, ANALYSIS_HEIGHT, gridSpec));
     if (prev) {
       const stats = analyzePair(prev, frame, blockOpts);
       for (const k of keys) series[k].push(stats[k]);
       series.t.push((pair + 0.5) / fps);
+      if (gridSpec) gridDiff.push(cellDiff(prev, frame, ANALYSIS_WIDTH, ANALYSIS_HEIGHT, gridSpec));
       pair++;
     }
     prev = frame;
@@ -310,7 +352,8 @@ async function analyzeMotion(inputPath, opts = {}) {
       saturatedPairs: series.saturatedFraction.map((s, i) => s > 0.3 ? i : -1).filter((i) => i >= 0),
       diffPeriodicityHz: dominantPeriodicity(d, fps)
     },
-    segments
+    segments,
+    ...gridSpec && { grid: { spec: gridSpec, diff: gridDiff, occupancy: gridOcc } }
   };
 }
 async function main() {