OnlineTcpVectorSource Class Reference

#include <online-tcp-source.h>

Inheritance diagram for OnlineTcpVectorSource:

Collaboration diagram for OnlineTcpVectorSource:

Public Member Functions
	OnlineTcpVectorSource (int32 socket)
	~OnlineTcpVectorSource ()
bool	Read (Vector< BaseFloat > *data)
bool	IsConnected ()
size_t	SamplesProcessed ()
void	ResetSamples ()
Public Member Functions inherited from OnlineAudioSourceItf
virtual	~OnlineAudioSourceItf ()

Private Member Functions
bool	ReadFull (char *buf, int32 len)
int32	GetNextPack ()
int32	FillFrame (int32 size)
	KALDI_DISALLOW_COPY_AND_ASSIGN (OnlineTcpVectorSource)

Private Attributes
int32	socket_desc
bool	connected
char *	pack
int32	pack_size
char *	frame
int32	frame_size
int32	last_pack_size
int32	last_pack_rem
size_t	samples_processed

Detailed Description

Definition at line 34 of file online-tcp-source.h.

Constructor & Destructor Documentation

OnlineTcpVectorSource()

OnlineTcpVectorSource

(

int32

socket

)

Definition at line 29 of file online-tcp-source.cc.

References OnlineTcpVectorSource::frame, OnlineTcpVectorSource::frame_size, OnlineTcpVectorSource::pack, and OnlineTcpVectorSource::pack_size.

    : socket_desc(socket),
      connected(true),
      pack_size(512),
      frame_size(512),
      last_pack_size(0),
      last_pack_rem(0),
      samples_processed(0) {
  pack = new char[pack_size];
  frame = new char[frame_size];
}

~OnlineTcpVectorSource()

~OnlineTcpVectorSource

(

)

Definition at line 41 of file online-tcp-source.cc.

References OnlineTcpVectorSource::frame, and OnlineTcpVectorSource::pack.

                                              {
  delete[] pack;
  delete[] frame;
}

Member Function Documentation

FillFrame()

int FillFrame ( int32  size )

private

Definition at line 93 of file online-tcp-source.cc.

References OnlineTcpVectorSource::frame, OnlineTcpVectorSource::GetNextPack(), OnlineTcpVectorSource::last_pack_rem, OnlineTcpVectorSource::last_pack_size, and OnlineTcpVectorSource::pack.

Referenced by OnlineTcpVectorSource::Read().

                                                   {
  int32 frame_offset = 0;
  if (last_pack_rem > 0) {
    int pack_offset = last_pack_size - last_pack_rem;
    int size = last_pack_rem < get_size ? last_pack_rem : get_size;

    memcpy(frame, pack + pack_offset, size);

    last_pack_rem -= size;
    get_size -= size;
    frame_offset += size;
  }

  while (get_size > 0) {
    int32 ret = GetNextPack();

    if (ret == 0)
      return frame_offset;

    int32 size = ret < get_size ? ret : get_size;

    memcpy(frame + frame_offset, pack, size);

    last_pack_size = ret;
    last_pack_rem = last_pack_size - size;
    get_size -= size;
    frame_offset += size;
  }

  return frame_offset;
}

GetNextPack()

int GetNextPack ( )

private

Definition at line 71 of file online-tcp-source.cc.

References KALDI_ERR, OnlineTcpVectorSource::pack, OnlineTcpVectorSource::pack_size, and OnlineTcpVectorSource::ReadFull().

Referenced by OnlineTcpVectorSource::FillFrame().

                                       {
  int32 size = 0;
  if (!ReadFull((char*) &size, 4))
    return 0;

  if (size % 2 != 0) {
    KALDI_ERR << "TCPVectorSource: Pack size must be even!";
    return 0;
  }

  if (pack_size < size) {
    pack_size = size;
    delete[] pack;
    pack = new char[pack_size];
  }

  if (!ReadFull(pack, size))
    return 0;

  return size;
}

IsConnected()

bool IsConnected

(

)

Definition at line 151 of file online-tcp-source.cc.

References OnlineTcpVectorSource::connected.

Referenced by main().

                                        {
  return connected;
}

KALDI_DISALLOW_COPY_AND_ASSIGN()

KALDI_DISALLOW_COPY_AND_ASSIGN ( OnlineTcpVectorSource  )

private

Read()

