Skip to content

Commit 0b6590d

Browse files
committed
getNextDecodedOutputNoDemux -> getNextFrameOutputNoDemuxInternal
1 parent 6bd363d commit 0b6590d

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

src/torchcodec/decoders/_core/VideoDecoder.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1027,7 +1027,7 @@ VideoDecoder::DecodedOutput VideoDecoder::getFrameAtIndexInternal(
10271027

10281028
int64_t pts = stream.allFrames[frameIndex].pts;
10291029
setCursorPtsInSeconds(ptsToSeconds(pts, stream.timeBase));
1030-
return getNextDecodedOutputNoDemux(preAllocatedOutputTensor);
1030+
return getNextFrameOutputNoDemuxInternal(preAllocatedOutputTensor);
10311031
}
10321032

10331033
VideoDecoder::BatchDecodedOutput VideoDecoder::getFramesAtIndices(
@@ -1268,12 +1268,12 @@ VideoDecoder::RawDecodedOutput VideoDecoder::getNextRawDecodedOutputNoDemux() {
12681268
}
12691269

12701270
VideoDecoder::DecodedOutput VideoDecoder::getNextFrameNoDemux() {
1271-
auto output = getNextDecodedOutputNoDemux();
1271+
auto output = getNextFrameOutputNoDemuxInternal();
12721272
output.frame = MaybePermuteHWC2CHW(output.streamIndex, output.frame);
12731273
return output;
12741274
}
12751275

1276-
VideoDecoder::DecodedOutput VideoDecoder::getNextDecodedOutputNoDemux(
1276+
VideoDecoder::DecodedOutput VideoDecoder::getNextFrameOutputNoDemuxInternal(
12771277
std::optional<torch::Tensor> preAllocatedOutputTensor) {
12781278
auto rawOutput = getNextRawDecodedOutputNoDemux();
12791279
return convertAVFrameToDecodedOutput(rawOutput, preAllocatedOutputTensor);

src/torchcodec/decoders/_core/VideoDecoder.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -161,8 +161,8 @@ class VideoDecoder {
161161

162162
// ---- SINGLE FRAME SEEK AND DECODING API ----
163163
// Places the cursor at the first frame on or after the position in seconds.
164-
// Calling getNextDecodedOutputNoDemux() will return the first frame at or
165-
// after this position.
164+
// Calling getNextFrameOutputNoDemuxInternal() will return the first frame at
165+
// or after this position.
166166
void setCursorPtsInSeconds(double seconds);
167167
// This is an internal structure that is used to store the decoded output
168168
// from decoding a frame through color conversion. Example usage is:
@@ -391,7 +391,7 @@ class VideoDecoder {
391391
int streamIndex,
392392
int64_t frameIndex,
393393
std::optional<torch::Tensor> preAllocatedOutputTensor = std::nullopt);
394-
DecodedOutput getNextDecodedOutputNoDemux(
394+
DecodedOutput getNextFrameOutputNoDemuxInternal(
395395
std::optional<torch::Tensor> preAllocatedOutputTensor = std::nullopt);
396396

397397
DecoderOptions options_;

0 commit comments

Comments
 (0)