aboutsummaryrefslogtreecommitdiffstats
path: root/mpm/lib/i2c/i2cdev_iface.cpp
blob: b346597a86216acf53e68bf45f5da74ed86bb931 (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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
//
// Copyright 2018 Ettus Research, a National Instruments Company
//
// SPDX-License-Identifier: GPL-3.0-or-later
//


#include "i2cdev.h"
#include <fcntl.h>
#include <linux/i2c-dev.h>
#include <linux/i2c.h>
#include <mpm/exception.hpp>
#include <mpm/i2c/i2c_iface.hpp>
#include <boost/format.hpp>
#include <iostream>

using namespace mpm::i2c;

/******************************************************************************
 * Implementation
 *****************************************************************************/
class i2cdev_iface_impl : public i2c_iface
{
public:
    i2cdev_iface_impl(const std::string& device,
        const uint16_t addr,
        const bool ten_bit_addr,
        const unsigned int timeout_ms,
        const bool do_open = false)
        : _device(device)
        , _addr(addr)
        , _ten_bit_addr(ten_bit_addr)
        , _timeout_ms(timeout_ms)
    {
        if (do_open)
            _open();
        else
            _fd = -ENODEV;
    }

    ~i2cdev_iface_impl()
    {
        if (_fd >= 0)
            close(_fd);
    }

    int transfer(uint8_t* tx, size_t tx_len, uint8_t* rx, size_t rx_len, bool do_close)
    {
        if (_fd < 0)
            _open();

        int ret = i2cdev_transfer(_fd, _addr, _ten_bit_addr, tx, tx_len, rx, rx_len);

        if (do_close) {
            close(_fd);
            _fd = -ENODEV;
        }

        if (ret) {
            throw mpm::runtime_error(str(boost::format("I2C Transaction failed!")));
        }

        return ret;
    }

    int transfer(std::vector<uint8_t>* tx, std::vector<uint8_t>* rx, bool do_close)
    {
        uint8_t *tx_data = NULL, *rx_data = NULL;
        size_t tx_len = 0, rx_len = 0;

        if (tx) {
            tx_data = tx->data();
            tx_len  = tx->size();
        }

        if (rx) {
            rx_data = rx->data();
            rx_len  = rx->size();
        }
        int ret = transfer(tx_data, tx_len, rx_data, rx_len, do_close);

        return ret;
    }

private:
    const std::string _device;
    int _fd;
    const uint16_t _addr;
    const bool _ten_bit_addr;
    const unsigned int _timeout_ms;

    int _open(void)
    {
        if (i2cdev_open(&_fd, _device.c_str(), _timeout_ms) < 0) {
            throw mpm::runtime_error(
                str(boost::format("Could not initialize i2cdev device %s") % _device));
        }

        if (_fd < 0) {
            throw mpm::runtime_error(
                str(boost::format("Could not open i2cdev device %s") % _device));
        }
    }
};

/******************************************************************************
 * Factory
 *****************************************************************************/
i2c_iface::sptr i2c_iface::make_i2cdev(const std::string& bus,
    const uint16_t addr,
    const bool ten_bit_addr,
    const int timeout_ms)
{
    return std::make_shared<i2cdev_iface_impl>(bus, addr, ten_bit_addr, timeout_ms);
}