Back to index

lightning-sunbird  0.9+nobinonly
nsCheapSets.h
Go to the documentation of this file.
00001 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
00002 /* ***** BEGIN LICENSE BLOCK *****
00003  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
00004  *
00005  * The contents of this file are subject to the Mozilla Public License Version
00006  * 1.1 (the "License"); you may not use this file except in compliance with
00007  * the License. You may obtain a copy of the License at
00008  * http://www.mozilla.org/MPL/
00009  *
00010  * Software distributed under the License is distributed on an "AS IS" basis,
00011  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
00012  * for the specific language governing rights and limitations under the
00013  * License.
00014  *
00015  * The Original Code is Mozilla Communicator client code.
00016  *
00017  * The Initial Developer of the Original Code is
00018  * Netscape Communications Corporation.
00019  * Portions created by the Initial Developer are Copyright (C) 1998
00020  * the Initial Developer. All Rights Reserved.
00021  *
00022  * Contributor(s):
00023  *
00024  * Alternatively, the contents of this file may be used under the terms of
00025  * either of the GNU General Public License Version 2 or later (the "GPL"),
00026  * or the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
00027  * in which case the provisions of the GPL or the LGPL are applicable instead
00028  * of those above. If you wish to allow use of your version of this file only
00029  * under the terms of either the GPL or the LGPL, and not to allow others to
00030  * use your version of this file under the terms of the MPL, indicate your
00031  * decision by deleting the provisions above and replace them with the notice
00032  * and other provisions required by the GPL or the LGPL. If you do not delete
00033  * the provisions above, a recipient may use your version of this file under
00034  * the terms of any one of the MPL, the GPL or the LGPL.
00035  *
00036  * ***** END LICENSE BLOCK ***** */
00037 
00038 #ifndef __nsCheapSets_h__
00039 #define __nsCheapSets_h__
00040 
00041 #include "nsHashSets.h"
00042 
00047 class NS_COM nsCheapStringSet {
00048 public:
00049   nsCheapStringSet() : mValOrHash(nsnull)
00050   {
00051   }
00052   ~nsCheapStringSet();
00053 
00058   nsresult Put(const nsAString& aVal);
00059 
00064   void Remove(const nsAString& aVal);
00065 
00071   PRBool Contains(const nsAString& aVal)
00072   {
00073     nsStringHashSet* set = GetHash();
00074     // Check the value from the hash if the hash is there
00075     if (set) {
00076       return set->Contains(aVal);
00077     }
00078 
00079     // Check whether the value is equal to the string if the string is there
00080     nsAString* str = GetStr();
00081     return str && str->Equals(aVal);
00082   }
00083 
00084 private:
00085   typedef unsigned long PtrBits;
00086 
00088   nsStringHashSet* GetHash()
00089   {
00090     return (PtrBits(mValOrHash) & 0x1) ? nsnull : (nsStringHashSet*)mValOrHash;
00091   }
00093   nsAString* GetStr()
00094   {
00095     return (PtrBits(mValOrHash) & 0x1)
00096            ? (nsAString*)(PtrBits(mValOrHash) & ~0x1)
00097            : nsnull;
00098   }
00100   nsresult SetStr(const nsAString& aVal)
00101   {
00102     nsString* str = new nsString(aVal);
00103     if (!str) {
00104       return NS_ERROR_OUT_OF_MEMORY;
00105     }
00106     mValOrHash = (nsAString*)(PtrBits(str) | 0x1);
00107     return NS_OK;
00108   }
00110   nsresult InitHash(nsStringHashSet** aSet);
00111 
00112 private:
00114   void* mValOrHash;
00115 };
00116 
00117 
00122 class NS_COM nsCheapInt32Set {
00123 public:
00124   nsCheapInt32Set() : mValOrHash(nsnull)
00125   {
00126   }
00127   ~nsCheapInt32Set();
00128 
00132   nsresult Put(PRInt32 aVal);
00133  
00138   void Remove(PRInt32 aVal);
00139 
00145   PRBool Contains(PRInt32 aVal)
00146   {
00147     nsInt32HashSet* set = GetHash();
00148     if (set) {
00149       return set->Contains(aVal);
00150     }
00151     if (IsInt()) {
00152       return GetInt() == aVal;
00153     }
00154     return PR_FALSE;
00155   }
00156 
00157 private:
00158   typedef unsigned long PtrBits;
00159 
00161   nsInt32HashSet* GetHash()
00162   {
00163     return PtrBits(mValOrHash) & 0x1 ? nsnull : (nsInt32HashSet*)mValOrHash;
00164   }
00166   PRBool IsInt()
00167   {
00168     return !!(PtrBits(mValOrHash) & 0x1);
00169   }
00171   PRInt32 GetInt()
00172   {
00173     return PtrBits(mValOrHash) >> 1;
00174   }
00176   void SetInt(PRInt32 aInt)
00177   {
00178     mValOrHash = (void*)((aInt << 1) | 0x1);
00179   }
00181   nsresult InitHash(nsInt32HashSet** aSet);
00182 
00183 private:
00185   void* mValOrHash;
00186 };
00187 
00188 
00194 #endif