blob: d908736cffe31e2acb1bfa787bc12b05265543b7 [file] [log] [blame]
Simon Glass39f76112014-02-26 15:59:23 -07001/*
2 * Copyright (c) 2013 Google, Inc
3 *
4 * SPDX-License-Identifier: GPL-2.0+
5 */
6
7#include <common.h>
8#include <dm.h>
9#include <errno.h>
10#include <fdtdec.h>
11#include <malloc.h>
12#include <dm-demo.h>
13#include <asm/io.h>
Simon Glassa02af4a2015-01-05 20:05:31 -070014#include <asm/gpio.h>
Simon Glass39f76112014-02-26 15:59:23 -070015
16DECLARE_GLOBAL_DATA_PTR;
17
18/* Shape size */
19#define WIDTH 8
20#define HEIGHT 6
21
22struct shape_data {
23 int num_chars; /* Number of non-space characters output so far */
Simon Glassa02af4a2015-01-05 20:05:31 -070024 struct gpio_desc gpio_desc[8];
25 int gpio_count;
Simon Glass39f76112014-02-26 15:59:23 -070026};
27
28/* Crazy little function to draw shapes on the console */
Heiko Schocher54c5d082014-05-22 12:43:05 +020029static int shape_hello(struct udevice *dev, int ch)
Simon Glass39f76112014-02-26 15:59:23 -070030{
31 const struct dm_demo_pdata *pdata = dev_get_platdata(dev);
32 struct shape_data *data = dev_get_priv(dev);
33 static const struct shape {
34 int start;
35 int end;
36 int dstart;
37 int dend;
38 } shapes[3] = {
39 { 0, 1, 0, 1 },
40 { 0, WIDTH, 0, 0 },
41 { HEIGHT / 2 - 1, WIDTH - HEIGHT / 2 + 1, -1, 1},
42 };
43 struct shape shape;
44 unsigned int index;
45 int line, pos, inside;
46 const char *colour = pdata->colour;
47 int first = 0;
48
49 if (!ch)
50 ch = pdata->default_char;
51 if (!ch)
52 ch = '@';
53
54 index = (pdata->sides / 2) - 1;
55 if (index >= ARRAY_SIZE(shapes))
56 return -EIO;
57 shape = shapes[index];
58
59 for (line = 0; line < HEIGHT; line++) {
60 first = 1;
61 for (pos = 0; pos < WIDTH; pos++) {
62 inside = pos >= shape.start && pos < shape.end;
63 if (inside) {
64 putc(first ? *colour++ : ch);
65 data->num_chars++;
66 first = 0;
67 if (!*colour)
68 colour = pdata->colour;
69 } else {
70 putc(' ');
71 }
72 }
73 putc('\n');
74 shape.start += shape.dstart;
75 shape.end += shape.dend;
76 if (shape.start < 0) {
77 shape.dstart = -shape.dstart;
78 shape.dend = -shape.dend;
79 shape.start += shape.dstart;
80 shape.end += shape.dend;
81 }
82 }
83
84 return 0;
85}
86
Heiko Schocher54c5d082014-05-22 12:43:05 +020087static int shape_status(struct udevice *dev, int *status)
Simon Glass39f76112014-02-26 15:59:23 -070088{
89 struct shape_data *data = dev_get_priv(dev);
90
91 *status = data->num_chars;
92 return 0;
93}
94
Simon Glassa02af4a2015-01-05 20:05:31 -070095static int set_light(struct udevice *dev, int light)
96{
97 struct shape_data *priv = dev_get_priv(dev);
98 struct gpio_desc *desc;
99 int ret;
100 int i;
101
102 desc = priv->gpio_desc;
103 for (i = 0; i < priv->gpio_count; i++, desc++) {
104 uint mask = 1 << i;
105
106 ret = dm_gpio_set_value(desc, light & mask);
107 if (ret < 0)
108 return ret;
109 }
110
111 return 0;
112}
113
114static int get_light(struct udevice *dev)
115{
116 struct shape_data *priv = dev_get_priv(dev);
117 struct gpio_desc *desc;
118 uint value = 0;
119 int ret;
120 int i;
121
122 desc = priv->gpio_desc;
123 for (i = 0; i < priv->gpio_count; i++, desc++) {
124 uint mask = 1 << i;
125
126 ret = dm_gpio_get_value(desc);
127 if (ret < 0)
128 return ret;
129 if (ret)
130 value |= mask;
131 }
132
133 return value;
134}
135
Simon Glass39f76112014-02-26 15:59:23 -0700136static const struct demo_ops shape_ops = {
137 .hello = shape_hello,
138 .status = shape_status,
Simon Glassa02af4a2015-01-05 20:05:31 -0700139 .get_light = get_light,
140 .set_light = set_light,
Simon Glass39f76112014-02-26 15:59:23 -0700141};
142
Heiko Schocher54c5d082014-05-22 12:43:05 +0200143static int shape_ofdata_to_platdata(struct udevice *dev)
Simon Glass39f76112014-02-26 15:59:23 -0700144{
145 struct dm_demo_pdata *pdata = dev_get_platdata(dev);
146 int ret;
147
148 /* Parse the data that is common with all demo devices */
149 ret = demo_parse_dt(dev);
150 if (ret)
151 return ret;
152
153 /* Parse the data that only we need */
154 pdata->default_char = fdtdec_get_int(gd->fdt_blob, dev->of_offset,
155 "character", '@');
156
157 return 0;
158}
159
Simon Glassa02af4a2015-01-05 20:05:31 -0700160static int dm_shape_probe(struct udevice *dev)
161{
162 struct shape_data *priv = dev_get_priv(dev);
163 int ret;
164
165 ret = gpio_request_list_by_name(dev, "light-gpios", priv->gpio_desc,
166 ARRAY_SIZE(priv->gpio_desc),
167 GPIOD_IS_OUT | GPIOD_IS_OUT_ACTIVE);
168 if (ret < 0)
169 return ret;
170 priv->gpio_count = ret;
171 debug("%s: %d GPIOs\n", __func__, priv->gpio_count);
172
173 return 0;
174}
175
176static int dm_shape_remove(struct udevice *dev)
177{
178 struct shape_data *priv = dev_get_priv(dev);
179
180 return gpio_free_list(dev, priv->gpio_desc, priv->gpio_count);
181}
182
Simon Glassae7f4512014-06-11 23:29:45 -0600183static const struct udevice_id demo_shape_id[] = {
Simon Glass39f76112014-02-26 15:59:23 -0700184 { "demo-shape", 0 },
185 { },
186};
187
188U_BOOT_DRIVER(demo_shape_drv) = {
189 .name = "demo_shape_drv",
190 .of_match = demo_shape_id,
191 .id = UCLASS_DEMO,
192 .ofdata_to_platdata = shape_ofdata_to_platdata,
193 .ops = &shape_ops,
Simon Glassa02af4a2015-01-05 20:05:31 -0700194 .probe = dm_shape_probe,
195 .remove = dm_shape_remove,
Simon Glass39f76112014-02-26 15:59:23 -0700196 .priv_auto_alloc_size = sizeof(struct shape_data),
197 .platdata_auto_alloc_size = sizeof(struct dm_demo_pdata),
198};