AutoLock.h

Go to the documentation of this file.

//=============================================================================
//  Computational Process Networks class library
//  Copyright (C) 1997-2006  Gregory E. Allen and The University of Texas
//
//  This library is free software; you can redistribute it and/or modify it
//  under the terms of the GNU Library General Public License as published
//  by the Free Software Foundation; either version 2 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
//  Library General Public License for more details.
//
//  The GNU Public License is available in the file LICENSE, or you
//  can write to the Free Software Foundation, Inc., 59 Temple Place -
//  Suite 330, Boston, MA 02111-1307, USA, or you can find it on the
//  World Wide Web at http://www.fsf.org.
//=============================================================================
#ifndef AUTOLOCK_H
#define AUTOLOCK_H
#pragma once
#include <cpn/common.h>

template<class Lockable>
class AutoLock {
public:
    AutoLock(Lockable& mutex_) : mutex(mutex_), count(0) {
        Lock();
    }
    AutoLock(Lockable& mutex_, bool lock) : mutex(mutex_), count(0) {
        if (lock) { Lock(); }
    }
    
    ~AutoLock() {
        while (count > 0)
            Unlock();
    }

    void Unlock() {
        --count;
        mutex.Unlock();
    }

    void Lock() {
        mutex.Lock();
        ++count;
    }

private:
    Lockable& mutex;
    int count;
};
#endif