gecko/content/media/video/src/nsChannelReader.cpp

/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim:set ts=2 sw=2 sts=2 et cindent: */
/* ***** BEGIN LICENSE BLOCK *****
 * Version: MPL 1.1/GPL 2.0/LGPL 2.1
 *
 * The contents of this file are subject to the Mozilla Public License Version
 * 1.1 (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.mozilla.org/MPL/
 *
 * Software distributed under the License is distributed on an "AS IS" basis,
 * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
 * for the specific language governing rights and limitations under the
 * License.
 *
 * The Original Code is Mozilla code.
 *
 * The Initial Developer of the Original Code is the Mozilla Corporation.
 * Portions created by the Initial Developer are Copyright (C) 2007
 * the Initial Developer. All Rights Reserved.
 *
 * Contributor(s):
 *  Chris Double <chris.double@double.co.nz>
 *
 * Alternatively, the contents of this file may be used under the terms of
 * either the GNU General Public License Version 2 or later (the "GPL"), or
 * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
 * in which case the provisions of the GPL or the LGPL are applicable instead
 * of those above. If you wish to allow use of your version of this file only
 * under the terms of either the GPL or the LGPL, and not to allow others to
 * use your version of this file under the terms of the MPL, indicate your
 * decision by deleting the provisions above and replace them with the notice
 * and other provisions required by the GPL or the LGPL. If you do not delete
 * the provisions above, a recipient may use your version of this file under
 * the terms of any one of the MPL, the GPL or the LGPL.
 *
 * ***** END LICENSE BLOCK ***** */
#include "nsAString.h"
#include "nsThreadUtils.h"
#include "nsNetUtil.h"
#include "prlog.h"
#include "nsOggDecoder.h"
#include "nsChannelReader.h"

nsChannelToPipeListener::nsChannelToPipeListener(nsOggDecoder* aDecoder) :
  mDecoder(aDecoder),
  mIntervalStart(0),
  mIntervalEnd(0),
  mTotalBytes(0)
{
}

nsresult nsChannelToPipeListener::Init() 
{
  nsresult rv = NS_NewPipe(getter_AddRefs(mInput), 
                           getter_AddRefs(mOutput),
                           0, 
                           PR_UINT32_MAX);
  NS_ENSURE_SUCCESS(rv, rv);

  return NS_OK;
}

void nsChannelToPipeListener::Stop()
{
  mDecoder = nsnull;
  mInput = nsnull;
  mOutput = nsnull;
  mDecoder = nsnull;
}

double nsChannelToPipeListener::BytesPerSecond() const
{
  return mTotalBytes / ((PR_IntervalToMilliseconds(mIntervalEnd-mIntervalStart)) / 1000.0);
}

void nsChannelToPipeListener::GetInputStream(nsIInputStream** aStream)
{
  if (aStream) {
    NS_IF_ADDREF(*aStream = mInput);
  }
}

nsresult nsChannelToPipeListener::OnStartRequest(nsIRequest* aRequest, nsISupports* aContext)
{
  mIntervalStart = PR_IntervalNow();
  mIntervalEnd = mIntervalStart;
  mTotalBytes = 0;
  mDecoder->UpdateBytesDownloaded(mTotalBytes);
  return NS_OK;
}

nsresult nsChannelToPipeListener::OnStopRequest(nsIRequest* aRequest, nsISupports* aContext, nsresult aStatus) 
{
  mOutput = nsnull;
  if (mDecoder) {
    mDecoder->ResourceLoaded();
  }
  return NS_OK;
}

nsresult nsChannelToPipeListener::OnDataAvailable(nsIRequest* aRequest, 
                                                nsISupports* aContext, 
                                                nsIInputStream* aStream,
                                                PRUint32 aOffset,
                                                PRUint32 aCount)
{
  if (mOutput) {
    PRUint32 bytes = 0;
  
    do {
      nsresult rv = mOutput->WriteFrom(aStream, aCount, &bytes);
      NS_ENSURE_SUCCESS(rv, rv);
      
      aCount -= bytes;
      mTotalBytes += bytes;
      mDecoder->UpdateBytesDownloaded(mTotalBytes);
    } while (aCount) ;
    
    nsresult rv = mOutput->Flush();
    NS_ENSURE_SUCCESS(rv, rv);
    mIntervalEnd = PR_IntervalNow();
  }
  return NS_OK;
}

