summaryrefslogtreecommitdiff
path: root/gr-zeromq/lib/base_impl.h
blob: c9b5245fdb3237a44e4ac1328bc729f66953583c (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
/* -*- c++ -*- */
/*
 * Copyright 2016,2019 Free Software Foundation, Inc.
 *
 * This file is part of GNU Radio.
 *
 * SPDX-License-Identifier: GPL-3.0-or-later
 *
 */

#ifndef INCLUDED_ZEROMQ_BASE_IMPL_H
#define INCLUDED_ZEROMQ_BASE_IMPL_H

#include "zmq_common_impl.h"
#include <gnuradio/sync_block.h>

namespace gr {
namespace zeromq {

class base_impl : public virtual gr::sync_block
{
public:
    base_impl(int type,
              size_t itemsize,
              size_t vlen,
              int timeout,
              bool pass_tags,
              const std::string& key = "");
    virtual ~base_impl();

protected:
    std::string last_endpoint();
    zmq::context_t* d_context;
    zmq::socket_t* d_socket;
    size_t d_vsize;
    int d_timeout;
    bool d_pass_tags;
    const std::string d_key;
};

class base_sink_impl : public base_impl
{
public:
    base_sink_impl(int type,
                   size_t itemsize,
                   size_t vlen,
                   char* address,
                   int timeout,
                   bool pass_tags,
                   int hwm,
                   const std::string& key = "");

protected:
    int send_message(const void* in_buf, const int in_nitems, const uint64_t in_offset);
};

class base_source_impl : public base_impl
{
public:
    base_source_impl(int type,
                     size_t itemsize,
                     size_t vlen,
                     char* address,
                     int timeout,
                     bool pass_tags,
                     int hwm,
                     const std::string& key = "");

protected:
    zmq::message_t d_msg;
    std::vector<gr::tag_t> d_tags;
    size_t d_consumed_bytes;
    int d_consumed_items;

    bool has_pending();
    int flush_pending(void* out_buf, const int out_nitems, const uint64_t out_offset);
    bool load_message(bool wait);
};

} // namespace zeromq
} // namespace gr

#endif /* INCLUDED_ZEROMQ_BASE_IMPL_H */

// vim: ts=2 sw=2 expandtab