rays 0.1.12 → 0.1.17

data/.doc/ext/rays/camera.cpp

@@ -0,0 +1,171 @@
+#include "rays/ruby/camera.h"
+
+
+#include "rays/ruby/image.h"
+#include "defs.h"
+
+
+RUCY_DEFINE_VALUE_FROM_TO(Rays::Camera)
+
+#define THIS  to<Rays::Camera*>(self)
+
+#define CHECK RUCY_CHECK_OBJECT(Rays::Camera, self)
+
+
+static
+VALUE alloc(VALUE klass)
+{
+	return new_type<Rays::Camera>(klass);
+}
+
+static
+VALUE setup(VALUE self, VALUE device_name, VALUE min_width, VALUE min_height, VALUE resize, VALUE crop)
+{
+	RUCY_CHECK_OBJ(Rays::Camera, self);
+
+	*THIS = Rays::Camera(
+		device_name ? to<const char*>(device_name) : NULL,
+		to<int>(min_width), to<int>(min_height),
+		to<bool>(resize), to<bool>(crop));
+	return self;
+}
+
+static
+VALUE start(VALUE self)
+{
+	CHECK;
+	return value(THIS->start());
+}
+
+static
+VALUE stop(VALUE self)
+{
+	CHECK;
+	THIS->stop();
+}
+
+static
+VALUE is_active(VALUE self)
+{
+	CHECK;
+	return value(THIS->is_active());
+}
+
+static
+VALUE set_min_width(VALUE self, VALUE width)
+{
+	CHECK;
+	THIS->set_min_width(to<int>(width));
+	return value(THIS->min_width());
+}
+
+static
+VALUE min_width(VALUE self)
+{
+	CHECK;
+	return value(THIS->min_width());
+}
+
+static
+VALUE set_min_height(VALUE self, VALUE height)
+{
+	CHECK;
+	THIS->set_min_height(to<int>(height));
+	return value(THIS->min_height());
+}
+
+static
+VALUE min_height(VALUE self)
+{
+	CHECK;
+	return value(THIS->min_height());
+}
+
+static
+VALUE set_resize(VALUE self, VALUE resize)
+{
+	CHECK;
+	THIS->set_resize(to<bool>(resize));
+	return value(THIS->is_resize());
+}
+
+static
+VALUE is_resize(VALUE self)
+{
+	CHECK;
+	return value(THIS->is_resize());
+}
+
+static
+VALUE set_crop(VALUE self, VALUE crop)
+{
+	CHECK;
+	THIS->set_crop(to<bool>(crop));
+	return value(THIS->is_crop());
+}
+
+static
+VALUE is_crop(VALUE self)
+{
+	CHECK;
+	return value(THIS->is_crop());
+}
+
+static
+VALUE image(VALUE self)
+{
+	CHECK;
+	const Rays::Image* img = THIS->image();
+	return img ? value(*img) : nil();
+}
+
+static
+VALUE device_names(VALUE self)
+{
+	auto names = Rays::get_camera_device_names();
+
+	std::vector<Value> v;
+	for (auto it = names.begin(), end = names.end(); it != end; ++it)
+		v.emplace_back(value(it->c_str()));
+	return value(v.size(), &v[0]);
+}
+
+
+static Class cCamera;
+
+void
+Init_camera ()
+{
+	Module mRays = rb_define_module("Rays");
+
+	cCamera = rb_define_class_under(mRays, "Camera", rb_cObject);
+	rb_define_alloc_func(cCamera, alloc);
+	rb_define_private_method(cCamera, "setup", RUBY_METHOD_FUNC(setup), 5);
+	rb_define_method(cCamera, "start", RUBY_METHOD_FUNC(start), 0);
+	rb_define_method(cCamera, "stop", RUBY_METHOD_FUNC(stop), 0);
+	cCamera.define_method("active?", is_active);
+	rb_define_method(cCamera, "min_width=", RUBY_METHOD_FUNC(set_min_width), 1);
+	rb_define_method(cCamera, "min_width", RUBY_METHOD_FUNC(min_width), 0);
+	rb_define_method(cCamera, "min_height=", RUBY_METHOD_FUNC(set_min_height), 1);
+	rb_define_method(cCamera, "min_height", RUBY_METHOD_FUNC(min_height), 0);
+	rb_define_method(cCamera, "resize=", RUBY_METHOD_FUNC(set_resize), 1);
+	cCamera.define_method("resize?",  is_resize);
+	rb_define_method(cCamera, "crop=", RUBY_METHOD_FUNC(set_crop), 1);
+	cCamera.define_method("crop?",    is_crop);
+	rb_define_method(cCamera, "image", RUBY_METHOD_FUNC(image), 0);
+	rb_define_module_function(cCamera, "device_names", RUBY_METHOD_FUNC(device_names), 0);
+}
+
+
+namespace Rays
+{
+
+
+	Class
+	camera_class ()
+	{
+		return cCamera;
+	}
+
+
+}// Rays

