223 lines
7.0 KiB
C++
223 lines
7.0 KiB
C++
/**
|
|
* vim: set ts=4 :
|
|
* =============================================================================
|
|
* SourceMod Sample Extension
|
|
* Copyright (C) 2004-2008 AlliedModders LLC. All rights reserved.
|
|
* =============================================================================
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it under
|
|
* the terms of the GNU General Public License, version 3.0, as published by the
|
|
* Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
|
|
* FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
|
|
* details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along with
|
|
* this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
* As a special exception, AlliedModders LLC gives you permission to link the
|
|
* code of this program (as well as its derivative works) to "Half-Life 2," the
|
|
* "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
|
|
* by the Valve Corporation. You must obey the GNU General Public License in
|
|
* all respects for all other code used. Additionally, AlliedModders LLC grants
|
|
* this exception to all derivative works. AlliedModders LLC defines further
|
|
* exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
|
|
* or <http://www.sourcemod.net/license.php>.
|
|
*
|
|
* Version: $Id$
|
|
*/
|
|
|
|
#ifndef _INCLUDE_SOURCEMOD_EXTENSION_PROPER_H_
|
|
#define _INCLUDE_SOURCEMOD_EXTENSION_PROPER_H_
|
|
|
|
#include <poll.h>
|
|
#include "smsdk_ext.h"
|
|
#include "celt_header.h"
|
|
#include "ringbuffer.h"
|
|
|
|
/**
|
|
* @file extension.h
|
|
* @brief Sample extension code header.
|
|
*/
|
|
|
|
#define MAX_CLIENTS 16
|
|
|
|
#ifdef _WIN32
|
|
typedef __int64 int64;
|
|
#else
|
|
typedef long long int64;
|
|
#endif
|
|
|
|
class CDetour;
|
|
class IClient;
|
|
typedef void (*t_SV_BroadcastVoiceData)(IClient *, int, unsigned char *, int64);
|
|
|
|
/**
|
|
* @brief Sample implementation of the SDK Extension.
|
|
* Note: Uncomment one of the pre-defined virtual functions in order to use it.
|
|
*/
|
|
class CVoice :
|
|
public SDKExtension,
|
|
public IConCommandBaseAccessor
|
|
{
|
|
public:
|
|
/**
|
|
* @brief This is called after the initial loading sequence has been processed.
|
|
*
|
|
* @param error Error message buffer.
|
|
* @param maxlength Size of error message buffer.
|
|
* @param late Whether or not the module was loaded after map load.
|
|
* @return True to succeed loading, false to fail.
|
|
*/
|
|
virtual bool SDK_OnLoad(char *error, size_t maxlength, bool late);
|
|
|
|
/**
|
|
* @brief This is called right before the extension is unloaded.
|
|
*/
|
|
virtual void SDK_OnUnload();
|
|
|
|
/**
|
|
* @brief This is called once all known extensions have been loaded.
|
|
* Note: It is is a good idea to add natives here, if any are provided.
|
|
*/
|
|
virtual void SDK_OnAllLoaded();
|
|
|
|
/**
|
|
* @brief Called when the pause state is changed.
|
|
*/
|
|
//virtual void SDK_OnPauseChange(bool paused);
|
|
|
|
/**
|
|
* @brief this is called when Core wants to know if your extension is working.
|
|
*
|
|
* @param error Error message buffer.
|
|
* @param maxlength Size of error message buffer.
|
|
* @return True if working, false otherwise.
|
|
*/
|
|
//virtual bool QueryRunning(char *error, size_t maxlength);
|
|
public:
|
|
#if defined SMEXT_CONF_METAMOD
|
|
/**
|
|
* @brief Called when Metamod is attached, before the extension version is called.
|
|
*
|
|
* @param error Error buffer.
|
|
* @param maxlength Maximum size of error buffer.
|
|
* @param late Whether or not Metamod considers this a late load.
|
|
* @return True to succeed, false to fail.
|
|
*/
|
|
virtual bool SDK_OnMetamodLoad(ISmmAPI *ismm, char *error, size_t maxlength, bool late);
|
|
|
|
/**
|
|
* @brief Called when Metamod is detaching, after the extension version is called.
|
|
* NOTE: By default this is blocked unless sent from SourceMod.
|
|
*
|
|
* @param error Error buffer.
|
|
* @param maxlength Maximum size of error buffer.
|
|
* @return True to succeed, false to fail.
|
|
*/
|
|
//virtual bool SDK_OnMetamodUnload(char *error, size_t maxlength);
|
|
|
|
/**
|
|
* @brief Called when Metamod's pause state is changing.
|
|
* NOTE: By default this is blocked unless sent from SourceMod.
|
|
*
|
|
* @param paused Pause state being set.
|
|
* @param error Error buffer.
|
|
* @param maxlength Maximum size of error buffer.
|
|
* @return True to succeed, false to fail.
|
|
*/
|
|
//virtual bool SDK_OnMetamodPauseChange(bool paused, char *error, size_t maxlength);
|
|
#endif
|
|
|
|
public: // IConCommandBaseAccessor
|
|
virtual bool RegisterConCommandBase(ConCommandBase *pVar);
|
|
|
|
public:
|
|
CVoice();
|
|
void OnGameFrame(bool simulating);
|
|
|
|
void ListenSocket();
|
|
void PushPlayerVoiceData(int playerSlot, int nBytes, char *data);
|
|
void BroadcastVoiceDataTorchlightCelt(IClient *pClient, int16_t *pPCM, int nSamples);
|
|
void TranscodeNoSteamToSteam(int playerSlot, int nBytes, char *data);
|
|
void HandlePlayerVoiceData(int playerSlot);
|
|
|
|
private:
|
|
int m_ListenSocket;
|
|
|
|
struct CClient
|
|
{
|
|
int m_Socket;
|
|
size_t m_BufferWriteIndex;
|
|
size_t m_LastLength;
|
|
double m_LastValidData;
|
|
bool m_New;
|
|
bool m_UnEven;
|
|
unsigned char m_Remainder;
|
|
} m_aClients[MAX_CLIENTS];
|
|
|
|
struct pollfd m_aPollFds[1 + MAX_CLIENTS];
|
|
int m_PollFds;
|
|
|
|
CRingBuffer m_Buffer;
|
|
|
|
OpusEncoder *m_OpusEncoder;
|
|
|
|
// Torchlight transcode state (48000 Hz stereo -> 22050 Hz mono -> CELT)
|
|
OpusDecoder *m_TorchlightOpusDecoder;
|
|
int16_t m_torchMonoAccum[4096];
|
|
int m_torchMonoAccumLen;
|
|
int m_torchResampleAccum;
|
|
|
|
// Player transcode state (24000 Hz mono -> 22050 Hz mono -> CELT)
|
|
OpusDecoder *m_PlayerOpusDecoder[SM_MAXPLAYERS + 1];
|
|
|
|
double m_AvailableTime;
|
|
double m_playerAvailableTime[SM_MAXPLAYERS + 1];
|
|
|
|
// NoSteam->Steam transcode state per player
|
|
CRingBuffer m_nosteamOpusPCMBuffer[SM_MAXPLAYERS + 1];
|
|
int m_nosteamResampleAccum[SM_MAXPLAYERS + 1];
|
|
OpusEncoder *m_nosteamOpusEncoder[SM_MAXPLAYERS + 1];
|
|
|
|
// CELT encoder (shared by both paths, same output format)
|
|
struct CEncoderSettings
|
|
{
|
|
int32_t SampleRate_Hz;
|
|
int32_t TargetBitRate_Kbps;
|
|
int32_t FrameSize;
|
|
int32_t PacketSize;
|
|
int32_t Complexity;
|
|
double FrameTime;
|
|
} m_CeltEncoderSettings;
|
|
|
|
CELTMode *m_pCeltMode;
|
|
CELTEncoder *m_pCeltCodec;
|
|
|
|
CELTMode *m_pCeltModePlayer;
|
|
CELTEncoder *m_pCeltCodecPlayer[SM_MAXPLAYERS + 1];
|
|
|
|
CELTDecoder *m_pCeltDecoder[SM_MAXPLAYERS + 1];
|
|
|
|
CRingBuffer m_playerVoiceBuffer[SM_MAXPLAYERS + 1];
|
|
|
|
t_SV_BroadcastVoiceData m_SV_BroadcastVoiceData;
|
|
CDetour *m_VoiceDetour;
|
|
CDetour *m_ProcessVoiceDataDetour;
|
|
|
|
uint16_t m_nosteamSeqNum[SM_MAXPLAYERS + 1];
|
|
|
|
double m_nosteamAvailableTime[SM_MAXPLAYERS + 1];
|
|
|
|
int m_TorchlightSeqNum;
|
|
|
|
void HandleNetwork();
|
|
void OnDataReceived(CClient *pClient, int16_t *pData, size_t Samples);
|
|
void HandleVoiceDataTorchlight();
|
|
void HandleNoSteamVoiceData();
|
|
};
|
|
|
|
#endif // _INCLUDE_SOURCEMOD_EXTENSION_PROPER_H_
|