mirror of
https://github.com/f4exb/sdrangel.git
synced 2024-12-23 10:05:46 -05:00
364 lines
11 KiB
C++
364 lines
11 KiB
C++
///////////////////////////////////////////////////////////////////////////////////
|
|
// Copyright (C) 2016-2019, 2022 Edouard Griffiths, F4EXB <f4exb06@gmail.com> //
|
|
// Copyright (C) 2017 Ziga S <ziga.svetina@gmail.com> //
|
|
// Copyright (C) 2020 Kacper Michajłow <kasper93@gmail.com> //
|
|
// Copyright (C) 2022 Jon Beniston, M7RCE <jon@beniston.com> //
|
|
// //
|
|
// This program is free software; you can redistribute it and/or modify //
|
|
// it under the terms of the GNU General Public License as published by //
|
|
// the Free Software Foundation as version 3 of the License, or //
|
|
// (at your option) any later version. //
|
|
// //
|
|
// This program is distributed in the hope that it will be useful, //
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of //
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the //
|
|
// GNU General Public License V3 for more details. //
|
|
// //
|
|
// You should have received a copy of the GNU General Public License //
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>. //
|
|
///////////////////////////////////////////////////////////////////////////////////
|
|
|
|
#ifndef __APPLE__
|
|
#include <fcntl.h>
|
|
#include <sys/stat.h>
|
|
#endif
|
|
|
|
#ifndef _MSC_VER
|
|
#include <dirent.h>
|
|
#include <unistd.h>
|
|
#include <libgen.h>
|
|
#endif
|
|
|
|
#if !defined(_WIN32) && !defined(__APPLE__)
|
|
#include <termios.h>
|
|
#include <sys/ioctl.h>
|
|
#include <linux/serial.h>
|
|
#include "util/serialutil.h"
|
|
#endif
|
|
|
|
#include <chrono>
|
|
#include <thread>
|
|
|
|
#include <QThread>
|
|
#include <QBuffer>
|
|
#include <QDataStream>
|
|
|
|
#include "ambeworker.h"
|
|
#include "ambeengine.h"
|
|
|
|
uint32_t AMBEEngine::AMBEController::getSuccessCount() const
|
|
{
|
|
return worker ? worker->getSuccessCount() : 0;
|
|
}
|
|
|
|
uint32_t AMBEEngine::AMBEController::getFailureCount() const
|
|
{
|
|
return worker ? worker->getFailureCount() : 0;
|
|
}
|
|
|
|
AMBEEngine::AMBEEngine()
|
|
{}
|
|
|
|
AMBEEngine::~AMBEEngine()
|
|
{
|
|
qDebug("AMBEEngine::~AMBEEngine: %lu controllers", m_controllers.size());
|
|
}
|
|
|
|
#if defined(_WIN32)
|
|
void AMBEEngine::getComList()
|
|
{
|
|
qDebug("AMBEEngine::getComList: Win");
|
|
m_comList.clear();
|
|
m_comList8250.clear();
|
|
char comCStr[16];
|
|
|
|
// Arbitrarily set the list to the 20 first COM ports
|
|
for (int i = 1; i <= 20; i++)
|
|
{
|
|
sprintf(comCStr, "COM%d", i);
|
|
m_comList.push_back(std::string(comCStr));
|
|
}
|
|
}
|
|
|
|
void AMBEEngine::scan(QList<QString>& ambeDevices)
|
|
{
|
|
qDebug("AMBEEngine::scan");
|
|
AMBEEngine::getComList();
|
|
std::vector<std::string>::const_iterator it = m_comList.begin();
|
|
ambeDevices.clear();
|
|
|
|
while (it != m_comList.end())
|
|
{
|
|
AMBEWorker *worker = new AMBEWorker();
|
|
qDebug("AMBEEngine::scan: com: %s", it->c_str());
|
|
|
|
if (worker->open(*it))
|
|
{
|
|
ambeDevices.push_back(QString(it->c_str()));
|
|
worker->close();
|
|
}
|
|
|
|
delete worker;
|
|
++it;
|
|
}
|
|
}
|
|
#elif defined(__APPLE__)
|
|
void AMBEEngine::getComList()
|
|
{
|
|
qDebug("AMBEEngine::getComList: Apple");
|
|
}
|
|
void AMBEEngine::scan(QList<QString>& ambeDevices)
|
|
{
|
|
(void) ambeDevices;
|
|
}
|
|
#else
|
|
void AMBEEngine::getComList()
|
|
{
|
|
qDebug("AMBEEngine::getComList: Linux");
|
|
m_comList.clear();
|
|
m_comList8250.clear();
|
|
const char* sysdir = "/sys/class/tty/";
|
|
|
|
std::vector<std::string> comPorts;
|
|
SerialUtil::getComPorts(comPorts, "ttyUSB[0-9]+"); // in Linux the AMBE devices will be listed as ttyUSBx
|
|
for (std::vector<std::string>::iterator it = comPorts.begin(); it != comPorts.end(); ++it)
|
|
{
|
|
// Construct full absolute file path
|
|
std::string devicedir = sysdir;
|
|
it->erase(0,5); // remove /dev/
|
|
devicedir += *it;
|
|
// Register the device
|
|
register_comport(m_comList, m_comList8250, devicedir);
|
|
}
|
|
|
|
// Only non-serial8250 has been added to comList without any further testing
|
|
// serial8250-devices must be probe to check for validity
|
|
probe_serial8250_comports(m_comList, m_comList8250);
|
|
}
|
|
#endif // not Windows nor Apple
|
|
|
|
#if !defined(_WIN32) && !defined(__APPLE__)
|
|
void AMBEEngine::register_comport(
|
|
std::vector<std::string>& comList,
|
|
std::vector<std::string>& comList8250,
|
|
const std::string& dir)
|
|
{
|
|
// Get the driver the device is using
|
|
std::string driver = get_driver(dir);
|
|
|
|
// Skip devices without a driver
|
|
if (driver.size() > 0)
|
|
{
|
|
//std::cerr << "register_comport: dir: "<< dir << " driver: " << driver << std::endl;
|
|
std::string devfile = std::string("/dev/") + basename((char *) dir.c_str());
|
|
|
|
// Put serial8250-devices in a separate list
|
|
if (driver == "serial8250") {
|
|
comList8250.push_back(devfile);
|
|
} else {
|
|
comList.push_back(devfile);
|
|
}
|
|
}
|
|
}
|
|
|
|
void AMBEEngine::probe_serial8250_comports(
|
|
std::vector<std::string>& comList,
|
|
std::vector<std::string> comList8250)
|
|
{
|
|
struct serial_struct serinfo;
|
|
std::vector<std::string>::iterator it = comList8250.begin();
|
|
|
|
// Iterate over all serial8250-devices
|
|
while (it != comList8250.end())
|
|
{
|
|
|
|
// Try to open the device
|
|
int fd = open((*it).c_str(), O_RDWR | O_NONBLOCK | O_NOCTTY);
|
|
|
|
if (fd >= 0)
|
|
{
|
|
// Get serial_info
|
|
if (ioctl(fd, TIOCGSERIAL, &serinfo) == 0)
|
|
{
|
|
// If device type is no PORT_UNKNOWN we accept the port
|
|
if (serinfo.type != PORT_UNKNOWN) {
|
|
comList.push_back(*it);
|
|
}
|
|
}
|
|
|
|
close(fd);
|
|
}
|
|
it++;
|
|
}
|
|
}
|
|
|
|
std::string AMBEEngine::get_driver(const std::string& tty)
|
|
{
|
|
struct stat st;
|
|
std::string devicedir = tty;
|
|
// Append '/device' to the tty-path
|
|
devicedir += "/device";
|
|
|
|
// Stat the devicedir and handle it if it is a symlink
|
|
if (lstat(devicedir.c_str(), &st) == 0 && S_ISLNK(st.st_mode))
|
|
{
|
|
char buffer[1024];
|
|
memset(buffer, 0, sizeof(buffer));
|
|
// Append '/driver' and return basename of the target
|
|
devicedir += "/driver";
|
|
|
|
if (readlink(devicedir.c_str(), buffer, sizeof(buffer)) > 0) {
|
|
return basename(buffer);
|
|
}
|
|
}
|
|
|
|
return "";
|
|
}
|
|
|
|
void AMBEEngine::scan(QList<QString>& ambeDevices)
|
|
{
|
|
qDebug("AMBEEngine::scan");
|
|
AMBEEngine::getComList();
|
|
std::vector<std::string>::const_iterator it = m_comList.begin();
|
|
ambeDevices.clear();
|
|
|
|
while (it != m_comList.end())
|
|
{
|
|
AMBEWorker *worker = new AMBEWorker();
|
|
qDebug("AMBEEngine::scan: com: %s", it->c_str());
|
|
|
|
if (worker->open(*it))
|
|
{
|
|
ambeDevices.push_back(QString(it->c_str()));
|
|
worker->close();
|
|
}
|
|
|
|
delete worker;
|
|
++it;
|
|
}
|
|
}
|
|
#endif // not Windows nor Apple
|
|
|
|
bool AMBEEngine::registerController(const std::string& deviceRef)
|
|
{
|
|
AMBEWorker *worker = new AMBEWorker();
|
|
|
|
if (worker->open(deviceRef))
|
|
{
|
|
qDebug("AMBEEngine::registerController: device: %s", deviceRef.c_str());
|
|
m_controllers.push_back(AMBEController());
|
|
m_controllers.back().worker = worker;
|
|
m_controllers.back().thread = new QThread();
|
|
m_controllers.back().device = deviceRef;
|
|
|
|
m_controllers.back().worker->moveToThread(m_controllers.back().thread);
|
|
connect(m_controllers.back().worker, SIGNAL(finished()), m_controllers.back().thread, SLOT(quit()));
|
|
connect(m_controllers.back().worker, SIGNAL(finished()), m_controllers.back().worker, SLOT(deleteLater()));
|
|
connect(m_controllers.back().thread, SIGNAL(finished()), m_controllers.back().thread, SLOT(deleteLater()));
|
|
connect(&m_controllers.back().worker->m_inputMessageQueue, SIGNAL(messageEnqueued()), m_controllers.back().worker, SLOT(handleInputMessages()));
|
|
std::this_thread::sleep_for(std::chrono::seconds(1));
|
|
m_controllers.back().thread->start();
|
|
|
|
return true;
|
|
}
|
|
|
|
qWarning("AMBEEngine::registerController: failed to register device: %s", deviceRef.c_str());
|
|
return false;
|
|
}
|
|
|
|
void AMBEEngine::releaseController(const std::string& deviceRef)
|
|
{
|
|
std::vector<AMBEController>::iterator it = m_controllers.begin();
|
|
|
|
while (it != m_controllers.end())
|
|
{
|
|
if (it->device == deviceRef)
|
|
{
|
|
disconnect(&it->worker->m_inputMessageQueue, SIGNAL(messageEnqueued()), it->worker, SLOT(handleInputMessages()));
|
|
it->worker->stop();
|
|
it->thread->wait(100);
|
|
it->worker->m_inputMessageQueue.clear();
|
|
it->worker->close();
|
|
qDebug() << "AMBEEngine::releaseController: closed device at: " << it->device.c_str();
|
|
m_controllers.erase(it);
|
|
break;
|
|
}
|
|
|
|
++it;
|
|
}
|
|
}
|
|
|
|
void AMBEEngine::releaseAll()
|
|
{
|
|
std::vector<AMBEController>::iterator it = m_controllers.begin();
|
|
|
|
while (it != m_controllers.end())
|
|
{
|
|
disconnect(&it->worker->m_inputMessageQueue, SIGNAL(messageEnqueued()), it->worker, SLOT(handleInputMessages()));
|
|
it->worker->stop();
|
|
it->thread->wait(100);
|
|
it->worker->m_inputMessageQueue.clear();
|
|
it->worker->close();
|
|
qDebug() << "AMBEEngine::release: closed device at: " << it->device.c_str();
|
|
++it;
|
|
}
|
|
|
|
m_controllers.clear();
|
|
}
|
|
|
|
void AMBEEngine::getDeviceRefs(QList<DeviceRef>& deviceRefs)
|
|
{
|
|
std::vector<AMBEController>::const_iterator it = m_controllers.begin();
|
|
|
|
while (it != m_controllers.end())
|
|
{
|
|
deviceRefs.push_back(DeviceRef{QString(it->device.c_str()), it->getSuccessCount(), it->getFailureCount()});
|
|
++it;
|
|
}
|
|
}
|
|
|
|
void AMBEEngine::pushMbeFrame(
|
|
const unsigned char *mbeFrame,
|
|
int mbeRateIndex,
|
|
int mbeVolumeIndex,
|
|
unsigned char channels,
|
|
bool useLP,
|
|
int upsampling,
|
|
AudioFifo *audioFifo)
|
|
{
|
|
std::vector<AMBEController>::iterator it = m_controllers.begin();
|
|
std::vector<AMBEController>::iterator itAvail = m_controllers.end();
|
|
bool done = false;
|
|
QMutexLocker locker(&m_mutex);
|
|
|
|
while (it != m_controllers.end())
|
|
{
|
|
if (it->worker->hasFifo(audioFifo))
|
|
{
|
|
it->worker->pushMbeFrame(mbeFrame, mbeRateIndex, mbeVolumeIndex, channels, useLP, upsampling, audioFifo);
|
|
done = true;
|
|
}
|
|
else if (it->worker->isAvailable())
|
|
{
|
|
itAvail = it;
|
|
}
|
|
|
|
++it;
|
|
}
|
|
|
|
if (!done)
|
|
{
|
|
if (itAvail != m_controllers.end())
|
|
{
|
|
int wNum = itAvail - m_controllers.begin();
|
|
|
|
qDebug("AMBEEngine::pushMbeFrame: push %p on empty queue %d", audioFifo, wNum);
|
|
itAvail->worker->pushMbeFrame(mbeFrame, mbeRateIndex, mbeVolumeIndex, channels, useLP, upsampling, audioFifo);
|
|
}
|
|
else
|
|
{
|
|
qDebug("AMBEEngine::pushMbeFrame: no DV device available. MBE frame dropped");
|
|
}
|
|
}
|
|
}
|