Files
geant4/source/visualization/RayTracer/src/G4RTOutBitStream.cc
T
2016-06-09 15:16:48 +02:00

165 lines
4.0 KiB
C++

//
// ********************************************************************
// * License and Disclaimer *
// * *
// * The Geant4 software is copyright of the Copyright Holders of *
// * the Geant4 Collaboration. It is provided under the terms and *
// * conditions of the Geant4 Software License, included in the file *
// * LICENSE and available at http://cern.ch/geant4/license . These *
// * include a list of copyright holders. *
// * *
// * Neither the authors of this software system, nor their employing *
// * institutes,nor the agencies providing financial support for this *
// * work make any representation or warranty, express or implied, *
// * regarding this software system or assume any liability for its *
// * use. Please see the license in the file LICENSE and URL above *
// * for the full disclaimer and the limitation of liability. *
// * *
// * This code implementation is the result of the scientific and *
// * technical work of the GEANT4 collaboration. *
// * By using, copying, modifying or distributing the software (or *
// * any work based on the software) you agree to acknowledge its *
// * use in resulting scientific publications, and indicate your *
// * acceptance of all terms of the Geant4 Software license. *
// ********************************************************************
//
//
// $Id: G4RTOutBitStream.cc,v 1.5 2006/06/29 21:24:01 gunter Exp $
// GEANT4 tag $Name: geant4-09-00 $
//
//
//
#include <string.h>
#include "G4RTJpeg.hh"
#include "G4RTOutBitStream.hh"
G4OutBitStream::G4OutBitStream(int size)
{
if(size < 1)
throw( G4MemoryError( size, "G4OutBitStream" ) );
mHeadOfBuf = mBuf = new u_char[size];
if( mHeadOfBuf == 0 )
throw( G4MemoryError( size, "G4OutBitStream" ) );
mEndOfBuf = mBuf + size;
memset( mHeadOfBuf, 0, size );
mBitPos = 7;
mWriteFlag = 1;
}
void
G4OutBitStream::IncBuf( void )
{
if( ++mBuf >= mEndOfBuf )
mWriteFlag = 0;
}
void
G4OutBitStream::SetBits(int v, int numBits)
{
if( numBits == 0 )
return;
if( numBits > 16 )
throw( G4BufferError( "SetBits:Max Bit Over" ) );
if( numBits > 8 ){
Set8Bits( u_char(v>>8), numBits-8 );
numBits = 8;
}
Set8Bits( u_char(v), numBits );
}
void
G4OutBitStream::SetFewBits(u_char v, int numBits)
{
v &= BitFullMaskT[numBits-1];
*mBuf |= v << (mBitPos + 1 - numBits);
if( (mBitPos -= numBits) < 0 ){
if( *mBuf == 0xff ){
IncBuf();
*mBuf = 0;
}
IncBuf();
mBitPos = 7;
}
}
void
G4OutBitStream::SetBits2Byte(u_char v, int numBits)
{
v &= BitFullMaskT[numBits-1];
int nextBits = numBits - (mBitPos + 1);
*mBuf |= ( v >> nextBits ) & BitFullMaskT[mBitPos];
if( *mBuf == 0xff ){
IncBuf();
*mBuf = 0;
}
IncBuf();
*mBuf = v << 8 - nextBits;
mBitPos = 7 - nextBits;
}
void
G4OutBitStream::Set8Bits(u_char v, int numBits)
{
if( mBitPos + 1 >= numBits )
SetFewBits( (u_char)v, numBits );
else
SetBits2Byte( (u_char)v, numBits );
}
void
G4OutBitStream::FullBit( void )
{
if( mBitPos != 7 )
SetFewBits( BitFullMaskT[mBitPos], mBitPos+1 );
}
void
G4OutBitStream::SetByte(u_char dat)
{
if( mWriteFlag ){
FullBit();
*mBuf = dat;
IncBuf();
return;
}
throw( G4BufferError( "SetByte" ) );
}
void
G4OutBitStream::SetWord(u_int dat)
{
if( mWriteFlag ){
FullBit();
*mBuf = (dat >> 8) & 0xff;
IncBuf();
*mBuf = dat & 0xff;
IncBuf();
return;
}
throw( G4BufferError( "SetWord" ) );
}
void
G4OutBitStream::CopyByte(const char* src, int n)
{
if( mBuf+n < mEndOfBuf ){
FullBit();
memcpy( mBuf, src, n );
mBuf += n;
return;
}
throw( G4BufferError( "CopyByte" ) );
}