mandelbrot Swift #3 program

/* The Computer Language Benchmarks Game
 https://salsa.debian.org/benchmarksgame-team/benchmarksgame/
 contributed by Ralph Ganszky
 modified for Swift 3.0 by Daniel Muellenborn
 */

import Glibc
import Dispatch

let Iter = 50

// Declare Vec8 as a tuple of 8 double
typealias Vec8 = (Double, Double, Double, Double,
   Double, Double, Double, Double)

let zero: Vec8 = (0.0, 0.0, 0.0, 0.0,
                  0.0, 0.0, 0.0, 0.0)

// Declare some basic operators on Vec8
func +(lhs: Vec8, rhs: Double) -> Vec8 {
   var res = lhs
   res.0 += rhs
   res.1 += rhs
   res.2 += rhs
   res.3 += rhs
   res.4 += rhs
   res.5 += rhs
   res.6 += rhs
   res.7 += rhs
   return res
}

func +(lhs: Vec8, rhs: Vec8) -> Vec8 {
   var res = lhs
   res.0 += rhs.0
   res.1 += rhs.1
   res.2 += rhs.2
   res.3 += rhs.3
   res.4 += rhs.4
   res.5 += rhs.5
   res.6 += rhs.6
   res.7 += rhs.7
   return res
}

func -(lhs: Vec8, rhs: Vec8) -> Vec8 {
   var res = lhs
   res.0 -= rhs.0
   res.1 -= rhs.1
   res.2 -= rhs.2
   res.3 -= rhs.3
   res.4 -= rhs.4
   res.5 -= rhs.5
   res.6 -= rhs.6
   res.7 -= rhs.7
   return res
}

func *(lhs: Vec8, rhs: Vec8) -> Vec8 {
   var res = lhs
   res.0 *= rhs.0
   res.1 *= rhs.1
   res.2 *= rhs.2
   res.3 *= rhs.3
   res.4 *= rhs.4
   res.5 *= rhs.5
   res.6 *= rhs.6
   res.7 *= rhs.7
   return res
}

func *(lhs: Double, rhs: Vec8) -> Vec8 {
   var res = rhs
   res.0 *= lhs
   res.1 *= lhs
   res.2 *= lhs
   res.3 *= lhs
   res.4 *= lhs
   res.5 *= lhs
   res.6 *= lhs
   res.7 *= lhs
   return res
}

func *(lhs: Vec8, rhs: Double) -> Vec8 {
   return rhs * lhs
}

func >(lhs: Vec8, rhs: Double) -> Bool {
   if lhs.0 > rhs &&
      lhs.1 > rhs &&
      lhs.2 > rhs &&
      lhs.3 > rhs &&
      lhs.4 > rhs &&
      lhs.5 > rhs &&
      lhs.6 > rhs &&
      lhs.7 > rhs {
      return true
   } else {
      return false
   }
}

// Calculate mandelbrot set for one Vec8 into one byte
func mand8(_ cr: Vec8, _ ci: Double) -> UInt8 {
   var Zr = zero
   var Zi = zero
   var Tr = zero
   var Ti = zero
   
   for _ in 0 ..< Iter {
      Zi = 2 * Zr * Zi + ci
      Zr = Tr - Ti + cr
      Tr = Zr * Zr
      Ti = Zi * Zi
      if Tr + Ti > 4.0 {
         break
      }
   }
   var byte: UInt8 = 0
   let t = Tr+Ti
   if t.0 <= 4.0 { byte |= 0x80 }
   if t.1 <= 4.0 { byte |= 0x40 }
   if t.2 <= 4.0 { byte |= 0x20 }
   if t.3 <= 4.0 { byte |= 0x10 }
   if t.4 <= 4.0 { byte |= 0x08 }
   if t.5 <= 4.0 { byte |= 0x04 }
   if t.6 <= 4.0 { byte |= 0x02 }
   if t.7 <= 4.0 { byte |= 0x01 }
   return byte
}

// Parse command line arguments
let n: Int

if CommandLine.arguments.count > 1 {
   n = Int(CommandLine.arguments[1]) ?? 200
} else {
   n = 200
}

let N = (n + 7) & ~0x7
let inv = 2.0 / Double(n)
var xvals = [Double](repeating: 0.0, count: N)
var yvals = [Double](repeating: 0.0, count: n)

for i in 0..<N {
   xvals[i] = Double(i) * inv - 1.5
   yvals[i] = Double(i) * inv - 1.0
}

var rows = [UInt8](repeating: 0, count: n*N/8)

let queue = DispatchQueue.global(qos: .default)
DispatchQueue.concurrentPerform(iterations: n) { y in
   let ci = yvals[y]
   for x in stride(from: 0, to: N, by: 8) {
      var cr = Vec8(xvals[x+0], xvals[x+1], xvals[x+2], xvals[x+3],
                    xvals[x+4], xvals[x+5], xvals[x+6], xvals[x+7])
      rows[y*N/8+x/8] = mand8(cr, ci)
   }
}

let header = "P4\n\(n) \(n)\n"
let headerStr = header.withCString { s in s }

let iov = UnsafeMutablePointer<iovec>.allocate(capacity: 2)
iov[0].iov_base = UnsafeMutableRawPointer(mutating: headerStr)
iov[0].iov_len = header.utf8CString.count - 1

let _ = rows.withUnsafeMutableBufferPointer {
   (p: inout UnsafeMutableBufferPointer) -> Int in
   iov[1].iov_base = UnsafeMutableRawPointer(p.baseAddress)
   return 0
}

iov[1].iov_len = rows.count
writev(STDOUT_FILENO, iov, 2)
iov.deallocate()
    

NOTES:
64-bit Ubuntu quad core
Swift version 5.1.3 (swift-5.1.3-RELEASE)
Target: x86_64-unknown-linux-gnu



Tue, 05 May 2020 20:02:22 GMT

MAKE:
/opt/src/swift-5.1.3-RELEASE-ubuntu18.04/usr/bin/swiftc mandelbrot.swift-3.swift -Ounchecked -target-cpu core2  -o mandelbrot.swift-3.swift_run
<unknown>:0: warning: argument unused during compilation: '-mcpu=core2'
mandelbrot.swift-3.swift:158:11: warning: variable 'cr' was never mutated; consider changing to 'let' constant
      var cr = Vec8(xvals[x+0], xvals[x+1], xvals[x+2], xvals[x+3],
      ~~~ ^
      let

14.16s to complete and log all make actions

COMMAND LINE:
./mandelbrot.swift-3.swift_run 16000

(BINARY) PROGRAM OUTPUT NOT SHOWN