satdump/src-core/common/detect_header.cpp

#include "detect_header.h"
#include "common/wav.h"
#include <cstdint>
#include "wav.h"
#ifdef BUILD_ZIQ
#include "ziq.h"
#endif

#include "logger.h"

#include "ziq2.h"

HeaderInfo try_parse_header(std::string file)
{
    HeaderInfo info;

    if (wav::isValidWav(wav::parseHeaderFromFileWav(file)))
    {
        logger->debug("File is wav!");
        wav::WavHeader wav_header = wav::parseHeaderFromFileWav(file);
        info.samplerate = wav_header.samplerate;
        info.valid = true;
        if (wav_header.bits_per_sample == 8 && wav_header.audio_Format == 1)
            info.type = "cu8";
        else if (wav_header.bits_per_sample == 16 && wav_header.audio_Format == 1)
            info.type = "cs16";
        else if (wav_header.bits_per_sample == 32 && wav_header.audio_Format == 1)
            info.type = "cs32";
        else if (wav_header.bits_per_sample == 32 && wav_header.audio_Format == 3)
            info.type = "cf32";
        else
        {
            info.valid = false;
            logger->warn("Unsupported WAV format! Bits per sample: %hu, Format: %hu",
                wav_header.bits_per_sample, wav_header.audio_Format);
        }
    }
    else if (wav::isValidRF64(wav::parseHeaderFromFileWav(file)))
    {
        wav::RF64Header rf64_header = wav::parseHeaderFromFileRF64(file);
        logger->debug("File is RF64!");
        info.samplerate = rf64_header.samplerate;
        info.valid = true;
        if (rf64_header.bits_per_sample == 8 && rf64_header.audio_Format == 1)
            info.type = "cu8";
        else if (rf64_header.bits_per_sample == 16 && rf64_header.audio_Format == 1)
            info.type = "cs16";
        else if (rf64_header.bits_per_sample == 32 && rf64_header.audio_Format == 1)
            info.type = "cs32";
        else if (rf64_header.bits_per_sample == 32 && rf64_header.audio_Format == 3)
            info.type = "cf32";
        else
        {
            info.valid = false;
            logger->warn("Unsupported RF64 format! Bits per sample: %hu, Format: %hu",
                rf64_header.bits_per_sample, rf64_header.audio_Format);
        }
    }
#ifdef BUILD_ZIQ
    else if (ziq::isValidZIQ(file))
    {
        logger->debug("File is ZIQ!");
        info.type = "ziq";
        info.valid = true;
        info.samplerate = ziq::getCfgFromFile(file).samplerate;
    }
#endif
#ifdef BUILD_ZIQ2
    else if (ziq2::ziq2_is_valid_ziq2(file))
    {
        logger->debug("File is ZIQ2!");
        info.type = "ziq2";
        info.valid = true;
        info.samplerate = ziq2::ziq2_try_parse_header(file);
    }
#endif

    return info;
}

void try_get_params_from_input_file(nlohmann::json &parameters, std::string input_file)
{
    if (std::filesystem::exists(input_file) && !std::filesystem::is_directory(input_file))
    {
        HeaderInfo hdr = try_parse_header(input_file);
        if (hdr.valid)
        {
            if (!parameters.contains("samplerate"))
                parameters["samplerate"] = hdr.samplerate;
            if (!parameters.contains("baseband_format"))
                parameters["baseband_format"] = hdr.type;
        }
        wav::FileMetadata md = wav::tryParseFilenameMetadata(input_file, false);
        if (!parameters.contains("start_timestamp") && md.timestamp != 0)
            parameters["start_timestamp"] = md.timestamp;
        if (!parameters.contains("samplerate") && md.samplerate != 0)
            parameters["samplerate"] = md.samplerate;
        if (!parameters.contains("frequency") && md.frequency != 0)
            parameters["frequency"] = md.frequency;
        if (!parameters.contains("baseband_format") && md.baseband_format != "")
            parameters["baseband_format"] = md.baseband_format;
    }
}