X-Git-Url: https://sigrok.org/gitweb/?p=pulseview.git;a=blobdiff_plain;f=pv%2Fdevicemanager.cpp;h=883175c7752eb8cf27767a254d285da3760eb539;hp=4018c0470ee031ce17f3d37664c115bb79865992;hb=83b1c8d251386ac1980284c4668cbdd8e425550f;hpb=107ca6d350b51186d12dac9273f6ed130b3f5dd7 diff --git a/pv/devicemanager.cpp b/pv/devicemanager.cpp index 4018c047..883175c7 100644 --- a/pv/devicemanager.cpp +++ b/pv/devicemanager.cpp @@ -14,122 +14,174 @@ * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License - * along with this program; if not, write to the Free Software - * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA + * along with this program; if not, see . */ -#include "devicemanager.h" +#include "devicemanager.hpp" +#include "session.hpp" #include -#include +#include #include +#include #include -#include +#include -using namespace std; +#include + +#include + +using std::bind; +using std::list; +using std::map; +using std::placeholders::_1; +using std::placeholders::_2; +using std::shared_ptr; +using std::string; + +using Glib::VariantBase; + +using sigrok::ConfigKey; +using sigrok::Context; +using sigrok::Driver; namespace pv { -DeviceManager::DeviceManager(struct sr_context *sr_ctx) : - _sr_ctx(sr_ctx) +DeviceManager::DeviceManager(shared_ptr context) : + context_(context) +{ + for (auto entry : context->drivers()) + driver_scan(entry.second, map()); +} + +const shared_ptr& DeviceManager::context() const { - init_drivers(); - scan_all_drivers(); + return context_; } -DeviceManager::~DeviceManager() +shared_ptr DeviceManager::context() { - release_devices(); + return context_; } -const std::list& DeviceManager::devices() const +const list< shared_ptr >& +DeviceManager::devices() const { - return _devices; + return devices_; } -list DeviceManager::driver_scan( - struct sr_dev_driver *const driver, GSList *const drvopts) +list< shared_ptr > +DeviceManager::driver_scan( + shared_ptr driver, map drvopts) { - list driver_devices; + list< shared_ptr > driver_devices; assert(driver); // Remove any device instances from this driver from the device // list. They will not be valid after the scan. - list::iterator i = _devices.begin(); - while (i != _devices.end()) { - if ((*i)->driver == driver) - i = _devices.erase(i); - else - i++; - } - - // Clear all the old device instances from this driver - sr_dev_clear(driver); + devices_.remove_if([&](shared_ptr device) { + return device->hardware_device()->driver() == driver; }); // Do the scan - GSList *const devices = sr_driver_scan(driver, drvopts); - for (GSList *l = devices; l; l = l->next) - driver_devices.push_back((sr_dev_inst*)l->data); - g_slist_free(devices); - driver_devices.sort(compare_devices); - - // Add the scanned devices to the main list - _devices.insert(_devices.end(), driver_devices.begin(), + auto devices = driver->scan(drvopts); + + // Add the scanned devices to the main list, set display names and sort. + for (shared_ptr device : devices) { + const shared_ptr d( + new devices::HardwareDevice(context_, device)); + driver_devices.push_back(d); + } + + devices_.insert(devices_.end(), driver_devices.begin(), driver_devices.end()); - _devices.sort(compare_devices); + devices_.sort(bind(&DeviceManager::compare_devices, this, _1, _2)); + driver_devices.sort(bind( + &DeviceManager::compare_devices, this, _1, _2)); return driver_devices; } -void DeviceManager::init_drivers() +const map DeviceManager::get_device_info( + shared_ptr device) { - // Initialise all libsigrok drivers - sr_dev_driver **const drivers = sr_driver_list(); - for (sr_dev_driver **driver = drivers; *driver; driver++) { - if (sr_driver_init(_sr_ctx, *driver) != SR_OK) { - throw runtime_error( - string("Failed to initialize driver ") + - string((*driver)->name)); - } - } + map result; + + assert(device); + + const shared_ptr sr_dev = device->device(); + if (sr_dev->vendor().length() > 0) + result["vendor"] = sr_dev->vendor(); + if (sr_dev->model().length() > 0) + result["model"] = sr_dev->model(); + if (sr_dev->version().length() > 0) + result["version"] = sr_dev->version(); + if (sr_dev->serial_number().length() > 0) + result["serial_num"] = sr_dev->serial_number(); + if (sr_dev->connection_id().length() > 0) + result["connection_id"] = sr_dev->connection_id(); + + return result; } -void DeviceManager::release_devices() +const shared_ptr DeviceManager::find_device_from_info( + const map search_info) { - sr_dev_driver **const drivers = sr_driver_list(); - for (sr_dev_driver **driver = drivers; *driver; driver++) - sr_dev_clear(*driver); -} + shared_ptr last_resort_dev; + map dev_info; + + for (shared_ptr dev : devices_) { + assert(dev); + dev_info = get_device_info(dev); + + // If present, vendor and model always have to match. + if (dev_info.count("vendor") > 0 && search_info.count("vendor") > 0) + if (dev_info.at("vendor") != search_info.at("vendor")) + continue; + + if (dev_info.count("model") > 0 && search_info.count("model") > 0) + if (dev_info.at("model") != search_info.at("model")) + continue; + + // Most unique match: vendor/model/serial_num (but don't match a S/N of 0) + if ((dev_info.count("serial_num") > 0) && (dev_info.at("serial_num") != "0") + && search_info.count("serial_num") > 0) + if (dev_info.at("serial_num") == search_info.at("serial_num") && + dev_info.at("serial_num") != "0") + return dev; + + // Second best match: vendor/model/connection_id + if (dev_info.count("connection_id") > 0 && + search_info.count("connection_id") > 0) + if (dev_info.at("connection_id") == search_info.at("connection_id")) + return dev; + + // Last resort: vendor/model/version + if (dev_info.count("version") > 0 && + search_info.count("version") > 0) + if (dev_info.at("version") == search_info.at("version") && + dev_info.at("version") != "0") + return dev; + + // For this device, we merely have a vendor/model match. + last_resort_dev = dev; + } -void DeviceManager::scan_all_drivers() -{ - // Scan all drivers for all devices. - struct sr_dev_driver **const drivers = sr_driver_list(); - for (struct sr_dev_driver **driver = drivers; *driver; driver++) - driver_scan(*driver); + // If there wasn't even a vendor/model/version match, we end up here. + // This is usually the case for devices with only vendor/model data. + // The selected device may be wrong with multiple such devices attached + // but it is the best we can do at this point. After all, there may be + // only one such device and we do want to select it in this case. + return last_resort_dev; } -bool DeviceManager::compare_devices(const sr_dev_inst *const a, - const sr_dev_inst *const b) +bool DeviceManager::compare_devices(shared_ptr a, + shared_ptr b) { assert(a); assert(b); - - const int vendor_cmp = strcasecmp(a->vendor, b->vendor); - if(vendor_cmp < 0) - return true; - else if(vendor_cmp > 0) - return false; - - const int model_cmp = strcasecmp(a->model, b->model); - if(model_cmp < 0) - return true; - else if(model_cmp > 0) - return false; - - return strcasecmp(a->version, b->version) < 0; + return a->display_name(*this).compare(b->display_name(*this)) < 0; } } // namespace pv