Gabriele Cosmo
134 lines
2.9 KiB
C++
134 lines
2.9 KiB
C++
#ifndef G4STRSTREAM_HH
|
|
#define G4STRSTREAM_HH
|
|
|
|
#include <iostream.h>
|
|
#ifdef WIN32
|
|
#include <Strstrea.h>
|
|
#else
|
|
#include <strstream.h>
|
|
#endif
|
|
#include "globals.hh"
|
|
#include "G4coutDestination.hh"
|
|
|
|
class G4strstreambuf;
|
|
extern G4strstreambuf G4coutbuf;
|
|
extern G4strstreambuf G4cerrbuf;
|
|
#ifndef G4STREAM_STREAMBUF_IMPLEMENTATION
|
|
|
|
|
|
class G4strstreambuf : public strstreambuf {
|
|
public:
|
|
G4strstreambuf() {
|
|
destination = NULL;
|
|
}
|
|
void SetDestination(G4coutDestination * value) {
|
|
destination = value;
|
|
}
|
|
int sync() {
|
|
G4String stringToSend;
|
|
int c;
|
|
int result;
|
|
result = EOF;
|
|
while (( c= sbumpc() ) != EOF) {
|
|
stringToSend += (char) c;
|
|
}
|
|
if(this == & G4coutbuf && destination != NULL) {
|
|
result = destination->ReceiveG4cout(stringToSend);
|
|
} else
|
|
if(this == & G4cerrbuf && destination != NULL) {
|
|
result = destination->ReceiveG4cerr(stringToSend);
|
|
} else
|
|
if(this == & G4coutbuf && destination == NULL) {
|
|
cout << stringToSend << flush;
|
|
result =0;
|
|
} else
|
|
if(this == & G4cerrbuf && destination == NULL) {
|
|
cerr << stringToSend << flush;
|
|
result =0;
|
|
}
|
|
return result;
|
|
};
|
|
|
|
private:
|
|
G4coutDestination *destination;
|
|
};
|
|
|
|
//
|
|
|
|
#else // G4STREAM_STREAMBUF_IMPLEMENTATION :
|
|
|
|
// On NT the upper implementation involves that
|
|
// first character is lost when sending to destination !
|
|
// Below implementation is correct on this platform.
|
|
|
|
|
|
class G4strstreambuf : public streambuf {
|
|
public:
|
|
G4strstreambuf() {
|
|
destination = NULL;
|
|
count = 0;
|
|
size = 127;
|
|
buffer = new char[size+1];
|
|
}
|
|
~G4strstreambuf() {
|
|
delete buffer;
|
|
}
|
|
void SetDestination(G4coutDestination * value) {
|
|
destination = value;
|
|
}
|
|
int overflow(int c=EOF) {
|
|
int result = 0;
|
|
if(count>=size) {
|
|
buffer[count] = '\0';
|
|
count = 0;
|
|
result = ReceiveString ();
|
|
}
|
|
buffer[count] = c;
|
|
count++;
|
|
if(c=='\n') {
|
|
buffer[count] = '\0';
|
|
count = 0;
|
|
result = ReceiveString ();
|
|
}
|
|
return result;
|
|
}
|
|
int sync() {
|
|
buffer[count] = '\0';
|
|
count = 0;
|
|
return ReceiveString ();
|
|
|
|
}
|
|
#ifdef WIN32
|
|
int underflow() {
|
|
return 0;
|
|
}
|
|
#endif
|
|
int ReceiveString () {
|
|
G4String stringToSend = buffer;
|
|
int result;
|
|
if(this == & G4coutbuf && destination != NULL) {
|
|
result = destination->ReceiveG4cout(stringToSend);
|
|
} else if(this == & G4cerrbuf && destination != NULL) {
|
|
result = destination->ReceiveG4cerr(stringToSend);
|
|
} else if(this == & G4coutbuf && destination == NULL) {
|
|
cout << stringToSend << flush;
|
|
result =0;
|
|
} else if(this == & G4cerrbuf && destination == NULL) {
|
|
cerr << stringToSend << flush;
|
|
result =0;
|
|
}
|
|
return result;
|
|
};
|
|
|
|
private:
|
|
G4coutDestination * destination;
|
|
char* buffer;
|
|
int count,size;
|
|
};
|
|
|
|
#endif // G4STREAM_STREAMBUF_IMPLEMENTATION :
|
|
|
|
|
|
|
|
#endif
|