threads.cpp

Go to the documentation of this file.

/*
Copyright(c) 1997-2001 Id Software, Inc.
Copyright(c) 2002 The Quakeforge Project.
Copyright(c) 2006 Quake2World.

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 2
of the License, or(at your option) any later version.

This program 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 General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

*/

#include "bsp.h"
#include "../../shared/thread.h"

#define MAX_THREADS 8

threadstate_t threadstate;

static int GetThreadWork (void)
{
    int r;
    int f;

    ThreadLock();

    if (threadstate.workindex == threadstate.workcount) {  /* done */
        ThreadUnlock();
        return -1;
    }

    /* update work fraction and output progress if desired */
    f = 10 * threadstate.workindex / threadstate.workcount;
    if (f != threadstate.workfrac) {
        threadstate.workfrac = f;
        if (threadstate.progress) {
            fprintf(stdout, "%i...", f);
            fflush(stdout);
        }
    } else if (threadstate.progress && threadstate.workindex % threadstate.worktick == 0) {
        fprintf(stdout, "%c\b", "-\\|/"[(threadstate.workindex / threadstate.worktick) & 3]);
        fflush(stdout);
    }

    /* assign the next work iteration */
    r = threadstate.workindex;
    threadstate.workindex++;

    ThreadUnlock();

    return r;
}


static void (*WorkFunction) (unsigned int);


static int ThreadWork (void* p)
{
    while (true) {
        int work = GetThreadWork();
        if (work == -1)
            break;
        WorkFunction(work);
    }

    return 0;
}


static SDL_mutex *lock = nullptr;

static void ThreadInit (void)
{
    if (lock != nullptr)
        Sys_Error("Mutex already created!");

    lock = SDL_CreateMutex();
}

static void ThreadRelease (void)
{
    SDL_DestroyMutex(lock);
    lock = nullptr;
}

void ThreadLock (void)
{
    if (threadstate.numthreads == 1) {
        /* do nothing */
    } else if (lock != nullptr && SDL_LockMutex(lock) != -1) {
        /* already locked */
    } else {
        Sys_Error("Couldn't lock mutex (%p)!", (void*)lock);
    }
}

void ThreadUnlock (void)
{
    if (threadstate.numthreads == 1) {
        /* do nothing */
    } else if (lock != nullptr && SDL_UnlockMutex(lock) != -1) {
        /* already locked */
    } else {
        Sys_Error("Couldn't unlock mutex (%p)!", (void*)lock);
    }
}

static void RunThreads (void)
{
    SDL_Thread *threads[MAX_THREADS];
    int i;

    if (threadstate.numthreads == 1) {
        ThreadWork(nullptr);
        return;
    }

    ThreadInit();

    for (i = 0; i < threadstate.numthreads; i++)
        threads[i] = Com_CreateThread(ThreadWork, "CompileWorker");

    for (i = 0; i < threadstate.numthreads; i++)
        SDL_WaitThread(threads[i], nullptr);

    ThreadRelease();
}


void RunThreadsOn (void (*func)(unsigned int), int unsigned workcount, bool progress, const char* id)
{
    int start, end;

    if (threadstate.numthreads < 1)  /* ensure safe thread counts */
        threadstate.numthreads = 1;

    if (threadstate.numthreads > MAX_THREADS)
        threadstate.numthreads = MAX_THREADS;

    threadstate.workindex = 0;
    threadstate.workcount = workcount;
    threadstate.workfrac = -1;
    threadstate.progress = progress;
    threadstate.worktick = sqrt(workcount) + 1;

    WorkFunction = func;

    start = time(nullptr);

    if (threadstate.progress) {
        fprintf(stdout, "%10s: ", id);
        fflush(stdout);
    }

    RunThreads();

    end = time(nullptr);

    if (threadstate.progress) {
        Verb_Printf(VERB_NORMAL, " (time: %6is, #: %i)\n", end - start, workcount);
    }
}

void RunSingleThreadOn (void (*func)(unsigned int), int unsigned workcount, bool progress, const char* id)
{
    int saved_numthreads = threadstate.numthreads;

    threadstate.numthreads = 1;

    RunThreadsOn(func, workcount, progress, id);

    threadstate.numthreads = saved_numthreads;
}