// Copyright 2020 Google LLC. All rights reserved.
//
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file or at
// https://developers.google.com/open-source/licenses/bsd

#ifndef PACKAGER_MEDIA_FORMATS_MP2T_ES_PARSER_DVB_H_
#define PACKAGER_MEDIA_FORMATS_MP2T_ES_PARSER_DVB_H_

#include <unordered_map>

#include <packager/media/base/byte_queue.h>
#include <packager/media/formats/dvb/dvb_sub_parser.h>
#include <packager/media/formats/mp2t/es_parser.h>
#include <functional>

namespace shaka {
namespace media {
namespace mp2t {

class EsParserDvb : public EsParser {
 public:
  EsParserDvb(uint32_t pid,
              const NewStreamInfoCB& new_stream_info_cb,
              const EmitTextSampleCB& emit_sample_cb,
              const uint8_t* descriptor,
              size_t descriptor_length);
  ~EsParserDvb() override;

  // EsParser implementation.
  bool Parse(const uint8_t* buf, int size, int64_t pts, int64_t dts) override;
  bool Flush() override;
  void Reset() override;

 private:
  EsParserDvb(const EsParserDvb&) = delete;
  EsParserDvb& operator=(const EsParserDvb&) = delete;

  bool ParseInternal(const uint8_t* data, size_t size, int64_t pts);

  // Callbacks:
  // - to signal a new audio configuration,
  // - to send ES buffers.
  NewStreamInfoCB new_stream_info_cb_;
  EmitTextSampleCB emit_sample_cb_;

  // A map of page_id to parser.
  std::unordered_map<uint16_t, DvbSubParser> parsers_;
  // A map of page_id to language.
  std::unordered_map<uint16_t, std::string> languages_;
  bool sent_info_ = false;
};

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

#endif  // PACKAGER_MEDIA_FORMATS_MP2T_ES_PARSER_DVB_H_