/// <summary> Initializes a new instance of the NbtWriter class. </summary>
/// <param name="stream"> Stream to write to. </param>
/// <param name="rootTagName"> Name to give to the root tag (written immediately). </param>
/// <param name="bigEndian"> Whether NBT data should be in Big-Endian encoding. </param>
/// <exception cref="ArgumentNullException"> <paramref name="stream"/> or <paramref name="rootTagName"/> is <c>null</c>. </exception>
/// <exception cref="ArgumentException"> <paramref name="stream"/> is not writable. </exception>
public NbtWriter([NotNull] Stream stream, [NotNull] string rootTagName, bool bigEndian)
{
if (rootTagName == null)
{
throw new ArgumentNullException("rootTagName");
}
writer = new NbtBinaryWriter(stream, bigEndian);
writer.Write((byte)NbtTagType.Compound);
writer.Write(rootTagName);
parentType = NbtTagType.Compound;
}
/// <summary> Saves this NBT file to a stream. Nothing is written to stream if RootTag is <c>null</c>. </summary>
/// <param name="stream"> Stream to write data to. May not be <c>null</c>. </param>
/// <param name="compression"> Compression mode to use for saving. May not be AutoDetect. </param>
/// <returns> Number of bytes written to the stream. </returns>
/// <exception cref="ArgumentNullException"> <paramref name="stream"/> is <c>null</c>. </exception>
/// <exception cref="ArgumentException"> If AutoDetect was given as the <paramref name="compression"/> mode. </exception>
/// <exception cref="ArgumentOutOfRangeException"> If an unrecognized/unsupported value was given for <paramref name="compression"/>. </exception>
/// <exception cref="InvalidDataException"> If given stream does not support writing. </exception>
/// <exception cref="NbtFormatException"> If RootTag is null;
/// or if RootTag is unnamed;
/// or if one of the NbtCompound tags contained unnamed tags;
/// or if an NbtList tag had Unknown list type and no elements. </exception>
public long SaveToStream([NotNull] Stream stream, NbtCompression compression)
{
if (stream == null)
{
throw new ArgumentNullException("stream");
}
switch (compression)
{
case NbtCompression.AutoDetect:
throw new ArgumentException("AutoDetect is not a valid NbtCompression value for saving.");
case NbtCompression.ZLib:
case NbtCompression.GZip:
case NbtCompression.None:
break;
default:
throw new ArgumentOutOfRangeException("compression");
}
if (rootTag.Name == null)
{
// This may trigger if root tag has been renamed
throw new NbtFormatException(
"Cannot save NbtFile: Root tag is not named. Its name may be an empty string, but not null.");
}
long startOffset = 0;
if (stream.CanSeek)
{
startOffset = stream.Position;
}
else
{
stream = new ByteCountingStream(stream);
}
switch (compression)
{
case NbtCompression.ZLib:
stream.WriteByte(0x78);
stream.WriteByte(0x01);
int checksum;
using (var compressStream = new ZLibStream(stream, CompressionMode.Compress, true))
{
var bufferedStream = new BufferedStream(compressStream, WriteBufferSize);
RootTag.WriteTag(new NbtBinaryWriter(bufferedStream, BigEndian));
bufferedStream.Flush();
checksum = compressStream.Checksum;
}
byte[] checksumBytes = BitConverter.GetBytes(checksum);
if (BitConverter.IsLittleEndian)
{
// Adler32 checksum is big-endian
Array.Reverse(checksumBytes);
}
stream.Write(checksumBytes, 0, checksumBytes.Length);
break;
case NbtCompression.GZip:
using (var compressStream = new GZipStream(stream, CompressionMode.Compress, true))
{
// use a buffered stream to avoid GZipping in small increments (which has a lot of overhead)
var bufferedStream = new BufferedStream(compressStream, WriteBufferSize);
RootTag.WriteTag(new NbtBinaryWriter(bufferedStream, BigEndian));
bufferedStream.Flush();
}
break;
case NbtCompression.None:
var writer = new NbtBinaryWriter(stream, BigEndian);
RootTag.WriteTag(writer);
break;
default:
throw new ArgumentOutOfRangeException("compression");
}
if (stream.CanSeek)
{
return(stream.Position - startOffset);
}
else
{
return(((ByteCountingStream)stream).BytesWritten);
}
}
