File: detect_header.cpp

package info (click to toggle)
satdump 1.2.2%2Bgb79af48-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 81,648 kB
  • sloc: cpp: 276,768; ansic: 164,598; lisp: 1,219; sh: 283; xml: 106; makefile: 7
file content (103 lines) | stat: -rw-r--r-- 3,733 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
#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;
    }
}