Drop timeBase from convertAVFrameToFrameOutput API

dvrogozh · dvrogozh · commit a27a9e6444d8 · 2025-09-02T19:04:59.000Z
Signed-off-by: Dmitry Rogozhkin &lt;dmitry.v.rogozhkin@intel.com&gt;
diff --git a/src/torchcodec/_core/CpuDeviceInterface.cpp b/src/torchcodec/_core/CpuDeviceInterface.cpp
@@ -109,7 +109,6 @@ std::unique_ptr<FiltersContext> CpuDeviceInterface::initializeFiltersContext(
 // `dimension_order` parameter. It's up to callers to re-shape it if needed.
 void CpuDeviceInterface::convertAVFrameToFrameOutput(
     const VideoStreamOptions& videoStreamOptions,
-    [[maybe_unused]] const AVRational& timeBase,
     UniqueAVFrame& avFrame,
     FrameOutput& frameOutput,
     std::optional<torch::Tensor> preAllocatedOutputTensor) {
diff --git a/src/torchcodec/_core/CpuDeviceInterface.h b/src/torchcodec/_core/CpuDeviceInterface.h
@@ -33,7 +33,6 @@ class CpuDeviceInterface : public DeviceInterface {
 
   void convertAVFrameToFrameOutput(
       const VideoStreamOptions& videoStreamOptions,
-      const AVRational& timeBase,
       UniqueAVFrame& avFrame,
       FrameOutput& frameOutput,
       std::optional<torch::Tensor> preAllocatedOutputTensor =
diff --git a/src/torchcodec/_core/CudaDeviceInterface.cpp b/src/torchcodec/_core/CudaDeviceInterface.cpp
@@ -265,7 +265,6 @@ std::unique_ptr<FiltersContext> CudaDeviceInterface::initializeFiltersContext(
 
 void CudaDeviceInterface::convertAVFrameToFrameOutput(
     const VideoStreamOptions& videoStreamOptions,
-    [[maybe_unused]] const AVRational& timeBase,
     UniqueAVFrame& avFrame,
     FrameOutput& frameOutput,
     std::optional<torch::Tensor> preAllocatedOutputTensor) {
@@ -284,7 +283,6 @@ void CudaDeviceInterface::convertAVFrameToFrameOutput(
     FrameOutput cpuFrameOutput;
     cpuInterface->convertAVFrameToFrameOutput(
         videoStreamOptions,
-        timeBase,
         avFrame,
         cpuFrameOutput,
         preAllocatedOutputTensor);
diff --git a/src/torchcodec/_core/CudaDeviceInterface.h b/src/torchcodec/_core/CudaDeviceInterface.h
@@ -28,7 +28,6 @@ class CudaDeviceInterface : public DeviceInterface {
 
   void convertAVFrameToFrameOutput(
       const VideoStreamOptions& videoStreamOptions,
-      const AVRational& timeBase,
       UniqueAVFrame& avFrame,
       FrameOutput& frameOutput,
       std::optional<torch::Tensor> preAllocatedOutputTensor =
diff --git a/src/torchcodec/_core/DeviceInterface.h b/src/torchcodec/_core/DeviceInterface.h
@@ -56,7 +56,6 @@ class DeviceInterface {
 
   virtual void convertAVFrameToFrameOutput(
       const VideoStreamOptions& videoStreamOptions,
-      const AVRational& timeBase,
       UniqueAVFrame& avFrame,
       FrameOutput& frameOutput,
       std::optional<torch::Tensor> preAllocatedOutputTensor = std::nullopt) = 0;
diff --git a/src/torchcodec/_core/SingleStreamDecoder.cpp b/src/torchcodec/_core/SingleStreamDecoder.cpp
@@ -1291,7 +1291,6 @@ FrameOutput SingleStreamDecoder::convertAVFrameToFrameOutput(
     }
     deviceInterface_->convertAVFrameToFrameOutput(
         streamInfo.videoStreamOptions,
-        streamInfo.timeBase,
         avFrame,
         frameOutput,
         preAllocatedOutputTensor);

Original file line number	Diff line number	Diff line change
`@@ -1291,7 +1291,6 @@ FrameOutput SingleStreamDecoder::convertAVFrameToFrameOutput(`
`1291`	`1291`	`}`
`1292`	`1292`	`deviceInterface_->convertAVFrameToFrameOutput(`
`1293`	`1293`	`streamInfo.videoStreamOptions,`
`1294`		`- streamInfo.timeBase,`
`1295`	`1294`	`avFrame,`
`1296`	`1295`	`frameOutput,`
`1297`	`1296`	`preAllocatedOutputTensor);`