123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164 |
- /*
- Copyright 2005-2013 Intel Corporation. All Rights Reserved.
- This file is part of Threading Building Blocks.
- Threading Building Blocks is free software; you can redistribute it
- and/or modify it under the terms of the GNU General Public License
- version 2 as published by the Free Software Foundation.
- Threading Building Blocks 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 Threading Building Blocks; if not, write to the Free Software
- Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
- As a special exception, you may use this file as part of a free software
- library without restriction. Specifically, if other files instantiate
- templates or use macros or inline functions from this file, or you compile
- this file and link it with other files to produce an executable, this
- file does not by itself cause the resulting executable to be covered by
- the GNU General Public License. This exception does not however
- invalidate any other reasons why the executable file might be covered by
- the GNU General Public License.
- */
- #ifndef __TBB_task_scheduler_observer_H
- #define __TBB_task_scheduler_observer_H
- #include "atomic.h"
- #if __TBB_TASK_ARENA
- #include "task_arena.h"
- #endif //__TBB_TASK_ARENA
- #if __TBB_SCHEDULER_OBSERVER
- namespace tbb {
- namespace interface6 {
- class task_scheduler_observer;
- }
- namespace internal {
- class observer_proxy;
- class observer_list;
- class task_scheduler_observer_v3 {
- friend class observer_proxy;
- friend class observer_list;
- friend class interface6::task_scheduler_observer;
- //! Pointer to the proxy holding this observer.
- /** Observers are proxied by the scheduler to maintain persistent lists of them. **/
- observer_proxy* my_proxy;
- //! Counter preventing the observer from being destroyed while in use by the scheduler.
- /** Valid only when observation is on. **/
- atomic<intptr_t> my_busy_count;
- public:
- //! Enable or disable observation
- /** For local observers the method can be used only when the current thread
- has the task scheduler initialized or is attached to an arena.
- Repeated calls with the same state are no-ops. **/
- void __TBB_EXPORTED_METHOD observe( bool state=true );
- //! Returns true if observation is enabled, false otherwise.
- bool is_observing() const {return my_proxy!=NULL;}
- //! Construct observer with observation disabled.
- task_scheduler_observer_v3() : my_proxy(NULL) { my_busy_count.store<relaxed>(0); }
- //! Entry notification
- /** Invoked from inside observe(true) call and whenever a worker enters the arena
- this observer is associated with. If a thread is already in the arena when
- the observer is activated, the entry notification is called before it
- executes the first stolen task.
- Obsolete semantics. For global observers it is called by a thread before
- the first steal since observation became enabled. **/
- virtual void on_scheduler_entry( bool /*is_worker*/ ) {}
- //! Exit notification
- /** Invoked from inside observe(false) call and whenever a worker leaves the
- arena this observer is associated with.
- Obsolete semantics. For global observers it is called by a thread before
- the first steal since observation became enabled. **/
- virtual void on_scheduler_exit( bool /*is_worker*/ ) {}
- //! Destructor automatically switches observation off if it is enabled.
- virtual ~task_scheduler_observer_v3() { if(my_proxy) observe(false);}
- };
- } // namespace internal
- #if TBB_PREVIEW_LOCAL_OBSERVER
- namespace interface6 {
- class task_scheduler_observer : public internal::task_scheduler_observer_v3 {
- friend class internal::task_scheduler_observer_v3;
- friend class internal::observer_proxy;
- friend class internal::observer_list;
- /** Negative numbers with the largest absolute value to minimize probability
- of coincidence in case of a bug in busy count usage. **/
- // TODO: take more high bits for version number
- static const intptr_t v6_trait = (intptr_t)((~(uintptr_t)0 >> 1) + 1);
- //! contains task_arena pointer or tag indicating local or global semantics of the observer
- intptr_t my_context_tag;
- enum { global_tag = 0, implicit_tag = 1 };
- public:
- //! Construct local or global observer in inactive state (observation disabled).
- /** For a local observer entry/exit notifications are invoked whenever a worker
- thread joins/leaves the arena of the observer's owner thread. If a thread is
- already in the arena when the observer is activated, the entry notification is
- called before it executes the first stolen task. **/
- /** TODO: Obsolete.
- Global observer semantics is obsolete as it violates master thread isolation
- guarantees and is not composable. Thus the current default behavior of the
- constructor is obsolete too and will be changed in one of the future versions
- of the library. **/
- task_scheduler_observer( bool local = false ) {
- my_busy_count.store<relaxed>(v6_trait);
- my_context_tag = local? implicit_tag : global_tag;
- }
- #if __TBB_TASK_ARENA
- //! Construct local observer for a given arena in inactive state (observation disabled).
- /** entry/exit notifications are invoked whenever a thread joins/leaves arena.
- If a thread is already in the arena when the observer is activated, the entry notification
- is called before it executes the first stolen task. **/
- task_scheduler_observer( task_arena & a) {
- my_busy_count.store<relaxed>(v6_trait);
- my_context_tag = (intptr_t)&a;
- }
- #endif //__TBB_TASK_ARENA
- //! The callback can be invoked in a worker thread before it leaves an arena.
- /** If it returns false, the thread remains in the arena. Will not be called for masters
- or if the worker leaves arena due to rebalancing or priority changes, etc.
- NOTE: The preview library must be linked for this method to take effect **/
- virtual bool on_scheduler_leaving() { return true; }
- //! Destructor additionally protects concurrent on_scheduler_leaving notification
- // It is recommended to disable observation before destructor of a derived class starts,
- // otherwise it can lead to concurrent notification callback on partly destroyed object
- virtual ~task_scheduler_observer() { if(my_proxy) observe(false);}
- };
- } //namespace interface6
- using interface6::task_scheduler_observer;
- #else /*TBB_PREVIEW_LOCAL_OBSERVER*/
- typedef tbb::internal::task_scheduler_observer_v3 task_scheduler_observer;
- #endif /*TBB_PREVIEW_LOCAL_OBSERVER*/
- } // namespace tbb
- #endif /* __TBB_SCHEDULER_OBSERVER */
- #endif /* __TBB_task_scheduler_observer_H */
|