add G711a, G711u and SPEEX codecs

3 files changed, 464 insertions, 0 deletions

diff --git a/source/QXmppCodec.cpp b/source/QXmppCodec.cpp
new file mode 100644
index 00000000..f7a0aba3
--- /dev/null
+++ b/source/QXmppCodec.cpp
@@ -0,0 +1,368 @@
+/*
+ * wiLink
+ * Copyright (C) 2009-2010 Bolloré telecom
+ * See AUTHORS file for a full list of contributors.
+ * 
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ * 
+ * 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/>.
+ */
+
+/*
+ *  G.711 based on reference implementation by Sun Microsystems, Inc.
+ */
+
+#include <QDataStream>
+#include <QDebug>
+
+#include "QXmppCodec.h"
+
+#ifdef QXMPP_USE_SPEEX
+#include <speex/speex.h>
+#endif
+
+#define BIAS        (0x84)  /* Bias for linear code. */
+#define CLIP        8159
+
+#define SIGN_BIT    (0x80)  /* Sign bit for a A-law byte. */
+#define QUANT_MASK  (0xf)   /* Quantization field mask. */
+#define NSEGS       (8)     /* Number of A-law segments. */
+#define SEG_SHIFT   (4)     /* Left shift for segment number. */
+#define SEG_MASK    (0x70)  /* Segment field mask. */
+
+static qint16 seg_aend[8] = {0x1F, 0x3F, 0x7F, 0xFF,
+                0x1FF, 0x3FF, 0x7FF, 0xFFF};
+static qint16 seg_uend[8] = {0x3F, 0x7F, 0xFF, 0x1FF,
+                0x3FF, 0x7FF, 0xFFF, 0x1FFF};
+
+static qint16 search(qint16 val, qint16 *table, qint16 size)
+{
+   qint16 i;
+   
+   for (i = 0; i < size; i++) {
+      if (val <= *table++)
+     return (i);
+   }
+   return (size);
+}
+
+/*
+ * linear2alaw() - Convert a 16-bit linear PCM value to 8-bit A-law
+ *
+ * Accepts a 16-bit integer and encodes it as A-law data.
+ *
+ *      Linear Input Code   Compressed Code
+ *  ------------------------    ---------------
+ *  0000000wxyza            000wxyz
+ *  0000001wxyza            001wxyz
+ *  000001wxyzab            010wxyz
+ *  00001wxyzabc            011wxyz
+ *  0001wxyzabcd            100wxyz
+ *  001wxyzabcde            101wxyz
+ *  01wxyzabcdef            110wxyz
+ *  1wxyzabcdefg            111wxyz
+ *
+ * For further information see John C. Bellamy's Digital Telephony, 1982,
+ * John Wiley & Sons, pps 98-111 and 472-476.
+ */
+quint8 linear2alaw(qint16 pcm_val)
+{
+   qint16 mask;
+   qint16 seg;
+   quint8 aval;
+   
+   pcm_val = pcm_val >> 3;
+
+   if (pcm_val >= 0) {
+      mask = 0xD5;      /* sign (7th) bit = 1 */
+   } else {
+      mask = 0x55;      /* sign bit = 0 */
+      pcm_val = -pcm_val - 1;
+   }
+   
+   /* Convert the scaled magnitude to segment number. */
+   seg = search(pcm_val, seg_aend, 8);
+   
+   /* Combine the sign, segment, and quantization bits. */
+   
+   if (seg >= 8)        /* out of range, return maximum value. */
+      return (quint8) (0x7F ^ mask);
+   else {
+      aval = (quint8) seg << SEG_SHIFT;
+      if (seg < 2)
+     aval |= (pcm_val >> 1) & QUANT_MASK;
+      else
+     aval |= (pcm_val >> seg) & QUANT_MASK;
+      return (aval ^ mask);
+   }
+}
+
+/*
+ * alaw2linear() - Convert an A-law value to 16-bit linear PCM
+ *
+ */
+qint16 alaw2linear(quint8 a_val)
+{
+   qint16 t;
+   qint16 seg;
+   
+   a_val ^= 0x55;
+   
+   t = (a_val & QUANT_MASK) << 4;
+   seg = ((qint16)a_val & SEG_MASK) >> SEG_SHIFT;
+   switch (seg) {
+   case 0:
+      t += 8;
+      break;
+   case 1:
+      t += 0x108;
+      break;
+   default:
+      t += 0x108;
+      t <<= seg - 1;
+   }
+   return ((a_val & SIGN_BIT) ? t : -t);
+}
+
+/*
+ * linear2ulaw() - Convert a linear PCM value to u-law
+ *
+ * In order to simplify the encoding process, the original linear magnitude
+ * is biased by adding 33 which shifts the encoding range from (0 - 8158) to
+ * (33 - 8191). The result can be seen in the following encoding table:
+ *
+ *   Biased Linear Input Code    Compressed Code
+ *   ------------------------    ---------------
+ *   00000001wxyza           000wxyz
+ *   0000001wxyzab           001wxyz
+ *   000001wxyzabc           010wxyz
+ *   00001wxyzabcd           011wxyz
+ *   0001wxyzabcde           100wxyz
+ *   001wxyzabcdef           101wxyz
+ *   01wxyzabcdefg           110wxyz
+ *   1wxyzabcdefgh           111wxyz
+ *
+ * Each biased linear code has a leading 1 which identifies the segment
+ * number. The value of the segment number is equal to 7 minus the number
+ * of leading 0's. The quantization interval is directly available as the
+ * four bits wxyz.  * The trailing bits (a - h) are ignored.
+ *
+ * Ordinarily the complement of the resulting code word is used for
+ * transmission, and so the code word is complemented before it is returned.
+ *
+ * For further information see John C. Bellamy's Digital Telephony, 1982,
+ * John Wiley & Sons, pps 98-111 and 472-476.
+ */
+quint8 linear2ulaw(qint16 pcm_val)
+{
+   qint16 mask;
+   qint16 seg;
+   quint8 uval;
+   
+   /* Get the sign and the magnitude of the value. */
+   pcm_val = pcm_val >> 2;
+   if (pcm_val < 0) {
+      pcm_val = -pcm_val;
+      mask = 0x7F;
+   } else {
+      mask = 0xFF;
+   }
+   if (pcm_val > CLIP) pcm_val = CLIP;        /* clip the magnitude */
+   pcm_val += (BIAS >> 2);
+   
+   /* Convert the scaled magnitude to segment number. */
+   seg = search(pcm_val, seg_uend, 8);
+   
+   /*
+   * Combine the sign, segment, quantization bits;
+   * and complement the code word.
+   */
+   if (seg >= 8)        /* out of range, return maximum value. */
+      return (quint8) (0x7F ^ mask);
+   else {
+      uval = (quint8) (seg << 4) | ((pcm_val >> (seg + 1)) & 0xF);
+      return (uval ^ mask);
+   }
+}
+
+/*
+ * ulaw2linear() - Convert a u-law value to 16-bit linear PCM
+ *
+ * First, a biased linear code is derived from the code word. An unbiased
+ * output can then be obtained by subtracting 33 from the biased code.
+ *
+ * Note that this function expects to be passed the complement of the
+ * original code word. This is in keeping with ISDN conventions.
+ */
+qint16 ulaw2linear(quint8 u_val)
+{
+   qint16 t;
+   
+   /* Complement to obtain normal u-law value. */
+   u_val = ~u_val;
+   
+   /*
+    * Extract and bias the quantization bits. Then
+    * shift up by the segment number and subtract out the bias.
+    */
+   t = ((u_val & QUANT_MASK) << 3) + BIAS;
+   t <<= ((unsigned)u_val & SEG_MASK) >> SEG_SHIFT;
+   
+   return ((u_val & SIGN_BIT) ? (BIAS - t) : (t - BIAS));
+}
+
+QXmppG711aCodec::QXmppG711aCodec(int clockrate)
+{
+    m_frequency = clockrate;
+}
+
+int QXmppG711aCodec::bitrate() const
+{
+    return m_frequency * 8;
+}
+
+qint64 QXmppG711aCodec::encode(QDataStream &input, QDataStream &output)
+{
+    qint64 samples = 0;
+    qint16 pcm;
+    while (!input.atEnd())
+    {
+        input >> pcm;
+        output << linear2alaw(pcm);
+        ++samples;
+    }
+    return samples;
+}
+
+qint64 QXmppG711aCodec::decode(QDataStream &input, QDataStream &output)
+{
+    qint64 samples = 0;
+    quint8 g711;
+    while (!input.atEnd())
+    {
+        input >> g711;
+        output << alaw2linear(g711);
+        ++samples;
+    }
+    return samples;
+}
+
+QXmppG711uCodec::QXmppG711uCodec(int clockrate)
+{
+    m_frequency = clockrate;
+}
+
+int QXmppG711uCodec::bitrate() const
+{
+    return m_frequency * 8;
+}
+
+qint64 QXmppG711uCodec::encode(QDataStream &input, QDataStream &output)
+{
+    qint64 samples = 0;
+    qint16 pcm;
+    while (!input.atEnd())
+    {
+        input >> pcm;
+        output << linear2ulaw(pcm);
+        ++samples;
+    }
+    return samples;
+}
+
+qint64 QXmppG711uCodec::decode(QDataStream &input, QDataStream &output)
+{
+    qint64 samples = 0;
+    quint8 g711;
+    while (!input.atEnd())
+    {
+        input >> g711;
+        output << ulaw2linear(g711);
+        ++samples;
+    }
+    return samples;
+}
+
+#ifdef QXMPP_USE_SPEEX
+QXmppSpeexCodec::QXmppSpeexCodec(int clockrate)
+{
+    const SpeexMode *mode = &speex_nb_mode;
+    if (clockrate == 32000)
+        mode = &speex_uwb_mode;
+    else if (clockrate == 16000)
+        mode = &speex_wb_mode;
+    else if (clockrate == 8000)
+        mode = &speex_nb_mode;
+    else
+        qWarning() << "QXmppSpeexCodec got invalid clockrate" << clockrate;
+
+    // encoder
+    encoder_bits = new SpeexBits;
+    speex_bits_init(encoder_bits);
+    encoder_state = speex_encoder_init(mode);
+
+    // decoder
+    decoder_bits = new SpeexBits;
+    speex_bits_init(decoder_bits);
+    decoder_state = speex_decoder_init(mode);
+
+    // get frame size in samples
+    speex_encoder_ctl(encoder_state, SPEEX_GET_FRAME_SIZE, &frame_samples);
+}
+
+QXmppSpeexCodec::~QXmppSpeexCodec()
+{
+    delete encoder_bits;
+    delete decoder_bits;
+}
+
+int QXmppSpeexCodec::bitrate() const
+{
+    int bitrate;
+    speex_encoder_ctl(encoder_state, SPEEX_GET_BITRATE, &bitrate);
+    return bitrate;
+}
+
+qint64 QXmppSpeexCodec::encode(QDataStream &input, QDataStream &output)
+{
+    qint64 samples = 0;
+    QByteArray pcm_buffer(frame_samples * 2, 0);
+    const int length = input.readRawData(pcm_buffer.data(), pcm_buffer.size());
+    if (length != pcm_buffer.size())
+    {
+        qWarning() << "Read only read" << length << "bytes";
+        return samples;
+    }
+    speex_bits_reset(encoder_bits);
+    speex_encode_int(encoder_state, (short*)pcm_buffer.data(), encoder_bits);
+    QByteArray speex_buffer(speex_bits_nbytes(encoder_bits), 0);
+    speex_bits_write(encoder_bits, speex_buffer.data(), speex_buffer.size());
+    output.writeRawData(speex_buffer.data(), speex_buffer.size());
+    return frame_samples;
+}
+
+qint64 QXmppSpeexCodec::decode(QDataStream &input, QDataStream &output)
+{
+    qint64 samples = 0;
+    quint8 g711;
+    const int length = input.device()->bytesAvailable();
+    QByteArray speex_buffer(length, 0);
+    input.readRawData(speex_buffer.data(), speex_buffer.size());
+    speex_bits_read_from(decoder_bits, speex_buffer.data(), speex_buffer.size());
+    QByteArray pcm_buffer(frame_samples * 2, 0);
+    speex_decode_int(decoder_state, decoder_bits, (short*)pcm_buffer.data());
+    output.writeRawData(pcm_buffer.data(), pcm_buffer.size());
+    return frame_samples;
+}
+
+#endif
+
diff --git a/source/QXmppCodec.h b/source/QXmppCodec.h
new file mode 100644
index 00000000..e8f6631c
--- /dev/null
+++ b/source/QXmppCodec.h
@@ -0,0 +1,94 @@
+/*
+ * Copyright (C) 2010 Bolloré telecom
+ *
+ * Author:
+ *	Jeremy Lainé
+ *
+ * Source:
+ *	http://code.google.com/p/qxmpp
+ *
+ * This file is a part of QXmpp library.
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library 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
+ * Lesser General Public License for more details.
+ *
+ */
+
+#ifndef QXMPPCODEC_H
+#define QXMPPCODEC_H
+
+#include <QtGlobal>
+
+/// The QXmppCodec class is the base class for audio codecs capable of
+/// encoding / decoding 16-bit mono samples.
+
+class QXmppCodec
+{
+public:
+    virtual int bitrate() const = 0;
+    virtual qint64 encode(QDataStream &input, QDataStream &output) = 0;
+    virtual qint64 decode(QDataStream &input, QDataStream &output) = 0;
+};
+
+/// The QXmppG711aCodec class represent a G.711 a-law PCM codec.
+
+class QXmppG711aCodec : public QXmppCodec
+{
+public:
+    QXmppG711aCodec(int clockrate);
+
+    int bitrate() const;
+    qint64 encode(QDataStream &input, QDataStream &output);
+    qint64 decode(QDataStream &input, QDataStream &output);
+
+private:
+    int m_frequency;
+};
+
+/// The QXmppG711uCodec class represent a G.711 u-law PCM codec.
+
+class QXmppG711uCodec : public QXmppCodec
+{
+public:
+    QXmppG711uCodec(int clockrate);
+
+    int bitrate() const;
+    qint64 encode(QDataStream &input, QDataStream &output);
+    qint64 decode(QDataStream &input, QDataStream &output);
+
+private:
+    int m_frequency;
+};
+
+#ifdef QXMPP_USE_SPEEX
+typedef struct SpeexBits SpeexBits;
+
+/// The QXmppSpeexCodec class represent a SPEEX codec.
+
+class QXmppSpeexCodec : public QXmppCodec
+{
+public:
+    QXmppSpeexCodec(int clockrate);
+    ~QXmppSpeexCodec();
+
+    int bitrate() const;
+    qint64 encode(QDataStream &input, QDataStream &output);
+    qint64 decode(QDataStream &input, QDataStream &output);
+
+private:
+    SpeexBits *encoder_bits;
+    void *encoder_state;
+    SpeexBits *decoder_bits;
+    void *decoder_state;
+    int frame_samples;
+};
+#endif
+
+#endif
diff --git a/source/source.pro b/source/source.pro
index 712ca506..287bc918 100644
--- a/source/source.pro
+++ b/source/source.pro
@@ -23,6 +23,7 @@ HEADERS += QXmppUtils.h \
     QXmppBind.h \
     QXmppByteStreamIq.h \
     QXmppClient.h \
+    QXmppCodec.h \
     QXmppConfiguration.h \
     QXmppConstants.h \
     QXmppDataForm.h \
@@ -64,6 +65,7 @@ SOURCES += QXmppUtils.cpp \
     QXmppBind.cpp \
     QXmppByteStreamIq.cpp \
     QXmppClient.cpp \
+    QXmppCodec.cpp \
     QXmppConfiguration.cpp \
     QXmppConstants.cpp \
     QXmppDataForm.cpp \