data/.doc/ext/rays/color.cpp

@@ -1,14 +1,11 @@
 #include "rays/ruby/color.h"
 
 
-#include <rucy.h>
+#include <map>
 #include "defs.h"
 
 
-using namespace Rucy;
-
-
-RUCY_DEFINE_VALUE_FROM_TO(Rays::Color)
+RUCY_DEFINE_VALUE_OR_ARRAY_FROM_TO(Rays::Color)
 
 #define THIS  to<Rays::Color*>(self)
 
@@ -22,34 +19,13 @@ VALUE alloc(VALUE klass)
 }
 
 static
-VALUE setup(VALUE self)
+VALUE initialize(VALUE self)
 {
 	CHECK;
 	check_arg_count(__FILE__, __LINE__, "Color#initialize", argc, 0, 1, 2, 3, 4);
 
-	if (argc == 0) return self;
-
-	switch (argc)
-	{
-		case 1:
-			*THIS = Rays::Color(to<float>(argv[0]));
-			break;
-
-		case 2:
-			*THIS = Rays::Color(to<float>(argv[0]), to<float>(argv[1]));
-			break;
-
-		case 3:
-			*THIS = Rays::Color(
-				to<float>(argv[0]), to<float>(argv[1]), to<float>(argv[2]));
-			break;
-
-		case 4:
-			*THIS = Rays::Color(
-				to<float>(argv[0]), to<float>(argv[1]),
-				to<float>(argv[2]), to<float>(argv[3]));
-			break;
-	}
+	if (argc >= 1)
+		*THIS = to<Rays::Color>(argc, argv);
 
 	return self;
 }
@@ -127,6 +103,82 @@ VALUE get_alpha(VALUE self)
 }
 
 
