Scala:泛型函数乘以不同类型的数字

时间:2017-04-12 13:49:19

标签: scala generics numeric implicit

我正在尝试编写通用的加权平均函数。 我想放宽对值和重量相同类型的要求。即,我想支持说:(value:Float,weight:Int)(value:Int,weight:Float)参数的序列而不只是:(value:Int,weight:Int)

为此,我首先需要实现一个带有两个通用数值并返回其产品的函数。

def times[A: Numeric, B: Numeric](x: B, y: A): (A, B) : ??? = {...}

编写签名并考虑返回类型,让我意识到我需要为Numerics定义某种层次结构以确定返回类型。即x:Float*y:Int=z:Floatx:Float*y:Double=z:Double

现在,Numeric类仅为相同类型的参数定义操作plustimes等。我想我需要实现一个类型:

class NumericConverter[Numeirc[A],Numeric[B]]{
type BiggerType=???
}

这样我就可以将我的时间函数写成:

def times[A: Numeric, B: Numeric](x: B, y: A): (A, B) :
NumericConverter[Numeirc[A],Numeric[B]].BiggerType= {...}

并转换"较小的类型"到了更大的一个"并将其提供给times()

我是否在正确的轨道上?我将如何实施" BiggerType

显然,我无法做到这样的事情:

type myType = if(...) Int else Float

因为它是动态评估的,所以它已经磨损了。

我明白我可能会这样做使用Scalaz等,但这是一个学术练习,我想理解如何编写一个静态返回基于参数类型的类型的函数

如果有更简单的方法,请随时告诉我。

更新

这就是我想出来的。

abstract class NumericsConvert[A: Numeric,B: Numeric]{

    def AisBiggerThanB: Boolean

    def timesA=new PartialFunction[(A,B), A] {
        override def isDefinedAt(x: (A, B)): Boolean = AisBiggerThanB
        override def apply(x: (A, B)): A = implicitly[Numeric[A]].times(x._1, x._2.asInstanceOf[A])
    }

    def timesB=new PartialFunction[(A,B), B] {
        override def isDefinedAt(x: (A, B)): Boolean = !AisBiggerThanB
        override def apply(x: (A, B)): B = implicitly[Numeric[B]].times(x._1.asInstanceOf[B], x._2)
    }
    def times: PartialFunction[(A, B), Any] = timesA orElse timesB
}

def times[A: Numeric, B: Numeric](x: B, y: A)= implicitly[NumericsConvert[A,B]].times(x,y)

这是愚蠢的,因为我将不得不为两个

创造暗示
IntDouble extends NumericsConvert[Int,Double]

DoubleInt extends NumericsConvert[Double,Int]

更不用说times的返回类型现在是Any,但无论如何,我的时间功能都会出错。我想我会在这里添加它,以防它可能有助于达成解决方案。所以附带问题:如何将一个类/函数的上下文绑定类型传递给另一个类/函数,就像我想要的那样

1 个答案:

答案 0 :(得分:9)

我认为你正在努力实现这一目标。

你需要"证据"这两个参数都是Numeric。有了这个,就让证据完成工作。 Scala将使用numeric widening,以便结果是两种接收类型中更为通用的。

def mult[T](a: T, b: T)(implicit ev:Numeric[T]): T =
  ev.times(a,b)

如果你想获得一点点发烧友,你可以提出所需的暗示。然后它更容易阅读和理解。

def mult[T: Numeric](a: T, b: T): T = {
  import Numeric.Implicits._
  a * b
}

证明:

mult(2.3f , 7)  //res0: Float = 16.1
mult(8, 2.1)    //res1: Double = 16.8
mult(3, 2)      //res2: Int = 6

有关泛型类型和数字扩展的更多信息,this question及其答案值得研究。