VoiceTranscoder/Main.h at master · WPMGPRoSToTeMa/VoiceTranscoder · 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
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
#pragma once

#include "VoiceCodec_Speex.h"
#include "VoiceCodec_SILK.h"
#include <EngineUTIL.h>
#include <rehlds_api.h>
#ifdef min
#undef min
#endif
#ifdef max
#undef max
#endif
#include <vector>
#include <memory>
#include <chrono>
#include <Optional.h>

using namespace std;
using namespace chrono_literals;
using namespace chrono;

class VoiceCodec_OPUS_PLC;

// Structs and classes
struct clientData_t {
	bool HasNewCodec;
	bool IsChecking;
	bool IsVguiRunScriptReceived;
	bool IsSpeaking;
	Optional<time_point<steady_clock>> VoiceEndTime;
	bool IsMuted;
	bool IsBlocked;
	Optional<time_point<steady_clock>> NextVoicePacketExpectedTime;
	VoiceCodec_SILK *NewCodec;
	VoiceCodec_OPUS_PLC *NewCodec2;
	Optional<int> SampleRate;
	VoiceCodec_Speex *OldCodec;
};

// TODO: samples16k
struct playSound_t {
	std::vector<int16_t> samples8k;
	std::vector<int16_t> samples16k;
	size_t currentSample;
	size_t PlayedNewEncodedSampleCount;
	client_t *receiver;
	time_point<steady_clock> OldEncodedDataSendTime;
	time_point<steady_clock> NewEncodedDataSendTime;
	std::unique_ptr<VoiceCodec_Speex> oldCodec;
	std::unique_ptr<VoiceCodec_SILK> newCodec;
};

// VoicePacketCommand
enum : size_t {
	VPC_VDATA_SILENCE = 0,
	VPC_VDATA_MILES = 1, // Really unknown, deprecated
	VPC_VDATA_SPEEX = 2, // Deprecated
	VPC_VDATA_RAW = 3,
	VPC_VDATA_SILK = 4,
	VPC_VDATA_OPUS_PLC = 6,
	VPC_UNKNOWN = 10,
	VPC_SETSAMPLERATE = 11
};

// Constants
const size_t MAX_VOICEPACKET_SIZE = 8192; // or 4096? or 8192?
const size_t MAX_DECOMPRESSED_VOICEPACKET_SAMPLES = 32768; // or 8192?
// uint64_t(steamid) + uint8_t(VPC_SETSAMPLERATE or VPC_VDATA_SILK or VPC_VDATA_SILENCE) + uint16_t(arg) + ... + uint32_t(CRC32 checksum)
constexpr auto MIN_VOICEPACKET_SIZE = sizeof(uint64_t) + sizeof(uint8_t) + sizeof(uint16_t) + sizeof(uint32_t);
const char VTC_CONFIGNAME[] = "VoiceTranscoder.cfg";
const size_t NEWCODEC_WANTED_SAMPLERATE = 16000;
const size_t NEWCODEC_WANTED_SAMPLERATE2 = 24000;
constexpr auto SPEAKING_TIMEOUT = 200ms;

// Externs
extern std::vector<playSound_t> g_playSounds;
extern clientData_t g_clientData[MAX_CLIENTS];

extern cvar_t *g_pcvarForceSendHLTV;
extern cvar_t *g_pcvarVolumeOldToNew;
extern cvar_t *g_pcvarVolumeNewToOld;

extern void OnClientCommandReceiving(edict_t *pClient);
extern qboolean OnClientConnected(edict_t *pClient, const char *pszName, const char *pszAddress, char *pszRejectReason);
extern void OnClientDisconnected(edict_t *pClient);
extern void OnServerActivated(edict_t *pEdictList, int nEdictCount, int nClientMax);
extern void OnFrameStarted();

extern void SV_ParseVoiceData_Hook(client_t *client);
extern void HandleNetCommand_Hook(IRehldsHook_HandleNetCommand *chain, IGameClient *client, uint8_t netcmd);

extern void VTC_InitCvars(void);
extern void VTC_InitAPI(void);
extern void VTC_ExecConfig(void);
extern void VTC_InitConfig(void);
extern void VTC_InitCodecs(void);
extern void VTC_UpdateCodecs();