FileInput.hx 2.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697
  1. /*
  2. * Copyright (C)2005-2018 Haxe Foundation
  3. *
  4. * Permission is hereby granted, free of charge, to any person obtaining a
  5. * copy of this software and associated documentation files (the "Software"),
  6. * to deal in the Software without restriction, including without limitation
  7. * the rights to use, copy, modify, merge, publish, distribute, sublicense,
  8. * and/or sell copies of the Software, and to permit persons to whom the
  9. * Software is furnished to do so, subject to the following conditions:
  10. *
  11. * The above copyright notice and this permission notice shall be included in
  12. * all copies or substantial portions of the Software.
  13. *
  14. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  15. * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  16. * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  17. * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  18. * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  19. * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  20. * DEALINGS IN THE SOFTWARE.
  21. */
  22. package sys.io;
  23. import lua.FileHandle;
  24. import lua.Io;
  25. import lua.NativeStringTools;
  26. import lua.Boot;
  27. import lua.Os;
  28. import haxe.io.Bytes;
  29. import haxe.io.Error;
  30. import haxe.io.Eof;
  31. class FileInput extends haxe.io.Input {
  32. var f:FileHandle;
  33. var _eof:Bool;
  34. public function new(f:FileHandle){
  35. if (f == null) throw 'Invalid filehandle : $f';
  36. this.bigEndian = Boot.platformBigEndian;
  37. this.f = f;
  38. this._eof = false;
  39. }
  40. inline public function seek( p : Int, pos : FileSeek ) : Void {
  41. var arg = switch(pos){
  42. case SeekBegin : "set";
  43. case SeekCur : "cur";
  44. case SeekEnd : "end";
  45. }
  46. _eof = false;
  47. return f.seek(arg, p);
  48. }
  49. inline public function tell() : Int {
  50. return f.seek();
  51. }
  52. inline public function eof() : Bool {
  53. return _eof;
  54. }
  55. override inline public function readByte() : Int {
  56. var byte = f.read(1);
  57. if (byte == null) {
  58. _eof = true;
  59. throw new haxe.io.Eof();
  60. }
  61. return NativeStringTools.byte(byte);
  62. }
  63. override function readBytes( s : Bytes, pos : Int, len : Int ) : Int {
  64. if(eof()) throw new haxe.io.Eof();
  65. return super.readBytes(s, pos, len);
  66. }
  67. override inline public function close() : Void {
  68. f.close();
  69. }
  70. override public function readAll( ?bufsize : Int ) : Bytes {
  71. if( bufsize == null )
  72. bufsize = (1 << 14); // 16 Ko
  73. var buf = Bytes.alloc(bufsize);
  74. var total = new haxe.io.BytesBuffer();
  75. try {
  76. while( true ) {
  77. var len = readBytes(buf,0,bufsize);
  78. if (len == 0) break;
  79. total.addBytes(buf,0,len);
  80. }
  81. } catch( e : Eof ) {
  82. _eof = true;
  83. }
  84. return total.getBytes();
  85. }
  86. }