summary refs log tree commit diff stats
path: root/nimlib/pure/variants.nim
blob: 40679c779ede6d2c667f7828cf50e6e2e925329c (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
#
#
#            Nimrod's Runtime Library
#        (c) Copyright 2009 Andreas Rumpf
#
#    See the file "copying.txt", included in this
#    distribution, for details about the copyright.
#

## This module implements Nimrod's support for the ``variant`` datatype.
## `TVariant` shows how the flexibility of dynamic typing is achieved
## within a static type system.

type
  TVarType* = enum
    vtNone,
    vtBool,
    vtChar,
    vtEnum,
    vtInt,
    vtFloat,
    vtString,
    vtSet,
    vtSeq,
    vtDict
  TVariant* {.final.} = object of TObject
    case vtype: TVarType
    of vtNone: nil
    of vtBool, vtChar, vtEnum, vtInt: vint: int64
    of vtFloat: vfloat: float64
    of vtString: vstring: string
    of vtSet, vtSeq: q: seq[TVariant]
    of vtDict: d: seq[tuple[key, val: TVariant]]

iterator objectFields*[T](x: T, skipInherited: bool): tuple[
  key: string, val: TVariant] {.magic: "ObjectFields"}

proc `<>`*(x: ordinal): TVariant =
  result.kind = vtEnum
  result.vint = x

proc `<>`*(x: biggestInt): TVariant =
  result.kind = vtInt
  result.vint = x

proc `<>`*(x: char): TVariant =
  result.kind = vtChar
  result.vint = ord(x)

proc `<>`*(x: bool): TVariant =
  result.kind = vtBool
  result.vint = ord(x)

proc `<>`*(x: biggestFloat): TVariant =
  result.kind = vtFloat
  result.vfloat = x

proc `<>`*(x: string): TVariant =
  result.kind = vtString
  result.vstring = x

proc `<>`*[T](x: openArray[T]): TVariant =
  result.kind = vtSeq
  newSeq(result.q, x.len)
  for i in 0..x.len-1: result.q[i] = <>x[i]

proc `<>`*[T](x: set[T]): TVariant =
  result.kind = vtSet
  result.q = @[]
  for a in items(x): result.q.add(<>a)

proc `<>`* [T: object](x: T): TVariant {.magic: "ToVariant".}
  ## this converts a value to a variant ("boxing")

proc `><`*[T](v: TVariant, typ: T): T {.magic: "FromVariant".}

[<>5, <>67, <>"hello"]
myVar><int


proc `==`* (x, y: TVariant): bool =
  if x.vtype == y.vtype:
    case x.vtype
    of vtNone: result = true
    of vtBool, vtChar, vtEnum, vtInt: result = x.vint == y.vint
    of vtFloat: result = x.vfloat == y.vfloat
    of vtString: result = x.vstring == y.vstring
    of vtSet:
      # complicated! We check that each a in x also occurs in y and that the
      # counts are identical:
      if x.q.len == y.q.len:
        for a in items(x.q):
          block inner:
            for b in items(y.q):
              if a == b: break inner
            return false
        result = true
    of vtSeq:
      if x.q.len == y.q.len:
        for i in 0..x.q.len-1:
          if x.q[i] != y.q[i]: return false
        result = true
    of vtDict:
      # it is an ordered dict:
      if x.d.len == y.d.len:
        for i in 0..x.d.len-1:
          if x.d[i].key != y.d[i].key: return false
          if x.d[i].val != y.d[i].val: return false
        result = true

proc `[]`* (a, b: TVariant): TVariant =
  case a.vtype
  of vtSeq:
    if b.vtype in {vtBool, vtChar, vtEnum, vtInt}:
      result = a.q[b.vint]
    else:
      variantError()
  of vtDict:
    for i in 0..a.d.len-1:
      if a.d[i].key == b: return a.d[i].val
    if b.vtype in {vtBool, vtChar, vtEnum, vtInt}:
      result = a.d[b.vint].val
    variantError()
  else: variantError()

proc `[]=`* (a, b, c: TVariant) =
  case a.vtype
  of vtSeq:
    if b.vtype in {vtBool, vtChar, vtEnum, vtInt}:
      a.q[b.vint] = b
    else:
      variantError()
  of vtDict:
    for i in 0..a.d.len-1:
      if a.d[i].key == b:
        a.d[i].val = c
        return
    if b.vtype in {vtBool, vtChar, vtEnum, vtInt}:
      a.d[b.vint].val = c
    variantError()
  else: variantError()

proc `[]`* (a: TVariant, b: int): TVariant {.inline} = return a[<>b]
proc `[]`* (a: TVariant, b: string): TVariant {.inline} = return a[<>b]
proc `[]=`* (a: TVariant, b: int, c: TVariant) {.inline} = a[<>b] = c
proc `[]=`* (a: TVariant, b: string, c: TVariant) {.inline} = a[<>b] = c

proc `+`* (x, y: TVariant): TVariant =
  case x.vtype
  of vtBool, vtChar, vtEnum, vtInt:
    if y.vtype == x.vtype:
      result.vtype = x.vtype
      result.vint = x.vint + y.vint
    else:
      case y.vtype
      of vtBool, vtChar, vtEnum, vtInt:



    vint: int64
  of vtFloat: vfloat: float64
  of vtString: vstring: string
  of vtSet, vtSeq: q: seq[TVariant]
  of vtDict: d: seq[tuple[key, val: TVariant]]

proc `-`* (x, y: TVariant): TVariant
proc `*`* (x, y: TVariant): TVariant
proc `/`* (x, y: TVariant): TVariant
proc `div`* (x, y: TVariant): TVariant
proc `mod`* (x, y: TVariant): TVariant
proc `&`* (x, y: TVariant): TVariant
proc `$`* (x: TVariant): string =
  # uses JS notation

proc parseVariant*(s: string): TVariant
proc `<`* (x, y: TVariant): bool
proc `<=`* (x, y: TVariant): bool

proc hash*(x: TVariant): int =