cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix timestamp computation when copying within first block.

Browse Source 
When computing the last copied sample's timestamp, first_block_offset_ needs to be taken into account.

PiperOrigin-RevId: 542643291
This commit is contained in:
MediaPipe Team 2023-06-22 12:58:43 -07:00 committed by Copybara-Service
parent 98d493f37a
commit 2f5fc16a38
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mediapipe/calculators/audio/time_series_framer_calculator.cc
View File

@ -208,6 +208,7 @@ Matrix TimeSeriesFramerCalculator::SampleBlockBuffer::CopySamples(
int offset = first_block_offset_; int offset = first_block_offset_;
int n; int n;
Timestamp last_block_ts; Timestamp last_block_ts;
int last_sample_index;
for (auto it = blocks_.begin(); it != blocks_.end() && count > 0; ++it) { for (auto it = blocks_.begin(); it != blocks_.end() && count > 0; ++it) {
n = std::min(it->num_samples() - offset, count); n = std::min(it->num_samples() - offset, count);
@ -216,12 +217,13 @@ Matrix TimeSeriesFramerCalculator::SampleBlockBuffer::CopySamples(
count -= n; count -= n;
num_copied += n; num_copied += n;
last_block_ts = it->timestamp; last_block_ts = it->timestamp;
last_sample_index = offset + n - 1;
offset = 0; // No samples have been discarded in subsequent blocks. offset = 0; // No samples have been discarded in subsequent blocks.
} }
// Compute the timestamp of the last copied sample. // Compute the timestamp of the last copied sample.
*last_timestamp = *last_timestamp =
last_block_ts + std::round(ts_units_per_sample_ * (n - 1)); last_block_ts + std::round(ts_units_per_sample_ * last_sample_index);
} }
if (count > 0) { if (count > 0) {