xref: /aosp_15_r20/external/libtextclassifier/native/utils/tokenfree/byte_encoder.cc (revision 993b0882672172b81d12fad7a7ac0c3e5c824a12)
1*993b0882SAndroid Build Coastguard Worker /*
2*993b0882SAndroid Build Coastguard Worker  * Copyright (C) 2018 The Android Open Source Project
3*993b0882SAndroid Build Coastguard Worker  *
4*993b0882SAndroid Build Coastguard Worker  * Licensed under the Apache License, Version 2.0 (the "License");
5*993b0882SAndroid Build Coastguard Worker  * you may not use this file except in compliance with the License.
6*993b0882SAndroid Build Coastguard Worker  * You may obtain a copy of the License at
7*993b0882SAndroid Build Coastguard Worker  *
8*993b0882SAndroid Build Coastguard Worker  *      http://www.apache.org/licenses/LICENSE-2.0
9*993b0882SAndroid Build Coastguard Worker  *
10*993b0882SAndroid Build Coastguard Worker  * Unless required by applicable law or agreed to in writing, software
11*993b0882SAndroid Build Coastguard Worker  * distributed under the License is distributed on an "AS IS" BASIS,
12*993b0882SAndroid Build Coastguard Worker  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13*993b0882SAndroid Build Coastguard Worker  * See the License for the specific language governing permissions and
14*993b0882SAndroid Build Coastguard Worker  * limitations under the License.
15*993b0882SAndroid Build Coastguard Worker  */
16*993b0882SAndroid Build Coastguard Worker 
17*993b0882SAndroid Build Coastguard Worker #include "utils/tokenfree/byte_encoder.h"
18*993b0882SAndroid Build Coastguard Worker 
19*993b0882SAndroid Build Coastguard Worker #include <vector>
20*993b0882SAndroid Build Coastguard Worker namespace libtextclassifier3 {
21*993b0882SAndroid Build Coastguard Worker 
Encode(StringPiece input_text,std::vector<int64_t> * encoded_text) const22*993b0882SAndroid Build Coastguard Worker bool ByteEncoder::Encode(StringPiece input_text,
23*993b0882SAndroid Build Coastguard Worker                          std::vector<int64_t>* encoded_text) const {
24*993b0882SAndroid Build Coastguard Worker   const int len = input_text.size();
25*993b0882SAndroid Build Coastguard Worker   if (len <= 0) {
26*993b0882SAndroid Build Coastguard Worker     *encoded_text = {};
27*993b0882SAndroid Build Coastguard Worker     return true;
28*993b0882SAndroid Build Coastguard Worker   }
29*993b0882SAndroid Build Coastguard Worker 
30*993b0882SAndroid Build Coastguard Worker   int size = input_text.size();
31*993b0882SAndroid Build Coastguard Worker   encoded_text->resize(size);
32*993b0882SAndroid Build Coastguard Worker 
33*993b0882SAndroid Build Coastguard Worker   const auto& text = input_text.ToString();
34*993b0882SAndroid Build Coastguard Worker   for (int i = 0; i < size; i++) {
35*993b0882SAndroid Build Coastguard Worker     int64_t encoding = static_cast<int64_t>(text[i]);
36*993b0882SAndroid Build Coastguard Worker     (*encoded_text)[i] = encoding;
37*993b0882SAndroid Build Coastguard Worker   }
38*993b0882SAndroid Build Coastguard Worker 
39*993b0882SAndroid Build Coastguard Worker   return true;
40*993b0882SAndroid Build Coastguard Worker }
41*993b0882SAndroid Build Coastguard Worker 
42*993b0882SAndroid Build Coastguard Worker }  // namespace libtextclassifier3
43