estimateSize.h

Go to the documentation of this file.

//
// Copyright 2025 Pixar
//
// Licensed under the terms set forth in the LICENSE.txt file available at
// https://openusd.org/license.
//
#ifndef PXR_EXEC_VDF_ESTIMATE_SIZE_H
#define PXR_EXEC_VDF_ESTIMATE_SIZE_H
 
 
#include "pxr/pxr.h"
 
#include "pxr/base/tf/smallVector.h"
 
#include <memory>
#include <vector>
 
PXR_NAMESPACE_OPEN_SCOPE
 
template < typename T >
inline size_t
VdfEstimateSize(const T &)
{
    return sizeof(T);
}
 
template < typename T, uint32_t N >
inline size_t
VdfEstimateSize(const TfSmallVector<T, N> &v)
{
    // It would be more accurate the iterate over v and estimate the size of
    // each element of v, but we are optimizing for performance rather than
    // accuracy for now.
    const size_t numExternal = (v.capacity() > N) ? v.capacity() : 0;
    const size_t elementSize = v.empty()
        ? sizeof(T) : VdfEstimateSize(v.front());
    return sizeof(TfSmallVector<T, N>) + numExternal * elementSize;
}
 
template < typename T, typename A >
inline size_t
VdfEstimateSize(const std::vector<T, A> &v)
{
    // It would be more accurate the iterate over v and estimate the size
    // of each element of v, but we are optimizing for performance rather
    // than accuracy for now.
    const size_t elementSize = v.empty()
        ? sizeof(T) : VdfEstimateSize(v.front());
    return sizeof(std::vector<T, A>) + v.capacity() * elementSize;
}
 
template < typename T >
inline size_t
VdfEstimateSize(const std::shared_ptr<T> &p)
{
    return sizeof(std::shared_ptr<T>) + (p ? VdfEstimateSize(*p) : 0);
}
 
PXR_NAMESPACE_CLOSE_SCOPE
 
#endif /* PXR_EXEC_VDF_ESTIMATE_SIZE_H */