//! By convention, root.zig is the root source file when making a library. If
//! you are making an executable, the convention is to delete this file and
//! start with main.zig instead.
const std = @import("std");
const builtin = @import("builtin");
const testing = std.testing;

pub const value = @import("libzisp/value.zig");
pub const read = @import("libzisp/read.zig");
pub const gc = @import("libzisp/gc.zig");

pub const Value = value.Value;
pub const Bucket = gc.Bucket;

test "double" {
    const d1: f64 = 0.123456789;
    const d2: f64 = -0.987654321;
    const v1 = value.double.pack(d1);
    const v2 = value.double.pack(d2);
    const v3 = value.double.add(v1, v2);
    const result = value.double.unpack(v3);

    try std.testing.expect(value.double.check(v1));
    try std.testing.expect(value.double.check(v2));
    try std.testing.expect(value.double.check(v3));

    try std.testing.expectEqual(d1 + d2, result);
}

test "fixnum" {
    const int1: i64 = 123456789;
    const int2: i64 = -987654321;
    const v1 = value.fixnum.pack(int1);
    const v2 = value.fixnum.pack(int2);
    const v3 = value.fixnum.add(v1, v2);
    const result = value.fixnum.unpack(v3);

    try std.testing.expect(value.fixnum.check(v1));
    try std.testing.expect(value.fixnum.check(v2));
    try std.testing.expect(value.fixnum.check(v3));

    try std.testing.expectEqual(int1 + int2, result);
}

test "ptr" {
    const ptr = value.ptr;

    const val: [*]Bucket = @ptrFromInt(256);
    const tag = ptr.Tag.string;

    const f = ptr.packForeign(val);
    try std.testing.expect(ptr.checkForeign(f));
    try std.testing.expectEqual(
        @intFromPtr(val),
        @intFromPtr(ptr.unpackForeign(f)),
    );

    const p = ptr.pack(val, tag);
    try std.testing.expect(ptr.check(p));
    try std.testing.expect(ptr.checkZisp(p, tag));
    try std.testing.expect(ptr.checkStrong(p));

    const pv, const pt = ptr.unpack(p);
    try std.testing.expectEqual(val, pv);
    try std.testing.expectEqual(tag, pt);

    var w = ptr.makeWeak(p);
    try std.testing.expect(ptr.check(w));
    try std.testing.expect(ptr.checkZisp(w, tag));
    try std.testing.expect(ptr.checkWeak(w));
    try std.testing.expectEqual(true, value.boole.unpack(ptr.predWeak(w)));
    try std.testing.expectEqual(false, value.boole.unpack(ptr.predWeakNull(w)));

    const wv, const wt = ptr.unpack(w);
    try std.testing.expectEqual(val, wv);
    try std.testing.expectEqual(tag, wt);

    const wv2, const wt2 = ptr.unpack(ptr.getWeak(w));
    try std.testing.expectEqual(val, wv2);
    try std.testing.expectEqual(tag, wt2);

    ptr.setWeakNull(&w);
    try std.testing.expect(ptr.check(w));
    try std.testing.expect(ptr.checkWeak(w));
    try std.testing.expect(ptr.isWeakNull(w));
    try std.testing.expectEqual(true, value.boole.unpack(ptr.predWeak(w)));
    try std.testing.expectEqual(true, value.boole.unpack(ptr.predWeakNull(w)));
    try std.testing.expectEqual(false, value.boole.unpack(ptr.getWeak(w)));
}

