Compress.hx 2.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364
  1. /*
  2. * Copyright (c) 2005, The haXe Project Contributors
  3. * All rights reserved.
  4. * Redistribution and use in source and binary forms, with or without
  5. * modification, are permitted provided that the following conditions are met:
  6. *
  7. * - Redistributions of source code must retain the above copyright
  8. * notice, this list of conditions and the following disclaimer.
  9. * - Redistributions in binary form must reproduce the above copyright
  10. * notice, this list of conditions and the following disclaimer in the
  11. * documentation and/or other materials provided with the distribution.
  12. *
  13. * THIS SOFTWARE IS PROVIDED BY THE HAXE PROJECT CONTRIBUTORS "AS IS" AND ANY
  14. * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  15. * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  16. * DISCLAIMED. IN NO EVENT SHALL THE HAXE PROJECT CONTRIBUTORS BE LIABLE FOR
  17. * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  18. * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  19. * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
  20. * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  21. * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  22. * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
  23. * DAMAGE.
  24. */
  25. package cpp.zip;
  26. class Compress {
  27. var s : Dynamic;
  28. public function new( level : Int ) {
  29. s = _deflate_init(level);
  30. }
  31. public function execute( src : haxe.io.Bytes, srcPos : Int, dst : haxe.io.Bytes, dstPos : Int ) : { done : Bool, read : Int, write : Int } {
  32. return _deflate_buffer(s,src.getData(),srcPos,dst.getData(),dstPos);
  33. }
  34. public function setFlushMode( f : Flush ) {
  35. _set_flush_mode(s,Std.string(f));
  36. }
  37. public function close() {
  38. _deflate_end(s);
  39. }
  40. public static function run( s : haxe.io.Bytes, level : Int ) : haxe.io.Bytes {
  41. var c = new Compress(level);
  42. c.setFlushMode(Flush.FINISH);
  43. var out = haxe.io.Bytes.alloc(_deflate_bound(c.s,s.length));
  44. var r = c.execute(s,0,out,0);
  45. c.close();
  46. if( !r.done || r.read != s.length )
  47. throw "Compression failed";
  48. return out.sub(0,r.write);
  49. }
  50. static var _deflate_init = cpp.Lib.load("zlib","deflate_init",1);
  51. static var _deflate_bound = cpp.Lib.load("zlib","deflate_bound",2);
  52. static var _deflate_buffer = cpp.Lib.load("zlib","deflate_buffer",5);
  53. static var _deflate_end = cpp.Lib.load("zlib","deflate_end",1);
  54. static var _set_flush_mode = cpp.Lib.load("zlib","set_flush_mode",2);
  55. }