RefCount.h

Go to the documentation of this file.

#include <shogun/lib/config.h>

#ifdef HAVE_CXX11_ATOMIC
#include <atomic>
#endif

#include <shogun/lib/common.h>
#include <shogun/lib/Lock.h>

#ifndef _REFCOUNT__H__
#define _REFCOUNT__H__

namespace shogun
{
class RefCount
{
public:
    RefCount(int32_t ref_start=0) : rc(ref_start) {}

    int32_t ref();

    int32_t unref();

    int32_t ref_count();

#ifdef HAVE_CXX11_ATOMIC
    volatile std::atomic<int> rc;
#else
    int32_t rc;

    CLock lock;
#endif
};
}

#endif //_REFCOUNT__H__