lzma-rs
lzma-rs copied to clipboard
Expose decoder/encoder primitives as public
I maintain a Rust implementation of MAME's Compressed Hunks of Data format, which is essentially chunks of data compressed using various compression algorithms, LZMA included. One of the quirks of this format is that because they use a very old LZMA 19.0, the encoding parameters are not saved into the output stream, and therefore every CHD decoder has to essentially mimic the defaults of LZMA 19.0.
Thankfully lzma-rs allows this but since the encode and decode modules aren't public, I had to fork and vendor lzma-rs to access those primitives like LzmaParams in my crate. Additionally I added LzmaParams::new to construct an instance manually for this purpose.
Ideally I would like to just be able to use mainline lzma-rs. Since decoder is mostly documented already, would it be feasible to expose those primitives as public without much work, possibly behind a feature flag?
Since
decoderis mostly documented already, would it be feasible to expose those primitives as public without much work, possibly behind a feature flag?
It surely would be feasible! Could you send a pull request for review?
I don't think a feature flag is needed here - just make sure that the newly exposed API is documented and that the GitHub workflows pass.
I did some internal refactoring to expose the raw decoder as a primitive where the LZMA compression parameters are encoded aas const generics. I also did a similar wrapper for LZMA2 although I'm not too sure about the usefulness of that in my fork.
I believe this is a better approach than just exposing LzmaParams and I was able to expose this API without changing the existing public API, but at the cost of having to make some relatively sizeable internal changes, particularly with DecoderState to also allow reusing the existing allocation.
If you could take a quick look to see if you're comfortable with those changes and the shape of the LzmaDecoder and Lzma2Decoder APIs I'd be happy to clean it up slightly and PR it upstream.
Some comments:
- In the current state of things, I'm not sure about the value of using const generics to encode the compression parameters. The
LzmaDecoderstruct just converts these const generics parameters into simply callingLzmaParams. It seems to me that something more useful could be achieved withfeature(generic_const_exprs)(i.e.literal_probscould be a const generic array rather than aVec), but this is still an unstable feature. I'm open to exploring that if it's possible to gate the use of const generics by a feature flag (i.e. on stable Rustlzma-rscompiles without using any const generics, and on nightly it can leverage const generics if the feature is enabled). - Having a
resetfunction that fills the default values in place rather than re-allocating and copying seems like a valuable addition :) It's unclear how much improvement this brings after compiler optimizations (the compiler may replace the array copy by a simple in-place writing of the default data) - but the intent is clearer and it also compiles better in debug mode. - What was the reason to moving the
outputoutside of the decoder state (https://github.com/gendx/lzma-rs/commit/07c2d7ee6f96599676bf747b504f35503711b60f)? This adds quite a bit of verbosity in passing it along on each function call.
Feel free to break down your changes into multiple PRs (exposing LzmaParams, improving the reset implementation, adding some const generics, etc.) to make it easier to review.
- Fair criticism of using const generics. In my fork the motivation was mostly for expressiveness (i.e. an
LzmaDecoder<3,0,2>would be type-incompatible with anLzmaDecoder<0,0,0>) but I'm open to feature-gating this for consumers that need it or just scrapping it all together in favour of usingLzmaParamsdirectly or anew_with_paramsconstructor. - After some unscientific benchmarking,
resetwas slightly faster in Release as well, but this is heavily dependent on compiler optimizations and specific usage. I think intentwise,resetis a good to have. - Taking
outputoutside of the decoder state was a necessary prerequisite for a simplerreset.- Conceptually,
resetprepares the decoder state for a new buffer of input that may be written into a new output. Keepingoutputin the decoder state tied the lifetime of the output buffer to the state of the decoder which wouldn't make much sense for a mutable reset function. This wasn't an issue before since the public APIs were one-shot. - Keeping the lifetime attached meant that a new
DecoderStatemust be created for each possible newoutputwhere'decoder : 'output. There is some added verbosity at the callsite but I think separating the lifetime of the decoder state and the output more than makes up for the complexity budget there. - Detaching the lifetime of
outputfromDecoderStatealso allowedLzmaDecoderto not be lifetime-parameterized and reuse the sameDecoderStatefor differentoutputbuffers.
- Conceptually,
With that said I'd be happy to split decoupling output from DecoderState into a separate PR since that doesn't affect any public APIs and build on top of that.