File: MsgFifo.h

package info (click to toggle)
supercollider 1%3A3.13.0%2Brepack-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 80,292 kB
  • sloc: cpp: 476,363; lisp: 84,680; ansic: 77,685; sh: 25,509; python: 7,909; makefile: 3,440; perl: 1,964; javascript: 974; xml: 826; java: 677; yacc: 314; lex: 175; objc: 152; ruby: 136
file content (124 lines) | stat: -rw-r--r-- 4,064 bytes parent folder | download | duplicates (4)
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
116
117
118
119
120
121
122
123
124
/*
    SuperCollider real time audio synthesis system
    Copyright (c) 2002 James McCartney. All rights reserved.
    http://www.audiosynth.com

    Copyright (c) 2013 Tim Blechmann

    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; either version 2 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 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 Street, Fifth Floor, Boston, MA 02110-1301  USA
*/

#pragma once

#include <atomic>

/////////////////////////////////////////////////////////////////////

namespace SCMsgFifo {

using std::memory_order_relaxed;
using std::memory_order_release;

template <class MsgType, int N> class MsgFifo {
public:
    MsgFifo(): mReadHead(0), mWriteHead(0), mFreeHead(0) {}

    void MakeEmpty() {
        int writeHead = mWriteHead.load();
        mReadHead.store(writeHead, memory_order_relaxed);
        mFreeHead.store(writeHead, memory_order_release);
    }
    bool IsEmpty() { return mReadHead.load(memory_order_relaxed) == mWriteHead.load(memory_order_relaxed); }
    bool HasData() { return mReadHead.load(memory_order_relaxed) != mWriteHead.load(memory_order_relaxed); }
    bool NeedsFree() { return mFreeHead.load(memory_order_relaxed) != mReadHead.load(memory_order_relaxed); }

    bool Write(MsgType& data) {
        unsigned int next = NextPos(mWriteHead.load(memory_order_relaxed));
        if (next == mFreeHead.load(memory_order_relaxed))
            return false; // fifo is full
        mItems[next] = data;

        mWriteHead.store(next, memory_order_release);

        return true;
    }

    void Perform() // get next and advance
    {
        while (HasData()) {
            unsigned int next = NextPos(mReadHead.load(memory_order_relaxed));
            mItems[next].Perform();
            mReadHead.store(next, memory_order_release);
        }
    }
    void Free() // reclaim messages
    {
        while (NeedsFree()) {
            unsigned int next = NextPos(mFreeHead.load(memory_order_relaxed));
            mItems[next].Free();
            mFreeHead.store(next, memory_order_release);
        }
    }

private:
    int NextPos(int inPos) { return (inPos + 1) & (N - 1); }

    std::atomic<int> mReadHead, mWriteHead, mFreeHead;
    MsgType mItems[N];
};

/////////////////////////////////////////////////////////////////////

template <class MsgType, int N> class MsgFifoNoFree {
public:
    MsgFifoNoFree(): mReadHead(0), mWriteHead(0) {}

    void MakeEmpty() { mReadHead.store(mWriteHead.load()); }
    bool IsEmpty() { return mReadHead.load(memory_order_relaxed) == mWriteHead.load(memory_order_relaxed); }
    bool HasData() { return mReadHead.load(memory_order_relaxed) != mWriteHead.load(memory_order_relaxed); }

    bool Write(MsgType& data) {
        unsigned int next = NextPos(mWriteHead.load(memory_order_relaxed));
        if (next == mReadHead.load(memory_order_relaxed))
            return false; // fifo is full
        mItems[next] = data;

        mWriteHead.store(next, memory_order_release);

        return true;
    }

    void Perform() // get next and advance
    {
        while (HasData()) {
            unsigned int next = NextPos(mReadHead.load(memory_order_relaxed));
            mItems[next].Perform();
            mReadHead.store(next, memory_order_release);
        }
    }

private:
    int NextPos(int inPos) { return (inPos + 1) & (N - 1); }

    std::atomic<int> mReadHead, mWriteHead;
    MsgType mItems[N];
};

}

using SCMsgFifo::MsgFifo;
using SCMsgFifo::MsgFifoNoFree;

/////////////////////////////////////////////////////////////////////