// derived from org.gagravarr.opus.OpusFile

/*

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *

 *     http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */

lib 1400511 // VorbisJava (to unpack ogg file)

lib 1400502 // Concentus 1.0 (to decode OPUS audio in ogg file)

import org.gagravarr.ogg.*;

import org.gagravarr.ogg.audio.*;

import org.gagravarr.opus.*;

import static org.gagravarr.ogg.OggStreamIdentifier.OggStreamType;

import org.concentus.*;

sclass OpusFromStream implements OggAudioStream, OggAudioHeaders, Closeable {

  InputStream in;

  OggStreamPacketReader r;

  int sid = -1;

  int channels;

  OpusDecoder decoder;

  bool printResets;

  OpusInfo info;

  OpusTags tags;

  // channels = 1 or 2 (mono/stereo)

  *(InputStream *in, int *channels) throws IOException, OpusException {

    decoder = new OpusDecoder(48000, channels);

    r = new OggStreamPacketReader(in);

  }

  bool initialized() { ret tags != null; }

  bool initStep() throws IOException {

    OggPacket p = null;

    // find first packet (info)

    if (sid < 0) {

      while ((p = getNextPacket()) != null) {

        if (p.isBeginningOfStream() && p.getData().length > 10) {

          if (OpusPacketFactory.isOpusStream(p)) {

            sid = p.getSid();

            info = (OpusInfo) OpusPacketFactory.create(p);

            ifdef OpusFromStream_debug

              printVars OpusFromStream(+sid, +info);

            endifdef

            break;

          }

        }

      }

    }

    // find second packet (tags)

    if (tags == null) {  

      if ((p = getNextPacketWithSid(sid)) != null) {

        tags = (OpusTags) OpusPacketFactory.create(p);

        ifdef OpusFromStream_debug

          printVars OpusFromStream(+tags);

        endifdef

      }

    }

    ret initialized();

  }

  public OpusAudioData getNextAudioPacket() throws IOException {

    if (!initStep()) null;

    OggPacket p = null;

    OpusPacket op = null;

    while ((p = getNextPacketWithSid(sid)) != null) {

      op = OpusPacketFactory.create(p);

      if (op cast OpusAudioData) ret op;

      else

        warn("Skipping non audio packet " + op + " mid audio stream");

    }

    null;

  }

  /**

   * Returns the Ogg Stream ID

   */

  public int getSid() {

    ret sid;

  }

  /**

   * This is an Opus file

   */

  public OggStreamType getType() {

    ret OggStreamIdentifier.OPUS_AUDIO;

  }

  public OpusInfo getInfo() {

    return info;

  }

  public OpusTags getTags() {

    return tags;

  }

  public void skipToGranule(long granulePosition) throws IOException {

    unimplemented();

  }

  /**

   * Opus doesn't have setup headers, so this is always null

   */

  public OggAudioSetupHeader getSetup() {

    null;

  }

  close {

    dispose r;

    dispose in;

  }

  // return some audio samples. null if none available

  short[] nextSamples aka moreSamples() throws OpusException, IOException {

    OpusAudioData data;

    while ((data = getNextAudioPacket()) != null) {

      byte[] bytes = data.getData();

      if (empty(bytes)) continue;

      // decode packet

      short[] pcm = new[5760*channels]; // max opus frame size

      int nSamples = decoder.decode(bytes, 0, l(bytes), pcm, 0, l(pcm), false);

      ret subShortArray(pcm, 0, nSamples);

    }

    null;

  }

  public OggPacket getNextPacket() throws IOException {

    in.mark(65536);

    OggPacket p;

    try {

      p = r.getNextPacket();

      //if (p != null && printResets) print("OpusFromStream: Got packet");

    } catch (EOFException e) {

      if (printResets) print("OpusFromStream: EOFException, resetting");

      ifdef OpusFromStream_debug

        print("OpusFromStream: EOFException, resetting");

      endifdef

      in.reset();

      null;

    }

    if (p == null) {

      if (printResets) print("OpusFromStream: packet null, resetting");

      ifdef OpusFromStream_debug

        print("OpusFromStream: packet null, resetting");

      endifdef

      in.reset();

    }

    ifdef OpusFromStream_debug

      printVars OpusFromStream(packet := p, +in);

    endifdef

    ret p;

  }

  /**

     * Returns the next packet with the given SID (Stream ID), or

     *  null if no more packets remain.

     * Any packets from other streams will be silently discarded.

     */

  public OggPacket getNextPacketWithSid(int sid) throws IOException {

    OggPacket p = null;

    while ((p = getNextPacket()) != null) {

      if (p.getSid() == sid) ret p;

    }

    null;

  }

  // e.g. for mark/reset

  static int bufferSizeNeeded() { ret 65536; }

}