cli_var.hpp (2827B)
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 | /* Relacy Race Detector * Copyright (c) 2008-2010, Dmitry S. Vyukov * All rights reserved. * This software is provided AS-IS with no warranty, either express or implied. * This software is distributed under a license and may not be copied, * modified or distributed except as expressly authorized under the * terms of the license contained in the file LICENSE.TXT in this distribution. */ #ifndef RL_CLI_VAR_HPP #define RL_CLI_VAR_HPP #ifdef _MSC_VER # pragma once #endif #include "base.hpp" #include "atomic.hpp" namespace rl { template<typename T> class nvar; template<typename T> class nvar_proxy { public: typedef typename atomic_add_type<T>::type add_type; template<typename Y> friend class nvar; operator T () const { return load(); } T operator = (T value) { store(value); return value; } T operator = (nvar_proxy const& r) { T const value = r.load(); store(value); return *this; } T operator ++ (int) { T tmp = load(); store(tmp + 1); return tmp; } T operator -- (int) { T tmp = load(); store(tmp - 1); return tmp; } T operator ++ () { T tmp = load(); store(tmp + 1); return tmp + 1; } T operator -- () { T tmp = load(); store(tmp - 1); return tmp - 1; } T operator += (add_type value) { T tmp = load(); store(tmp + value); return tmp + value; } T operator -= (add_type value) { T tmp = load(); store(tmp - value); return tmp - value; } private: nvar<T>& var_; debug_info info_; nvar_proxy(nvar<T>& var, debug_info_param info) : var_(var) , info_(info) { } T load() const { return var_.load(mo_relaxed, info_); } void store(T value) { var_.store(value, mo_relaxed, info_); } }; template<typename T> class nvar : public generic_atomic<T, true> { public: typedef nvar_proxy<T> proxy_t; friend class nvar_proxy<T>; nvar() { } explicit nvar(T value) { this->store(value, mo_relaxed, $); } nvar(nvar const& r) { T const value = r.load(mo_relaxed, $); this->store(value, mo_relaxed, $); } nvar(proxy_t const& r) { T const value = r.load(); this->store(value, mo_relaxed, r.info_); } proxy_t operator () (debug_info_param info) { return proxy_t(*this, info); } private: nvar& operator = (nvar const&); }; } #endif |