assign.go 5.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177
  1. package rules
  2. import (
  3. "fmt"
  4. "fw/cp"
  5. "fw/cp/constant"
  6. "fw/cp/constant/operation"
  7. "fw/cp/constant/statement"
  8. "fw/cp/node"
  9. "fw/rt2"
  10. "fw/rt2/context"
  11. "fw/rt2/frame"
  12. "fw/rt2/scope"
  13. "reflect"
  14. )
  15. func incSeq(f frame.Frame) (seq frame.Sequence, ret frame.WAIT) {
  16. n := rt2.NodeOf(f)
  17. op := node.New(constant.DYADIC, int(cp.SomeAdr())).(node.OperationNode)
  18. op.SetOperation(operation.PLUS)
  19. op.SetLeft(n.Left())
  20. op.SetRight(n.Right())
  21. rt2.Push(rt2.New(op), f)
  22. seq = func(f frame.Frame) (seq frame.Sequence, ret frame.WAIT) {
  23. sc := f.Domain().Discover(context.SCOPE).(scope.Manager)
  24. sc.Update(n.Left().Object().Adr(), scope.Simple(rt2.ValueOf(f)[op.Adr()]))
  25. return frame.End()
  26. }
  27. ret = frame.LATER
  28. return seq, ret
  29. }
  30. func decSeq(f frame.Frame) (seq frame.Sequence, ret frame.WAIT) {
  31. n := rt2.NodeOf(f)
  32. op := node.New(constant.DYADIC, int(cp.SomeAdr())).(node.OperationNode)
  33. op.SetOperation(operation.MINUS)
  34. op.SetLeft(n.Left())
  35. op.SetRight(n.Right())
  36. rt2.Push(rt2.New(op), f)
  37. seq = func(f frame.Frame) (seq frame.Sequence, ret frame.WAIT) {
  38. sc := f.Domain().Discover(context.SCOPE).(scope.Manager)
  39. sc.Update(n.Left().Object().Adr(), scope.Simple(rt2.ValueOf(f)[op.Adr()]))
  40. return frame.End()
  41. }
  42. ret = frame.LATER
  43. return seq, ret
  44. }
  45. func assignSeq(f frame.Frame) (seq frame.Sequence, ret frame.WAIT) {
  46. a := rt2.NodeOf(f)
  47. var left scope.Value
  48. var rightId cp.ID
  49. right := func(f frame.Frame) (seq frame.Sequence, ret frame.WAIT) {
  50. vleft := left.(scope.Variable)
  51. switch a.Right().(type) {
  52. case node.ConstantNode:
  53. seq = func(f frame.Frame) (seq frame.Sequence, ret frame.WAIT) {
  54. sc := f.Domain().Discover(context.SCOPE).(scope.Manager)
  55. vleft.Set(sc.Provide(a.Right())(nil)) //scope.Simple(a.Right().(node.ConstantNode).Data()))
  56. return frame.End()
  57. }
  58. ret = frame.NOW
  59. case node.VariableNode, node.ParameterNode:
  60. seq = func(f frame.Frame) (seq frame.Sequence, ret frame.WAIT) {
  61. sc := f.Domain().Discover(context.SCOPE).(scope.Manager)
  62. vleft.Set(sc.Select(a.Right().Object().Adr()))
  63. return frame.End()
  64. }
  65. ret = frame.NOW
  66. case node.OperationNode, node.CallNode, node.DerefNode:
  67. rt2.Push(rt2.New(a.Right()), f)
  68. rt2.Assert(f, func(f frame.Frame) (bool, int) {
  69. return rt2.ValueOf(f)[a.Right().Adr()] != nil, 61
  70. })
  71. seq = func(f frame.Frame) (seq frame.Sequence, ret frame.WAIT) {
  72. //sc := f.Domain().Discover(context.SCOPE).(scope.Manager)
  73. vleft.Set(rt2.ValueOf(f)[a.Right().Adr()])
  74. return frame.End()
  75. }
  76. ret = frame.LATER
  77. case node.IndexNode:
  78. rightId = a.Right().Adr()
  79. rt2.Push(rt2.New(a.Right()), f)
  80. seq = func(f frame.Frame) (seq frame.Sequence, ret frame.WAIT) {
  81. /* rightId.Index = new(int64)
  82. *rightId.Index = int64(rt2.DataOf(f)[a.Right()].(int32))
  83. sc := f.Domain().Discover(context.SCOPE).(scope.Manager)
  84. sc.Update(leftId, func(interface{}) interface{} {
  85. return sc.Select(rightId)
  86. }) */
  87. return frame.End()
  88. }
  89. ret = frame.LATER
  90. case node.ProcedureNode:
  91. sc := f.Domain().Discover(context.SCOPE).(scope.Manager)
  92. vleft.Set(sc.Provide(a.Right().Object())(nil))
  93. return frame.End()
  94. default:
  95. fmt.Println(reflect.TypeOf(a.Right()))
  96. panic("wrong right")
  97. }
  98. return seq, ret
  99. }
  100. switch a.(node.AssignNode).Statement() {
  101. case statement.ASSIGN:
  102. switch l := a.Left().(type) {
  103. case node.VariableNode, node.ParameterNode:
  104. sc := f.Domain().Discover(context.SCOPE).(scope.Manager)
  105. left = sc.Select(a.Left().Object().Adr())
  106. seq, ret = right(f)
  107. case node.FieldNode:
  108. rt2.Push(rt2.New(l), f)
  109. rt2.Assert(f, func(f frame.Frame) (bool, int) {
  110. return rt2.ValueOf(f)[l.Adr()] != nil, 62
  111. })
  112. seq = func(f frame.Frame) (frame.Sequence, frame.WAIT) {
  113. left = rt2.ValueOf(f)[l.Adr()]
  114. return right(f)
  115. }
  116. ret = frame.LATER
  117. case node.IndexNode:
  118. // leftId = scope.Designator(a.Left())
  119. rt2.Push(rt2.New(a.Left()), f)
  120. seq = func(f frame.Frame) (seq frame.Sequence, ret frame.WAIT) {
  121. /* leftId.Index = new(int64)
  122. *leftId.Index = int64(rt2.DataOf(f)[a.Left()].(int32))*/
  123. return right(f)
  124. }
  125. ret = frame.LATER
  126. case node.DerefNode:
  127. // rt2.DataOf(f)[a.Left()] = scope.ID{}
  128. rt2.Push(rt2.New(a.Left()), f)
  129. seq = func(f frame.Frame) (frame.Sequence, frame.WAIT) {
  130. // leftId = rt2.DataOf(f)[a.Left()].(scope.ID)
  131. return right(f)
  132. }
  133. ret = frame.LATER
  134. default:
  135. fmt.Println(reflect.TypeOf(a.Left()))
  136. panic("wrong left")
  137. }
  138. case statement.INC, statement.INCL:
  139. switch a.Left().(type) {
  140. case node.VariableNode, node.ParameterNode:
  141. seq, ret = incSeq(f)
  142. default:
  143. panic(fmt.Sprintln("wrong left", reflect.TypeOf(a.Left())))
  144. }
  145. case statement.DEC, statement.EXCL:
  146. switch a.Left().(type) {
  147. case node.VariableNode, node.ParameterNode:
  148. seq, ret = decSeq(f)
  149. default:
  150. panic(fmt.Sprintln("wrong left", reflect.TypeOf(a.Left())))
  151. }
  152. case statement.NEW:
  153. // sc := f.Domain().Discover(context.SCOPE).(scope.Manager)
  154. // heap := scope.This(f.Domain().Discover(context.HEAP))
  155. if a.Right() != nil {
  156. seq, ret = This(expectExpr(f, a.Right(), func(...IN) OUT {
  157. fmt.Println("NEW", rt2.DataOf(f)[a.Right()], "here")
  158. // sc.Update(scope.Designator(a.Left()), heap.Target().(scope.HeapAllocator).Allocate(a.Left(), rt2.DataOf(f)[a.Right()]))
  159. return End()
  160. }))
  161. } else {
  162. fmt.Println("NEW here")
  163. // sc.Update(scope.Designator(a.Left()), heap.Target().(scope.HeapAllocator).Allocate(a.Left()))
  164. return frame.End()
  165. }
  166. default:
  167. panic(fmt.Sprintln("wrong statement", a.(node.AssignNode).Statement()))
  168. }
  169. return seq, ret
  170. }