Start line:  
End line:  

Snippet Preview

Snippet HTML Code

Stack Overflow Questions
  /*
   **** BEGIN LICENSE BLOCK *****
   * Version: EPL 1.0/GPL 2.0/LGPL 2.1
   *
   * The contents of this file are subject to the Eclipse Public
   * License Version 1.0 (the "License"); you may not use this file
   * except in compliance with the License. You may obtain a copy of
   * the License at http://www.eclipse.org/legal/epl-v10.html
   *
  * Software distributed under the License is distributed on an "AS
  * IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
  * implied. See the License for the specific language governing
  * rights and limitations under the License.
  *
  * Alternatively, the contents of this file may be used under the terms of
  * either of the GNU General Public License Version 2 or later (the "GPL"),
  * or the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  * in which case the provisions of the GPL or the LGPL are applicable instead
  * of those above. If you wish to allow use of your version of this file only
  * under the terms of either the GPL or the LGPL, and not to allow others to
  * use your version of this file under the terms of the EPL, indicate your
  * decision by deleting the provisions above and replace them with the notice
  * and other provisions required by the GPL or the LGPL. If you do not delete
  * the provisions above, a recipient may use your version of this file under
  * the terms of any one of the EPL, the GPL or the LGPL.
  ***** END LICENSE BLOCK *****/
 package org.jruby.ext.zlib;
 
 import static org.jruby.CompatVersion.RUBY1_8;
 import static org.jruby.CompatVersion.RUBY1_9;
 import org.jruby.Ruby;
 import static org.jruby.runtime.Visibility.PRIVATE;
 
 @JRubyClass(name = "Zlib::GzipWriter", parent = "Zlib::GzipFile")
 public class JZlibRubyGzipWriter extends RubyGzipFile {
     protected static final ObjectAllocator GZIPWRITER_ALLOCATOR = new ObjectAllocator() {
         @Override
         public IRubyObject allocate(Ruby runtimeRubyClass klass) {
             return new JZlibRubyGzipWriter(runtimeklass);
         }
     };
 
     @JRubyMethod(name = "new", rest = true, meta = true)
     public static JZlibRubyGzipWriter newInstance(IRubyObject recvIRubyObject[] argsBlock block) {
         RubyClass klass = (RubyClassrecv;
         JZlibRubyGzipWriter result = (JZlibRubyGzipWriterklass.allocate();
         
         result.callInit(argsblock);
         
         return result;
     }
 
     @JRubyMethod(name = "open", required = 1, optional = 2, meta = true, compat = )
     public static IRubyObject open18(ThreadContext contextIRubyObject recvIRubyObject[] argsBlock block) {
         Ruby runtime = recv.getRuntime();
         
         args[0] = Helpers.invoke(contextruntime.getFile(), "open"args[0], runtime.newString("wb"));
         
         JZlibRubyGzipWriter gzio = newInstance(recvargsblock);
         
         return RubyGzipFile.wrapBlock(contextgzioblock);
     }
 
     @JRubyMethod(name = "open", required = 1, optional = 3, meta = true, compat = )
     public static IRubyObject open19(ThreadContext contextIRubyObject recvIRubyObject[] argsBlock block) {
         Ruby runtime = recv.getRuntime();
         
        args[0] = Helpers.invoke(contextruntime.getFile(), "open"args[0], runtime.newString("wb"));
        
        JZlibRubyGzipWriter gzio = newInstance(recvargsblock);
        
        return RubyGzipFile.wrapBlock(contextgzioblock);
    }
    public JZlibRubyGzipWriter(Ruby runtimeRubyClass type) {
        super(runtimetype);
    }
    @JRubyMethod(name = "initialize", required = 1, rest = true, visibility = , compat = )
    public IRubyObject initialize(IRubyObject[] args) {
        Ruby runtime = getRuntime();
        
         = processLevel(args.lengthargsruntime);
        
        // unused; could not figure out how to get JZlib to take this right
        /*int strategy = */processStrategy(args.lengthargs);
        
        return initializeCommon(args[0], );
    }
    @JRubyMethod(name = "initialize", rest = true, visibility = , compat = )
    public IRubyObject initialize19(ThreadContext contextIRubyObject[] argsBlock unused) {
        Ruby runtime = context.getRuntime();
        IRubyObject opt = context.nil;
        
        int argc = args.length;
        if (argc > 1) {
            opt = TypeConverter.checkHashType(runtimeopt);
            if (!opt.isNil()) argc--;
        }
        
         = processLevel(argcargsruntime);
        
        // unused; could not figure out how to get JZlib to take this right
        /*int strategy = */processStrategy(argcargs);
        
        initializeCommon(args[0], );
        
        ecopts(contextopt);
        
        return this;
    }
    // These methods are here to avoid defining a singleton #path on every instance, as in MRI
    @JRubyMethod(compat = )
    public IRubyObject path(ThreadContext context) {
        return this..callMethod(context"path");
    }
    
    @JRubyMethod(name = "respond_to?", frame = true, compat = )
    public IRubyObject respond_to(ThreadContext contextIRubyObject name) {
        if (name.asJavaString().equals("path")) {
            return this..callMethod(context"respond_to?"name);
        }
        
        return Helpers.invokeSuper(contextthisname.);
    }
    private int processStrategy(int argcIRubyObject[] args) {
        return argc < 3 ? . : RubyZlib.FIXNUMARG(args[2], .);
    }
    private int processLevel(int argcIRubyObject[] argsRuby runtime) {
        int level = argc < 2 ? . : RubyZlib.FIXNUMARG(args[1], .);
        checkLevel(runtimelevel);
        return level;
    }
    private IRubyObject initializeCommon(IRubyObject streamint level) {
         = (RubyObjectstream;
        try {
            // the 15+16 here is copied from a Deflater default constructor
            Deflater deflater = new Deflater(level, 15+16, false);
             = new GZIPOutputStream(new IOOutputStream(falsefalse), deflater, 512, false);
            return this;
        } catch (IOException ioe) {
            throw getRuntime().newIOErrorFromException(ioe);
        }
    }
    
    private static void checkLevel(Ruby runtimeint level) {
        if (level != . && (level < . || level > .)) {
            throw RubyZlib.newStreamError(runtime"stream error: invalid level");
        }
    }
    @Override
    @JRubyMethod(name = "close")
    public IRubyObject close() {
        if (!) {
            try {
                .close();
                if (.respondsTo("close")) {
                    .callMethod(.getRuntime().getCurrentContext(), "close");
                }
            } catch (IOException ioe) {
                throw getRuntime().newIOErrorFromException(ioe);
            }
        }
        
        this. = true;
        
        return getRuntime().getNil();
    }
    @JRubyMethod(name = {"append""<<"}, required = 1)
    public IRubyObject append(IRubyObject p1) {
        this.write(p1);
        
        return this;
    }
    @JRubyMethod(name = "printf", required = 1, rest = true)
    public IRubyObject printf(ThreadContext contextIRubyObject[] args) {
        write(RubyKernel.sprintf(contextthisargs));
        
        return context.getRuntime().getNil();
    }
    @JRubyMethod(name = "print", rest = true)
    public IRubyObject print(IRubyObject[] args) {
        if (args.length != 0) {
            for (int i = 0, j = args.lengthi < ji++) {
                write(args[i]);
            }
        }
        IRubyObject sep = getRuntime().getGlobalVariables().get("$\\");
        if (!sep.isNil()) write(sep);
        return getRuntime().getNil();
    }
    @JRubyMethod(name = {"pos""tell"})
    public IRubyObject pos() {
        return RubyNumeric.int2fix(getRuntime(), .getTotalIn());
    }
    @JRubyMethod(name = "orig_name=", required = 1)
    public IRubyObject set_orig_name(IRubyObject obj) {
         = obj.convertToString();
        
        try {
            .setName(.toString());
        } catch (GZIPException e) {
            throw RubyZlib.newGzipFileError(getRuntime(), "header is already written");
        }
        
        return obj;
    }
    @JRubyMethod(name = "comment=", required = 1)
    public IRubyObject set_comment(IRubyObject obj) {
         = obj.convertToString();
        
        try {
            .setComment(.toString());
        } catch (GZIPException e) {
            throw RubyZlib.newGzipFileError(getRuntime(), "header is already written");
        }
        
        return obj;
    }
    private void ensureNonNull(RubyString obj) {
        String str = obj.toString();
        
        if (str.indexOf('\0') >= 0) {
            String trim = str.substring(0, str.toString().indexOf('\0'));
            obj.setValue(new ByteList(trim.getBytes()));
        }
    }
    @JRubyMethod(name = "putc", required = 1)
    public IRubyObject putc(IRubyObject p1) {
        try {
            .write(RubyNumeric.num2chr(p1));
            
            return p1;
        } catch (IOException ioe) {
            throw getRuntime().newIOErrorFromException(ioe);
        }
    }
    @JRubyMethod(name = "puts", rest = true)
    public IRubyObject puts(ThreadContext contextIRubyObject[] args) {
        IRubyObject _sio = getRuntime().getClass("StringIO").newInstance(contextnew IRubyObject[0], .);
        if (_sio instanceof RubyStringIO) {
            RubyStringIO sio = (RubyStringIO)_sio;
            sio.puts(contextargs);
            write(sio.string());
        } else {
            StringIO sio = (StringIO)_sio;
            sio.puts(contextargs);
            write(sio.string(context));
        }
        return getRuntime().getNil();
    }
    @Override
    public IRubyObject finish() {
        if (!) {
            try {
                .finish();
            } catch (IOException ioe) {
                throw getRuntime().newIOErrorFromException(ioe);
            }
        }
        
         = true;
        
        return ;
    }
    @JRubyMethod(name = "flush", optional = 1)
    public IRubyObject flush(IRubyObject[] args) {
        int flush = .;
        
        if (args.length > 0 && !args[0].isNil()) {
            flush = RubyNumeric.fix2int(args[0]);
        }
        
        boolean tmp = .getSyncFlush();
        try {
            if (flush != 0 /*
                     * NO_FLUSH
                     */) {
                .setSyncFlush(true);
            }
            .flush();
        } catch (IOException ioe) {
            throw getRuntime().newIOErrorFromException(ioe);
        } finally {
            .setSyncFlush(tmp);
        }
        
        return getRuntime().getNil();
    }
    @JRubyMethod(name = "mtime=", required = 1)
    public IRubyObject set_mtime(IRubyObject arg) {
        if (arg instanceof RubyTime) {
            this. = ((RubyTimearg);
        } else if (arg.isNil()) {
            // ...nothing
        } else {
            this..setDateTime(new DateTime(RubyNumeric.fix2long(arg) * 1000));
        }
        
        try {
            .setModifiedTime(this..to_i().getLongValue());
        } catch (GZIPException e) {
            throw RubyZlib.newGzipFileError(getRuntime(), "header is already written");
        }
        
        return getRuntime().getNil();
    }
    @Override
    @JRubyMethod(name = "crc")
    public IRubyObject crc() {
        long crc = 0L;
        
        try {
            crc = .getCRC();
        } catch (GZIPException e) {
            // not calculated yet
        }
        
        return getRuntime().newFixnum(crc);
    }
    @JRubyMethod(name = "write", required = 1)
    public IRubyObject write(IRubyObject p1) {
        Ruby runtime = getRuntime();
        ByteList bytes = p1.asString().getByteList();
        
        if (runtime.is1_9()) {
            if ( != null
                    &&  != .) {
                bytes = Transcoder.strConvEncOpts(runtime.getCurrentContext(), bytesbytes.getEncoding(),
                        , 0, runtime.getNil());
            }
        }
        
        try {
            // TODO: jzlib-1.1.0.jar throws IndexOutOfBoundException for zero length buffer.
            if (bytes.length() > 0) {
                .write(bytes.getUnsafeBytes(), bytes.begin(), bytes.length());
            }
            
            return runtime.newFixnum(bytes.length());
        } catch (IOException ioe) {
            throw runtime.newIOErrorFromException(ioe);
        }
    }
    @Override
    public IRubyObject set_sync(IRubyObject arg) {
        IRubyObject s = super.set_sync(arg);
        
        .setSyncFlush();
        
        return s;
    }
    
    private GZIPOutputStream io;
New to GrepCode? Check out our FAQ X