1 /**********************************************************************
2 *
3 * Project: GDAL
4 * Purpose: Global thread pool
5 * Author: Even Rouault, <even dot rouault at spatialys dot com>
6 *
7 **********************************************************************
8 * Copyright (c) 2020, Even Rouault, <even dot rouault at spatialys dot com>
9 *
10 * Permission is hereby granted, free of charge, to any person obtaining a
11 * copy of this software and associated documentation files (the "Software"),
12 * to deal in the Software without restriction, including without limitation
13 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
14 * and/or sell copies of the Software, and to permit persons to whom the
15 * Software is furnished to do so, subject to the following conditions:
16 *
17 * The above copyright notice and this permission notice shall be included
18 * in all copies or substantial portions of the Software.
19 *
20 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
21 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
22 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
23 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
24 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
25 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
26 * DEALINGS IN THE SOFTWARE.
27 ****************************************************************************/
28
29 #include "gdal_thread_pool.h"
30
31 #include <mutex>
32
33 static std::mutex gMutexThreadPool;
34 static CPLWorkerThreadPool *gpoCompressThreadPool = nullptr;
35
GDALGetGlobalThreadPool(int nThreads)36 CPLWorkerThreadPool* GDALGetGlobalThreadPool(int nThreads)
37 {
38 std::lock_guard<std::mutex> oGuard(gMutexThreadPool);
39 if( gpoCompressThreadPool == nullptr )
40 {
41 gpoCompressThreadPool = new CPLWorkerThreadPool();
42 if( !gpoCompressThreadPool->Setup(nThreads, nullptr, nullptr) )
43 {
44 delete gpoCompressThreadPool;
45 gpoCompressThreadPool = nullptr;
46 }
47 }
48 else if( nThreads > gpoCompressThreadPool->GetThreadCount() )
49 {
50 // Increase size of thread pool
51 gpoCompressThreadPool->Setup(nThreads, nullptr, nullptr, false);
52 }
53 return gpoCompressThreadPool;
54 }
55
GDALDestroyGlobalThreadPool()56 void GDALDestroyGlobalThreadPool()
57 {
58 delete gpoCompressThreadPool;
59 gpoCompressThreadPool = nullptr;
60 }
61