1//===- Solution.h - PBQP Solution -------------------------------*- C++ -*-===// 2// 3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4// See https://llvm.org/LICENSE.txt for license information. 5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6// 7//===----------------------------------------------------------------------===// 8// 9// PBQP Solution class. 10// 11//===----------------------------------------------------------------------===// 12 13#ifndef LLVM_CODEGEN_PBQP_SOLUTION_H 14#define LLVM_CODEGEN_PBQP_SOLUTION_H 15 16#include "llvm/CodeGen/PBQP/Graph.h" 17#include <cassert> 18#include <map> 19 20namespace llvm { 21namespace PBQP { 22 23 /// Represents a solution to a PBQP problem. 24 /// 25 /// To get the selection for each node in the problem use the getSelection method. 26 class Solution { 27 private: 28 using SelectionsMap = std::map<GraphBase::NodeId, unsigned>; 29 SelectionsMap selections; 30 31 public: 32 /// Initialise an empty solution. 33 Solution() = default; 34 35 /// Set the selection for a given node. 36 /// @param nodeId Node id. 37 /// @param selection Selection for nodeId. 38 void setSelection(GraphBase::NodeId nodeId, unsigned selection) { 39 selections[nodeId] = selection; 40 } 41 42 /// Get a node's selection. 43 /// @param nodeId Node id. 44 /// @return The selection for nodeId; 45 unsigned getSelection(GraphBase::NodeId nodeId) const { 46 SelectionsMap::const_iterator sItr = selections.find(nodeId); 47 assert(sItr != selections.end() && "No selection for node."); 48 return sItr->second; 49 } 50 }; 51 52} // end namespace PBQP 53} // end namespace llvm 54 55#endif // LLVM_CODEGEN_PBQP_SOLUTION_H 56