// RUN: mlir-tblgen -gen-python-op-bindings -bind-dialect=test -I %S/../../include %s | FileCheck %s

include "mlir/IR/OpBase.td"
include "mlir/Bindings/Python/Attributes.td"

// CHECK: @_cext.register_dialect
// CHECK: class _Dialect(_ir.Dialect):
  // CHECK: DIALECT_NAMESPACE = "test"
  // CHECK: pass
def Test_Dialect : Dialect {
  let name = "test";
  let cppNamespace = "Test";
}
class TestOp<string mnemonic, list<OpTrait> traits = []> :
    Op<Test_Dialect, mnemonic, traits>;

// CHECK: @_cext.register_operation(_Dialect)
// CHECK: class AttrSizedOperandsOp(_ir.OpView):
// CHECK-LABEL: OPERATION_NAME = "test.attr_sized_operands"
def AttrSizedOperandsOp : TestOp<"attr_sized_operands",
                                 [AttrSizedOperandSegments]> {
  // CHECK: def __init__(self, variadic1, non_variadic, variadic2, loc=None, ip=None):
  // CHECK:   operands = []
  // CHECK:   results = []
  // CHECK:   attributes = {}
  // CHECK:   operand_segment_sizes = array.array('L')
  // CHECK:   operands += [*variadic1]
  // CHECK:   operand_segment_sizes.append(len(variadic1))
  // CHECK:   operands.append(non_variadic)
  // CHECK:   operand_segment_sizes.append(1)
  // CHECK:   if variadic2 is not None: operands.append(variadic2)
  // CHECK:   operand_segment_sizes.append(0 if variadic2 is None else 1)
  // CHECK:   attributes["operand_segment_sizes"] = _ir.DenseElementsAttr.get(operand_segment_sizes,
  // CHECK:       context=_get_default_loc_context(loc))
  // CHECK:   super().__init__(_ir.Operation.create(
  // CHECK:     "test.attr_sized_operands", attributes=attributes, operands=operands, results=results,
  // CHECK:     loc=loc, ip=ip))

  // CHECK: @property
  // CHECK: def variadic1(self):
  // CHECK:   operand_range = _segmented_accessor(
  // CHECK:       self.operation.operands,
  // CHECK:       self.operation.attributes["operand_segment_sizes"], 0)
  // CHECK:   return operand_range
  //
  // CHECK: @property
  // CHECK: def non_variadic(self):
  // CHECK:   operand_range = _segmented_accessor(
  // CHECK:       self.operation.operands,
  // CHECK:       self.operation.attributes["operand_segment_sizes"], 1)
  // CHECK:   return operand_range[0]
  //
  // CHECK: @property
  // CHECK: def variadic2(self):
  // CHECK:   operand_range = _segmented_accessor(
  // CHECK:       self.operation.operands,
  // CHECK:       self.operation.attributes["operand_segment_sizes"], 2)
  // CHECK:   return operand_range[0] if len(operand_range) > 0 else None
  let arguments = (ins Variadic<AnyType>:$variadic1, AnyType:$non_variadic,
                   Optional<AnyType>:$variadic2);
}

// CHECK: @_cext.register_operation(_Dialect)
// CHECK: class AttrSizedResultsOp(_ir.OpView):
// CHECK-LABEL: OPERATION_NAME = "test.attr_sized_results"
def AttrSizedResultsOp : TestOp<"attr_sized_results",
                               [AttrSizedResultSegments]> {
  // CHECK: def __init__(self, variadic1, non_variadic, variadic2, loc=None, ip=None):
  // CHECK:   operands = []
  // CHECK:   results = []
  // CHECK:   attributes = {}
  // CHECK:   result_segment_sizes = array.array('L')
  // CHECK:   if variadic1 is not None: results.append(variadic1)
  // CHECK:   result_segment_sizes.append(0 if variadic1 is None else 1)
  // CHECK:   results.append(non_variadic)
  // CHECK:   result_segment_sizes.append(1) # non_variadic
  // CHECK:   if variadic2 is not None: results.append(variadic2)
  // CHECK:   result_segment_sizes.append(0 if variadic2 is None else 1)
  // CHECK:   attributes["result_segment_sizes"] = _ir.DenseElementsAttr.get(result_segment_sizes,
  // CHECK:       context=_get_default_loc_context(loc))
  // CHECK:   super().__init__(_ir.Operation.create(
  // CHECK:     "test.attr_sized_results", attributes=attributes, operands=operands, results=results,
  // CHECK:     loc=loc, ip=ip))

  // CHECK: @property
  // CHECK: def variadic1(self):
  // CHECK:   result_range = _segmented_accessor(
  // CHECK:       self.operation.results,
  // CHECK:       self.operation.attributes["result_segment_sizes"], 0)
  // CHECK:   return result_range[0] if len(result_range) > 0 else None
  //
  // CHECK: @property
  // CHECK: def non_variadic(self):
  // CHECK:   result_range = _segmented_accessor(
  // CHECK:       self.operation.results,
  // CHECK:       self.operation.attributes["result_segment_sizes"], 1)
  // CHECK:   return result_range[0]
  //
  // CHECK: @property
  // CHECK: def variadic2(self):
  // CHECK:   result_range = _segmented_accessor(
  // CHECK:       self.operation.results,
  // CHECK:       self.operation.attributes["result_segment_sizes"], 2)
  // CHECK:   return result_range
  let results = (outs Optional<AnyType>:$variadic1, AnyType:$non_variadic,
                 Optional<AnyType>:$variadic2);
}


