[OpenMP] Аналог rwlock

#ifndef _OMP_EXT_H_
#define _OMP_EXT_H_

#include <assert.h>
#include <omp.h>


typedef struct
{
	omp_lock_t lock;
	unsigned readers;
} omp_rwlock_t;


void  omp_init_rwlock (omp_rwlock_t *);
void  omp_destroy_rwlock (omp_rwlock_t *);
void  omp_set_rlock (omp_rwlock_t *);
void  omp_set_wlock (omp_rwlock_t *);
void  omp_unset_rlock (omp_rwlock_t *);
void  omp_unset_wlock (omp_rwlock_t *);
int   omp_test_rlock (omp_rwlock_t *);
int   omp_test_wlock (omp_rwlock_t *);


#endif //_OMP_EXT_H_

#include "omp-ext.h"


void omp_init_rwlock(omp_rwlock_t *lock)
{
	omp_init_lock(&lock->lock);
	lock->readers = 0;
}


void omp_destroy_rwlock(omp_rwlock_t *lock)
{
	omp_destroy_lock(&lock->lock);
}


void omp_set_rlock(omp_rwlock_t *lock)
{
#pragma omp critical(__omp_rwlock__)
{
	if(lock->readers == 0)
		omp_set_lock(&lock->lock);
	lock->readers++;
}
}


void omp_set_wlock(omp_rwlock_t *lock)
{
	omp_set_lock(&lock->lock);
	assert(lock->readers == 0);
}


void omp_unset_rlock(omp_rwlock_t *lock)
{
#pragma omp critical(__omp_rwlock__)
{
	assert(lock->readers > 0);
	lock->readers--;
	if(lock->readers == 0)
		omp_unset_lock(&lock->lock);
}
}


void omp_unset_wlock(omp_rwlock_t *lock)
{
	assert(lock->readers == 0);
	omp_unset_lock(&lock->lock);	
}


int omp_test_rlock(omp_rwlock_t *lock)
{
	int res = 1;
#pragma omp critical(__omp_rwlock__)
{
	if(lock->readers == 0)
		res = omp_test_lock(&lock->lock);
	if(res)
		lock->readers++;
}
	return res;
}


int omp_test_wlock(omp_rwlock_t *lock)
{
	int res = omp_test_lock(&lock->lock);
	if(res)
		assert(lock->readers == 0);
	return res;
}