src/media/formats/webm/cluster_builder.h

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef MEDIA_FORMATS_WEBM_CLUSTER_BUILDER_H_
   6 #define MEDIA_FORMATS_WEBM_CLUSTER_BUILDER_H_
   7
   8 #include "base/basictypes.h"
   9 #include "base/memory/scoped_ptr.h"
  10 #include "media/base/buffers.h"
  11
  12 namespace media {
  13
  14 class Cluster {
  15  public:
  16   Cluster(scoped_ptr<uint8[]> data, int size);
  17   ~Cluster();
  18
  19   const uint8* data() const { return data_.get(); }
  20   int size() const { return size_; }
  21
  22  private:
  23   scoped_ptr<uint8[]> data_;
  24   int size_;
  25
  26   DISALLOW_IMPLICIT_CONSTRUCTORS(Cluster);
  27 };
  28
  29 class ClusterBuilder {
  30  public:
  31   ClusterBuilder();
  32   ~ClusterBuilder();
  33
  34   void SetClusterTimecode(int64 cluster_timecode);
  35   void AddSimpleBlock(int track_num, int64 timecode, int flags,
  36                       const uint8* data, int size);
  37   void AddBlockGroup(int track_num, int64 timecode, int duration, int flags,
  38                      const uint8* data, int size);
  39   void AddBlockGroupWithoutBlockDuration(int track_num, int64 timecode,
  40                      int flags, const uint8* data, int size);
  41
  42   scoped_ptr<Cluster> Finish();
  43
  44  private:
  45   void AddBlockGroupInternal(int track_num, int64 timecode,
  46                              bool include_block_duration, int duration,
  47                              int flags, const uint8* data, int size);
  48   void Reset();
  49   void ExtendBuffer(int bytes_needed);
  50   void UpdateUInt64(int offset, int64 value);
  51   void WriteBlock(uint8* buf, int track_num, int64 timecode, int flags,
  52                   const uint8* data, int size);
  53
  54   scoped_ptr<uint8[]> buffer_;
  55   int buffer_size_;
  56   int bytes_used_;
  57   int64 cluster_timecode_;
  58
  59   DISALLOW_COPY_AND_ASSIGN(ClusterBuilder);
  60 };
  61
  62 }  // namespace media
  63
  64 #endif  // MEDIA_FORMATS_WEBM_CLUSTER_BUILDER_H_