sonnum/zigsonnum/soundnode.zig

119 lines
2.9 KiB
Zig

const std = @import("std");
const print = std.debug.print;
const math = std.math;
const Endian = std.builtin.Endian;
const ArrayList = std.ArrayList;
const AutoHashMap = std.AutoHashMap;
const Allocator = std.mem.Allocator;
const Pnt = @import("point.zig").Pnt;
const SoundSettings = @import("settings.zig").SoundSettings;
const utility = @import("utility.zig");
const prnt = utility.prnt;
pub const SoundNode = struct {
name: []const u8 = "soundnode",
location: Pnt = Pnt{.x = 0, .y = 0, .z = 0},
air_in: ArrayList(*SoundNode),
wire_in: ArrayList(*SoundNode),
freqmap: AutoHashMap(u64, f16),
pub fn init(allocator: Allocator, name: []const u8) !SoundNode {
var air_in = ArrayList(*SoundNode).init(allocator);
defer air_in.deinit();
var wire_in = ArrayList(*SoundNode).init(allocator);
defer wire_in.deinit();
var freqmap = AutoHashMap(u64, f16).init(allocator);
defer freqmap.deinit();
return .{
.air_in = air_in,
.wire_in = wire_in,
.name = name,
.freqmap = freqmap,
};
}
};
pub fn singleSineTick(st: *SoundSettings, freq: f16, t: u64) f64 {
const ft: f64 = @floatFromInt(t);
return math.sin(st.sine_multiplier * @as(f64, freq) * ft);
}
pub fn main() !void {
var gpa = std.heap.GeneralPurposeAllocator(.{}){};
const allocator = gpa.allocator();
var sn1: SoundNode = try SoundNode.init(allocator, "left_sink");
var sn2: SoundNode = try SoundNode.init(allocator, "right_sink");
sn2.location.x = -5;
sn1.location.x = 4;
try sn1.air_in.append(&sn2);
var st = SoundSettings{};
st.printSettings();
const duration_in_sec: u32 = 5*60;
prnt("start");
const stdout_file = std.io.getStdOut().writer();
var bw = std.io.bufferedWriter(stdout_file);
const stdout = bw.writer();
//WRITING WAV HEADER
try stdout.writeAll("RIFF");
try stdout.writeInt(
u32,
36 + ((duration_in_sec * st.sample_rate) * 2 * st.sample_width),
Endian.little,
);
try stdout.writeAll("WAVE");
try stdout.writeAll("fmt ");
try stdout.writeInt(u32, 16, Endian.little);
try stdout.writeInt(u16, 1, Endian.little);
try stdout.writeInt(u16, 2, Endian.little);
try stdout.writeInt(u32, st.sample_rate, Endian.little);
const block_align: u16 = @intCast(2 * st.sample_width);
try stdout.writeInt(u32, @as(u32, block_align) * st.sample_rate, Endian.little);
try stdout.writeInt(u16, block_align, Endian.little);
try stdout.writeInt(u16, st.bit_depth, Endian.little);
try stdout.writeAll("data");
try stdout.writeInt(
u32,
(duration_in_sec * st.sample_rate) * 2 * st.sample_width,
Endian.little,
);
var sample: i24 = 0.0;
var amp: f64 = undefined;
for (0 .. st.sample_rate * duration_in_sec) |t| {
amp = singleSineTick(&st, 440, t);
sample = @intFromFloat(amp * @as(f64, @floatFromInt(st.max_amp)));
try stdout.writeInt(i24, sample, Endian.little);
try stdout.writeInt(i24, sample, Endian.little);
}
prnt("done");
try bw.flush();
}