| // Copyright 2018 The Chromium OS Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef ML_MACHINE_LEARNING_SERVICE_IMPL_H_ |
| #define ML_MACHINE_LEARNING_SERVICE_IMPL_H_ |
| |
| #include <map> |
| #include <memory> |
| #include <string> |
| |
| #include <base/callback_forward.h> |
| #include <base/macros.h> |
| #include <mojo/public/cpp/bindings/pending_receiver.h> |
| #include <mojo/public/cpp/bindings/receiver.h> |
| #include <mojo/public/cpp/bindings/receiver_set.h> |
| |
| #include "ml/model_metadata.h" |
| #include "ml/mojom/machine_learning_service.mojom.h" |
| |
| namespace ml { |
| |
| class MachineLearningServiceImpl |
| : public chromeos::machine_learning::mojom::MachineLearningService { |
| public: |
| // Creates an instance bound to `pipe`. The specified `disconnect_handler` |
| // will be invoked if the binding encounters a connection error or is closed. |
| MachineLearningServiceImpl(mojo::ScopedMessagePipeHandle pipe, |
| base::Closure disconnect_handler); |
| |
| // A interface to change `text_classifier_model_filename_` for testing. Should |
| // not be used outside of tests. |
| void SetTextClassifierModelFilenameForTesting(const std::string& filename); |
| |
| protected: |
| // Testing constructor that allows overriding of the model dir. Should not be |
| // used outside of tests. |
| MachineLearningServiceImpl(mojo::ScopedMessagePipeHandle pipe, |
| base::Closure disconnect_handler, |
| const std::string& model_dir); |
| |
| private: |
| // chromeos::machine_learning::mojom::MachineLearningService: |
| void Clone(mojo::PendingReceiver< |
| chromeos::machine_learning::mojom::MachineLearningService> |
| receiver) override; |
| void LoadBuiltinModel( |
| chromeos::machine_learning::mojom::BuiltinModelSpecPtr spec, |
| mojo::PendingReceiver<chromeos::machine_learning::mojom::Model> receiver, |
| LoadBuiltinModelCallback callback) override; |
| void LoadFlatBufferModel( |
| chromeos::machine_learning::mojom::FlatBufferModelSpecPtr spec, |
| mojo::PendingReceiver<chromeos::machine_learning::mojom::Model> receiver, |
| LoadFlatBufferModelCallback callback) override; |
| void LoadTextClassifier( |
| mojo::PendingReceiver<chromeos::machine_learning::mojom::TextClassifier> |
| receiver, |
| LoadTextClassifierCallback callback) override; |
| void LoadHandwritingModel( |
| mojo::PendingReceiver< |
| chromeos::machine_learning::mojom::HandwritingRecognizer> receiver, |
| LoadHandwritingModelCallback callback) override; |
| void LoadHandwritingModelWithSpec( |
| chromeos::machine_learning::mojom::HandwritingRecognizerSpecPtr spec, |
| mojo::PendingReceiver< |
| chromeos::machine_learning::mojom::HandwritingRecognizer> receiver, |
| LoadHandwritingModelCallback callback) override; |
| |
| // Init the icu data if it is not initialized yet. |
| void InitIcuIfNeeded(); |
| |
| // Used to hold the icu data read from file. |
| char* icu_data_; |
| |
| std::string text_classifier_model_filename_; |
| |
| // Metadata required to load builtin models. Initialized at construction. |
| const std::map<chromeos::machine_learning::mojom::BuiltinModelId, |
| BuiltinModelMetadata> |
| builtin_model_metadata_; |
| |
| const std::string model_dir_; |
| |
| // Primordial receiver bootstrapped over D-Bus. Once opened, is never closed. |
| mojo::Receiver<chromeos::machine_learning::mojom::MachineLearningService> |
| receiver_; |
| |
| // Additional receivers bound via `Clone`. |
| mojo::ReceiverSet<chromeos::machine_learning::mojom::MachineLearningService> |
| clone_receivers_; |
| |
| DISALLOW_COPY_AND_ASSIGN(MachineLearningServiceImpl); |
| }; |
| |
| } // namespace ml |
| |
| #endif // ML_MACHINE_LEARNING_SERVICE_IMPL_H_ |