tiny.choir.compiler

1const std = @import("std");2const choir = @import("../root.zig");3const ir = choir.ir;4const dialects = choir.dialects;5const extensions = choir.extensions;67pub const default_compiler_dialects = [_][]const u8{8    "builtin",9    "arith",10    "memref",11    "scf",12    "func",13};1415pub const ContextOptions = struct {16    context_limits: ir.Context.Limits,17    packages: []const extensions.PackageExtension = &.{},18    preload_dialects: []const []const u8 = &default_compiler_dialects,19    preload_all_choir_dialects: bool = false,20};2122pub const Session = struct {23    allocator: std.mem.Allocator,24    ctx: ir.Context,25    extension_registry: extensions.ExtensionRegistry,26    live: bool = true,2728    pub fn init(allocator: std.mem.Allocator, options: ContextOptions) !Session {29        var ctx = try ir.Context.init(allocator, options.context_limits);30        errdefer ctx.deinit(allocator);3132        try registerCoreDialects(&ctx);3334        var extension_registry = extensions.ExtensionRegistry.init(allocator);35        errdefer extension_registry.deinit();36        for (options.packages) |package| {37            try extension_registry.registerPackage(&ctx, package);38        }39        try preloadRequestedDialects(&ctx, options);4041        return .{42            .allocator = allocator,43            .ctx = ctx,44            .extension_registry = extension_registry,45        };46    }4748    pub fn deinit(self: *Session) void {49        if (!self.live) return;50        self.extension_registry.deinit();51        self.ctx.deinit(self.allocator);52        self.live = false;53    }5455    pub fn context(self: *Session) *ir.Context {56        return &self.ctx;57    }5859    pub fn extensionRegistry(self: *Session) *extensions.ExtensionRegistry {60        return &self.extension_registry;61    }62};6364pub fn initContext(allocator: std.mem.Allocator, options: ContextOptions) !ir.Context {65    var ctx = try ir.Context.init(allocator, options.context_limits);66    errdefer ctx.deinit(allocator);6768    try registerCoreDialects(&ctx);69    for (options.packages) |package| {70        try package.registerContext(&ctx);71    }72    try preloadRequestedDialects(&ctx, options);7374    return ctx;75}7677fn registerCoreDialects(ctx: *ir.Context) !void {78    try ctx.requireRegistered();79    try dialects.registerChoirDialect(ctx);80}8182fn preloadRequestedDialects(ctx: *ir.Context, options: ContextOptions) !void {83    if (options.preload_all_choir_dialects) {84        inline for (dialects.choir_registry.entries) |entry| {85            _ = try ctx.getOrLoadDialect(entry.name);86        }87    }8889    for (options.preload_dialects) |name| {90        _ = try ctx.getOrLoadDialect(name);91    }92}

1pub const compiler = @import("compiler/root.zig");