Vector [N]的编解码器,其中N确定向量的结束

时间:2017-08-10 04:05:37

标签: scodec

我正在使用Scodec来解码Flac元数据。其中一个规范是有一个Header和一个Block可以一起重复多次。标头有一个标志,指示当前的标题/块组合是否是最后一个。

我已经能够解码Header和Block,但我们如何根据此规范创建Vector。

这是分解的代码

  //isLastBlock determines if this is the last Header/Block combo to decode.
  case class Header(isLastBlock: Boolean)
  //Some example data.
  case class Block(someData: Int)

  object Codec {
    //Codec for Header
    val headerCodec : Codec[Header] = bool.as[Header]
    //Coded for Block
    val blockCodec: Codec[Block] = int32.as[Block]

    //We are guaranteed at least one Header/Block Combo, but how can we do this?
    val headerBlock: Codec[(Header, Block, Vector[(Header, Block)])] = ???
  }

不确定scodec是否提供此功能。 2个方法vectorOfN和sizedVector不起作用,因为它们需要在解码之前知道项目数。

1 个答案:

答案 0 :(得分:0)

我目前正在做同样的练习 - 很有兴趣知道你是如何进步的。

我也在尝试使用VectorTerminatedByIsLastIndicator问题,尽管采用的方法不同。

这是我创建的一些测试代码(对于Bar读取MetadataBlockHeader)。请注意,MetadataBlockHeader的case类不包含isLast指示符 - 它只是在编码期间添加,并在解码期间删除。

case class Bar(value: Int)

case class Foo(list1: List[Bar], list2: List[Bar])

object Foo {

  def main(args: Array[String]) : Unit = {

    implicit val barCodec : Codec[Bar] = {
      ("value" | int32).hlist
    }.as[Bar]

    implicit val barListCodec : Codec[List[Bar]] = new Codec[List[Bar]] {

      case class IsLast[A](isLast: Boolean, thing: A)

      implicit val lastBarCodec : Codec[IsLast[Bar]] = {
        ("isLast" | bool) :: ("bar" | Codec[Bar])
      }.as[IsLast[Bar]]

      override def sizeBound: SizeBound = SizeBound.unknown

      override def encode(bars: List[Bar]): Attempt[BitVector] = {
        if (bars.size == 0) {
          Failure(Err("Cannot encode zero length list"))
        } else {
          val zippedBars = bars.zipWithIndex
          val lastBars = zippedBars.map(bi => IsLast(bi._2 + 1 == bars.size, bi._1))
          Codec.encodeSeq(lastBarCodec)(lastBars)
        }
      }

      override def decode(b: BitVector): Attempt[DecodeResult[List[Bar]]] = {
        val lastBars = decode(b, List.empty)
        val bars = lastBars.map(dr => dr.value.thing)
        Successful(DecodeResult(bars, lastBars.last.remainder))
      }

      @tailrec
      private def decode(b: BitVector, lastBars: List[DecodeResult[IsLast[Bar]]]) : List[DecodeResult[IsLast[Bar]]] = {
        val lastBar = lastBarCodec.decode(b).require
        val lastBarsInterim = lastBars :+ lastBar
        if (lastBar.value.isLast) lastBarsInterim
        else decode(lastBar.remainder, lastBarsInterim)
      }
    }

    implicit val fooCodec : Codec[Foo] = {
      (("list1" | Codec[List[Bar]])
      :: ("list2" | Codec[List[Bar]]))
    }.as[Foo]

    val bar1 = Bar(1)
    val bar2 = Bar(2)
    val bar3 = Bar(3)
    val bar4 = Bar(4)

    val aFoo = Foo(Seq(bar1, bar2).toList, Seq(bar3, bar4).toList)
    println("aFoo:       " + aFoo)

    val encodedFoo = fooCodec.encode(aFoo).require
    println("encodedFoo: " + encodedFoo)

    val decodedFoo = fooCodec.decode(encodedFoo).require.value
    println("decodedFoo: " + decodedFoo)

    assert(decodedFoo == aFoo)
  }

}

可以找到此示例的更多背景here