sprint/thread/semaphore.h

Go to the documentation of this file.

/******************************************************************************
*   SPRINT::semaphore
*
*   Copyright (C) 2005-2010  Paolo Medici <www.pmx.it>
*
 *  This library is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU Lesser General Public
 *  License as published by the Free Software Foundation; either
 *  version 2.1 of the License, or (at your option) any later version.
 *
 *  This library is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 *  Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public
 *  License along with this library; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 *
 *******************************************************************************/

#ifndef _SPRINT_SEMAPHORE_H
#define _SPRINT_SEMAPHORE_H

#include <sysdef.h>

namespace sprint {

#ifdef WIN32

class semaphore {
private:
        HANDLE hSemaphore;
public:
        semaphore(unsigned int queuesize)
        {
                hSemaphore = ::CreateSemaphore(NULL, 0, queuesize, NULL);
        }
        ~semaphore()
        {
                ::CloseHandle(hSemaphore);
        }

        inline long release()
        {
                long pc;
                ::ReleaseSemaphore(hSemaphore, 1, &pc);
                return pc;
        }

        inline bool wait()
        {
                return ::WaitForSingleObject(hSemaphore, INFINITE)==WAIT_OBJECT_0;
        }

        inline HANDLE gethandle()
        {
                return hSemaphore;
        }
};


#elif defined(__unix__)

#include <semaphore.h>

class semaphore {
 sem_t sem;
public:
        semaphore(unsigned int queuesize)
        {
        sem_init (&sem, FALSE, queuesize);
        }
        ~semaphore()
        {
        sem_destroy(&sem);
        }

        inline int release()
        {
        int value;
        sem_post(&sem);
        sem_getvalue(&sem, &value); /* se serve! */
        return value;
        }

        inline bool wait()
        {
        sem_wait(&sem);
        return true;
        }

};


#endif  // WIN32/UNIX

}

#endif  // _SPRINT_SEMAPHORE_H