folly/detail/RangeCommon.cpp

   1 /*
   2  * Copyright 2016 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
  17 #include <folly/detail/RangeCommon.h>
  18
  19 #include <bitset>
  20 #include <folly/SparseByteSet.h>
  21
  22 namespace folly {
  23
  24 namespace detail {
  25
  26 size_t qfind_first_byte_of_bitset(const StringPieceLite haystack,
  27                                   const StringPieceLite needles) {
  28   std::bitset<256> s;
  29   for (auto needle : needles) {
  30     s[(uint8_t)needle] = true;
  31   }
  32   for (size_t index = 0; index < haystack.size(); ++index) {
  33     if (s[(uint8_t)haystack[index]]) {
  34       return index;
  35     }
  36   }
  37   return std::string::npos;
  38 }
  39
  40 size_t qfind_first_byte_of_byteset(const StringPieceLite haystack,
  41                                    const StringPieceLite needles) {
  42   SparseByteSet s;
  43   for (auto needle: needles) {
  44     s.add(uint8_t(needle));
  45   }
  46   for (size_t index = 0; index < haystack.size(); ++index) {
  47     if (s.contains(uint8_t(haystack[index]))) {
  48       return index;
  49     }
  50   }
  51   return std::string::npos;
  52 }
  53
  54 }
  55
  56 }