SmallVectorExtras.h

Go to the documentation of this file.

//===- llvm/ADT/SmallVectorExtras.h -----------------------------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
///
/// \file
/// This file defines less commonly used SmallVector utilities.
///
//===----------------------------------------------------------------------===//
 
#ifndef LLVM_ADT_SMALLVECTOREXTRAS_H
#define LLVM_ADT_SMALLVECTOREXTRAS_H
 
#include "llvm/ADT/STLExtras.h"
#include "llvm/ADT/SmallVector.h"
 
namespace llvm {
 
/// Filter a range to a SmallVector with the element types deduced.
template <unsigned Size, class ContainerTy, class PredicateFn>
auto filter_to_vector(ContainerTy &&C, PredicateFn &&Pred) {
  return to_vector<Size>(make_filter_range(std::forward<ContainerTy>(C),
                                           std::forward<PredicateFn>(Pred)));
}
 
/// Filter a range to a SmallVector with the element types deduced.
template <class ContainerTy, class PredicateFn>
auto filter_to_vector(ContainerTy &&C, PredicateFn &&Pred) {
  return to_vector(make_filter_range(std::forward<ContainerTy>(C),
                                     std::forward<PredicateFn>(Pred)));
}
 
/// Map a range to a SmallVector with element types deduced from the mapping.
/// \p F can be a function, lambda, or member pointer.
template <unsigned Size, class ContainerTy, class FuncTy>
auto map_to_vector(ContainerTy &&C, FuncTy &&F) {
  return to_vector<Size>(
      map_range(std::forward<ContainerTy>(C), std::forward<FuncTy>(F)));
}
 
/// Map a range to a SmallVector with element types deduced from the mapping.
/// \p F can be a function, lambda, or member pointer.
template <class ContainerTy, class FuncTy>
auto map_to_vector(ContainerTy &&C, FuncTy &&F) {
  return to_vector(
      map_range(std::forward<ContainerTy>(C), std::forward<FuncTy>(F)));
}
 
} // namespace llvm
 
#endif // LLVM_ADT_SMALLVECTOREXTRAS_H