// Like lab-eval.mjs, but also feeds enrolled voiceprints into clusterWindows.
// Usage: node lab-eval2.mjs <emb.json> [expect] [numSpeakers] [markers.json] [vp.json]
import fs from 'node:fs';
import { clusterWindows } from './diar-v2.js';
const [file, expect, numSp, markerFile, vpFile] = process.argv.slice(2);
const d = JSON.parse(fs.readFileSync(file, 'utf8'));
const ok = d.wins.filter(w => w.v).map(w => ({ ti: w.ti, start: w.s, end: w.e, vec: w.v }));
const markers = markerFile && markerFile !== '-' ? JSON.parse(fs.readFileSync(markerFile, 'utf8')) : [];
const vps = vpFile ? JSON.parse(fs.readFileSync(vpFile, 'utf8')) : [];
const segs = clusterWindows(d.turns, ok, +(numSp || 0), markers, vps);
const talk = new Map(), first = new Map(), last = new Map();
for (const s of segs) {
  talk.set(s.speaker, (talk.get(s.speaker) || 0) + (s.end - s.start));
  if (!first.has(s.speaker)) first.set(s.speaker, s.start);
  last.set(s.speaker, s.end);
}
const order = [...talk.entries()].sort((a, b) => b[1] - a[1]);
console.log(`RESULT speakers=${order.length}${expect ? ` (expect ${expect})` : ''} segs=${segs.length}`);
console.log('   ' + order.map(([sp, t]) => `${sp.replace('speaker_','')}:${t.toFixed(0)}s[${(first.get(sp)/60).toFixed(1)}-${(last.get(sp)/60).toFixed(1)}]`).join(' '));