move to new cspot

components/spotify/cspot/bell/main/io/EncodedAudioStream.cpp

@@ -0,0 +1,175 @@
+#include "EncodedAudioStream.h"
+
+#include <iostream>
+using namespace bell;
+
+EncodedAudioStream::EncodedAudioStream() {
+  bell::decodersInstance->ensureAAC();
+  bell::decodersInstance->ensureMP3();
+  inputBuffer = std::vector<uint8_t>(AAC_READBUF_SIZE * 4);
+  outputBuffer = std::vector<short>(AAC_MAX_NCHANS * AAC_MAX_NSAMPS * 4 * 4);
+  decodePtr = inputBuffer.data();
+}
+
+EncodedAudioStream::~EncodedAudioStream() {
+  this->innerStream->close();
+}
+
+void EncodedAudioStream::openWithStream(
+    std::unique_ptr<bell::ByteStream> byteStream) {
+  if (this->innerStream) {
+    this->innerStream->close();
+  }
+  this->innerStream = std::move(byteStream);
+  this->guessDataFormat();
+}
+
+bool EncodedAudioStream::vectorStartsWith(std::vector<uint8_t>& vec,
+                                          std::vector<uint8_t>& start) {
+  if (vec.size() < start.size()) {
+    return false;
+  }
+
+  for (int i = 0; i < start.size(); i++) {
+    if (vec[i] != start[i]) {
+      return false;
+    }
+  }
+
+  return true;
+}
+
+size_t EncodedAudioStream::decodeFrame(uint8_t* dst) {
+  if (innerStream->size() != 0 &&
+      innerStream->position() >= innerStream->size()) {
+    return 0;
+  }
+
+  switch (this->codec) {
+    case AudioCodec::AAC:
+      return decodeFrameAAC(dst);
+      break;
+    case AudioCodec::MP3:
+      return decodeFrameMp3(dst);
+      break;
+    default:
+      break;
+  }
+
+  return 0;
+}
+
+size_t EncodedAudioStream::decodeFrameMp3(uint8_t* dst) {
+  size_t writtenBytes = 0;
+  int bufSize = MP3_READBUF_SIZE;
+
+  int readBytes = innerStream->read(inputBuffer.data() + bytesInBuffer,
+                                    bufSize - bytesInBuffer);
+  if (readBytes > 0) {
+    bytesInBuffer += readBytes;
+    decodePtr = inputBuffer.data();
+    offset = MP3FindSyncWord(inputBuffer.data(), bytesInBuffer);
+
+    if (offset != -1) {
+      bytesInBuffer -= offset;
+      decodePtr += offset;
+
+      int decodeStatus =
+          MP3Decode(bell::decodersInstance->mp3Decoder, &decodePtr,
+                    &bytesInBuffer, outputBuffer.data(), 0);
+      MP3GetLastFrameInfo(bell::decodersInstance->mp3Decoder, &mp3FrameInfo);
+      if (decodeStatus == ERR_MP3_NONE) {
+        decodedSampleRate = mp3FrameInfo.samprate;
+        writtenBytes =
+            (mp3FrameInfo.bitsPerSample / 8) * mp3FrameInfo.outputSamps;
+
+        memcpy(dst, outputBuffer.data(), writtenBytes);
+
+      } else {
+        BELL_LOG(info, TAG, "Error in frame, moving two bytes %d",
+                 decodeStatus);
+        decodePtr += 1;
+        bytesInBuffer -= 1;
+      }
+    } else {
+      BELL_LOG(info, TAG, "Unexpected error in data, skipping a word");
+      decodePtr += 3800;
+      bytesInBuffer -= 3800;
+    }
+
+    memmove(inputBuffer.data(), decodePtr, bytesInBuffer);
+  }
+  return writtenBytes;
+}
+
+bool EncodedAudioStream::isReadable() {
+  return this->codec != AudioCodec::NONE;
+}
+
+size_t EncodedAudioStream::decodeFrameAAC(uint8_t* dst) {
+  size_t writtenBytes = 0;
+  auto bufSize = AAC_READBUF_SIZE;
+
+  int readBytes = innerStream->read(inputBuffer.data() + bytesInBuffer,
+                                    bufSize - bytesInBuffer);
+  if (readBytes > 0) {
+    bytesInBuffer += readBytes;
+    decodePtr = inputBuffer.data();
+    offset = AACFindSyncWord(inputBuffer.data(), bytesInBuffer);
+
+    if (offset != -1) {
+      bytesInBuffer -= offset;
+      decodePtr += offset;
+
+      int decodeStatus =
+          AACDecode(bell::decodersInstance->aacDecoder, &decodePtr,
+                    &bytesInBuffer, outputBuffer.data());
+      AACGetLastFrameInfo(bell::decodersInstance->aacDecoder, &aacFrameInfo);
+      if (decodeStatus == ERR_AAC_NONE) {
+        decodedSampleRate = aacFrameInfo.sampRateOut;
+        writtenBytes =
+            (aacFrameInfo.bitsPerSample / 8) * aacFrameInfo.outputSamps;
+
+        memcpy(dst, outputBuffer.data(), writtenBytes);
+
+      } else {
+        BELL_LOG(info, TAG, "Error in frame, moving two bytes %d",
+                 decodeStatus);
+        decodePtr += 1;
+        bytesInBuffer -= 1;
+      }
+    } else {
+      BELL_LOG(info, TAG, "Unexpected error in data, skipping a word");
+      decodePtr += 3800;
+      bytesInBuffer -= 3800;
+    }
+
+    memmove(inputBuffer.data(), decodePtr, bytesInBuffer);
+  }
+  return writtenBytes;
+}
+
+void EncodedAudioStream::guessDataFormat() {
+  // Read 14 bytes from the stream
+  this->innerStream->read(inputBuffer.data(), 14);
+  bytesInBuffer = 14;
+
+  BELL_LOG(info, TAG, "No codec set, reading secret bytes");
+
+  if (vectorStartsWith(inputBuffer, this->mp3MagicBytesIdc) ||
+      vectorStartsWith(inputBuffer, this->mp3MagicBytesUntagged)) {
+    BELL_LOG(info, TAG, "Detected MP3");
+    codec = AudioCodec::MP3;
+  } else if (vectorStartsWith(inputBuffer, this->aacMagicBytes) ||
+             vectorStartsWith(inputBuffer, this->aacMagicBytes4)) {
+    BELL_LOG(info, TAG, "Detected AAC");
+    codec = AudioCodec::AAC;
+  }
+
+  if (codec == AudioCodec::NONE) {
+    throw std::runtime_error("Codec not supported");
+  }
+}
+
+void EncodedAudioStream::readFully(uint8_t* dst, size_t nbytes) {
+}