summaryrefslogtreecommitdiff
path: root/tools/specview.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tools/specview.cc')
-rw-r--r--tools/specview.cc94
1 files changed, 94 insertions, 0 deletions
diff --git a/tools/specview.cc b/tools/specview.cc
new file mode 100644
index 0000000..77b362c
--- /dev/null
+++ b/tools/specview.cc
@@ -0,0 +1,94 @@
+#include <stdio.h>
+#include <string.h>
+#include <vector>
+#include <algorithm> // For std::max_element
+
+// Redefine SpecHeader from spectool.cc
+struct SpecHeader {
+ char magic[4];
+ int32_t version;
+ int32_t dct_size;
+ int32_t num_frames;
+};
+
+void print_usage() {
+ printf("Usage: specview <input.spec>\n");
+ printf("Displays an ASCII representation of a spectrogram file.\n");
+}
+
+int main(int argc, char** argv) {
+ if (argc < 2) {
+ print_usage();
+ return 1;
+ }
+
+ const char* input_path = argv[1];
+
+ FILE* f_in = fopen(input_path, "rb");
+ if (!f_in) {
+ printf("Error: Failed to open input file: %s\n", input_path);
+ return 1;
+ }
+
+ SpecHeader header;
+ if (fread(&header, sizeof(SpecHeader), 1, f_in) != 1 || strncmp(header.magic, "SPEC", 4) != 0) {
+ printf("Error: Invalid spectrogram file format.\n");
+ fclose(f_in);
+ return 1;
+ }
+
+ if (header.version != 1) {
+ printf("Error: Unsupported spectrogram version %d.\n", header.version);
+ fclose(f_in);
+ return 1;
+ }
+
+ std::vector<float> spec_data(header.num_frames * header.dct_size);
+ if (fread(spec_data.data(), sizeof(float), spec_data.size(), f_in) != spec_data.size()) {
+ printf("Error: Failed to read all spectrogram data.\n");
+ fclose(f_in);
+ return 1;
+ }
+ fclose(f_in);
+
+ printf("Spectrogram: %s\n", input_path);
+ printf(" DCT Size: %d\n", header.dct_size);
+ printf(" Num Frames: %d\n", header.num_frames);
+
+ // Find max magnitude for normalization
+ float max_mag = 0.0f;
+ for (float val : spec_data) {
+ max_mag = std::max(max_mag, fabsf(val));
+ }
+ if (max_mag == 0.0f) max_mag = 1.0f; // Avoid division by zero
+
+ // ASCII visualization
+ const char* gradient = " .:-=+*#%@";
+ int gradient_len = strlen(gradient);
+
+ printf("\nASCII Visualization:\n");
+ for (int frame = 0; frame < header.num_frames; ++frame) {
+ printf("%4d: ", frame);
+ const float* current_frame_data = spec_data.data() + frame * header.dct_size;
+
+ // Average bins into fewer columns for better fit on console
+ const int display_cols = 80; // Max console width
+ const int bins_per_col = header.dct_size / display_cols; // Divide into 80 bins
+
+ for (int col = 0; col < display_cols; ++col) {
+ float sum_mag = 0.0f;
+ for (int bin_idx = 0; bin_idx < bins_per_col; ++bin_idx) {
+ int current_bin = col * bins_per_col + bin_idx;
+ if (current_bin < header.dct_size) {
+ sum_mag += fabsf(current_frame_data[current_bin]);
+ }
+ }
+ float avg_mag = sum_mag / bins_per_col;
+ int char_idx = (int)((avg_mag / max_mag) * (gradient_len - 1));
+ printf("%c", gradient[char_idx]);
+ }
+ printf("\n");
+ }
+
+ return 0;
+}