1.2.0 simd + hasSimd macros

nimsimd.nimble

@@ -1,4 +1,4 @@
-version     = "1.1.11"
+version     = "1.2.0"
 author      = "Ryan Oldenburg"
 description = "Pleasant Nim bindings for SIMD instruction sets."
 license     = "MIT"

src/nimsimd/hassimd.nim

@@ -0,0 +1,118 @@
+import std/macros, std/tables
+
+var simdProcs* {.compiletime.}: Table[string, NimNode]
+
+proc procName(procedure: NimNode): string =
+  ## Given a procedure this returns the name as a string.
+  let nameNode = procedure[0]
+  if nameNode.kind == nnkPostfix:
+    nameNode[1].strVal
+  else:
+    nameNode.strVal
+
+proc procArguments(procedure: NimNode): seq[NimNode] =
+  ## Given a procedure this gets the arguments as a list.
+  for i, arg in procedure[3]:
+    if i > 0:
+      for j in 0 ..< arg.len - 2:
+        result.add(arg[j])
+
+proc procReturnType(procedure: NimNode): NimNode =
+  ## Given a procedure this gets the return type.
+  procedure[3][0]
+
+proc procSignature(procedure: NimNode): string =
+  ## Given a procedure this returns the signature as a string.
+  result = "("
+
+  for i, arg in procedure[3]:
+    if i > 0:
+      for j in 0 ..< arg.len - 2:
+        result &= arg[^2].repr & ", "
+
+  if procedure[3].len > 1:
+    result = result[0 ..^ 3]
+
+  result &= ")"
+
+  let ret = procedure.procReturnType()
+  if ret.kind != nnkEmpty:
+    result &= ": " & ret.repr
+
+proc callAndReturn(name: NimNode, procedure: NimNode): NimNode =
+  ## Produces a procedure call with arguments.
+  let
+    retType = procedure.procReturnType()
+    call = newNimNode(nnkCall)
+  call.add(name)
+  for arg in procedure.procArguments():
+    call.add(arg)
+  if retType.kind == nnkEmpty:
+    result = quote do:
+      `call`
+      return
+  else:
+    result = quote do:
+      return `call`
+
+macro simd*(procedure: untyped) =
+  let signature = procedure.procName() & procSignature(procedure)
+  simdProcs[signature] = procedure.copy()
+  return procedure
+
+macro hasSimd*(procedure: untyped) =
+  let
+    name = procedure.procName()
+    nameNeon = name & "Neon"
+    nameSse2 = name & "Sse2"
+    nameAvx = name & "Avx"
+    nameAvx2 = name & "Avx2"
+    callNeon = callAndReturn(ident(nameNeon), procedure)
+    callSse2 = callAndReturn(ident(nameSse2), procedure)
+    callAvx = callAndReturn(ident(nameAvx), procedure)
+    callAvx2 = callAndReturn(ident(nameAvx2), procedure)
+
+  var
+    foundSimd: bool
+
+  if procedure[6].kind != nnkStmtList:
+    error("hasSimd proc body must start with nnkStmtList")
+
+  var insertIdx = 0
+  if procedure[6][0].kind == nnkCommentStmt:
+    insertIdx = 1
+
+  when defined(amd64):
+    if nameAvx2 & procSignature(procedure) in simdProcs:
+      foundSimd = true
+      procedure[6].insert(insertIdx, quote do:
+        if cpuHasAvx2:
+          `callAvx2`
+      )
+      inc insertIdx
+    if nameAvx & procSignature(procedure) in simdProcs:
+      foundSimd = true
+      procedure[6].insert(insertIdx, quote do:
+        if cpuHasAvx:
+          `callAvx`
+      )
+      inc insertIdx
+    if nameSse2 & procSignature(procedure) in simdProcs:
+      foundSimd = true
+      procedure[6].insert(insertIdx, quote do:
+        `callSse2`
+      )
+      inc insertIdx
+      while procedure[6].len > insertIdx:
+        procedure[6].del(insertIdx)
+  elif defined(arm64):
+    if nameNeon & procSignature(procedure) in simdProcs:
+      foundSimd = true
+      procedure[6].insert(insertIdx, quote do:
+        `callNeon`
+      )
+      inc insertIdx
+      while procedure[6].len > insertIdx:
+        procedure[6].del(insertIdx)
+
+  return procedure