rocm-systems/rocclr/runtime/platform/object.cpp

//
// Copyright (c) 2010 Advanced Micro Devices, Inc. All rights reserved.
//

#include "platform/object.hpp"

#include <cstring>

namespace amd {

Atomic<ObjectMetadata::Key>
ObjectMetadata::nextKey_ = 1;


ObjectMetadata::Destructor
ObjectMetadata::destructors_[OCL_MAX_KEYS] = { NULL };


bool
ObjectMetadata::check(Key key)
{
    return key > 0 && key <= OCL_MAX_KEYS;
}

ObjectMetadata::Key
ObjectMetadata::createKey(Destructor destructor)
{
    Key key = nextKey_++;

    if (!check(key)) {
        return 0;
    }

    destructors_[key-1] = destructor;
    return key;
}

ObjectMetadata::~ObjectMetadata()
{
    if (!values_) {
        return;
    }

    for (size_t i = 0; i < OCL_MAX_KEYS; ++i) {
        if (values_[i] && destructors_[i]) {
            destructors_[i](values_[i]);
        }
    }

    delete[] values_;
}

void*
ObjectMetadata::getValueForKey(Key key) const
{
    if (!values_ || !check(key)) {
        return NULL;
    }

    return values_[key-1];
}

bool
ObjectMetadata::setValueForKey(Key key, Value value)
{
    if (!check(key)) {
        return false;
    }

    while (!values_) {
         Value* values = new Value[OCL_MAX_KEYS];
         memset(values, '\0', sizeof(Value) * OCL_MAX_KEYS);

         if (!values_.compareAndSet(NULL, values)) {
             delete[] values;
         }
    }

    size_t index = key-1;
    Value prev = AtomicOperation::swap(value, &values_[index]);
    if (prev && destructors_[index] != NULL) {
        destructors_[index](prev);
    }

    return true;
}

} // namespace amd