LLZKEnforceNoOverwritePass.cpp

Go to the documentation of this file.

//===-- LLZKEnforceNoOverwritePass.cpp --------------------------*- 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
//
//===----------------------------------------------------------------------===//
//===----------------------------------------------------------------------===//
 
#include "llzk/Analysis/AnalysisUtil.h"
#include "llzk/Analysis/MemberOverwriteAnalysis.h"
#include "llzk/Dialect/Function/IR/Ops.h"
#include "llzk/Dialect/Struct/IR/Ops.h"
#include "llzk/Transforms/LLZKTransformationPasses.h"
 
#include <mlir/Analysis/DataFlowFramework.h>
 
#include <llvm/Support/Debug.h>
 
#include <memory>
 
namespace llzk {
#define GEN_PASS_DEF_ENFORCENOMEMBEROVERWRITEPASS
#include "llzk/Transforms/LLZKTransformationPasses.h.inc"
} // namespace llzk
 
#define DEBUG_TYPE "llzk-enforce-no-overwrites-pass"
 
namespace {
 
using namespace llzk;
 
class PassImpl : public llzk::impl::EnforceNoMemberOverwritePassBase<PassImpl> {
  using Base = EnforceNoMemberOverwritePassBase<PassImpl>;
  using Base::Base;
 
  void runOnOperation() override {
    getOperation()->walk([this](component::StructDefOp structDef) {
      auto result = analyzeStruct(structDef);
      if (failed(result)) {
        signalPassFailure();
      }
      const auto &[overwrites, written] = *result;
 
      for (auto member : structDef.getMemberDefs()) {
        if (!written.contains(member.getSymName())) {
          signalPassFailure();
        }
      }
 
      if (!overwrites.empty()) {
        signalPassFailure();
      }
    });
  }
};
 
} // namespace