diff options
Diffstat (limited to 'src/prop/cryptominisat/Solver/StreamBuffer.h')
-rw-r--r-- | src/prop/cryptominisat/Solver/StreamBuffer.h | 78 |
1 files changed, 78 insertions, 0 deletions
diff --git a/src/prop/cryptominisat/Solver/StreamBuffer.h b/src/prop/cryptominisat/Solver/StreamBuffer.h new file mode 100644 index 000000000..6dc390415 --- /dev/null +++ b/src/prop/cryptominisat/Solver/StreamBuffer.h @@ -0,0 +1,78 @@ +/******************************************************************************************[Main.C] +MiniSat -- Copyright (c) 2003-2006, Niklas Een, Niklas Sorensson + +Permission is hereby granted, free of charge, to any person obtaining a copy of this software and +associated documentation files (the "Software"), to deal in the Software without restriction, +including without limitation the rights to use, copy, modify, merge, publish, distribute, +sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is +furnished to do so, subject to the following conditions: + +The above copyright notice and this permission notice shall be included in all copies or +substantial portions of the Software. + +THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT +NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND +NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, +DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT +OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. +**************************************************************************************************/ + +#ifndef STREAMBUFFER_H +#define STREAMBUFFER_H + +#define CHUNK_LIMIT 1048576 + +#ifndef DISABLE_ZLIB +#include <zlib.h> +#endif // DISABLE_ZLIB + +namespace CMSat +{ + +class StreamBuffer +{ + #ifdef DISABLE_ZLIB + FILE * in; + #else + gzFile in; + #endif // DISABLE_ZLIB + char buf[CHUNK_LIMIT]; + int pos; + int size; + + void assureLookahead() { + if (pos >= size) { + pos = 0; + #ifdef DISABLE_ZLIB + #ifdef VERBOSE_DEBUG + printf("buf = %08X\n", buf); + printf("sizeof(buf) = %u\n", sizeof(buf)); + #endif //VERBOSE_DEBUG + size = fread(buf, 1, sizeof(buf), in); + #else + size = gzread(in, buf, sizeof(buf)); + #endif // DISABLE_ZLIB + } + } + +public: + #ifdef DISABLE_ZLIB + StreamBuffer(FILE * i) : in(i), pos(0), size(0) { + #else + StreamBuffer(gzFile i) : in(i), pos(0), size(0) { + #endif // DISABLE_ZLIB + assureLookahead(); + } + + int operator * () { + return (pos >= size) ? EOF : buf[pos]; + } + void operator ++ () { + pos++; + assureLookahead(); + } +}; + +} + +#endif //STREAMBUFFER_H |