prj/cam/src/object/linkset.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/linkset.cpp,v 1.4 1998/01/27 18:49:35 mahk Exp $
   7 #include <lnkquer_.h>
   8 #include <linkset.h>
   9 #include <lststtem.h>
  10 #include <linkid.h>
  11
  12 // Must be last header
  13 #include <dbmem.h>
  14
  15 #ifdef __MSVC
  16 template cDList<cSimpleDListNode<LinkID>,1>;
  17 template cSimpleDList<LinkID>;
  18 template cSimpleListSet<LinkID>;
  19 #endif
  20
  21
  22 ////////////////////////////////////////////////////////////
  23 //
  24 // INTERNAL CLASS: cSimpleLinkSetQuery
  25 //
  26 ////////////////////////////////////////////////////////////
  27
  28 class cSimpleLinkSetQuery : public cBaseLinkQuery
  29 {
  30    cLinkSet::cIter Iter;
  31
  32    BOOL Eligible(LinkID id);
  33 public:
  34
  35    cSimpleLinkSetQuery(const cLinkSet* set,
  36                  tQueryDate bday = QUERY_DATE_NONE)
  37       : Iter(set->Iter()),
  38         cBaseLinkQuery(bday)
  39    { while (!Iter.Done() && !Eligible(Iter.Value())) Iter.Next();} ;
  40
  41    virtual ~cSimpleLinkSetQuery() {};
  42
  43    STDMETHOD_(BOOL,Done)() const;
  44    STDMETHOD(Link)(sLink* link) const;
  45    STDMETHOD(Next)();
  46    STDMETHOD_(LinkID,ID)() const { return (LinkID)Iter.Value();};
  47 };
  48
  49
  50 STDMETHODIMP_(BOOL) cSimpleLinkSetQuery::Done() const
  51 {
  52    return Iter.Done();
  53 }
  54
  55 STDMETHODIMP cSimpleLinkSetQuery::Link(sLink* link) const
  56 {
  57    if (Iter.Done()) return E_FAIL;
  58    LinkID val = Iter.Value();
  59    LinkMan()->Get(val,link);
  60    return S_OK;
  61 }
  62
  63 BOOL cSimpleLinkSetQuery::Eligible(LinkID link)
  64 {
  65    return VerifyLink(link);
  66 }
  67
  68 STDMETHODIMP cSimpleLinkSetQuery::Next()
  69 {
  70    for (Iter.Next(); !Iter.Done(); Iter.Next())
  71    {
  72       if (Eligible(Iter.Value()))
  73          return S_OK;
  74    }
  75    return E_FAIL;
  76 }
  77
  78 ////////////////////////////////////////////////////////////
  79 //
  80 // INTERNAL CLASS: cLinkSetQuery
  81 //
  82 ////////////////////////////////////////////////////////////
  83
  84 class cLinkSetQuery : public cBaseLinkQuery
  85 {
  86    sLinkTemplate Pattern;
  87    cLinkSet::cIter Iter;
  88
  89    BOOL Eligible(LinkID id);
  90 public:
  91
  92    cLinkSetQuery(const cLinkSet& set,
  93                  const sLinkTemplate& pattern,
  94                  tQueryDate bday = QUERY_DATE_NONE)
  95       : Iter(set.Iter()),
  96         Pattern(pattern),
  97         cBaseLinkQuery(bday)
  98    { while (!Iter.Done() && !Eligible(Iter.Value())) Iter.Next();} ;
  99
 100    virtual ~cLinkSetQuery() {};
 101
 102    STDMETHOD_(BOOL,Done)() const;
 103    STDMETHOD(Link)(sLink* link) const;
 104    STDMETHOD(Next)();
 105    STDMETHOD_(LinkID,ID)() const { return (LinkID)Iter.Value();};
 106 };
 107
 108
 109 STDMETHODIMP_(BOOL) cLinkSetQuery::Done() const
 110 {
 111    return Iter.Done();
 112 }
 113
 114 STDMETHODIMP cLinkSetQuery::Link(sLink* link) const
 115 {
 116    if (Iter.Done()) return E_FAIL;
 117    LinkID val = Iter.Value();
 118    LinkMan()->Get(val,link);
 119    return S_OK;
 120 }
 121
 122 BOOL cLinkSetQuery::Eligible(LinkID link)
 123 {
 124    if (!VerifyLink(link)) return FALSE;
 125
 126       // check relation
 127    RelationID relid = LINKID_RELATION(link);
 128
 129    // don't yield inverse links on wildcard.
 130    // cuz that's the spec
 131
 132    if (!RELID_MATCH(Pattern.flavor,relid))
 133       return FALSE;
 134
 135       // we always assume the source is right.
 136    if (Pattern.dest != LINKOBJ_WILDCARD)  // must test dest
 137    {
 138       sLinkTemplate temp;
 139       LinkMan()->Get(link,(sLink*)&temp);
 140       if (temp != Pattern) return FALSE;
 141    }
 142    return TRUE;
 143 }
 144
 145 STDMETHODIMP cLinkSetQuery::Next()
 146 {
 147    for (Iter.Next(); !Iter.Done(); Iter.Next())
 148    {
 149       if (Eligible(Iter.Value()))
 150          return S_OK;
 151    }
 152    return E_FAIL;
 153 }
 154
 155
 156
 157 ////////////////////////////////////////////////////////////
 158 //
 159 // CLASS: cLinkSet
 160 //
 161 //
 162
 163 ILinkQuery* cLinkSet::Query(tQueryDate bday)
 164 {
 165    return new cSimpleLinkSetQuery(this,bday);
 166 }
 167
 168 ILinkQuery* cLinkSet::PatternQuery(const struct sLinkTemplate* pattern, tQueryDate bday)
 169 {
 170    return new cLinkSetQuery(*this,*pattern,bday);
 171 }
 172
 173
 174
 175
 176