1 //===- llvm/CodeGen/RegAllocRegistry.h --------------------------*- 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 // This file contains the implementation for register allocator function
10 // pass registry (RegisterRegAlloc).
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #ifndef LLVM_CODEGEN_REGALLOCREGISTRY_H
15 #define LLVM_CODEGEN_REGALLOCREGISTRY_H
16 
17 #include "llvm/CodeGen/MachinePassRegistry.h"
18 
19 namespace llvm {
20 
21 class FunctionPass;
22 
23 //===----------------------------------------------------------------------===//
24 ///
25 /// RegisterRegAllocBase class - Track the registration of register allocators.
26 ///
27 //===----------------------------------------------------------------------===//
28 template <class SubClass>
29 class RegisterRegAllocBase : public MachinePassRegistryNode<FunctionPass *(*)()> {
30 public:
31   using FunctionPassCtor = FunctionPass *(*)();
32 
33   static MachinePassRegistry<FunctionPassCtor> Registry;
34 
35   RegisterRegAllocBase(const char *N, const char *D, FunctionPassCtor C)
36       : MachinePassRegistryNode(N, D, C) {
37     Registry.Add(this);
38   }
39 
40   ~RegisterRegAllocBase() { Registry.Remove(this); }
41 
42   // Accessors.
43   SubClass *getNext() const {
44     return static_cast<SubClass *>(MachinePassRegistryNode::getNext());
45   }
46 
47   static SubClass *getList() {
48     return static_cast<SubClass *>(Registry.getList());
49   }
50 
51   static FunctionPassCtor getDefault() { return Registry.getDefault(); }
52 
53   static void setDefault(FunctionPassCtor C) { Registry.setDefault(C); }
54 
55   static void setListener(MachinePassRegistryListener<FunctionPassCtor> *L) {
56     Registry.setListener(L);
57   }
58 };
59 
60 class RegisterRegAlloc : public RegisterRegAllocBase<RegisterRegAlloc> {
61 public:
62   RegisterRegAlloc(const char *N, const char *D, FunctionPassCtor C)
63     : RegisterRegAllocBase(N, D, C) {}
64 };
65 
66 /// RegisterRegAlloc's global Registry tracks allocator registration.
67 template <class T>
68 MachinePassRegistry<RegisterRegAlloc::FunctionPassCtor>
69 RegisterRegAllocBase<T>::Registry;
70 
71 } // end namespace llvm
72 
73 #endif // LLVM_CODEGEN_REGALLOCREGISTRY_H
74