Geant4 Cross Reference |
1 // 1 // 2 // ******************************************* 2 // ******************************************************************** 3 // * License and Disclaimer 3 // * License and Disclaimer * 4 // * 4 // * * 5 // * The Geant4 software is copyright of th 5 // * The Geant4 software is copyright of the Copyright Holders of * 6 // * the Geant4 Collaboration. It is provided 6 // * the Geant4 Collaboration. It is provided under the terms and * 7 // * conditions of the Geant4 Software License 7 // * conditions of the Geant4 Software License, included in the file * 8 // * LICENSE and available at http://cern.ch/ 8 // * LICENSE and available at http://cern.ch/geant4/license . These * 9 // * include a list of copyright holders. 9 // * include a list of copyright holders. * 10 // * 10 // * * 11 // * Neither the authors of this software syst 11 // * Neither the authors of this software system, nor their employing * 12 // * institutes,nor the agencies providing fin 12 // * institutes,nor the agencies providing financial support for this * 13 // * work make any representation or warran 13 // * work make any representation or warranty, express or implied, * 14 // * regarding this software system or assum 14 // * regarding this software system or assume any liability for its * 15 // * use. Please see the license in the file 15 // * use. Please see the license in the file LICENSE and URL above * 16 // * for the full disclaimer and the limitatio 16 // * for the full disclaimer and the limitation of liability. * 17 // * 17 // * * 18 // * This code implementation is the result 18 // * This code implementation is the result of the scientific and * 19 // * technical work of the GEANT4 collaboratio 19 // * technical work of the GEANT4 collaboration. * 20 // * By using, copying, modifying or distri 20 // * By using, copying, modifying or distributing the software (or * 21 // * any work based on the software) you ag 21 // * any work based on the software) you agree to acknowledge its * 22 // * use in resulting scientific publicati 22 // * use in resulting scientific publications, and indicate your * 23 // * acceptance of all terms of the Geant4 Sof 23 // * acceptance of all terms of the Geant4 Software license. * 24 // ******************************************* 24 // ******************************************************************** 25 // 25 // 26 // 26 // >> 27 // $Id: G4RTOutBitStream.cc 66373 2012-12-18 09:41:34Z gcosmo $ 27 // 28 // 28 // 29 // 29 // 30 // 30 31 31 #include <string.h> 32 #include <string.h> 32 #include "G4RTJpeg.hh" 33 #include "G4RTJpeg.hh" 33 #include "G4RTOutBitStream.hh" 34 #include "G4RTOutBitStream.hh" 34 35 35 36 36 37 37 G4OutBitStream::G4OutBitStream(int size) 38 G4OutBitStream::G4OutBitStream(int size) 38 { 39 { 39 if(size < 1) 40 if(size < 1) 40 throw( G4MemoryError( size, "G4OutBitS 41 throw( G4MemoryError( size, "G4OutBitStream" ) ); 41 42 42 mHeadOfBuf = mBuf = new u_char[size]; 43 mHeadOfBuf = mBuf = new u_char[size]; 43 if( mHeadOfBuf == 0 ) 44 if( mHeadOfBuf == 0 ) 44 throw( G4MemoryError( size, "G4OutBitS 45 throw( G4MemoryError( size, "G4OutBitStream" ) ); 45 46 46 mEndOfBuf = mBuf + size; 47 mEndOfBuf = mBuf + size; 47 48 48 memset( mHeadOfBuf, 0, size ); 49 memset( mHeadOfBuf, 0, size ); 49 50 50 mBitPos = 7; 51 mBitPos = 7; 51 mWriteFlag = 1; 52 mWriteFlag = 1; 52 } 53 } 53 54 54 G4OutBitStream::~G4OutBitStream() 55 G4OutBitStream::~G4OutBitStream() 55 { 56 { 56 delete mBuf; 57 delete mBuf; 57 } 58 } 58 59 59 void 60 void 60 G4OutBitStream::IncBuf( void ) 61 G4OutBitStream::IncBuf( void ) 61 { 62 { 62 if( ++mBuf >= mEndOfBuf ) 63 if( ++mBuf >= mEndOfBuf ) 63 mWriteFlag = 0; 64 mWriteFlag = 0; 64 } 65 } 65 66 66 67 67 68 68 void 69 void 69 G4OutBitStream::SetBits(int v, int numBits) 70 G4OutBitStream::SetBits(int v, int numBits) 70 { 71 { 71 if( numBits == 0 ) 72 if( numBits == 0 ) 72 return; 73 return; 73 if( numBits > 16 ) 74 if( numBits > 16 ) 74 throw( G4BufferError( "SetBits:Max Bit 75 throw( G4BufferError( "SetBits:Max Bit Over" ) ); 75 if( numBits > 8 ){ 76 if( numBits > 8 ){ 76 Set8Bits( u_char(v>>8), numBits-8 ); 77 Set8Bits( u_char(v>>8), numBits-8 ); 77 numBits = 8; 78 numBits = 8; 78 } 79 } 79 Set8Bits( u_char(v), numBits ); 80 Set8Bits( u_char(v), numBits ); 80 } 81 } 81 82 82 void 83 void 83 G4OutBitStream::SetFewBits(u_char v, int numBi 84 G4OutBitStream::SetFewBits(u_char v, int numBits) 84 { 85 { 85 v &= BitFullMaskT[numBits-1]; 86 v &= BitFullMaskT[numBits-1]; 86 *mBuf |= v << (mBitPos + 1 - numBits); 87 *mBuf |= v << (mBitPos + 1 - numBits); 87 if( (mBitPos -= numBits) < 0 ){ 88 if( (mBitPos -= numBits) < 0 ){ 88 if( *mBuf == 0xff ){ 89 if( *mBuf == 0xff ){ 89 IncBuf(); 90 IncBuf(); 90 *mBuf = 0; 91 *mBuf = 0; 91 } 92 } 92 IncBuf(); 93 IncBuf(); 93 mBitPos = 7; 94 mBitPos = 7; 94 } 95 } 95 } 96 } 96 97 97 void 98 void 98 G4OutBitStream::SetBits2Byte(u_char v, int num 99 G4OutBitStream::SetBits2Byte(u_char v, int numBits) 99 { 100 { 100 v &= BitFullMaskT[numBits-1]; 101 v &= BitFullMaskT[numBits-1]; 101 int nextBits = numBits - (mBitPos + 1); 102 int nextBits = numBits - (mBitPos + 1); 102 *mBuf |= ( v >> nextBits ) & BitFullMaskT[mB 103 *mBuf |= ( v >> nextBits ) & BitFullMaskT[mBitPos]; 103 if( *mBuf == 0xff ){ 104 if( *mBuf == 0xff ){ 104 IncBuf(); 105 IncBuf(); 105 *mBuf = 0; 106 *mBuf = 0; 106 } 107 } 107 IncBuf(); 108 IncBuf(); 108 109 109 *mBuf = v << (8 - nextBits); 110 *mBuf = v << (8 - nextBits); 110 mBitPos = 7 - nextBits; 111 mBitPos = 7 - nextBits; 111 } 112 } 112 113 113 void 114 void 114 G4OutBitStream::Set8Bits(u_char v, int numBits 115 G4OutBitStream::Set8Bits(u_char v, int numBits) 115 { 116 { 116 if( mBitPos + 1 >= numBits ) 117 if( mBitPos + 1 >= numBits ) 117 SetFewBits( (u_char)v, numBits ); 118 SetFewBits( (u_char)v, numBits ); 118 else 119 else 119 SetBits2Byte( (u_char)v, numBits ); 120 SetBits2Byte( (u_char)v, numBits ); 120 } 121 } 121 122 122 123 123 void 124 void 124 G4OutBitStream::FullBit( void ) 125 G4OutBitStream::FullBit( void ) 125 { 126 { 126 if( mBitPos != 7 ) 127 if( mBitPos != 7 ) 127 SetFewBits( BitFullMaskT[mBitPos], mBi 128 SetFewBits( BitFullMaskT[mBitPos], mBitPos+1 ); 128 } 129 } 129 130 130 void 131 void 131 G4OutBitStream::SetByte(u_char dat) 132 G4OutBitStream::SetByte(u_char dat) 132 { 133 { 133 if( mWriteFlag ){ 134 if( mWriteFlag ){ 134 FullBit(); 135 FullBit(); 135 *mBuf = dat; 136 *mBuf = dat; 136 IncBuf(); 137 IncBuf(); 137 return; 138 return; 138 } 139 } 139 throw( G4BufferError( "SetByte" ) ); 140 throw( G4BufferError( "SetByte" ) ); 140 } 141 } 141 142 142 void 143 void 143 G4OutBitStream::SetWord(u_int dat) 144 G4OutBitStream::SetWord(u_int dat) 144 { 145 { 145 if( mWriteFlag ){ 146 if( mWriteFlag ){ 146 FullBit(); 147 FullBit(); 147 *mBuf = (dat >> 8) & 0xff; 148 *mBuf = (dat >> 8) & 0xff; 148 IncBuf(); 149 IncBuf(); 149 *mBuf = dat & 0xff; 150 *mBuf = dat & 0xff; 150 IncBuf(); 151 IncBuf(); 151 return; 152 return; 152 } 153 } 153 throw( G4BufferError( "SetWord" ) ); 154 throw( G4BufferError( "SetWord" ) ); 154 } 155 } 155 156 156 void 157 void 157 G4OutBitStream::CopyByte(const char* src, int 158 G4OutBitStream::CopyByte(const char* src, int n) 158 { 159 { 159 if( mBuf+n < mEndOfBuf ){ 160 if( mBuf+n < mEndOfBuf ){ 160 FullBit(); 161 FullBit(); 161 memcpy( mBuf, src, n ); 162 memcpy( mBuf, src, n ); 162 mBuf += n; 163 mBuf += n; 163 return; 164 return; 164 } 165 } 165 throw( G4BufferError( "CopyByte" ) ); 166 throw( G4BufferError( "CopyByte" ) ); 166 } 167 } 167 168 168 169