prj/cam/src/object/traisrch.cpp

   1 /*
   2 @Copyright Looking Glass Studios, Inc.
   3 1996,1997,1998,1999,2000 Unpublished Work.
   4 */
   5
   6 // $Header: r:/t2repos/thief2/src/object/traisrch.cpp,v 1.9 1998/10/05 17:28:33 mahk Exp $
   7 #include <traisrch.h>
   8 #include <dlisttem.h>
   9 #include <osysbase.h>
  10 #include <traitbas.h>
  11
  12 // Must be last header
  13 #include <dbmem.h>
  14
  15
  16 #ifdef __MSVC
  17 template LinkQueryList;
  18 template LinkQueryList::cParent;
  19 //template cSimpleStack<QueryPtr>;
  20 template cDList<class cSimpleDListNode<class QueryPtr>,1>;
  21 #endif
  22
  23 ////////////////////////////////////////////////////////////
  24 //
  25 // cDepthFirstDonorQuery
  26 //
  27 ////////////////////////////////////////////////////////////
  28
  29 STDMETHODIMP_(BOOL) cDepthFirstObjectQuery::Done()
  30 {
  31    return CurObj == OBJ_NULL;
  32 }
  33
  34 STDMETHODIMP_(ObjID) cDepthFirstObjectQuery::Object()
  35 {
  36    return CurObj;
  37 }
  38
  39 STDMETHODIMP cDepthFirstObjectQuery::Next()
  40 {
  41    if (!Expand())
  42    {
  43       if (!Contract())
  44       {
  45          CurObj = OBJ_NULL;
  46          return S_OK;
  47       }
  48    }
  49    CurObj = Queries.Top()->Object();
  50    return S_OK;
  51 }
  52
  53
  54 BOOL cDepthFirstObjectQuery::Expand()
  55 {
  56    if (CurObj == OBJ_NULL)
  57       return FALSE;
  58
  59    IObjectQuery* query = Successors(CurObj);
  60    Queries.Push(query);
  61    return !query->Done();
  62 }
  63
  64 BOOL cDepthFirstObjectQuery::Contract()
  65 {
  66    if (Queries.Top() == NULL)
  67       return FALSE;
  68
  69    while(Queries.Top()->Done())
  70    {
  71       Queries.Pop()->Release();
  72       if (Queries.Top() == NULL)
  73          return FALSE;
  74       Queries.Top()->Next();
  75    }
  76    return TRUE;
  77 }
  78
  79 ////////////////////////////////////////
  80
  81 BOOL cConcretenessFilterQuery::Filter(ObjID obj)
  82 {
  83    switch(Which)
  84    {
  85       case kObjectConcrete:
  86          return OBJ_IS_CONCRETE(obj);
  87
  88       case kObjectAbstract:
  89          return OBJ_IS_ABSTRACT(obj);
  90
  91       case kObjectAll:
  92          return TRUE;
  93
  94       default:
  95          Warning(("Unknown concreteness for query: %d\n",Which));
  96    }
  97    return FALSE;
  98 }
  99
 100