OnlineSpeexEncoder Class Reference

#include <online-speex-wrapper.h>

Collaboration diagram for OnlineSpeexEncoder:

Public Member Functions
	OnlineSpeexEncoder (const SpeexOptions &config)
	~OnlineSpeexEncoder ()
void	AcceptWaveform (int32 sample_rate, const VectorBase< BaseFloat > &waveform)
void	InputFinished ()
void	GetSpeexBits (std::vector< char > *spx_bits)

Private Member Functions
void	Encode (const VectorBase< BaseFloat > &wave, std::vector< char > *speex_encoder_bits)

Private Attributes
int32	speex_frame_size_
int32	speex_encoded_frame_size_
SPEEXBITS	speex_bits_
Vector< BaseFloat >	waveform_remainder_
std::vector< char >	speex_encoded_char_bits_
BaseFloat	sample_rate_
bool	input_finished_

Detailed Description

Definition at line 74 of file online-speex-wrapper.h.

Constructor & Destructor Documentation

OnlineSpeexEncoder()

OnlineSpeexEncoder

(

const SpeexOptions &

config

)

Definition at line 25 of file online-speex-wrapper.cc.

References KALDI_ERR, SpeexOptions::sample_rate, OnlineSpeexEncoder::sample_rate_, OnlineSpeexEncoder::speex_bits_, SpeexOptions::speex_bits_frame_size, OnlineSpeexEncoder::speex_encoded_frame_size_, OnlineSpeexEncoder::speex_frame_size_, SpeexOptions::speex_quality, and SpeexOptions::speex_wave_frame_size.

                                                                :
 input_finished_(false) {
  speex_frame_size_ = config.speex_bits_frame_size;
  speex_encoded_frame_size_ = config.speex_wave_frame_size;
  sample_rate_ = config.sample_rate;

  if(sizeof(speex_bits_) == 1) {
    KALDI_ERR << "OnlineSpeexEncoder called but Speex not installed."
      << "You should run tools/extras/install_speex.sh first, then "
      << "re-run configure in src/ and then make Kaldi again.\n";
  }

#ifdef HAVE_SPEEX
  speex_state_ = speex_encoder_init(&speex_wb_mode);  // init speex with wideband mode
  int32 tmp = config.speex_quality;
  speex_encoder_ctl(speex_state_, SPEEX_SET_QUALITY, &tmp);
  tmp = (int)(sample_rate_);
  speex_encoder_ctl(speex_state_, SPEEX_SET_SAMPLING_RATE, &tmp);
  speex_bits_init(&speex_bits_);
#endif
}

~OnlineSpeexEncoder()

~OnlineSpeexEncoder

(

)

Definition at line 47 of file online-speex-wrapper.cc.

References OnlineSpeexEncoder::speex_bits_.

                                        {
#ifdef HAVE_SPEEX
  speex_bits_destroy(&speex_bits_);
  speex_encoder_destroy(speex_state_);
#endif
}

Member Function Documentation

AcceptWaveform()

void AcceptWaveform	(	int32	sample_rate,
		const VectorBase< BaseFloat > &	waveform
	)

Definition at line 54 of file online-speex-wrapper.cc.

References VectorBase< Real >::Dim(), OnlineSpeexEncoder::Encode(), OnlineSpeexEncoder::input_finished_, KALDI_ERR, VectorBase< Real >::Range(), Vector< Real >::Resize(), OnlineSpeexEncoder::sample_rate_, OnlineSpeexEncoder::speex_encoded_char_bits_, and OnlineSpeexEncoder::waveform_remainder_.

Referenced by main().

                                                           {
  if (waveform.Dim() == 0) {
    return;   // Nothing to do.
  }
  if (input_finished_) {
    KALDI_ERR << "AcceptWaveform called after InputFinished() was called.";
  }
  if (sample_rate != sample_rate_) {
    KALDI_ERR << "Sampling frequency mismatch, expected "
              << sample_rate_ << ", got " << sample_rate;
  }

  Vector<BaseFloat> appended_wave;
  const VectorBase<BaseFloat> &wave_to_use = (waveform_remainder_.Dim() != 0 ?
                                              appended_wave : waveform);
  if (waveform_remainder_.Dim() != 0) {
    appended_wave.Resize(waveform_remainder_.Dim() +
                         waveform.Dim());
    appended_wave.Range(0, waveform_remainder_.Dim()).CopyFromVec(
        waveform_remainder_);
    appended_wave.Range(waveform_remainder_.Dim(),
                        waveform.Dim()).CopyFromVec(waveform);
  }
  waveform_remainder_.Resize(0);

  std::vector<char> spx_bits;
  Encode(wave_to_use, &spx_bits);

  if (spx_bits.size() > 0) {
    speex_encoded_char_bits_.insert(speex_encoded_char_bits_.end(),
                                    spx_bits.begin(), spx_bits.end());
  }
}

Encode()

void Encode ( const VectorBase< BaseFloat > &  wave, std::vector< char > *  speex_encoder_bits  )

private

Definition at line 106 of file online-speex-wrapper.cc.

References VectorBase< Real >::Data(), VectorBase< Real >::Dim(), rnnlm::i, KALDI_ERR, kaldi::kCopyData, OnlineSpeexEncoder::speex_bits_, OnlineSpeexEncoder::speex_encoded_frame_size_, OnlineSpeexEncoder::speex_frame_size_, and OnlineSpeexEncoder::waveform_remainder_.

