aboutsummaryrefslogtreecommitdiffstats
path: root/src/lib/bs_encode/encode.h
diff options
context:
space:
mode:
authorDaniil Cherednik <dan.cherednik@gmail.com>2025-05-31 22:39:38 +0200
committerDaniil Cherednik <dan.cherednik@gmail.com>2025-05-31 22:39:38 +0200
commit1c7f2f821fb965af468cdf2a14df3ff75cc1c352 (patch)
tree1bc92237122b75c67afc326af207cf3cc9eb3d6c /src/lib/bs_encode/encode.h
parent272af27a3d148bd13e8f15640e53ca70c64ccb9b (diff)
parent6dfc60e9d4791c3385908c61ad75c4a0093ea1eb (diff)
downloadatracdenc-1c7f2f821fb965af468cdf2a14df3ff75cc1c352.tar.gz
Merge branch 'at3plus-dev'
It looks like we are able to encode ATRAC3PLUS compatible bitstream so we can merge at3p development branch in to the main branch. Current limitation for AT3P mode: - Only 352 Kbps (proper bit allocation and some psychoacoustic must be implemented) - GHA sometime works with error (but huge bitrate hide it) - No VLC, VQ, delta encoding - No noise substitution - No gain control - No window shape switching
Diffstat (limited to 'src/lib/bs_encode/encode.h')
-rw-r--r--src/lib/bs_encode/encode.h71
1 files changed, 71 insertions, 0 deletions
diff --git a/src/lib/bs_encode/encode.h b/src/lib/bs_encode/encode.h
new file mode 100644
index 0000000..3302d10
--- /dev/null
+++ b/src/lib/bs_encode/encode.h
@@ -0,0 +1,71 @@
+/*
+ * 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 <memory>
+#include <vector>
+#include <functional>
+
+namespace NBitStream {
+class TBitStream;
+}
+
+namespace NAtracDEnc {
+
+class TBitAllocHandler {
+public:
+ void Start(size_t targetBits, float minLambda, float maxLambda) noexcept;
+ float Continue() noexcept;
+ bool Check(size_t gitBits) const noexcept;
+ void Submit(size_t gotBits) noexcept;
+
+ // Returns consumption of all previous encoded parts (except part from this method called)
+ uint32_t GetCurGlobalConsumption() const noexcept;
+};
+
+class IBitStreamPartEncoder {
+public:
+ using TPtr = std::unique_ptr<IBitStreamPartEncoder>;
+ enum class EStatus {
+ Ok, // Ok, go to the next stage
+ Repeat, // Repeat from first stage
+ };
+
+ virtual ~IBitStreamPartEncoder() = default;
+ virtual EStatus Encode(void* frameData, TBitAllocHandler& ba) = 0;
+ virtual void Dump(NBitStream::TBitStream& bs) = 0;
+ virtual void Reset() noexcept {};
+ virtual uint32_t GetConsumption() const noexcept = 0;
+};
+
+class TBitStreamEncoder {
+public:
+ class TImpl;
+ explicit TBitStreamEncoder(std::vector<IBitStreamPartEncoder::TPtr>&& encoders);
+ ~TBitStreamEncoder();
+
+ void Do(void* frameData, NBitStream::TBitStream& bs);
+ TBitStreamEncoder(const TBitStreamEncoder&) = delete;
+ TBitStreamEncoder& operator=(const TBitStreamEncoder&) = delete;
+private:
+ std::vector<IBitStreamPartEncoder::TPtr> Encoders;
+ TImpl* Impl;
+};
+
+}