Sirikata
|
00001 /* Sirikata 00002 * CountResourceMonitor.hpp 00003 * 00004 * Copyright (c) 2009, Ewen Cheslack-Postava 00005 * All rights reserved. 00006 * 00007 * Redistribution and use in source and binary forms, with or without 00008 * modification, are permitted provided that the following conditions are 00009 * met: 00010 * * Redistributions of source code must retain the above copyright 00011 * notice, this list of conditions and the following disclaimer. 00012 * * Redistributions in binary form must reproduce the above copyright 00013 * notice, this list of conditions and the following disclaimer in 00014 * the documentation and/or other materials provided with the 00015 * distribution. 00016 * * Neither the name of Sirikata nor the names of its contributors may 00017 * be used to endorse or promote products derived from this software 00018 * without specific prior written permission. 00019 * 00020 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS 00021 * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 00022 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A 00023 * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER 00024 * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 00025 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 00026 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 00027 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 00028 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 00029 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 00030 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 00031 */ 00032 00033 #ifndef _SIRIKATA_COUNT_RESOURCE_MONITOR_HPP_ 00034 #define _SIRIKATA_COUNT_RESOURCE_MONITOR_HPP_ 00035 00036 #include <sirikata/core/util/Platform.hpp> 00037 #include <sirikata/core/util/AtomicTypes.hpp> 00038 00039 namespace Sirikata { 00040 00041 class CountResourceMonitor { 00042 Sirikata::AtomicValue<int32> mSize; 00043 const int32 mLimit; 00044 public: 00045 void reset() { 00046 mSize=0; 00047 } 00048 uint32 maxSize() const { 00049 return (uint32)mLimit; 00050 } 00051 uint32 filledSize() const { 00052 return (uint32)mSize.read(); 00053 } 00054 CountResourceMonitor(uint32 limit) 00055 : mSize(0), 00056 mLimit((int32)limit) 00057 { 00058 assert(mLimit>=0); 00059 } 00060 00061 template <class T> 00062 bool preIncrement(T value, bool force) { 00063 if ((mSize+=1)>=mLimit&&mLimit&&!force) { 00064 mSize -= 1; 00065 return false; 00066 } 00067 return true; 00068 } 00069 00070 template <class T> 00071 bool preIncrement(T* value, bool force) { 00072 if ((mSize+=1)>=mLimit&&mLimit&&!force) { 00073 mSize -= 1; 00074 return false; 00075 } 00076 return true; 00077 } 00078 00079 template <class T> 00080 void postDecrement(T value) { 00081 int32 check=(mSize-=1); 00082 assert(check>=0); 00083 } 00084 00085 template <class T> 00086 void postDecrement(T* value) { 00087 int32 check=(mSize-=1); 00088 assert(check>=0); 00089 } 00090 00091 template <class T> 00092 bool probablyCanPush(T t) { 00093 return mSize.read()+1<=mLimit||!mLimit; 00094 } 00095 00096 template <class T> 00097 bool probablyCanPush(T* t) { 00098 return mSize.read()+1<=mLimit||!mLimit; 00099 } 00100 00101 bool probablyCanPush(size_t size) { 00102 return mSize.read()+1<=mLimit||!mLimit; 00103 } 00104 }; 00105 00106 } // namespace Sirikata 00107 00108 #endif //_SIRIKATA_COUNT_RESOURCE_MONITOR_HPP_