1 //===-------------------- Graph.h - PBQP Graph ------------------*- C++ -*-===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 // 10 // PBQP Graph class. 11 // 12 //===----------------------------------------------------------------------===// 13 14 15 #ifndef LLVM_CODEGEN_PBQP_GRAPH_H 16 #define LLVM_CODEGEN_PBQP_GRAPH_H 17 18 #include "llvm/ADT/ilist.h" 19 #include "llvm/ADT/ilist_node.h" 20 #include "llvm/Support/Debug.h" 21 #include <list> 22 #include <map> 23 #include <set> 24 25 namespace llvm { 26 namespace PBQP { 27 28 class GraphBase { 29 public: 30 typedef unsigned NodeId; 31 typedef unsigned EdgeId; 32 33 /// @brief Returns a value representing an invalid (non-existent) node. 34 static NodeId invalidNodeId() { 35 return std::numeric_limits<NodeId>::max(); 36 } 37 38 /// @brief Returns a value representing an invalid (non-existent) edge. 39 static EdgeId invalidEdgeId() { 40 return std::numeric_limits<EdgeId>::max(); 41 } 42 }; 43 44 /// PBQP Graph class. 45 /// Instances of this class describe PBQP problems. 46 /// 47 template <typename SolverT> 48 class Graph : public GraphBase { 49 private: 50 typedef typename SolverT::CostAllocator CostAllocator; 51 public: 52 typedef typename SolverT::RawVector RawVector; 53 typedef typename SolverT::RawMatrix RawMatrix; 54 typedef typename SolverT::Vector Vector; 55 typedef typename SolverT::Matrix Matrix; 56 typedef typename CostAllocator::VectorPtr VectorPtr; 57 typedef typename CostAllocator::MatrixPtr MatrixPtr; 58 typedef typename SolverT::NodeMetadata NodeMetadata; 59 typedef typename SolverT::EdgeMetadata EdgeMetadata; 60 typedef typename SolverT::GraphMetadata GraphMetadata; 61 62 private: 63 64 class NodeEntry { 65 public: 66 typedef std::vector<EdgeId> AdjEdgeList; 67 typedef AdjEdgeList::size_type AdjEdgeIdx; 68 typedef AdjEdgeList::const_iterator AdjEdgeItr; 69 70 static AdjEdgeIdx getInvalidAdjEdgeIdx() { 71 return std::numeric_limits<AdjEdgeIdx>::max(); 72 } 73 74 NodeEntry(VectorPtr Costs) : Costs(Costs) {} 75 76 AdjEdgeIdx addAdjEdgeId(EdgeId EId) { 77 AdjEdgeIdx Idx = AdjEdgeIds.size(); 78 AdjEdgeIds.push_back(EId); 79 return Idx; 80 } 81 82 void removeAdjEdgeId(Graph &G, NodeId ThisNId, AdjEdgeIdx Idx) { 83 // Swap-and-pop for fast removal. 84 // 1) Update the adj index of the edge currently at back(). 85 // 2) Move last Edge down to Idx. 86 // 3) pop_back() 87 // If Idx == size() - 1 then the setAdjEdgeIdx and swap are 88 // redundant, but both operations are cheap. 89 G.getEdge(AdjEdgeIds.back()).setAdjEdgeIdx(ThisNId, Idx); 90 AdjEdgeIds[Idx] = AdjEdgeIds.back(); 91 AdjEdgeIds.pop_back(); 92 } 93 94 const AdjEdgeList& getAdjEdgeIds() const { return AdjEdgeIds; } 95 96 VectorPtr Costs; 97 NodeMetadata Metadata; 98 private: 99 AdjEdgeList AdjEdgeIds; 100 }; 101 102 class EdgeEntry { 103 public: 104 EdgeEntry(NodeId N1Id, NodeId N2Id, MatrixPtr Costs) 105 : Costs(Costs) { 106 NIds[0] = N1Id; 107 NIds[1] = N2Id; 108 ThisEdgeAdjIdxs[0] = NodeEntry::getInvalidAdjEdgeIdx(); 109 ThisEdgeAdjIdxs[1] = NodeEntry::getInvalidAdjEdgeIdx(); 110 } 111 112 void invalidate() { 113 NIds[0] = NIds[1] = Graph::invalidNodeId(); 114 ThisEdgeAdjIdxs[0] = ThisEdgeAdjIdxs[1] = 115 NodeEntry::getInvalidAdjEdgeIdx(); 116 Costs = nullptr; 117 } 118 119 void connectToN(Graph &G, EdgeId ThisEdgeId, unsigned NIdx) { 120 assert(ThisEdgeAdjIdxs[NIdx] == NodeEntry::getInvalidAdjEdgeIdx() && 121 "Edge already connected to NIds[NIdx]."); 122 NodeEntry &N = G.getNode(NIds[NIdx]); 123 ThisEdgeAdjIdxs[NIdx] = N.addAdjEdgeId(ThisEdgeId); 124 } 125 126 void connectTo(Graph &G, EdgeId ThisEdgeId, NodeId NId) { 127 if (NId == NIds[0]) 128 connectToN(G, ThisEdgeId, 0); 129 else { 130 assert(NId == NIds[1] && "Edge does not connect NId."); 131 connectToN(G, ThisEdgeId, 1); 132 } 133 } 134 135 void connect(Graph &G, EdgeId ThisEdgeId) { 136 connectToN(G, ThisEdgeId, 0); 137 connectToN(G, ThisEdgeId, 1); 138 } 139 140 void setAdjEdgeIdx(NodeId NId, typename NodeEntry::AdjEdgeIdx NewIdx) { 141 if (NId == NIds[0]) 142 ThisEdgeAdjIdxs[0] = NewIdx; 143 else { 144 assert(NId == NIds[1] && "Edge not connected to NId"); 145 ThisEdgeAdjIdxs[1] = NewIdx; 146 } 147 } 148 149 void disconnectFromN(Graph &G, unsigned NIdx) { 150 assert(ThisEdgeAdjIdxs[NIdx] != NodeEntry::getInvalidAdjEdgeIdx() && 151 "Edge not connected to NIds[NIdx]."); 152 NodeEntry &N = G.getNode(NIds[NIdx]); 153 N.removeAdjEdgeId(G, NIds[NIdx], ThisEdgeAdjIdxs[NIdx]); 154 ThisEdgeAdjIdxs[NIdx] = NodeEntry::getInvalidAdjEdgeIdx(); 155 } 156 157 void disconnectFrom(Graph &G, NodeId NId) { 158 if (NId == NIds[0]) 159 disconnectFromN(G, 0); 160 else { 161 assert(NId == NIds[1] && "Edge does not connect NId"); 162 disconnectFromN(G, 1); 163 } 164 } 165 166 NodeId getN1Id() const { return NIds[0]; } 167 NodeId getN2Id() const { return NIds[1]; } 168 MatrixPtr Costs; 169 EdgeMetadata Metadata; 170 private: 171 NodeId NIds[2]; 172 typename NodeEntry::AdjEdgeIdx ThisEdgeAdjIdxs[2]; 173 }; 174 175 // ----- MEMBERS ----- 176 177 GraphMetadata Metadata; 178 CostAllocator CostAlloc; 179 SolverT *Solver; 180 181 typedef std::vector<NodeEntry> NodeVector; 182 typedef std::vector<NodeId> FreeNodeVector; 183 NodeVector Nodes; 184 FreeNodeVector FreeNodeIds; 185 186 typedef std::vector<EdgeEntry> EdgeVector; 187 typedef std::vector<EdgeId> FreeEdgeVector; 188 EdgeVector Edges; 189 FreeEdgeVector FreeEdgeIds; 190 191 // ----- INTERNAL METHODS ----- 192 193 NodeEntry &getNode(NodeId NId) { 194 assert(NId < Nodes.size() && "Out of bound NodeId"); 195 return Nodes[NId]; 196 } 197 const NodeEntry &getNode(NodeId NId) const { 198 assert(NId < Nodes.size() && "Out of bound NodeId"); 199 return Nodes[NId]; 200 } 201 202 EdgeEntry& getEdge(EdgeId EId) { return Edges[EId]; } 203 const EdgeEntry& getEdge(EdgeId EId) const { return Edges[EId]; } 204 205 NodeId addConstructedNode(NodeEntry N) { 206 NodeId NId = 0; 207 if (!FreeNodeIds.empty()) { 208 NId = FreeNodeIds.back(); 209 FreeNodeIds.pop_back(); 210 Nodes[NId] = std::move(N); 211 } else { 212 NId = Nodes.size(); 213 Nodes.push_back(std::move(N)); 214 } 215 return NId; 216 } 217 218 EdgeId addConstructedEdge(EdgeEntry E) { 219 assert(findEdge(E.getN1Id(), E.getN2Id()) == invalidEdgeId() && 220 "Attempt to add duplicate edge."); 221 EdgeId EId = 0; 222 if (!FreeEdgeIds.empty()) { 223 EId = FreeEdgeIds.back(); 224 FreeEdgeIds.pop_back(); 225 Edges[EId] = std::move(E); 226 } else { 227 EId = Edges.size(); 228 Edges.push_back(std::move(E)); 229 } 230 231 EdgeEntry &NE = getEdge(EId); 232 233 // Add the edge to the adjacency sets of its nodes. 234 NE.connect(*this, EId); 235 return EId; 236 } 237 238 Graph(const Graph &Other) {} 239 void operator=(const Graph &Other) {} 240 241 public: 242 243 typedef typename NodeEntry::AdjEdgeItr AdjEdgeItr; 244 245 class NodeItr { 246 public: 247 typedef std::forward_iterator_tag iterator_category; 248 typedef NodeId value_type; 249 typedef int difference_type; 250 typedef NodeId* pointer; 251 typedef NodeId& reference; 252 253 NodeItr(NodeId CurNId, const Graph &G) 254 : CurNId(CurNId), EndNId(G.Nodes.size()), FreeNodeIds(G.FreeNodeIds) { 255 this->CurNId = findNextInUse(CurNId); // Move to first in-use node id 256 } 257 258 bool operator==(const NodeItr &O) const { return CurNId == O.CurNId; } 259 bool operator!=(const NodeItr &O) const { return !(*this == O); } 260 NodeItr& operator++() { CurNId = findNextInUse(++CurNId); return *this; } 261 NodeId operator*() const { return CurNId; } 262 263 private: 264 NodeId findNextInUse(NodeId NId) const { 265 while (NId < EndNId && 266 std::find(FreeNodeIds.begin(), FreeNodeIds.end(), NId) != 267 FreeNodeIds.end()) { 268 ++NId; 269 } 270 return NId; 271 } 272 273 NodeId CurNId, EndNId; 274 const FreeNodeVector &FreeNodeIds; 275 }; 276 277 class EdgeItr { 278 public: 279 EdgeItr(EdgeId CurEId, const Graph &G) 280 : CurEId(CurEId), EndEId(G.Edges.size()), FreeEdgeIds(G.FreeEdgeIds) { 281 this->CurEId = findNextInUse(CurEId); // Move to first in-use edge id 282 } 283 284 bool operator==(const EdgeItr &O) const { return CurEId == O.CurEId; } 285 bool operator!=(const EdgeItr &O) const { return !(*this == O); } 286 EdgeItr& operator++() { CurEId = findNextInUse(++CurEId); return *this; } 287 EdgeId operator*() const { return CurEId; } 288 289 private: 290 EdgeId findNextInUse(EdgeId EId) const { 291 while (EId < EndEId && 292 std::find(FreeEdgeIds.begin(), FreeEdgeIds.end(), EId) != 293 FreeEdgeIds.end()) { 294 ++EId; 295 } 296 return EId; 297 } 298 299 EdgeId CurEId, EndEId; 300 const FreeEdgeVector &FreeEdgeIds; 301 }; 302 303 class NodeIdSet { 304 public: 305 NodeIdSet(const Graph &G) : G(G) { } 306 NodeItr begin() const { return NodeItr(0, G); } 307 NodeItr end() const { return NodeItr(G.Nodes.size(), G); } 308 bool empty() const { return G.Nodes.empty(); } 309 typename NodeVector::size_type size() const { 310 return G.Nodes.size() - G.FreeNodeIds.size(); 311 } 312 private: 313 const Graph& G; 314 }; 315 316 class EdgeIdSet { 317 public: 318 EdgeIdSet(const Graph &G) : G(G) { } 319 EdgeItr begin() const { return EdgeItr(0, G); } 320 EdgeItr end() const { return EdgeItr(G.Edges.size(), G); } 321 bool empty() const { return G.Edges.empty(); } 322 typename NodeVector::size_type size() const { 323 return G.Edges.size() - G.FreeEdgeIds.size(); 324 } 325 private: 326 const Graph& G; 327 }; 328 329 class AdjEdgeIdSet { 330 public: 331 AdjEdgeIdSet(const NodeEntry &NE) : NE(NE) { } 332 typename NodeEntry::AdjEdgeItr begin() const { 333 return NE.getAdjEdgeIds().begin(); 334 } 335 typename NodeEntry::AdjEdgeItr end() const { 336 return NE.getAdjEdgeIds().end(); 337 } 338 bool empty() const { return NE.getAdjEdgeIds().empty(); } 339 typename NodeEntry::AdjEdgeList::size_type size() const { 340 return NE.getAdjEdgeIds().size(); 341 } 342 private: 343 const NodeEntry &NE; 344 }; 345 346 /// @brief Construct an empty PBQP graph. 347 Graph() : Solver(nullptr) {} 348 349 /// @brief Construct an empty PBQP graph with the given graph metadata. 350 Graph(GraphMetadata Metadata) : Metadata(Metadata), Solver(nullptr) {} 351 352 /// @brief Get a reference to the graph metadata. 353 GraphMetadata& getMetadata() { return Metadata; } 354 355 /// @brief Get a const-reference to the graph metadata. 356 const GraphMetadata& getMetadata() const { return Metadata; } 357 358 /// @brief Lock this graph to the given solver instance in preparation 359 /// for running the solver. This method will call solver.handleAddNode for 360 /// each node in the graph, and handleAddEdge for each edge, to give the 361 /// solver an opportunity to set up any requried metadata. 362 void setSolver(SolverT &S) { 363 assert(!Solver && "Solver already set. Call unsetSolver()."); 364 Solver = &S; 365 for (auto NId : nodeIds()) 366 Solver->handleAddNode(NId); 367 for (auto EId : edgeIds()) 368 Solver->handleAddEdge(EId); 369 } 370 371 /// @brief Release from solver instance. 372 void unsetSolver() { 373 assert(Solver && "Solver not set."); 374 Solver = nullptr; 375 } 376 377 /// @brief Add a node with the given costs. 378 /// @param Costs Cost vector for the new node. 379 /// @return Node iterator for the added node. 380 template <typename OtherVectorT> 381 NodeId addNode(OtherVectorT Costs) { 382 // Get cost vector from the problem domain 383 VectorPtr AllocatedCosts = CostAlloc.getVector(std::move(Costs)); 384 NodeId NId = addConstructedNode(NodeEntry(AllocatedCosts)); 385 if (Solver) 386 Solver->handleAddNode(NId); 387 return NId; 388 } 389 390 /// @brief Add a node bypassing the cost allocator. 391 /// @param Costs Cost vector ptr for the new node (must be convertible to 392 /// VectorPtr). 393 /// @return Node iterator for the added node. 394 /// 395 /// This method allows for fast addition of a node whose costs don't need 396 /// to be passed through the cost allocator. The most common use case for 397 /// this is when duplicating costs from an existing node (when using a 398 /// pooling allocator). These have already been uniqued, so we can avoid 399 /// re-constructing and re-uniquing them by attaching them directly to the 400 /// new node. 401 template <typename OtherVectorPtrT> 402 NodeId addNodeBypassingCostAllocator(OtherVectorPtrT Costs) { 403 NodeId NId = addConstructedNode(NodeEntry(Costs)); 404 if (Solver) 405 Solver->handleAddNode(NId); 406 return NId; 407 } 408 409 /// @brief Add an edge between the given nodes with the given costs. 410 /// @param N1Id First node. 411 /// @param N2Id Second node. 412 /// @param Costs Cost matrix for new edge. 413 /// @return Edge iterator for the added edge. 414 template <typename OtherVectorT> 415 EdgeId addEdge(NodeId N1Id, NodeId N2Id, OtherVectorT Costs) { 416 assert(getNodeCosts(N1Id).getLength() == Costs.getRows() && 417 getNodeCosts(N2Id).getLength() == Costs.getCols() && 418 "Matrix dimensions mismatch."); 419 // Get cost matrix from the problem domain. 420 MatrixPtr AllocatedCosts = CostAlloc.getMatrix(std::move(Costs)); 421 EdgeId EId = addConstructedEdge(EdgeEntry(N1Id, N2Id, AllocatedCosts)); 422 if (Solver) 423 Solver->handleAddEdge(EId); 424 return EId; 425 } 426 427 /// @brief Add an edge bypassing the cost allocator. 428 /// @param N1Id First node. 429 /// @param N2Id Second node. 430 /// @param Costs Cost matrix for new edge. 431 /// @return Edge iterator for the added edge. 432 /// 433 /// This method allows for fast addition of an edge whose costs don't need 434 /// to be passed through the cost allocator. The most common use case for 435 /// this is when duplicating costs from an existing edge (when using a 436 /// pooling allocator). These have already been uniqued, so we can avoid 437 /// re-constructing and re-uniquing them by attaching them directly to the 438 /// new edge. 439 template <typename OtherMatrixPtrT> 440 NodeId addEdgeBypassingCostAllocator(NodeId N1Id, NodeId N2Id, 441 OtherMatrixPtrT Costs) { 442 assert(getNodeCosts(N1Id).getLength() == Costs->getRows() && 443 getNodeCosts(N2Id).getLength() == Costs->getCols() && 444 "Matrix dimensions mismatch."); 445 // Get cost matrix from the problem domain. 446 EdgeId EId = addConstructedEdge(EdgeEntry(N1Id, N2Id, Costs)); 447 if (Solver) 448 Solver->handleAddEdge(EId); 449 return EId; 450 } 451 452 /// @brief Returns true if the graph is empty. 453 bool empty() const { return NodeIdSet(*this).empty(); } 454 455 NodeIdSet nodeIds() const { return NodeIdSet(*this); } 456 EdgeIdSet edgeIds() const { return EdgeIdSet(*this); } 457 458 AdjEdgeIdSet adjEdgeIds(NodeId NId) { return AdjEdgeIdSet(getNode(NId)); } 459 460 /// @brief Get the number of nodes in the graph. 461 /// @return Number of nodes in the graph. 462 unsigned getNumNodes() const { return NodeIdSet(*this).size(); } 463 464 /// @brief Get the number of edges in the graph. 465 /// @return Number of edges in the graph. 466 unsigned getNumEdges() const { return EdgeIdSet(*this).size(); } 467 468 /// @brief Set a node's cost vector. 469 /// @param NId Node to update. 470 /// @param Costs New costs to set. 471 template <typename OtherVectorT> 472 void setNodeCosts(NodeId NId, OtherVectorT Costs) { 473 VectorPtr AllocatedCosts = CostAlloc.getVector(std::move(Costs)); 474 if (Solver) 475 Solver->handleSetNodeCosts(NId, *AllocatedCosts); 476 getNode(NId).Costs = AllocatedCosts; 477 } 478 479 /// @brief Get a VectorPtr to a node's cost vector. Rarely useful - use 480 /// getNodeCosts where possible. 481 /// @param NId Node id. 482 /// @return VectorPtr to node cost vector. 483 /// 484 /// This method is primarily useful for duplicating costs quickly by 485 /// bypassing the cost allocator. See addNodeBypassingCostAllocator. Prefer 486 /// getNodeCosts when dealing with node cost values. 487 const VectorPtr& getNodeCostsPtr(NodeId NId) const { 488 return getNode(NId).Costs; 489 } 490 491 /// @brief Get a node's cost vector. 492 /// @param NId Node id. 493 /// @return Node cost vector. 494 const Vector& getNodeCosts(NodeId NId) const { 495 return *getNodeCostsPtr(NId); 496 } 497 498 NodeMetadata& getNodeMetadata(NodeId NId) { 499 return getNode(NId).Metadata; 500 } 501 502 const NodeMetadata& getNodeMetadata(NodeId NId) const { 503 return getNode(NId).Metadata; 504 } 505 506 typename NodeEntry::AdjEdgeList::size_type getNodeDegree(NodeId NId) const { 507 return getNode(NId).getAdjEdgeIds().size(); 508 } 509 510 /// @brief Update an edge's cost matrix. 511 /// @param EId Edge id. 512 /// @param Costs New cost matrix. 513 template <typename OtherMatrixT> 514 void updateEdgeCosts(EdgeId EId, OtherMatrixT Costs) { 515 MatrixPtr AllocatedCosts = CostAlloc.getMatrix(std::move(Costs)); 516 if (Solver) 517 Solver->handleUpdateCosts(EId, *AllocatedCosts); 518 getEdge(EId).Costs = AllocatedCosts; 519 } 520 521 /// @brief Get a MatrixPtr to a node's cost matrix. Rarely useful - use 522 /// getEdgeCosts where possible. 523 /// @param EId Edge id. 524 /// @return MatrixPtr to edge cost matrix. 525 /// 526 /// This method is primarily useful for duplicating costs quickly by 527 /// bypassing the cost allocator. See addNodeBypassingCostAllocator. Prefer 528 /// getEdgeCosts when dealing with edge cost values. 529 const MatrixPtr& getEdgeCostsPtr(EdgeId EId) const { 530 return getEdge(EId).Costs; 531 } 532 533 /// @brief Get an edge's cost matrix. 534 /// @param EId Edge id. 535 /// @return Edge cost matrix. 536 const Matrix& getEdgeCosts(EdgeId EId) const { 537 return *getEdge(EId).Costs; 538 } 539 540 EdgeMetadata& getEdgeMetadata(EdgeId EId) { 541 return getEdge(EId).Metadata; 542 } 543 544 const EdgeMetadata& getEdgeMetadata(EdgeId EId) const { 545 return getEdge(EId).Metadata; 546 } 547 548 /// @brief Get the first node connected to this edge. 549 /// @param EId Edge id. 550 /// @return The first node connected to the given edge. 551 NodeId getEdgeNode1Id(EdgeId EId) const { 552 return getEdge(EId).getN1Id(); 553 } 554 555 /// @brief Get the second node connected to this edge. 556 /// @param EId Edge id. 557 /// @return The second node connected to the given edge. 558 NodeId getEdgeNode2Id(EdgeId EId) const { 559 return getEdge(EId).getN2Id(); 560 } 561 562 /// @brief Get the "other" node connected to this edge. 563 /// @param EId Edge id. 564 /// @param NId Node id for the "given" node. 565 /// @return The iterator for the "other" node connected to this edge. 566 NodeId getEdgeOtherNodeId(EdgeId EId, NodeId NId) { 567 EdgeEntry &E = getEdge(EId); 568 if (E.getN1Id() == NId) { 569 return E.getN2Id(); 570 } // else 571 return E.getN1Id(); 572 } 573 574 /// @brief Get the edge connecting two nodes. 575 /// @param N1Id First node id. 576 /// @param N2Id Second node id. 577 /// @return An id for edge (N1Id, N2Id) if such an edge exists, 578 /// otherwise returns an invalid edge id. 579 EdgeId findEdge(NodeId N1Id, NodeId N2Id) { 580 for (auto AEId : adjEdgeIds(N1Id)) { 581 if ((getEdgeNode1Id(AEId) == N2Id) || 582 (getEdgeNode2Id(AEId) == N2Id)) { 583 return AEId; 584 } 585 } 586 return invalidEdgeId(); 587 } 588 589 /// @brief Remove a node from the graph. 590 /// @param NId Node id. 591 void removeNode(NodeId NId) { 592 if (Solver) 593 Solver->handleRemoveNode(NId); 594 NodeEntry &N = getNode(NId); 595 // TODO: Can this be for-each'd? 596 for (AdjEdgeItr AEItr = N.adjEdgesBegin(), 597 AEEnd = N.adjEdgesEnd(); 598 AEItr != AEEnd;) { 599 EdgeId EId = *AEItr; 600 ++AEItr; 601 removeEdge(EId); 602 } 603 FreeNodeIds.push_back(NId); 604 } 605 606 /// @brief Disconnect an edge from the given node. 607 /// 608 /// Removes the given edge from the adjacency list of the given node. 609 /// This operation leaves the edge in an 'asymmetric' state: It will no 610 /// longer appear in an iteration over the given node's (NId's) edges, but 611 /// will appear in an iteration over the 'other', unnamed node's edges. 612 /// 613 /// This does not correspond to any normal graph operation, but exists to 614 /// support efficient PBQP graph-reduction based solvers. It is used to 615 /// 'effectively' remove the unnamed node from the graph while the solver 616 /// is performing the reduction. The solver will later call reconnectNode 617 /// to restore the edge in the named node's adjacency list. 618 /// 619 /// Since the degree of a node is the number of connected edges, 620 /// disconnecting an edge from a node 'u' will cause the degree of 'u' to 621 /// drop by 1. 622 /// 623 /// A disconnected edge WILL still appear in an iteration over the graph 624 /// edges. 625 /// 626 /// A disconnected edge should not be removed from the graph, it should be 627 /// reconnected first. 628 /// 629 /// A disconnected edge can be reconnected by calling the reconnectEdge 630 /// method. 631 void disconnectEdge(EdgeId EId, NodeId NId) { 632 if (Solver) 633 Solver->handleDisconnectEdge(EId, NId); 634 635 EdgeEntry &E = getEdge(EId); 636 E.disconnectFrom(*this, NId); 637 } 638 639 /// @brief Convenience method to disconnect all neighbours from the given 640 /// node. 641 void disconnectAllNeighborsFromNode(NodeId NId) { 642 for (auto AEId : adjEdgeIds(NId)) 643 disconnectEdge(AEId, getEdgeOtherNodeId(AEId, NId)); 644 } 645 646 /// @brief Re-attach an edge to its nodes. 647 /// 648 /// Adds an edge that had been previously disconnected back into the 649 /// adjacency set of the nodes that the edge connects. 650 void reconnectEdge(EdgeId EId, NodeId NId) { 651 EdgeEntry &E = getEdge(EId); 652 E.connectTo(*this, EId, NId); 653 if (Solver) 654 Solver->handleReconnectEdge(EId, NId); 655 } 656 657 /// @brief Remove an edge from the graph. 658 /// @param EId Edge id. 659 void removeEdge(EdgeId EId) { 660 if (Solver) 661 Solver->handleRemoveEdge(EId); 662 EdgeEntry &E = getEdge(EId); 663 E.disconnect(); 664 FreeEdgeIds.push_back(EId); 665 Edges[EId].invalidate(); 666 } 667 668 /// @brief Remove all nodes and edges from the graph. 669 void clear() { 670 Nodes.clear(); 671 FreeNodeIds.clear(); 672 Edges.clear(); 673 FreeEdgeIds.clear(); 674 } 675 }; 676 677 } // namespace PBQP 678 } // namespace llvm 679 680 #endif // LLVM_CODEGEN_PBQP_GRAPH_HPP 681