1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73
|
// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROMEOS_ASH_COMPONENTS_MANTIS_MEDIA_APP_MANTIS_UNTRUSTED_SERVICE_H_
#define CHROMEOS_ASH_COMPONENTS_MANTIS_MEDIA_APP_MANTIS_UNTRUSTED_SERVICE_H_
#include "ash/webui/media_app_ui/media_app_ui_untrusted.mojom.h"
#include "base/component_export.h"
#include "base/functional/callback_forward.h"
#include "base/sequence_checker.h"
#include "chromeos/ash/components/mantis/mojom/mantis_processor.mojom.h"
#include "mojo/public/cpp/bindings/pending_receiver.h"
#include "mojo/public/cpp/bindings/pending_remote.h"
#include "mojo/public/cpp/bindings/receiver.h"
#include "mojo/public/cpp/bindings/remote.h"
namespace ash {
// A Mojo connection between Mantis processor in CrOS and Mantis untrusted
// service in media app. Its lifetime is bound to the Mojo connection and
// inherently the image file opened in the media app, i.e. it gets destructed
// when the media app window opens a new file, or the media app window closes or
// navigates.
class COMPONENT_EXPORT(CHROMEOS_ASH_COMPONENTS_MANTIS_MEDIA_APP)
MantisUntrustedService
: public media_app_ui::mojom::MantisUntrustedService {
public:
explicit MantisUntrustedService(
mojo::PendingRemote<mantis::mojom::MantisProcessor> processor);
MantisUntrustedService(const MantisUntrustedService&) = delete;
MantisUntrustedService& operator=(const MantisUntrustedService&) = delete;
~MantisUntrustedService() override;
mojo::PendingRemote<media_app_ui::mojom::MantisUntrustedService>
BindNewPipeAndPassRemote(base::OnceClosure disconnect_handler);
// Implements `media_app_ui::mojom::MantisUntrustedService`:
void SegmentImage(const std::vector<uint8_t>& image,
const std::vector<uint8_t>& selection,
SegmentImageCallback callback) override;
void GenerativeFillImage(const std::vector<uint8_t>& image,
const std::vector<uint8_t>& mask,
const std::string& text,
uint32_t seed,
GenerativeFillImageCallback callback) override;
void InpaintImage(const std::vector<uint8_t>& image,
const std::vector<uint8_t>& mask,
uint32_t seed,
InpaintImageCallback callback) override;
void ClassifyImageSafety(const std::vector<uint8_t>& image,
ClassifyImageSafetyCallback callback) override;
void OutpaintImage(const std::vector<uint8_t>& image,
const std::vector<uint8_t>& mask,
uint32_t seed,
OutpaintImageCallback callback) override;
void InferSegmentationMode(std::vector<mantis::mojom::TouchPointPtr> gesture,
InferSegmentationModeCallback callback) override;
private:
mojo::Receiver<media_app_ui::mojom::MantisUntrustedService> receiver_;
mojo::Remote<mantis::mojom::MantisProcessor> processor_;
SEQUENCE_CHECKER(sequence_checker_);
};
} // namespace ash
#endif // CHROMEOS_ASH_COMPONENTS_MANTIS_MEDIA_APP_MANTIS_UNTRUSTED_SERVICE_H_
|