1 //===- InterfaceSupport.cpp - MLIR Interface Support Classes --------------===//
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 defines several support classes for defining interfaces.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #include "mlir/Support/InterfaceSupport.h"
14 #include "llvm/Support/Debug.h"
15 #include "llvm/Support/raw_ostream.h"
16 
17 #define DEBUG_TYPE "interfaces"
18 
19 using namespace mlir;
20 
21 detail::InterfaceMap::InterfaceMap(
22     MutableArrayRef<std::pair<TypeID, void *>> elements)
23     : interfaces(elements.begin(), elements.end()) {
24   llvm::sort(interfaces, [](const auto &lhs, const auto &rhs) {
25     return compare(lhs.first, rhs.first);
26   });
27 }
28 
29 void detail::InterfaceMap::insert(
30     ArrayRef<std::pair<TypeID, void *>> elements) {
31   // Insert directly into the right position to keep the interfaces sorted.
32   for (auto &element : elements) {
33     TypeID id = element.first;
34     auto *it = llvm::lower_bound(interfaces, id, [](const auto &it, TypeID id) {
35       return compare(it.first, id);
36     });
37     if (it != interfaces.end() && it->first == id) {
38       LLVM_DEBUG(llvm::dbgs() << "Ignoring repeated interface registration");
39       free(element.second);
40       continue;
41     }
42     interfaces.insert(it, element);
43   }
44 }
45