base/i18n/string_search.cc

   1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include <stdint.h>
   6
   7 #include "base/i18n/string_search.h"
   8 #include "base/logging.h"
   9
  10 #include "third_party/icu/source/i18n/unicode/usearch.h"
  11
  12 namespace base {
  13 namespace i18n {
  14
  15 FixedPatternStringSearchIgnoringCaseAndAccents::
  16 FixedPatternStringSearchIgnoringCaseAndAccents(const string16& find_this)
  17     : find_this_(find_this) {
  18   // usearch_open requires a valid string argument to be searched, even if we
  19   // want to set it by usearch_setText afterwards. So, supplying a dummy text.
  20   const string16& dummy = find_this_;
  21
  22   UErrorCode status = U_ZERO_ERROR;
  23   search_ = usearch_open(find_this_.data(), find_this_.size(), dummy.data(),
  24                          dummy.size(), uloc_getDefault(),
  25                          nullptr,  // breakiter
  26                          &status);
  27   if (U_SUCCESS(status)) {
  28     UCollator* collator = usearch_getCollator(search_);
  29     ucol_setStrength(collator, UCOL_PRIMARY);
  30     usearch_reset(search_);
  31   }
  32 }
  33
  34 FixedPatternStringSearchIgnoringCaseAndAccents::
  35 ~FixedPatternStringSearchIgnoringCaseAndAccents() {
  36   if (search_)
  37     usearch_close(search_);
  38 }
  39
  40 bool FixedPatternStringSearchIgnoringCaseAndAccents::Search(
  41     const string16& in_this, size_t* match_index, size_t* match_length) {
  42   UErrorCode status = U_ZERO_ERROR;
  43   usearch_setText(search_, in_this.data(), in_this.size(), &status);
  44
  45   // Default to basic substring search if usearch fails. According to
  46   // http://icu-project.org/apiref/icu4c/usearch_8h.html, usearch_open will fail
  47   // if either |find_this| or |in_this| are empty. In either case basic
  48   // substring search will give the correct return value.
  49   if (!U_SUCCESS(status)) {
  50     size_t index = in_this.find(find_this_);
  51     if (index == string16::npos)
  52       return false;
  53     if (match_index)
  54       *match_index = index;
  55     if (match_length)
  56       *match_length = find_this_.size();
  57     return true;
  58   }
  59
  60   int32_t index = usearch_first(search_, &status);
  61   if (!U_SUCCESS(status) || index == USEARCH_DONE)
  62     return false;
  63   if (match_index)
  64     *match_index = static_cast<size_t>(index);
  65   if (match_length)
  66     *match_length = static_cast<size_t>(usearch_getMatchedLength(search_));
  67   return true;
  68 }
  69
  70 bool StringSearchIgnoringCaseAndAccents(const string16& find_this,
  71                                         const string16& in_this,
  72                                         size_t* match_index,
  73                                         size_t* match_length) {
  74   return FixedPatternStringSearchIgnoringCaseAndAccents(find_this).Search(
  75       in_this, match_index, match_length);
  76 }
  77
  78 }  // namespace i18n
  79 }  // namespace base