// CHECK: @_cext.register_operation(_Dialect)
// CHECK: class AttributedOp(_ir.OpView):
// CHECK-LABEL: OPERATION_NAME = "test.attributed_op"
def AttributedOp : TestOp<"attributed_op"> {
  // CHECK: def __init__(self, i32attr, optionalF32Attr, unitAttr, in_, loc=None, ip=None):
  // CHECK:   operands = []
  // CHECK:   results = []
  // CHECK:   attributes = {}
  // CHECK:   attributes["i32attr"] = i32attr
  // CHECK:   if optionalF32Attr is not None: attributes["optionalF32Attr"] = optionalF32Attr
  // CHECK:   if bool(unitAttr): attributes["unitAttr"] = _ir.UnitAttr.get(
  // CHECK:     _get_default_loc_context(loc))
  // CHECK:   attributes["in"] = in_
  // CHECK:   super().__init__(_ir.Operation.create(
  // CHECK:     "test.attributed_op", attributes=attributes, operands=operands, results=results,
  // CHECK:     loc=loc, ip=ip))

  // CHECK: @property
  // CHECK: def i32attr(self):
  // CHECK:   return _ir.IntegerAttr(self.operation.attributes["i32attr"])

  // CHECK: @property
  // CHECK: def optionalF32Attr(self):
  // CHECK:   if "optionalF32Attr" not in self.operation.attributes:
  // CHECK:     return None
  // CHECK:   return _ir.FloatAttr(self.operation.attributes["optionalF32Attr"])

  // CHECK: @property
  // CHECK: def unitAttr(self):
  // CHECK:   return "unitAttr" in self.operation.attributes

  // CHECK: @property
  // CHECK: def in_(self):
  // CHECK:   return _ir.IntegerAttr(self.operation.attributes["in"])
  let arguments = (ins I32Attr:$i32attr, OptionalAttr<F32Attr>:$optionalF32Attr,
                   UnitAttr:$unitAttr, I32Attr:$in);
}

// CHECK: @_cext.register_operation(_Dialect)
// CHECK: class AttributedOpWithOperands(_ir.OpView):
// CHECK-LABEL: OPERATION_NAME = "test.attributed_op_with_operands"
def AttributedOpWithOperands : TestOp<"attributed_op_with_operands"> {
  // CHECK: def __init__(self, _gen_arg_0, in_, _gen_arg_2, is_, loc=None, ip=None):
  // CHECK:   operands = []
  // CHECK:   results = []
  // CHECK:   attributes = {}
  // CHECK:   operands.append(_gen_arg_0)
  // CHECK:   operands.append(_gen_arg_2)
  // CHECK:   if bool(in_): attributes["in"] = _ir.UnitAttr.get(
  // CHECK:     _get_default_loc_context(loc))
  // CHECK:   if is_ is not None: attributes["is"] = is_
  // CHECK:   super().__init__(_ir.Operation.create(
  // CHECK:     "test.attributed_op_with_operands", attributes=attributes, operands=operands, results=results,
  // CHECK:     loc=loc, ip=ip))

  // CHECK: @property
  // CHECK: def in_(self):
  // CHECK:   return "in" in self.operation.attributes

  // CHECK: @property
  // CHECK: def is_(self):
  // CHECK:   if "is" not in self.operation.attributes:
  // CHECK:     return None
  // CHECK:   return _ir.FloatAttr(self.operation.attributes["is"])
  let arguments = (ins I32, UnitAttr:$in, F32, OptionalAttr<F32Attr>:$is);
}


