thomas.rijpstra/ytdlp-segment-downloader
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: output top 10 segment candidates with timestamps and duration

Browse Source
This commit is contained in:
Kilo Code Cloud
2026-01-14 20:05:28 +00:00
parent 3c263a78c6
commit 9ea4d4ab33
3 changed files with 50 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

67
src/cli/downloader.ts
View File

@@ -6,6 +6,7 @@ export interface DownloadOptions {
url: string;
outputDir: string;
format: string;
topN: number;
}
interface RawHeatmapSegment {
@@ -142,9 +143,10 @@ function getIntensity(segment: RawHeatmapSegment): number {
return segment.intensity ?? segment.heat ?? segment.value ?? 0;
}
function findHighestIntegralJump(
segments: RawHeatmapSegment[]
): ProcessedSegment | null {
function getTopSegmentsByIntegral(
segments: RawHeatmapSegment[],
topN: number
): ProcessedSegment[] {
// Convert to processed format and filter valid segments
const validSegments = segments
.map(seg => ({
@@ -161,10 +163,10 @@ function findHighestIntegralJump(
);
if (validSegments.length === 0) {
return null;
return [];
}
// Calculate integral jump for each segment (intensity × duration)
// Calculate integral for each segment and sort by highest
const withIntegral = validSegments.map(seg => {
const segmentDuration = seg.end - seg.start;
const integralJump = seg.intensity * segmentDuration;
@@ -174,15 +176,13 @@ function findHighestIntegralJump(
};
});
// Find the segment with the highest integral jump (biggest bump in the integral)
// This is the segment that contributed most to the total integral
return withIntegral.reduce((max, current) => {
return current.integralJump > max.integralJump ? current : max;
});
// Sort by integral jump (highest first) and return top N
withIntegral.sort((a, b) => b.integralJump - a.integralJump);
return withIntegral.slice(0, topN);
}
export async function downloadMostWatchedSegment(options: DownloadOptions): Promise<void> {
const { url, outputDir, format } = options;
const { url, outputDir, format, topN } = options;
// Create output directory if it doesn't exist
if (!existsSync(outputDir)) {
@@ -208,38 +208,47 @@ export async function downloadMostWatchedSegment(options: DownloadOptions): Prom
}
console.log(`\nHeatmap data found: ${info.heatmap.length} segments`);
console.log(`\nTop ${topN} segments by integral jump:\n`);
// Find segment with highest integral jump (biggest bump)
const topSegment = findHighestIntegralJump(info.heatmap);
// Get top segments
const topSegments = getTopSegmentsByIntegral(info.heatmap, topN);
if (!topSegment) {
if (topSegments.length === 0) {
console.log("No valid segments found. Downloading full video...");
const outputPath = join(outputDir, `${safeTitle}.%(ext)s`);
await downloadSegment(url, outputPath, 0, info.duration, format);
return;
}
console.log(`\nSegment with highest integral jump:`);
console.log(` Time: ${formatTime(topSegment.start)} - ${formatTime(topSegment.end)}`);
console.log(` Duration: ${formatTime(topSegment.end - topSegment.start)}`);
console.log(` Intensity: ${(topSegment.intensity * 100).toFixed(1)}%`);
console.log(` Integral Jump: ${topSegment.integralJump.toFixed(4)}`);
// Output the top segments
for (let i = 0; i < topSegments.length; i++) {
const seg = topSegments[i];
const duration = seg.end - seg.start;
console.log(`${i + 1}. ${formatTime(seg.start)} - ${formatTime(seg.end)} | Duration: ${formatTime(duration)} | Integral: ${seg.integralJump.toFixed(4)}`);
}
// Download the segment
console.log("");
// Download the top segment
const topSegment = topSegments[0];
const outputPath = join(outputDir, `${safeTitle}_most_watched.%(ext)s`);
console.log(`\nDownloading segment...`);
console.log(`Downloading segment: ${formatTime(topSegment.start)} - ${formatTime(topSegment.end)}`);
await downloadSegment(url, outputPath, topSegment.start, topSegment.end, format);
// Save segment info
const segmentInfoPath = join(outputDir, `${safeTitle}_segment_info.txt`);
const segmentInfo = `# ${info.title}\n\n` +
`Segment with highest integral jump (biggest bump in heatmap):\n` +
` Start: ${formatTime(topSegment.start)} (${topSegment.start.toFixed(1)}s)\n` +
` End: ${formatTime(topSegment.end)} (${topSegment.end.toFixed(1)}s)\n` +
` Duration: ${formatTime(topSegment.end - topSegment.start)}\n` +
` Intensity: ${(topSegment.intensity * 100).toFixed(1)}%\n` +
` Integral Jump: ${topSegment.integralJump.toFixed(4)}\n`;
const segmentInfoPath = join(outputDir, `${safeTitle}_top_segments.txt`);
let segmentInfo = `# ${info.title}\n\n`;
segmentInfo += `Top ${topN} segments by integral jump:\n\n`;
for (let i = 0; i < topSegments.length; i++) {
const seg = topSegments[i];
const duration = seg.end - seg.start;
segmentInfo += `${i + 1}. ${formatTime(seg.start)} - ${formatTime(seg.end)}\n`;
segmentInfo += ` Duration: ${formatTime(duration)}\n`;
segmentInfo += ` Integral: ${seg.integralJump.toFixed(4)}\n`;
segmentInfo += ` Intensity: ${(seg.intensity * 100).toFixed(1)}%\n\n`;
}
writeFileSync(segmentInfoPath, segmentInfo);
console.log(`\nSegment info saved to: ${segmentInfoPath}`);