folly/test/function_benchmark/benchmark_impl.h

   1 /*
   2  * Copyright 2015 Facebook, Inc.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *   http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 #ifndef BENCHMARK_IMPL_H_
  17 #define BENCHMARK_IMPL_H_
  18
  19 #include <functional>
  20
  21 class TestClass;
  22 class VirtualClass;
  23
  24 void BM_fn_ptr_invoke_impl(int iters, void (*fn)());
  25 void BM_std_function_invoke_impl(int iters, const std::function<void()>& fn);
  26 void BM_mem_fn_invoke_impl(int iters,
  27                            TestClass* tc,
  28                            void (TestClass::*memfn)());
  29 void BM_virtual_fn_invoke_impl(int iters, VirtualClass* vc);
  30
  31 // Inlined version of BM_fn_ptr_invoke_impl().
  32 // The compiler could potentially even optimize the call to the function
  33 // pointer if it is a constexpr.
  34 inline void BM_fn_ptr_invoke_inlined_impl(int iters, void (*fn)()) {
  35   for (int n = 0; n < iters; ++n) {
  36     fn();
  37   }
  38 }
  39
  40 // Invoke a function object as a template parameter.
  41 // This can be used to directly invoke lambda functions
  42 template<typename T>
  43 void BM_invoke_fn_template_impl(int iters, const T& fn) {
  44   for (int n = 0; n < iters; ++n) {
  45     fn();
  46   }
  47 }
  48
  49 #endif // BENCHMARK_IMPL_H_