9.1/sources/libs/ds3d/dataloader/lidarsource/lidar_file_source_impl.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2021-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: Apache-2.0
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
#ifndef _DS3D_DATALOADER_LIDARSOURCE_LIDAR_FILE_SOURCE_IMPL_H
#define _DS3D_DATALOADER_LIDARSOURCE_LIDAR_FILE_SOURCE_IMPL_H
 
#include "lidar_file_source.h"
#include "lidar_file_config.h"
#include "ds3d/common/hpp/profiling.hpp"
#include "ds3d/common/impl/impl_frames.h"
#include "ds3d/common/helper/memdata.h"
#include "ds3d/common/helper/cuda_utils.h"
 
namespace ds3d { namespace impl { namespace lidarsource {
 
using MemPtr = std::unique_ptr<MemData>;
 
class LidarFileSourceImpl : public SyncImplDataLoader {
public:
    LidarFileSourceImpl() = default;
    ~LidarFileSourceImpl() override = default;
 
protected:
    ErrCode startImpl(const std::string& content, const std::string& path) override;
    ErrCode readDataImpl(GuardDataMap& datamap) override;
    ErrCode flushImpl() final { return ErrCode::kGood; }
    ErrCode stopImpl() override;
 
private:
    ErrCode reserveMem(
        Ptr<BufferPool<MemPtr>>& pool, size_t memSize, uint32_t count, const std::string& name);
 
    Ptr<BufferPool<MemPtr>> _bufMem;
    std::vector<uint8_t> _tmpStrideBuf;
    Ptr<MemData> _cpuSwapBuf;
    Config _config;
    profiling::FileReader _dataReader;
    uint32_t _bytesPerFrame = 0;
    uint32_t _bytesStrideFrame = 0;
    uint32_t _totalNumFrames = 0;
    uint32_t _totalFrameDuration = 0;
    uint32_t _readFrameCount = 0;
    bool _isFirstFrame = true;
};
 
}}}
 
#endif