io.kaitai.struct.languages.components.ExtraAttrs.scala Maven / Gradle / Ivy

Go to download
package io.kaitai.struct.languages.components

import io.kaitai.struct.datatype.DataType
import io.kaitai.struct.datatype.DataType._
import io.kaitai.struct.format._

/**
  * Generates list of extra attributes required to store intermediate /
  * virtual stuff for every attribute like:
  *
  * * buffered raw value byte arrays
  * * IO objects (?)
  * * unprocessed / postprocessed byte arrays
  */
object ExtraAttrs {
  def forAttr(attr: AttrLikeSpec): List[AttrSpec] =
    forAttr(attr.id, attr.dataType)

  def forAttr(id: Identifier, dataType: DataType): List[AttrSpec] = {
    dataType match {
      case bt: BytesType =>
        bt.process match {
          case None => List()
          case Some(_) =>
            val rawId = RawIdentifier(id)
            List(AttrSpec(List(), rawId, bt))
        }
      case utb: UserTypeFromBytes =>
        val rawId = RawIdentifier(id)
        List(AttrSpec(List(), rawId, utb.bytes)) ++ forAttr(rawId, utb.bytes)
      case _ =>
        List()
    }
  }
}