9.1/sources/includes/ds3d/common/helper/memdata.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_COMMON_HELPER_MEMDATA_H
#define DS3D_COMMON_HELPER_MEMDATA_H
 
#include "ds3d/common/helper/safe_queue.h"
 
namespace ds3d {
 
struct MemData {
    void* data = nullptr;
    size_t byteSize = 0;
    int devId = 0;
    MemType type = MemType::kCpu;
    MemData() = default;
    virtual ~MemData() = default;
 
protected:
    DS3D_DISABLE_CLASS_COPY(MemData);
};
 
class CpuMemBuf : public MemData {
public:
    CpuMemBuf() {}
    static std::unique_ptr<CpuMemBuf> CreateBuf(size_t size)
    {
        auto mem = std::make_unique<CpuMemBuf>();
        void* data = malloc(size);
        if (!data) {
            LOG_ERROR("DS3D, malloc out of memory");
            return nullptr;
        }
        mem->data = data;
        mem->byteSize = size;
        mem->devId = 0;
        mem->type = MemType::kCpu;
        return mem;
    }
    ~CpuMemBuf()
    {
        if (data) {
            free(data);
        }
    }
};
 
};
#endif