Coverage Report

Created: 2025-07-01 06:18

/src/WasmEdge/include/host/mock/wasmedge_stablediffusion_func.h
Line
Count
Source (jump to first uncovered line)
1
#pragma once
2
3
#include "common/errcode.h"
4
#include "host/mock/log.h"
5
#include "runtime/callingframe.h"
6
#include "runtime/hostfunc.h"
7
8
namespace WasmEdge {
9
namespace Host {
10
namespace WasmEdgeStableDiffusionMock {
11
12
using namespace std::literals;
13
static inline constexpr const uint32_t kStableDiffusionError = 1U;
14
15
class CreateContext : public Runtime::HostFunction<CreateContext> {
16
public:
17
  Expect<uint32_t> body(const Runtime::CallingFrame &, uint32_t, uint32_t,
18
                        uint32_t, uint32_t, uint32_t, uint32_t, uint32_t,
19
                        uint32_t, uint32_t, uint32_t, uint32_t, uint32_t,
20
                        uint32_t, uint32_t, uint32_t, uint32_t, uint32_t,
21
                        uint32_t, uint32_t, uint32_t, uint32_t, uint32_t,
22
                        uint32_t, uint32_t, uint32_t, uint32_t, uint32_t,
23
                        uint32_t, uint32_t, uint32_t, uint32_t, uint32_t,
24
0
                        uint32_t) {
25
0
    printPluginMock("WasmEdge-Stable-Diffusion"sv);
26
0
    return kStableDiffusionError;
27
0
  }
28
};
29
30
class TextToImage : public Runtime::HostFunction<TextToImage> {
31
public:
32
  Expect<uint32_t> body(const Runtime::CallingFrame &, uint32_t, uint32_t,
33
                        uint32_t, uint32_t, uint32_t, uint32_t, uint32_t, float,
34
                        uint32_t, uint32_t, int32_t, float, uint32_t, uint32_t,
35
                        uint32_t, uint32_t, float, float, uint32_t, uint32_t,
36
                        uint32_t, uint32_t, uint32_t, uint32_t, uint32_t,
37
                        uint32_t, uint32_t, float, float, float, uint32_t,
38
0
                        uint32_t, uint32_t, uint32_t, uint32_t) {
39
0
    printPluginMock("WasmEdge-Stable-Diffusion"sv);
40
0
    return kStableDiffusionError;
41
0
  }
42
};
43
class ImageToImage : public Runtime::HostFunction<ImageToImage> {
44
public:
45
  Expect<uint32_t> body(const Runtime::CallingFrame &, uint32_t, uint32_t,
46
                        uint32_t, uint32_t, uint32_t, float, uint32_t, uint32_t,
47
                        uint32_t, uint32_t, uint32_t, uint32_t, uint32_t,
48
                        uint32_t, int32_t, float, uint32_t, uint32_t, float,
49
                        uint32_t, uint32_t, float, float, uint32_t, uint32_t,
50
                        uint32_t, uint32_t, uint32_t, uint32_t, uint32_t,
51
                        uint32_t, uint32_t, float, float, float, uint32_t,
52
0
                        uint32_t, uint32_t, uint32_t, uint32_t) {
53
0
    printPluginMock("WasmEdge-Stable-Diffusion"sv);
54
0
    return kStableDiffusionError;
55
0
  }
56
};
57
class Convert : public Runtime::HostFunction<Convert> {
58
public:
59
  Expect<uint32_t> body(const Runtime::CallingFrame &, uint32_t, uint32_t,
60
0
                        uint32_t, uint32_t, uint32_t, uint32_t, uint32_t) {
61
0
    printPluginMock("WasmEdge-Stable-Diffusion"sv);
62
0
    return kStableDiffusionError;
63
0
  }
64
};
65
} // namespace WasmEdgeStableDiffusionMock
66
} // namespace Host
67
} // namespace WasmEdge