common cpp

examples/CMakeLists.txt

@@ -1,4 +1,6 @@
-include_directories(${CMAKE_CURRENT_SOURCE_DIR})
+add_library(common STATIC common.cpp)
+target_include_directories(common PUBLIC ${CMAKE_CURRENT_SOURCE_DIR})
+target_compile_features(common PRIVATE cxx_std_11)
 
 add_subdirectory(main)
 add_subdirectory(quantize)

examples/common.cpp

@@ -0,0 +1,23 @@
+#include <string> 
+#include <vector>
+
+#define DR_WAV_IMPLEMENTATION
+#include "dr_wav.h"
+
+#define SAMPLE_RATE 24000
+
+void write_wav_on_disk(std::vector<float> & audio_arr, std::string dest_path) {
+ drwav_data_format format;
+ format.bitsPerSample = 32;
+ format.sampleRate = SAMPLE_RATE;
+ format.container = drwav_container_riff;
+ format.channels = 1;
+ format.format = DR_WAVE_FORMAT_IEEE_FLOAT;
+
+ drwav wav;
+ drwav_init_file_write(&wav, dest_path.c_str(), &format, NULL);
+ drwav_uint64 frames = drwav_write_pcm_frames(&wav, audio_arr.size(), audio_arr.data());
+ drwav_uninit(&wav);
+
+ fprintf(stderr, "%s: Number of frames written = %lld.\n", __func__, frames);
+}

examples/common.h

@@ -0,0 +1,12 @@
+#include <string>
+#include <vector>
+
+/**
+ * @brief Reads a WAV file from disk and stores the audio data in a vector of floats.
+ *
+ * @param in_path Path to the input WAV file.
+ * @param audio_arr Vector to store the audio data.
+ * @return true If the file was successfully read.
+ * @return false If the file could not be read.
+ */
+bool read_wav_from_disk(std::string in_path, std::vector<float> & audio_arr);