NS_IMPL_ISUPPORTS2(nsChannelToPipeListener, nsIRequestObserver, nsIStreamListener)

PRUint32 nsChannelReader::Available()
{
  PRUint32 available = 0;
  mInput->Available(&available);
  return available;
}

double nsChannelReader::BytesPerSecond() const
{
  return mListener->BytesPerSecond();
}

OggPlayErrorCode nsChannelReader::initialise(int aBlock)
{
  return E_OGGPLAY_OK;
}

OggPlayErrorCode nsChannelReader::destroy()
{
  mChannel->Cancel(NS_BINDING_ABORTED);
  mChannel = nsnull;
  mInput->Close();
  mInput = nsnull;
  mListener->Stop();
  mListener = nsnull;

  return E_OGGPLAY_OK;
}

size_t nsChannelReader::io_read(char* aBuffer, size_t aCount)
{
  PRUint32 bytes = 0;
  nsresult rv = mInput->Read(aBuffer, aCount, &bytes);
  if (!NS_SUCCEEDED(rv)) {
    bytes = 0;
  }

  mCurrentPosition += bytes;
  return bytes == 0 ? static_cast<size_t>(OGGZ_ERR_SYSTEM) : bytes;
}

int nsChannelReader::io_seek(long aOffset, int aWhence)
{
  // How to implement seek in channels?
  // fseek(file, offset, whence);

  return -1;
}

long nsChannelReader::io_tell()
{
  return mCurrentPosition;
}

static OggPlayErrorCode oggplay_channel_reader_initialise(OggPlayReader* aReader, int aBlock) 
{
  nsChannelReader * me = static_cast<nsChannelReader*>(aReader);

  if (me == NULL) {
    return E_OGGPLAY_BAD_READER;
  }
  return me->initialise(aBlock);
}

static OggPlayErrorCode oggplay_channel_reader_destroy(OggPlayReader* aReader) 
{
  nsChannelReader* me = static_cast<nsChannelReader*>(aReader);
  OggPlayErrorCode result = me->destroy();
  delete me;
  return result;
}

static size_t oggplay_channel_reader_io_read(void* aReader, void* aBuffer, size_t aCount) 
{
  nsChannelReader* me = static_cast<nsChannelReader*>(aReader);
  return me->io_read(static_cast<char*>(aBuffer), aCount);
}

static int oggplay_channel_reader_io_seek(void* aReader, long aOffset, int aWhence) 
{
  nsChannelReader* me = static_cast<nsChannelReader*>(aReader);
  return me->io_seek(aOffset, aWhence);
}

static long oggplay_channel_reader_io_tell(void* aReader) 
{
  nsChannelReader* me = static_cast<nsChannelReader*>(aReader);
  return me->io_tell();
}

nsresult nsChannelReader::Init(nsOggDecoder* aDecoder, nsIURI* aURI)
{
  nsresult rv;

  mCurrentPosition = 0;
  mListener = new nsChannelToPipeListener(aDecoder);
  NS_ENSURE_TRUE(mListener, NS_ERROR_OUT_OF_MEMORY);

  rv = mListener->Init();
  NS_ENSURE_SUCCESS(rv, rv);
  
  rv = NS_NewChannel(getter_AddRefs(mChannel), 
                     aURI, 
                     nsnull,
                     nsnull,
                     nsnull,
                     nsIRequest::LOAD_NORMAL);
  NS_ENSURE_SUCCESS(rv, rv);

  rv = mChannel->AsyncOpen(mListener, nsnull);
  NS_ENSURE_SUCCESS(rv, rv);
  
  mListener->GetInputStream(getter_AddRefs(mInput));

  return NS_OK;
}

nsChannelReader::nsChannelReader() 
{
  OggPlayReader* reader = this;
  reader->initialise = &oggplay_channel_reader_initialise;
  reader->destroy = &oggplay_channel_reader_destroy;
  reader->seek = nsnull;
  reader->io_read = &oggplay_channel_reader_io_read;
  reader->io_seek = &oggplay_channel_reader_io_seek;
  reader->io_tell = &oggplay_channel_reader_io_tell;
}