1 // Ceres Solver - A fast non-linear least squares minimizer
2 // Copyright 2010, 2011, 2012 Google Inc. All rights reserved.
3 // http://code.google.com/p/ceres-solver/
4 //
5 // Redistribution and use in source and binary forms, with or without
6 // modification, are permitted provided that the following conditions are met:
7 //
8 // * Redistributions of source code must retain the above copyright notice,
9 // this list of conditions and the following disclaimer.
10 // * Redistributions in binary form must reproduce the above copyright notice,
11 // this list of conditions and the following disclaimer in the documentation
12 // and/or other materials provided with the distribution.
13 // * Neither the name of Google Inc. nor the names of its contributors may be
14 // used to endorse or promote products derived from this software without
15 // specific prior written permission.
16 //
17 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
18 // AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 // ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
21 // LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
22 // CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
23 // SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
24 // INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
25 // CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
26 // ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
27 // POSSIBILITY OF SUCH DAMAGE.
28 //
29 // Author: Craig Silverstein.
30 //
31 // A simple mutex wrapper, supporting locks and read-write locks.
32 // You should assume the locks are *not* re-entrant.
33 //
34 // This class is meant to be internal-only and should be wrapped by an
35 // internal namespace. Before you use this module, please give the
36 // name of your internal namespace for this module. Or, if you want
37 // to expose it, you'll want to move it to the Google namespace. We
38 // cannot put this class in global namespace because there can be some
39 // problems when we have multiple versions of Mutex in each shared object.
40 //
41 // NOTE: by default, we have #ifdef'ed out the TryLock() method.
42 // This is for two reasons:
43 // 1) TryLock() under Windows is a bit annoying (it requires a
44 // #define to be defined very early).
45 // 2) TryLock() is broken for NO_THREADS mode, at least in NDEBUG
46 // mode.
47 // If you need TryLock(), and either these two caveats are not a
48 // problem for you, or you're willing to work around them, then
49 // feel free to #define GMUTEX_TRYLOCK, or to remove the #ifdefs
50 // in the code below.
51 //
52 // CYGWIN NOTE: Cygwin support for rwlock seems to be buggy:
53 // http://www.cygwin.com/ml/cygwin/2008-12/msg00017.html
54 // Because of that, we might as well use windows locks for
55 // cygwin. They seem to be more reliable than the cygwin pthreads layer.
56 //
57 // TRICKY IMPLEMENTATION NOTE:
58 // This class is designed to be safe to use during
59 // dynamic-initialization -- that is, by global constructors that are
60 // run before main() starts. The issue in this case is that
61 // dynamic-initialization happens in an unpredictable order, and it
62 // could be that someone else's dynamic initializer could call a
63 // function that tries to acquire this mutex -- but that all happens
64 // before this mutex's constructor has run. (This can happen even if
65 // the mutex and the function that uses the mutex are in the same .cc
66 // file.) Basically, because Mutex does non-trivial work in its
67 // constructor, it's not, in the naive implementation, safe to use
68 // before dynamic initialization has run on it.
69 //
70 // The solution used here is to pair the actual mutex primitive with a
71 // bool that is set to true when the mutex is dynamically initialized.
72 // (Before that it's false.) Then we modify all mutex routines to
73 // look at the bool, and not try to lock/unlock until the bool makes
74 // it to true (which happens after the Mutex constructor has run.)
75 //
76 // This works because before main() starts -- particularly, during
77 // dynamic initialization -- there are no threads, so a) it's ok that
78 // the mutex operations are a no-op, since we don't need locking then
79 // anyway; and b) we can be quite confident our bool won't change
80 // state between a call to Lock() and a call to Unlock() (that would
81 // require a global constructor in one translation unit to call Lock()
82 // and another global constructor in another translation unit to call
83 // Unlock() later, which is pretty perverse).
84 //
85 // That said, it's tricky, and can conceivably fail; it's safest to
86 // avoid trying to acquire a mutex in a global constructor, if you
87 // can. One way it can fail is that a really smart compiler might
88 // initialize the bool to true at static-initialization time (too
89 // early) rather than at dynamic-initialization time. To discourage
90 // that, we set is_safe_ to true in code (not the constructor
91 // colon-initializer) and set it to true via a function that always
92 // evaluates to true, but that the compiler can't know always
93 // evaluates to true. This should be good enough.
94
95 #ifndef CERES_INTERNAL_MUTEX_H_
96 #define CERES_INTERNAL_MUTEX_H_
97
98 #include "ceres/internal/port.h"
99
100 #if defined(CERES_NO_THREADS)
101 typedef int MutexType; // to keep a lock-count
102 #elif defined(_WIN32) || defined(__CYGWIN32__) || defined(__CYGWIN64__)
103 # define CERES_WIN32_LEAN_AND_MEAN // We only need minimal includes
104 # ifdef CERES_GMUTEX_TRYLOCK
105 // We need Windows NT or later for TryEnterCriticalSection(). If you
106 // don't need that functionality, you can remove these _WIN32_WINNT
107 // lines, and change TryLock() to assert(0) or something.
108 # ifndef _WIN32_WINNT
109 # define _WIN32_WINNT 0x0400
110 # endif
111 # endif
112 // Unfortunately, windows.h defines a bunch of macros with common
113 // names. Two in particular need avoiding: ERROR and min/max.
114 // To avoid macro definition of ERROR.
115 # define NOGDI
116 // To avoid macro definition of min/max.
117 # ifndef NOMINMAX
118 # define NOMINMAX
119 # endif
120 # include <windows.h>
121 typedef CRITICAL_SECTION MutexType;
122 #elif defined(CERES_HAVE_PTHREAD) && defined(CERES_HAVE_RWLOCK)
123 // Needed for pthread_rwlock_*. If it causes problems, you could take it
124 // out, but then you'd have to unset CERES_HAVE_RWLOCK (at least on linux --
125 // it *does* cause problems for FreeBSD, or MacOSX, but isn't needed for
126 // locking there.)
127 # if defined(__linux__) && !defined(_XOPEN_SOURCE)
128 # define _XOPEN_SOURCE 500 // may be needed to get the rwlock calls
129 # endif
130 # include <pthread.h>
131 typedef pthread_rwlock_t MutexType;
132 #elif defined(CERES_HAVE_PTHREAD)
133 # include <pthread.h>
134 typedef pthread_mutex_t MutexType;
135 #else
136 # error Need to implement mutex.h for your architecture, or #define NO_THREADS
137 #endif
138
139 // We need to include these header files after defining _XOPEN_SOURCE
140 // as they may define the _XOPEN_SOURCE macro.
141 #include <assert.h>
142 #include <stdlib.h> // for abort()
143
144 namespace ceres {
145 namespace internal {
146
147 class Mutex {
148 public:
149 // Create a Mutex that is not held by anybody. This constructor is
150 // typically used for Mutexes allocated on the heap or the stack.
151 // See below for a recommendation for constructing global Mutex
152 // objects.
153 inline Mutex();
154
155 // Destructor
156 inline ~Mutex();
157
158 inline void Lock(); // Block if needed until free then acquire exclusively
159 inline void Unlock(); // Release a lock acquired via Lock()
160 #ifdef CERES_GMUTEX_TRYLOCK
161 inline bool TryLock(); // If free, Lock() and return true, else return false
162 #endif
163 // Note that on systems that don't support read-write locks, these may
164 // be implemented as synonyms to Lock() and Unlock(). So you can use
165 // these for efficiency, but don't use them anyplace where being able
166 // to do shared reads is necessary to avoid deadlock.
167 inline void ReaderLock(); // Block until free or shared then acquire a share
168 inline void ReaderUnlock(); // Release a read share of this Mutex
WriterLock()169 inline void WriterLock() { Lock(); } // Acquire an exclusive lock
WriterUnlock()170 inline void WriterUnlock() { Unlock(); } // Release a lock from WriterLock()
171
172 // TODO(hamaji): Do nothing, implement correctly.
AssertHeld()173 inline void AssertHeld() {}
174
175 private:
176 MutexType mutex_;
177 // We want to make sure that the compiler sets is_safe_ to true only
178 // when we tell it to, and never makes assumptions is_safe_ is
179 // always true. volatile is the most reliable way to do that.
180 volatile bool is_safe_;
181
SetIsSafe()182 inline void SetIsSafe() { is_safe_ = true; }
183
184 // Catch the error of writing Mutex when intending MutexLock.
Mutex(Mutex *)185 Mutex(Mutex* /*ignored*/) {}
186 // Disallow "evil" constructors
187 Mutex(const Mutex&);
188 void operator=(const Mutex&);
189 };
190
191 // Now the implementation of Mutex for various systems
192 #if defined(CERES_NO_THREADS)
193
194 // When we don't have threads, we can be either reading or writing,
195 // but not both. We can have lots of readers at once (in no-threads
196 // mode, that's most likely to happen in recursive function calls),
197 // but only one writer. We represent this by having mutex_ be -1 when
198 // writing and a number > 0 when reading (and 0 when no lock is held).
199 //
200 // In debug mode, we assert these invariants, while in non-debug mode
201 // we do nothing, for efficiency. That's why everything is in an
202 // assert.
203
Mutex()204 Mutex::Mutex() : mutex_(0) { }
~Mutex()205 Mutex::~Mutex() { assert(mutex_ == 0); }
Lock()206 void Mutex::Lock() { assert(--mutex_ == -1); }
Unlock()207 void Mutex::Unlock() { assert(mutex_++ == -1); }
208 #ifdef CERES_GMUTEX_TRYLOCK
TryLock()209 bool Mutex::TryLock() { if (mutex_) return false; Lock(); return true; }
210 #endif
ReaderLock()211 void Mutex::ReaderLock() { assert(++mutex_ > 0); }
ReaderUnlock()212 void Mutex::ReaderUnlock() { assert(mutex_-- > 0); }
213
214 #elif defined(_WIN32) || defined(__CYGWIN32__) || defined(__CYGWIN64__)
215
Mutex()216 Mutex::Mutex() { InitializeCriticalSection(&mutex_); SetIsSafe(); }
~Mutex()217 Mutex::~Mutex() { DeleteCriticalSection(&mutex_); }
Lock()218 void Mutex::Lock() { if (is_safe_) EnterCriticalSection(&mutex_); }
Unlock()219 void Mutex::Unlock() { if (is_safe_) LeaveCriticalSection(&mutex_); }
220 #ifdef GMUTEX_TRYLOCK
TryLock()221 bool Mutex::TryLock() { return is_safe_ ?
222 TryEnterCriticalSection(&mutex_) != 0 : true; }
223 #endif
ReaderLock()224 void Mutex::ReaderLock() { Lock(); } // we don't have read-write locks
ReaderUnlock()225 void Mutex::ReaderUnlock() { Unlock(); }
226
227 #elif defined(CERES_HAVE_PTHREAD) && defined(CERES_HAVE_RWLOCK)
228
229 #define CERES_SAFE_PTHREAD(fncall) do { /* run fncall if is_safe_ is true */ \
230 if (is_safe_ && fncall(&mutex_) != 0) abort(); \
231 } while (0)
232
Mutex()233 Mutex::Mutex() {
234 SetIsSafe();
235 if (is_safe_ && pthread_rwlock_init(&mutex_, NULL) != 0) abort();
236 }
~Mutex()237 Mutex::~Mutex() { CERES_SAFE_PTHREAD(pthread_rwlock_destroy); }
Lock()238 void Mutex::Lock() { CERES_SAFE_PTHREAD(pthread_rwlock_wrlock); }
Unlock()239 void Mutex::Unlock() { CERES_SAFE_PTHREAD(pthread_rwlock_unlock); }
240 #ifdef CERES_GMUTEX_TRYLOCK
TryLock()241 bool Mutex::TryLock() { return is_safe_ ?
242 pthread_rwlock_trywrlock(&mutex_) == 0 :
243 true; }
244 #endif
ReaderLock()245 void Mutex::ReaderLock() { CERES_SAFE_PTHREAD(pthread_rwlock_rdlock); }
ReaderUnlock()246 void Mutex::ReaderUnlock() { CERES_SAFE_PTHREAD(pthread_rwlock_unlock); }
247 #undef CERES_SAFE_PTHREAD
248
249 #elif defined(CERES_HAVE_PTHREAD)
250
251 #define CERES_SAFE_PTHREAD(fncall) do { /* run fncall if is_safe_ is true */ \
252 if (is_safe_ && fncall(&mutex_) != 0) abort(); \
253 } while (0)
254
Mutex()255 Mutex::Mutex() {
256 SetIsSafe();
257 if (is_safe_ && pthread_mutex_init(&mutex_, NULL) != 0) abort();
258 }
~Mutex()259 Mutex::~Mutex() { CERES_SAFE_PTHREAD(pthread_mutex_destroy); }
Lock()260 void Mutex::Lock() { CERES_SAFE_PTHREAD(pthread_mutex_lock); }
Unlock()261 void Mutex::Unlock() { CERES_SAFE_PTHREAD(pthread_mutex_unlock); }
262 #ifdef CERES_GMUTEX_TRYLOCK
TryLock()263 bool Mutex::TryLock() { return is_safe_ ?
264 pthread_mutex_trylock(&mutex_) == 0 : true; }
265 #endif
ReaderLock()266 void Mutex::ReaderLock() { Lock(); }
ReaderUnlock()267 void Mutex::ReaderUnlock() { Unlock(); }
268 #undef CERES_SAFE_PTHREAD
269
270 #endif
271
272 // --------------------------------------------------------------------------
273 // Some helper classes
274
275 // Note: The weird "Ceres" prefix for the class is a workaround for having two
276 // similar mutex.h files included in the same translation unit. This is a
277 // problem because macros do not respect C++ namespaces, and as a result, this
278 // does not work well (e.g. inside Chrome). The offending macros are
279 // "MutexLock(x) COMPILE_ASSERT(false)". To work around this, "Ceres" is
280 // prefixed to the class names; this permits defining the classes.
281
282 // CeresMutexLock(mu) acquires mu when constructed and releases it
283 // when destroyed.
284 class CeresMutexLock {
285 public:
CeresMutexLock(Mutex * mu)286 explicit CeresMutexLock(Mutex *mu) : mu_(mu) { mu_->Lock(); }
~CeresMutexLock()287 ~CeresMutexLock() { mu_->Unlock(); }
288 private:
289 Mutex * const mu_;
290 // Disallow "evil" constructors
291 CeresMutexLock(const CeresMutexLock&);
292 void operator=(const CeresMutexLock&);
293 };
294
295 // CeresReaderMutexLock and CeresWriterMutexLock do the same, for rwlocks
296 class CeresReaderMutexLock {
297 public:
CeresReaderMutexLock(Mutex * mu)298 explicit CeresReaderMutexLock(Mutex *mu) : mu_(mu) { mu_->ReaderLock(); }
~CeresReaderMutexLock()299 ~CeresReaderMutexLock() { mu_->ReaderUnlock(); }
300 private:
301 Mutex * const mu_;
302 // Disallow "evil" constructors
303 CeresReaderMutexLock(const CeresReaderMutexLock&);
304 void operator=(const CeresReaderMutexLock&);
305 };
306
307 class CeresWriterMutexLock {
308 public:
CeresWriterMutexLock(Mutex * mu)309 explicit CeresWriterMutexLock(Mutex *mu) : mu_(mu) { mu_->WriterLock(); }
~CeresWriterMutexLock()310 ~CeresWriterMutexLock() { mu_->WriterUnlock(); }
311 private:
312 Mutex * const mu_;
313 // Disallow "evil" constructors
314 CeresWriterMutexLock(const CeresWriterMutexLock&);
315 void operator=(const CeresWriterMutexLock&);
316 };
317
318 // Catch bug where variable name is omitted, e.g. MutexLock (&mu);
319 #define CeresMutexLock(x) \
320 COMPILE_ASSERT(0, ceres_mutex_lock_decl_missing_var_name)
321 #define CeresReaderMutexLock(x) \
322 COMPILE_ASSERT(0, ceres_rmutex_lock_decl_missing_var_name)
323 #define CeresWriterMutexLock(x) \
324 COMPILE_ASSERT(0, ceres_wmutex_lock_decl_missing_var_name)
325
326 } // namespace internal
327 } // namespace ceres
328
329 #endif // CERES_INTERNAL_MUTEX_H_
330