]> sigrok.org Git - libsigrok.git/blob - src/hardware/atten-pps3xxx/api.c
rigol-ds: improve robustness in samplerate getting code path
[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         /* This is how the vendor software scans for hardware. */
117         memset(packet, 0, PACKET_SIZE);
118         packet[0] = 0xaa;
119         packet[1] = 0xaa;
120         delay_ms = serial_timeout(serial, PACKET_SIZE);
121         if (serial_write_blocking(serial, packet, PACKET_SIZE, delay_ms) < PACKET_SIZE) {
122                 sr_err("Unable to write while probing for hardware.");
123                 return NULL;
124         }
125         /* The device responds with a 24-byte packet when it receives a packet.
126          * At 9600 baud, 300ms is long enough for it to have arrived. */
127         g_usleep(300 * 1000);
128         memset(packet, 0, PACKET_SIZE);
129         if ((ret = serial_read_nonblocking(serial, packet, PACKET_SIZE)) < 0) {
130                 sr_err("Unable to read while probing for hardware: %s",
131                                 sr_strerror(ret));
132                 return NULL;
133         }
134         if (ret != PACKET_SIZE || packet[0] != 0xaa || packet[1] != 0xaa) {
135                 /* Doesn't look like an Atten PPS. */
136                 return NULL;
137         }
138
139         model = NULL;
140         for (i = 0; i < ARRAY_SIZE(models); i++) {
141                 if (models[i].modelid == modelid) {
142                         model = &models[i];
143                         break;
144                 }
145         }
146         if (!model) {
147                 sr_err("Unknown modelid %d", modelid);
148                 return NULL;
149         }
150
151         sdi = g_malloc0(sizeof(struct sr_dev_inst));
152         sdi->status = SR_ST_INACTIVE;
153         sdi->vendor = g_strdup("Atten");
154         sdi->model = g_strdup(model->name);
155         sdi->inst_type = SR_INST_SERIAL;
156         sdi->conn = serial;
157         for (i = 0; i < MAX_CHANNELS; i++) {
158                 snprintf(channel, 10, "CH%d", i + 1);
159                 ch = sr_channel_new(sdi, i, SR_CHANNEL_ANALOG, TRUE, channel);
160                 cg = g_malloc(sizeof(struct sr_channel_group));
161                 cg->name = g_strdup(channel);
162                 cg->channels = g_slist_append(NULL, ch);
163                 cg->priv = NULL;
164                 sdi->channel_groups = g_slist_append(sdi->channel_groups, cg);
165         }
166
167         devc = g_malloc0(sizeof(struct dev_context));
168         devc->model = model;
169         devc->config = g_malloc0(sizeof(struct per_channel_config) * model->num_channels);
170         devc->delay_ms = delay_ms;
171         sdi->priv = devc;
172
173         serial_close(serial);
174
175         return std_scan_complete(di, g_slist_append(NULL, sdi));
176 }
177
178 static GSList *scan_3203(struct sr_dev_driver *di, GSList *options)
179 {
180         return scan(di, options, PPS_3203T_3S);
181 }
182
183 static int config_get(uint32_t key, GVariant **data,
184         const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
185 {
186         struct dev_context *devc;
187         struct sr_channel *ch;
188         int channel;
189
190         if (!sdi)
191                 return SR_ERR_ARG;
192
193         devc = sdi->priv;
194
195         if (!cg) {
196                 switch (key) {
197                 case SR_CONF_CHANNEL_CONFIG:
198                         *data = g_variant_new_string(channel_modes[devc->channel_mode]);
199                         break;
200                 case SR_CONF_OVER_CURRENT_PROTECTION_ENABLED:
201                         *data = g_variant_new_boolean(devc->over_current_protection);
202                         break;
203                 default:
204                         return SR_ERR_NA;
205                 }
206         } else {
207                 /* We only ever have one channel per channel group in this driver. */
208                 ch = cg->channels->data;
209                 channel = ch->index;
210
211                 switch (key) {
212                 case SR_CONF_VOLTAGE:
213                         *data = g_variant_new_double(devc->config[channel].output_voltage_last);
214                         break;
215                 case SR_CONF_VOLTAGE_TARGET:
216                         *data = g_variant_new_double(devc->config[channel].output_voltage_max);
217                         break;
218                 case SR_CONF_CURRENT:
219                         *data = g_variant_new_double(devc->config[channel].output_current_last);
220                         break;
221                 case SR_CONF_CURRENT_LIMIT:
222                         *data = g_variant_new_double(devc->config[channel].output_current_max);
223                         break;
224                 case SR_CONF_ENABLED:
225                         *data = g_variant_new_boolean(devc->config[channel].output_enabled);
226                         break;
227                 default:
228                         return SR_ERR_NA;
229                 }
230         }
231
232         return SR_OK;
233 }
234
235 static int config_set(uint32_t key, GVariant *data,
236         const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
237 {
238         struct dev_context *devc;
239         struct sr_channel *ch;
240         gdouble dval;
241         int channel, ival;
242         gboolean bval;
243
244         devc = sdi->priv;
245
246         if (!cg) {
247                 switch (key) {
248                 case SR_CONF_CHANNEL_CONFIG:
249                         if ((ival = std_str_idx(data, ARRAY_AND_SIZE(channel_modes))) < 0)
250                                 return SR_ERR_ARG;
251                         if (devc->model->channel_modes && (1 << ival) == 0)
252                                 return SR_ERR_ARG; /* Not supported on this model. */
253                         if (ival == devc->channel_mode_set)
254                                 break; /* Nothing to do. */
255                         devc->channel_mode_set = ival;
256                         devc->config_dirty = TRUE;
257                         break;
258                 case SR_CONF_OVER_CURRENT_PROTECTION_ENABLED:
259                         bval = g_variant_get_boolean(data);
260                         if (bval == devc->over_current_protection_set)
261                                 break; /* Nothing to do. */
262                         devc->over_current_protection_set = bval;
263                         devc->config_dirty = TRUE;
264                         break;
265                 default:
266                         return SR_ERR_NA;
267                 }
268         } else {
269                 /* We only ever have one channel per channel group in this driver. */
270                 ch = cg->channels->data;
271                 channel = ch->index;
272
273                 switch (key) {
274                 case SR_CONF_VOLTAGE_TARGET:
275                         dval = g_variant_get_double(data);
276                         if (dval < 0 || dval > devc->model->channels[channel].voltage[1])
277                                 return SR_ERR_ARG;
278                         devc->config[channel].output_voltage_max = dval;
279                         devc->config_dirty = TRUE;
280                         break;
281                 case SR_CONF_CURRENT_LIMIT:
282                         dval = g_variant_get_double(data);
283                         if (dval < 0 || dval > devc->model->channels[channel].current[1])
284                                 return SR_ERR_ARG;
285                         devc->config[channel].output_current_max = dval;
286                         devc->config_dirty = TRUE;
287                         break;
288                 case SR_CONF_ENABLED:
289                         bval = g_variant_get_boolean(data);
290                         if (bval == devc->config[channel].output_enabled_set)
291                                 break; /* Nothing to do. */
292                         devc->config[channel].output_enabled_set = bval;
293                         devc->config_dirty = TRUE;
294                         break;
295                 default:
296                         return SR_ERR_NA;
297                 }
298         }
299
300         return SR_OK;
301 }
302
303 static int config_list(uint32_t key, GVariant **data,
304         const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
305 {
306         struct dev_context *devc;
307         struct sr_channel *ch;
308         int channel;
309
310         devc = (sdi) ? sdi->priv : NULL;
311
312         if (!cg) {
313                 switch (key) {
314                 case SR_CONF_SCAN_OPTIONS:
315                 case SR_CONF_DEVICE_OPTIONS:
316                         return STD_CONFIG_LIST(key, data, sdi, cg, scanopts, drvopts, devopts);
317                 case SR_CONF_CHANNEL_CONFIG:
318                         if (!devc || !devc->model)
319                                 return SR_ERR_ARG;
320                         if (devc->model->channel_modes == CHANMODE_INDEPENDENT) {
321                                 /* The 1-channel models. */
322                                 *data = g_variant_new_strv(channel_modes, 1);
323                         } else {
324                                 /* The other models support all modes. */
325                                 *data = g_variant_new_strv(ARRAY_AND_SIZE(channel_modes));
326                         }
327                         break;
328                 default:
329                         return SR_ERR_NA;
330                 }
331         } else {
332                 /* We only ever have one channel per channel group in this driver. */
333                 ch = cg->channels->data;
334                 channel = ch->index;
335
336                 switch (key) {
337                 case SR_CONF_DEVICE_OPTIONS:
338                         *data = std_gvar_array_u32(ARRAY_AND_SIZE(devopts_cg));
339                         break;
340                 case SR_CONF_VOLTAGE_TARGET:
341                         if (!devc || !devc->model)
342                                 return SR_ERR_ARG;
343                         *data = std_gvar_min_max_step_array(devc->model->channels[channel].voltage);
344                         break;
345                 case SR_CONF_CURRENT_LIMIT:
346                         if (!devc || !devc->model)
347                                 return SR_ERR_ARG;
348                         *data = std_gvar_min_max_step_array(devc->model->channels[channel].current);
349                         break;
350                 default:
351                         return SR_ERR_NA;
352                 }
353         }
354
355         return SR_OK;
356 }
357
358 static int dev_close(struct sr_dev_inst *sdi)
359 {
360         struct dev_context *devc;
361
362         devc = sdi->priv;
363
364         if (devc->config_dirty)
365                 /* Some configuration changes were queued up but didn't
366                  * get sent to the device, likely because we were never
367                  * in acquisition mode. Send them out now. */
368                 send_config(sdi);
369
370         return std_serial_dev_close(sdi);
371 }
372
373 static int dev_acquisition_start(const struct sr_dev_inst *sdi)
374 {
375         struct dev_context *devc;
376         struct sr_serial_dev_inst *serial;
377         uint8_t packet[PACKET_SIZE];
378
379         devc = sdi->priv;
380         memset(devc->packet, 0x44, PACKET_SIZE);
381         devc->packet_size = 0;
382
383         devc->acquisition_running = TRUE;
384
385         serial = sdi->conn;
386         serial_source_add(sdi->session, serial, G_IO_IN, 50,
387                         atten_pps3xxx_receive_data, (void *)sdi);
388         std_session_send_df_header(sdi);
389
390         /* Send a "channel" configuration packet now. */
391         memset(packet, 0, PACKET_SIZE);
392         packet[0] = 0xaa;
393         packet[1] = 0xaa;
394         send_packet(sdi, packet);
395
396         return SR_OK;
397 }
398
399 static int dev_acquisition_stop(struct sr_dev_inst *sdi)
400 {
401         struct dev_context *devc;
402
403         devc = sdi->priv;
404         devc->acquisition_running = FALSE;
405
406         return SR_OK;
407 }
408
409 static struct sr_dev_driver atten_pps3203_driver_info = {
410         .name = "atten-pps3203",
411         .longname = "Atten PPS3203T-3S",
412         .api_version = 1,
413         .init = std_init,
414         .cleanup = std_cleanup,
415         .scan = scan_3203,
416         .dev_list = std_dev_list,
417         .dev_clear = std_dev_clear,
418         .config_get = config_get,
419         .config_set = config_set,
420         .config_list = config_list,
421         .dev_open = std_serial_dev_open,
422         .dev_close = dev_close,
423         .dev_acquisition_start = dev_acquisition_start,
424         .dev_acquisition_stop = dev_acquisition_stop,
425         .context = NULL,
426 };
427 SR_REGISTER_DEV_DRIVER(atten_pps3203_driver_info);