103 lines
3.2 KiB
C++
103 lines
3.2 KiB
C++
// Copyright 2014 Google Inc. 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
|
|
//
|
|
// MP4Fragmenter is responsible for the generation of MP4 fragments, i.e. traf
|
|
// and the corresponding mdat. The samples are also encrypted if encryption is
|
|
// requested.
|
|
|
|
#ifndef MEDIA_MP4_MP4_FRAGMENTER_H_
|
|
#define MEDIA_MP4_MP4_FRAGMENTER_H_
|
|
|
|
#include <vector>
|
|
|
|
#include "base/memory/ref_counted.h"
|
|
#include "base/memory/scoped_ptr.h"
|
|
#include "media/base/status.h"
|
|
|
|
namespace media {
|
|
|
|
class AesCtrEncryptor;
|
|
class BufferWriter;
|
|
class MediaSample;
|
|
|
|
namespace mp4 {
|
|
|
|
class SegmentReference;
|
|
class TrackFragment;
|
|
|
|
class MP4Fragmenter {
|
|
public:
|
|
// Caller retains the ownership of |traf| and transfers ownership of
|
|
// |encryptor|. |clear_time| specifies clear time in the current track
|
|
// timescale. |nalu_length_size| specifies NAL unit length size, for
|
|
// subsample encryption. |normalize_presentation_timestamp| defines whether
|
|
// PTS should be normalized to start from zero.
|
|
MP4Fragmenter(TrackFragment* traf,
|
|
scoped_ptr<AesCtrEncryptor> encryptor,
|
|
int64 clear_time,
|
|
uint8 nalu_length_size,
|
|
bool normalize_presentation_timestamp);
|
|
~MP4Fragmenter();
|
|
|
|
virtual Status AddSample(scoped_refptr<MediaSample> sample);
|
|
|
|
// Initialize the fragment with default data.
|
|
void InitializeFragment();
|
|
|
|
// Finalize and optimize the fragment.
|
|
void FinalizeFragment();
|
|
|
|
// Fill in |reference| with current fragment information.
|
|
void GenerateSegmentReference(SegmentReference* reference);
|
|
|
|
uint64 fragment_duration() const { return fragment_duration_; }
|
|
uint64 first_sap_time() const { return first_sap_time_; }
|
|
uint64 earliest_presentation_time() const {
|
|
return earliest_presentation_time_;
|
|
}
|
|
bool fragment_finalized() const { return fragment_finalized_; }
|
|
BufferWriter* data() { return data_.get(); }
|
|
BufferWriter* aux_data() { return aux_data_.get(); }
|
|
|
|
private:
|
|
void EncryptBytes(uint8* data, uint32 size);
|
|
Status EncryptSample(scoped_refptr<MediaSample> sample);
|
|
|
|
// Should we enable encrytion for the current fragment?
|
|
bool ShouldEncryptFragment() {
|
|
return (encryptor_ != NULL && clear_time_ <= 0);
|
|
}
|
|
|
|
// Should we enable subsample encryption?
|
|
bool IsSubsampleEncryptionRequired() { return nalu_length_size_ != 0; }
|
|
|
|
// Check if the current fragment starts with SAP.
|
|
bool StartsWithSAP();
|
|
|
|
scoped_ptr<AesCtrEncryptor> encryptor_;
|
|
// If this stream contains AVC, subsample encryption specifies that the size
|
|
// and type of NAL units remain unencrypted. This field specifies the size of
|
|
// the size field. Can be 1, 2 or 4 bytes.
|
|
uint8 nalu_length_size_;
|
|
TrackFragment* traf_;
|
|
bool fragment_finalized_;
|
|
uint64 fragment_duration_;
|
|
bool normalize_presentation_timestamp_;
|
|
int64 presentation_start_time_;
|
|
uint64 earliest_presentation_time_;
|
|
uint64 first_sap_time_;
|
|
int64 clear_time_;
|
|
scoped_ptr<BufferWriter> data_;
|
|
scoped_ptr<BufferWriter> aux_data_;
|
|
|
|
DISALLOW_COPY_AND_ASSIGN(MP4Fragmenter);
|
|
};
|
|
|
|
} // namespace mp4
|
|
} // namespace media
|
|
|
|
#endif // MEDIA_MP4_MP4_FRAGMENTER_H_
|