Initial commit

Torque/SDK/engine/ts/tsIntegerSet.h

@@ -0,0 +1,96 @@
+//-----------------------------------------------------------------------------
+// Torque Game Engine
+// Copyright (C) GarageGames.com, Inc.
+//-----------------------------------------------------------------------------
+
+#ifndef _TSINTEGERSET_H_
+#define _TSINTEGERSET_H_
+
+#ifndef _PLATFORM_H_
+#include "platform/platform.h"
+#endif
+#ifndef _TVECTOR_H_
+#include "core/tVector.h"
+#endif
+
+
+#if defined(TORQUE_LIB)
+#define MAX_TS_SET_DWORDS 32
+#else
+#define MAX_TS_SET_DWORDS 6
+#endif
+
+#define MAX_TS_SET_SIZE   (32*MAX_TS_SET_DWORDS)
+
+class Stream;
+
+/// The standard mathmatical set, where there are no duplicates.  However,
+/// this set uses bits instead of numbers.
+class TSIntegerSet
+{
+   /// The bits!
+   U32 bits[MAX_TS_SET_DWORDS];
+
+public:
+
+   /// Sets this bit to false
+   void clear(S32 index);
+   /// Set this bit to true
+   void set(S32 index);
+   /// Is this bit true?
+   bool test(S32 index) const;
+
+   /// Sets all bits to false
+   void clearAll(S32 upto = MAX_TS_SET_SIZE);
+   /// Sets all bits to true
+   void setAll(S32 upto = MAX_TS_SET_SIZE);
+   /// Tests all bits for true
+   bool testAll(S32 upto = MAX_TS_SET_SIZE) const;
+
+   /// intersection (a & b)
+   void intersect(const TSIntegerSet&);
+   /// union (a | b)
+   void overlap(const TSIntegerSet&);
+   /// xor (a | b) & ( !(a & b) )
+   void difference(const TSIntegerSet&);
+   /// subtraction (a - b)
+   void takeAway(const TSIntegerSet&);
+
+   /// copy one integer set into another
+   void copy(const TSIntegerSet&);
+
+   void operator=(const TSIntegerSet& otherSet) { copy(otherSet); }
+
+   S32 start() const;
+   S32 end() const;
+   void next(S32 & i) const;
+
+   void read(Stream *);
+   void write(Stream *);
+
+   TSIntegerSet();
+   TSIntegerSet(const TSIntegerSet&);
+};
+
+inline void TSIntegerSet::clear(S32 index)
+{
+   AssertFatal(index>=0 && index<MAX_TS_SET_SIZE,"TS::IntegerSet::clear");
+
+   bits[index>>5] &= ~(1 << (index & 31));
+}
+
+inline void TSIntegerSet::set(S32 index)
+{
+   AssertFatal(index>=0 && index<MAX_TS_SET_SIZE,"TS::IntegerSet::set");
+
+   bits[index>>5] |= 1 << (index & 31);
+}
+
+inline bool TSIntegerSet::test(S32 index) const
+{
+   AssertFatal(index>=0 && index<MAX_TS_SET_SIZE,"TS::IntegerSet::test");
+
+   return ((bits[index>>5] & (1 << (index & 31)))!=0);
+}
+
+#endif