tiny.chant.ast.types.shape

1pub const shape = shape_mod;

1const std = @import("std");2const types = @import("types.zig");34const Kind = types.Kind;5const Type = types.Type;67pub fn isPointerLike(c_type: *const Type) bool {8    return c_type.kind == .pointer or c_type.kind == .array;9}1011pub fn element(c_type: *const Type) ?*const Type {12    return switch (c_type.kind) {13        .pointer, .array => c_type.child,14        else => null,15    };16}1718test "pointer-like types expose their element" {19    const pointer = Type{ .kind = .pointer, .child = &types.double_type };20    const array = Type{ .kind = .array, .child = &types.int_type, .array_len = 4 };2122    try std.testing.expect(isPointerLike(&pointer));23    try std.testing.expect(isPointerLike(&array));24    try std.testing.expectEqual(Kind.double_type, element(&pointer).?.kind);25    try std.testing.expectEqual(Kind.int_type, element(&array).?.kind);26    try std.testing.expect(element(&types.int_type) == null);27}