+typedef std::map<Rays::String, Rays::Color> ColorMap;
+
+static ColorMap&
+get_color_map ()
+{
+	static ColorMap map;
+	if (map.empty())
+	{
+		map["no"]          =
+		map["none"]        =
+		map["transp"]      =
+		map["transparent"] = Rays::gray(0, 0);
+
+		map["black"]      = Rays::rgb8(  0,   0,   0);
+		map["white"]      = Rays::rgb8(255, 241, 232);
+		map["gray"]       =
+		map["lightgray"]  = Rays::rgb8(194, 195, 199);
+		map["darkgray"]   = Rays::rgb8( 95,  87,  79);
+		map["brown"]      = Rays::rgb8(171,  82,  54);
+		map["red"]        = Rays::rgb8(255,   0,  77);
+		map["orange"]     = Rays::rgb8(255, 163,   0);
+		map["yellow"]     = Rays::rgb8(255, 236,  39);
+		map["green"]      = Rays::rgb8(  0, 228,  54);
+		map["darkgreen"]  = Rays::rgb8(  0, 135,  81);
+		map["blue"]       = Rays::rgb8( 41, 173, 255);
+		map["darkblue"]   = Rays::rgb8( 29,  43,  83);
+		map["indigo"]     = Rays::rgb8(131, 118, 156);
+		map["pink"]       = Rays::rgb8(255, 119, 168);
+		map["peach"]      = Rays::rgb8(255, 204, 170);
+		map["darkpurple"] = Rays::rgb8(126,  37,  83);
+	}
+	return map;
+}
+
+static const Rays::String
+to_color_name (const char* name)
+{
+	return Rays::String(name).downcase();
+}
+
+static const Rays::Color&
+find_color (const char* name)
+{
+	assert(name);
+
+	const ColorMap& map = get_color_map();
+	ColorMap::const_iterator it = map.find(to_color_name(name));
+	if (it == map.end())
+		argument_error(__FILE__, __LINE__, "color '%s' is not found.", name);
+
+	return it->second;
+}
+
+static
+VALUE hsv(VALUE self)
+{
+	check_arg_count(__FILE__, __LINE__, "Color.hsv", argc, 3, 4);
+
+	float h = to<float>(argv[0]);
+	float s = to<float>(argv[1]);
+	float v = to<float>(argv[2]);
+	if (argc >= 4)
+		return value(Rays::hsv(h, s, v, to<float>(argv[3])));
+	else
+		return value(Rays::hsv(h, s, v));
+}
+
+static
+VALUE set_palette_color(VALUE self)
+{
+	check_arg_count(__FILE__, __LINE__, "Color.set_palette_color", argc, 2, 3, 4, 5);
+
+	get_color_map()[to_color_name(argv[0].c_str())] = to<Rays::Color>(argc - 1, &argv[1]);
+}
+
+
 static Class cColor;
 
 void
@@ -136,7 +188,7 @@ Init_color ()
 
 	cColor = rb_define_class_under(mRays, "Color", rb_cObject);
 	rb_define_alloc_func(cColor, alloc);
-	rb_define_private_method(cColor, "setup", RUBY_METHOD_FUNC(setup), -1);
+	rb_define_private_method(cColor, "initialize", RUBY_METHOD_FUNC(initialize), -1);
 	rb_define_private_method(cColor, "initialize_copy", RUBY_METHOD_FUNC(initialize_copy), 1);
 	rb_define_method(cColor, "red=", RUBY_METHOD_FUNC(set_red), 1);
 	rb_define_method(cColor, "red", RUBY_METHOD_FUNC(get_red), 0);
@@ -146,6 +198,8 @@ Init_color ()
 	rb_define_method(cColor, "blue", RUBY_METHOD_FUNC(get_blue), 0);
 	rb_define_method(cColor, "alpha=", RUBY_METHOD_FUNC(set_alpha), 1);
 	rb_define_method(cColor, "alpha", RUBY_METHOD_FUNC(get_alpha), 0);
+	rb_define_module_function(cColor, "hsv", RUBY_METHOD_FUNC(hsv), -1);
+	rb_define_module_function(cColor, "set_palette_color", RUBY_METHOD_FUNC(set_palette_color), -1);
 }
 
 
