tmove_objconstr.nim 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195
  1. discard """
  2. output: '''test created
  3. test destroyed 0
  4. Pony is dying!
  5. 1
  6. 2
  7. 3
  8. 4
  9. Pony is dying!'''
  10. joinable: false
  11. target: "C"
  12. """
  13. # bug #4214
  14. type
  15. Data = object
  16. data: string
  17. rc: int
  18. proc `=destroy`(d: var Data) =
  19. dec d.rc
  20. echo d.data, " destroyed ", d.rc
  21. proc `=`(dst: var Data, src: Data) =
  22. echo src.data, " copied"
  23. dst.data = src.data & " (copy)"
  24. dec dst.rc
  25. inc dst.rc
  26. proc initData(s: string): Data =
  27. result = Data(data: s, rc: 1)
  28. echo s, " created"
  29. proc pointlessWrapper(s: string): Data =
  30. result = initData(s)
  31. proc main =
  32. var x = pointlessWrapper"test"
  33. when true:
  34. main()
  35. # bug #985
  36. type
  37. Pony = object
  38. name: string
  39. proc `=destroy`(o: var Pony) =
  40. echo "Pony is dying!"
  41. proc getPony: Pony =
  42. result.name = "Sparkles"
  43. iterator items(p: Pony): int =
  44. for i in 1..4:
  45. yield i
  46. for x in getPony():
  47. echo x
  48. #------------------------------------------------------------
  49. #-- Move into tuple constructor and move on tuple unpacking
  50. #------------------------------------------------------------
  51. type
  52. MySeqNonCopyable* = object
  53. len: int
  54. data: ptr UncheckedArray[float]
  55. proc `=destroy`*(m: var MySeqNonCopyable) {.inline.} =
  56. if m.data != nil:
  57. deallocShared(m.data)
  58. m.data = nil
  59. proc `=`*(m: var MySeqNonCopyable, m2: MySeqNonCopyable) {.error.}
  60. proc `=sink`*(m: var MySeqNonCopyable, m2: MySeqNonCopyable) {.inline.} =
  61. if m.data != m2.data:
  62. if m.data != nil:
  63. `=destroy`(m)
  64. m.len = m2.len
  65. m.data = m2.data
  66. proc len*(m: MySeqNonCopyable): int {.inline.} = m.len
  67. proc `[]`*(m: MySeqNonCopyable; i: int): float {.inline.} =
  68. m.data[i.int]
  69. proc `[]=`*(m: var MySeqNonCopyable; i, val: float) {.inline.} =
  70. m.data[i.int] = val
  71. proc setTo(s: var MySeqNonCopyable, val: float) =
  72. for i in 0..<s.len.int:
  73. s.data[i] = val
  74. proc newMySeq*(size: int, initial_value = 0.0): MySeqNonCopyable =#
  75. result.len = size
  76. if size > 0:
  77. result.data = cast[ptr UncheckedArray[float]](createShared(float, size))
  78. result.setTo(initial_value)
  79. proc myfunc(x, y: int): (MySeqNonCopyable, MySeqNonCopyable) =
  80. result = (newMySeq(x, 1.0), newMySeq(y, 5.0))
  81. proc myfunc2(x, y: int): tuple[a: MySeqNonCopyable, b:int, c:MySeqNonCopyable] =
  82. var cc = newMySeq(y, 5.0)
  83. (a: case x:
  84. of 1:
  85. let (z1, z2) = myfunc(x,y)
  86. z2
  87. elif x > 5: raise newException(ValueError, "new error")
  88. else: newMySeq(x, 1.0),
  89. b: 0,
  90. c: block:
  91. var tmp = if y > 0: move(cc) else: newMySeq(1, 3.0)
  92. tmp[0] = 5
  93. tmp
  94. )
  95. let (seq1, seq2) = myfunc(2, 3)
  96. doAssert seq1.len == 2
  97. doAssert seq1[0] == 1.0
  98. doAssert seq2.len == 3
  99. doAssert seq2[0] == 5.0
  100. var (seq3, i, _) = myfunc2(2, 3)
  101. doAssert seq3.len == 2
  102. doAssert seq3[0] == 1.0
  103. var seq4, seq5: MySeqNonCopyable
  104. (seq4, i, seq5) = myfunc2(2, 3)
  105. seq4 = block:
  106. var tmp = newMySeq(4, 1.0)
  107. tmp[0] = 3.0
  108. tmp
  109. doAssert seq4[0] == 3.0
  110. import macros
  111. seq4 =
  112. if i > 0: newMySeq(2, 5.0)
  113. elif i < -100: raise newException(ValueError, "Parse Error")
  114. else: newMySeq(2, 3.0)
  115. seq4 =
  116. case (char) i:
  117. of 'A', {'W'..'Z'}: newMySeq(2, 5.0)
  118. of 'B': quit(-1)
  119. else:
  120. let (x1, x2, x3) = myfunc2(2, 3)
  121. x3
  122. #------------------------------------------------------------
  123. #-- Move into array constructor
  124. #------------------------------------------------------------
  125. var ii = 1
  126. let arr2 = [newMySeq(2, 5.0), if i > 1: newMySeq(3, 1.0) else: newMySeq(0, 0.0)]
  127. var seqOfSeq2 = @[newMySeq(2, 5.0), newMySeq(3, 1.0)]
  128. ## issue #10462
  129. proc myfuncLoop(x: int): MySeqNonCopyable =
  130. for i in 0..<x:
  131. var cc = newMySeq(i, 5.0)
  132. result = cc
  133. discard myfuncLoop(3)
  134. #------------------------------------------------------------
  135. # Move into table via openarray
  136. #------------------------------------------------------------
  137. type
  138. TableNonCopyable = object
  139. x: seq[(string, MySeqNonCopyable)]
  140. proc toTable(pairs: sink openArray[(string, MySeqNonCopyable)]): TableNonCopyable =
  141. discard
  142. let mytable = {"a": newMySeq(2, 5.0)}.toTable