ManhuntRIBber/main.cpp at master · winterheart/ManhuntRIBber · GitHub

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
/* SPDX-FileCopyrightText: Copyright 2024-2025 Azamat H. Hackimov <azamat.hackimov@gmail.com> */
/* SPDX-License-Identifier: LGPL-2.1-or-later */

#include <cstdint>
#include <filesystem>
#include <format>
#include <fstream>
#include <iostream>
#include <vector>

#include "CLI11.hpp"
#include "byteswap.h"
#include "codec.h"
#include "manhuntribber_version.h"

void decode(const std::filesystem::path &in_file, const std::filesystem::path& out_file, bool is_mono, uint32_t frequency, uint32_t nb_streams) {
  Codec codec(is_mono, frequency, nb_streams);
  codec.decode(in_file, out_file);
}

void encode(const std::vector<std::filesystem::path>& in_files, const std::filesystem::path& out_file) {
  std::ifstream input_file(in_files.front(), std::ios::binary);

  if (!input_file.is_open()) {
    std::cout << std::format("Can't open input file for reading {}", in_files.front().string()) << std::endl;
    exit(1);
  }

  wav_hdr wave_header;
  input_file.read(reinterpret_cast<char *>(&wave_header), sizeof(wav_hdr));
  input_file.close();
  Codec codec(UTILS::convert_le(wave_header.NumOfChan) == 1, UTILS::convert_le(wave_header.SamplesPerSec), in_files.size());
  codec.encode(in_files, out_file);
}

int main(int argc, char *argv[]) {

  std::filesystem::path in_file;
  std::vector<std::filesystem::path> in_files;
  std::filesystem::path out_file;
  bool is_complex = false;
  bool is_mono = false;
  uint32_t frequency = 44100;

  CLI::App app{"ManhuntRIBber - encode/decode RIB files from Rockstar's Manhunt PC game"};
  app.set_version_flag("-v", MANHUNTRIBBER_VERSION);
  argv = app.ensure_utf8(argv);
  std::cout << std::format("ManhuntRIBber {} https://github.com/winterheart/ManhuntRIBber\n"
                           "(c) 2024-2025 Azamat H. Hackimov <azamat.hackimov@gmail.com>\n",
                           app.version())
            << std::endl;

  auto encode_cmd =
      app.add_subcommand("encode", "Encode WAV file to RIB")->callback([&]() { encode(in_files, out_file); });
  encode_cmd->add_option("input", in_files, "Input WAV file(s)")->required()->check(CLI::ExistingFile)->expected(1, 6);
  encode_cmd->add_option("-o,--output", out_file, "Output RIB file");

  auto decode_cmd =
      app.add_subcommand("decode", "Decode RIB file to WAV")->callback([&]() {
        decode(in_file, out_file, is_mono, frequency, is_complex ? 6 : 1);
      });
  decode_cmd->add_flag("-c", is_complex, "Threats input file as Complex stream")->default_val(is_complex);
  decode_cmd->add_option("-f", frequency, "Frequency of the stream")->default_val(frequency);
  decode_cmd->add_flag("-m", is_mono, "Threats input file as Mono stream")->default_val(is_mono);
  decode_cmd->add_option("input", in_file, "Input WAV file")->required()->check(CLI::ExistingFile);
  decode_cmd->add_option("-o,--output", out_file, "Output RIB file");

  CLI11_PARSE(app, argc, argv);

  return 0;
}