shithub: tinygl

ref: 03db060992601a3e4b8be519c3ecdca2af0020a8
dir: /include-demo/lockstepthread.h/

/* Public Domain / CC0 3d Lock-Step Threading Implementation


Written by Gek (DMHSW) in 2020


*/

#ifndef LOCKSTEPTHREAD_H
#define LOCKSTEPTHREAD_H
#include <pthread.h>
#include <stdlib.h>
typedef struct {
	pthread_mutex_t myMutex;
	pthread_barrier_t myBarrier;
	pthread_t myThread;
	int isThreadLive;
	int shouldKillThread;
	int state;
	void (*execute)(void*);
	void* argument;
} lsthread;
void init_lsthread(lsthread* t);
void start_lsthread(lsthread* t);
void kill_lsthread(lsthread* t);
void destroy_lsthread(lsthread* t);
void lock(lsthread* t);
void step(lsthread* t);
void* lsthread_func(void* me_void);
#ifdef LOCKSTEPTHREAD_IMPL
//function declarations

void init_lsthread(lsthread* t){
	t->myMutex = (pthread_mutex_t)PTHREAD_MUTEX_INITIALIZER;
	pthread_barrier_init(&t->myBarrier, NULL, 2);
	t->isThreadLive = 0;
	t->shouldKillThread = 0;
	t->state = 0;
	t->execute = NULL;
	t->argument = NULL;
}
void destroy_lsthread(lsthread* t){
	pthread_mutex_destroy(&t->myMutex);
	pthread_barrier_destroy(&t->myBarrier);
}
void lock(lsthread* t){
	if(t->state == 1)return;//if already locked, nono
	if(!t->isThreadLive)return;
	//exit(1)
	pthread_barrier_wait(&t->myBarrier);
	//exit(1)
	if(pthread_mutex_lock(&t->myMutex))
		exit(1);
	t->state = 1;
	//exit(1)
}

void step(lsthread* t){
	if(t->state == -1)return; //if already stepping, nono
	if(!t->isThreadLive)return;
	//exit(1)
	if(pthread_mutex_unlock(&(t->myMutex)))
		exit(1);
	//exit(1)
	pthread_barrier_wait(&t->myBarrier);
	t->state = -1;
	//exit(1)
}
void kill_lsthread(lsthread* t){
	if(!t->isThreadLive)return;
	//exit(1)
	if(t->state != 1){
		lock(t);
		//exit(1)
	}
	t->shouldKillThread = 1;
	
	step(t);
	//exit(1)
	pthread_join(t->myThread,NULL);
	//if(pthread_kill(t->myThread)){
	//	exit(1)
	//}
	t->isThreadLive = 0;
	t->shouldKillThread = 0;
}
void* lsthread_func(void* me_void){
	lsthread* me = (lsthread*) me_void;
	int ret = 0;
	if (!me)pthread_exit(NULL);
	while (1) {
		//ret = pthread_cond_wait(&(me->myCond), &(me->myMutex));
		pthread_barrier_wait(&me->myBarrier);
		//exit(1)
		pthread_mutex_lock(&me->myMutex);
		//exit(1)
		//if(ret)pthread_exit(NULL);
		if (!(me->shouldKillThread) && me->execute)
			me->execute(me->argument);
		else if(me->shouldKillThread){
			pthread_mutex_unlock(&me->myMutex);
			//exit(1)
			//pthread_barrier_wait(&me->myBarrier);
			//exit(1)
			pthread_exit(NULL);
		}
		//exit(1)
		pthread_mutex_unlock(&me->myMutex);
		//exit(1)
		pthread_barrier_wait(&me->myBarrier);
		//exit(1)
	}
	pthread_exit(NULL);
}
void start_lsthread(lsthread* t){
	if(t->isThreadLive)return;
	t->isThreadLive = 1;
	t->shouldKillThread = 0;
	if(pthread_mutex_lock(&t->myMutex))
		exit(1);
	t->state = 1; //LOCKED
	pthread_create(
		&t->myThread,
		NULL,
		lsthread_func,
		(void*)t
	);
}
#endif
//end of implementation

#endif
//end of header