// Copyright 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef MEDIA_CAST_AUDIO_RECEIVER_AUDIO_DECODER_H_
#define MEDIA_CAST_AUDIO_RECEIVER_AUDIO_DECODER_H_

#include "base/callback.h"
#include "base/synchronization/lock.h"
#include "media/cast/cast_config.h"
#include "media/cast/cast_environment.h"
#include "media/cast/framer/cast_message_builder.h"
#include "media/cast/framer/frame_id_map.h"
#include "media/cast/rtp_receiver/rtp_receiver_defines.h"

namespace webrtc {
class AudioCodingModule;
}

namespace media {
namespace cast {

typedef std::map<uint32, uint32> FrameIdRtpTimestampMap;

// Thread safe class.
class AudioDecoder {
 public:
  AudioDecoder(scoped_refptr<CastEnvironment> cast_environment,
               const AudioReceiverConfig& audio_config,
               RtpPayloadFeedback* incoming_payload_feedback);
  virtual ~AudioDecoder();

  // Extract a raw audio frame from the decoder.
  // Set the number of desired 10ms blocks and frequency.
  // Should be called from the cast audio decoder thread; however that is not
  // required.
  bool GetRawAudioFrame(int number_of_10ms_blocks,
                        int desired_frequency,
                        PcmAudioFrame* audio_frame,
                        uint32* rtp_timestamp);

  // Insert an RTP packet to the decoder.
  // Should be called from the main cast thread; however that is not required.
  void IncomingParsedRtpPacket(const uint8* payload_data,
                               size_t payload_size,
                               const RtpCastHeader& rtp_header);

  bool TimeToSendNextCastMessage(base::TimeTicks* time_to_send);
  void SendCastMessage();

 private:
  scoped_refptr<CastEnvironment> cast_environment_;

  // The webrtc AudioCodingModule is threadsafe.
  scoped_ptr<webrtc::AudioCodingModule> audio_decoder_;

  FrameIdMap frame_id_map_;
  CastMessageBuilder cast_message_builder_;

  base::Lock lock_;
  bool have_received_packets_;
  FrameIdRtpTimestampMap frame_id_rtp_timestamp_map_;
  uint32 last_played_out_timestamp_;

  DISALLOW_COPY_AND_ASSIGN(AudioDecoder);
};

}  // namespace cast
}  // namespace media

#endif  // MEDIA_CAST_AUDIO_RECEIVER_AUDIO_DECODER_H_