Parallel.h

Go to the documentation of this file.

/*
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * Written (W) 1999-2009 Soeren Sonnenburg
 * Copyright (C) 1999-2009 Fraunhofer Institute FIRST and Max-Planck-Society
 */

#ifndef PARALLEL_H__
#define PARALLEL_H__

#include <shogun/lib/config.h>
#include <shogun/lib/common.h>
#include <shogun/io/SGIO.h>

#ifdef HAVE_PTHREAD
#ifdef USE_SPINLOCKS
    #define PTHREAD_LOCK_T pthread_spinlock_t
    #define PTHREAD_LOCK_INIT(lock) pthread_spin_init(lock, 0)
    #define PTHREAD_LOCK_DESTROY(lock) pthread_spin_destroy(lock)
    #define PTHREAD_LOCK(lock) pthread_spin_lock(lock)
    #define PTHREAD_UNLOCK(lock) pthread_spin_unlock(lock)
#else
    #define PTHREAD_LOCK_T pthread_mutex_t
    #define PTHREAD_LOCK_INIT(lock) pthread_mutex_init(lock, NULL)
    #define PTHREAD_LOCK_DESTROY(lock) pthread_mutex_destroy(lock)
    #define PTHREAD_LOCK(lock) pthread_mutex_lock(lock)
    #define PTHREAD_UNLOCK(lock) pthread_mutex_unlock(lock)
#endif
#endif

namespace shogun
{
class Parallel
{
public:
    Parallel();

    Parallel(const Parallel& orig);

    virtual ~Parallel();

    int32_t get_num_cpus() const;

    inline void set_num_threads(int32_t n)
    {
#ifndef HAVE_PTHREAD
        ASSERT(n==1);
#endif
        num_threads=n;
    }

    inline int32_t get_num_threads() const
    {
        return num_threads;
    }

    inline int32_t ref()
    {
        ++refcount;
        return refcount;
    }

    inline int32_t ref_count() const
    {
        return refcount;
    }

    inline int32_t unref()
    {
        if (refcount==0 || --refcount==0)
        {
            delete this;
            return 0;
        }
        else
            return refcount;
    }

private:

    int32_t refcount;

    int32_t num_threads;
};
}
#endif