#include "boo/audiodev/MIDIDecoder.hpp"
#include "MIDICommon.hpp"
#include <memory>
#include <algorithm>

namespace boo {

static inline uint8_t clamp7(uint8_t val) { return std::max(0, std::min(127, int(val))); }

bool MIDIDecoder::_readContinuedValue(std::vector<uint8_t>::const_iterator& it,
                                      std::vector<uint8_t>::const_iterator end, uint32_t& valOut) {
  uint8_t a = *it++;
  valOut = a & 0x7f;

  if (a & 0x80) {
    if (it == end)
      return false;
    valOut <<= 7;
    a = *it++;
    valOut |= a & 0x7f;

    if (a & 0x80) {
      if (it == end)
        return false;
      valOut <<= 7;
      a = *it++;
      valOut |= a & 0x7f;
    }
  }

  return true;
}

std::vector<uint8_t>::const_iterator MIDIDecoder::receiveBytes(std::vector<uint8_t>::const_iterator begin,
                                                               std::vector<uint8_t>::const_iterator end) {
  std::vector<uint8_t>::const_iterator it = begin;
  while (it != end) {
    uint8_t a = *it++;
    uint8_t b;
    if (a & 0x80)
      m_status = a;
    else
      it--;

    if (m_status == 0xff) {
      /* Meta events (ignored for now) */
      if (it == end)
        return begin;
      a = *it++;

      uint32_t length;
      _readContinuedValue(it, end, length);
      it += length;
    } else {
      uint8_t chan = m_status & 0xf;
      switch (Status(m_status & 0xf0)) {
      case Status::NoteOff: {
        if (it == end)
          return begin;
        a = *it++;
        if (it == end)
          return begin;
        b = *it++;
        m_out.noteOff(chan, clamp7(a), clamp7(b));
        break;
      }
      case Status::NoteOn: {
        if (it == end)
          return begin;
        a = *it++;
        if (it == end)
          return begin;
        b = *it++;
        m_out.noteOn(chan, clamp7(a), clamp7(b));
        break;
      }
      case Status::NotePressure: {
        if (it == end)
          return begin;
        a = *it++;
        if (it == end)
          return begin;
        b = *it++;
        m_out.notePressure(chan, clamp7(a), clamp7(b));
        break;
      }
      case Status::ControlChange: {
        if (it == end)
          return begin;
        a = *it++;
        if (it == end)
          return begin;
        b = *it++;
        m_out.controlChange(chan, clamp7(a), clamp7(b));
        break;
      }
      case Status::ProgramChange: {
        if (it == end)
          return begin;
        a = *it++;
        m_out.programChange(chan, clamp7(a));
        break;
      }
      case Status::ChannelPressure: {
        if (it == end)
          return begin;
        a = *it++;
        m_out.channelPressure(chan, clamp7(a));
        break;
      }
      case Status::PitchBend: {
        if (it == end)
          return begin;
        a = *it++;
        if (it == end)
          return begin;
        b = *it++;
        m_out.pitchBend(chan, clamp7(b) * 128 + clamp7(a));
        break;
      }
      case Status::SysEx: {
        switch (Status(m_status & 0xff)) {
        case Status::SysEx: {
          uint32_t len;
          if (!_readContinuedValue(it, end, len) || end - it < len)
            return begin;
          m_out.sysex(&*it, len);
          break;
        }
        case Status::TimecodeQuarterFrame: {
          if (it == end)
            return begin;
          a = *it++;
          m_out.timeCodeQuarterFrame(a >> 4 & 0x7, a & 0xf);
          break;
        }
        case Status::SongPositionPointer: {
          if (it == end)
            return begin;
          a = *it++;
          if (it == end)
            return begin;
          b = *it++;
          m_out.songPositionPointer(clamp7(b) * 128 + clamp7(a));
          break;
        }
        case Status::SongSelect: {
          if (it == end)
            return begin;
          a = *it++;
          m_out.songSelect(clamp7(a));
          break;
        }
        case Status::TuneRequest:
          m_out.tuneRequest();
          break;
        case Status::Start:
          m_out.startSeq();
          break;
        case Status::Continue:
          m_out.continueSeq();
          break;
        case Status::Stop:
          m_out.stopSeq();
          break;
        case Status::Reset:
          m_out.reset();
          break;
        case Status::SysExTerm:
        case Status::TimingClock:
        case Status::ActiveSensing:
        default:
          break;
        }
        break;
      }
      default:
        break;
      }
    }
  }
  return it;
}

} // namespace boo