prj/cam/src/editor/linkdraw.cpp

   1 /*
   2 @Copyright Looking Glass Studios, Inc.
   3 1996,1997,1998,1999,2000 Unpublished Work.
   4 */
   5
   6 ////////////////////////////////////////////////////////////////////////////////
   7 // $Header: r:/t2repos/thief2/src/editor/linkdraw.cpp,v 1.7 2000/02/11 14:27:44 bfarquha Exp $
   8 //
   9 // Link-drawing system
  10 //
  11
  12 #include <lg.h>
  13 #include <string.h>
  14
  15 #include <appagg.h>
  16 #include <comtools.h>
  17
  18 #include <dlist.h>
  19 #include <relation.h>
  20 #include <lnkquery.h>
  21 #include <linkbase.h>
  22 #include <linkman.h>
  23
  24 #include <wrtype.h>
  25 #include <objpos.h>
  26
  27 #include <ged_line.h>
  28 #include <linkdraw.h>
  29
  30 #include <editbr_.h>
  31 #include <editbr.h>
  32 #include <brlist.h>
  33 #include <brinfo.h>
  34
  35 // Must be last header
  36 #include <dbmem.h>
  37
  38
  39 class cLinkDraw;
  40
  41 typedef cDList<cLinkDraw, 0>     cLinkDrawList;
  42 typedef cDListNode<cLinkDraw, 0> cLinkDrawNode;
  43
  44 struct sColor
  45 {
  46    int r, g, b;
  47    sColor(int rr = 0, int gg = 0, int bb = 0) : r(rr),g(gg),b(bb) {};
  48 };
  49
  50 class cLinkDraw : public cLinkDrawNode
  51 {
  52 public:
  53    char name[64];
  54    sColor color;
  55 };
  56
  57
  58 static cLinkDrawList LinkDrawList;
  59
  60 ////////////////////////////////////////////////////////////////////////////////
  61
  62 void InitDrawnRelations()
  63 {
  64    // Hey, we might want to do something here someday...
  65 }
  66
  67 ////////////////////////////////////////
  68
  69 void TermDrawnRelations()
  70 {
  71    cLinkDraw *pLinkDraw;
  72
  73    pLinkDraw = LinkDrawList.GetFirst();
  74    while (pLinkDraw != NULL)
  75    {
  76       delete LinkDrawList.Remove(pLinkDraw);
  77       pLinkDraw = LinkDrawList.GetFirst();
  78    }
  79 }
  80
  81 ////////////////////////////////////////
  82
  83 void ResetDrawnRelations()
  84 {
  85    TermDrawnRelations();
  86    InitDrawnRelations();
  87 }
  88
  89 ////////////////////////////////////////////////////////////////////////////////
  90
  91 static cLinkDraw *GetLinkDrawNode(const char *name)
  92 {
  93    cLinkDraw *pLinkDraw;
  94
  95    pLinkDraw = LinkDrawList.GetFirst();
  96    while (pLinkDraw != NULL)
  97    {
  98       if (!strcmp(name, pLinkDraw->name))
  99          return pLinkDraw;
 100
 101       pLinkDraw = pLinkDraw->GetNext();
 102    }
 103
 104    return NULL;
 105 }
 106
 107 ////////////////////////////////////////
 108
 109 void AddDrawnRelation(const char *name, ubyte r, ubyte g, ubyte b)
 110 {
 111    cLinkDraw *pLinkDraw;
 112
 113    if (GetLinkDrawNode(name) == NULL)
 114    {
 115       pLinkDraw = new cLinkDraw;
 116
 117       strncpy(pLinkDraw->name, name, 64);
 118       pLinkDraw->color = sColor(r,g,b);
 119
 120       LinkDrawList.Append(pLinkDraw);
 121    }
 122 }
 123
 124 ////////////////////////////////////////
 125
 126 void AddDrawnRelationID(RelationID relationID, ubyte r, ubyte g, ubyte b)
 127 {
 128    AutoAppIPtr_(LinkManager, pLinkMan);
 129
 130    IRelation *rel = pLinkMan->GetRelation(relationID);
 131
 132    if (rel != NULL)
 133       AddDrawnRelation(rel->Describe()->name, r, g, b);
 134    else
 135       Warning(("AddDrawnRelationID: Relation id %d not found!\n", relationID));
 136
 137    SafeRelease(rel);
 138 }
 139
 140 ////////////////////////////////////////
 141
 142 void RemoveDrawnRelation(const char *name)
 143 {
 144    cLinkDraw *pLinkDraw;
 145
 146    if ((pLinkDraw = GetLinkDrawNode(name)) != NULL)
 147       delete LinkDrawList.Remove(pLinkDraw);
 148 }
 149
 150 ////////////////////////////////////////
 151
 152 void RemoveDrawnRelationID(RelationID relationID)
 153 {
 154    AutoAppIPtr_(LinkManager, pLinkMan);
 155
 156    IRelation *rel = pLinkMan->GetRelation(relationID);
 157
 158    if (rel != NULL)
 159       RemoveDrawnRelation(rel->Describe()->name);
 160    else
 161       Warning(("RemoveDrawnRelationID: Relation id %d not found!\n", relationID));
 162
 163    SafeRelease(rel);
 164 }
 165
 166 ////////////////////////////////////////////////////////////////////////////////
 167
 168 // from brrend.c:
 169 extern "C"
 170 {
 171    bool pointOBBIntersect(editBrush *point, editBrush *brush);
 172 }
 173
 174 // Determine if link originates inside brush.
 175 static BOOL LinkOriginatesInBrush(sLink *pLink, editBrush *us)
 176 {
 177    editBrush point;
 178    point.pos = ObjPosGet(pLink->source)->loc.vec;
 179    return pointOBBIntersect(&point, us);
 180 }
 181
 182
 183 int g_nLinkDrawSourceId = 0;
 184 int g_nLinkDrawSplitId = 0;
 185
 186
 187 // Is this id a child of g_nLinkDrawSplitId?
 188 // *** We're assuming there are no circular patrols that don't include startid
 189 // To do this right, we'd need to mark previously visited nodes, but that's probably overkill
 190 // given our problem space. Still, let's at least not crash:
 191
 192 static int nCrashDetector;
 193
 194 static BOOL IsChild(ObjID startid, ObjID id, IRelation *pRelation, int nLoopCount = 0)
 195 {
 196    if (startid==id) // We've gone circular, or are just beginning.
 197    {
 198       if (nLoopCount)
 199       {
 200          nCrashDetector--;
 201          return FALSE;
 202       }
 203       else
 204          nLoopCount++;
 205    }
 206
 207    if (nCrashDetector > 40)
 208       return FALSE;
 209
 210    nCrashDetector++;
 211
 212    ILinkQuery *query = pRelation->Query(LINKOBJ_WILDCARD, id);
 213
 214    for (; !query->Done(); query->Next())
 215    {
 216       sLink link;
 217       query->Link(&link);
 218       if (link.source == g_nLinkDrawSplitId)
 219       {
 220          SafeRelease(query);
 221          return TRUE;
 222       }
 223       else if (IsChild(startid, link.source, pRelation, nLoopCount))
 224       {
 225          SafeRelease(query);
 226          return TRUE;
 227       }
 228    }
 229
 230    SafeRelease(query);
 231    nCrashDetector--;
 232    return FALSE;
 233 }
 234
 235
 236 // Is this id a parent of g_nLinkDrawSplitId?
 237 // *** We're assuming there are no circular patrols that don't include startid
 238 // To do this right, we'd need to mark previously visited nodes, but that's probably overkill
 239 // given our problem space.
 240 static BOOL IsParent(ObjID startid, ObjID id, IRelation *pRelation, int nLoopCount = 0)
 241 {
 242    if (startid==id) // We've gone circular, or are just beginning.
 243    {
 244       if (nLoopCount)
 245       {
 246          nCrashDetector--;
 247          return FALSE;
 248       }
 249       else
 250          nLoopCount++;
 251    }
 252
 253    if (nCrashDetector > 40)
 254       return FALSE;
 255
 256    nCrashDetector++;
 257
 258    ILinkQuery *query = pRelation->Query(id, LINKOBJ_WILDCARD);
 259
 260    for (; !query->Done(); query->Next())
 261    {
 262       sLink link;
 263       query->Link(&link);
 264       if (link.dest == g_nLinkDrawSplitId)
 265       {
 266          SafeRelease(query);
 267          return TRUE;
 268       }
 269       else if (IsParent(startid, link.dest, pRelation, nLoopCount))
 270       {
 271          SafeRelease(query);
 272          return TRUE;
 273       }
 274    }
 275
 276    SafeRelease(query);
 277    nCrashDetector--;
 278    return FALSE;
 279 }
 280
 281
 282 // This is SLOW, but then again, it doesn't need to be fast.
 283 // See if this Obj is anywhere on branching relation.
 284 static BOOL ObjIsOnSplitPath(ObjID id, IRelation *pRelation)
 285 {
 286    if (id==g_nLinkDrawSplitId)
 287       return TRUE;
 288
 289    nCrashDetector = 0;
 290    if (IsParent(id, id, pRelation))
 291       return TRUE;
 292
 293    nCrashDetector = 0;
 294    if (IsChild(id, id, pRelation))
 295       return TRUE;
 296
 297    return FALSE;
 298 }
 299
 300
 301 // Go through and see if any brushes are me-only. If so, only draw links that originate within this brush.
 302 static BOOL LinkIsDrawable(sLink *pLink, IRelation *pRelation)
 303 {
 304    editBrush *us;
 305    int hIter;
 306    BOOL bFoundBrush = FALSE;
 307
 308    if ((g_nLinkDrawSourceId > 0) && (pLink->source != g_nLinkDrawSourceId))
 309       return FALSE;
 310
 311    if ((g_nLinkDrawSplitId > 0) && !ObjIsOnSplitPath(pLink->source, pRelation))
 312       return FALSE;
 313
 314    us=blistIterStart(&hIter);
 315    while ((us!=NULL) && !bFoundBrush)
 316    {
 317       if ((brushGetType(us)==brType_HOTREGION) && brHot_IsMEONLY(us))
 318       {
 319          bFoundBrush = TRUE;
 320          blistIterDone(hIter);
 321       }
 322       else
 323          us=blistIterNext(hIter);
 324    }
 325
 326    return !bFoundBrush || LinkOriginatesInBrush(pLink, us);
 327 }
 328
 329
 330
 331 void DrawRelations()
 332 {
 333    AutoAppIPtr_(LinkManager, pLinkMan);
 334    cLinkDraw *pLinkDraw;
 335
 336    gedDeleteChannelLines(1<<LINE_CH_LINKS);
 337
 338    ged_line_load_channel = LINE_CH_LINKS;
 339    ged_line_view_channels |= (1<<LINE_CH_LINKS);
 340
 341    pLinkDraw = LinkDrawList.GetFirst();
 342    while (pLinkDraw != NULL)
 343    {
 344       IRelation *pRelation = pLinkMan->GetRelationNamed(pLinkDraw->name);
 345
 346       if (pRelation != NULL)
 347       {
 348          ILinkQuery *query = pRelation->Query(LINKOBJ_WILDCARD, LINKOBJ_WILDCARD);
 349
 350          for (; !query->Done(); query->Next())
 351          {
 352             sLink link;
 353             query->Link(&link);
 354
 355             if (LinkIsDrawable(&link, pRelation))
 356                gedLineAddRGB(&ObjPosGet(link.source)->loc.vec, &ObjPosGet(link.dest)->loc.vec, LINE_FLG_DIR, pLinkDraw->color.r, pLinkDraw->color.g, pLinkDraw->color.b);
 357          }
 358
 359          SafeRelease(query);
 360       }
 361       else
 362          Warning(("DrawRelations: relation %s not found!\n", pLinkDraw->name));
 363
 364       SafeRelease(pRelation);
 365
 366       pLinkDraw = pLinkDraw->GetNext();
 367    }
 368 }
 369
 370 ////////////////////////////////////////////////////////////////////////////////
 371
 372 BOOL ObjHasDrawnLinks(ObjID objID)
 373 {
 374    AutoAppIPtr_(LinkManager, pLinkMan);
 375    cLinkDraw  *pLinkDraw;
 376    ILinkQuery *query;
 377
 378    pLinkDraw = LinkDrawList.GetFirst();
 379    while (pLinkDraw != NULL)
 380    {
 381       IRelation *pRelation = pLinkMan->GetRelationNamed(pLinkDraw->name);
 382
 383       if (pRelation != NULL)
 384       {
 385          query = pRelation->Query(objID, LINKOBJ_WILDCARD);
 386
 387          if (!query->Done())
 388          {
 389             SafeRelease(query);
 390             SafeRelease(pRelation);
 391
 392             return TRUE;
 393          }
 394
 395          SafeRelease(query);
 396
 397          query = pRelation->Query(LINKOBJ_WILDCARD, objID);
 398
 399          if (!query->Done())
 400          {
 401             SafeRelease(query);
 402             SafeRelease(pRelation);
 403
 404             return TRUE;
 405          }
 406
 407          SafeRelease(query);
 408       }
 409       else
 410          Warning(("DrawRelations: relation %s not found!\n", pLinkDraw->name));
 411
 412       SafeRelease(pRelation);
 413
 414       pLinkDraw = pLinkDraw->GetNext();
 415    }
 416
 417    return FALSE;
 418 }
 419
 420
 421