]> sigrok.org Git - libsigrok.git/blob - src/hardware/atten-pps3xxx/api.c
drivers: Factor out std_gvar_min_max_step{,_array}().
[libsigrok.git] / src / hardware / atten-pps3xxx / api.c
1 /*
2  * This file is part of the libsigrok project.
3  *
4  * Copyright (C) 2014 Bert Vermeulen <bert@biot.com>
5  *
6  * This program is free software: you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation, either version 3 of the License, or
9  * (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
18  */
19
20 #include <config.h>
21 #include <string.h>
22 #include "protocol.h"
23
24 /*
25  * The default serial communication settings on the device are 9600
26  * baud, 9 data bits. The 9th bit isn't actually used, and the vendor
27  * software uses Mark parity to absorb the extra bit.
28  *
29  * Since 9 data bits is not a standard available in POSIX, we use two
30  * stop bits to skip over the extra bit instead.
31  */
32 #define SERIALCOMM "9600/8n2"
33
34 static const uint32_t scanopts[] = {
35         SR_CONF_CONN,
36         SR_CONF_SERIALCOMM,
37 };
38
39 static const uint32_t drvopts[] = {
40         SR_CONF_POWER_SUPPLY,
41 };
42
43 static const uint32_t devopts[] = {
44         SR_CONF_CONTINUOUS,
45         SR_CONF_CHANNEL_CONFIG | SR_CONF_GET | SR_CONF_SET | SR_CONF_LIST,
46         SR_CONF_OVER_CURRENT_PROTECTION_ENABLED | SR_CONF_GET | SR_CONF_SET,
47 };
48
49 static const uint32_t devopts_cg[] = {
50         SR_CONF_VOLTAGE | SR_CONF_GET,
51         SR_CONF_VOLTAGE_TARGET | SR_CONF_GET | SR_CONF_SET | SR_CONF_LIST,
52         SR_CONF_CURRENT | SR_CONF_GET,
53         SR_CONF_CURRENT_LIMIT | SR_CONF_GET | SR_CONF_SET | SR_CONF_LIST,
54         SR_CONF_ENABLED | SR_CONF_GET | SR_CONF_SET,
55 };
56
57 static const char *channel_modes[] = {
58         "Independent",
59         "Series",
60         "Parallel",
61 };
62
63 static const struct pps_model models[] = {
64         { PPS_3203T_3S, "PPS3203T-3S",
65                 CHANMODE_INDEPENDENT | CHANMODE_SERIES | CHANMODE_PARALLEL,
66                 3,
67                 {
68                         /* Channel 1 */
69                         { { 0, 32, 0.01 }, { 0, 3, 0.001 } },
70                         /* Channel 2 */
71                         { { 0, 32, 0.01 }, { 0, 3, 0.001 } },
72                         /* Channel 3 */
73                         { { 0, 6, 0.01 }, { 0, 3, 0.001 } },
74                 },
75         },
76 };
77
78 static GSList *scan(struct sr_dev_driver *di, GSList *options, int modelid)
79 {
80         struct sr_dev_inst *sdi;
81         struct dev_context *devc;
82         struct sr_config *src;
83         struct sr_channel *ch;
84         struct sr_channel_group *cg;
85         struct sr_serial_dev_inst *serial;
86         GSList *l;
87         const struct pps_model *model;
88         uint8_t packet[PACKET_SIZE];
89         unsigned int i;
90         int delay_ms, ret;
91         const char *conn, *serialcomm;
92         char channel[10];
93
94         conn = serialcomm = NULL;
95         for (l = options; l; l = l->next) {
96                 src = l->data;
97                 switch (src->key) {
98                 case SR_CONF_CONN:
99                         conn = g_variant_get_string(src->data, NULL);
100                         break;
101                 case SR_CONF_SERIALCOMM:
102                         serialcomm = g_variant_get_string(src->data, NULL);
103                         break;
104                 }
105         }
106         if (!conn)
107                 return NULL;
108         if (!serialcomm)
109                 serialcomm = SERIALCOMM;
110
111         serial = sr_serial_dev_inst_new(conn, serialcomm);
112
113         if (serial_open(serial, SERIAL_RDWR) != SR_OK)
114                 return NULL;
115
116         serial_flush(serial);
117
118         /* This is how the vendor software scans for hardware. */
119         memset(packet, 0, PACKET_SIZE);
120         packet[0] = 0xaa;
121         packet[1] = 0xaa;
122         delay_ms = serial_timeout(serial, PACKET_SIZE);
123         if (serial_write_blocking(serial, packet, PACKET_SIZE, delay_ms) < PACKET_SIZE) {
124                 sr_err("Unable to write while probing for hardware.");
125                 return NULL;
126         }
127         /* The device responds with a 24-byte packet when it receives a packet.
128          * At 9600 baud, 300ms is long enough for it to have arrived. */
129         g_usleep(300 * 1000);
130         memset(packet, 0, PACKET_SIZE);
131         if ((ret = serial_read_nonblocking(serial, packet, PACKET_SIZE)) < 0) {
132                 sr_err("Unable to read while probing for hardware: %s",
133                                 sr_strerror(ret));
134                 return NULL;
135         }
136         if (ret != PACKET_SIZE || packet[0] != 0xaa || packet[1] != 0xaa) {
137                 /* Doesn't look like an Atten PPS. */
138                 return NULL;
139         }
140
141         model = NULL;
142         for (i = 0; i < ARRAY_SIZE(models); i++) {
143                 if (models[i].modelid == modelid) {
144                         model = &models[i];
145                         break;
146                 }
147         }
148         if (!model) {
149                 sr_err("Unknown modelid %d", modelid);
150                 return NULL;
151         }
152
153         sdi = g_malloc0(sizeof(struct sr_dev_inst));
154         sdi->status = SR_ST_INACTIVE;
155         sdi->vendor = g_strdup("Atten");
156         sdi->model = g_strdup(model->name);
157         sdi->inst_type = SR_INST_SERIAL;
158         sdi->conn = serial;
159         for (i = 0; i < MAX_CHANNELS; i++) {
160                 snprintf(channel, 10, "CH%d", i + 1);
161                 ch = sr_channel_new(sdi, i, SR_CHANNEL_ANALOG, TRUE, channel);
162                 cg = g_malloc(sizeof(struct sr_channel_group));
163                 cg->name = g_strdup(channel);
164                 cg->channels = g_slist_append(NULL, ch);
165                 cg->priv = NULL;
166                 sdi->channel_groups = g_slist_append(sdi->channel_groups, cg);
167         }
168
169         devc = g_malloc0(sizeof(struct dev_context));
170         devc->model = model;
171         devc->config = g_malloc0(sizeof(struct per_channel_config) * model->num_channels);
172         devc->delay_ms = delay_ms;
173         sdi->priv = devc;
174
175         serial_close(serial);
176
177         return std_scan_complete(di, g_slist_append(NULL, sdi));
178 }
179
180 static GSList *scan_3203(struct sr_dev_driver *di, GSList *options)
181 {
182         return scan(di, options, PPS_3203T_3S);
183 }
184
185 static int config_get(uint32_t key, GVariant **data, const struct sr_dev_inst *sdi,
186                 const struct sr_channel_group *cg)
187 {
188         struct dev_context *devc;
189         struct sr_channel *ch;
190         int channel;
191
192         if (!sdi)
193                 return SR_ERR_ARG;
194
195         devc = sdi->priv;
196
197         if (!cg) {
198                 switch (key) {
199                 case SR_CONF_CHANNEL_CONFIG:
200                         *data = g_variant_new_string(channel_modes[devc->channel_mode]);
201                         break;
202                 case SR_CONF_OVER_CURRENT_PROTECTION_ENABLED:
203                         *data = g_variant_new_boolean(devc->over_current_protection);
204                         break;
205                 default:
206                         return SR_ERR_NA;
207                 }
208         } else {
209                 /* We only ever have one channel per channel group in this driver. */
210                 ch = cg->channels->data;
211                 channel = ch->index;
212
213                 switch (key) {
214                 case SR_CONF_VOLTAGE:
215                         *data = g_variant_new_double(devc->config[channel].output_voltage_last);
216                         break;
217                 case SR_CONF_VOLTAGE_TARGET:
218                         *data = g_variant_new_double(devc->config[channel].output_voltage_max);
219                         break;
220                 case SR_CONF_CURRENT:
221                         *data = g_variant_new_double(devc->config[channel].output_current_last);
222                         break;
223                 case SR_CONF_CURRENT_LIMIT:
224                         *data = g_variant_new_double(devc->config[channel].output_current_max);
225                         break;
226                 case SR_CONF_ENABLED:
227                         *data = g_variant_new_boolean(devc->config[channel].output_enabled);
228                         break;
229                 default:
230                         return SR_ERR_NA;
231                 }
232         }
233
234         return SR_OK;
235 }
236
237 static int find_str(const char *str, const char **strings, int array_size)
238 {
239         int idx, i;
240
241         idx = -1;
242         for (i = 0; i < array_size; i++) {
243                 if (!strcmp(str, strings[i])) {
244                         idx = i;
245                         break;
246                 }
247         }
248
249         return idx;
250 }
251
252 static int config_set(uint32_t key, GVariant *data, const struct sr_dev_inst *sdi,
253                 const struct sr_channel_group *cg)
254 {
255         struct dev_context *devc;
256         struct sr_channel *ch;
257         gdouble dval;
258         int channel, ret, ival;
259         const char *sval;
260         gboolean bval;
261
262         ret = SR_OK;
263         devc = sdi->priv;
264         if (!cg) {
265                 switch (key) {
266                 case SR_CONF_CHANNEL_CONFIG:
267                         sval = g_variant_get_string(data, NULL);
268                         if ((ival = find_str(sval, channel_modes,
269                                                         ARRAY_SIZE(channel_modes))) == -1) {
270                                 ret = SR_ERR_ARG;
271                                 break;
272                         }
273                         if (devc->model->channel_modes && (1 << ival) == 0) {
274                                 /* Not supported on this model. */
275                                 ret = SR_ERR_ARG;
276                         }
277                         if (ival == devc->channel_mode_set)
278                                 /* Nothing to do. */
279                                 break;
280                         devc->channel_mode_set = ival;
281                         devc->config_dirty = TRUE;
282                         break;
283                 case SR_CONF_OVER_CURRENT_PROTECTION_ENABLED:
284                         bval = g_variant_get_boolean(data);
285                         if (bval == devc->over_current_protection_set)
286                                 /* Nothing to do. */
287                                 break;
288                         devc->over_current_protection_set = bval;
289                         devc->config_dirty = TRUE;
290                         break;
291                 default:
292                         return SR_ERR_NA;
293                 }
294         } else {
295                 /* We only ever have one channel per channel group in this driver. */
296                 ch = cg->channels->data;
297                 channel = ch->index;
298
299                 switch (key) {
300                 case SR_CONF_VOLTAGE_TARGET:
301                         dval = g_variant_get_double(data);
302                         if (dval < 0 || dval > devc->model->channels[channel].voltage[1])
303                                 ret = SR_ERR_ARG;
304                         devc->config[channel].output_voltage_max = dval;
305                         devc->config_dirty = TRUE;
306                         break;
307                 case SR_CONF_CURRENT_LIMIT:
308                         dval = g_variant_get_double(data);
309                         if (dval < 0 || dval > devc->model->channels[channel].current[1])
310                                 ret = SR_ERR_ARG;
311                         devc->config[channel].output_current_max = dval;
312                         devc->config_dirty = TRUE;
313                         break;
314                 case SR_CONF_ENABLED:
315                         bval = g_variant_get_boolean(data);
316                         if (bval == devc->config[channel].output_enabled_set)
317                                 /* Nothing to do. */
318                                 break;
319                         devc->config[channel].output_enabled_set = bval;
320                         devc->config_dirty = TRUE;
321                         break;
322                 default:
323                         ret = SR_ERR_NA;
324                 }
325         }
326
327         return ret;
328 }
329
330 static int config_list(uint32_t key, GVariant **data, const struct sr_dev_inst *sdi,
331                 const struct sr_channel_group *cg)
332 {
333         struct dev_context *devc;
334         struct sr_channel *ch;
335         int channel;
336
337         devc = (sdi) ? sdi->priv : NULL;
338
339         if (!cg) {
340                 switch (key) {
341                 case SR_CONF_SCAN_OPTIONS:
342                 case SR_CONF_DEVICE_OPTIONS:
343                         return STD_CONFIG_LIST(key, data, sdi, cg, scanopts, drvopts, devopts);
344                 case SR_CONF_CHANNEL_CONFIG:
345                         if (devc->model->channel_modes == CHANMODE_INDEPENDENT) {
346                                 /* The 1-channel models. */
347                                 *data = g_variant_new_strv(channel_modes, 1);
348                         } else {
349                                 /* The other models support all modes. */
350                                 *data = g_variant_new_strv(channel_modes, ARRAY_SIZE(channel_modes));
351                         }
352                         break;
353                 default:
354                         return SR_ERR_NA;
355                 }
356         } else {
357                 /* We only ever have one channel per channel group in this driver. */
358                 ch = cg->channels->data;
359                 channel = ch->index;
360
361                 switch (key) {
362                 case SR_CONF_DEVICE_OPTIONS:
363                         *data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
364                                         devopts_cg, ARRAY_SIZE(devopts_cg), sizeof(uint32_t));
365                         break;
366                 case SR_CONF_VOLTAGE_TARGET:
367                         *data = std_gvar_min_max_step_array(devc->model->channels[channel].voltage);
368                         break;
369                 case SR_CONF_CURRENT_LIMIT:
370                         *data = std_gvar_min_max_step_array(devc->model->channels[channel].current);
371                         break;
372                 default:
373                         return SR_ERR_NA;
374                 }
375         }
376
377         return SR_OK;
378 }
379
380 static int dev_close(struct sr_dev_inst *sdi)
381 {
382         struct dev_context *devc;
383
384         devc = sdi->priv;
385
386         if (devc->config_dirty)
387                 /* Some configuration changes were queued up but didn't
388                  * get sent to the device, likely because we were never
389                  * in acquisition mode. Send them out now. */
390                 send_config(sdi);
391
392         return std_serial_dev_close(sdi);
393 }
394
395 static int dev_acquisition_start(const struct sr_dev_inst *sdi)
396 {
397         struct dev_context *devc;
398         struct sr_serial_dev_inst *serial;
399         uint8_t packet[PACKET_SIZE];
400
401         devc = sdi->priv;
402         memset(devc->packet, 0x44, PACKET_SIZE);
403         devc->packet_size = 0;
404
405         devc->acquisition_running = TRUE;
406
407         serial = sdi->conn;
408         serial_source_add(sdi->session, serial, G_IO_IN, 50,
409                         atten_pps3xxx_receive_data, (void *)sdi);
410         std_session_send_df_header(sdi);
411
412         /* Send a "channel" configuration packet now. */
413         memset(packet, 0, PACKET_SIZE);
414         packet[0] = 0xaa;
415         packet[1] = 0xaa;
416         send_packet(sdi, packet);
417
418         return SR_OK;
419 }
420
421 static int dev_acquisition_stop(struct sr_dev_inst *sdi)
422 {
423         struct dev_context *devc;
424
425         devc = sdi->priv;
426         devc->acquisition_running = FALSE;
427
428         return SR_OK;
429 }
430
431 static struct sr_dev_driver atten_pps3203_driver_info = {
432         .name = "atten-pps3203",
433         .longname = "Atten PPS3203T-3S",
434         .api_version = 1,
435         .init = std_init,
436         .cleanup = std_cleanup,
437         .scan = scan_3203,
438         .dev_list = std_dev_list,
439         .dev_clear = std_dev_clear,
440         .config_get = config_get,
441         .config_set = config_set,
442         .config_list = config_list,
443         .dev_open = std_serial_dev_open,
444         .dev_close = dev_close,
445         .dev_acquisition_start = dev_acquisition_start,
446         .dev_acquisition_stop = dev_acquisition_stop,
447         .context = NULL,
448 };
449 SR_REGISTER_DEV_DRIVER(atten_pps3203_driver_info);