Skip to content

Commit d8f899e

Browse files
committed
Update names
1 parent 760d6da commit d8f899e

File tree

3 files changed

+21
-21
lines changed

3 files changed

+21
-21
lines changed

src/torchcodec/decoders/_core/FFMPEGCommon.cpp

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -10,22 +10,22 @@
1010

1111
namespace facebook::torchcodec {
1212

13-
AutoFreedAVPacket::AutoFreedAVPacket() : avPacket_(av_packet_alloc()) {
13+
AutoAVPacket::AutoAVPacket() : avPacket_(av_packet_alloc()) {
1414
TORCH_CHECK(avPacket_ != nullptr, "Couldn't allocate avPacket.");
1515
}
16-
AutoFreedAVPacket::~AutoFreedAVPacket() {
16+
AutoAVPacket::~AutoAVPacket() {
1717
av_packet_free(&avPacket_);
1818
}
1919

20-
AutoUnrefedAVPacket::AutoUnrefedAVPacket(AutoFreedAVPacket& shared)
20+
ReferenceAVPacket::ReferenceAVPacket(AutoAVPacket& shared)
2121
: avPacket_(shared.avPacket_) {}
22-
AutoUnrefedAVPacket::~AutoUnrefedAVPacket() {
22+
ReferenceAVPacket::~ReferenceAVPacket() {
2323
av_packet_unref(avPacket_);
2424
}
25-
AVPacket* AutoUnrefedAVPacket::get() {
25+
AVPacket* ReferenceAVPacket::get() {
2626
return avPacket_;
2727
}
28-
AVPacket* AutoUnrefedAVPacket::operator->() {
28+
AVPacket* ReferenceAVPacket::operator->() {
2929
return avPacket_;
3030
}
3131

src/torchcodec/decoders/_core/FFMPEGCommon.h

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -71,37 +71,37 @@ using UniqueSwsContext =
7171
// These 2 classes share the same underlying AVPacket object. They are meant to
7272
// be used in tandem, like so:
7373
//
74-
// AutoFreedAVPacket autoFreedAVPacket; // <-- malloc for AVPacket happens here
74+
// AutoAVPacket autoAVPacket; // <-- malloc for AVPacket happens here
7575
// while(...){
76-
// AutoUnrefedAVPacket packet(autoFreedAVPacket);
76+
// ReferenceAVPacket packet(autoAVPacket);
7777
// av_read_frame(..., packet.get()); <-- av_packet_ref() called by FFmpeg
7878
// } <-- av_packet_unref() called here
7979
//
8080
// This achieves a few desirable things:
8181
// - Memory allocation of the underlying AVPacket happens only once, when
82-
// autoFreedAVPacket is created.
83-
// - av_packet_free() is called when autoFreedAVPacket gets out of scope
82+
// autoAVPacket is created.
83+
// - av_packet_free() is called when autoAVPacket gets out of scope
8484
// - av_packet_unref() is automatically called when needed, i.e. at the end of
8585
// each loop iteration (or when hitting break / continue). This prevents the
8686
// risk of us forgetting to call it.
87-
class AutoFreedAVPacket {
88-
friend class AutoUnrefedAVPacket;
87+
class AutoAVPacket {
88+
friend class ReferenceAVPacket;
8989

9090
private:
9191
AVPacket* avPacket_;
9292

9393
public:
94-
AutoFreedAVPacket();
95-
~AutoFreedAVPacket();
94+
AutoAVPacket();
95+
~AutoAVPacket();
9696
};
9797

98-
class AutoUnrefedAVPacket {
98+
class ReferenceAVPacket {
9999
private:
100100
AVPacket* avPacket_;
101101

102102
public:
103-
AutoUnrefedAVPacket(AutoFreedAVPacket& shared);
104-
~AutoUnrefedAVPacket();
103+
ReferenceAVPacket(AutoAVPacket& shared);
104+
~ReferenceAVPacket();
105105
AVPacket* get();
106106
AVPacket* operator->();
107107
};

src/torchcodec/decoders/_core/VideoDecoder.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -573,9 +573,9 @@ void VideoDecoder::scanFileAndUpdateMetadataAndIndex() {
573573
return;
574574
}
575575

576-
AutoFreedAVPacket autoFreedAVPacket;
576+
AutoAVPacket autoAVPacket;
577577
while (true) {
578-
AutoUnrefedAVPacket packet(autoFreedAVPacket);
578+
ReferenceAVPacket packet(autoAVPacket);
579579

580580
// av_read_frame is a misleading name: it gets the next **packet**.
581581
int ffmpegStatus = av_read_frame(formatContext_.get(), packet.get());
@@ -806,7 +806,7 @@ VideoDecoder::RawDecodedOutput VideoDecoder::getDecodedOutputWithFilter(
806806
}
807807
// Need to get the next frame or error from PopFrame.
808808
UniqueAVFrame frame(av_frame_alloc());
809-
AutoFreedAVPacket autoFreedAVPacket;
809+
AutoAVPacket autoAVPacket;
810810
int ffmpegStatus = AVSUCCESS;
811811
bool reachedEOF = false;
812812
int frameStreamIndex = -1;
@@ -846,7 +846,7 @@ VideoDecoder::RawDecodedOutput VideoDecoder::getDecodedOutputWithFilter(
846846
// pulling frames from its internal buffers.
847847
continue;
848848
}
849-
AutoUnrefedAVPacket packet(autoFreedAVPacket);
849+
ReferenceAVPacket packet(autoAVPacket);
850850
ffmpegStatus = av_read_frame(formatContext_.get(), packet.get());
851851
decodeStats_.numPacketsRead++;
852852
if (ffmpegStatus == AVERROR_EOF) {

0 commit comments

Comments
 (0)