Question

以下C代码使用枚举和数组作为从枚举到任何内容的有效“映射”：

enum Color { ColorRed, ColorGreen, ColorBlue, ColorSize};


void f() {
  int x[ColorSize];
  x[ColorRed]   = 12;
  x[ColorGreen] = 33;
  x[ColorBlue]  = 4;
  return x[ColorGreen];
}

Scala可以实现吗？即从case类到某个东西的“map”，实现为高效数组而不是树或hashmap。然而，我希望能够仅使用不具有Int。

的特定类型进行索引

更新：简而言之，我想让Scala数组通过某种枚举（case class or Enumeration）索引。

Answer 1

object Color extends Enumeration{
  val ColorRed, ColorGreen, ColorBlue = Value
}

import Color._
def f:Map[Color.Value,Int] = 
  Map(ColorRed -> 12 , ColorGreen -> 33, ColorBlue -> 4)

Answer 2

对于小枚举，您可以“模拟”C行为：

abstract sealed class Color(val index: Int)

object Color {
  implicit def col2int(color:Color) = color.index
}

case object ColorRed extends Color(0)
case object ColorGreen extends Color(1)
case object ColorBlue extends Color(2)

...

import Color._
val array = Array(1,2,3)
array(ColorRed) = 12

但是，我怀疑这会被认为是好风格，特别是因为它不安全。使用地图是一种更好的方法，或者您可以将数组包装在处理Color indizes的专门数据结构中：

class ColorArray[T:ClassManifest] {
  val array = new Array[T] (3)
  def apply(color: Color) = array(color.index)
  def update(color: Color, value: T) = array(color.index) = value
}

...

val cArray = new ColorArray[Int]()
cArray(ColorRed) = 12
println(cArray(ColorRed))

Answer 3

如果你想要完整的C性能，你可以这样做：

trait CEnum {
private var size = 0;
def value = { size += 1; size-1 }
}

object Color extends CEnum {
  val colorRed = value 
  val colorGreen = value 
  val colorBlue = value 
  val colorSize = 3
}

import Color._

def f() = {
  val x = Array[Int](colorSize)
  x(colorRed) = 12
  x(colorGreen) = 33
  x(colorBlue) = 4
  x(colorGreen)
}

它与C＆amp; C中的方法同样不安全。同样高效。然而，这是非常不安全的。

将案例类作为Scala中的密钥的高效映射？

3 个答案: