Move implementation details into corresponding dirs (#52)HEADmaster

1 files changed, 95 insertions, 0 deletions

diff --git a/src/atrac/at3/atrac3_bitstream.h b/src/atrac/at3/atrac3_bitstream.h
new file mode 100644
index 0000000..17a8548
--- /dev/null
+++ b/src/atrac/at3/atrac3_bitstream.h
@@ -0,0 +1,95 @@
+/*
+ * 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 "atrac3.h"
+#include <compressed_io.h>
+#include <atrac/atrac_scale.h>
+#include <vector>
+#include <utility>
+
+namespace NBitStream {
+    class TBitStream;
+}
+
+namespace NAtracDEnc {
+namespace NAtrac3 {
+
+struct TTonalBlock {
+    TTonalBlock(const TAtrac3Data::TTonalVal* valPtr, const TScaledBlock& scaledBlock)
+        : ValPtr(valPtr)
+        , ScaledBlock(scaledBlock)
+    {}
+    const TAtrac3Data::TTonalVal* ValPtr = nullptr;
+    TScaledBlock ScaledBlock;
+};
+
+class TAtrac3BitStreamWriter {
+public:
+    struct TSingleChannelElement {
+        TAtrac3Data::SubbandInfo SubbandInfo;
+        std::vector<TTonalBlock> TonalBlocks;
+        std::vector<TScaledBlock> ScaledBlocks;
+        float Loudness;
+    };
+private:
+    static std::vector<float> ATH;
+
+    struct TTonalComponentsSubGroup {
+        std::vector<uint8_t> SubGroupMap;
+        std::vector<const TTonalBlock*> SubGroupPtr;
+    };
+    ICompressedOutput* Container;
+    const TContainerParams Params;
+    const uint32_t BfuIdxConst;
+    std::vector<char> OutBuffer;
+
+    uint32_t CLCEnc(const uint32_t selector, const int mantissas[TAtrac3Data::MaxSpecsPerBlock],
+                    const uint32_t blockSize, NBitStream::TBitStream* bitStream);
+
+    uint32_t VLCEnc(const uint32_t selector, const int mantissas[TAtrac3Data::MaxSpecsPerBlock],
+                    const uint32_t blockSize, NBitStream::TBitStream* bitStream);
+
+    std::vector<uint32_t> CalcBitsAllocation(const std::vector<TScaledBlock>& scaledBlocks,
+                                             uint32_t bfuNum, float spread, float shift, float loudness);
+
+    std::pair<uint8_t, std::vector<uint32_t>> CreateAllocation(const TSingleChannelElement& sce,
+                                                               uint16_t targetBits, int mt[TAtrac3Data::MaxSpecs], float laudness);
+
+    std::pair<uint8_t, uint32_t> CalcSpecsBitsConsumption(const TSingleChannelElement& sce,
+                                                          const std::vector<uint32_t>& precisionPerEachBlocks,
+                                                          int* mantisas, std::vector<float>& energyErr);
+
+    void EncodeSpecs(const TSingleChannelElement& sce, NBitStream::TBitStream* bitStream,
+                     const std::pair<uint8_t, std::vector<uint32_t>>&, const int mt[TAtrac3Data::MaxSpecs]);
+
+    uint8_t GroupTonalComponents(const std::vector<TTonalBlock>& tonalComponents,
+                                 const std::vector<uint32_t>& allocTable,
+                                 TTonalComponentsSubGroup groups[64]);
+
+    uint16_t EncodeTonalComponents(const TSingleChannelElement& sce,
+                                   const std::vector<uint32_t>& allocTable,
+                                   NBitStream::TBitStream* bitStream);
+public:
+    TAtrac3BitStreamWriter(ICompressedOutput* container, const TContainerParams& params, uint32_t bfuIdxConst);
+
+    void WriteSoundUnit(const std::vector<TSingleChannelElement>& singleChannelElements, float laudness);
+};
+
+} // namespace NAtrac3
+} // namespace NAtracDEnc