1 /*
2 * Copyright (c) 2024 Ambiq Micro Inc. <www.ambiq.com>
3 * SPDX-License-Identifier: Apache-2.0
4 */
5
6 #include <zephyr/device.h>
7 #include <zephyr/drivers/gpio.h>
8 #include <zephyr/drivers/mspi.h>
9 #include <zephyr/drivers/mspi_emul.h>
10 #include <zephyr/ztest.h>
11
12 #define TEST_MSPI_REINIT 1
13
14 /* add else if for other SoC platforms */
15 #if defined(CONFIG_SOC_POSIX)
16 typedef struct mspi_timing_cfg mspi_timing_cfg;
17 typedef enum mspi_timing_param mspi_timing_param;
18 #elif defined(CONFIG_SOC_FAMILY_AMBIQ)
19 #include "mspi_ambiq.h"
20 typedef struct mspi_ambiq_timing_cfg mspi_timing_cfg;
21 typedef enum mspi_ambiq_timing_param mspi_timing_param;
22 #endif
23
24 #define MSPI_BUS_NODE DT_ALIAS(mspi0)
25
26 static const struct device *mspi_devices[] = {
27 DT_FOREACH_CHILD_STATUS_OKAY_SEP(MSPI_BUS_NODE, DEVICE_DT_GET, (,))
28 };
29
30 static struct gpio_dt_spec ce_gpios[] = MSPI_CE_GPIOS_DT_SPEC_GET(MSPI_BUS_NODE);
31
32 #if TEST_MSPI_REINIT
33 struct mspi_cfg hardware_cfg = {
34 .channel_num = 0,
35 .op_mode = DT_ENUM_IDX_OR(MSPI_BUS_NODE, op_mode, MSPI_OP_MODE_CONTROLLER),
36 .duplex = DT_ENUM_IDX_OR(MSPI_BUS_NODE, duplex, MSPI_HALF_DUPLEX),
37 .dqs_support = DT_PROP_OR(MSPI_BUS_NODE, dqs_support, false),
38 .ce_group = ce_gpios,
39 .num_ce_gpios = ARRAY_SIZE(ce_gpios),
40 .num_periph = DT_CHILD_NUM(MSPI_BUS_NODE),
41 .max_freq = DT_PROP(MSPI_BUS_NODE, clock_frequency),
42 .re_init = true,
43 };
44 #endif
45
46
47 static struct mspi_dev_id dev_id[] = {
48 DT_FOREACH_CHILD_STATUS_OKAY_SEP(MSPI_BUS_NODE, MSPI_DEVICE_ID_DT, (,))
49 };
50
51 static struct mspi_dev_cfg device_cfg[] = {
52 DT_FOREACH_CHILD_STATUS_OKAY_SEP(MSPI_BUS_NODE, MSPI_DEVICE_CONFIG_DT, (,))
53 };
54
55 static struct mspi_xip_cfg xip_cfg[] = {
56 DT_FOREACH_CHILD_STATUS_OKAY_SEP(MSPI_BUS_NODE, MSPI_XIP_CONFIG_DT, (,))
57 };
58
59 static struct mspi_scramble_cfg scramble_cfg[] = {
60 DT_FOREACH_CHILD_STATUS_OKAY_SEP(MSPI_BUS_NODE, MSPI_SCRAMBLE_CONFIG_DT, (,))
61 };
62
ZTEST(mspi_api,test_mspi_api)63 ZTEST(mspi_api, test_mspi_api)
64 {
65 int ret = 0;
66 const struct device *mspi_bus = DEVICE_DT_GET(MSPI_BUS_NODE);
67
68 zassert_true(device_is_ready(mspi_bus), "mspi_bus is not ready");
69
70 #if TEST_MSPI_REINIT
71 const struct mspi_dt_spec spec = {
72 .bus = mspi_bus,
73 .config = hardware_cfg,
74 };
75
76 ret = mspi_config(&spec);
77 zassert_equal(ret, 0, "mspi_config failed.");
78 #endif
79
80 for (int dev_idx = 0; dev_idx < ARRAY_SIZE(mspi_devices); ++dev_idx) {
81
82 zassert_true(device_is_ready(mspi_devices[dev_idx]), "mspi_device is not ready");
83
84 ret = mspi_dev_config(mspi_bus, &dev_id[dev_idx],
85 MSPI_DEVICE_CONFIG_ALL, &device_cfg[dev_idx]);
86 zassert_equal(ret, 0, "mspi_dev_config failed.");
87
88 #if CONFIG_MSPI_XIP
89 ret = mspi_xip_config(mspi_bus, &dev_id[dev_idx], &xip_cfg[dev_idx]);
90 zassert_equal(ret, 0, "mspi_xip_config failed.");
91 #endif
92
93 #if CONFIG_MSPI_SCRAMBLE
94 ret = mspi_scramble_config(mspi_bus, &dev_id[dev_idx], &scramble_cfg[dev_idx]);
95 zassert_equal(ret, 0, "mspi_scramble_config failed.");
96 #endif
97
98 #if CONFIG_MSPI_TIMING
99 mspi_timing_cfg timing_cfg;
100 mspi_timing_param timing_cfg_mask = 0;
101
102 ret = mspi_timing_config(mspi_bus, &dev_id[dev_idx], timing_cfg_mask, &timing_cfg);
103 zassert_equal(ret, 0, "mspi_timing_config failed.");
104 #endif
105
106 ret = mspi_register_callback(mspi_bus, &dev_id[dev_idx],
107 MSPI_BUS_XFER_COMPLETE, NULL, NULL);
108 zassert_equal(ret, 0, "mspi_register_callback failed.");
109
110 ret = mspi_get_channel_status(mspi_bus, 0);
111 zassert_equal(ret, 0, "mspi_get_channel_status failed.");
112
113 }
114 }
115
116 ZTEST_SUITE(mspi_api, NULL, NULL, NULL, NULL, NULL);
117
