blitz-research
161 lines
3.2 KiB
C++
161 lines
3.2 KiB
C++
|
|
#include "std.h"
|
|
#include "bbstream.h"
|
|
|
|
static set<bbStream*> stream_set;
|
|
|
|
void debugStream( bbStream *s ){
|
|
if( stream_set.count(s) ) return;
|
|
RTEX( "Stream does not exist" );
|
|
}
|
|
|
|
bbStream::bbStream(){
|
|
stream_set.insert( this );
|
|
}
|
|
|
|
bbStream::~bbStream(){
|
|
stream_set.erase( this );
|
|
}
|
|
|
|
int bbEof( bbStream *s ){
|
|
if( debug ) debugStream( s );
|
|
return s->eof();
|
|
}
|
|
|
|
int bbReadAvail( bbStream *s ){
|
|
if( debug ) debugStream( s );
|
|
return s->avail();
|
|
}
|
|
|
|
int bbReadByte( bbStream *s ){
|
|
if( debug ) debugStream( s );
|
|
int n=0;
|
|
s->read( (char*)&n,1 );
|
|
return n;
|
|
}
|
|
|
|
int bbReadShort( bbStream *s ){
|
|
if( debug ) debugStream( s );
|
|
int n=0;
|
|
s->read( (char*)&n,2 );
|
|
return n;
|
|
}
|
|
|
|
int bbReadInt( bbStream *s ){
|
|
if( debug ) debugStream( s );
|
|
int n=0;
|
|
s->read( (char*)&n,4 );
|
|
return n;
|
|
}
|
|
|
|
float bbReadFloat( bbStream *s ){
|
|
if( debug ) debugStream( s );
|
|
float n=0;
|
|
s->read( (char*)&n,4 );
|
|
return n;
|
|
}
|
|
|
|
BBStr *bbReadString( bbStream *s ){
|
|
if( debug ) debugStream( s );
|
|
int len;
|
|
BBStr *str=d_new BBStr();
|
|
if( s->read( (char*)&len,4 ) ){
|
|
char *buff=d_new char[len];
|
|
if( s->read( buff,len ) ){
|
|
*str=string( buff,len );
|
|
}
|
|
delete[] buff;
|
|
}
|
|
return str;
|
|
}
|
|
|
|
BBStr *bbReadLine( bbStream *s ){
|
|
if( debug ) debugStream( s );
|
|
unsigned char c;
|
|
BBStr *str=d_new BBStr();
|
|
for(;;){
|
|
if( s->read( (char*)&c,1 )!=1 ) break;
|
|
if( c=='\n' ) break;
|
|
if( c!='\r' ) *str+=c;
|
|
}
|
|
return str;
|
|
}
|
|
|
|
void bbWriteByte( bbStream *s,int n ){
|
|
if( debug ) debugStream( s );
|
|
s->write( (char*)&n,1 );
|
|
}
|
|
|
|
void bbWriteShort( bbStream *s,int n ){
|
|
if( debug ) debugStream( s );
|
|
s->write( (char*)&n,2 );
|
|
}
|
|
|
|
void bbWriteInt( bbStream *s,int n ){
|
|
if( debug ) debugStream( s );
|
|
s->write( (char*)&n,4 );
|
|
}
|
|
|
|
void bbWriteFloat( bbStream *s,float n ){
|
|
if( debug ) debugStream( s );
|
|
s->write( (char*)&n,4 );
|
|
}
|
|
|
|
void bbWriteString( bbStream *s,BBStr *t ){
|
|
if( debug ) debugStream( s );
|
|
int n=t->size();
|
|
s->write( (char*)&n,4 );
|
|
s->write( t->data(),t->size() );
|
|
delete t;
|
|
}
|
|
|
|
void bbWriteLine( bbStream *s,BBStr *t ){
|
|
if( debug ) debugStream( s );
|
|
s->write( t->data(),t->size() );
|
|
s->write( "\r\n",2 );
|
|
delete t;
|
|
}
|
|
|
|
void bbCopyStream( bbStream *s,bbStream *d,int buff_size ){
|
|
if( debug ){
|
|
debugStream( s );debugStream( d );
|
|
if( buff_size<1 || buff_size>1024*1024 ) RTEX( "Illegal buffer size" );
|
|
}
|
|
char *buff=d_new char[buff_size];
|
|
while( s->eof()==0 && d->eof()==0 ){
|
|
int n=s->read( buff,buff_size );
|
|
d->write( buff,n );
|
|
if( n<buff_size ) break;
|
|
}
|
|
delete buff;
|
|
}
|
|
|
|
bool stream_create(){
|
|
return true;
|
|
}
|
|
|
|
bool stream_destroy(){
|
|
return true;
|
|
}
|
|
|
|
void stream_link( void(*rtSym)(const char*,void*) ){
|
|
rtSym( "%Eof%stream",bbEof );
|
|
rtSym( "%ReadAvail%stream",bbReadAvail );
|
|
rtSym( "%ReadByte%stream",bbReadByte );
|
|
rtSym( "%ReadShort%stream",bbReadShort );
|
|
rtSym( "%ReadInt%stream",bbReadInt );
|
|
rtSym( "#ReadFloat%stream",bbReadFloat );
|
|
rtSym( "$ReadString%stream",bbReadString );
|
|
rtSym( "$ReadLine%stream",bbReadLine );
|
|
rtSym( "WriteByte%stream%byte",bbWriteByte );
|
|
rtSym( "WriteShort%stream%short",bbWriteShort );
|
|
rtSym( "WriteInt%stream%int",bbWriteInt );
|
|
rtSym( "WriteFloat%stream#float",bbWriteFloat );
|
|
rtSym( "WriteString%stream$string",bbWriteString );
|
|
rtSym( "WriteLine%stream$string",bbWriteLine );
|
|
rtSym( "CopyStream%src_stream%dest_stream%buffer_size=16384",bbCopyStream );
|
|
}
|
|
|
|
|
|
|