swift-mirror/stdlib/public/core/EmbeddedPrint.swift

//===----------------------------------------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//

import SwiftShims

// TODO: This is all a stop-gap so that at least some types are printable in
// embedded Swift, in an embedded-programming friendly way (we mainly need
// printing to not need to heap allocate).

@_extern(c, "putchar")
@discardableResult
func putchar(_: CInt) -> CInt

public func print(_ string: StaticString, terminator: StaticString = "\n") {
  var p = unsafe string.utf8Start
  while unsafe p.pointee != 0 {
    putchar(CInt(unsafe p.pointee))
    unsafe p += 1
  }
  unsafe p = terminator.utf8Start
  while unsafe p.pointee != 0 {
    putchar(CInt(unsafe p.pointee))
    unsafe p += 1
  }
}

@_disfavoredOverload
public func print(_ string: String, terminator: StaticString = "\n") {
  var string = string
  string.withUTF8 { buf in
    for unsafe c in unsafe buf {
      putchar(CInt(c))
    }
  }
  var p = unsafe terminator.utf8Start
  while unsafe p.pointee != 0 {
    putchar(CInt(unsafe p.pointee))
    unsafe p += 1
  }
}

@_disfavoredOverload
public func print(_ object: some CustomStringConvertible, terminator: StaticString = "\n") {
  var string = object.description
  string.withUTF8 { buf in
    for unsafe c in unsafe buf {
      putchar(CInt(c))
    }
  }
  var p = unsafe terminator.utf8Start
  while unsafe p.pointee != 0 {
    putchar(CInt(unsafe p.pointee))
    unsafe p += 1
  }
}

func print(_ buf: UnsafeBufferPointer<UInt8>, terminator: StaticString = "\n") {
  for c in buf {
    putchar(CInt(c))
  }
  var p = unsafe terminator.utf8Start
  while unsafe p.pointee != 0 {
    unsafe putchar(CInt(p.pointee))
    p += 1
  }
}

func printCharacters(_ buf: UnsafeRawBufferPointer) {
  for unsafe c in unsafe buf {
    putchar(CInt(c))
  }
}

func printCharacters(_ buf: UnsafeBufferPointer<UInt8>) {
  unsafe printCharacters(UnsafeRawBufferPointer(buf))
}

extension BinaryInteger {
  internal func _toStringImpl(
    _ buffer: UnsafeMutablePointer<UTF8.CodeUnit>,
    _ bufferLength: UInt,
    _ radix: Int,
    _ uppercase: Bool
  ) -> Int {
    if self == (0 as Self) {
      unsafe buffer[0] = UInt8(("0" as Unicode.Scalar).value)
      return 1
    }
    
    func _ascii(_ digit: UInt8) -> UTF8.CodeUnit {
      if digit < 10 {
        UInt8(("0" as Unicode.Scalar).value) + digit
      } else {
        UInt8(("a" as Unicode.Scalar).value) + (digit - 10)
      }
    }
    let isNegative = Self.isSigned && self < (0 as Self)
    var value = magnitude
    
    var index = Int(bufferLength - 1)
    while value != 0 {
      let (quotient, remainder) = value.quotientAndRemainder(dividingBy: Magnitude(radix))
      unsafe buffer[index] = _ascii(UInt8(truncatingIfNeeded: remainder))
      index -= 1
      value = quotient
    }
    if isNegative {
      unsafe buffer[index] = UInt8(("-" as Unicode.Scalar).value)
      index -= 1
    }
    let start = index + 1
    let end = Int(bufferLength - 1)
    let count = end - start + 1
    
    let intermediate = unsafe UnsafeBufferPointer(start: buffer.advanced(by: start), count: count)
    let destination = unsafe UnsafeMutableRawBufferPointer(start: buffer, count: Int(bufferLength))
    unsafe destination.copyMemory(from: UnsafeRawBufferPointer(intermediate))
    
    return count
  }

  func writeToStdout() {
    // Avoid withUnsafeTemporaryAllocation which is not typed-throws ready yet
    let byteCount = 64
    let stackBuffer = Builtin.stackAlloc(byteCount._builtinWordValue,
         1._builtinWordValue, 1._builtinWordValue)
    let buffer = unsafe UnsafeMutableRawBufferPointer(start: .init(stackBuffer),
        count: byteCount).baseAddress!.assumingMemoryBound(to: UInt8.self)

    let count = unsafe _toStringImpl(buffer, 64, 10, false)

    unsafe printCharacters(UnsafeBufferPointer(start: buffer, count: count))

    Builtin.stackDealloc(stackBuffer)
  }
}

public func print(_ integer: some BinaryInteger, terminator: StaticString = "\n") {
  integer.writeToStdout()
  print("", terminator: terminator)
}

public func print(_ boolean: Bool, terminator: StaticString = "\n") {
  if boolean {
    print("true", terminator: terminator)
  } else {
    print("false", terminator: terminator)
  }
}