Skip to content

Commit 86dd879

Browse files
committed
Simplify code and remove using-declarations
1 parent be68f3e commit 86dd879

File tree

2 files changed

+42
-64
lines changed

2 files changed

+42
-64
lines changed

src/FramePlayer.cpp

Lines changed: 32 additions & 43 deletions
Original file line numberDiff line numberDiff line change
@@ -24,11 +24,11 @@
2424
#include <vrs/IndexRecord.h>
2525

2626
namespace rerun_vrs {
27-
RerunFramePlayer::RerunFramePlayer(StreamId id, rerun::RecordingStream& rec)
27+
RerunFramePlayer::RerunFramePlayer(vrs::StreamId id, rerun::RecordingStream& rec)
2828
: id_{id}, rec_{rec} {}
2929

3030
bool RerunFramePlayer::onDataLayoutRead(
31-
const CurrentRecord& record, size_t blockIndex, DataLayout& layout
31+
const vrs::CurrentRecord& record, size_t blockIndex, vrs::DataLayout& layout
3232
) {
3333
if (!enabled_)
3434
return false;
@@ -55,11 +55,11 @@ namespace rerun_vrs {
5555
}
5656

5757
bool RerunFramePlayer::onImageRead(
58-
const CurrentRecord& record, size_t /*blockIndex*/, const ContentBlock& contentBlock
58+
const vrs::CurrentRecord& record, size_t /*blockIndex*/, const vrs::ContentBlock& contentBlock
5959
) {
6060
/* std::cout << "onImageRead" << std::endl; */
6161
const auto& spec = contentBlock.image();
62-
std::shared_ptr<PixelFrame> frame = getFrame(true);
62+
std::shared_ptr<vrs::utils::PixelFrame> frame;
6363
const auto& imageFormat = spec.getImageFormat();
6464
const auto& imageFormatStr = spec.getImageFormatAsString();
6565
bool frameValid = false;
@@ -68,7 +68,7 @@ namespace rerun_vrs {
6868
/* << std::endl; */
6969

7070
if (imageFormat == vrs::ImageFormat::RAW) {
71-
frameValid = PixelFrame::readRawFrame(frame, record.reader, spec);
71+
frameValid = vrs::utils::PixelFrame::readRawFrame(frame, record.reader, spec);
7272
if (frameValid) {
7373
// Log image to Rerun
7474
// NOTE currently we need to construct a vector to log an image, this will
@@ -153,56 +153,45 @@ namespace rerun_vrs {
153153
/* } */
154154
}
155155

156-
void RerunFramePlayer::convertFrame(std::shared_ptr<PixelFrame>& frame) {
157-
if (blankMode_) {
158-
makeBlankFrame(frame);
159-
} else {
160-
std::shared_ptr<PixelFrame> convertedFrame =
161-
needsConvertedFrame_ ? getFrame(false) : nullptr;
162-
PixelFrame::normalizeFrame(frame, convertedFrame, false);
163-
needsConvertedFrame_ = (frame != convertedFrame); // for next time!
164-
if (needsConvertedFrame_) {
165-
recycle(frame, true);
166-
frame = std::move(convertedFrame);
167-
}
168-
}
156+
void RerunFramePlayer::convertFrame(std::shared_ptr<vrs::utils::PixelFrame>& frame) {
157+
/* if (blankMode_) { */
158+
/* makeBlankFrame(frame); */
159+
/* } else { */
160+
/* std::shared_ptr<vrs::utils::PixelFrame> convertedFrame = */
161+
/* needsConvertedFrame_ ? getFrame(false) : nullptr; */
162+
/* vrs::utils::PixelFrame::normalizeFrame(frame, convertedFrame, false); */
163+
/* needsConvertedFrame_ = (frame != convertedFrame); // for next time! */
164+
/* if (needsConvertedFrame_) { */
165+
/* recycle(frame, true); */
166+
/* frame = std::move(convertedFrame); */
167+
/* } */
168+
/* } */
169169
}
170170

171-
void RerunFramePlayer::makeBlankFrame(std::shared_ptr<PixelFrame>& frame) {
171+
void RerunFramePlayer::makeBlankFrame(std::shared_ptr<vrs::utils::PixelFrame>& frame) {
172172
frame->init(vrs::PixelFormat::GREY8, frame->getWidth(), frame->getHeight());
173173
frame->blankFrame();
174174
}
175175

176-
std::shared_ptr<PixelFrame> RerunFramePlayer::getFrame(bool inputNotConvertedFrame) {
177-
std::vector<std::shared_ptr<PixelFrame>>& frames =
178-
inputNotConvertedFrame ? inputFrames_ : convertedframes_;
179-
if (frames.empty()) {
180-
return nullptr;
181-
}
182-
std::shared_ptr<PixelFrame> frame = std::move(frames.back());
183-
frames.pop_back();
184-
return frame;
185-
}
186-
187176
void RerunFramePlayer::recycle(
188-
std::shared_ptr<PixelFrame>& frame, bool inputNotConvertedFrame
177+
std::shared_ptr<vrs::utils::PixelFrame>& frame, bool inputNotConvertedFrame
189178
) {
190-
if (frame) {
191-
{
192-
std::vector<std::shared_ptr<PixelFrame>>& frames =
193-
inputNotConvertedFrame ? inputFrames_ : convertedframes_;
194-
if (frames.size() < 10) {
195-
frames.emplace_back(std::move(frame));
196-
}
197-
}
198-
frame.reset();
199-
}
179+
/* if (frame) { */
180+
/* { */
181+
/* std::vector<std::shared_ptr<vrs::utils::PixelFrame>>& frames = */
182+
/* inputNotConvertedFrame ? inputFrames_ : convertedframes_; */
183+
/* if (frames.size() < 10) { */
184+
/* frames.emplace_back(std::move(frame)); */
185+
/* } */
186+
/* } */
187+
/* frame.reset(); */
188+
/* } */
200189
}
201190

202191
void RerunFramePlayer::imageJobsThreadActivity() {
203192
std::unique_ptr<ImageJob> job;
204193
while (imageJobs_.waitForJob(job)) {
205-
std::shared_ptr<PixelFrame> frame = std::move(job->frame);
194+
std::shared_ptr<vrs::utils::PixelFrame> frame = std::move(job->frame);
206195
// if we're behind, we just drop images except the last one!
207196
while (imageJobs_.getJob(job)) {
208197
recycle(frame, true);
@@ -214,7 +203,7 @@ namespace rerun_vrs {
214203
frameValid = (frame != nullptr);
215204
} else {
216205
if (!frame) {
217-
frame = std::make_shared<PixelFrame>();
206+
frame = std::make_shared<vrs::utils::PixelFrame>();
218207
}
219208
frameValid = frame->readCompressedFrame(job->buffer, job->imageFormat);
220209
}

src/FramePlayer.h

Lines changed: 10 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -32,30 +32,22 @@ enum class FileReaderState;
3232

3333
namespace rerun_vrs {
3434

35-
using ::vrs::ContentBlock;
36-
using ::vrs::CurrentRecord;
37-
using ::vrs::DataLayout;
38-
using ::vrs::StreamId;
39-
using ::vrs::StreamPlayer;
40-
using ::vrs::utils::PixelFrame;
41-
using ::vrs::utils::VideoRecordFormatStreamPlayer;
42-
4335
struct ImageJob {
4436
ImageJob(vrs::ImageFormat imageFormat) : imageFormat{imageFormat} {}
4537

4638
vrs::ImageFormat imageFormat;
47-
std::shared_ptr<PixelFrame> frame;
39+
std::shared_ptr<vrs::utils::PixelFrame> frame;
4840
std::vector<uint8_t> buffer;
4941
};
5042

51-
class RerunFramePlayer : public VideoRecordFormatStreamPlayer {
43+
class RerunFramePlayer : public vrs::utils::VideoRecordFormatStreamPlayer {
5244
public:
53-
explicit RerunFramePlayer(StreamId id, rerun::RecordingStream& rec);
45+
explicit RerunFramePlayer(vrs::StreamId id, rerun::RecordingStream& rec);
5446

55-
bool onDataLayoutRead(const CurrentRecord& r, size_t blockIndex, DataLayout&) override;
56-
bool onImageRead(const CurrentRecord& r, size_t blockIndex, const ContentBlock&) override;
47+
bool onDataLayoutRead(const vrs::CurrentRecord& r, size_t blockIndex, vrs::DataLayout&) override;
48+
bool onImageRead(const vrs::CurrentRecord& r, size_t blockIndex, const vrs::ContentBlock&) override;
5749

58-
StreamId getId() const {
50+
vrs::StreamId getId() const {
5951
return id_;
6052
}
6153

@@ -67,10 +59,8 @@ namespace rerun_vrs {
6759

6860
private:
6961
rerun::RecordingStream& rec_;
70-
std::vector<std::shared_ptr<PixelFrame>> inputFrames_;
71-
std::vector<std::shared_ptr<PixelFrame>> convertedframes_;
7262
bool needsConvertedFrame_{false};
73-
StreamId id_;
63+
vrs::StreamId id_;
7464
/* MetaDataCollector descriptions_; */
7565
bool blankMode_{true};
7666
bool enabled_{true};
@@ -81,10 +71,9 @@ namespace rerun_vrs {
8171

8272
vrs::JobQueueWithThread<std::unique_ptr<ImageJob>> imageJobs_;
8373

84-
void convertFrame(std::shared_ptr<PixelFrame>& frame);
85-
void makeBlankFrame(std::shared_ptr<PixelFrame>& frame);
86-
std::shared_ptr<PixelFrame> getFrame(bool inputNotConvertedFrame);
87-
void recycle(std::shared_ptr<PixelFrame>& frame, bool inputNotConvertedFrame);
74+
void convertFrame(std::shared_ptr<vrs::utils::PixelFrame>& frame);
75+
void makeBlankFrame(std::shared_ptr<vrs::utils::PixelFrame>& frame);
76+
void recycle(std::shared_ptr<vrs::utils::PixelFrame>& frame, bool inputNotConvertedFrame);
8877
};
8978

9079
} // namespace rerun_vrs

0 commit comments

Comments
 (0)