test "sstr" {
    const impls = .{
        .{ value.sstr.pack, value.sstr.unpack },
        // .{ value.sstr.pack1, value.sstr.unpack1 },
        // .{ value.sstr.pack2, value.sstr.unpack2 },
        // .{ value.sstr.pack3, value.sstr.unpack3 },
        // .{ value.sstr.pack4, value.sstr.unpack4 },
    };

    inline for (impls, 0..) |impl, i| {
        const pack, const unpack = impl;

        const ss1 = pack("1");
        const ss2 = pack("123");
        const ss3 = pack("123456");

        const s1, const l1 = unpack(ss1);
        const s2, const l2 = unpack(ss2);
        const s3, const l3 = unpack(ss3);

        try std.testing.expect(value.sstr.check(ss1));
        try std.testing.expect(value.sstr.check(ss2));
        try std.testing.expect(value.sstr.check(ss3));

        try std.testing.expectEqual(1, l1);
        try std.testing.expectEqualStrings("1", s1[0..l1]);

        try std.testing.expectEqual(3, l2);
        try std.testing.expectEqualStrings("123", s2[0..l2]);

        try std.testing.expectEqual(6, l3);
        try std.testing.expectEqualStrings("123456", s3[0..l3]);

        var timer = try std.time.Timer.start();
        var ns: f64 = undefined;
        var secs: f64 = undefined;

        const iters = 1;
        // const iters = 10_000_000; // standard
        // const iters = 1_000_000_000; // ReleaseFast
        if (iters > 1) {
            for (0..iters) |_i| {
                _ = _i;
                std.mem.doNotOptimizeAway(pack("1"));
                std.mem.doNotOptimizeAway(pack("123"));
                std.mem.doNotOptimizeAway(pack("123456"));
            }

            ns = @floatFromInt(timer.lap());
            secs = ns / 1_000_000_000;

            std.debug.print("pack{}: {d:.3}s\t", .{ i, secs });

            for (0..iters) |_i| {
                _ = _i;
                std.mem.doNotOptimizeAway(unpack(ss1));
                std.mem.doNotOptimizeAway(unpack(ss2));
                std.mem.doNotOptimizeAway(unpack(ss3));
            }

            ns = @floatFromInt(timer.lap());
            secs = ns / 1_000_000_000;

            std.debug.print("unpack{}: {d:.3}s\n", .{ i, secs });
        }
    }
}

test "char" {
    const c1 = value.char.pack('\x00');
    try std.testing.expect(value.char.check(c1));
    try std.testing.expectEqual('\x00', value.char.unpack(c1));

    const c2 = value.char.pack('😀');
    try std.testing.expect(value.char.check(c2));
    try std.testing.expectEqual('😀', value.char.unpack(c2));
}

test "misc" {
    const f = value.boole.pack(false);
    try std.testing.expect(value.boole.check(f));
    try std.testing.expectEqual(false, value.boole.unpack(f));
    try std.testing.expect(value.boole.unpack(value.boole.pred(f)));

    const t = value.boole.pack(true);
    try std.testing.expect(value.boole.check(t));
    try std.testing.expectEqual(true, value.boole.unpack(t));
    try std.testing.expect(value.boole.unpack(value.boole.pred(t)));

    const nil = value.nil.get();
    try std.testing.expect(value.nil.check(nil));
    try std.testing.expect(value.boole.unpack(value.nil.pred(nil)));

    const eof = value.eof.get();
    try std.testing.expect(value.eof.check(eof));
    try std.testing.expect(value.boole.unpack(value.eof.pred(eof)));
}

test "pair" {
    const v1 = value.fixnum.pack(1);
    const v2 = value.fixnum.pack(2);

    const v3 = value.fixnum.pack(3);
    const v4 = value.fixnum.pack(4);

    const p = value.pair.cons(v1, v2);
    try std.testing.expect(value.pair.check(p));
    try std.testing.expect(value.boole.unpack(value.pair.pred(p)));

    const car = value.pair.car(p);
    const cdr = value.pair.cdr(p);
    try std.testing.expectEqual(1, value.fixnum.unpack(car));
    try std.testing.expectEqual(2, value.fixnum.unpack(cdr));

    value.pair.setcar(p, v3);
    value.pair.setcdr(p, v4);

    const car2 = value.pair.car(p);
    const cdr2 = value.pair.cdr(p);
    try std.testing.expectEqual(3, value.fixnum.unpack(car2));
    try std.testing.expectEqual(4, value.fixnum.unpack(cdr2));
}

test "read" {
    const val = read.read("\"foo\"");
    const s, const l = value.sstr.unpack(value.pair.car(value.pair.cdr(val)));
    try std.testing.expectEqualStrings("foo", s[0..l]);
    try std.testing.expectEqual(3, l);
}