Referenced by OnlineSpeexEncoder::AcceptWaveform(), and OnlineSpeexEncoder::InputFinished().

                                      {
  if (wave.Dim() == 0) {
    return;
  }

  int32 to_encode = wave.Dim();
  int32 has_encode = 0;
  char cbits[200];
  std::vector<char> encoded_bits;
  while (to_encode > speex_encoded_frame_size_) {
    SubVector<BaseFloat> wave_frame(wave, has_encode,
      speex_encoded_frame_size_);
    int32 nbytes = 0;
#ifdef HAVE_SPEEX
    speex_bits_reset(&speex_bits_);
    speex_encode(speex_state_, wave_frame.Data(), &speex_bits_);
    nbytes = speex_bits_nbytes(&speex_bits_);
    if (nbytes != speex_frame_size_) {
      KALDI_ERR << "The number of bytes of Speex encoded frame mismatch,"
        << "expected " << speex_frame_size_ << ", got " << nbytes;
    }
    nbytes = speex_bits_write(&speex_bits_, cbits, 200);
#endif

    int32 encoded_bits_len = encoded_bits.size();
    encoded_bits.resize(encoded_bits_len + nbytes);
    for (int32 i = 0; i < nbytes; i++) {
      encoded_bits[i+encoded_bits_len] = cbits[i];
    }

    has_encode += speex_encoded_frame_size_;
    to_encode -= speex_encoded_frame_size_;
  }

  if (to_encode > 0) {
    SubVector<BaseFloat> wave_left(wave, has_encode, to_encode);
    int32 dim = waveform_remainder_.Dim();
    if (dim != 0) {
      waveform_remainder_.Resize(dim + to_encode, kCopyData);
      waveform_remainder_.Range(dim, to_encode).CopyFromVec(wave_left);
    } else {
      waveform_remainder_ = wave_left;
    }
  }

  *speex_encoder_bits = encoded_bits;
}

GetSpeexBits()

void GetSpeexBits ( std::vector< char > *  spx_bits )

inline

Definition at line 84 of file online-speex-wrapper.h.

Referenced by main().

                                                 {  // call it after AcceptWaveform
      *spx_bits = speex_encoded_char_bits_;
      speex_encoded_char_bits_.clear();
    }

InputFinished()

void InputFinished

(

)

Definition at line 90 of file online-speex-wrapper.cc.

References OnlineSpeexEncoder::Encode(), OnlineSpeexEncoder::input_finished_, KALDI_ASSERT, VectorBase< Real >::Range(), OnlineSpeexEncoder::speex_encoded_char_bits_, OnlineSpeexEncoder::speex_encoded_frame_size_, and OnlineSpeexEncoder::waveform_remainder_.

Referenced by main().

                                       {
  input_finished_ = true;

  int32 dim = waveform_remainder_.Dim();
  if (dim != 0) {
    KALDI_ASSERT(dim <= speex_encoded_frame_size_);
    Vector<BaseFloat> wave_last(speex_encoded_frame_size_);
    std::vector<char> spx_bits;
    wave_last.Range(0, dim).CopyFromVec(waveform_remainder_);
    Encode(wave_last, &spx_bits);

    speex_encoded_char_bits_.insert(speex_encoded_char_bits_.end(),
                                    spx_bits.begin(), spx_bits.end());
  }
}

Member Data Documentation

input_finished_

bool input_finished_

private

Definition at line 104 of file online-speex-wrapper.h.

Referenced by OnlineSpeexEncoder::AcceptWaveform(), and OnlineSpeexEncoder::InputFinished().

sample_rate_

BaseFloat sample_rate_

private

Definition at line 103 of file online-speex-wrapper.h.

Referenced by OnlineSpeexEncoder::AcceptWaveform(), and OnlineSpeexEncoder::OnlineSpeexEncoder().

speex_bits_

SPEEXBITS speex_bits_

private

Definition at line 94 of file online-speex-wrapper.h.

Referenced by OnlineSpeexDecoder::Decode(), OnlineSpeexEncoder::Encode(), OnlineSpeexDecoder::OnlineSpeexDecoder(), OnlineSpeexEncoder::OnlineSpeexEncoder(), OnlineSpeexDecoder::~OnlineSpeexDecoder(), and OnlineSpeexEncoder::~OnlineSpeexEncoder().

speex_encoded_char_bits_

std::vector<char> speex_encoded_char_bits_

private

Definition at line 101 of file online-speex-wrapper.h.

Referenced by OnlineSpeexEncoder::AcceptWaveform(), and OnlineSpeexEncoder::InputFinished().

speex_encoded_frame_size_

int32 speex_encoded_frame_size_

private

Definition at line 90 of file online-speex-wrapper.h.

Referenced by OnlineSpeexEncoder::Encode(), OnlineSpeexEncoder::InputFinished(), and OnlineSpeexEncoder::OnlineSpeexEncoder().

speex_frame_size_

int32 speex_frame_size_

private

Definition at line 89 of file online-speex-wrapper.h.

Referenced by OnlineSpeexDecoder::Decode(), OnlineSpeexEncoder::Encode(), OnlineSpeexDecoder::OnlineSpeexDecoder(), and OnlineSpeexEncoder::OnlineSpeexEncoder().

waveform_remainder_

Vector<BaseFloat> waveform_remainder_

private

Definition at line 96 of file online-speex-wrapper.h.

Referenced by OnlineSpeexEncoder::AcceptWaveform(), OnlineSpeexEncoder::Encode(), and OnlineSpeexEncoder::InputFinished().

---

The documentation for this class was generated from the following files:

• online2/online-speex-wrapper.h
• online2/online-speex-wrapper.cc