aboutsummaryrefslogtreecommitdiffstats
path: root/src/lib/bitstream/bitstream.h
diff options
context:
space:
mode:
authorDaniil Cherednik <dan.cherednik@gmail.com>2024-06-17 20:12:45 +0000
committerDaniil Cherednik <dan.cherednik@gmail.com>2024-06-17 22:21:52 +0200
commit23a4e5f1dd7ce24f65a2af0598d1f92af4b5c424 (patch)
tree8a259ca8363c5b15fd3605b760518cb37e6ac63c /src/lib/bitstream/bitstream.h
parent73dbd1609445a0142e1e138b6b44ec6d1925cbb8 (diff)
downloadatracdenc-23a4e5f1dd7ce24f65a2af0598d1f92af4b5c424.tar.gz
[refactoring] move some libraries in to library directory
Diffstat (limited to 'src/lib/bitstream/bitstream.h')
-rw-r--r--src/lib/bitstream/bitstream.h47
1 files changed, 47 insertions, 0 deletions
diff --git a/src/lib/bitstream/bitstream.h b/src/lib/bitstream/bitstream.h
new file mode 100644
index 0000000..cfba790
--- /dev/null
+++ b/src/lib/bitstream/bitstream.h
@@ -0,0 +1,47 @@
+/*
+ * This file is part of AtracDEnc.
+ *
+ * AtracDEnc is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * AtracDEnc 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
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with AtracDEnc; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
+ */
+
+#pragma once
+#include <vector>
+
+#include <iostream>
+
+namespace NBitStream {
+
+static inline int MakeSign(int val, unsigned bits) {
+ unsigned shift = 8 * sizeof(int) - bits;
+ union { unsigned u; int s; } v = { (unsigned) val << shift };
+ return v.s >> shift;
+}
+
+class TBitStream {
+ std::vector<char> Buf;
+ int BitsUsed = 0;
+ int ReadPos = 0;
+ public:
+ TBitStream(const char* buf, int size);
+ TBitStream();
+ void Write(uint32_t val, int n);
+ uint32_t Read(int n);
+ unsigned long long GetSizeInBits() const;
+ uint32_t GetBufSize() const { return Buf.size(); };
+ const std::vector<char>& GetBytes() const {
+ return Buf;
+ }
+};
+} //NBitStream