ByteBuffer.h

Go to the documentation of this file.

// Jolt Physics Library (https://github.com/jrouwe/JoltPhysics)
// SPDX-FileCopyrightText: 2021 Jorrit Rouwe
// SPDX-License-Identifier: MIT
 
#pragma once
 
#include <Jolt/Core/STLAlignedAllocator.h>
 
JPH_NAMESPACE_BEGIN
 
using ByteBufferVector = Array<uint8, STLAlignedAllocator<uint8, JPH_CACHE_LINE_SIZE>>;
 
class ByteBuffer : public ByteBufferVector
{
public:
    size_t          Align(size_t inSize)
    {
        // Assert power of 2
        JPH_ASSERT(IsPowerOf2(inSize));
 
        // Calculate new size and resize buffer
        size_t s = AlignUp(size(), inSize);
        resize(s, 0);
 
        return s;
    }
 
    template <class Type>
    Type *          Allocate(size_t inSize = 1)
    {
        // Reserve space
        size_t s = size();
        resize(s + inSize * sizeof(Type));
 
        // Get data pointer
        Type *data = reinterpret_cast<Type *>(&at(s));
 
        // Construct elements
        for (Type *d = data, *d_end = data + inSize; d < d_end; ++d)
            new (d) Type;
 
        // Return pointer
        return data;
    }
 
    template <class Type>
    void            AppendVector(const Array<Type> &inData)
    {
        size_t size = inData.size() * sizeof(Type);
        uint8 *data = Allocate<uint8>(size);
        memcpy(data, &inData[0], size);
    }
 
    template <class Type>
    const Type *    Get(size_t inPosition) const
    {
        return reinterpret_cast<const Type *>(&at(inPosition));
    }
 
    template <class Type>
    Type *          Get(size_t inPosition)
    {
        return reinterpret_cast<Type *>(&at(inPosition));
    }
};
 
JPH_NAMESPACE_END