aboutsummaryrefslogtreecommitdiffstats
path: root/gui/stmdsp.hpp
blob: 030038d891487a275b99e278d5e76b8737184577 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
#ifndef STMDSP_HPP_
#define STMDSP_HPP_

#include <cstdint>
#include <list>
#include <serial/serial.h>
#include <string>

namespace stmdsp
{
    class scanner
    {
    private:
        constexpr static const char *STMDSP_USB_ID = "USB VID:PID=0483:5740";

    public:
        std::list<std::string>& scan();
        auto& devices() {
            return m_available_devices;
        }

    private:
        std::list<std::string> m_available_devices;
    };

    using adcsample_t = uint16_t;

    class device
    {
    public:
        device(const std::string& file) :
            m_serial(file, 115200, serial::Timeout::simpleTimeout(1000)) {}

        ~device() {
            m_serial.close();
        }

        bool connected() {
            return m_serial.isOpen() && (m_serial.write("i"), m_serial.read(6) == "stmdsp");
        }

        std::vector<adcsample_t> sample(unsigned long int count = 1) {
            if (connected()) {
                m_serial.write(std::vector<uint8_t> {'r',
                                                     static_cast<uint8_t>(count),
                                                     static_cast<uint8_t>(count >> 8)});
                std::vector<adcsample_t> data (count);
                m_serial.read(reinterpret_cast<uint8_t *>(data.data()),
                              data.size() * sizeof(adcsample_t));
                return data;
            } else {
                return {};
            }
        }

    private:
        serial::Serial m_serial;
    };
}

#endif // STMDSP_HPP_