Swift 5:将两个泛型函数与 <T:BinaryInteger> 和 <T:BinaryFloatingPoint> 合并,并指定输出类型



我想做什么

我的最终目标是让数组数据(如[Int][UInt][Float]传递一个泛型函数,该函数转换任何这些类型并输出像素数据的[UInt8]0-255 值。 这样我就可以制作一个指向此UInt8数据的指针,从此指针创建一个CFData(内置类型),然后最终将CFData提供给CGDataProvider并将其作为输入以从CGImage.init()生成灰度图像。(我认为这个过程可能更简单,但这就是我想通的方式。

我将拥有的阅读功能如下所示

mutating func fromfileBI<T : BinaryInteger>(count : Int = 1) -> [T]?  {
let byteSize = MemoryLayout<T>.stride
var swiftTypeOut : [T] = []
if count > 1 {
swiftTypeOut = [T].init(repeating: 0, count: count)
for i in 0...count-1 {
swiftTypeOut[i] = data.subdata(in: off + i * byteSize..<off + (i + 1) * byteSize ).withUnsafeBytes{ $0.load(as: T.self )}
}
off += count * byteSize
}
else if count == 1 {
swiftTypeOut =  [ data.subdata(in: off..<(off+1) * byteSize ).withUnsafeBytes{ $0.load(as: T.self )} ]
off += byteSize
}
else if count == 0 {

return []
} else if count < 0 {
print("Warning, fRead.int8Read( count : Int = 1) called with a negative count, returning empty array.")
return []
}
return swiftTypeOut
}

然而,BinaryFloatingPoint(BF) 也有一个相同的主体函数。

mutating func fromfileBF<T : BinaryFloatingPoint>(count : Int = 1) -> [T]?

以及格式化图像数据的功能(最后一步是我所关心的......所有数据都将进入图像,而无需共享数据的原始类型(数据将从此函数返回 [UInt8])。

下面是BinaryInteger符合类型的一个:

func formatArrayDataforImage<T>(dataSet : Array< T >, count:  Int, numpyTypeName : String = "") -> [UInt8]! where T: BinaryInteger {

var f32Data = [Float](repeating: 0, count: count)
let obj = f32Data as NSObject // Object to synchronise
var UInt8BytesOut = [UInt8](repeating: 0, count: count)
DispatchQueue.concurrentPerform(iterations: count){ index in
synchronized(obj) {
let dataSetValue = dataSet[index]
f32Data[index] = Float( dataSetValue )
}
} // This DispatchQueue thing was necessary when I was dealing with pointers as input, Now I dont know.
guard let max = f32Data.max() else { fatalError("Find max fail.")}
guard let min = f32Data.min() else { fatalError("find min fail.")}
for i in 0..<f32Data.count {
f32Data[i] -= min
f32Data[i] =  255 * f32Data[i] / max
UInt8BytesOut[i] = UInt8( f32Data[i] )
if i<100 {
print(UInt8BytesOut[i])
}
}
return UInt8BytesOut

} 我想其他formatArrayDataforImage<T: BinaryFloatingPoint>只会有所不同Float(dataSetValue)哪些也应该起作用,我认为Float32可以投射到FloatFloat8等。但是函数调用要求一致性BinaryInteger所以我们会看到。

漫步:(跳过它仅用于上下文)––––––––

我已经修复了一个读取字节并将其输出为所需类型的函数,我已经硬编码了一直到将一个数据集加载为类型 [UInt16],这很有效。 但我从一开始就知道,我必须学习如何使类型的处理完全自动化。 (项目代码有点尴尬...我知道我必须打开从源文件中读取的字符串值才能确定 Swift 数据类型一次。但我最终在阅读时经常需要这个开关,现在我可能不得不再次对开关进行硬编码。我将如何让调用它的函数知道 Swift 数据类型? 我不想使用单独的情况,例如打开数据类型然后强制转换var example_variable = dataWhosTypeIKnowOnly_FromRawString as [Int32] // <--I know your type from hardcoding the switch我怀疑我的问题的答案隐藏在这个答案的某个地方——C# 泛型:将泛型类型转换为值类型 但鉴于我正在处理希望为创建指针指定我的泛型类型数组,我不确定这会如何改变事情。

––––

END希望一劳永逸地学习如何清楚地了解正确的 Swift 类型处理。

如果我理解正确的话,你只想formatArrayDataforImage接受任何BinaryInteger或BinaryFloatingPoint数组,而无需重写整个内容。

为此,请将其写入浮点数,然后从整数版本调用它。

我相信这是您的浮点版本:

func formatArrayDataforImage<Element>(dataSet: [Element]) -> [UInt8]
where Element: BinaryFloatingPoint {
guard let max = dataSet.max() else { fatalError("Find max fail.")}
guard let min = dataSet.min() else { fatalError("find min fail.")}
// Note this is a little dangerous since it it crash if
// min and max aren't in the range 0 - 1. I'd probably add
// assertions at least.
return dataSet.map { UInt8(255 * ($0 - min) / max) }
}

没有特别的理由将其转换为浮动。它适用于任何BinaryFloatingPoint。

然后,要为 BinaryInteger 调用它,您只需要将值映射到浮点类型(如 Float),就像您当前所做的那样。

func formatArrayDataforImage<Element>(dataSet: [Element]) -> [UInt8]
where Element: BinaryInteger  // <=== Note different `where` clause
{
// Since this creates a [Float] it will call the other function
formatArrayDataforImage(dataSet: dataSet.map(Float.init))
}

请注意,您的 concurrentPerformance 正在尝试并行执行非法操作(即使您正在修改不同的索引,您也不能同时修改多个线程上的数组)。事实证明这并不重要,因为您的同步只会再次使并发执行串行。所有这些代码都等同于我的dataSet.map(Float.init).

最新更新