| /* |
| * Copyright (C) 2017 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #ifndef LIBTEXTCLASSIFIER_COMMON_MEMORY_IMAGE_DATA_STORE_H_ |
| #define LIBTEXTCLASSIFIER_COMMON_MEMORY_IMAGE_DATA_STORE_H_ |
| |
| #include <string> |
| |
| #include "common/memory_image/data-store.pb.h" |
| #include "common/memory_image/memory-image-reader.h" |
| #include "util/strings/stringpiece.h" |
| |
| namespace libtextclassifier { |
| namespace nlp_core { |
| namespace memory_image { |
| |
| // Class to access a data store. See usage example in comments for |
| // DataStoreBuilder. |
| class DataStore { |
| public: |
| // Constructs a DataStore using the indicated bytes, i.e., bytes.size() bytes |
| // starting at address bytes.data(). These bytes should contain the |
| // serialization of a data store, see DataStoreBuilder::SerializeAsString(). |
| explicit DataStore(StringPiece bytes); |
| |
| // Retrieves (start_addr, num_bytes) info for piece of memory that contains |
| // the data associated with the indicated name. Note: this piece of memory is |
| // inside the [start, start + size) (see constructor). This piece of memory |
| // starts at an offset from start which is a multiple of the alignment |
| // specified when the data store was built using DataStoreBuilder. |
| // |
| // If the alignment is a low power of 2 (e..g, 4, 8, or 16) and "start" passed |
| // to constructor corresponds to the beginning of a memory page or an address |
| // returned by new or malloc(), then start_addr is divisible with alignment. |
| StringPiece GetData(const std::string &name) const; |
| |
| private: |
| MemoryImageReader<DataStoreProto> reader_; |
| }; |
| |
| } // namespace memory_image |
| } // namespace nlp_core |
| } // namespace libtextclassifier |
| |
| #endif // LIBTEXTCLASSIFIER_COMMON_MEMORY_IMAGE_DATA_STORE_H_ |