lib/CodeGen/PBQP/Solution.h

   1 //===-- Solution.h ------- PBQP Solution ------------------------*- 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 Solution class.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #ifndef LLVM_CODEGEN_PBQP_SOLUTION_H
  15 #define LLVM_CODEGEN_PBQP_SOLUTION_H
  16
  17 #include "Math.h"
  18 #include "Graph.h"
  19
  20 #include <map>
  21
  22 namespace PBQP {
  23
  24   /// \brief Represents a solution to a PBQP problem.
  25   ///
  26   /// To get the selection for each node in the problem use the getSelection method.
  27   class Solution {
  28   private:
  29     typedef std::map<Graph::NodeItr, unsigned, NodeItrComparator> SelectionsMap;
  30     SelectionsMap selections;
  31
  32   public:
  33
  34     /// \brief Number of nodes for which selections have been made.
  35     /// @return Number of nodes for which selections have been made.
  36     unsigned numNodes() const { return selections.size(); }
  37
  38     /// \brief Set the selection for a given node.
  39     /// @param nItr Node iterator.
  40     /// @param selection Selection for nItr.
  41     void setSelection(Graph::NodeItr nItr, unsigned selection) {
  42       selections[nItr] = selection;
  43     }
  44
  45     /// \brief Get a node's selection.
  46     /// @param nItr Node iterator.
  47     /// @return The selection for nItr;
  48     unsigned getSelection(Graph::NodeItr nItr) const {
  49       SelectionsMap::const_iterator sItr = selections.find(nItr);
  50       assert(sItr != selections.end() && "No selection for node.");
  51       return sItr->second;
  52     }
  53
  54   };
  55
  56 }
  57
  58 #endif // LLVM_CODEGEN_PBQP_SOLUTION_H