device.c

   1 /*
   2  * This file is part of the sigrok project.
   3  *
   4  * Copyright (C) 2010 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 <stdio.h>
  21 #include <glib.h>
  22 #include <sigrok.h>
  23 #include <sigrok-internal.h>
  24
  25 extern struct sr_global *global;
  26
  27 GSList *devices = NULL;
  28
  29 void sr_device_scan(void)
  30 {
  31         GSList *plugins, *l;
  32         struct sr_device_plugin *plugin;
  33
  34         plugins = sr_list_hwplugins();
  35
  36         /*
  37          * Initialize all plugins first. Since the init() call may involve
  38          * a firmware upload and associated delay, we may as well get all
  39          * of these out of the way first.
  40          */
  41         for (l = plugins; l; l = l->next) {
  42                 plugin = l->data;
  43                 sr_device_plugin_init(plugin);
  44         }
  45
  46 }
  47
  48 int sr_device_plugin_init(struct sr_device_plugin *plugin)
  49 {
  50         int num_devices, num_probes, i;
  51
  52         sr_info("initializing %s plugin", plugin->name);
  53         num_devices = plugin->init(NULL);
  54         for (i = 0; i < num_devices; i++) {
  55                 num_probes = (int)plugin->get_device_info(i, SR_DI_NUM_PROBES);
  56                 sr_device_new(plugin, i, num_probes);
  57         }
  58
  59         return num_devices;
  60 }
  61
  62 void sr_device_close_all(void)
  63 {
  64         int ret;
  65         struct sr_device *device;
  66
  67         while (devices) {
  68                 device = devices->data;
  69                 if (device->plugin && device->plugin->closedev) {
  70                         ret = device->plugin->closedev(device->plugin_index);
  71                         if (ret != SR_OK) {
  72                                 sr_err("dev: %s: could not close device %d",
  73                                        __func__, device->plugin_index);
  74                         }
  75                 }
  76                 sr_device_destroy(device);
  77         }
  78 }
  79
  80 GSList *sr_device_list(void)
  81 {
  82
  83         if (!devices)
  84                 sr_device_scan();
  85
  86         return devices;
  87 }
  88
  89 struct sr_device *sr_device_new(struct sr_device_plugin *plugin, int plugin_index,
  90                              int num_probes)
  91 {
  92         struct sr_device *device;
  93         int i;
  94
  95         if (!(device = g_try_malloc0(sizeof(struct sr_device)))) {
  96                 sr_err("dev: %s: device malloc failed", __func__);
  97                 return NULL;
  98         }
  99
 100         device->plugin = plugin;
 101         device->plugin_index = plugin_index;
 102         devices = g_slist_append(devices, device);
 103
 104         for (i = 0; i < num_probes; i++)
 105                 sr_device_probe_add(device, NULL);
 106
 107         return device;
 108 }
 109
 110 void sr_device_clear(struct sr_device *device)
 111 {
 112         unsigned int pnum;
 113
 114         /* TODO: Plugin-specific clear call? */
 115
 116         if (!device->probes)
 117                 return;
 118
 119         for (pnum = 1; pnum <= g_slist_length(device->probes); pnum++)
 120                 sr_device_probe_clear(device, pnum);
 121 }
 122
 123 void sr_device_destroy(struct sr_device *device)
 124 {
 125         unsigned int pnum;
 126
 127         /*
 128          * TODO: Plugin-specific destroy call, need to decrease refcount
 129          * in plugin.
 130          */
 131
 132         devices = g_slist_remove(devices, device);
 133         if (device->probes) {
 134                 for (pnum = 1; pnum <= g_slist_length(device->probes); pnum++)
 135                         sr_device_probe_clear(device, pnum);
 136                 g_slist_free(device->probes);
 137         }
 138         g_free(device);
 139 }
 140
 141 void sr_device_probe_clear(struct sr_device *device, int probenum)
 142 {
 143         struct sr_probe *p;
 144
 145         p = sr_device_probe_find(device, probenum);
 146         if (!p)
 147                 return;
 148
 149         if (p->name) {
 150                 g_free(p->name);
 151                 p->name = NULL;
 152         }
 153
 154         if (p->trigger) {
 155                 g_free(p->trigger);
 156                 p->trigger = NULL;
 157         }
 158 }
 159
 160 void sr_device_probe_add(struct sr_device *device, const char *name)
 161 {
 162         struct sr_probe *p;
 163         char probename[16];
 164         int probenum;
 165
 166         probenum = g_slist_length(device->probes) + 1;
 167
 168         if (!(p = g_try_malloc0(sizeof(struct sr_probe)))) {
 169                 sr_err("dev: %s: p malloc failed", __func__);
 170                 // return SR_ERR_MALLOC;
 171                 return; /* FIXME: should return int. */
 172         }
 173
 174         p->index = probenum;
 175         p->enabled = TRUE;
 176         if (name) {
 177                 p->name = g_strdup(name);
 178         } else {
 179                 snprintf(probename, 16, "%d", probenum);
 180                 p->name = g_strdup(probename);
 181         }
 182         p->trigger = NULL;
 183         device->probes = g_slist_append(device->probes, p);
 184 }
 185
 186 struct sr_probe *sr_device_probe_find(struct sr_device *device, int probenum)
 187 {
 188         GSList *l;
 189         struct sr_probe *p, *found_probe;
 190
 191         found_probe = NULL;
 192         for (l = device->probes; l; l = l->next) {
 193                 p = l->data;
 194                 if (p->index == probenum) {
 195                         found_probe = p;
 196                         break;
 197                 }
 198         }
 199
 200         return found_probe;
 201 }
 202
 203 /* TODO: return SR_ERR if probenum not found */
 204 void sr_device_probe_name(struct sr_device *device, int probenum,
 205                           const char *name)
 206 {
 207         struct sr_probe *p;
 208
 209         p = sr_device_probe_find(device, probenum);
 210         if (!p)
 211                 return;
 212
 213         if (p->name)
 214                 g_free(p->name);
 215         p->name = g_strdup(name);
 216 }
 217
 218 /* TODO: return SR_ERR if probenum not found */
 219 void sr_device_trigger_clear(struct sr_device *device)
 220 {
 221         struct sr_probe *p;
 222         unsigned int pnum;
 223
 224         if (!device->probes)
 225                 return;
 226
 227         for (pnum = 1; pnum <= g_slist_length(device->probes); pnum++) {
 228                 p = sr_device_probe_find(device, pnum);
 229                 if (p && p->trigger) {
 230                         g_free(p->trigger);
 231                         p->trigger = NULL;
 232                 }
 233         }
 234 }
 235
 236 /* TODO: return SR_ERR if probenum not found */
 237 void sr_device_trigger_set(struct sr_device *device, int probenum,
 238                            const char *trigger)
 239 {
 240         struct sr_probe *p;
 241
 242         p = sr_device_probe_find(device, probenum);
 243         if (!p)
 244                 return;
 245
 246         if (p->trigger)
 247                 g_free(p->trigger);
 248
 249         p->trigger = g_strdup(trigger);
 250
 251 }
 252
 253 gboolean sr_device_has_hwcap(struct sr_device *device, int hwcap)
 254 {
 255         int *capabilities, i;
 256
 257         if (!device || !device->plugin)
 258                 return FALSE;
 259
 260         if ((capabilities = device->plugin->get_capabilities()))
 261                 for (i = 0; capabilities[i]; i++)
 262                         if (capabilities[i] == hwcap)
 263                                 return TRUE;
 264
 265         return FALSE;
 266 }