// Copyright 2014 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 PACKAGER_MEDIA_FORMATS_MP2T_TS_SECTION_PMT_H_
#define PACKAGER_MEDIA_FORMATS_MP2T_TS_SECTION_PMT_H_

#include <functional>
#include <string>

#include <packager/macros/classes.h>
#include <packager/media/formats/mp2t/ts_audio_type.h>
#include <packager/media/formats/mp2t/ts_section_psi.h>
#include <packager/media/formats/mp2t/ts_stream_type.h>

namespace shaka {
namespace media {
namespace mp2t {

class TsSectionPmt : public TsSectionPsi {
 public:
  // RegisterPesCb::Run(int pes_pid, int stream_type, uint32_t max_bitrate,
  //  const string& lang, TsAudioType audio_type, uint8_t* descriptor,
  // size_t desriptor_size);
  // Stream type is defined in
  // "Table 2-34 – Stream type assignments" in H.222
  // Audio type is defined in
  // "Table 2-60 - Audio type values" in H.222
  typedef std::function<void(int,
                             TsStreamType,
                             uint32_t,
                             const std::string&,
                             TsAudioType,
                             const uint8_t*,
                             size_t)>
      RegisterPesCb;

  explicit TsSectionPmt(const RegisterPesCb& register_pes_cb);
  ~TsSectionPmt() override;

  // Mpeg2TsPsiParser implementation.
  bool ParsePsiSection(BitReader* bit_reader) override;
  void ResetPsiSection() override;

 private:
  RegisterPesCb register_pes_cb_;

  DISALLOW_COPY_AND_ASSIGN(TsSectionPmt);
};

}  // namespace mp2t
}  // namespace media
}  // namespace shaka

#endif