uk.co.real_logic.sbe.generation.rust.RustGenerator Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of sbe-tool Show documentation
Show all versions of sbe-tool Show documentation
FIX/SBE - OSI layer 6 presentation for encoding and decoding application messages in binary format for low-latency applications.
The newest version!
/*
* Copyright 2013-2024 Real Logic Limited.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package uk.co.real_logic.sbe.generation.rust;
import org.agrona.Strings;
import org.agrona.Verify;
import org.agrona.generation.OutputManager;
import uk.co.real_logic.sbe.PrimitiveType;
import uk.co.real_logic.sbe.generation.CodeGenerator;
import uk.co.real_logic.sbe.generation.Generators;
import uk.co.real_logic.sbe.generation.java.JavaUtil;
import uk.co.real_logic.sbe.ir.Encoding;
import uk.co.real_logic.sbe.ir.Ir;
import uk.co.real_logic.sbe.ir.Signal;
import uk.co.real_logic.sbe.ir.Token;
import java.io.IOException;
import java.io.UncheckedIOException;
import java.io.Writer;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import static java.lang.String.format;
import static uk.co.real_logic.sbe.generation.rust.RustUtil.*;
import static uk.co.real_logic.sbe.ir.GenerationUtil.collectFields;
import static uk.co.real_logic.sbe.ir.GenerationUtil.collectGroups;
import static uk.co.real_logic.sbe.ir.GenerationUtil.collectVarData;
import static uk.co.real_logic.sbe.ir.Signal.BEGIN_ENUM;
import static uk.co.real_logic.sbe.ir.Signal.BEGIN_SET;
/**
* Generate codecs for the Rust programming language.
*/
public class RustGenerator implements CodeGenerator
{
static final String WRITE_BUF_TYPE = "WriteBuf";
static final String READ_BUF_TYPE = "ReadBuf";
static final String BUF_LIFETIME = "'a";
enum CodecType
{
Decoder
{
String bufType()
{
return READ_BUF_TYPE;
}
},
Encoder
{
String bufType()
{
return WRITE_BUF_TYPE;
}
};
abstract String bufType();
}
interface ParentDef
{
SubGroup addSubGroup(String name, int level, Token groupToken);
}
private final Ir ir;
private final RustOutputManager outputManager;
private final String crateVersion;
/**
* Create a new Rust language {@link CodeGenerator}.
*
* @param ir for the messages and types.
* @param crateVersion for the generated crate.
* @param outputManager for generating the codecs to.
*/
public RustGenerator(
final Ir ir,
final String crateVersion,
final OutputManager outputManager)
{
Verify.notNull(ir, "ir");
Verify.notNull(crateVersion, "crateVersion");
Verify.notNull(outputManager, "outputManager");
this.ir = ir;
this.crateVersion = crateVersion;
this.outputManager = (RustOutputManager)outputManager;
}
@Override
public void generate() throws IOException
{
// create Cargo.toml
try (Writer writer = outputManager.createCargoToml())
{
final String packageName = toLowerSnakeCase(ir.packageName()).replaceAll("[.-]", "_");
final String namespace;
if (ir.namespaceName() == null || ir.namespaceName().equalsIgnoreCase(packageName))
{
namespace = packageName.toLowerCase();
}
else
{
namespace = (ir.namespaceName() + "_" + packageName).toLowerCase();
}
indent(writer, 0, "[package]\n");
indent(writer, 0, "name = \"%s\"\n", namespace);
indent(writer, 0, "version = \"%s\"\n", crateVersion);
indent(writer, 0, "authors = [\"sbetool\"]\n");
indent(writer, 0, "description = \"%s\"\n", ir.description());
indent(writer, 0, "edition = \"2021\"\n\n");
indent(writer, 0, "[lib]\n");
indent(writer, 0, "name = \"%s\"\n", namespace);
indent(writer, 0, "path = \"src/lib.rs\"\n");
}
// lib.rs
final LibRsDef libRsDef = new LibRsDef(outputManager, ir.byteOrder(), schemaVersionType());
generateEnums(ir, outputManager);
generateBitSets(ir, outputManager);
generateComposites(schemaVersionType(), ir, outputManager);
for (final List tokens : ir.messages())
{
final Token msgToken = tokens.get(0);
final String codecModName = codecModName(msgToken.name());
final List messageBody = tokens.subList(1, tokens.size() - 1);
int i = 0;
final List fields = new ArrayList<>();
i = collectFields(messageBody, i, fields);
final List groups = new ArrayList<>();
i = collectGroups(messageBody, i, groups);
final List varData = new ArrayList<>();
collectVarData(messageBody, i, varData);
try (Writer out = outputManager.createOutput(codecModName))
{
indent(out, 0, "use crate::*;\n\n");
indent(out, 0, "pub use decoder::%sDecoder;\n", formatStructName(msgToken.name()));
indent(out, 0, "pub use encoder::%sEncoder;\n\n", formatStructName(msgToken.name()));
final String blockLengthType = blockLengthType();
final String templateIdType = rustTypeName(ir.headerStructure().templateIdType());
final String schemaIdType = rustTypeName(ir.headerStructure().schemaIdType());
final String schemaVersionType = schemaVersionType();
final String semanticVersion = ir.semanticVersion() == null ? "" : ir.semanticVersion();
indent(out, 0, "pub const SBE_BLOCK_LENGTH: %s = %d;\n", blockLengthType, msgToken.encodedLength());
indent(out, 0, "pub const SBE_TEMPLATE_ID: %s = %d;\n", templateIdType, msgToken.id());
indent(out, 0, "pub const SBE_SCHEMA_ID: %s = %d;\n", schemaIdType, ir.id());
indent(out, 0, "pub const SBE_SCHEMA_VERSION: %s = %d;\n", schemaVersionType, ir.version());
indent(out, 0, "pub const SBE_SEMANTIC_VERSION: &str = \"%s\";\n\n", semanticVersion);
MessageCoderDef.generateEncoder(ir, out, msgToken, fields, groups, varData);
MessageCoderDef.generateDecoder(ir, out, msgToken, fields, groups, varData);
}
}
libRsDef.generate();
}
String blockLengthType()
{
return rustTypeName(ir.headerStructure().blockLengthType());
}
String schemaVersionType()
{
return rustTypeName(ir.headerStructure().schemaVersionType());
}
static String withLifetime(final String typeName)
{
return format("%s<%s>", typeName, BUF_LIFETIME);
}
static void appendImplWithLifetimeHeader(
final Appendable appendable,
final String typeName) throws IOException
{
indent(appendable, 1, "impl<%s> %s<%s> {\n", BUF_LIFETIME, typeName, BUF_LIFETIME);
}
static String getBufOffset(final Token token)
{
final int offset = token.offset();
if (offset > 0)
{
return "offset + " + offset;
}
return "offset";
}
static void generateEncoderFields(
final StringBuilder sb,
final List tokens,
final int level)
{
Generators.forEachField(
tokens,
(fieldToken, typeToken) ->
{
try
{
final String name = fieldToken.name();
switch (typeToken.signal())
{
case ENCODING:
generatePrimitiveEncoder(sb, level, typeToken, name);
break;
case BEGIN_ENUM:
generateEnumEncoder(sb, level, fieldToken, typeToken, name);
break;
case BEGIN_SET:
generateBitSetEncoder(sb, level, typeToken, name);
break;
case BEGIN_COMPOSITE:
generateCompositeEncoder(sb, level, typeToken, name);
break;
default:
break;
}
}
catch (final IOException ex)
{
throw new UncheckedIOException(ex);
}
});
}
static void generateEncoderGroups(
final StringBuilder sb,
final List tokens,
final int level,
final ParentDef parentDef) throws IOException
{
for (int i = 0, size = tokens.size(); i < size; i++)
{
final Token groupToken = tokens.get(i);
if (groupToken.signal() != Signal.BEGIN_GROUP)
{
throw new IllegalStateException("tokens must begin with BEGIN_GROUP: token=" + groupToken);
}
++i;
final int index = i;
final int groupHeaderTokenCount = tokens.get(i).componentTokenCount();
i += groupHeaderTokenCount;
final List fields = new ArrayList<>();
i = collectFields(tokens, i, fields);
final List groups = new ArrayList<>();
i = collectGroups(tokens, i, groups);
final List varData = new ArrayList<>();
i = collectVarData(tokens, i, varData);
final String groupName = encoderName(formatStructName(groupToken.name()));
final Token numInGroupToken = Generators.findFirst("numInGroup", tokens, index);
final PrimitiveType numInGroupPrimitiveType = numInGroupToken.encoding().primitiveType();
final String description = groupToken.description();
if (!Strings.isEmpty(description))
{
indent(sb, level, "/// GROUP ENCODER (id=%s, description='%s')\n",
groupToken.id(), description);
}
else
{
indent(sb, level, "/// GROUP ENCODER (id=%s)\n",
groupToken.id());
}
assert 4 == groupHeaderTokenCount;
indent(sb, level, "#[inline]\n");
indent(sb, level, "pub fn %s(self, count: %s, %1$s: %3$s) -> %3$s {\n",
formatFunctionName(groupName),
rustTypeName(numInGroupPrimitiveType),
groupName);
indent(sb, level + 1, "%s.wrap(self, count)\n", toLowerSnakeCase(groupName));
indent(sb, level, "}\n\n");
final SubGroup subGroup = parentDef.addSubGroup(groupName, level, groupToken);
subGroup.generateEncoder(tokens, fields, groups, varData, index);
}
}
static void generateEncoderVarData(
final StringBuilder sb,
final List tokens,
final int level) throws IOException
{
for (int i = 0, size = tokens.size(); i < size; )
{
final Token varDataToken = tokens.get(i);
if (varDataToken.signal() != Signal.BEGIN_VAR_DATA)
{
throw new IllegalStateException("tokens must begin with BEGIN_VAR_DATA: token=" + varDataToken);
}
final String characterEncoding = characterEncoding(tokens.get(i + 3).encoding());
final String propertyName = toLowerSnakeCase(varDataToken.name());
final Token lengthToken = tokens.get(i + 2);
final Encoding lengthEncoding = lengthToken.encoding();
final PrimitiveType lengthType = lengthEncoding.primitiveType();
final String varDataType;
final String toBytesFn;
if (JavaUtil.isUtf8Encoding(characterEncoding))
{
varDataType = "&str";
toBytesFn = ".as_bytes()";
}
else
{
varDataType = "&[u8]";
toBytesFn = "";
}
// function to write slice ... todo - handle character encoding ?
indent(sb, level, "/// VAR_DATA ENCODER - character encoding: '%s'\n", characterEncoding);
indent(sb, level, "#[inline]\n");
indent(sb, level, "pub fn %s(&mut self, value: %s) {\n", propertyName, varDataType);
indent(sb, level + 1, "let limit = self.get_limit();\n");
indent(sb, level + 1, "let data_length = value.len();\n");
indent(sb, level + 1, "self.set_limit(limit + %d + data_length);\n", lengthType.size());
indent(sb, level + 1,
"self.get_buf_mut().put_%s_at(limit, data_length as %1$s);\n",
rustTypeName(lengthType));
indent(sb, level + 1, "self.get_buf_mut().put_slice_at(limit + %d, value%s);\n",
lengthType.size(),
toBytesFn);
indent(sb, level, "}\n\n");
i += varDataToken.componentTokenCount();
}
}
private static void generatePrimitiveEncoder(
final StringBuilder sb,
final int level,
final Token typeToken,
final String name) throws IOException
{
final Encoding encoding = typeToken.encoding();
final PrimitiveType primitiveType = encoding.primitiveType();
final String rustPrimitiveType = rustTypeName(primitiveType);
final int arrayLength = typeToken.arrayLength();
if (arrayLength > 1)
{
indent(sb, level, "/// primitive array field '%s'\n", name);
indent(sb, level, "/// - min value: %s\n", encoding.applicableMinValue());
indent(sb, level, "/// - max value: %s\n", encoding.applicableMaxValue());
indent(sb, level, "/// - null value: %s\n", encoding.applicableNullValue());
indent(sb, level, "/// - characterEncoding: %s\n", encoding.characterEncoding());
indent(sb, level, "/// - semanticType: %s\n", encoding.semanticType());
indent(sb, level, "/// - encodedOffset: %d\n", typeToken.offset());
indent(sb, level, "/// - encodedLength: %d\n", typeToken.encodedLength());
indent(sb, level, "/// - version: %d\n", typeToken.version());
indent(sb, level, "#[inline]\n");
indent(sb, level, "pub fn %s(&mut self, value: &[%s; %d]) {\n",
formatFunctionName(name),
rustPrimitiveType,
arrayLength);
// NB: must create variable 'offset' before calling mutable self.get_buf_mut()
indent(sb, level + 1, "let offset = self.%s;\n", getBufOffset(typeToken));
indent(sb, level + 1, "let buf = self.get_buf_mut();\n");
if (rustPrimitiveType.equals("u8"))
{
indent(sb, level + 1, "buf.put_bytes_at(offset, value);\n");
indent(sb, level, "}\n\n");
return;
}
for (int i = 0; i < arrayLength; i++)
{
if (i == 0)
{
indent(sb, level + 1, "buf.put_%s_at(offset, value[%d]);\n", rustPrimitiveType, i);
}
else
{
indent(sb, level + 1, "buf.put_%s_at(offset + %d, value[%d]);\n",
rustPrimitiveType,
i * primitiveType.size(),
i);
}
}
indent(sb, level, "}\n\n");
}
else
{
if (encoding.presence() == Encoding.Presence.CONSTANT)
{
indent(sb, level, "// skipping CONSTANT %s\n\n", name);
}
else
{
indent(sb, level, "/// primitive field '%s'\n", name);
indent(sb, level, "/// - min value: %s\n", encoding.applicableMinValue());
indent(sb, level, "/// - max value: %s\n", encoding.applicableMaxValue());
indent(sb, level, "/// - null value: %s\n", encoding.applicableNullValue());
indent(sb, level, "/// - characterEncoding: %s\n", encoding.characterEncoding());
indent(sb, level, "/// - semanticType: %s\n", encoding.semanticType());
indent(sb, level, "/// - encodedOffset: %d\n", typeToken.offset());
indent(sb, level, "/// - encodedLength: %d\n", typeToken.encodedLength());
indent(sb, level, "/// - version: %d\n", typeToken.version());
indent(sb, level, "#[inline]\n");
indent(sb, level, "pub fn %s(&mut self, value: %s) {\n",
formatFunctionName(name),
rustPrimitiveType);
// NB: must create variable 'offset' before calling mutable self.get_buf_mut()
indent(sb, level + 1, "let offset = self.%s;\n", getBufOffset(typeToken));
indent(sb, level + 1, "self.get_buf_mut().put_%s_at(offset, value);\n", rustPrimitiveType);
indent(sb, level, "}\n\n");
}
}
}
private static void generateEnumEncoder(
final StringBuilder sb,
final int level,
final Token fieldToken,
final Token typeToken,
final String name) throws IOException
{
final String referencedName = typeToken.referencedName();
final String enumType = format("%s::%s",
toLowerSnakeCase(referencedName == null ? typeToken.name() : referencedName),
formatStructName(referencedName == null ? typeToken.name() : referencedName));
if (fieldToken.isConstantEncoding())
{
indent(sb, level, "// skipping CONSTANT enum '%s'\n\n", name);
}
else
{
final Encoding encoding = typeToken.encoding();
final String rustPrimitiveType = rustTypeName(encoding.primitiveType());
indent(sb, level, "/// REQUIRED enum\n");
indent(sb, level, "#[inline]\n");
indent(sb, level, "pub fn %s(&mut self, value: %s) {\n", formatFunctionName(name), enumType);
// NB: must create variable 'offset' before calling mutable self.get_buf_mut()
indent(sb, level + 1, "let offset = self.%s;\n", getBufOffset(typeToken));
indent(sb, level + 1, "self.get_buf_mut().put_%s_at(offset, value as %1$s)\n", rustPrimitiveType);
indent(sb, level, "}\n\n");
}
}
private static void generateBitSetEncoder(
final StringBuilder sb,
final int level,
final Token bitsetToken,
final String name) throws IOException
{
final Encoding encoding = bitsetToken.encoding();
final String rustPrimitiveType = rustTypeName(encoding.primitiveType());
final String referencedName = bitsetToken.referencedName();
final String structTypeName = format("%s::%s",
toLowerSnakeCase(referencedName == null ? bitsetToken.name() : referencedName),
formatStructName(bitsetToken.applicableTypeName()));
indent(sb, level, "#[inline]\n");
indent(sb, level, "pub fn %s(&mut self, value: %s) {\n", formatFunctionName(name), structTypeName);
// NB: must create variable 'offset' before calling mutable self.get_buf_mut()
indent(sb, level + 1, "let offset = self.%s;\n", getBufOffset(bitsetToken));
indent(sb, level + 1, "self.get_buf_mut().put_%s_at(offset, value.0)\n", rustPrimitiveType);
indent(sb, level, "}\n\n");
}
private static void generateCompositeEncoder(
final StringBuilder sb,
final int level,
final Token typeToken,
final String name) throws IOException
{
final String encoderName = toLowerSnakeCase(encoderName(name));
final String encoderTypeName = format("%s::%s",
codecModName(typeToken.referencedName() == null ? typeToken.name() : typeToken.referencedName()),
encoderName(formatStructName(typeToken.applicableTypeName())));
indent(sb, level, "/// COMPOSITE ENCODER\n");
indent(sb, level, "#[inline]\n");
indent(sb, level, "pub fn %s(self) -> %2$s {\n",
encoderName,
encoderTypeName);
// NB: must create variable 'offset' before moving 'self'
indent(sb, level + 1, "let offset = self.%s;\n", getBufOffset(typeToken));
indent(sb, level + 1, "%s::default().wrap(self, offset)\n", encoderTypeName);
indent(sb, level, "}\n\n");
}
static void generateDecoderFields(
final StringBuilder sb,
final List tokens,
final int level)
{
Generators.forEachField(
tokens,
(fieldToken, typeToken) ->
{
try
{
final String name = fieldToken.name();
final Encoding encoding = typeToken.encoding();
switch (typeToken.signal())
{
case ENCODING:
generatePrimitiveDecoder(sb, level, fieldToken, typeToken, name, encoding);
break;
case BEGIN_ENUM:
generateEnumDecoder(sb, level, fieldToken, typeToken, name);
break;
case BEGIN_SET:
generateBitSetDecoder(sb, level, typeToken, name);
break;
case BEGIN_COMPOSITE:
generateCompositeDecoder(sb, level, fieldToken, typeToken, name);
break;
default:
throw new UnsupportedOperationException("Unable to handle: " + typeToken);
}
}
catch (final IOException ex)
{
throw new UncheckedIOException(ex);
}
});
}
private static void generateCompositeDecoder(
final StringBuilder sb,
final int level,
final Token fieldToken,
final Token typeToken,
final String name) throws IOException
{
final String decoderName = toLowerSnakeCase(decoderName(name));
final String referencedName = typeToken.referencedName();
final String decoderTypeName = format("%s::%s",
codecModName(referencedName == null ? typeToken.name() : referencedName),
decoderName(formatStructName(typeToken.applicableTypeName())));
indent(sb, level, "/// COMPOSITE DECODER\n");
indent(sb, level, "#[inline]\n");
if (fieldToken.version() > 0)
{
indent(sb, level, "pub fn %s(self) -> Either> {\n",
decoderName,
decoderTypeName);
indent(sb, level + 1, "if self.acting_version() < %d {\n", fieldToken.version());
indent(sb, level + 2, "return Either::Left(self);\n");
indent(sb, level + 1, "}\n\n");
indent(sb, level + 1, "let offset = self.%s;\n", getBufOffset(fieldToken));
indent(sb, level + 1, "Either::Right(%s::default().wrap(self, offset))\n",
decoderTypeName);
}
else
{
indent(sb, level, "pub fn %s(self) -> %s {\n",
decoderName,
decoderTypeName);
indent(sb, level + 1, "let offset = self.%s;\n", getBufOffset(fieldToken));
indent(sb, level + 1, "%s::default().wrap(self, offset)\n", decoderTypeName);
}
indent(sb, level, "}\n\n");
}
private static void generateBitSetDecoder(
final StringBuilder sb,
final int level,
final Token bitsetToken,
final String name) throws IOException
{
final Encoding encoding = bitsetToken.encoding();
final String rustPrimitiveType = rustTypeName(encoding.primitiveType());
final String referencedName = bitsetToken.referencedName();
final String structTypeName = format("%s::%s",
toLowerSnakeCase(referencedName == null ? bitsetToken.name() : referencedName),
formatStructName(bitsetToken.applicableTypeName()));
indent(sb, level, "/// BIT SET DECODER\n");
indent(sb, level, "#[inline]\n");
indent(sb, level, "pub fn %s(&self) -> %s {\n", formatFunctionName(name), structTypeName);
if (bitsetToken.version() > 0)
{
indent(sb, level + 1, "if self.acting_version() < %d {\n", bitsetToken.version());
indent(sb, level + 2, "return %s::default();\n", structTypeName);
indent(sb, level + 1, "}\n\n");
}
indent(sb, level + 1, "%s::new(self.get_buf().get_%s_at(self.%s))\n",
structTypeName,
rustPrimitiveType,
getBufOffset(bitsetToken));
indent(sb, level, "}\n\n");
}
private static void generatePrimitiveDecoder(
final StringBuilder sb,
final int level,
final Token fieldToken,
final Token typeToken,
final String name,
final Encoding encoding) throws IOException
{
if (typeToken.arrayLength() > 1)
{
generatePrimitiveArrayDecoder(sb, level, fieldToken, typeToken, name, encoding);
}
else if (encoding.presence() == Encoding.Presence.CONSTANT)
{
generatePrimitiveConstantDecoder(sb, level, name, encoding);
}
else if (encoding.presence() == Encoding.Presence.OPTIONAL)
{
generatePrimitiveOptionalDecoder(sb, level, fieldToken, name, encoding);
}
else
{
generatePrimitiveRequiredDecoder(sb, level, fieldToken, name, encoding);
}
}
private static void generatePrimitiveArrayDecoder(
final StringBuilder sb,
final int level,
final Token fieldToken,
final Token typeToken,
final String name,
final Encoding encoding) throws IOException
{
final PrimitiveType primitiveType = encoding.primitiveType();
final String rustPrimitiveType = rustTypeName(primitiveType);
final int arrayLength = typeToken.arrayLength();
assert arrayLength > 1;
indent(sb, level, "#[inline]\n");
indent(sb, level, "pub fn %s(&self) -> [%s; %d] {\n",
formatFunctionName(name),
rustPrimitiveType,
arrayLength);
if (fieldToken.version() > 0)
{
indent(sb, level + 1, "if self.acting_version() < %d {\n", fieldToken.version());
indent(sb, level + 2, "return [%s; %d];\n", encoding.applicableNullValue(), arrayLength);
indent(sb, level + 1, "}\n\n");
}
indent(sb, level + 1, "let buf = self.get_buf();\n");
if (rustPrimitiveType.equals("u8"))
{
indent(sb, level + 1, "ReadBuf::get_bytes_at(buf.data, self.%s)\n",
getBufOffset(typeToken));
indent(sb, level, "}\n\n");
return;
}
indent(sb, level + 1, "[\n");
for (int i = 0; i < arrayLength; i++)
{
if (i == 0)
{
indent(sb, level + 2, "buf.get_%s_at(self.%s),\n",
rustPrimitiveType,
getBufOffset(typeToken));
}
else
{
indent(sb, level + 2, "buf.get_%s_at(self.%s + %d),\n",
rustPrimitiveType,
getBufOffset(typeToken),
i * primitiveType.size());
}
}
indent(sb, level + 1, "]\n");
indent(sb, level, "}\n\n");
}
private static void generatePrimitiveConstantDecoder(
final StringBuilder sb,
final int level,
final String name,
final Encoding encoding) throws IOException
{
assert encoding.presence() == Encoding.Presence.CONSTANT;
final PrimitiveType primitiveType = encoding.primitiveType();
final String rustPrimitiveType = rustTypeName(primitiveType);
final String characterEncoding = encoding.characterEncoding();
indent(sb, level, "/// CONSTANT \n");
final String rawConstValue = encoding.constValue().toString();
if (characterEncoding != null)
{
indent(sb, level, "/// characterEncoding: '%s'\n", characterEncoding);
indent(sb, level, "#[inline]\n");
if (JavaUtil.isAsciiEncoding(characterEncoding))
{
indent(sb, level, "pub fn %s(&self) -> &'static [u8] {\n",
formatFunctionName(name));
indent(sb, level + 1, "b\"%s\"\n", rawConstValue);
}
else if (JavaUtil.isUtf8Encoding(characterEncoding))
{
indent(sb, level, "pub fn %s(&self) -> &'static str {\n", formatFunctionName(name));
indent(sb, level + 1, "\"%s\"\n", rawConstValue);
}
else
{
throw new IllegalArgumentException("Unsupported encoding: " + characterEncoding);
}
indent(sb, level, "}\n\n");
}
else
{
indent(sb, level, "#[inline]\n");
indent(sb, level, "pub fn %s(&self) -> %s {\n",
formatFunctionName(name),
rustPrimitiveType);
indent(sb, level + 1, "%s\n", rawConstValue);
indent(sb, level, "}\n\n");
}
}
private static void generatePrimitiveOptionalDecoder(
final StringBuilder sb,
final int level,
final Token fieldToken,
final String name,
final Encoding encoding) throws IOException
{
assert encoding.presence() == Encoding.Presence.OPTIONAL;
final PrimitiveType primitiveType = encoding.primitiveType();
final String rustPrimitiveType = rustTypeName(primitiveType);
final String characterEncoding = encoding.characterEncoding();
indent(sb, level, "/// primitive field - '%s' { null_value: '%s' }\n",
encoding.presence(), encoding.applicableNullValue());
if (characterEncoding != null)
{
// ASCII and UTF-8
indent(sb, level, "/// characterEncoding: '%s'\n", characterEncoding);
}
indent(sb, level, "#[inline]\n");
indent(sb, level, "pub fn %s(&self) -> Option<%s> {\n",
formatFunctionName(name),
rustPrimitiveType);
indent(sb, level + 1, "let value = self.get_buf().get_%s_at(self.%s);\n",
rustPrimitiveType,
getBufOffset(fieldToken));
final String literal = generateRustLiteral(primitiveType, encoding.applicableNullValue().toString());
if (literal.endsWith("::NAN"))
{
indent(sb, level + 1, "if value.is_nan() {\n");
}
else
{
indent(sb, level + 1, "if value == %s {\n", literal);
}
indent(sb, level + 2, "None\n");
indent(sb, level + 1, "} else {\n");
indent(sb, level + 2, "Some(value)\n");
indent(sb, level + 1, "}\n");
indent(sb, level, "}\n\n");
}
private static void generatePrimitiveRequiredDecoder(
final StringBuilder sb,
final int level,
final Token fieldToken,
final String name,
final Encoding encoding) throws IOException
{
assert encoding.presence() == Encoding.Presence.REQUIRED;
final PrimitiveType primitiveType = encoding.primitiveType();
final String rustPrimitiveType = rustTypeName(primitiveType);
final String characterEncoding = encoding.characterEncoding();
indent(sb, level, "/// primitive field - '%s'\n", encoding.presence());
if (characterEncoding != null)
{
// ASCII and UTF-8
indent(sb, level, "/// characterEncoding: '%s'\n", characterEncoding);
}
indent(sb, level, "#[inline]\n");
indent(sb, level, "pub fn %s(&self) -> %s {\n",
formatFunctionName(name),
rustPrimitiveType);
if (fieldToken.version() > 0)
{
indent(sb, level + 1, "if self.acting_version() < %d {\n", fieldToken.version());
indent(sb, level + 2, "return %s;\n",
generateRustLiteral(encoding.primitiveType(), encoding.applicableNullValue().toString()));
indent(sb, level + 1, "}\n\n");
}
indent(sb, level + 1, "self.get_buf().get_%s_at(self.%s)\n",
rustPrimitiveType,
getBufOffset(fieldToken));
indent(sb, level, "}\n\n");
}
private static void generateEnumDecoder(
final StringBuilder sb,
final int level,
final Token fieldToken,
final Token typeToken,
final String name) throws IOException
{
final String referencedName = typeToken.referencedName();
final String enumType = format("%s::%s",
toLowerSnakeCase(referencedName == null ? typeToken.name() : referencedName),
formatStructName(referencedName == null ? typeToken.name() : referencedName));
if (fieldToken.isConstantEncoding())
{
indent(sb, level, "/// CONSTANT enum\n");
final Encoding encoding = fieldToken.encoding();
final String rawConstValueName = encoding.constValue().toString();
final int indexOfDot = rawConstValueName.indexOf('.');
final String constValueName = -1 == indexOfDot ?
rawConstValueName :
rawConstValueName.substring(indexOfDot + 1);
final String constantRustExpression = enumType + "::" + constValueName;
appendConstAccessor(sb, name, enumType, constantRustExpression, level);
}
else
{
final Encoding encoding = typeToken.encoding();
final String rustPrimitiveType = rustTypeName(encoding.primitiveType());
indent(sb, level, "/// REQUIRED enum\n");
indent(sb, level, "#[inline]\n");
indent(sb, level, "pub fn %s(&self) -> %s {\n", formatFunctionName(name), enumType);
if (fieldToken.version() > 0)
{
indent(sb, level + 1, "if self.acting_version() < %d {\n", fieldToken.version());
indent(sb, level + 2, "return %s::default();\n", enumType);
indent(sb, level + 1, "}\n\n");
}
indent(sb, level + 1, "self.get_buf().get_%s_at(self.%s).into()\n",
rustPrimitiveType,
getBufOffset(typeToken));
indent(sb, level, "}\n\n");
}
}
static void generateDecoderGroups(
final String schemaVersionType,
final StringBuilder sb,
final List tokens,
final int level,
final ParentDef parentDef) throws IOException
{
for (int i = 0, size = tokens.size(); i < size; i++)
{
final Token groupToken = tokens.get(i);
if (groupToken.signal() != Signal.BEGIN_GROUP)
{
throw new IllegalStateException("tokens must begin with BEGIN_GROUP: token=" + groupToken);
}
++i;
final int index = i;
final int groupHeaderTokenCount = tokens.get(i).componentTokenCount();
i += groupHeaderTokenCount;
final List fields = new ArrayList<>();
i = collectFields(tokens, i, fields);
final List groups = new ArrayList<>();
i = collectGroups(tokens, i, groups);
final List varData = new ArrayList<>();
i = collectVarData(tokens, i, varData);
final String groupName = decoderName(formatStructName(groupToken.name()));
final String description = groupToken.description();
if (!Strings.isEmpty(description))
{
indent(sb, level, "/// GROUP DECODER (id=%s, description='%s')\n",
groupToken.id(), description);
}
else
{
indent(sb, level, "/// GROUP DECODER (id=%s)\n",
groupToken.id());
}
assert 4 == groupHeaderTokenCount;
indent(sb, level, "#[inline]\n");
if (groupToken.version() > 0)
{
indent(sb, level, "pub fn %s(self) -> Option<%2$s> {\n",
formatFunctionName(groupName), groupName);
indent(sb, level + 1, "if self.acting_version() < %d {\n", groupToken.version());
indent(sb, level + 2, "return None;\n");
indent(sb, level + 1, "}\n\n");
indent(sb, level + 1, "Some(%s::default().wrap(self))\n", groupName);
}
else
{
indent(sb, level, "pub fn %s(self) -> %2$s {\n",
formatFunctionName(groupName), groupName);
indent(sb, level + 1, "%s::default().wrap(self)\n", groupName);
}
indent(sb, level, "}\n\n");
final SubGroup subGroup = parentDef.addSubGroup(groupName, level, groupToken);
subGroup.generateDecoder(schemaVersionType, tokens, fields, groups, varData, index);
}
}
static void generateDecoderVarData(
final StringBuilder sb,
final List tokens,
final int level,
final boolean isSubGroup) throws IOException
{
for (int i = 0, size = tokens.size(); i < size; )
{
final Token varDataToken = tokens.get(i);
if (varDataToken.signal() != Signal.BEGIN_VAR_DATA)
{
throw new IllegalStateException("tokens must begin with BEGIN_VAR_DATA: token=" + varDataToken);
}
final String characterEncoding = characterEncoding(tokens.get(i + 3).encoding());
final String propertyName = toLowerSnakeCase(varDataToken.name());
final Token lengthToken = tokens.get(i + 2);
final Encoding lengthEncoding = lengthToken.encoding();
final PrimitiveType lengthType = lengthEncoding.primitiveType();
indent(sb, level, "/// VAR_DATA DECODER - character encoding: '%s'\n", characterEncoding);
indent(sb, level, "#[inline]\n");
indent(sb, level, "pub fn %s_decoder(&mut self) -> (usize, usize) {\n", propertyName);
if (isSubGroup)
{
if (varDataToken.version() > 0)
{
indent(sb, level + 1, "if self.acting_version() < %d {\n",
varDataToken.version());
indent(sb, level + 2, "return (self.parent.as_ref().unwrap().get_limit(), 0);\n");
indent(sb, level + 1, "}\n\n");
}
indent(sb, level + 1, "let offset = self.parent.as_ref().expect(\"parent missing\").get_limit();\n");
indent(sb, level + 1, "let data_length = self.get_buf().get_%s_at(offset) as usize;\n",
rustTypeName(lengthType));
indent(sb, level + 1, "self.parent.as_mut().unwrap().set_limit(offset + %d + data_length);\n",
lengthType.size());
}
else
{
if (varDataToken.version() > 0)
{
indent(sb, level + 1, "if self.acting_version() < %d {\n",
varDataToken.version());
indent(sb, level + 2, "return (self.get_limit(), 0);\n");
indent(sb, level + 1, "}\n\n");
}
indent(sb, level + 1, "let offset = self.get_limit();\n");
indent(sb, level + 1, "let data_length = self.get_buf().get_%s_at(offset) as usize;\n",
rustTypeName(lengthType));
indent(sb, level + 1, "self.set_limit(offset + %d + data_length);\n", lengthType.size());
}
indent(sb, level + 1, "(offset + %d, data_length)\n", lengthType.size());
indent(sb, level, "}\n\n");
// function to return slice form given coord
indent(sb, level, "#[inline]\n");
indent(sb, level, "pub fn %s_slice(&'a self, coordinates: (usize, usize)) -> &'a [u8] {\n", propertyName);
if (varDataToken.version() > 0)
{
indent(sb, level + 1, "if self.acting_version() < %d {\n",
varDataToken.version());
indent(sb, level + 2, "return &[] as &[u8];\n");
indent(sb, level + 1, "}\n\n");
}
indent(sb, level + 1, "debug_assert!(self.get_limit() >= coordinates.0 + coordinates.1);\n");
indent(sb, level + 1, "self.get_buf().get_slice_at(coordinates.0, coordinates.1)\n");
indent(sb, level, "}\n\n");
i += varDataToken.componentTokenCount();
}
}
private static void generateBitSets(
final Ir ir,
final RustOutputManager outputManager) throws IOException
{
for (final List tokens : ir.types())
{
if (!tokens.isEmpty() && tokens.get(0).signal() == BEGIN_SET)
{
final Token beginToken = tokens.get(0);
final String bitSetType = formatStructName(beginToken.applicableTypeName());
try (Writer out = outputManager.createOutput(bitSetType))
{
generateSingleBitSet(bitSetType, tokens, out);
}
}
}
}
private static void generateSingleBitSet(
final String bitSetType,
final List tokens,
final Appendable writer) throws IOException
{
final Token beginToken = tokens.get(0);
final String rustPrimitiveType = rustTypeName(beginToken.encoding().primitiveType());
indent(writer, 0, "#[derive(Default, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]\n");
indent(writer, 0, "pub struct %s(pub %s);\n", bitSetType, rustPrimitiveType);
indent(writer, 0, "impl %s {\n", bitSetType);
indent(writer, 1, "#[inline]\n");
indent(writer, 1, "pub fn new(value: %s) -> Self {\n", rustPrimitiveType);
indent(writer, 2, "%s(value)\n", bitSetType);
indent(writer, 1, "}\n\n");
indent(writer, 1, "#[inline]\n");
indent(writer, 1, "pub fn clear(&mut self) -> &mut Self {\n");
indent(writer, 2, "self.0 = 0;\n");
indent(writer, 2, "self\n");
indent(writer, 1, "}\n");
for (final Token token : tokens)
{
if (Signal.CHOICE != token.signal())
{
continue;
}
final String choiceName = formatFunctionName(token.name());
final Encoding encoding = token.encoding();
final String choiceBitIndex = encoding.constValue().toString();
indent(writer, 0, "\n");
indent(writer, 1, "#[inline]\n");
indent(writer, 1, "pub fn get_%s(&self) -> bool {\n", choiceName);
indent(writer, 2, "0 != self.0 & (1 << %s)\n", choiceBitIndex);
indent(writer, 1, "}\n\n");
indent(writer, 1, "#[inline]\n");
indent(writer, 1, "pub fn set_%s(&mut self, value: bool) -> &mut Self {\n", choiceName);
indent(writer, 2, "self.0 = if value {\n");
indent(writer, 3, "self.0 | (1 << %s)\n", choiceBitIndex);
indent(writer, 2, "} else {\n");
indent(writer, 3, "self.0 & !(1 << %s)\n", choiceBitIndex);
indent(writer, 2, "};\n");
indent(writer, 2, "self\n");
indent(writer, 1, "}\n");
}
indent(writer, 0, "}\n");
indent(writer, 0, "impl core::fmt::Debug for %s {\n", bitSetType);
indent(writer, 1, "#[inline]\n");
indent(writer, 1, "fn fmt(&self, fmt: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {\n");
indent(writer, 2, "write!(fmt, \"%s[", bitSetType);
final StringBuilder builder = new StringBuilder();
final StringBuilder arguments = new StringBuilder();
boolean comma = false;
for (final Token token : tokens)
{
if (Signal.CHOICE != token.signal())
{
continue;
}
final String choiceName = formatFunctionName(token.name());
final String choiceBitIndex = token.encoding().constValue().toString();
if (comma)
{
builder.append(",");
}
builder.append(choiceName).append("(").append(choiceBitIndex).append(")={}");
arguments.append("self.get_").append(choiceName).append("(),");
comma = true;
}
writer.append(builder.toString()).append("]\",\n");
indent(writer, 3, arguments + ")\n");
indent(writer, 1, "}\n");
indent(writer, 0, "}\n");
}
static void appendImplEncoderTrait(
final Appendable out,
final String typeName) throws IOException
{
indent(out, 1, "impl<%s> %s for %s {\n", BUF_LIFETIME, withLifetime("Writer"), withLifetime(typeName));
indent(out, 2, "#[inline]\n");
indent(out, 2, "fn get_buf_mut(&mut self) -> &mut WriteBuf<'a> {\n");
indent(out, 3, "&mut self.buf\n");
indent(out, 2, "}\n");
indent(out, 1, "}\n\n");
indent(out, 1, "impl<%s> %s for %s {\n", BUF_LIFETIME, withLifetime("Encoder"), withLifetime(typeName));
indent(out, 2, "#[inline]\n");
indent(out, 2, "fn get_limit(&self) -> usize {\n");
indent(out, 3, "self.limit\n");
indent(out, 2, "}\n\n");
indent(out, 2, "#[inline]\n");
indent(out, 2, "fn set_limit(&mut self, limit: usize) {\n");
indent(out, 3, "self.limit = limit;\n");
indent(out, 2, "}\n");
indent(out, 1, "}\n\n");
}
static void appendImplDecoderTrait(
final String schemaVersionType,
final Appendable out,
final String typeName) throws IOException
{
indent(out, 1, "impl<%s> %s for %s {\n", BUF_LIFETIME, "ActingVersion", withLifetime(typeName));
indent(out, 2, "#[inline]\n");
indent(out, 2, "fn acting_version(&self) -> %s {\n", schemaVersionType);
indent(out, 3, "self.acting_version\n");
indent(out, 2, "}\n");
indent(out, 1, "}\n\n");
indent(out, 1, "impl<%s> %s for %s {\n", BUF_LIFETIME, withLifetime("Reader"), withLifetime(typeName));
indent(out, 2, "#[inline]\n");
indent(out, 2, "fn get_buf(&self) -> &ReadBuf<'a> {\n");
indent(out, 3, "&self.buf\n");
indent(out, 2, "}\n");
indent(out, 1, "}\n\n");
indent(out, 1, "impl<%s> %s for %s {\n", BUF_LIFETIME, withLifetime("Decoder"), withLifetime(typeName));
indent(out, 2, "#[inline]\n");
indent(out, 2, "fn get_limit(&self) -> usize {\n");
indent(out, 3, "self.limit\n");
indent(out, 2, "}\n\n");
indent(out, 2, "#[inline]\n");
indent(out, 2, "fn set_limit(&mut self, limit: usize) {\n");
indent(out, 3, "self.limit = limit;\n");
indent(out, 2, "}\n");
indent(out, 1, "}\n\n");
}
private static void generateEnums(
final Ir ir,
final RustOutputManager outputManager) throws IOException
{
final Set enumTypeNames = new HashSet<>();
for (final List tokens : ir.types())
{
if (tokens.isEmpty())
{
continue;
}
final Token beginToken = tokens.get(0);
if (beginToken.signal() != BEGIN_ENUM)
{
continue;
}
final String typeName = beginToken.applicableTypeName();
if (!enumTypeNames.add(typeName))
{
continue;
}
try (Writer out = outputManager.createOutput(typeName))
{
generateEnum(tokens, out);
}
}
}
private static void generateEnum(
final List enumTokens,
final Appendable writer) throws IOException
{
final String originalEnumName = enumTokens.get(0).applicableTypeName();
final String enumRustName = formatStructName(originalEnumName);
final List messageBody = enumTokens.subList(1, enumTokens.size() - 1);
if (messageBody.isEmpty())
{
throw new IllegalArgumentException("No valid values provided for enum " + originalEnumName);
}
indent(writer, 0, "#[derive(Clone, Copy, Debug, Default, PartialEq, Eq, PartialOrd, Ord, Hash)]\n");
final String primitiveType = rustTypeName(messageBody.get(0).encoding().primitiveType());
indent(writer, 0, "#[repr(%s)]\n", primitiveType);
indent(writer, 0, "pub enum %s {\n", enumRustName);
for (final Token token : messageBody)
{
final Encoding encoding = token.encoding();
final String literal = generateRustLiteral(encoding.primitiveType(), encoding.constValue().toString());
indent(writer, 1, "%s = %s, \n", token.name(), literal);
}
// null value
{
final Encoding encoding = messageBody.get(0).encoding();
final CharSequence nullVal = generateRustLiteral(encoding.primitiveType(),
encoding.applicableNullValue().toString());
indent(writer, 1, "#[default]\n");
indent(writer, 1, "NullVal = %s, \n", nullVal);
}
indent(writer, 0, "}\n");
// From impl
indent(writer, 0, "impl From<%s> for %s {\n", primitiveType, enumRustName);
indent(writer, 1, "#[inline]\n");
indent(writer, 1, "fn from(v: %s) -> Self {\n", primitiveType);
indent(writer, 2, "match v {\n");
for (final Token token : messageBody)
{
final Encoding encoding = token.encoding();
final String literal = generateRustLiteral(encoding.primitiveType(), encoding.constValue().toString());
indent(writer, 3, "%s => Self::%s, \n", literal, token.name());
}
// default => NullVal
indent(writer, 3, "_ => Self::NullVal,\n");
indent(writer, 2, "}\n");
indent(writer, 1, "}\n");
indent(writer, 0, "}\n");
}
private static void generateComposites(
final String schemaVersionType,
final Ir ir,
final RustOutputManager outputManager) throws IOException
{
for (final List tokens : ir.types())
{
if (!tokens.isEmpty() && tokens.get(0).signal() == Signal.BEGIN_COMPOSITE)
{
generateComposite(schemaVersionType, tokens, outputManager);
}
}
}
private static void generateComposite(
final String schemaVersionType,
final List tokens,
final RustOutputManager outputManager) throws IOException
{
final Token token = tokens.get(0);
final String compositeName = token.applicableTypeName();
final String compositeModName = codecModName(compositeName);
try (Writer out = outputManager.createOutput(compositeModName))
{
indent(out, 0, "use crate::*;\n\n");
indent(out, 0, "pub use encoder::%sEncoder;\n", formatStructName(compositeName));
indent(out, 0, "pub use decoder::%sDecoder;\n\n", formatStructName(compositeName));
final int encodedLength = tokens.get(0).encodedLength();
if (encodedLength > 0)
{
indent(out, 0, "pub const ENCODED_LENGTH: usize = %d;\n\n", encodedLength);
}
generateCompositeEncoder(tokens, encoderName(compositeName), out);
indent(out, 0, "\n");
generateCompositeDecoder(schemaVersionType, tokens, decoderName(compositeName), out);
}
}
static void appendImplWriterForComposite(
final Appendable out,
final int level,
final String name) throws IOException
{
// impl Decoder...
indent(out, level, "impl<'a, P> Writer<'a> for %s where P: Writer<'a> + Default {\n", name);
indent(out, level + 1, "#[inline]\n");
indent(out, level + 1, "fn get_buf_mut(&mut self) -> &mut WriteBuf<'a> {\n");
indent(out, level + 2, "if let Some(parent) = self.parent.as_mut() {\n");
indent(out, level + 3, "parent.get_buf_mut()\n");
indent(out, level + 2, "} else {\n");
indent(out, level + 3, "panic!(\"parent was None\")\n");
indent(out, level + 2, "}\n");
indent(out, level + 1, "}\n");
indent(out, level, "}\n\n");
}
static void appendImplEncoderForComposite(
final Appendable out,
final int level,
final String name) throws IOException
{
appendImplWriterForComposite(out, level, name);
// impl Encoder...
indent(out, level, "impl<'a, P> Encoder<'a> for %s
where P: Encoder<'a> + Default {\n", name);
indent(out, level + 1, "#[inline]\n");
indent(out, level + 1, "fn get_limit(&self) -> usize {\n");
indent(out, level + 2, "self.parent.as_ref().expect(\"parent missing\").get_limit()\n");
indent(out, level + 1, "}\n\n");
indent(out, level + 1, "#[inline]\n");
indent(out, level + 1, "fn set_limit(&mut self, limit: usize) {\n");
indent(out, level + 2, "self.parent.as_mut().expect(\"parent missing\").set_limit(limit);\n");
indent(out, level + 1, "}\n");
indent(out, level, "}\n\n");
}
static void appendImplReaderForComposite(
final String schemaVersionType,
final int version,
final Appendable out,
final int level,
final String name) throws IOException
{
indent(out, level, "impl<'a, P> ActingVersion for %s
where P: Reader<'a> + ActingVersion + Default {\n",
name);
indent(out, level + 1, "#[inline]\n");
indent(out, level + 1, "fn acting_version(&self) -> %s {\n", schemaVersionType);
indent(out, level + 2, "self.parent.as_ref().unwrap().acting_version()\n");
indent(out, level + 1, "}\n");
indent(out, level, "}\n\n");
// impl Reader...
indent(out, level, "impl<'a, P> Reader<'a> for %s
where P: Reader<'a> %s+ Default {\n",
name, version > 0 ? "+ ActingVersion " : "");
indent(out, level + 1, "#[inline]\n");
indent(out, level + 1, "fn get_buf(&self) -> &ReadBuf<'a> {\n");
indent(out, level + 2, "self.parent.as_ref().expect(\"parent missing\").get_buf()\n");
indent(out, level + 1, "}\n");
indent(out, level, "}\n\n");
}
static void appendImplDecoderForComposite(
final String schemaVersionType,
final int version,
final Appendable out,
final int level,
final String name) throws IOException
{
appendImplReaderForComposite(schemaVersionType, version, out, level, name);
// impl Decoder...
indent(out, level, "impl<'a, P> Decoder<'a> for %s
where P: Decoder<'a> + ActingVersion + Default {\n",
name);
indent(out, level + 1, "#[inline]\n");
indent(out, level + 1, "fn get_limit(&self) -> usize {\n");
indent(out, level + 2, "self.parent.as_ref().expect(\"parent missing\").get_limit()\n");
indent(out, level + 1, "}\n\n");
indent(out, level + 1, "#[inline]\n");
indent(out, level + 1, "fn set_limit(&mut self, limit: usize) {\n");
indent(out, level + 2, "self.parent.as_mut().expect(\"parent missing\").set_limit(limit);\n");
indent(out, level + 1, "}\n");
indent(out, level, "}\n\n");
}
private static void generateCompositeEncoder(
final List tokens,
final String encoderName,
final Writer out) throws IOException
{
indent(out, 0, "pub mod encoder {\n");
indent(out, 1, "use super::*;\n\n");
// define struct...
indent(out, 1, "#[derive(Debug, Default)]\n");
indent(out, 1, "pub struct %s {\n", encoderName);
indent(out, 2, "parent: Option
,\n");
indent(out, 2, "offset: usize,\n");
indent(out, 1, "}\n\n");
appendImplWriterForComposite(out, 1, encoderName);
// impl<'a> start
indent(out, 1, "impl<'a, P> %s
where P: Writer<'a> + Default {\n", encoderName);
indent(out, 2, "pub fn wrap(mut self, parent: P, offset: usize) -> Self {\n");
indent(out, 3, "self.parent = Some(parent);\n");
indent(out, 3, "self.offset = offset;\n");
indent(out, 3, "self\n");
indent(out, 2, "}\n\n");
// parent fn...
indent(out, 2, "#[inline]\n");
indent(out, 2, "pub fn parent(&mut self) -> SbeResult
{\n");
indent(out, 3, "self.parent.take().ok_or(SbeErr::ParentNotSet)\n");
indent(out, 2, "}\n\n");
for (int i = 1, end = tokens.size() - 1; i < end; )
{
final Token encodingToken = tokens.get(i);
final StringBuilder sb = new StringBuilder();
switch (encodingToken.signal())
{
case ENCODING:
generatePrimitiveEncoder(sb, 2, encodingToken, encodingToken.name());
break;
case BEGIN_ENUM:
generateEnumEncoder(sb, 2, encodingToken, encodingToken, encodingToken.name());
break;
case BEGIN_SET:
generateBitSetEncoder(sb, 2, encodingToken, encodingToken.name());
break;
case BEGIN_COMPOSITE:
generateCompositeEncoder(sb, 2, encodingToken, encodingToken.name());
break;
default:
break;
}
out.append(sb);
i += encodingToken.componentTokenCount();
}
indent(out, 1, "}\n"); // end impl
indent(out, 0, "} // end encoder mod \n");
}
private static void generateCompositeDecoder(
final String schemaVersionType,
final List tokens,
final String decoderName,
final Writer out) throws IOException
{
indent(out, 0, "pub mod decoder {\n");
indent(out, 1, "use super::*;\n\n");
indent(out, 1, "#[derive(Debug, Default)]\n");
indent(out, 1, "pub struct %s {\n", decoderName);
indent(out, 2, "parent: Option
,\n");
indent(out, 2, "offset: usize,\n");
indent(out, 1, "}\n\n");
final int version = tokens.stream().findFirst().get().version();
appendImplReaderForComposite(schemaVersionType, version, out, 1, decoderName);
// impl<'a, P> start
indent(out, 1, "impl<'a, P> %s
where P: Reader<'a> %s+ Default {\n",
decoderName, version > 0 ? "+ ActingVersion " : "");
indent(out, 2, "pub fn wrap(mut self, parent: P, offset: usize) -> Self {\n");
indent(out, 3, "self.parent = Some(parent);\n");
indent(out, 3, "self.offset = offset;\n");
indent(out, 3, "self\n");
indent(out, 2, "}\n\n");
// parent fn...
indent(out, 2, "#[inline]\n");
indent(out, 2, "pub fn parent(&mut self) -> SbeResult
{\n");
indent(out, 3, "self.parent.take().ok_or(SbeErr::ParentNotSet)\n");
indent(out, 2, "}\n\n");
for (int i = 1, end = tokens.size() - 1; i < end; )
{
final Token encodingToken = tokens.get(i);
final StringBuilder sb = new StringBuilder();
switch (encodingToken.signal())
{
case ENCODING:
generatePrimitiveDecoder(
sb, 2, encodingToken, encodingToken, encodingToken.name(), encodingToken.encoding());
break;
case BEGIN_ENUM:
generateEnumDecoder(sb, 2, encodingToken, encodingToken, encodingToken.name());
break;
case BEGIN_SET:
generateBitSetDecoder(sb, 2, encodingToken, encodingToken.name());
break;
case BEGIN_COMPOSITE:
generateCompositeDecoder(sb, 2, encodingToken, encodingToken, encodingToken.name());
break;
default:
break;
}
out.append(sb);
i += encodingToken.componentTokenCount();
}
indent(out, 1, "}\n"); // end impl
indent(out, 0, "} // end decoder mod \n");
}
private static void appendConstAccessor(
final Appendable writer,
final String name,
final String rustTypeName,
final String rustExpression,
final int level) throws IOException
{
indent(writer, level, "#[inline]\n");
indent(writer, level, "pub fn %s(&self) -> %s {\n", formatFunctionName(name), rustTypeName);
indent(writer, level + 1, rustExpression + "\n");
indent(writer, level, "}\n\n");
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy