summaryrefslogtreecommitdiffstats
path: root/dom/media/encoder/fmp4_muxer/AVCBox.cpp
blob: a45cda8b7fa17baccf54e5d4d09612d56afad69c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-*/
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this file,
 * You can obtain one at http://mozilla.org/MPL/2.0/. */

#include <climits>
#include "ISOControl.h"
#include "ISOMediaBoxes.h"
#include "AVCBox.h"

namespace mozilla {

nsresult
AVCSampleEntry::Generate(uint32_t* aBoxSize)
{
  uint32_t avc_box_size = 0;
  nsresult rv;
  rv = avcConfigBox->Generate(&avc_box_size);
  NS_ENSURE_SUCCESS(rv, rv);

  size += avc_box_size;

  *aBoxSize = size;

  return NS_OK;
}

nsresult
AVCSampleEntry::Write()
{
  BoxSizeChecker checker(mControl, size);
  nsresult rv;
  rv = VisualSampleEntry::Write();
  NS_ENSURE_SUCCESS(rv, rv);
  rv = avcConfigBox->Write();
  NS_ENSURE_SUCCESS(rv, rv);

  return NS_OK;
}

AVCSampleEntry::AVCSampleEntry(ISOControl* aControl)
  : VisualSampleEntry(NS_LITERAL_CSTRING("avc1"), aControl)
{
  avcConfigBox = new AVCConfigurationBox(aControl);
  MOZ_COUNT_CTOR(AVCSampleEntry);
}

AVCSampleEntry::~AVCSampleEntry()
{
  MOZ_COUNT_DTOR(AVCSampleEntry);
}

AVCConfigurationBox::AVCConfigurationBox(ISOControl* aControl)
  : Box(NS_LITERAL_CSTRING("avcC"), aControl)
{
  MOZ_COUNT_CTOR(AVCConfigurationBox);
}

AVCConfigurationBox::~AVCConfigurationBox()
{
  MOZ_COUNT_DTOR(AVCConfigurationBox);
}

nsresult
AVCConfigurationBox::Generate(uint32_t* aBoxSize)
{
  nsresult rv;
  FragmentBuffer* frag = mControl->GetFragment(Video_Track);
  rv = frag->GetCSD(avcConfig);
  NS_ENSURE_SUCCESS(rv, rv);
  size += avcConfig.Length();
  *aBoxSize = size;
  return NS_OK;
}

nsresult
AVCConfigurationBox::Write()
{
  BoxSizeChecker checker(mControl, size);
  Box::Write();

  mControl->Write(avcConfig.Elements(), avcConfig.Length());

  return NS_OK;
}

}