- Main Page
- Related Pages
- Classes
- Files

- File List

# ThreadPool.h

    00001 #ifndef OPENMM_THREAD_POOL_H_
    00002 #define OPENMM_THREAD_POOL_H_
    00003
    00004 /* -------------------------------------------------------------------------- *
    00005  *                                   OpenMM                                   *
    00006  * -------------------------------------------------------------------------- *
    00007  * This is part of the OpenMM molecular simulation toolkit originating from   *
    00008  * Simbios, the NIH National Center for Physics-Based Simulation of           *
    00009  * Biological Structures at Stanford, funded under the NIH Roadmap for        *
    00010  * Medical Research, grant U54 GM072970. See https://simtk.org.               *
    00011  *                                                                            *
    00012  * Portions copyright (c) 2013 Stanford University and the Authors.           *
    00013  * Authors: Peter Eastman                                                     *
    00014  * Contributors:                                                              *
    00015  *                                                                            *
    00016  * Permission is hereby granted, free of charge, to any person obtaining a    *
    00017  * copy of this software and associated documentation files (the "Software"), *
    00018  * to deal in the Software without restriction, including without limitation  *
    00019  * the rights to use, copy, modify, merge, publish, distribute, sublicense,   *
    00020  * and/or sell copies of the Software, and to permit persons to whom the      *
    00021  * Software is furnished to do so, subject to the following conditions:       *
    00022  *                                                                            *
    00023  * The above copyright notice and this permission notice shall be included in *
    00024  * all copies or substantial portions of the Software.                        *
    00025  *                                                                            *
    00026  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR *
    00027  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,   *
    00028  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL    *
    00029  * THE AUTHORS, CONTRIBUTORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,    *
    00030  * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR      *
    00031  * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE  *
    00032  * USE OR OTHER DEALINGS IN THE SOFTWARE.                                     *
    00033  * -------------------------------------------------------------------------- */
    00034
    00035 #define NOMINMAX
    00036 #include "windowsExport.h"
    00037 #include <pthread.h>
    00038 #include <vector>
    00039
    00040 namespace OpenMM {
    00041
    00052 class OPENMM_EXPORT ThreadPool {
    00053 public:
    00054     class Task;
    00055     class ThreadData;
    00062     ThreadPool(int numThreads=0);
    00063     ~ThreadPool();
    00067     int getNumThreads() const;
    00071     void execute(Task& task);
    00076     void syncThreads();
    00081     void waitForThreads();
    00085     void resumeThreads();
    00086 private:
    00087     bool isDeleted;
    00088     int numThreads, waitCount;
    00089     std::vector<pthread_t> thread;
    00090     std::vector<ThreadData*> threadData;
    00091     pthread_cond_t startCondition, endCondition;
    00092     pthread_mutex_t lock;
    00093 };
    00094
    00098 class OPENMM_EXPORT ThreadPool::Task {
    00099 public:
    00106     virtual void execute(ThreadPool& pool, int threadIndex) = 0;
    00107 };
    00108
    00109 } // namespace OpenMM
    00110
    00111 #endif // OPENMM_THREAD_POOL_H_