// CHECK: @_cext.register_operation(_Dialect)
// CHECK: class EmptyOp(_ir.OpView):
// CHECK-LABEL: OPERATION_NAME = "test.empty"
def EmptyOp : TestOp<"empty">;
  // CHECK: def __init__(self, loc=None, ip=None):
  // CHECK:   operands = []
  // CHECK:   results = []
  // CHECK:   attributes = {}
  // CHECK:   super().__init__(_ir.Operation.create(
  // CHECK:     "test.empty", attributes=attributes, operands=operands, results=results,
  // CHECK:     loc=loc, ip=ip))

// CHECK: @_cext.register_operation(_Dialect)
// CHECK: class MissingNamesOp(_ir.OpView):
// CHECK-LABEL: OPERATION_NAME = "test.missing_names"
def MissingNamesOp : TestOp<"missing_names"> {
  // CHECK: def __init__(self, i32, _gen_res_1, i64, _gen_arg_0, f32, _gen_arg_2, loc=None, ip=None):
  // CHECK:   operands = []
  // CHECK:   results = []
  // CHECK:   attributes = {}
  // CHECK:   results.append(i32)
  // CHECK:   results.append(_gen_res_1)
  // CHECK:   results.append(i64)
  // CHECK:   operands.append(_gen_arg_0)
  // CHECK:   operands.append(f32)
  // CHECK:   operands.append(_gen_arg_2)
  // CHECK:   super().__init__(_ir.Operation.create(
  // CHECK:     "test.missing_names", attributes=attributes, operands=operands, results=results,
  // CHECK:     loc=loc, ip=ip))

  // CHECK: @property
  // CHECK: def f32(self):
  // CHECK:   return self.operation.operands[1]
  let arguments = (ins I32, F32:$f32, I64);

  // CHECK: @property
  // CHECK: def i32(self):
  // CHECK:   return self.operation.results[0]
  //
  // CHECK: @property
  // CHECK: def i64(self):
  // CHECK:   return self.operation.results[2]
  let results = (outs I32:$i32, F32, I64:$i64);
}

// CHECK: @_cext.register_operation(_Dialect)
// CHECK: class OneVariadicOperandOp(_ir.OpView):
// CHECK-LABEL: OPERATION_NAME = "test.one_variadic_operand"
def OneVariadicOperandOp : TestOp<"one_variadic_operand"> {
  // CHECK: def __init__(self, non_variadic, variadic, loc=None, ip=None):
  // CHECK:   operands = []
  // CHECK:   results = []
  // CHECK:   attributes = {}
  // CHECK:   operands.append(non_variadic)
  // CHECK:   operands += [*variadic]
  // CHECK:   super().__init__(_ir.Operation.create(
  // CHECK:     "test.one_variadic_operand", attributes=attributes, operands=operands, results=results,
  // CHECK:     loc=loc, ip=ip))

  // CHECK: @property
  // CHECK: def non_variadic(self):
  // CHECK:   return self.operation.operands[0]
  //
  // CHECK: @property
  // CHECK: def variadic(self):
  // CHECK:   variadic_group_length = len(self.operation.operands) - 2 + 1
  // CHECK:   return self.operation.operands[1:1 + variadic_group_length]
  let arguments = (ins AnyType:$non_variadic, Variadic<AnyType>:$variadic);
}

// CHECK: @_cext.register_operation(_Dialect)
// CHECK: class OneVariadicResultOp(_ir.OpView):
// CHECK-LABEL: OPERATION_NAME = "test.one_variadic_result"
def OneVariadicResultOp : TestOp<"one_variadic_result"> {
  // CHECK: def __init__(self, variadic, non_variadic, loc=None, ip=None):
  // CHECK:   operands = []
  // CHECK:   results = []
  // CHECK:   attributes = {}
  // CHECK:   results += [*variadic]
  // CHECK:   results.append(non_variadic)
  // CHECK:   super().__init__(_ir.Operation.create(
  // CHECK:     "test.one_variadic_result", attributes=attributes, operands=operands, results=results,
  // CHECK:     loc=loc, ip=ip))

  // CHECK: @property
  // CHECK: def variadic(self):
  // CHECK:   variadic_group_length = len(self.operation.results) - 2 + 1
  // CHECK:   return self.operation.results[0:0 + variadic_group_length]
  //
  // CHECK: @property
  // CHECK: def non_variadic(self):
  // CHECK:   variadic_group_length = len(self.operation.results) - 2 + 1
  // CHECK:   return self.operation.results[1 + variadic_group_length - 1]
  let results = (outs Variadic<AnyType>:$variadic, AnyType:$non_variadic);
}

