diff options
Diffstat (limited to 'engine/voice_codecs/celt/voiceencoder_celt.cpp')
| -rw-r--r-- | engine/voice_codecs/celt/voiceencoder_celt.cpp | 237 |
1 files changed, 237 insertions, 0 deletions
diff --git a/engine/voice_codecs/celt/voiceencoder_celt.cpp b/engine/voice_codecs/celt/voiceencoder_celt.cpp new file mode 100644 index 0000000..5c3ee62 --- /dev/null +++ b/engine/voice_codecs/celt/voiceencoder_celt.cpp @@ -0,0 +1,237 @@ +//========= Copyright Valve Corporation, All rights reserved. ============// +// +// Purpose: +// +// $NoKeywords: $ +// +//=============================================================================// + +/* This product contains Speex software. The license terms of the Speex +software, distributed with this product, are as follows: + +� 2002-2003, Jean-Marc Valin/Xiph.Org Foundation + +Redistribution and use in source and binary forms, with or without +modification, are permitted provided that the following conditions are met: + +- Redistributions of source code must retain the above copyright notice, +this list of conditions and the following disclaimer. + +- Redistributions in binary form must reproduce the above copyright notice, +this list of conditions and the following disclaimer in the documentation +and/or +other materials provided with the distribution. + +Neither the name of the Xiph.org Foundation nor the names of its +contributors may be used to endorse or promote products derived from this +software without specific prior written permission. + +This software is provided by the copyright holders and contributors "as is" +and any express or implied warranties, including, but not limited to, the +implied warranties of merchantability and fitness for a particular purpose +are disclaimed. In no event shall the foundation or contributors be liable +for any direct, indirect, incidental, special, exemplary, or consequential +damages (including, but not limited to, procurement of substitute goods or +services; loss of use, data, or profits; or business interruption) however +caused and on any theory of liability, whether in contract, strict +liability, or tort (including negligence or otherwise) arising in any way +out of the use of this software, even if advised of the possibility of such +damage. */ + +#include "ivoicecodec.h" +#include "iframeencoder.h" + +#ifdef POSIX +#include "source/osx/config.h" +#else +#include "source/msvc/config.h" +#endif +#include <stdio.h> +#include "celt.h" + + +// NOTE: This has to be the last file included! +#include "tier0/memdbgon.h" + + +#define CHANNELS 1 + +struct celt_versions +{ + int iSampleRate; + int iRawFrameSize; + int iPacketSize; +}; + +#define CELT_VERSION 4 + +celt_versions g_CeltVersion[CELT_VERSION] = +{ + { + 44100, 256, 120 + }, + + { + 22050, 120, 60 + }, + + { + 22050, 256, 60 + }, + + { + 22050, 512, 64 + }, +}; + +class VoiceEncoder_Celt : public IFrameEncoder +{ +public: + VoiceEncoder_Celt(); + virtual ~VoiceEncoder_Celt(); + + // Interfaces IFrameDecoder + + bool Init(int quality, int &rawFrameSize, int &encodedFrameSize); + void Release(); + void DecodeFrame(const char *pCompressed, char *pDecompressedBytes); + void EncodeFrame(const char *pUncompressedBytes, char *pCompressed); + bool ResetState(); + +private: + + bool InitStates(); + void TermStates(); + + CELTEncoder *m_EncoderState; // Celt internal encoder state + CELTDecoder *m_DecoderState; // Celt internal decoder state + CELTMode *m_Mode; + + int m_iVersion; +}; + +extern IVoiceCodec* CreateVoiceCodec_Frame(IFrameEncoder *pEncoder); + +void* CreateCeltVoiceCodec() +{ + IFrameEncoder *pEncoder = new VoiceEncoder_Celt; + return CreateVoiceCodec_Frame( pEncoder ); +} + +EXPOSE_INTERFACE_FN(CreateCeltVoiceCodec, IVoiceCodec, "vaudio_celt") + +////////////////////////////////////////////////////////////////////// +// Construction/Destruction +////////////////////////////////////////////////////////////////////// + +VoiceEncoder_Celt::VoiceEncoder_Celt() +{ + m_EncoderState = NULL; + m_DecoderState = NULL; + m_Mode = NULL; + m_iVersion = 0; +} + +VoiceEncoder_Celt::~VoiceEncoder_Celt() +{ + TermStates(); +} + +bool VoiceEncoder_Celt::Init( int quality, int &rawFrameSize, int &encodedFrameSize) +{ + if ( quality >= CELT_VERSION ) + return false; + + m_iVersion = quality; + + rawFrameSize = g_CeltVersion[m_iVersion].iRawFrameSize * BYTES_PER_SAMPLE; + + int iError = 0; + + m_Mode = celt_mode_create( g_CeltVersion[m_iVersion].iSampleRate, g_CeltVersion[m_iVersion].iRawFrameSize, &iError ); + m_EncoderState = celt_encoder_create_custom( m_Mode, CHANNELS, NULL); + m_DecoderState = celt_decoder_create_custom( m_Mode, CHANNELS, NULL); + + if ( !InitStates() ) + return false; + + encodedFrameSize = g_CeltVersion[m_iVersion].iPacketSize; + + return true; +} + +void VoiceEncoder_Celt::Release() +{ + delete this; +} + +void VoiceEncoder_Celt::EncodeFrame(const char *pUncompressedBytes, char *pCompressed) +{ + unsigned char output[1024]; + + celt_encode( m_EncoderState, (celt_int16*)pUncompressedBytes, g_CeltVersion[m_iVersion].iRawFrameSize, output, g_CeltVersion[m_iVersion].iPacketSize ); + + for ( int i = 0; i < g_CeltVersion[m_iVersion].iPacketSize; i++ ) + { + *pCompressed = (char)output[i]; + pCompressed++; + } +} + +void VoiceEncoder_Celt::DecodeFrame(const char *pCompressed, char *pDecompressedBytes) +{ + unsigned char output[1024]; + char *out = (char *)pCompressed; + + if ( !pCompressed ) + { + celt_decode( m_DecoderState, NULL, g_CeltVersion[m_iVersion].iPacketSize, (celt_int16 *)pDecompressedBytes, g_CeltVersion[m_iVersion].iRawFrameSize ); + return; + } + + for ( int i = 0; i < g_CeltVersion[m_iVersion].iPacketSize; i++ ) + { + output[i] = ( unsigned char ) ( ( *out < 0 ) ? (*out + 256) : *out ); + out++; + } + + + //celt_decoder_ctl( m_DecoderState, CELT_RESET_STATE_REQUEST, NULL ); + celt_decode( m_DecoderState, output, g_CeltVersion[m_iVersion].iPacketSize, (celt_int16 *)pDecompressedBytes, g_CeltVersion[m_iVersion].iRawFrameSize ); +} + +bool VoiceEncoder_Celt::ResetState() +{ + celt_encoder_ctl(m_EncoderState, CELT_RESET_STATE_REQUEST , NULL ); + celt_decoder_ctl(m_DecoderState, CELT_RESET_STATE_REQUEST , NULL ); + + return true; +} + +bool VoiceEncoder_Celt::InitStates() +{ + if ( !m_EncoderState || !m_DecoderState ) + return false; + + celt_encoder_ctl( m_EncoderState, CELT_RESET_STATE_REQUEST , NULL ); + celt_decoder_ctl( m_DecoderState, CELT_RESET_STATE_REQUEST , NULL ); + + return true; +} + +void VoiceEncoder_Celt::TermStates() +{ + if( m_EncoderState ) + { + celt_encoder_destroy( m_EncoderState ); + m_EncoderState = NULL; + } + + if( m_DecoderState ) + { + celt_decoder_destroy( m_DecoderState ); + m_DecoderState = NULL; + } + + celt_mode_destroy( m_Mode ); +} |