epeg 1.0

data/epeg/vendor/libjpeg-turbo-2.0.4/wrtarga.c

@@ -0,0 +1,261 @@
+/*
+ * wrtarga.c
+ *
+ * This file was part of the Independent JPEG Group's software:
+ * Copyright (C) 1991-1996, Thomas G. Lane.
+ * libjpeg-turbo Modifications:
+ * Copyright (C) 2017, D. R. Commander.
+ * For conditions of distribution and use, see the accompanying README.ijg
+ * file.
+ *
+ * This file contains routines to write output images in Targa format.
+ *
+ * These routines may need modification for non-Unix environments or
+ * specialized applications.  As they stand, they assume output to
+ * an ordinary stdio stream.
+ *
+ * Based on code contributed by Lee Daniel Crocker.
+ */
+
+#include "cdjpeg.h"             /* Common decls for cjpeg/djpeg applications */
+
+#ifdef TARGA_SUPPORTED
+
+
+/*
+ * To support 12-bit JPEG data, we'd have to scale output down to 8 bits.
+ * This is not yet implemented.
+ */
+
+#if BITS_IN_JSAMPLE != 8
+  Sorry, this code only copes with 8-bit JSAMPLEs. /* deliberate syntax err */
+#endif
+
+
+/* Private version of data destination object */
+
+typedef struct {
+  struct djpeg_dest_struct pub; /* public fields */
+
+  char *iobuffer;               /* physical I/O buffer */
+  JDIMENSION buffer_width;      /* width of one row */
+} tga_dest_struct;
+
+typedef tga_dest_struct *tga_dest_ptr;
+
+
+LOCAL(void)
+write_header(j_decompress_ptr cinfo, djpeg_dest_ptr dinfo, int num_colors)
+/* Create and write a Targa header */
+{
+  char targaheader[18];
+
+  /* Set unused fields of header to 0 */
+  MEMZERO(targaheader, sizeof(targaheader));
+
+  if (num_colors > 0) {
+    targaheader[1] = 1;         /* color map type 1 */
+    targaheader[5] = (char)(num_colors & 0xFF);
+    targaheader[6] = (char)(num_colors >> 8);
+    targaheader[7] = 24;        /* 24 bits per cmap entry */
+  }
+
+  targaheader[12] = (char)(cinfo->output_width & 0xFF);
+  targaheader[13] = (char)(cinfo->output_width >> 8);
+  targaheader[14] = (char)(cinfo->output_height & 0xFF);
+  targaheader[15] = (char)(cinfo->output_height >> 8);
+  targaheader[17] = 0x20;       /* Top-down, non-interlaced */
+
+  if (cinfo->out_color_space == JCS_GRAYSCALE) {
+    targaheader[2] = 3;         /* image type = uncompressed grayscale */
+    targaheader[16] = 8;        /* bits per pixel */
+  } else {                      /* must be RGB */
+    if (num_colors > 0) {
+      targaheader[2] = 1;       /* image type = colormapped RGB */
+      targaheader[16] = 8;
+    } else {
+      targaheader[2] = 2;       /* image type = uncompressed RGB */
+      targaheader[16] = 24;
+    }
+  }
+
+  if (JFWRITE(dinfo->output_file, targaheader, 18) != (size_t)18)
+    ERREXIT(cinfo, JERR_FILE_WRITE);
+}
+
+
+/*
+ * Write some pixel data.
+ * In this module rows_supplied will always be 1.
+ */
+
+METHODDEF(void)
+put_pixel_rows(j_decompress_ptr cinfo, djpeg_dest_ptr dinfo,
+               JDIMENSION rows_supplied)
+/* used for unquantized full-color output */
+{
+  tga_dest_ptr dest = (tga_dest_ptr)dinfo;
+  register JSAMPROW inptr;
+  register char *outptr;
+  register JDIMENSION col;
+
+  inptr = dest->pub.buffer[0];
+  outptr = dest->iobuffer;
+  for (col = cinfo->output_width; col > 0; col--) {
+    outptr[0] = (char)GETJSAMPLE(inptr[2]); /* RGB to BGR order */
+    outptr[1] = (char)GETJSAMPLE(inptr[1]);
+    outptr[2] = (char)GETJSAMPLE(inptr[0]);
+    inptr += 3, outptr += 3;
+  }
+  (void)JFWRITE(dest->pub.output_file, dest->iobuffer, dest->buffer_width);
+}
+
+METHODDEF(void)
+put_gray_rows(j_decompress_ptr cinfo, djpeg_dest_ptr dinfo,
+              JDIMENSION rows_supplied)
+/* used for grayscale OR quantized color output */
+{
+  tga_dest_ptr dest = (tga_dest_ptr)dinfo;
+  register JSAMPROW inptr;
+  register char *outptr;
+  register JDIMENSION col;
+
+  inptr = dest->pub.buffer[0];
+  outptr = dest->iobuffer;
+  for (col = cinfo->output_width; col > 0; col--) {
+    *outptr++ = (char)GETJSAMPLE(*inptr++);
+  }
+  (void)JFWRITE(dest->pub.output_file, dest->iobuffer, dest->buffer_width);
+}
+
+
+/*
+ * Write some demapped pixel data when color quantization is in effect.
+ * For Targa, this is only applied to grayscale data.
+ */
+
+METHODDEF(void)
+put_demapped_gray(j_decompress_ptr cinfo, djpeg_dest_ptr dinfo,
+                  JDIMENSION rows_supplied)
+{
+  tga_dest_ptr dest = (tga_dest_ptr)dinfo;
+  register JSAMPROW inptr;
+  register char *outptr;
+  register JSAMPROW color_map0 = cinfo->colormap[0];
+  register JDIMENSION col;
+
+  inptr = dest->pub.buffer[0];
+  outptr = dest->iobuffer;
+  for (col = cinfo->output_width; col > 0; col--) {
+    *outptr++ = (char)GETJSAMPLE(color_map0[GETJSAMPLE(*inptr++)]);
+  }
+  (void)JFWRITE(dest->pub.output_file, dest->iobuffer, dest->buffer_width);
+}
+
+
+/*
+ * Startup: write the file header.
+ */
+
+METHODDEF(void)
+start_output_tga(j_decompress_ptr cinfo, djpeg_dest_ptr dinfo)
+{
+  tga_dest_ptr dest = (tga_dest_ptr)dinfo;
+  int num_colors, i;
+  FILE *outfile;
+
+  if (cinfo->out_color_space == JCS_GRAYSCALE) {
+    /* Targa doesn't have a mapped grayscale format, so we will */
+    /* demap quantized gray output.  Never emit a colormap. */
+    write_header(cinfo, dinfo, 0);
+    if (cinfo->quantize_colors)
+      dest->pub.put_pixel_rows = put_demapped_gray;
+    else
+      dest->pub.put_pixel_rows = put_gray_rows;
+  } else if (cinfo->out_color_space == JCS_RGB) {
+    if (cinfo->quantize_colors) {
+      /* We only support 8-bit colormap indexes, so only 256 colors */
+      num_colors = cinfo->actual_number_of_colors;
+      if (num_colors > 256)
+        ERREXIT1(cinfo, JERR_TOO_MANY_COLORS, num_colors);
+      write_header(cinfo, dinfo, num_colors);
+      /* Write the colormap.  Note Targa uses BGR byte order */
+      outfile = dest->pub.output_file;
+      for (i = 0; i < num_colors; i++) {
+        putc(GETJSAMPLE(cinfo->colormap[2][i]), outfile);
+        putc(GETJSAMPLE(cinfo->colormap[1][i]), outfile);
+        putc(GETJSAMPLE(cinfo->colormap[0][i]), outfile);
+      }
+      dest->pub.put_pixel_rows = put_gray_rows;
+    } else {
+      write_header(cinfo, dinfo, 0);
+      dest->pub.put_pixel_rows = put_pixel_rows;
+    }
+  } else {
+    ERREXIT(cinfo, JERR_TGA_COLORSPACE);
+  }
+}
+
+
+/*
+ * Finish up at the end of the file.
+ */
+
+METHODDEF(void)
+finish_output_tga(j_decompress_ptr cinfo, djpeg_dest_ptr dinfo)
+{
+  /* Make sure we wrote the output file OK */
+  fflush(dinfo->output_file);
+  if (ferror(dinfo->output_file))
+    ERREXIT(cinfo, JERR_FILE_WRITE);
+}
+
+
+/*
+ * Re-calculate buffer dimensions based on output dimensions.
+ */
+
+METHODDEF(void)
+calc_buffer_dimensions_tga(j_decompress_ptr cinfo, djpeg_dest_ptr dinfo)
+{
+  tga_dest_ptr dest = (tga_dest_ptr)dinfo;
+
+  dest->buffer_width = cinfo->output_width * cinfo->output_components;
+}
+
+
+/*
+ * The module selection routine for Targa format output.
+ */
+
+GLOBAL(djpeg_dest_ptr)
+jinit_write_targa(j_decompress_ptr cinfo)
+{
+  tga_dest_ptr dest;
+
+  /* Create module interface object, fill in method pointers */
+  dest = (tga_dest_ptr)
+    (*cinfo->mem->alloc_small) ((j_common_ptr)cinfo, JPOOL_IMAGE,
+                                sizeof(tga_dest_struct));
+  dest->pub.start_output = start_output_tga;
+  dest->pub.finish_output = finish_output_tga;
+  dest->pub.calc_buffer_dimensions = calc_buffer_dimensions_tga;
+
+  /* Calculate output image dimensions so we can allocate space */
+  jpeg_calc_output_dimensions(cinfo);
+
+  /* Create I/O buffer. */
+  dest->pub.calc_buffer_dimensions(cinfo, (djpeg_dest_ptr)dest);
+  dest->iobuffer = (char *)
+    (*cinfo->mem->alloc_small) ((j_common_ptr)cinfo, JPOOL_IMAGE,
+                                (size_t)(dest->buffer_width * sizeof(char)));
+
+  /* Create decompressor output buffer. */
+  dest->pub.buffer = (*cinfo->mem->alloc_sarray)
+    ((j_common_ptr)cinfo, JPOOL_IMAGE, dest->buffer_width, (JDIMENSION)1);
+  dest->pub.buffer_height = 1;
+
+  return (djpeg_dest_ptr)dest;
+}
+
+#endif /* TARGA_SUPPORTED */

