File Coverage

src/panda/cast.h

Criterion	Covered	Total	%
statement	14	16	87.5
branch	9	22	40.9
condition			n/a
subroutine			n/a
pod			n/a
total	23	38	60.5

#pragma once
#include
#include
#include
#include
#include

namespace panda {

namespace detail { namespace cast {
    typedef std::map DynCastCacheMap;

    template
    DynCastCacheMap& get_map () {
        thread_local DynCastCacheMap* map;
        if (!map) {
            thread_local DynCastCacheMap _map;
            map = &_map;
        }
        return *map;
    }

    constexpr const ptrdiff_t INCORRECT_PTRDIFF = PTRDIFF_MAX;
}}

template
DERIVED_PTR dyn_cast (BASE* obj) {
    using namespace detail::cast;
    using DERIVED = typename std::remove_pointer::type;

    if (std::is_same::value) return (DERIVED_PTR)((void*)obj);
    if (!obj) return NULL;

    intptr_t key = (intptr_t)typeid(*obj).name();
    auto&    map = get_map();
    //auto& map = DynCastCache::map;
    DynCastCacheMap::iterator it = map.find(key);
    if (it != map.end())
        return it->second != INCORRECT_PTRDIFF ? reinterpret_cast((char*)obj - it->second) : NULL;
    DERIVED* ret = dynamic_cast(obj);
    if (ret) map[key] = (char*)obj - (char*)ret;
    else map[key] = INCORRECT_PTRDIFF;
    return ret;
}

template
DERIVED_REF dyn_cast (BASE& obj) {
    using namespace detail::cast;
    using DERIVED = typename std::remove_reference::type;

    if (std::is_same::value) return reinterpret_cast(obj);

    intptr_t key = (intptr_t)typeid(obj).name();
    auto&    map = get_map();
    DynCastCacheMap::iterator it = map.find(key);
    if (it != map.end() && it->second != INCORRECT_PTRDIFF)
        return *(reinterpret_cast((char*)&obj - it->second));
    // dont cache fails, as exceptions are much slower than dynamic_cast, let it always fall here
    DERIVED& ret = dynamic_cast(obj);
    map[key] = (char*)&obj - (char*)&ret;
    return ret;
}

}