diff options
Diffstat (limited to 'mavlink-library/src/main/scala/com/github/jodersky/mavlink/Generator.scala')
-rw-r--r-- | mavlink-library/src/main/scala/com/github/jodersky/mavlink/Generator.scala | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/mavlink-library/src/main/scala/com/github/jodersky/mavlink/Generator.scala b/mavlink-library/src/main/scala/com/github/jodersky/mavlink/Generator.scala index ff7970d..71e0f54 100644 --- a/mavlink-library/src/main/scala/com/github/jodersky/mavlink/Generator.scala +++ b/mavlink-library/src/main/scala/com/github/jodersky/mavlink/Generator.scala @@ -10,8 +10,9 @@ import trees._ /** * Generates Scala code implementing the MAVLink protocol. * @param dialect a specific MAVLink dialect for which to generate code + * @param name name of the dialect */ -class Generator(dialect: Dialect) { +class Generator(dialect: Dialect, name: String) { import Generator._ lazy val maxPayloadLength = dialect.messages.map(_.length).max @@ -29,7 +30,10 @@ class Generator(dialect: Dialect) { case class Target(path: String, generate: () => String) def targets: List[Target] = { - val context = Context(dialect.version) + val context = Context( + dialect.version, + name + ) List( Target(targetFiles(0), () => org.mavlink.txt.Assembler(context).body), Target(targetFiles(1), () => org.mavlink.txt.Crc(context).body), |