data/epeg.c

@@ -0,0 +1,131 @@
+#include <ruby.h>
+#include <ruby/intern.h>
+#include <Epeg.h>
+
+VALUE cEpeg;
+ID idsize;
+ID idclosed;
+
+static VALUE rb_epeg_thumbnail(klass, fname, x, y)
+	VALUE klass, fname, x, y;
+{
+	Epeg_Image *image = NULL;
+	unsigned char *data = NULL;
+	int size = 0;
+	VALUE retval;
+
+	Check_Type(fname, T_STRING);
+	if(StringValueCStr(fname)[0] != '/') {
+		fname = rb_file_s_expand_path(1, &fname);
+	}
+
+	image = epeg_file_open(StringValueCStr(fname));
+	if(!image) {
+		rb_raise(rb_eStandardError, "Can't open image");
+	}
+	epeg_decode_size_set(image, NUM2INT(x), NUM2INT(y));
+	epeg_memory_output_set(image, &data, &size);
+	if(epeg_encode(image) != 0) {
+		rb_raise(rb_eStandardError, "Can't encode image");
+	}
+	epeg_close(image);
+	retval = rb_str_new((const char *)data, size);
+	free(data);
+	return retval;
+}
+
+static VALUE
+rb_epeg_new(klass, fname)
+	VALUE klass, fname;
+{
+	VALUE retval;
+	Epeg_Image *image;
+	image = epeg_file_open(StringValueCStr(fname));
+	if(!image) {
+		rb_raise(rb_eStandardError, "Can't open image");
+	}
+	retval = Data_Wrap_Struct(klass, NULL, NULL, image);
+	rb_obj_call_init(retval, 0, NULL);
+	return retval;
+}
+
+static VALUE
+rb_epeg_get_size(obj)
+	VALUE obj;
+{
+	int x, y;
+	Epeg_Image *image;
+	VALUE retval;
+	retval = rb_ivar_get(obj, idsize);
+	if(RTEST(retval))
+		return retval;
+	Data_Get_Struct(obj, Epeg_Image, image);
+	epeg_size_get(image, &x, &y);
+	retval = rb_ary_new3(2, INT2NUM(x), INT2NUM(y));
+	rb_ivar_set(obj, idsize, retval);
+	return retval;
+}
+
+static VALUE
+rb_epeg_set_output_size(obj, x, y)
+	VALUE obj, x, y;
+{
+	Epeg_Image *image;
+	if(OBJ_FROZEN(obj)) {
+		rb_raise(rb_eStandardError, "can't set size now");
+	}
+	Data_Get_Struct(obj, Epeg_Image, image);
+	epeg_decode_size_set(image, NUM2INT(x), NUM2INT(y));
+	return Qnil;
+}
+
+static VALUE
+rb_epeg_finish(argc, args, obj)
+	int argc;
+	VALUE *args;
+	VALUE obj;
+{
+	Epeg_Image *image;
+	VALUE retval;
+	unsigned char *data;
+	int size;
+
+	if(OBJ_FROZEN(obj)) {
+		rb_raise(rb_eStandardError, "Epeg finished");
+	}
+
+	Data_Get_Struct(obj, Epeg_Image, image);
+
+	if(argc == 0) { // Return as string
+		epeg_memory_output_set(image, &data, &size);
+	} else if(argc == 1) { // Filename
+		Check_Type(args[0], T_STRING);
+		epeg_file_output_set(image, RSTRING_PTR(args[0]));
+	} else {
+		rb_raise(rb_eRuntimeError, "invalid arguments");
+	}
+
+	if(epeg_encode(image) != 0) {
+		rb_raise(rb_eStandardError, "Can't encode image");
+	}
+	epeg_close(image);
+	OBJ_FREEZE(obj);
+	if(argc == 0) {
+		retval = rb_str_new((const char *)data, size);
+		free(data);
+		return retval;
+	} else {
+		return Qnil;
+	}
+}
+
+void Init_epeg () {
+	idclosed = rb_intern("closed");
+	idsize = rb_intern("size");
+	cEpeg = rb_define_class("Epeg", rb_cObject);
+	rb_define_singleton_method(cEpeg, "thumbnail", rb_epeg_thumbnail, 3);
+	rb_define_singleton_method(cEpeg, "new", rb_epeg_new, 1);
+	rb_define_method(cEpeg, "size", rb_epeg_get_size, 0);
+	rb_define_method(cEpeg, "set_output_size", rb_epeg_set_output_size, 2);
+	rb_define_method(cEpeg, "finish", rb_epeg_finish, -1);
+}

