Search  

---

C:/temp/src/j2k/Net/TFTP/DataFactory.cpp

Go to the documentation of this file.

#ifndef __J2K__Data_Factory_CPP__
#define __J2K__Data_Factory_CPP__

#include <j2k/Net/TFTP/DataFactory.hpp>
// Creates packets from the file. 

DataFactory::DataFactory() 
 : buffer( NULL ), packet( NULL ), blocksize( 0 ), 
   totalSize( 0 ), dummySended( FALSE ),
   data( NULL ), blockNo( 0 ), sz( 0 ), neof( 1 ), 
   initialized( FALSE ),
   Basic_PThread()
{
  option.rwlock5_data_queue.write();
  option.data_queue = new DList_pPacket();
  option.rwlock5_data_queue.releaseWrite();
  blocksize = option.getBlockSizeValue();
  printf( "DataFactory Created.\n");
  fflush( stdout );
}

void DataFactory::init()
{
  if ( initialized == TRUE ) return;

  initialized = TRUE;
  blocksize = option.getBlockSizeValue();
  sz   = blocksize;
  neof = 1;

  option.rwlock2_file.read();
  option.myFile->setBegin();
  option.rwlock2_file.releaseRead();

  buffer = new char[ blocksize + 1 ];
}

void DataFactory::run() 
{
  init();

  while( sz >= blocksize && neof > -1 )
  {
    createDataPacket();
    if ( blockNo % 2 == 0 ) sched_yield();
  }

  // Save Last
  option.LastData = blockNo;
  option.Done     = FALSE;

  option.rwlock5_data_queue.write();
#ifdef DEBUG
  option.data_queue->display();
#endif
  option.rwlock5_data_queue.releaseWrite();

  printf( "DataFactory Finished.Should send %d bytes\n",totalSize);
  
  fflush( stdout );
}

void DataFactory::createDataPacket()
{
  blockNo++;    

  // Read data
  option.rwlock2_file.read();
  sz = option.myFile->getData( buffer, blocksize );

  if ( sz < 0 ) {
    neof = -1;
    sz  *= -1;
  }

  if ( sz > 0 && buffer[sz-1] == '\0' ) sz--;
  totalSize += sz;
#ifdef DEBUG
  printf("DataFact:[%d|%d|%s|%d]", neof, sz, buffer+4, blocksize );
#endif
  option.rwlock2_file.releaseRead();

  printf( "enqueuing data\n" );
  fflush( stdout );

  packet = new TFTP_Packet();
  packet->createData( blockNo, buffer, sz );

  option.rwlock5_data_queue.write();
  option.data_queue->enqueue( packet );
  option.rwlock5_data_queue.releaseWrite();

  if (  (neof < 0)  && (( totalSize % blocksize ) == 0)  )
  {
    dummyPacket();
  }

}

void DataFactory::dummyPacket() 
{
    if ( dummySended ) return;
    dummySended = TRUE;

    packet = new TFTP_Packet();
    memset( buffer, 0 , sz );

    blockNo++;
    packet->createData( blockNo, buffer, 0 );
   
    option.rwlock5_data_queue.write();
    option.data_queue->enqueue( packet ); 
    option.rwlock5_data_queue.releaseWrite();  
} 

#endif

---