StructSpecialization.h

Go to the documentation of this file.

//===-- StructSpecialization.h ----------------------------------*- C++ -*-===//
//
// Part of the LLZK Project, under the Apache License v2.0.
// See LICENSE.txt for license information.
// Copyright 2026 Project LLZK
// SPDX-License-Identifier: Apache-2.0
//
//===----------------------------------------------------------------------===//
 
#pragma once
 
#include "llzk/Dialect/Struct/IR/Ops.h"
 
#include <mlir/IR/BuiltinOps.h>
#include <mlir/IR/SymbolTable.h>
 
#include <llvm/ADT/DenseMap.h>
#include <llvm/ADT/SmallVector.h>
 
#include <optional>
 
namespace llzk::polymorphic {

class StructSpecializationResolver {
  mlir::ModuleOp rootModule;
  std::optional<llvm::DenseMap<int64_t, llvm::SmallVector<component::StructDefOp, 1>>> definitions;
 
public:
  explicit StructSpecializationResolver(mlir::ModuleOp module) : rootModule(module) {}

  mlir::FailureOr<component::StructDefOp> resolve(
      mlir::SymbolTableCollection &tables, component::StructType type, int64_t id,
      mlir::Operation *site, mlir::StringRef diagnosticContext
  );
};
 
} // namespace llzk::polymorphic