Add assertions

Author: jurihock

src/Robocode/Assert.h

@@ -0,0 +1,71 @@
+#pragma once
+
+#include <cstdarg>
+#include <filesystem>
+#include <string>
+#include <utility>
+
+#define assert_true(condition, ...)  \
+  do                                 \
+  {                                  \
+    assert_throw_if(                 \
+      !(condition),                  \
+      (#condition),                  \
+      __FILE__,                      \
+      __LINE__,                      \
+      __VA_ARGS__);                  \
+  }                                  \
+  while(0)                          //
+
+#define assert_false(condition, ...) \
+  do                                 \
+  {                                  \
+    assert_throw_if(                 \
+      (condition),                   \
+      (#condition),                  \
+      __FILE__,                      \
+      __LINE__,                      \
+      __VA_ARGS__);                  \
+  }                                  \
+  while(0)                          //
+
+template<typename... Args>
+static void assert_throw_if(
+  const bool condition,
+  const char* expression,
+  const char* file,
+  const int line,
+  const char* message,
+  Args&&... args)
+{
+  if (!condition)
+  {
+    return;
+  }
+
+  auto format = [](const char* format, ...)
+  {
+    va_list args;
+    va_list temp;
+
+    va_start(args, format);
+
+    va_copy(temp, args);
+    auto size = std::vsnprintf(nullptr, 0, format, temp);
+    va_end(temp);
+
+    std::string buffer(size + 1, '\0');
+    std::vsnprintf(buffer.data(), buffer.size(), format, args);
+    if (!buffer.empty()) { buffer.pop_back(); }
+
+    va_end(args);
+
+    return buffer;
+  };
+
+  auto filename = std::filesystem::path(file).filename();
+
+  throw std::runtime_error(
+    format(message, std::forward<Args>(args)...) +
+    format(" (%s in %s at %d)", expression, filename.c_str(), line));
+}

src/Robocode/Channel.cpp

@@ -9,6 +9,8 @@ Channel::Channel(const size_t index, const size_t dftsize, const double samplera
   // https://newt.phys.unsw.edu.au/jw/notes.html
   const double hz = std::pow(2, (double(index) - 69) / 12) * concertpitch;
 
+  config.dftsize = dftsize;
+
   config.freq = hz;
   config.gain = 0;
 
@@ -49,6 +51,8 @@ double Channel::synthesize(const std::span<std::complex<double>> dft,
                            const std::span<const double> pvcfreqs,
                            const double gestalt) // TODO: gestalt
 {
+  assert_true(dft.size() == config.dftsize, "Invalid DFT size!");
+
   const double tophase = (2 * std::numbers::pi) / config.samplerate;
 
   const double freq = config.freq;
@@ -62,12 +66,12 @@ double Channel::synthesize(const std::span<std::complex<double>> dft,
     const double f0 = pvcfreqs[i];
     const double f1 = freq * i;
 
-    freqs[i] = f1 * f0 / dftfreqs[i]; // TODO: f1 + (f0 - dftfreqs[i]) * (f1 / dftfreqs[i])
+    freqs[i] = f1 * f0 / dftfreqs[i];
   }
 
   for (size_t i = 1; i < dft.size() - 1; ++i)
   {
-    phase[i] += freqs[i] * tophase; // TODO: omega[i] + (freqs[i] - dftfreqs[i]) * tophase
+    phase[i] += freqs[i] * tophase;
 
     if (freqs[i] <= dftfreqs.front())
     {

src/Robocode/Channel.h

@@ -25,6 +25,8 @@ class Channel final
 
   struct
   {
+    size_t dftsize;
+
     double freq;
     double gain;
 

src/Robocode/Channels.cpp

@@ -25,7 +25,7 @@ void Channels::synthesize(const std::span<const std::complex<double>> dftanal,
     sum += channels[channel].synthesize(dftsynth, dftfreqs, pvcfreqs, gestalt);
   }
 
-  for (size_t i = 1; i < dftsynth.size() - 1; ++i)
+  for (size_t i = 1; i < dftfreqs.size() - 1; ++i)
   {
     const double abs = std::abs(dftanal[i]);
 

src/Robocode/Header.h

@@ -15,3 +15,5 @@
 #include <stdexcept>
 #include <utility>
 #include <vector>
+
+#include <Robocode/Assert.h>

src/Robocode/SDFT.cpp

@@ -5,6 +5,7 @@ SDFT::SDFT(const double samplerate, const size_t dftsize) :
   input(dftsize),
   output(dftsize)
 {
+  assert_true(dftsize > 0, "Invalid DFT size!");
 }
 
 float SDFT::transform(const float x, const std::function<void(

src/Robocode/Vocoder.cpp

@@ -13,17 +13,17 @@ inline double angle(const std::complex<double>& z)
 
 Vocoder::Vocoder(const double samplerate, const std::vector<double>& frequencies) :
   samplerate(samplerate),
-  frequencies(frequencies)
+  dftsize(frequencies.size())
 {
-  const size_t dftsize = frequencies.size();
-
   cache.resize(dftsize);
   freqs.resize(dftsize);
 }
 
 void Vocoder::analyze(const std::span<const std::complex<double>> dft,
                       const std::function<void(const std::span<const double> pvcfreqs)> callback)
 {
+  assert_true(dft.size() == dftsize, "Invalid DFT size!");
+
   const double tofreq = samplerate / (2 * std::numbers::pi);
 
   for (size_t i = 0; i < dft.size(); ++i)

src/Robocode/Vocoder.h

@@ -15,7 +15,7 @@ class Vocoder final
 private:
 
   const double samplerate;
-  const std::vector<double> frequencies;
+  const size_t dftsize;
 
   std::vector<std::complex<double>> cache;
   std::vector<double> freqs;