/*
 * Copyright (c) 2025 Huawei Technologies Co., Ltd.
 * This program is free software, you can redistribute it and/or modify it under the terms and conditions of
 * CANN Open Software License Agreement Version 2.0 (the "License").
 * Please refer to the License for details. You may not use this file except in compliance with the License.
 * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED,
 * INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.
 * See LICENSE in the root of the software repository for the full text of the License.
 */

#ifndef ASCIR_DIALECT_ASC_UTILS_UTILS_H
#define ASCIR_DIALECT_ASC_UTILS_UTILS_H

#include "mlir/Dialect/Func/IR/FuncOps.h"
#include "mlir/IR/DialectRegistry.h"
#include "mlir/IR/Dominance.h"
#include "mlir/IR/Operation.h"
#include "mlir/IR/PatternMatch.h"

namespace mlir {
namespace ascendc {

template <typename OpT>
struct HoistOpPattern : public OpRewritePattern<OpT> {
    using OpRewritePattern<OpT>::OpRewritePattern;

    virtual bool hoistable(OpT) const { return true; }

    LogicalResult matchAndRewrite(OpT op, PatternRewriter& rewriter) const override
    {
        Operation* parent = op->getParentOp();
        if (isa<func::FuncOp>(parent))
            return failure();
        if (!hoistable(op))
            return failure();
        DominanceInfo di;
        bool dominatedByOperands =
            llvm::all_of(op->getOperands(), [&](Value opnd) { return di.dominates(opnd, parent); });
        if (!dominatedByOperands)
            return failure();
        rewriter.setInsertionPoint(parent);
        rewriter.replaceOp(op, rewriter.clone(*op.getOperation())->getResults());
        return success();
    }
};

bool opPrecedes(Operation* lhs, Operation* rhs);

bool opPrecedes(Operation* lhs, Operation* rhs, DominanceInfo& di);

void registerInlinerInterfaces(DialectRegistry& registry);

} // namespace ascendc
} // namespace mlir

#endif // ASCIR_DIALECT_ASC_UTILS_UTILS_H