curioustorvald/tsvm
1
0
Fork 0
mirror of https://github.com/curioustorvald/tsvm.git synced 2026-03-07 11:51:49 +09:00
Code Issues Packages Projects Releases Wiki Activity

tav fix: webm being recognised as still image

Browse Source
This commit is contained in:
minjaesong
2026-01-21 21:21:11 +09:00
parent b10d5d3a34
commit 9310885260
2 changed files with 15 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
video_encoder/src/decoder_tav.c
View File

@@ -1679,7 +1679,6 @@ static int collect_audio_packets(decoder_context_t *ctx) {
// Scan through file // Scan through file
while (1) { while (1) {
long packet_pos = ftell(ctx->input_fp);
uint8_t packet_type; uint8_t packet_type;
if (fread(&packet_type, 1, 1, ctx->input_fp) != 1) break; if (fread(&packet_type, 1, 1, ctx->input_fp) != 1) break;

26
video_encoder/src/encoder_tav.c
View File

@@ -133,10 +133,10 @@ typedef struct frame_analysis {
double highband_ratio; // highband_energy / total_energy double highband_ratio; // highband_energy / total_energy
// Per-band entropies (Shannon entropy of coefficient magnitudes) // Per-band entropies (Shannon entropy of coefficient magnitudes)
double entropy_ll; // double entropy_ll;
double entropy_lh[ANALYSIS_DWT_LEVELS]; // double entropy_lh[ANALYSIS_DWT_LEVELS];
double entropy_hl[ANALYSIS_DWT_LEVELS]; // double entropy_hl[ANALYSIS_DWT_LEVELS];
double entropy_hh[ANALYSIS_DWT_LEVELS]; // double entropy_hh[ANALYSIS_DWT_LEVELS];
// Texture change indicators // Texture change indicators
double zero_crossing_rate; // Zero crossing rate in highbands double zero_crossing_rate; // Zero crossing rate in highbands
@@ -490,6 +490,10 @@ static int is_input_still_image(const char *input_file) {
return 1; // Known image extension return 1; // Known image extension
} }
} }
if (strcmp(ext, ".webm") == 0 || strcmp(ext, ".WEBM") == 0) {
return 0; // Known video extension
}
} }
// Use ffprobe to check if it's a single-frame input // Use ffprobe to check if it's a single-frame input
@@ -1351,7 +1355,7 @@ static float* resize_frame_to_analysis(const uint8_t *rgb_frame, int src_width,
} }
// Calculate Shannon entropy of coefficient magnitudes // Calculate Shannon entropy of coefficient magnitudes
static double calculate_shannon_entropy(const float *coeffs, int count) { /*static double calculate_shannon_entropy(const float *coeffs, int count) {
if (count == 0) return 0.0; if (count == 0) return 0.0;
// Build histogram of coefficient magnitudes (use 256 bins) // Build histogram of coefficient magnitudes (use 256 bins)
@@ -1389,10 +1393,10 @@ static double calculate_shannon_entropy(const float *coeffs, int count) {
return entropy; return entropy;
#undef HIST_BINS #undef HIST_BINS
} }*/
// Extract subband from DWT coefficients (helper for entropy calculation) // Extract subband from DWT coefficients (helper for entropy calculation)
static void extract_subband(const float *dwt_data, int width, int height, int level, /*static void extract_subband(const float *dwt_data, int width, int height, int level,
int band, float *output, int *out_count) { int band, float *output, int *out_count) {
// band: 0=LL, 1=LH, 2=HL, 3=HH // band: 0=LL, 1=LH, 2=HL, 3=HH
// For level L, subbands are in top-left quadrant of size (width>>L, height>>L) // For level L, subbands are in top-left quadrant of size (width>>L, height>>L)
@@ -1429,7 +1433,7 @@ static void extract_subband(const float *dwt_data, int width, int height, int le
} }
*out_count = count; *out_count = count;
} }*/
// Compute comprehensive frame analysis metrics // Compute comprehensive frame analysis metrics
static void compute_frame_metrics(const float *dwt_current, const float *dwt_previous, static void compute_frame_metrics(const float *dwt_current, const float *dwt_previous,
@@ -1492,7 +1496,7 @@ static void compute_frame_metrics(const float *dwt_current, const float *dwt_pre
metrics->highband_ratio = total_energy > 0 ? (highband_energy / total_energy) : 0; metrics->highband_ratio = total_energy > 0 ? (highband_energy / total_energy) : 0;
// Metric 4: Per-band entropies // Metric 4: Per-band entropies
float *subband_buffer = malloc(num_pixels * sizeof(float)); /*float *subband_buffer = malloc(num_pixels * sizeof(float));
int subband_count; int subband_count;
// LL band entropy // LL band entropy
@@ -1512,7 +1516,7 @@ static void compute_frame_metrics(const float *dwt_current, const float *dwt_pre
// HH band // HH band
extract_subband(dwt_current, width, height, level, 3, subband_buffer, &subband_count); extract_subband(dwt_current, width, height, level, 3, subband_buffer, &subband_count);
metrics->entropy_hh[level] = calculate_shannon_entropy(subband_buffer, subband_count); metrics->entropy_hh[level] = calculate_shannon_entropy(subband_buffer, subband_count);
} }*/
// Metric 5: Zero crossing rate in highbands (texture change indicator) // Metric 5: Zero crossing rate in highbands (texture change indicator)
int zero_crossings = 0; int zero_crossings = 0;
@@ -1527,7 +1531,7 @@ static void compute_frame_metrics(const float *dwt_current, const float *dwt_pre
metrics->zero_crossing_rate = (double)zero_crossings / highband_coeffs; metrics->zero_crossing_rate = (double)zero_crossings / highband_coeffs;
} }
free(subband_buffer); //free(subband_buffer);
} }
// Hybrid scene change detector with adaptive thresholds // Hybrid scene change detector with adaptive thresholds