]> git.saurik.com Git - wxWidgets.git/blob - src/common/stream.cpp
fix for toolbar radio buttons event generation
[wxWidgets.git] / src / common / stream.cpp
1 /////////////////////////////////////////////////////////////////////////////
2 // Name: src/common/stream.cpp
3 // Purpose: wxStream base classes
4 // Author: Guilhem Lavaux
5 // Modified by: VZ (23.11.00) to fix realloc()ing new[]ed memory,
6 // general code review
7 // Created: 11/07/98
8 // RCS-ID: $Id$
9 // Copyright: (c) Guilhem Lavaux
10 // Licence: wxWindows license
11 /////////////////////////////////////////////////////////////////////////////
12
13 // ============================================================================
14 // declarations
15 // ============================================================================
16
17 // ----------------------------------------------------------------------------
18 // headers
19 // ----------------------------------------------------------------------------
20
21 #ifdef __GNUG__
22 #pragma implementation "stream.h"
23 #endif
24
25 // For compilers that support precompilation, includes "wx.h".
26 #include "wx/wxprec.h"
27
28 #ifdef __BORLANDC__
29 #pragma hdrstop
30 #endif
31
32 #ifndef WX_PRECOMP
33 #include "wx/defs.h"
34 #endif
35
36 #if wxUSE_STREAMS
37
38 #include <ctype.h>
39 #include "wx/stream.h"
40 #include "wx/datstrm.h"
41 #include "wx/textfile.h"
42 #include "wx/log.h"
43
44 // ----------------------------------------------------------------------------
45 // constants
46 // ----------------------------------------------------------------------------
47
48 // the temporary buffer size used when copying from stream to stream
49 #define BUF_TEMP_SIZE 10000
50
51 // ============================================================================
52 // implementation
53 // ============================================================================
54
55 // ----------------------------------------------------------------------------
56 // wxStreamBuffer
57 // ----------------------------------------------------------------------------
58
59 void wxStreamBuffer::SetError(wxStreamError err)
60 {
61 if ( m_stream->m_lasterror == wxStream_NOERROR )
62 m_stream->m_lasterror = err;
63 }
64
65 void wxStreamBuffer::InitBuffer()
66 {
67 m_buffer_start =
68 m_buffer_end =
69 m_buffer_pos = NULL;
70 m_buffer_size = 0;
71
72 // if we are going to allocate the buffer, we should free it later as well
73 m_destroybuf = TRUE;
74 }
75
76 void wxStreamBuffer::Init()
77 {
78 InitBuffer();
79
80 m_fixed = TRUE;
81 }
82
83 wxStreamBuffer::wxStreamBuffer(wxStreamBase& stream, BufMode mode)
84 {
85 Init();
86
87 m_stream = &stream;
88 m_mode = mode;
89
90 m_flushable = TRUE;
91 m_destroystream = FALSE;
92 }
93
94 wxStreamBuffer::wxStreamBuffer(BufMode mode)
95 {
96 Init();
97
98 wxASSERT_MSG(mode != read_write, wxT("you have to use the other ctor for read_write mode") );
99 if ( mode == read )
100 m_stream = new wxInputStream;
101 else if ( mode == write)
102 m_stream = new wxOutputStream;
103 else
104 m_stream = NULL;
105
106 m_mode = mode;
107
108 m_flushable = FALSE;
109 m_destroystream = TRUE;
110 }
111
112 wxStreamBuffer::wxStreamBuffer(const wxStreamBuffer& buffer)
113 {
114 // doing this has big chances to lead to a crashwhen the source buffer is
115 // destroyed (otherwise assume the caller knows what he does)
116 wxASSERT_MSG( !buffer.m_destroybuf && !buffer.m_destroystream,
117 _T("it's a bad idea to copy this buffer") );
118
119 m_buffer_start = buffer.m_buffer_start;
120 m_buffer_end = buffer.m_buffer_end;
121 m_buffer_pos = buffer.m_buffer_pos;
122 m_buffer_size = buffer.m_buffer_size;
123 m_fixed = buffer.m_fixed;
124 m_flushable = buffer.m_flushable;
125 m_stream = buffer.m_stream;
126 m_mode = buffer.m_mode;
127 m_destroybuf = FALSE;
128 m_destroystream = FALSE;
129 }
130
131 void wxStreamBuffer::FreeBuffer()
132 {
133 if ( m_destroybuf )
134 free(m_buffer_start);
135 }
136
137 wxStreamBuffer::~wxStreamBuffer()
138 {
139 FreeBuffer();
140
141 if ( m_destroystream )
142 delete m_stream;
143 }
144
145 wxInputStream *wxStreamBuffer::GetInputStream() const
146 {
147 return m_mode == write ? NULL : (wxInputStream *)m_stream;
148 }
149
150 wxOutputStream *wxStreamBuffer::GetOutputStream() const
151 {
152 return m_mode == read ? NULL : (wxOutputStream *)m_stream;
153 }
154
155 void wxStreamBuffer::SetBufferIO(void *buffer_start,
156 void *buffer_end,
157 bool takeOwnership)
158 {
159 SetBufferIO(buffer_start, (char *)buffer_end - (char *)buffer_start,
160 takeOwnership);
161 }
162
163 void wxStreamBuffer::SetBufferIO(void *start,
164 size_t len,
165 bool takeOwnership)
166 {
167 // start by freeing the old buffer
168 FreeBuffer();
169
170 m_buffer_start = (char *)start;
171 m_buffer_end = m_buffer_start + len;
172
173 m_buffer_size = len;
174
175 // if we own it, we free it
176 m_destroybuf = takeOwnership;
177
178 ResetBuffer();
179 }
180
181 void wxStreamBuffer::SetBufferIO(size_t bufsize)
182 {
183 // start by freeing the old buffer
184 FreeBuffer();
185
186 if ( bufsize )
187 {
188 SetBufferIO(malloc(bufsize), bufsize, TRUE /* take ownership */);
189 }
190 else // no buffer size => no buffer
191 {
192 InitBuffer();
193 }
194 }
195
196 void wxStreamBuffer::ResetBuffer()
197 {
198 wxCHECK_RET( m_stream, _T("should have a stream in wxStreamBuffer") );
199
200 m_stream->m_lasterror = wxStream_NOERROR;
201 m_stream->m_lastcount = 0;
202 if (m_mode == read && m_flushable)
203 m_buffer_pos = m_buffer_end;
204 else
205 m_buffer_pos = m_buffer_start;
206 }
207
208 // fill the buffer with as much data as possible (only for read buffers)
209 bool wxStreamBuffer::FillBuffer()
210 {
211 wxInputStream *inStream = GetInputStream();
212
213 wxCHECK_MSG( inStream, FALSE, _T("should have a stream in wxStreamBuffer") );
214
215 size_t count = inStream->OnSysRead(m_buffer_start, m_buffer_size);
216 if ( !count )
217 return FALSE;
218
219 m_buffer_end = m_buffer_start + count;
220 m_buffer_pos = m_buffer_start;
221
222 return TRUE;
223 }
224
225 // write the buffer contents to the stream (only for write buffers)
226 bool wxStreamBuffer::FlushBuffer()
227 {
228 wxCHECK_MSG( m_flushable, FALSE, _T("can't flush this buffer") );
229
230 // FIXME: what is this check for? (VZ)
231 if ( m_buffer_pos == m_buffer_start )
232 return FALSE;
233
234 wxOutputStream *outStream = GetOutputStream();
235
236 wxCHECK_MSG( outStream, FALSE, _T("should have a stream in wxStreamBuffer") );
237
238 size_t current = m_buffer_pos - m_buffer_start;
239 size_t count = outStream->OnSysWrite(m_buffer_start, current);
240 if ( count != current )
241 return FALSE;
242
243 m_buffer_pos = m_buffer_start;
244
245 return TRUE;
246 }
247
248 size_t wxStreamBuffer::GetDataLeft()
249 {
250 /* Why is this done? RR. */
251 if ( m_buffer_pos == m_buffer_end && m_flushable)
252 FillBuffer();
253
254 return GetBytesLeft();
255 }
256
257 // copy up to size bytes from our buffer into the provided one
258 void wxStreamBuffer::GetFromBuffer(void *buffer, size_t size)
259 {
260 // don't get more bytes than left in the buffer
261 size_t left = GetBytesLeft();
262
263 if ( size > left )
264 size = left;
265
266 memcpy(buffer, m_buffer_pos, size);
267 m_buffer_pos += size;
268 }
269
270 // copy the contents of the provided buffer into this one
271 void wxStreamBuffer::PutToBuffer(const void *buffer, size_t size)
272 {
273 size_t left = GetBytesLeft();
274 if ( size > left )
275 {
276 if ( m_fixed )
277 {
278 // we can't realloc the buffer, so just copy what we can
279 size = left;
280 }
281 else // !m_fixed
282 {
283 // realloc the buffer to have enough space for the data
284 size_t delta = m_buffer_pos - m_buffer_start;
285
286 char *startOld = m_buffer_start;
287 m_buffer_size += size;
288 m_buffer_start = (char *)realloc(m_buffer_start, m_buffer_size);
289 if ( !m_buffer_start )
290 {
291 // don't leak memory if realloc() failed
292 m_buffer_start = startOld;
293 m_buffer_size -= size;
294
295 // what else can we do?
296 return;
297 }
298
299 // adjust the pointers invalidated by realloc()
300 m_buffer_pos = m_buffer_start + delta;
301 m_buffer_end = m_buffer_start + m_buffer_size;
302 }
303 }
304
305 memcpy(m_buffer_pos, buffer, size);
306 m_buffer_pos += size;
307 }
308
309 void wxStreamBuffer::PutChar(char c)
310 {
311 wxOutputStream *outStream = GetOutputStream();
312
313 wxCHECK_RET( outStream, _T("should have a stream in wxStreamBuffer") );
314
315 // if we don't have buffer at all, just forward this call to the stream,
316 if ( !HasBuffer() )
317 {
318 outStream->OnSysWrite(&c, 1);
319 }
320 else
321 {
322 // otherwise check we have enough space left
323 if ( !GetDataLeft() && !FlushBuffer() )
324 {
325 // we don't
326 SetError(wxStream_WRITE_ERR);
327 }
328 else
329 {
330 PutToBuffer(&c, 1);
331 m_stream->m_lastcount = 1;
332 }
333 }
334 }
335
336 char wxStreamBuffer::Peek()
337 {
338 wxCHECK_MSG( m_stream && HasBuffer(), 0,
339 _T("should have the stream and the buffer in wxStreamBuffer") );
340
341 if ( !GetDataLeft() )
342 {
343 SetError(wxStream_READ_ERR);
344 return 0;
345 }
346
347 char c;
348 GetFromBuffer(&c, 1);
349 m_buffer_pos--;
350
351 return c;
352 }
353
354 char wxStreamBuffer::GetChar()
355 {
356 wxInputStream *inStream = GetInputStream();
357
358 wxCHECK_MSG( inStream, 0, _T("should have a stream in wxStreamBuffer") );
359
360 char c;
361 if ( !HasBuffer() )
362 {
363 inStream->OnSysRead(&c, 1);
364 }
365 else
366 {
367 if ( !GetDataLeft() )
368 {
369 SetError(wxStream_READ_ERR);
370 c = 0;
371 }
372 else
373 {
374 GetFromBuffer(&c, 1);
375 m_stream->m_lastcount = 1;
376 }
377 }
378
379 return c;
380 }
381
382 size_t wxStreamBuffer::Read(void *buffer, size_t size)
383 {
384 wxInputStream *inStream = GetInputStream();
385
386 wxCHECK_MSG( inStream, 0, _T("should have a stream in wxStreamBuffer") );
387
388 // lasterror is reset before all new IO calls
389 m_stream->m_lasterror = wxStream_NOERROR;
390
391 if ( !HasBuffer() )
392 {
393 m_stream->m_lastcount = inStream->OnSysRead(buffer, size);
394 }
395 else // we have a buffer, use it
396 {
397 size_t orig_size = size;
398
399 while ( size > 0 )
400 {
401 size_t left = GetDataLeft();
402
403 // if the requested number of bytes if greater than the buffer
404 // size, read data in chunks
405 if ( size > left )
406 {
407 GetFromBuffer(buffer, left);
408 size -= left;
409 buffer = (char *)buffer + left;
410
411 if ( !FillBuffer() )
412 {
413 SetError(wxStream_EOF);
414 break;
415 }
416 }
417 else // otherwise just do it in one gulp
418 {
419 GetFromBuffer(buffer, size);
420 size = 0;
421 }
422 }
423
424 m_stream->m_lastcount = orig_size - size;
425 }
426
427 return m_stream->m_lastcount;
428 }
429
430 // this should really be called "Copy()"
431 size_t wxStreamBuffer::Read(wxStreamBuffer *dbuf)
432 {
433 wxCHECK_MSG( m_mode != write, 0, _T("can't read from this buffer") );
434
435 char buf[BUF_TEMP_SIZE];
436 size_t nRead,
437 total = 0;
438
439 do
440 {
441 nRead = Read(dbuf, WXSIZEOF(buf));
442 if ( nRead )
443 {
444 nRead = dbuf->Write(buf, nRead);
445 total += nRead;
446 }
447 }
448 while ( nRead );
449
450 return total;
451 }
452
453 size_t wxStreamBuffer::Write(const void *buffer, size_t size)
454 {
455 wxOutputStream *outStream = GetOutputStream();
456
457 wxCHECK_MSG( outStream, 0, _T("should have a stream in wxStreamBuffer") );
458
459 // lasterror is reset before all new IO calls
460 m_stream->m_lasterror = wxStream_NOERROR;
461
462 if ( !HasBuffer() && m_fixed )
463 {
464 // no buffer, just forward the call to the stream
465 m_stream->m_lastcount = outStream->OnSysWrite(buffer, size);
466 }
467 else // we [may] have a buffer, use it
468 {
469 size_t orig_size = size;
470
471 while ( size > 0 )
472 {
473 size_t left = GetBytesLeft();
474
475 // if the buffer is too large to fit in the stream buffer, split
476 // it in smaller parts
477 //
478 // NB: If stream buffer isn't fixed (as for wxMemoryOutputStream),
479 // we always go to the second case.
480 //
481 // FIXME: fine, but if it fails we should (re)try writing it by
482 // chunks as this will (hopefully) always work (VZ)
483 if ( size > left && m_fixed )
484 {
485 PutToBuffer(buffer, left);
486 size -= left;
487 buffer = (char *)buffer + left;
488
489 if ( !FlushBuffer() )
490 {
491 SetError(wxStream_WRITE_ERR);
492
493 break;
494 }
495
496 m_buffer_pos = m_buffer_start;
497 }
498 else // we can do it in one gulp
499 {
500 PutToBuffer(buffer, size);
501 size = 0;
502 }
503 }
504
505 m_stream->m_lastcount = orig_size - size;
506 }
507
508 return m_stream->m_lastcount;
509 }
510
511 size_t wxStreamBuffer::Write(wxStreamBuffer *sbuf)
512 {
513 wxCHECK_MSG( m_mode != read, 0, _T("can't write to this buffer") );
514 wxCHECK_MSG( sbuf->m_mode != write, 0, _T("can't read from that buffer") );
515
516 char buf[BUF_TEMP_SIZE];
517 size_t nWrite,
518 total = 0;
519
520 do
521 {
522 size_t nRead = sbuf->Read(buf, WXSIZEOF(buf));
523 if ( nRead )
524 {
525 nWrite = Write(buf, nRead);
526 if ( nWrite < nRead )
527 {
528 // put back data we couldn't copy
529 wxInputStream *in_stream = (wxInputStream *)sbuf->GetStream();
530
531 in_stream->Ungetch(buf + nWrite, nRead - nWrite);
532 }
533
534 total += nWrite;
535 }
536 else
537 {
538 nWrite = 0;
539 }
540 }
541 while ( nWrite == WXSIZEOF(buf) );
542
543 return total;
544 }
545
546 off_t wxStreamBuffer::Seek(off_t pos, wxSeekMode mode)
547 {
548 off_t ret_off, diff;
549
550 off_t last_access = GetLastAccess();
551
552 if ( !m_flushable )
553 {
554 switch (mode)
555 {
556 case wxFromStart:
557 diff = pos;
558 break;
559
560 case wxFromCurrent:
561 diff = pos + GetIntPosition();
562 break;
563
564 case wxFromEnd:
565 diff = pos + last_access;
566 break;
567
568 default:
569 wxFAIL_MSG( _T("invalid seek mode") );
570
571 return wxInvalidOffset;
572 }
573 if (diff < 0 || diff > last_access)
574 return wxInvalidOffset;
575 SetIntPosition(diff);
576 return diff;
577 }
578
579 switch ( mode )
580 {
581 case wxFromStart:
582 // We'll try to compute an internal position later ...
583 ret_off = m_stream->OnSysSeek(pos, wxFromStart);
584 ResetBuffer();
585 return ret_off;
586
587 case wxFromCurrent:
588 diff = pos + GetIntPosition();
589
590 if ( (diff > last_access) || (diff < 0) )
591 {
592 // We must take into account the fact that we have read
593 // something previously.
594 ret_off = m_stream->OnSysSeek(diff-last_access, wxFromCurrent);
595 ResetBuffer();
596 return ret_off;
597 }
598 else
599 {
600 SetIntPosition(diff);
601 return pos;
602 }
603
604 case wxFromEnd:
605 // Hard to compute: always seek to the requested position.
606 ret_off = m_stream->OnSysSeek(pos, wxFromEnd);
607 ResetBuffer();
608 return ret_off;
609 }
610
611 return wxInvalidOffset;
612 }
613
614 off_t wxStreamBuffer::Tell() const
615 {
616 off_t pos = m_stream->OnSysTell();
617 if ( pos == wxInvalidOffset )
618 return wxInvalidOffset;
619
620 pos += GetIntPosition();
621
622 if ( m_mode == read && m_flushable )
623 pos -= GetLastAccess();
624
625 return pos;
626 }
627
628 // ----------------------------------------------------------------------------
629 // wxStreamBase
630 // ----------------------------------------------------------------------------
631
632 wxStreamBase::wxStreamBase()
633 {
634 m_lasterror = wxStream_NOERROR;
635 m_lastcount = 0;
636 }
637
638 wxStreamBase::~wxStreamBase()
639 {
640 }
641
642 off_t wxStreamBase::OnSysSeek(off_t WXUNUSED(seek), wxSeekMode WXUNUSED(mode))
643 {
644 return wxInvalidOffset;
645 }
646
647 off_t wxStreamBase::OnSysTell() const
648 {
649 return wxInvalidOffset;
650 }
651
652 // ----------------------------------------------------------------------------
653 // wxInputStream
654 // ----------------------------------------------------------------------------
655
656 wxInputStream::wxInputStream()
657 {
658 m_wback = NULL;
659 m_wbacksize =
660 m_wbackcur = 0;
661 }
662
663 wxInputStream::~wxInputStream()
664 {
665 free(m_wback);
666 }
667
668 size_t wxInputStream::OnSysRead(void * WXUNUSED(buffer),
669 size_t WXUNUSED(bufsize))
670 {
671 return 0;
672 }
673
674 bool wxInputStream::Eof() const
675 {
676 wxInputStream *self = wxConstCast(this, wxInputStream);
677
678 char c;
679 self->Read(&c, 1);
680
681 // some streams can know that they're at EOF before actually trying to
682 // read beyond the end of stream (e.g. files) while others have no way of
683 // knowing it, so to provide the same behaviour in all cases we only
684 // return TRUE from here if the character really couldn't be read
685 if ( !self->LastRead() && GetLastError() == wxSTREAM_EOF )
686 {
687 return TRUE;
688 }
689
690 self->Ungetch(c);
691
692 return FALSE;
693 }
694
695 char *wxInputStream::AllocSpaceWBack(size_t needed_size)
696 {
697 // get number of bytes left from previous wback buffer
698 size_t toget = m_wbacksize - m_wbackcur;
699
700 // allocate a buffer large enough to hold prev + new data
701 char *temp_b = (char *)malloc(needed_size + toget);
702
703 if (!temp_b)
704 return NULL;
705
706 // copy previous data (and free old buffer) if needed
707 if (m_wback)
708 {
709 memmove(temp_b + needed_size, m_wback + m_wbackcur, toget);
710 free(m_wback);
711 }
712
713 // done
714 m_wback = temp_b;
715 m_wbackcur = 0;
716 m_wbacksize = needed_size + toget;
717
718 return m_wback;
719 }
720
721 size_t wxInputStream::GetWBack(void *buf, size_t bsize)
722 {
723 if (!m_wback)
724 return 0;
725
726 // how many bytes do we have in the buffer?
727 size_t toget = m_wbacksize - m_wbackcur;
728
729 if ( bsize < toget )
730 {
731 // we won't read everything
732 toget = bsize;
733 }
734
735 // copy the data from the cache
736 memcpy(buf, m_wback + m_wbackcur, toget);
737
738 m_wbackcur += toget;
739 if ( m_wbackcur == m_wbacksize )
740 {
741 // TODO: should we really free it here all the time? maybe keep it?
742 free(m_wback);
743 m_wback = NULL;
744 m_wbacksize = 0;
745 m_wbackcur = 0;
746 }
747
748 // return the number of bytes copied
749 return toget;
750 }
751
752 size_t wxInputStream::Ungetch(const void *buf, size_t bufsize)
753 {
754 char *ptrback = AllocSpaceWBack(bufsize);
755 if (!ptrback)
756 return 0;
757
758 memcpy(ptrback, buf, bufsize);
759 return bufsize;
760 }
761
762 bool wxInputStream::Ungetch(char c)
763 {
764 void *ptrback = AllocSpaceWBack(1);
765 if (!ptrback)
766 return FALSE;
767
768 *(char *)ptrback = c;
769 return TRUE;
770 }
771
772 char wxInputStream::GetC()
773 {
774 char c;
775 Read(&c, 1);
776 return c;
777 }
778
779 wxInputStream& wxInputStream::Read(void *buf, size_t size)
780 {
781 size_t retsize = GetWBack(buf, size);
782 if (retsize == size)
783 {
784 m_lastcount = size;
785 m_lasterror = wxStream_NOERROR;
786 return *this;
787 }
788 size -= retsize;
789 buf = (char *)buf + retsize;
790
791 m_lastcount = OnSysRead(buf, size) + retsize;
792 return *this;
793 }
794
795 char wxInputStream::Peek()
796 {
797 char c;
798 Read(&c, 1);
799 if (m_lasterror == wxStream_NOERROR)
800 {
801 Ungetch(c);
802 return c;
803 }
804
805 return 0;
806 }
807
808 wxInputStream& wxInputStream::Read(wxOutputStream& stream_out)
809 {
810 char buf[BUF_TEMP_SIZE];
811 size_t bytes_read = BUF_TEMP_SIZE;
812
813 while (bytes_read == BUF_TEMP_SIZE)
814 {
815 bytes_read = Read(buf, bytes_read).LastRead();
816 bytes_read = stream_out.Write(buf, bytes_read).LastWrite();
817 }
818 return *this;
819 }
820
821 off_t wxInputStream::SeekI(off_t pos, wxSeekMode mode)
822 {
823 // RR: This code is duplicated in wxBufferedInputStream. This is
824 // not really a good design, but buffered stream are different
825 // from all other in that they handle two stream-related objects,
826 // the stream buffer and parent stream.
827
828 // I don't know whether it should be put as well in wxFileInputStream::OnSysSeek
829 if (m_lasterror==wxSTREAM_EOF)
830 m_lasterror=wxSTREAM_NOERROR;
831
832 /* RR: A call to SeekI() will automatically invalidate any previous
833 call to Ungetch(), otherwise it would be possible to SeekI() to
834 one position, unread some bytes there, SeekI() to another position
835 and the data would be corrupted.
836
837 GRG: Could add code here to try to navigate within the wback
838 buffer if possible, but is it really needed? It would only work
839 when seeking in wxFromCurrent mode, else it would invalidate
840 anyway... */
841
842 if (m_wback)
843 {
844 wxLogDebug( wxT("Seeking in stream which has data written back to it.") );
845
846 free(m_wback);
847 m_wback = NULL;
848 m_wbacksize = 0;
849 m_wbackcur = 0;
850 }
851
852 return OnSysSeek(pos, mode);
853 }
854
855 off_t wxInputStream::TellI() const
856 {
857 off_t pos = OnSysTell();
858
859 if (pos != wxInvalidOffset)
860 pos -= (m_wbacksize - m_wbackcur);
861
862 return pos;
863 }
864
865
866 // ----------------------------------------------------------------------------
867 // wxOutputStream
868 // ----------------------------------------------------------------------------
869
870 wxOutputStream::wxOutputStream()
871 {
872 }
873
874 wxOutputStream::~wxOutputStream()
875 {
876 }
877
878 size_t wxOutputStream::OnSysWrite(const void * WXUNUSED(buffer),
879 size_t WXUNUSED(bufsize))
880 {
881 return 0;
882 }
883
884 void wxOutputStream::PutC(char c)
885 {
886 Write(&c, 1);
887 }
888
889 wxOutputStream& wxOutputStream::Write(const void *buffer, size_t size)
890 {
891 m_lastcount = OnSysWrite(buffer, size);
892 return *this;
893 }
894
895 wxOutputStream& wxOutputStream::Write(wxInputStream& stream_in)
896 {
897 stream_in.Read(*this);
898 return *this;
899 }
900
901 off_t wxOutputStream::TellO() const
902 {
903 return OnSysTell();
904 }
905
906 off_t wxOutputStream::SeekO(off_t pos, wxSeekMode mode)
907 {
908 return OnSysSeek(pos, mode);
909 }
910
911 void wxOutputStream::Sync()
912 {
913 }
914
915
916 // ----------------------------------------------------------------------------
917 // wxCountingOutputStream
918 // ----------------------------------------------------------------------------
919
920 wxCountingOutputStream::wxCountingOutputStream ()
921 {
922 m_currentPos = 0;
923 }
924
925 size_t wxCountingOutputStream::GetSize() const
926 {
927 return m_lastcount;
928 }
929
930 size_t wxCountingOutputStream::OnSysWrite(const void *WXUNUSED(buffer),
931 size_t size)
932 {
933 m_currentPos += size;
934 if (m_currentPos > m_lastcount)
935 m_lastcount = m_currentPos;
936
937 return m_currentPos;
938 }
939
940 off_t wxCountingOutputStream::OnSysSeek(off_t pos, wxSeekMode mode)
941 {
942 switch ( mode )
943 {
944 case wxFromStart:
945 m_currentPos = pos;
946 break;
947
948 case wxFromEnd:
949 m_currentPos = m_lastcount + pos;
950 break;
951
952 case wxFromCurrent:
953 m_currentPos += pos;
954 break;
955
956 default:
957 wxFAIL_MSG( _T("invalid seek mode") );
958 return wxInvalidOffset;
959 }
960
961 if (m_currentPos > m_lastcount)
962 m_lastcount = m_currentPos;
963
964 return m_currentPos;
965 }
966
967 off_t wxCountingOutputStream::OnSysTell() const
968 {
969 return m_currentPos;
970 }
971
972 // ----------------------------------------------------------------------------
973 // wxFilterInputStream
974 // ----------------------------------------------------------------------------
975
976 wxFilterInputStream::wxFilterInputStream()
977 {
978 m_parent_i_stream = NULL;
979 }
980
981 wxFilterInputStream::wxFilterInputStream(wxInputStream& stream)
982 {
983 m_parent_i_stream = &stream;
984 }
985
986 wxFilterInputStream::~wxFilterInputStream()
987 {
988 }
989
990 // ----------------------------------------------------------------------------
991 // wxFilterOutputStream
992 // ----------------------------------------------------------------------------
993
994 wxFilterOutputStream::wxFilterOutputStream()
995 {
996 m_parent_o_stream = NULL;
997 }
998
999 wxFilterOutputStream::wxFilterOutputStream(wxOutputStream& stream)
1000 {
1001 m_parent_o_stream = &stream;
1002 }
1003
1004 wxFilterOutputStream::~wxFilterOutputStream()
1005 {
1006 }
1007
1008 // ----------------------------------------------------------------------------
1009 // wxBufferedInputStream
1010 // ----------------------------------------------------------------------------
1011
1012 wxBufferedInputStream::wxBufferedInputStream(wxInputStream& s,
1013 wxStreamBuffer *buffer)
1014 : wxFilterInputStream(s)
1015 {
1016 if ( buffer )
1017 {
1018 // use the buffer provided by the user
1019 m_i_streambuf = buffer;
1020 }
1021 else // create a default buffer
1022 {
1023 m_i_streambuf = new wxStreamBuffer(*this, wxStreamBuffer::read);
1024
1025 m_i_streambuf->SetBufferIO(1024);
1026 }
1027 }
1028
1029 wxBufferedInputStream::~wxBufferedInputStream()
1030 {
1031 m_parent_i_stream->SeekI(-(off_t)m_i_streambuf->GetBytesLeft(),
1032 wxFromCurrent);
1033
1034 delete m_i_streambuf;
1035 }
1036
1037 char wxBufferedInputStream::Peek()
1038 {
1039 return m_i_streambuf->Peek();
1040 }
1041
1042 wxInputStream& wxBufferedInputStream::Read(void *buf, size_t size)
1043 {
1044 // reset the error flag
1045 m_lasterror = wxStream_NOERROR;
1046
1047 // first read from the already cached data
1048 m_lastcount = GetWBack(buf, size);
1049
1050 // do we have to read anything more?
1051 if ( m_lastcount < size )
1052 {
1053 size -= m_lastcount;
1054 buf = (char *)buf + m_lastcount;
1055
1056 // the call to wxStreamBuffer::Read() below will reset our m_lastcount,
1057 // so save it
1058 size_t countOld = m_lastcount;
1059
1060 m_i_streambuf->Read(buf, size);
1061
1062 m_lastcount += countOld;
1063 }
1064
1065 return *this;
1066 }
1067
1068 off_t wxBufferedInputStream::SeekI(off_t pos, wxSeekMode mode)
1069 {
1070 // RR: Look at wxInputStream for comments.
1071
1072 if (m_lasterror==wxSTREAM_EOF)
1073 m_lasterror=wxSTREAM_NOERROR;
1074
1075 if (m_wback)
1076 {
1077 wxLogDebug( wxT("Seeking in stream which has data written back to it.") );
1078
1079 free(m_wback);
1080 m_wback = NULL;
1081 m_wbacksize = 0;
1082 m_wbackcur = 0;
1083 }
1084
1085 return m_i_streambuf->Seek(pos, mode);
1086 }
1087
1088 off_t wxBufferedInputStream::TellI() const
1089 {
1090 off_t pos = m_i_streambuf->Tell();
1091
1092 if (pos != wxInvalidOffset)
1093 pos -= (m_wbacksize - m_wbackcur);
1094
1095 return pos;
1096 }
1097
1098 size_t wxBufferedInputStream::OnSysRead(void *buffer, size_t bufsize)
1099 {
1100 return m_parent_i_stream->Read(buffer, bufsize).LastRead();
1101 }
1102
1103 off_t wxBufferedInputStream::OnSysSeek(off_t seek, wxSeekMode mode)
1104 {
1105 return m_parent_i_stream->SeekI(seek, mode);
1106 }
1107
1108 off_t wxBufferedInputStream::OnSysTell() const
1109 {
1110 return m_parent_i_stream->TellI();
1111 }
1112
1113 void wxBufferedInputStream::SetInputStreamBuffer(wxStreamBuffer *buffer)
1114 {
1115 wxCHECK_RET( buffer, _T("wxBufferedInputStream needs buffer") );
1116
1117 delete m_i_streambuf;
1118 m_i_streambuf = buffer;
1119 }
1120
1121 // ----------------------------------------------------------------------------
1122 // wxBufferedOutputStream
1123 // ----------------------------------------------------------------------------
1124
1125 wxBufferedOutputStream::wxBufferedOutputStream(wxOutputStream& s,
1126 wxStreamBuffer *buffer)
1127 : wxFilterOutputStream(s)
1128 {
1129 if ( buffer )
1130 {
1131 m_o_streambuf = buffer;
1132 }
1133 else // create a default one
1134 {
1135 m_o_streambuf = new wxStreamBuffer(*this, wxStreamBuffer::write);
1136
1137 m_o_streambuf->SetBufferIO(1024);
1138 }
1139 }
1140
1141 wxBufferedOutputStream::~wxBufferedOutputStream()
1142 {
1143 Sync();
1144 delete m_o_streambuf;
1145 }
1146
1147 wxOutputStream& wxBufferedOutputStream::Write(const void *buffer, size_t size)
1148 {
1149 m_lastcount = 0;
1150 m_o_streambuf->Write(buffer, size);
1151 return *this;
1152 }
1153
1154 off_t wxBufferedOutputStream::SeekO(off_t pos, wxSeekMode mode)
1155 {
1156 Sync();
1157 return m_o_streambuf->Seek(pos, mode);
1158 }
1159
1160 off_t wxBufferedOutputStream::TellO() const
1161 {
1162 return m_o_streambuf->Tell();
1163 }
1164
1165 void wxBufferedOutputStream::Sync()
1166 {
1167 m_o_streambuf->FlushBuffer();
1168 m_parent_o_stream->Sync();
1169 }
1170
1171 size_t wxBufferedOutputStream::OnSysWrite(const void *buffer, size_t bufsize)
1172 {
1173 return m_parent_o_stream->Write(buffer, bufsize).LastWrite();
1174 }
1175
1176 off_t wxBufferedOutputStream::OnSysSeek(off_t seek, wxSeekMode mode)
1177 {
1178 return m_parent_o_stream->SeekO(seek, mode);
1179 }
1180
1181 off_t wxBufferedOutputStream::OnSysTell() const
1182 {
1183 return m_parent_o_stream->TellO();
1184 }
1185
1186 size_t wxBufferedOutputStream::GetSize() const
1187 {
1188 return m_parent_o_stream->GetSize() + m_o_streambuf->GetIntPosition();
1189 }
1190
1191 void wxBufferedOutputStream::SetOutputStreamBuffer(wxStreamBuffer *buffer)
1192 {
1193 wxCHECK_RET( buffer, _T("wxBufferedOutputStream needs buffer") );
1194
1195 delete m_o_streambuf;
1196 m_o_streambuf = buffer;
1197 }
1198
1199 // ----------------------------------------------------------------------------
1200 // Some IOManip function
1201 // ----------------------------------------------------------------------------
1202
1203 wxOutputStream& wxEndL(wxOutputStream& stream)
1204 {
1205 static const wxChar *eol = wxTextFile::GetEOL();
1206
1207 return stream.Write(eol, wxStrlen(eol));
1208 }
1209
1210 #endif
1211 // wxUSE_STREAMS