data/epeg.gemspec

@@ -0,0 +1,18 @@
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+
+Gem::Specification.new do |spec|
+  spec.name          = "epeg"
+  spec.version       = 1.0
+  spec.authors       = ["Aria Stewart"]
+  spec.email         = ["aredridel@dinhe.net"]
+  spec.summary       = %q{Ruby extension for the epeg library.}
+  spec.homepage      = "http://github.com/aredridel/ruby-epeg"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["."]
+  spec.extensions    = ["extconf.rb"]
+end

data/extconf.rb

@@ -0,0 +1,80 @@
+require 'mkmf'
+
+require 'timeout'
+
+RbConfig::MAKEFILE_CONFIG['CC'] = ENV['CC'] if ENV['CC']
+
+$CFLAGS << " #{ENV["CFLAGS"]}"
+$CFLAGS << " -g"
+$CFLAGS << " -O3" unless $CFLAGS[/-O\d/]
+$CFLAGS << " -Wall -Wno-comment"
+
+cmake_flags = [ ENV["CMAKE_FLAGS"] ]
+
+def sys(cmd)
+  puts " -- #{cmd}"
+  unless ret = xsystem(cmd)
+    raise "ERROR: '#{cmd}' failed"
+  end
+  ret
+end
+
+# Thrown when we detect CMake is taking too long and we killed it
+class CMakeTimeout < StandardError
+end
+
+def self.run_cmake(timeout, args)
+  # Set to process group so we can kill it and its children
+  pgroup = Gem.win_platform? ? :new_pgroup : :pgroup
+  pid = Process.spawn("cmake #{args}", pgroup => true)
+
+  Timeout.timeout(timeout) do
+    Process.waitpid(pid)
+  end
+
+rescue Timeout::Error
+  # Kill it, #detach is essentially a background wait, since we don't actually
+  # care about waiting for it now
+  Process.kill(-9, pid)
+  Process.detach(pid)
+  raise CMakeTimeout.new("cmake has exceeded its timeout of #{timeout}s")
+end
+
+MAKE = if Gem.win_platform?
+  # On Windows, Ruby-DevKit only has 'make'.
+  find_executable('make')
+else
+  find_executable('gmake') || find_executable('make')
+end
+
+if !MAKE
+  abort "ERROR: GNU make is required to build epeg."
+end
+
+CWD = File.expand_path(File.dirname(__FILE__))
+EPEG_DIR = File.join(CWD, 'epeg')
+
+if !find_executable('cmake')
+  abort "ERROR: CMake is required to build epeg."
+end
+
+Dir.chdir(EPEG_DIR) do
+  Dir.mkdir("build") if !Dir.exist?("build")
+
+  Dir.chdir("build") do
+	# On Windows, Ruby-DevKit is MSYS-based, so ensure to use MSYS Makefiles.
+	generator = "-G \"MSYS Makefiles\"" if Gem.win_platform?
+	run_cmake(5 * 60, ".. #{cmake_flags.join(' ')} #{generator}")
+	sys(MAKE)
+  end
+end
+
+# Prepend the vendored epeg build dir to the $DEFLIBPATH.
+$DEFLIBPATH.unshift("#{EPEG_DIR}/build")
+dir_config('epeg', "#{EPEG_DIR}", "#{EPEG_DIR}/build")
+
+unless have_library 'epeg' and have_header 'Epeg.h'
+  abort "ERROR: Failed to build epeg"
+end
+
+create_makefile('epeg')

data/test.rb

@@ -0,0 +1,42 @@
+require 'epeg'
+require 'test/unit'
+
+class TestEpeg < Test::Unit::TestCase
+	def setup
+		@epeg = Epeg.new('test.jpg')
+	end
+	def test_new
+		assert_equal(Epeg, @epeg.class)
+	end
+	def test_size
+		assert_equal(@epeg.size, [182, 170])
+	end
+	def test_finish
+		assert(@epeg.finish)
+		assert_raises(StandardError) { @epeg.finish }
+	end
+	def test_output_size
+		assert_nothing_raised { @epeg.set_output_size(16,16) }
+		@epeg.finish
+		assert_raises(StandardError) { @epeg.set_output_size(16,16) }
+	end
+	def test_finish_return
+		@epeg.set_output_size(128,128)
+		assert_equal(Epeg.thumbnail('test.jpg', 128, 128), @epeg.finish)
+	end
+end
+
+class TestEpegBad < Test::Unit::TestCase
+	def test_bad_file
+		assert_raises(StandardError) {
+			Epeg.new('/dev/null')
+		}
+	end
+
+	def test_nonexistant_file
+		assert_raises(StandardError) {
+			Epeg.new('/askdfhalskdfhasldfahsdf')
+		}
+	end
+end
+