| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400 |
- "use strict";
- Object.defineProperty(exports, "__esModule", { value: true });
- exports.BitStream = void 0;
- const bit_1 = require("./bit");
- const byte_stream_1 = require("./byte_stream");
- class BitStream {
- constructor(parameters) {
- this.buffer = new ArrayBuffer(0);
- this.view = new Uint8Array(this.buffer);
- this.bitsCount = 0;
- if (parameters) {
- if ("byteStream" in parameters) {
- this.fromByteStream(parameters.byteStream);
- }
- if ("view" in parameters) {
- this.fromUint8Array(parameters.view);
- }
- if ("buffer" in parameters) {
- this.fromArrayBuffer(parameters.buffer);
- }
- if ("string" in parameters) {
- this.fromString(parameters.string);
- }
- if ("uint32" in parameters) {
- this.fromUint32(parameters.uint32);
- }
- if ("bitsCount" in parameters && parameters.bitsCount) {
- this.bitsCount = parameters.bitsCount;
- }
- }
- }
- clear() {
- this.buffer = new ArrayBuffer(0);
- this.view = new Uint8Array(this.buffer);
- this.bitsCount = 0;
- }
- fromByteStream(stream) {
- this.fromUint8Array(stream.view);
- }
- fromArrayBuffer(array) {
- this.buffer = array;
- this.view = new Uint8Array(array);
- this.bitsCount = this.view.length << 3;
- }
- fromUint8Array(array) {
- this.fromArrayBuffer(new Uint8Array(array).buffer);
- }
- fromString(string) {
- const stringLength = string.length;
- this.buffer = new ArrayBuffer((stringLength >> 3) + ((stringLength % 8) ? 1 : 0));
- this.view = new Uint8Array(this.buffer);
- this.bitsCount = ((stringLength >> 3) + 1) << 3;
- let byteIndex = 0;
- for (let i = 0; i < stringLength; i++) {
- if (string[i] == "1")
- this.view[byteIndex] |= 1 << (7 - (i % 8));
- if (i && (((i + 1) % 8) == 0))
- byteIndex++;
- }
- if (stringLength % 8)
- this.shiftRight(8 - (stringLength % 8));
- this.bitsCount = stringLength;
- }
- fromUint32(uint32) {
- this.buffer = new ArrayBuffer(4);
- this.view = new Uint8Array(this.buffer);
- const value = new Uint32Array([uint32]);
- const view = new Uint8Array(value.buffer);
- for (let i = 3; i >= 0; i--)
- this.view[i] = view[3 - i];
- this.bitsCount = 32;
- }
- toString(start, length) {
- if (start == null) {
- start = 0;
- }
- if ((start >= this.view.length) || (start < 0)) {
- start = 0;
- }
- if (length == null) {
- length = this.view.length - start;
- }
- if ((length >= this.view.length) || (length < 0)) {
- length = this.view.length - start;
- }
- const result = [];
- for (let i = start; i < (start + length); i++) {
- result.push(bit_1.bitsToStringArray[this.view[i]]);
- }
- return result.join("").substring((this.view.length << 3) - this.bitsCount);
- }
- shiftRight(shift, needShrink = true) {
- if (this.view.length == 0) {
- return;
- }
- if ((shift < 0) || (shift > 8)) {
- throw new Error("The \"shift\" parameter must be in range 0-8");
- }
- if (shift > this.bitsCount) {
- throw new Error("The \"shift\" parameter can not be bigger than \"this.bitsCount\"");
- }
- const shiftMask = 0xFF >> (8 - shift);
- this.view[this.view.length - 1] >>= shift;
- for (let i = (this.view.length - 2); i >= 0; i--) {
- this.view[i + 1] |= (this.view[i] & shiftMask) << (8 - shift);
- this.view[i] >>= shift;
- }
- this.bitsCount -= shift;
- if (this.bitsCount == 0) {
- this.clear();
- }
- if (needShrink) {
- this.shrink();
- }
- }
- shiftLeft(shift) {
- if (this.view.length == 0) {
- return;
- }
- if ((shift < 0) || (shift > 8)) {
- throw new Error("The \"shift\" parameter must be in range 0-8");
- }
- if (shift > this.bitsCount) {
- throw new Error("The \"shift\" parameter can not be bigger than \"this.bitsCount\"");
- }
- const bitsOffset = this.bitsCount & 0x07;
- if (bitsOffset > shift) {
- this.view[0] &= 0xFF >> (bitsOffset + shift);
- }
- else {
- const view = this.view.slice(1);
- view[0] &= 0xFF >> (shift - bitsOffset);
- this.buffer = view.buffer;
- this.view = view;
- }
- this.bitsCount -= shift;
- if (this.bitsCount == 0) {
- this.clear();
- }
- }
- slice(start = 0, end = 0) {
- let valueShift = 0;
- if (this.bitsCount % 8) {
- valueShift = (8 - (this.bitsCount % 8));
- }
- start += valueShift;
- end += valueShift;
- const maxEnd = (this.view.length << 3) - 1;
- if ((start < 0) || (start > maxEnd)) {
- return new BitStream();
- }
- if (!end) {
- end = maxEnd;
- }
- if ((end < 0) || (end > maxEnd)) {
- return new BitStream();
- }
- if ((end - start + 1) > this.bitsCount) {
- return new BitStream();
- }
- const startIndex = start >> 3;
- const startOffset = start & 0x07;
- const endIndex = end >> 3;
- const endOffset = end & 0x07;
- const bitsLength = ((endIndex - startIndex) == 0) ? 1 : (endIndex - startIndex + 1);
- const result = new BitStream({
- buffer: this.buffer.slice(startIndex, startIndex + bitsLength),
- bitsCount: bitsLength << 3,
- });
- result.view[0] &= (0xFF >> startOffset);
- result.view[bitsLength] &= (0xFF << (7 - endOffset));
- if (7 - endOffset) {
- result.shiftRight(7 - endOffset, false);
- }
- result.bitsCount = (end - start + 1);
- result.shrink();
- return result;
- }
- copy(start = 0, length = 0) {
- const maxEnd = (this.view.length << 3) - 1;
- if ((start < 0) || (start > maxEnd)) {
- return new BitStream();
- }
- if (!length) {
- length = (this.view.length << 3) - start - 1;
- }
- if (length > this.bitsCount) {
- return new BitStream();
- }
- return this.slice(start, start + length - 1);
- }
- shrink() {
- const currentLength = (this.bitsCount >> 3) + ((this.bitsCount % 8) ? 1 : 0);
- if (currentLength < this.view.length) {
- const view = this.view.slice(this.view.length - currentLength, (this.view.length - currentLength) + currentLength);
- this.view = view;
- this.buffer = view.buffer;
- }
- }
- reverseBytes() {
- for (let i = 0; i < this.view.length; i++) {
- this.view[i] = ((this.view[i] * 0x0802 & 0x22110) | (this.view[i] * 0x8020 & 0x88440)) * 0x10101 >> 16;
- }
- if (this.bitsCount % 8) {
- const currentLength = (this.bitsCount >> 3) + ((this.bitsCount % 8) ? 1 : 0);
- this.view[this.view.length - currentLength] >>= (8 - (this.bitsCount & 0x07));
- }
- }
- reverseValue() {
- const initialValue = this.toString();
- const initialValueLength = initialValue.length;
- const reversedValue = new Array(initialValueLength);
- for (let i = 0; i < initialValueLength; i++) {
- reversedValue[initialValueLength - 1 - i] = initialValue[i];
- }
- this.fromString(reversedValue.join(""));
- }
- getNumberValue() {
- const byteLength = (this.view.length - 1);
- if (byteLength > 3) {
- return (-1);
- }
- if (byteLength == (-1)) {
- return 0;
- }
- const value = new Uint32Array(1);
- const view = new Uint8Array(value.buffer);
- for (let i = byteLength; i >= 0; i--) {
- view[byteLength - i] = this.view[i];
- }
- return value[0];
- }
- findPattern(pattern, start, length, backward) {
- const stringStream = new byte_stream_1.ByteStream({
- string: this.toString(),
- });
- const stringPattern = new byte_stream_1.ByteStream({
- string: pattern.toString()
- });
- return stringStream.findPattern(stringPattern, start, length, backward);
- }
- findFirstIn(patterns, start, length, backward) {
- const stringStream = new byte_stream_1.ByteStream({
- string: this.toString(),
- });
- const stringPatterns = new Array(patterns.length);
- for (let i = 0; i < patterns.length; i++) {
- stringPatterns[i] = new byte_stream_1.ByteStream({
- string: patterns[i].toString()
- });
- }
- return stringStream.findFirstIn(stringPatterns, start, length, backward);
- }
- findAllIn(patterns, start, length) {
- const stringStream = new byte_stream_1.ByteStream({
- string: this.toString()
- });
- const stringPatterns = new Array(patterns.length);
- for (let i = 0; i < patterns.length; i++) {
- stringPatterns[i] = new byte_stream_1.ByteStream({
- string: patterns[i].toString()
- });
- }
- return stringStream.findAllIn(stringPatterns, start, length);
- }
- findAllPatternIn(pattern, start, length) {
- const stringStream = new byte_stream_1.ByteStream({
- string: this.toString()
- });
- const stringPattern = new byte_stream_1.ByteStream({
- string: pattern.toString()
- });
- return stringStream.findAllPatternIn(stringPattern, start, length);
- }
- findFirstNotIn(patterns, start, length, backward) {
- const stringStream = new byte_stream_1.ByteStream({
- string: this.toString()
- });
- const stringPatterns = new Array(patterns.length);
- for (let i = 0; i < patterns.length; i++) {
- stringPatterns[i] = new byte_stream_1.ByteStream({
- string: patterns[i].toString()
- });
- }
- return stringStream.findFirstNotIn(stringPatterns, start, length, backward);
- }
- findAllNotIn(patterns, start, length) {
- const stringStream = new byte_stream_1.ByteStream({
- string: this.toString()
- });
- const stringPatterns = new Array(patterns.length);
- for (let i = 0; i < patterns.length; i++) {
- stringPatterns[i] = new byte_stream_1.ByteStream({
- string: patterns[i].toString()
- });
- }
- return stringStream.findAllNotIn(stringPatterns, start, length);
- }
- findFirstSequence(patterns, start, length, backward) {
- const stringStream = new byte_stream_1.ByteStream({
- string: this.toString()
- });
- const stringPatterns = new Array(patterns.length);
- for (let i = 0; i < patterns.length; i++) {
- stringPatterns[i] = new byte_stream_1.ByteStream({
- string: patterns[i].toString()
- });
- }
- return stringStream.findFirstSequence(stringPatterns, start, length, backward);
- }
- findAllSequences(patterns, start, length) {
- const stringStream = new byte_stream_1.ByteStream({
- string: this.toString()
- });
- const stringPatterns = new Array(patterns.length);
- for (let i = 0; i < patterns.length; i++) {
- stringPatterns[i] = new byte_stream_1.ByteStream({
- string: patterns[i].toString()
- });
- }
- return stringStream.findAllSequences(stringPatterns, start, length);
- }
- findPairedPatterns(leftPattern, rightPattern, start, length) {
- const stringStream = new byte_stream_1.ByteStream({
- string: this.toString()
- });
- const stringLeftPattern = new byte_stream_1.ByteStream({
- string: leftPattern.toString()
- });
- const stringRightPattern = new byte_stream_1.ByteStream({
- string: rightPattern.toString()
- });
- return stringStream.findPairedPatterns(stringLeftPattern, stringRightPattern, start, length);
- }
- findPairedArrays(inputLeftPatterns, inputRightPatterns, start, length) {
- const stringStream = new byte_stream_1.ByteStream({
- string: this.toString()
- });
- const stringLeftPatterns = new Array(inputLeftPatterns.length);
- for (let i = 0; i < inputLeftPatterns.length; i++) {
- stringLeftPatterns[i] = new byte_stream_1.ByteStream({
- string: inputLeftPatterns[i].toString()
- });
- }
- const stringRightPatterns = new Array(inputRightPatterns.length);
- for (let i = 0; i < inputRightPatterns.length; i++) {
- stringRightPatterns[i] = new byte_stream_1.ByteStream({
- string: inputRightPatterns[i].toString()
- });
- }
- return stringStream.findPairedArrays(stringLeftPatterns, stringRightPatterns, start, length);
- }
- replacePattern(searchPattern, replacePattern, start, length) {
- const stringStream = new byte_stream_1.ByteStream({
- string: this.toString()
- });
- const stringSearchPattern = new byte_stream_1.ByteStream({
- string: searchPattern.toString()
- });
- const stringReplacePattern = new byte_stream_1.ByteStream({
- string: replacePattern.toString()
- });
- if (stringStream.replacePattern(stringSearchPattern, stringReplacePattern, start, length)) {
- this.fromString(stringStream.toString());
- return true;
- }
- return false;
- }
- skipPatterns(patterns, start, length, backward) {
- const stringStream = new byte_stream_1.ByteStream({
- string: this.toString()
- });
- const stringPatterns = new Array(patterns.length);
- for (let i = 0; i < patterns.length; i++) {
- stringPatterns[i] = new byte_stream_1.ByteStream({
- string: patterns[i].toString()
- });
- }
- return stringStream.skipPatterns(stringPatterns, start, length, backward);
- }
- skipNotPatterns(patterns, start, length, backward) {
- const stringStream = new byte_stream_1.ByteStream({
- string: this.toString()
- });
- const stringPatterns = new Array(patterns.length);
- for (let i = 0; i < patterns.length; i++) {
- stringPatterns[i] = new byte_stream_1.ByteStream({
- string: patterns[i].toString()
- });
- }
- return stringStream.skipNotPatterns(stringPatterns, start, length, backward);
- }
- append(stream) {
- this.fromString([
- this.toString(),
- stream.toString()
- ].join(""));
- }
- }
- exports.BitStream = BitStream;
|