feat: initial commit
This commit is contained in:
commit
78f58a8fc7
|
@ -0,0 +1,2 @@
|
||||||
|
zig-cache/
|
||||||
|
zig-out/
|
|
@ -0,0 +1,44 @@
|
||||||
|
const std = @import("std");
|
||||||
|
|
||||||
|
// Although this function looks imperative, note that its job is to
|
||||||
|
// declaratively construct a build graph that will be executed by an external
|
||||||
|
// runner.
|
||||||
|
pub fn build(b: *std.Build) void {
|
||||||
|
// Standard target options allows the person running `zig build` to choose
|
||||||
|
// what target to build for. Here we do not override the defaults, which
|
||||||
|
// means any target is allowed, and the default is native. Other options
|
||||||
|
// for restricting supported target set are available.
|
||||||
|
const target = b.standardTargetOptions(.{});
|
||||||
|
|
||||||
|
// Standard optimization options allow the person running `zig build` to select
|
||||||
|
// between Debug, ReleaseSafe, ReleaseFast, and ReleaseSmall. Here we do not
|
||||||
|
// set a preferred release mode, allowing the user to decide how to optimize.
|
||||||
|
const optimize = b.standardOptimizeOption(.{});
|
||||||
|
|
||||||
|
const lib = b.addStaticLibrary(.{
|
||||||
|
.name = "zba-util",
|
||||||
|
// In this case the main source file is merely a path, however, in more
|
||||||
|
// complicated build scripts, this could be a generated file.
|
||||||
|
.root_source_file = .{ .path = "src/lib.zig" },
|
||||||
|
.target = target,
|
||||||
|
.optimize = optimize,
|
||||||
|
});
|
||||||
|
|
||||||
|
// This declares intent for the library to be installed into the standard
|
||||||
|
// location when the user invokes the "install" step (the default step when
|
||||||
|
// running `zig build`).
|
||||||
|
lib.install();
|
||||||
|
|
||||||
|
// Creates a step for unit testing.
|
||||||
|
const main_tests = b.addTest(.{
|
||||||
|
.root_source_file = .{ .path = "src/main.zig" },
|
||||||
|
.target = target,
|
||||||
|
.optimize = optimize,
|
||||||
|
});
|
||||||
|
|
||||||
|
// This creates a build step. It will be visible in the `zig build --help` menu,
|
||||||
|
// and can be selected like this: `zig build test`
|
||||||
|
// This will evaluate the `test` step rather than the default, which is "install".
|
||||||
|
const test_step = b.step("test", "Run library tests");
|
||||||
|
test_step.dependOn(&main_tests.step);
|
||||||
|
}
|
|
@ -0,0 +1,69 @@
|
||||||
|
const std = @import("std");
|
||||||
|
|
||||||
|
pub fn RingBuffer(comptime T: type) type {
|
||||||
|
return struct {
|
||||||
|
const Self = @This();
|
||||||
|
const Index = usize;
|
||||||
|
|
||||||
|
const Atomic = std.atomic.Atomic;
|
||||||
|
|
||||||
|
const max_capacity = (@as(Index, 1) << @typeInfo(Index).Int.bits - 1) - 1; // half the range of index type
|
||||||
|
const log = std.log.scoped(.RingBuffer);
|
||||||
|
|
||||||
|
read: Atomic(Index),
|
||||||
|
write: Atomic(Index),
|
||||||
|
buf: []T,
|
||||||
|
|
||||||
|
const Error = error{buffer_full};
|
||||||
|
|
||||||
|
pub fn init(buf: []T) Self {
|
||||||
|
std.debug.assert(std.math.isPowerOfTwo(buf.len)); // capacity must be a power of two
|
||||||
|
std.debug.assert(buf.len <= max_capacity);
|
||||||
|
|
||||||
|
return .{
|
||||||
|
.read = Atomic(Index).init(0),
|
||||||
|
.write = Atomic(Index).init(0),
|
||||||
|
.buf = buf,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn push(self: *Self, value: T) Error!void {
|
||||||
|
const read_idx = self.read.load(.Acquire);
|
||||||
|
const write_idx = self.write.load(.Acquire);
|
||||||
|
|
||||||
|
// Check to see if Queue is full
|
||||||
|
if (write_idx - read_idx == self.buf.len) return Error.buffer_full;
|
||||||
|
|
||||||
|
self.buf[self.mask(write_idx)] = value;
|
||||||
|
|
||||||
|
std.atomic.fence(.Release);
|
||||||
|
self.write.store(write_idx + 1, .Release);
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn pop(self: *Self) ?T {
|
||||||
|
const read_idx = self.read.load(.Acquire);
|
||||||
|
const write_idx = self.write.load(.Acquire);
|
||||||
|
|
||||||
|
if (read_idx == write_idx) return null;
|
||||||
|
|
||||||
|
std.atomic.fence(.Acquire);
|
||||||
|
const value = self.buf[self.mask(self.read)];
|
||||||
|
|
||||||
|
std.atomic.fence(.Release);
|
||||||
|
self.read.store(read_idx + 1, .Release);
|
||||||
|
|
||||||
|
return value;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn len(self: *const Self) Index {
|
||||||
|
const read_idx = self.read.load(.Acquire);
|
||||||
|
const write_idx = self.write.load(.Acquire);
|
||||||
|
|
||||||
|
return write_idx - read_idx;
|
||||||
|
}
|
||||||
|
|
||||||
|
fn mask(self: *const Self, idx: Index) Index {
|
||||||
|
return idx & (self.buf.len - 1);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
Loading…
Reference in New Issue