Initial commit

1 files changed, 131 insertions, 0 deletions

diff --git a/src/configuration.hpp b/src/configuration.hpp
new file mode 100644
index 0000000..126a10d
--- /dev/null
+++ b/src/configuration.hpp
@@ -0,0 +1,131 @@
+/*
+ * Copyright (c) 2020-2021 Vasile Vilvoiu <vasi.vilvoiu@gmail.com>
+ *
+ * specgram is free software; you can redistribute it and/or modify
+ * it under the terms of the MIT license. See LICENSE for details.
+ */
+
+#ifndef _CONFIGURATION_HPP_
+#define _CONFIGURATION_HPP_
+
+#include "color-map.hpp"
+#include "value-map.hpp"
+#include "window-function.hpp"
+
+#include <SFML/Graphics/Color.hpp>
+#include <string>
+#include <optional>
+#include <tuple>
+
+class Configuration {
+private:
+    std::optional<std::string> input_filename_;
+    std::optional<std::string> output_filename_;
+
+    std::size_t block_size_;
+    double rate_;
+    DataType datatype_;
+    bool has_complex_input_;
+    double prescale_factor_;
+
+    std::size_t fft_width_;
+    std::size_t fft_stride_;
+    WindowFunctionType window_function_;
+    std::size_t average_count_;
+    bool alias_negative_;
+
+    bool no_resampling_;
+    std::size_t width_;
+    double min_freq_;
+    double max_freq_;
+    ValueMapType scale_;
+    double scale_lower_bound_;
+    ColorMapType color_map_;
+    sf::Color color_map_custom_color_;
+    sf::Color background_color_;
+    sf::Color foreground_color_;
+    bool has_axes_;
+    bool has_legend_;
+    bool is_horizontal_;
+    bool print_input_;
+    bool print_fft_;
+    bool print_output_;
+
+    bool live_;
+    std::size_t count_;
+    std::string title_;
+
+    bool has_live_window_;
+
+    std::size_t margin_size_;
+    std::size_t live_margin_size_;
+    std::size_t minimum_margin_size_;
+    std::size_t legend_height_;
+    std::size_t live_fft_height_;
+    std::size_t axis_font_size_;
+
+    Configuration();
+
+    static sf::Color StringToColor(const std::string& str);
+
+public:
+    /* parse command line arguments and return a configuration object */
+    static std::tuple<Configuration, int, bool> FromArgs(int argc, char **argv);
+
+    /* generic getters */
+    Configuration GetForLive() const;
+    const auto & GetInputFilename() const { return input_filename_; }
+    const auto & GetOutputFilename() const { return output_filename_; }
+
+    /* input getters */
+    auto GetBlockSize() const { return block_size_; }
+    auto GetRate() const { return rate_; }
+    auto GetDataType() const { return datatype_; }
+    auto HasComplexInput() const { return has_complex_input_; }
+    auto GetPrescaleFactor() const { return prescale_factor_; }
+
+    /* FFT getters */
+    auto GetFFTWidth() const { return fft_width_; }
+    auto GetFFTStride() const { return fft_stride_; }
+    auto GetWindowFunction() const { return window_function_; }
+    auto IsAliasingNegativeFrequencies() const { return alias_negative_; }
+    auto GetAverageCount() const { return average_count_; }
+
+    /* display getters */
+    auto CanResample() const { return !no_resampling_; }
+    auto GetWidth() const { return width_; }
+    auto GetMinFreq() const { return min_freq_; }
+    auto GetMaxFreq() const { return max_freq_; }
+    auto GetScale() const { return scale_; }
+    auto GetScaleLowerBound() const { return scale_lower_bound_; }
+    auto GetColorMap() const { return color_map_; }
+    auto GetColorMapCustomColor() const { return color_map_custom_color_; }
+    auto GetBackgroundColor() const { return background_color_; }
+    auto GetForegroundColor() const { return foreground_color_; }
+    auto HasAxes() const { return has_axes_ || has_legend_; }
+    auto HasLegend() const { return has_legend_; }
+    auto IsHorizontal() const { return is_horizontal_; }
+    auto MustPrintInput() const { return print_input_; }
+    auto MustPrintFFT() const { return print_fft_; }
+    auto MustPrintOutput() const { return print_output_; }
+
+    /* live options */
+    auto IsLive() const { return live_; }
+    auto GetCount() const { return count_; }
+    auto GetTitle() const { return title_; }
+
+    /* internal options */
+    auto HasLiveWindow() const { return has_live_window_; }
+
+    auto GetMarginSize() const { return margin_size_; }
+    auto GetLiveMarginSize() const { return live_margin_size_; }
+    auto GetMinimumMarginSize() const { return minimum_margin_size_; }
+    auto GetLegendHeight() const { return legend_height_; }
+    auto GetLiveFFTHeight() const { return live_fft_height_; }
+
+    auto GetAxisFontSize() const { return axis_font_size_; }
+
+    sf::Color GetLiveGuidelinesColor() const;
+};
+
+#endif
\ No newline at end of file