Files
harbour-core/harbour/contrib/xhb/stream.prg
Viktor Szakats 81c9b0506c 2010-06-04 15:21 UTC+0200 Viktor Szakats (harbour.01 syenar.hu)
* contrib/*
    * Deleted 'www.' from harbour-project.org website name.
      (www.harbour-project.org -> harbour-project.org)
2010-06-04 13:32:23 +00:00

274 lines
7.0 KiB
Plaintext

/*
* $Id$
*/
/*
* Harbour Project source code:
* xhb stream classes
*
* Copyright 2009 {list of individual authors and e-mail addresses}
* www - http://harbour-project.org
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2, or (at your option)
* any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this software; see the file COPYING. If not, write to
* the Free Software Foundation, Inc., 59 Temple Place, Suite 330,
* Boston, MA 02111-1307 USA (or visit the web site http://www.gnu.org/).
*
* As a special exception, the Harbour Project gives permission for
* additional uses of the text contained in its release of Harbour.
*
* The exception is that, if you link the Harbour libraries with other
* files to produce an executable, this does not by itself cause the
* resulting executable to be covered by the GNU General Public License.
* Your use of that executable is in no way restricted on account of
* linking the Harbour library code into it.
*
* This exception does not however invalidate any other reasons why
* the executable file might be covered by the GNU General Public License.
*
* This exception applies only to the code released by the Harbour
* Project under the name Harbour. If you copy code from other
* Harbour Project or Free Software Foundation releases into a copy of
* Harbour, as the General Public License permits, the exception does
* not apply to the code that you add in this way. To avoid misleading
* anyone as to the status of such modified files, you must delete
* this exception notice from them.
*
* If you write modifications of your own for Harbour, it is your choice
* whether to permit this exception to apply to your modifications.
* If you do not wish that, delete this exception notice.
*
*/
#include "hbclass.ch"
#include "common.ch"
#include "fileio.ch"
#xtranslate THROW(<oErr>) => (Eval(ErrorBlock(), <oErr>), Break(<oErr>))
#define BUFFER_SIZE 16384
FUNCTION FileReader( cFile, nMode )
RETURN TStreamFileReader():New( cFile, nMode )
FUNCTION FileWriter( cFile, nMode )
RETURN TStreamFileWriter():New( cFile, nMode )
CREATE CLASS TStream
VAR lCanRead INIT .F.
VAR lCanWrite INIT .F.
VAR lCanSeek INIT .F.
VAR nLength INIT 0
VAR nPosition INIT 0
//METHOD BeginRead( sBuffer, nOffset, nCount, bCallback, oState )
//METHOD EndRead( oAsync ) VIRTUAL
//METHOD BeginWrite( sBuffer, nOffset, nCount, bCallback, oState )
//METHOD EndWrite( oAsync ) VIRTUAL
METHOD Close() VIRTUAL
METHOD Flush() VIRTUAL
METHOD Seek( nOffset, Origin ) VIRTUAL
METHOD Read( sBuffer, nOffset, nCount ) VIRTUAL
METHOD ReadByte() VIRTUAL
METHOD Write( sBuffer, nOffset, nCount ) VIRTUAL
METHOD WriteByte() VIRTUAL
METHOD CopyTo( oTargetStream )
ENDCLASS
METHOD PROCEDURE CopyTo( oTargetStream ) CLASS TStream
LOCAL nBytesToRead := ::nLength
LOCAL sBuffer := Space( BUFFER_SIZE )
LOCAL nRead
LOCAL nPosition
IF oTargetStream:lCanWrite == .F.
Throw( ErrorNew( "Stream", 0, 1001, ProcName(), "Target not writable.", HB_aParams() ) )
ENDIF
// Save.
nPosition := ::nPosition
::Seek( 0, FS_SET )
oTargetStream:Seek( 0, FS_SET )
DO WHILE nBytesToRead > 0
nRead := ::Read( @sBuffer, 0, BUFFER_SIZE )
oTargetStream:Write( sBuffer, 0, nRead )
nBytesToRead -= nRead
ENDDO
//Truncate incase it was a bigger file.
oTargetStream:Write( "", 0, 0 )
// Restore.
::Seek( nPosition, FS_SET )
RETURN
CREATE CLASS TStreamFileReader FROM TStream
VAR cFile
VAR Handle
METHOD New( cFile, nMode ) CONSTRUCTOR
METHOD Close() INLINE IIF( ::Handle > 0, FClose( ::Handle ), ), ::Handle := -2
METHOD Seek( nOffset Origin ) INLINE FSeek( ::Handle, nOffset, Origin )
METHOD Read( sBuffer, nOffset, nCount )
METHOD ReadByte()
DESTRUCTOR Finalize
ENDCLASS
METHOD New( cFile, nMode ) CLASS TStreamFileReader
::lCanRead := .T.
::cFile := cFile
IF nMode == NIL
nMode := 2
ENDIF
::Handle := FOpen( cFile, nMode )
IF ::Handle <= 0
Throw( ErrorNew( "Stream", 0, 1004, ProcName(), "Open Error: " + Str( FError() ), HB_aParams() ) )
ENDIF
::nPosition := 0
::nLength := FSeek( ::Handle, 0, FS_END )
FSeek( ::Handle, 0, FS_SET )
RETURN Self
METHOD PROCEDURE Finalize CLASS TStreamFileReader
::Close()
RETURN
METHOD Read( sBuffer, nOffset, nCount ) CLASS TStreamFileReader
LOCAL nRead
/*
IF ! HB_IsByRef( @sBuffer )
Throw( ErrorNew( "Stream", 0, 1002, ProcName(), "Buffer not BYREF.", HB_aParams() ) )
ENDIF
*/
nRead := FRead( ::Handle, @sBuffer, nCount, nOffset )
::nPosition += nRead
RETURN nRead
METHOD ReadByte() CLASS TStreamFileReader
LOCAL sBuffer := " "
IF FRead( ::Handle, @sBuffer, 1 ) == 1
::nPosition++
RETURN sBuffer
ENDIF
RETURN ""
CREATE CLASS TStreamFileWriter FROM TStream
VAR cFile
VAR Handle
METHOD New( cFile, nMode ) CONSTRUCTOR
METHOD Close() INLINE IIF( ::Handle > 0, FClose( ::Handle ), ), ::Handle := -2
METHOD Seek( nOffset Origin ) INLINE ::nPosition := FSeek( ::Handle, nOffset, Origin )
METHOD Write( sBuffer, nOffset, nCount )
METHOD WriteByte( cByte )
DESTRUCTOR Finalize
ENDCLASS
METHOD New( cFile, nMode ) CLASS TStreamFileWriter
::lCanWrite := .T.
::cFile := cFile
IF hb_FileExists( cFile )
DEFAULT nMode TO FO_READWRITE
::Handle := FOpen( cFile, nMode )
IF ::Handle <= 0
Throw( ErrorNew( "Stream", 0, 1004, ProcName(), "Open Error: " + Str( FError() ), HB_aParams() ) )
ENDIF
::nLength := FSeek( ::Handle, 0, FS_END )
::nPosition := ::nLength
ELSE
DEFAULT nMode TO FC_NORMAL
::Handle := FCreate( cFile, nMode )
IF ::Handle <= 0
Throw( ErrorNew( "Stream", 0, 1004, ProcName(), "Create Error: " + Str( FError() ), HB_aParams() ) )
ENDIF
::nPosition := 0
::nLength := 0
ENDIF
RETURN Self
METHOD PROCEDURE Finalize CLASS TStreamFileWriter
::Close()
RETURN
METHOD Write( sBuffer, nOffset, nCount ) CLASS TStreamFileWriter
LOCAL nWritten
nWritten := FWrite( ::Handle, sBuffer, nCount, nOffset )
::nPosition += nWritten
IF nWritten != nCount
Throw( ErrorNew( "Stream", 0, 1003, ProcName(), "Write failed - written:" + Str( nWritten ) + " bytes", HB_aParams() ) )
ENDIF
RETURN nWritten
METHOD PROCEDURE WriteByte( cByte ) CLASS TStreamFileWriter
LOCAL nWritten := FWrite( ::Handle, cByte, 1 )
::nPosition += nWritten
IF nWritten != 1
Throw( ErrorNew( "Stream", 0, 1006, ProcName(), "Write failed", HB_aParams() ) )
ENDIF
RETURN