include/llvm/CodeGen/SchedulerRegistry.h

   1 //===-- llvm/CodeGen/SchedulerRegistry.h ------------------------*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file was developed by the James M. Laskey and is distributed under
   6 // the University of Illinois Open Source License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file contains the implementation for instruction scheduler function
  11 // pass registry (RegisterScheduler).
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #ifndef LLVM_CODEGENSCHEDULERREGISTRY_H
  16 #define LLVM_CODEGENSCHEDULERREGISTRY_H
  17
  18 #include "llvm/CodeGen/MachinePassRegistry.h"
  19
  20 namespace llvm {
  21
  22 //===----------------------------------------------------------------------===//
  23 ///
  24 /// RegisterScheduler class - Track the registration of instruction schedulers.
  25 ///
  26 //===----------------------------------------------------------------------===//
  27
  28 class SelectionDAGISel;
  29 class ScheduleDAG;
  30 class SelectionDAG;
  31 class MachineBasicBlock;
  32
  33 class RegisterScheduler : public MachinePassRegistryNode {
  34
  35 public:
  36
  37   typedef ScheduleDAG *(*FunctionPassCtor)(SelectionDAGISel*, SelectionDAG*,
  38                                            MachineBasicBlock*);
  39
  40   static MachinePassRegistry Registry;
  41
  42   RegisterScheduler(const char *N, const char *D, FunctionPassCtor C)
  43   : MachinePassRegistryNode(N, D, (MachinePassCtor)C)
  44   { Registry.Add(this); }
  45   ~RegisterScheduler() { Registry.Remove(this); }
  46
  47
  48   // Accessors.
  49   //
  50   RegisterScheduler *getNext() const {
  51     return (RegisterScheduler *)MachinePassRegistryNode::getNext();
  52   }
  53   static RegisterScheduler *getList() {
  54     return (RegisterScheduler *)Registry.getList();
  55   }
  56   static FunctionPassCtor getDefault() {
  57     return (FunctionPassCtor)Registry.getDefault();
  58   }
  59   static void setDefault(FunctionPassCtor C) {
  60     Registry.setDefault((MachinePassCtor)C);
  61   }
  62   static void setListener(MachinePassRegistryListener *L) {
  63     Registry.setListener(L);
  64   }
  65
  66 };
  67
  68 } // end namespace llvm
  69
  70
  71 #endif