bool Read ( Vector< BaseFloat > *  data )

virtual

Implements OnlineAudioSourceItf.

Definition at line 125 of file online-tcp-source.cc.

References OnlineTcpVectorSource::connected, VectorBase< Real >::Dim(), OnlineTcpVectorSource::FillFrame(), OnlineTcpVectorSource::frame, OnlineTcpVectorSource::frame_size, rnnlm::i, and OnlineTcpVectorSource::samples_processed.

                                                        {
  if (!connected)
    return false;

  int32 n_elem = static_cast<uint32>(data->Dim());

  int32 n_bytes = n_elem * 2;

  if (frame_size < n_bytes) {
    frame_size = n_bytes;
    delete[] frame;
    frame = new char[frame_size];
  }

  int32 b_read = FillFrame(n_bytes);
  int32 n_read = b_read / 2;

  short* s_frame = (short*) frame;
  for (int32 i = 0; i < n_read; i++)
    (*data)(i) = s_frame[i];

  samples_processed += n_read;

  return (n_read == n_elem);
}

ReadFull()

bool ReadFull ( char *  buf, int32  len  )

private

Definition at line 53 of file online-tcp-source.cc.

References OnlineTcpVectorSource::connected, and OnlineTcpVectorSource::socket_desc.

Referenced by OnlineTcpVectorSource::GetNextPack().

                                                         {
  int32 to_read = len;
  int32 has_read = 0;
  int32 ret;

  while (to_read > 0) {
    ret = read(socket_desc, buf + has_read, to_read);
    if (ret <= 0) {
      connected = false;
      return false;
    }
    to_read -= ret;
    has_read += ret;
  }

  return true;
}

ResetSamples()

void ResetSamples

(

)

Definition at line 49 of file online-tcp-source.cc.

References OnlineTcpVectorSource::samples_processed.

Referenced by main().

                                         {
  samples_processed = 0;
}

SamplesProcessed()

size_t SamplesProcessed

(

)

Definition at line 46 of file online-tcp-source.cc.

References OnlineTcpVectorSource::samples_processed.

Referenced by main().

                                               {
  return samples_processed;
}

Member Data Documentation

connected

bool connected

private

Definition at line 52 of file online-tcp-source.h.

Referenced by OnlineTcpVectorSource::IsConnected(), OnlineTcpVectorSource::Read(), and OnlineTcpVectorSource::ReadFull().

frame

char* frame

private

Definition at line 55 of file online-tcp-source.h.

Referenced by OnlineTcpVectorSource::FillFrame(), OnlineTcpVectorSource::OnlineTcpVectorSource(), OnlineTcpVectorSource::Read(), and OnlineTcpVectorSource::~OnlineTcpVectorSource().

frame_size

int32 frame_size

private

Definition at line 56 of file online-tcp-source.h.

Referenced by OnlineTcpVectorSource::OnlineTcpVectorSource(), and OnlineTcpVectorSource::Read().

last_pack_rem

int32 last_pack_rem

private

Definition at line 58 of file online-tcp-source.h.

Referenced by OnlineTcpVectorSource::FillFrame().

last_pack_size

int32 last_pack_size

private

Definition at line 58 of file online-tcp-source.h.

Referenced by OnlineTcpVectorSource::FillFrame().

pack

char* pack

private

Definition at line 53 of file online-tcp-source.h.

Referenced by OnlineTcpVectorSource::FillFrame(), OnlineTcpVectorSource::GetNextPack(), OnlineTcpVectorSource::OnlineTcpVectorSource(), and OnlineTcpVectorSource::~OnlineTcpVectorSource().

pack_size

int32 pack_size

private

Definition at line 54 of file online-tcp-source.h.

Referenced by OnlineTcpVectorSource::GetNextPack(), and OnlineTcpVectorSource::OnlineTcpVectorSource().

samples_processed

size_t samples_processed

private

Definition at line 60 of file online-tcp-source.h.

Referenced by OnlineTcpVectorSource::Read(), OnlineTcpVectorSource::ResetSamples(), and OnlineTcpVectorSource::SamplesProcessed().

socket_desc

int32 socket_desc

private

Definition at line 51 of file online-tcp-source.h.

Referenced by OnlineTcpVectorSource::ReadFull().

---

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

• online/online-tcp-source.h
• online/online-tcp-source.cc