@@ -153,31 +207,155 @@ namespace Rucy
 {
 
 
+	static int
+	char2hex (char c)
+	{
+		     if ('0' <= c && c <= '9') return      c - '0';
+		else if ('a' <= c && c <= 'f') return 10 + c - 'a';
+		else if ('A' <= c && c <= 'F') return 10 + c - 'A';
+		else                           return -1;
+	}
+
+	static bool
+	parse_channel (float* channel, const char* str, size_t nchars, size_t index)
+	{
+		assert(channel && str && 1 <= nchars && nchars <= 2 && 0 <= index && index <= 3);
+
+		const char* p = str + index * nchars;
+
+		if (nchars == 1)
+		{
+			int c0 = char2hex(p[0]);
+			if (c0 < 0)
+				return false;
+
+			assert(c0 < 16);
+
+			*channel = c0 / 15.f;
+			return true;
+		}
+		else
+		{
+			int c0 = char2hex(p[0]);
+			int c1 = char2hex(p[1]);
+			if (c0 < 0 || c1 < 0)
+				return false;
+
+			assert(c0 < 16 && c1 < 16);
+
+			*channel = (c0 * 16 + c1) / 255.f;
+			return true;
+		}
+	}
+
+	static bool
+	parse_string (Rays::Color* color, const char* str)
+	{
+		assert(color && str);
+
+		if (*str != '#') return false;
+		++str;
+
+		size_t len = strlen(str);
+		switch (len)
+		{
+			case 3:
+			case 6:
+			{
+				size_t nchars = len / 3;
+
+				float r, g, b;
+				if (
+					!parse_channel(&r, str, nchars, 0) ||
+					!parse_channel(&g, str, nchars, 1) ||
+					!parse_channel(&b, str, nchars, 2))
+				{
+					return false;
+				}
+
+				color->reset(r, g, b);
+				break;
+			}
+
+			case 4:
+			case 8:
+			{
+				size_t nchars = len / 4;
+
+				float r, g, b, a;
+				if (
+					!parse_channel(&r, str, nchars, 0) ||
+					!parse_channel(&g, str, nchars, 1) ||
+					!parse_channel(&b, str, nchars, 2) ||
+					!parse_channel(&a, str, nchars, 3))
+				{
+					return false;
+				}
+
+				color->reset(r, g, b, a);
+				break;
+			}
+
+			default: return false;
+		}
+
+		return true;
+	}
+
+	static Rays::Color
+	str2color (const char* str)
+	{
+		if (!str)
+			argument_error(__FILE__, __LINE__);
+
+		Rays::String str_ = str;
+		str_ = str_.strip();
+
+		Rays::Color color;
+		if (parse_string(&color, str_.c_str()))
+			return color;
+		else
+			return find_color(str_.c_str());
+	}
+
 	template <> Rays::Color
-	value_to<Rays::Color> (Value value, bool convert)
+	value_to<Rays::Color> (int argc, const Value*argv, bool convert)
 	{
+		if (argc == 1 && argv->is_array())
+		{
+			argc = argv->size();
+			argv = argv->as_array();
+		}
+
+		assert(argc == 0 || (argc > 0 && argv));
+
 		if (convert)
 		{
-			if (value.is_i() || value.is_f())
-				return Rays::Color(value.as_f(true));
-			else if (value.is_array())
+			if (argc == 0)
+				return Rays::Color();
+			else if (argv->is_nil())
+				return str2color("none");
+			else if (argv->is_s() || argv->is_sym())
+				return str2color(argv[0].c_str());
+			else if (argv->is_num())
 			{
-				int size = value.size();
-				if (size <= 0 || 4 < size)
-					Rucy::argument_error(__FILE__, __LINE__);
-
-				Value* a = value.as_array();
-				switch (size)
+				switch (argc)
 				{
-					case 1: return Rays::Color(a[0].as_f(true));
-					case 2: return Rays::Color(a[0].as_f(true), a[1].as_f(true));
-					case 3: return Rays::Color(a[0].as_f(true), a[1].as_f(true), a[2].as_f(true));
-					case 4: return Rays::Color(a[0].as_f(true), a[1].as_f(true), a[2].as_f(true), a[3].as_f(true));
+					#define V(i) argv[i].as_f(true)
+					case 1: return Rays::Color(V(0));
+					case 2: return Rays::Color(V(0), V(1));
+					case 3: return Rays::Color(V(0), V(1), V(2));
+					case 4: return Rays::Color(V(0), V(1), V(2), V(3));
+					#undef V
+					default: argument_error(__FILE__, __LINE__, "invalid array size.");
 				}
 			}
 		}
 
-		return value_to<Rays::Color&>(value, convert);
+		if (argc != 1)
+			argument_error(__FILE__, __LINE__);
+
+		return value_to<Rays::Color&>(*argv, convert);
 	}