prj/cam/src/script/linkscpt.cpp

   1 /*
   2 @Copyright Looking Glass Studios, Inc.
   3 1996,1997,1998,1999,2000 Unpublished Work.
   4 */
   5
   6 // $Header: r:/t2repos/thief2/src/script/linkscpt.cpp,v 1.17 1999/02/16 16:05:53 Justin Exp $
   7
   8
   9 #include <comtools.h>
  10 #include <appagg.h>
  11
  12 #include <scrptapi.h>
  13 #include <scrptbas.h>
  14 #include <scrptsrv.h>
  15
  16 #include <linkscpt.h>
  17
  18 #include <bintrait.h>
  19
  20 #include <linkbase.h>
  21 #include <linkman.h>
  22 #include <relation.h>
  23
  24 #include <sdesbase.h>
  25 #include <sdestool.h>
  26 #include <sdesparm.h>
  27
  28 #include <objlpars.h>
  29
  30 #include <dbmem.h>
  31
  32 ////////////////////////////////////////////////////////////
  33 // LINK SCRIPT SERVICE
  34 //
  35
  36 static ILinkManager* LinkMan()
  37 {
  38    static ILinkManager* da_man = NULL;
  39    if (!da_man)
  40    {
  41       da_man = AppGetObj(ILinkManager);
  42    }
  43    return da_man;
  44 }
  45
  46 static IStructDescTools* SdescTools()
  47 {
  48    static IStructDescTools* tools = NULL;
  49    if (tools == NULL)
  50    {
  51       tools = AppGetObj(IStructDescTools);
  52    }
  53    return tools;
  54 }
  55
  56 //
  57 // Link Kind class
  58 //
  59
  60 linkkind::linkkind(const char* name)
  61 {
  62    IRelation* rel = LinkMan()->GetRelationNamed(name);
  63    id = rel->GetID();
  64
  65    AssertMsg1(id != RELID_NULL, "%s is not a link flavor", name);
  66    SafeRelease(rel);
  67 }
  68
  69 linkkind::operator string () const
  70 {
  71    cAutoIPtr<IRelation> rel ( LinkMan()->GetRelation(id) );
  72    return rel->Describe()->name;
  73 }
  74
  75 //
  76 // Link class
  77 //
  78
  79 object link::From() const
  80 {
  81    sLink link;
  82    LinkMan()->Get(id,&link);
  83    return link.source;
  84 }
  85
  86 object link::To() const
  87 {
  88    sLink link;
  89    LinkMan()->Get(id,&link);
  90    return link.dest;
  91 }
  92
  93 cMultiParm link::GetData(const char* field ) const
  94 {
  95    cAutoIPtr<IRelation> rel ( LinkMan()->GetRelation(LINKID_RELATION(id)) );
  96    void* data = rel->GetData(id);
  97    if (!data) return 0;
  98
  99    const sRelationDataDesc* ddesc = rel->DescribeData();
 100    const sStructDesc* sdesc = SdescTools()->Lookup(ddesc->type);
 101    if (!sdesc) return 0;
 102
 103    AssertMsg1(field != NULL || sdesc->nfields == 1,"%s links have more than one data field.", rel->Describe()->name);
 104
 105    const sFieldDesc* fdesc = NULL;
 106    if (field)
 107    {
 108       fdesc = NULL;
 109       for (int i = 0; i < sdesc->nfields; i++)
 110          if (stricmp(field,sdesc->fields[i].name) == 0)
 111          {
 112             fdesc = &sdesc->fields[i];
 113             break;
 114          }
 115       AssertMsg2(fdesc,"%s links have no data field named %s.",rel->Describe()->name,field);
 116    }
 117    else
 118       fdesc = &sdesc->fields[0];
 119
 120    return GetParmFromField(fdesc,data);
 121 }
 122
 123
 124 void link::SetData(const char* field, const cMultiParm& value)
 125 {
 126    cAutoIPtr<IRelation> rel ( LinkMan()->GetRelation(LINKID_RELATION(id)) );
 127    const sRelationDataDesc* ddesc = rel->DescribeData();
 128
 129    void* data = rel->GetData(id);
 130
 131    BOOL deldata = FALSE;
 132    if (!data)
 133    {
 134       data = new char[ddesc->size];
 135       deldata = TRUE;
 136    }
 137
 138    const sStructDesc* sdesc = SdescTools()->Lookup(ddesc->type);
 139    AssertMsg1(sdesc, "%s links have unknown data type.", rel->Describe()->name);
 140    if (!sdesc) return ;
 141
 142    AssertMsg1(field != NULL || sdesc->nfields == 1,"Relation %s has more than one data field.", rel->Describe()->name);
 143
 144    const sFieldDesc* fdesc = NULL;
 145    if (field)
 146    {
 147       fdesc = NULL;
 148       for (int i = 0; i < sdesc->nfields; i++)
 149          if (stricmp(field,sdesc->fields[i].name) == 0)
 150          {
 151             fdesc = &sdesc->fields[i];
 152             break;
 153          }
 154       AssertMsg2(fdesc,"%s links have no field named %s.",rel->Describe()->name,field);
 155    }
 156    else
 157       fdesc = &sdesc->fields[0];
 158
 159    SetFieldFromParm(fdesc,data,value);
 160    rel->SetData(id,data);
 161
 162    if (deldata)
 163       delete [] (char*)data;
 164 }
 165
 166 //
 167 // THE LINK SERVICE PROPER
 168 //
 169
 170 DECLARE_SCRIPT_SERVICE_IMPL(cLinkSrv, Link)
 171 {
 172    STDMETHOD_(void, Init)()
 173    {
 174    }
 175
 176    STDMETHOD_(link,Create)(linkkind kind, object from, object to)
 177    {
 178       return LinkMan()->Add(ScriptObjID(from),ScriptObjID(to),kind);
 179    }
 180
 181    STDMETHOD(Destroy)(link destroy_me)
 182    {
 183       return LinkMan()->Remove(destroy_me);
 184    }
 185
 186    STDMETHOD_(boolean,AnyExist)(linkkind kind, object from, object to)
 187    {
 188       cAutoIPtr<ILinkQuery> q ( LinkMan()->Query(ScriptObjID(from),ScriptObjID(to),kind) );
 189       return !q->Done();
 190    }
 191
 192    //
 193    // Get all the links that match a particular pattern
 194    //
 195    STDMETHOD_(linkset,GetAll)(linkkind kind, object from, object to)
 196    {
 197       cAutoIPtr<ILinkQuery> q ( LinkMan()->Query(ScriptObjID(from),ScriptObjID(to),kind) );
 198       return linkset(q);
 199    }
 200
 201    //
 202    // Get just one link.
 203    //
 204    STDMETHOD_(link,GetOne)(linkkind kind, object from, object to)
 205    {
 206       cAutoIPtr<ILinkQuery> q ( LinkMan()->Query(ScriptObjID(from),ScriptObjID(to),kind) );
 207
 208       return (q->Done()) ? 0 : q->ID();
 209    }
 210
 211    // @NOTE (justin 2/16/99): As of this writing, networking only supports
 212    // ObjIDs as the type for linkdata. This will be fixed some time when
 213    // I have time to properly network cMultiParms...
 214    STDMETHOD(BroadcastOnAllLinks)(const object ref SelfObj,
 215                                   const char *Message,
 216                                   linkkind recipients,
 217                                   const cMultiParm ref linkdata)
 218    {
 219       AutoAppIPtr(LinkManager);
 220       AutoAppIPtr(ScriptMan);
 221
 222       ObjID self = ScriptObjID(SelfObj);
 223
 224       if (pLinkManager->AnyLinks(recipients,self,LINKOBJ_WILDCARD))
 225       {
 226          ILinkQuery *Controls = pLinkManager->Query(self,LINKOBJ_WILDCARD,recipients);
 227
 228          while (!Controls->Done())
 229          {
 230             // so maybe the linkdata should be interepreted in a questbit-specific way someday?
 231             if (Controls->Data() == NULL || link(Controls->ID()).GetData() == linkdata)
 232             {
 233                sLink slink;
 234
 235                Controls->Link(&slink);
 236
 237                // Debug.MPrint(SelfObj," sending message ",Message," to ",thislink.To());
 238                //pScriptMan->PostMessage2(SelfObj,slink.dest,Message,linkdata,NULL,NULL);
 239                // We currently need to set up the message, to get the flag in:
 240                sScrMsg * pMsg = new sScrMsg(self,
 241                                             slink.dest,
 242                                             Message,
 243                                             linkdata);
 244                pMsg->flags |= kSMF_MsgPostToOwner;
 245                pScriptMan->PostMessage(pMsg);
 246                pMsg->Release();
 247             }
 248             Controls->Next();
 249          }
 250          SafeRelease(Controls);
 251       }
 252
 253       return(S_OK);
 254    }
 255
 256    STDMETHOD(BroadcastOnAllLinks)(const object ref SelfObj,
 257                                   const char *Message,
 258                                   linkkind recipients)
 259    {
 260       AutoAppIPtr(LinkManager);
 261       AutoAppIPtr(ScriptMan);
 262
 263       ObjID self = ScriptObjID(SelfObj);
 264
 265       if (pLinkManager->AnyLinks(recipients,self,LINKOBJ_WILDCARD))
 266       {
 267          ILinkQuery *Controls = pLinkManager->Query(self,LINKOBJ_WILDCARD,recipients);
 268
 269          while (!Controls->Done())
 270          {
 271             sLink slink;
 272             Controls->Link(&slink);
 273
 274             //pScriptMan->PostMessage2(SelfObj,slink.dest,Message,NULL,NULL,NULL);
 275             // We currently need to set up the message, to get the flag in:
 276             sScrMsg * pMsg = new sScrMsg(self,
 277                                          slink.dest,
 278                                          Message,
 279                                          NULL);
 280             pMsg->flags |= kSMF_MsgPostToOwner;
 281             pScriptMan->PostMessage(pMsg);
 282             pMsg->Release();
 283             Controls->Next();
 284          }
 285          SafeRelease(Controls);
 286       }
 287
 288       return(S_OK);
 289    }
 290
 291    //
 292    // Create multiple links.
 293    //
 294    STDMETHOD(CreateMany)(linkkind kind, const string ref FromSet, const string ref ToSet)
 295    {
 296       cObjListParser   parser;
 297       cDynArray<ObjID> sources;
 298       cDynArray<ObjID> destinations;
 299       IRelation *      pRelation = LinkMan()->GetRelation(kind);
 300
 301       if (!pRelation)
 302          return E_FAIL;
 303
 304       parser.Parse(FromSet, &sources);
 305       parser.Parse(ToSet, &destinations);
 306
 307       for (int i = 0; i < sources.Size(); i++)
 308          for (int j = 0; j < destinations.Size(); j++)
 309             pRelation->Add(sources[i], destinations[j]);
 310
 311       pRelation->Release();
 312       return S_OK;
 313    }
 314
 315    //
 316    // Destroy multiple link
 317    //
 318    STDMETHOD(DestroyMany)(linkkind kind, const string ref FromSet, const string ref ToSet)
 319    {
 320       cObjListParser   parser;
 321       cDynArray<ObjID> sources;
 322       cDynArray<ObjID> destinations;
 323       IRelation *      pRelation = LinkMan()->GetRelation(kind);
 324       ILinkQuery *     pQuery;
 325
 326       if (!pRelation)
 327          return E_FAIL;
 328
 329       parser.Parse(FromSet, &sources);
 330       parser.Parse(ToSet, &destinations);
 331
 332       for (int i = 0; i < sources.Size(); i++)
 333          for (int j = 0; j < destinations.Size(); j++)
 334          {
 335             pQuery = pRelation->Query(sources[i], destinations[j]);
 336
 337             for (; !pQuery->Done(); pQuery->Next())
 338                pRelation->Remove(pQuery->ID());
 339
 340             pQuery->Release();
 341          }
 342
 343       pRelation->Release();
 344       return S_OK;
 345    }
 346
 347    STDMETHOD_(linkset,GetAllInherited)(linkkind kind, object from, object to)
 348    {
 349       cAutoIPtr<IRelation> rel = LinkMan()->GetRelation(kind);
 350       cAutoIPtr<ILinkQuery> q  = QueryInheritedLinksUncached(rel, from, to);
 351       return linkset(q);
 352    }
 353
 354    STDMETHOD_(linkset,GetAllInheritedSingle)(linkkind kind, object from, object to)
 355    {
 356       cAutoIPtr<IRelation> rel = LinkMan()->GetRelation(kind);
 357       cAutoIPtr<ILinkQuery> q  = QueryInheritedLinksSingleUncached(rel, from, to);
 358       return linkset(q);
 359    }
 360
 361 };
 362
 363 IMPLEMENT_SCRIPT_SERVICE_IMPL(cLinkSrv, Link);
 364
 365
 366 DECLARE_SCRIPT_SERVICE_IMPL(cLinkToolsSrv, LinkTools)
 367 {
 368    STDMETHOD_(long, LinkKindNamed)(const char* name)
 369    {
 370       return linkkind(name);
 371    }
 372
 373    STDMETHOD_(string, LinkKindName)(long id)
 374    {
 375       cAutoIPtr<IRelation> rel (LinkMan()->GetRelation(id) );
 376       AssertMsg1(rel != 0,"A script referenced non-existent link kind %d",id);
 377       return rel->Describe()->name;
 378    }
 379
 380    STDMETHOD(LinkGet)(long id, sLink& link)
 381    {
 382       VerifyMsg1(LinkMan()->Get(id,&link),"A script tried to access non-existent link %d",id);
 383       return S_OK;
 384    }
 385
 386    STDMETHOD_(cMultiParm,LinkGetData)(long id, const char* field)
 387    {
 388       return link(id).GetData(field);
 389    }
 390
 391    STDMETHOD(LinkSetData)(long id, const char* field, const cMultiParm ref val)
 392    {
 393       link(id).SetData(field,val);
 394       return S_OK;
 395    }
 396 };
 397
 398 IMPLEMENT_SCRIPT_SERVICE_IMPL(cLinkToolsSrv, LinkTools);
 399
 400
 401