//
// This file is distributed under the MIT License. See LICENSE.md for details.
//

#ifndef MLIR_CLIFT_OP_INTERFACES_CONTROL_FLOW
#define MLIR_CLIFT_OP_INTERFACES_CONTROL_FLOW

include "revng/Clift/CliftOpInterfacesJump.td"

def Clift_BranchOpInterface
  : Clift_OpInterface<"BranchOpInterface",
                      [Clift_StatementOpInterface,
                       Clift_StatementRegionOpInterface,
                       Clift_ExpressionRegionOpInterface]> {

  let description = [{
    Common interface for branch statements, providing generic access to
    potentially evaluated branch regions. Whenever a branch operation is
    evaluated, exactly one of its branch regions is also evaluated.
  }];

  let methods = [
    InterfaceMethod<
      /*desc=*/"Returns the region describing the branch condition.",
      "mlir::Region &", "getBranchCondition", (ins),
      /*methodBody=*/[{}],
      /*defaultImplementation=*/ [{
        return $_op.getCondition();
      }]
    >,
    InterfaceMethod<
      /*desc=*/"Returns the regions describing the branches of this operation.",
      "llvm::MutableArrayRef<mlir::Region>", "getBranchRegions", (ins),
      /*methodBody=*/[{}],
      /*defaultImplementation=*/ [{
        return $_op->getRegions().drop_front(1);
      }]
    >,
  ];
}

def Clift_LoopOpInterface
  : Clift_OpInterface<"LoopOpInterface", [Clift_LabelAssignmentOpInterface,
                                          Clift_StatementRegionOpInterface,
                                          Clift_ExpressionRegionOpInterface]> {

  let description = [{
    Common interface for loop statements, providing generic access to assigned
    labels and the loop body region.
  }];

  let methods = [
    InterfaceMethod<
      /*desc=*/"Returns the loop break label (if any).",
      "mlir::Value", "getBreakLabel", (ins),
      /*methodBody=*/[{}],
      /*defaultImplementation=*/[{
        return clift::impl::getLoopLabel($_op, 0);
      }]
    >,
    InterfaceMethod<
      /*desc=*/"Replaces the loop break label (if any).",
      "void", "setBreakLabel", (ins "mlir::Value":$Label),
      /*methodBody=*/[{}],
      /*defaultImplementation=*/[{
        return clift::impl::setLoopLabel($_op, 0, Label);
      }]
    >,
    InterfaceMethod<
      /*desc=*/"Returns the loop continue label (if any).",
      "mlir::Value", "getContinueLabel", (ins),
      /*methodBody=*/[{}],
      /*defaultImplementation=*/[{
        return clift::impl::getLoopLabel($_op, 1);
      }]
    >,
    InterfaceMethod<
      /*desc=*/"Replaces the loop continue label (if any).",
      "void", "setContinueLabel", (ins "mlir::Value":$Label),
      /*methodBody=*/[{}],
      /*defaultImplementation=*/[{
        return clift::impl::setLoopLabel($_op, 1, Label);
      }]
    >,
    InterfaceMethod<
      /*desc=*/"Returns the loop body region.",
      "mlir::Region &", "getBody", (ins),
      /*methodBody=*/[{}],
      /*defaultImplementation=*/[{
        return $_op.getBody();
      }]
    >,
  ];
}

#endif