TBB基础之初始化&终止

开始深入的TBB之旅之前，我们先看看怎么初始化和终止TBB库吧，毕竟这是使用TBB的一个基础～～～

TBB里提供了一个class：task_scheduler_init，该class会在constructor中初始化TBB，在destructor中终止TBB库。

这样我们就知道了最简单的初始化&终止TBB的方法：

1 #include "tbb/task_scheduler_init.h" 
2 using namespace tbb;
3  
4 int main() {
5     task_scheduler_init init;
6       ...
7     return 0;
8 }

 

这样，由task_scheduler_init的constructor和destructor就提供了初始化和终止的功能。

下面我们来看看task_scheduler_init的constructor是怎样的，打开header file：tbb/task_scheduler_init.h，找到它的constructor的declaration：

 1 //! Class representing reference to tbb scheduler. 
 2 /** A thread must construct a task_scheduler_init, and keep it alive,
 3     during the time that it uses the services of class task.
 4     @ingroup task_scheduling */ 
 5 class task_scheduler_init: internal::no_copy {
 6     /** NULL if not currently initialized. */ 
 7     internal::scheduler* my_scheduler;
 8 public:
 9     //! Typedef for number of threads that is automatic.
 10     static const int automatic = -1;
 11 
 12     //! Argument to initialize() or constructor that causes initialization to be deferred.
 13     static const int deferred = -2;
 14 
 15     //! Ensure that scheduler exists for this thread
 16     /** A value of -1 lets tbb decide on the number 
 17         of threads, which is typically the number of hardware threads. 
 18         For production code, the default value of -1 should be used, 
 19         particularly if the client code is mixed with third party clients 
 20         that might also use tbb.
 21 
 22         The number_of_threads is ignored if any other task_scheduler_inits 
 23         currently exist.  A thread may construct multiple task_scheduler_inits.  
 24         Doing so does no harm because the underlying scheduler is reference counted. */
 25     void initialize( int number_of_threads=automatic );
 26 
 27     //! Inverse of method initialize.
 28     void terminate();
 29 
 30     //! Shorthand for default constructor followed by call to intialize(number_of_threads).
 31     task_scheduler_init( int number_of_threads=automatic ) : my_scheduler(NULL)  {
 32         initialize( number_of_threads );
 33     }
 34   
 35     //! Destroy scheduler for this thread if thread has no other live task_scheduler_inits.
 36     ~task_scheduler_init() {
 37         if( my_scheduler ) 
 38             terminate();
 39         internal::poison_pointer( my_scheduler );
 40     }
 41 };

 

我们看到task_scheduler_init的constructor有一个缺省参数number_of_threads，默认情况下取值为automatic（-1），它的含义是指在constructor时自动调用initialize函数，并创建出线程调度器。

number_of_threads可能的取值包括：

• automatic（-1）：constructor时自动调用initialize()函数，创建合适的线程调度器
• deferred（-2）：表示在consturctor时不要调用initialize()函数初始化，而等到后面手动初始化
• 任意正整数：指定期望的线程数量，一般不要自己指定，除非你已经针对平台进行过特定的调节；当然我们也可以先使用deferred创建，然后在通过initialize(number_of_threads)来指定线程数量

我们来看一个典型的利用deferred参数来动态设定的例子：

 1 int main( int argc, char* argv[] ) {
 2     int nthread = strtol(argv[0],0,0);
 3     task_scheduler_init init(task_scheduler_init::deferred);
 4     if( nthread>=1 )
 5         init.initialize(nthread);
 6     // ... code that uses task scheduler only if nthread>=1 ... 
 7     if( nthread>=1 )
 8         init.terminate();
 9     return 0;
10 }

 

一个要注意的是：task_scheduler_init的构造是很费时的，不要每次使用TBB时都创建它，而是在main或者入口的地方创建一次就可以了。

 

在TBB深入部分我们会去看看task_scheduler_init是怎么实现线程调度器的～～～（待续）