// CHECK: @_cext.register_operation(_Dialect)
// CHECK: class PythonKeywordOp(_ir.OpView):
// CHECK-LABEL: OPERATION_NAME = "test.python_keyword"
def PythonKeywordOp : TestOp<"python_keyword"> {
  // CHECK: def __init__(self, in_, loc=None, ip=None):
  // CHECK:   operands = []
  // CHECK:   results = []
  // CHECK:   attributes = {}
  // CHECK:   operands.append(in_)
  // CHECK:   super().__init__(_ir.Operation.create(
  // CHECK:     "test.python_keyword", attributes=attributes, operands=operands, results=results,
  // CHECK:     loc=loc, ip=ip))

  // CHECK: @property
  // CHECK: def in_(self):
  // CHECK:   return self.operation.operands[0]
  let arguments = (ins AnyType:$in);
}

// CHECK: @_cext.register_operation(_Dialect)
// CHECK: class SameVariadicOperandSizeOp(_ir.OpView):
// CHECK-LABEL: OPERATION_NAME = "test.same_variadic_operand"
def SameVariadicOperandSizeOp : TestOp<"same_variadic_operand",
                                       [SameVariadicOperandSize]> {
  // CHECK: @property
  // CHECK: def variadic1(self):
  // CHECK:   start, pg = _equally_sized_accessor(operation.operands, 2, 0, 0)
  // CHECK:   return self.operation.operands[start:start + pg]
  //
  // CHECK: @property
  // CHECK: def non_variadic(self):
  // CHECK:   start, pg = _equally_sized_accessor(operation.operands, 2, 0, 1)
  // CHECK:   return self.operation.operands[start]
  //
  // CHECK: @property
  // CHECK: def variadic2(self):
  // CHECK:   start, pg = _equally_sized_accessor(operation.operands, 2, 1, 1)
  // CHECK:   return self.operation.operands[start:start + pg]
  let arguments = (ins Variadic<AnyType>:$variadic1, AnyType:$non_variadic,
                   Variadic<AnyType>:$variadic2);
}

// CHECK: @_cext.register_operation(_Dialect)
// CHECK: class SameVariadicResultSizeOp(_ir.OpView):
// CHECK-LABEL: OPERATION_NAME = "test.same_variadic_result"
def SameVariadicResultSizeOp : TestOp<"same_variadic_result",
                                      [SameVariadicResultSize]> {
  // CHECK: @property
  // CHECK: def variadic1(self):
  // CHECK:   start, pg = _equally_sized_accessor(operation.results, 2, 0, 0)
  // CHECK:   return self.operation.results[start:start + pg]
  //
  // CHECK: @property
  // CHECK: def non_variadic(self):
  // CHECK:   start, pg = _equally_sized_accessor(operation.results, 2, 0, 1)
  // CHECK:   return self.operation.results[start]
  //
  // CHECK: @property
  // CHECK: def variadic2(self):
  // CHECK:   start, pg = _equally_sized_accessor(operation.results, 2, 1, 1)
  // CHECK:   return self.operation.results[start:start + pg]
  let results = (outs Variadic<AnyType>:$variadic1, AnyType:$non_variadic,
                 Variadic<AnyType>:$variadic2);
}

// CHECK: @_cext.register_operation(_Dialect)
// CHECK: class SimpleOp(_ir.OpView):
// CHECK-LABEL: OPERATION_NAME = "test.simple"
def SimpleOp : TestOp<"simple"> {
  // CHECK: def __init__(self, i64, f64, i32, f32, loc=None, ip=None):
  // CHECK:   operands = []
  // CHECK:   results = []
  // CHECK:   attributes = {}
  // CHECK:   results.append(i64)
  // CHECK:   results.append(f64)
  // CHECK:   operands.append(i32)
  // CHECK:   operands.append(f32)
  // CHECK:   super().__init__(_ir.Operation.create(
  // CHECK:     "test.simple", attributes=attributes, operands=operands, results=results,
  // CHECK:     loc=loc, ip=ip))

  // CHECK: @property
  // CHECK: def i32(self):
  // CHECK:   return self.operation.operands[0]
  //
  // CHECK: @property
  // CHECK: def f32(self):
  // CHECK:   return self.operation.operands[1]
  let arguments = (ins I32:$i32, F32:$f32);

  // CHECK: @property
  // CHECK: def i64(self):
  // CHECK:   return self.operation.results[0]
  //
  // CHECK: @property
  // CHECK: def f64(self):
  // CHECK:   return self.operation.results[1]
  let results = (outs I64:$i64, F64:$f64);
}