mirror of
https://github.com/NGSolve/netgen.git
synced 2024-12-26 13:50:33 +05:00
192 lines
2.5 KiB
C++
192 lines
2.5 KiB
C++
#ifndef FILE_PARTHREADS
|
|
#define FILE_PARTHREADS
|
|
|
|
/**************************************************************************/
|
|
/* File: parthreads.hh */
|
|
/* Author: Joachim Schoeberl */
|
|
/* Date: 22. Nov. 2000 */
|
|
/**************************************************************************/
|
|
|
|
/*
|
|
Parallel thread, Mutex,
|
|
*/
|
|
|
|
namespace netgen
|
|
{
|
|
|
|
#ifdef NO_PARALLEL_THREADS
|
|
|
|
class NgMutex { };
|
|
|
|
class NgLock
|
|
{
|
|
public:
|
|
NgLock (NgMutex & mut, bool lock = 0) { ; }
|
|
void Lock () { ; }
|
|
void UnLock () { ; }
|
|
};
|
|
|
|
|
|
#else
|
|
|
|
#ifdef _MSC_VER
|
|
|
|
#ifdef MSVC_EXPRESS
|
|
// #include <pthread.h>
|
|
|
|
class NgMutex
|
|
{
|
|
pthread_mutex_t mut;
|
|
public:
|
|
NgMutex ()
|
|
{
|
|
pthread_mutex_init (&mut, NULL);
|
|
}
|
|
friend class NgLock;
|
|
};
|
|
|
|
class NgLock
|
|
{
|
|
pthread_mutex_t & mut;
|
|
bool locked;
|
|
public:
|
|
NgLock (NgMutex & ngmut, bool lock = false)
|
|
: mut (ngmut.mut)
|
|
{
|
|
if (lock)
|
|
pthread_mutex_lock (&mut);
|
|
|
|
locked = lock;
|
|
};
|
|
|
|
~NgLock()
|
|
{
|
|
if (locked)
|
|
pthread_mutex_unlock (&mut);
|
|
}
|
|
|
|
void Lock ()
|
|
{
|
|
pthread_mutex_lock (&mut);
|
|
locked = true;
|
|
}
|
|
void UnLock ()
|
|
{
|
|
pthread_mutex_unlock (&mut);
|
|
locked = false;
|
|
}
|
|
/*
|
|
int TryLock ()
|
|
{
|
|
return pthread_mutex_trylock (&mut);
|
|
}
|
|
*/
|
|
};
|
|
|
|
#else // Using MS VC++ Standard / Enterprise / Professional edition...
|
|
|
|
|
|
class NgMutex
|
|
{
|
|
CCriticalSection cs;
|
|
|
|
public:
|
|
NgMutex ()
|
|
{ ; }
|
|
friend class NgLock;
|
|
};
|
|
|
|
class NgLock
|
|
{
|
|
CSingleLock sl;
|
|
bool locked;
|
|
public:
|
|
NgLock (NgMutex & mut, bool lock = 0)
|
|
: sl(&mut.cs)
|
|
{
|
|
if (lock) sl.Lock();
|
|
locked = lock;
|
|
}
|
|
|
|
~NgLock ()
|
|
{
|
|
if (locked) sl.Unlock();
|
|
}
|
|
|
|
void Lock ()
|
|
{
|
|
sl.Lock();
|
|
locked = 1;
|
|
}
|
|
|
|
void UnLock ()
|
|
{
|
|
sl.Unlock();
|
|
locked = 0;
|
|
}
|
|
};
|
|
|
|
#endif // MSVC_EXPRESS
|
|
|
|
#else
|
|
|
|
|
|
// #include <pthread.h>
|
|
|
|
class NgMutex
|
|
{
|
|
pthread_mutex_t mut;
|
|
public:
|
|
NgMutex ()
|
|
{
|
|
pthread_mutex_init (&mut, NULL);
|
|
}
|
|
friend class NgLock;
|
|
};
|
|
|
|
class NgLock
|
|
{
|
|
pthread_mutex_t & mut;
|
|
bool locked;
|
|
public:
|
|
NgLock (NgMutex & ngmut, bool lock = false)
|
|
: mut (ngmut.mut)
|
|
{
|
|
if (lock)
|
|
pthread_mutex_lock (&mut);
|
|
|
|
locked = lock;
|
|
};
|
|
|
|
~NgLock()
|
|
{
|
|
if (locked)
|
|
pthread_mutex_unlock (&mut);
|
|
}
|
|
|
|
void Lock ()
|
|
{
|
|
pthread_mutex_lock (&mut);
|
|
locked = true;
|
|
}
|
|
void UnLock ()
|
|
{
|
|
pthread_mutex_unlock (&mut);
|
|
locked = false;
|
|
}
|
|
/*
|
|
int TryLock ()
|
|
{
|
|
return pthread_mutex_trylock (&mut);
|
|
}
|
|
*/
|
|
};
|
|
|
|
#endif
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
#endif
|