wiretap 0.1

data/ext/format.cpp

@@ -0,0 +1,347 @@
+#include "wiretap.h"
+
+void wiretap_format_free(WireTapClipFormat* format) {
+	delete format;
+}
+
+
+static VALUE wiretap_clipformat_allocate(VALUE klass) {
+	return Data_Wrap_Struct(klass, 0, wiretap_format_free, new WireTapClipFormat());
+}
+
+static VALUE wiretap_audioformat_allocate(VALUE klass) {
+	return Data_Wrap_Struct(klass, 0, wiretap_format_free, new WireTapAudioFormat());
+}
+
+/*
+ * Get width of clip
+ */
+static VALUE wiretap_format_get_width(VALUE self) {
+	WireTapClipFormat* format;
+	Data_Get_Struct(self, WireTapClipFormat, format);
+	return INT2FIX(format->width());
+}
+
+/*
+ * Set width of clip
+ */
+static VALUE wiretap_format_set_width(VALUE self, VALUE value_) {
+	VALUE value = rb_funcall(value_, to_i, 0);
+
+	WireTapClipFormat* format;
+	Data_Get_Struct(self, WireTapClipFormat, format);
+	format->setWidth(NUM2INT(value));
+	return self;
+}
+
+/*
+ * Get height of clip
+ */
+static VALUE wiretap_format_get_height(VALUE self) {
+	WireTapClipFormat* format;
+	Data_Get_Struct(self, WireTapClipFormat, format);
+	return INT2FIX(format->height());
+}
+
+/*
+ * Set height of clip
+ */
+static VALUE wiretap_format_set_height(VALUE self, VALUE value_){
+	VALUE value = rb_funcall(value_, to_i, 0);
+
+	WireTapClipFormat* format;
+	Data_Get_Struct(self, WireTapClipFormat, format);
+	format->setHeight(NUM2INT(value));
+	return self;
+}
+
+/*
+ * Get bits per pixel in clip
+ */
+static VALUE wiretap_format_get_bpp(VALUE self) {
+	WireTapClipFormat* format;
+	Data_Get_Struct(self, WireTapClipFormat, format);
+	return INT2FIX(format->bitsPerPixel());
+}
+
+/*
+ * Sets bits per pixel in clip
+ */
+static VALUE wiretap_format_set_bpp(VALUE self, VALUE value_){
+	VALUE value = rb_funcall(value_, to_i, 0);
+
+	WireTapClipFormat* format;
+	Data_Get_Struct(self, WireTapClipFormat, format);
+	format->setBitsPerPixel(NUM2INT(value));
+	return self;
+}
+
+/*
+ * Get number of channels in clip
+ */
+static VALUE wiretap_format_get_channels(VALUE self) {
+	WireTapClipFormat* format;
+	Data_Get_Struct(self, WireTapClipFormat, format);
+	return INT2FIX(format->numChannels());
+}
+
+/*
+ * Set number channels in clip
+ */
+static VALUE wiretap_format_set_channels(VALUE self, VALUE value_){
+	VALUE value = rb_funcall(value_, to_i, 0);
+
+	WireTapClipFormat* format;
+	Data_Get_Struct(self, WireTapClipFormat, format);
+	format->setNumChannels(NUM2INT(value));
+	return self;
+}
+
+/*
+ * Get buffer size in clip. It is size of one frame (really a bit more).
+ * Usually, it is not required from ruby
+ */
+static VALUE wiretap_format_get_buffer_size(VALUE self) {
+	WireTapClipFormat* format;
+	Data_Get_Struct(self, WireTapClipFormat, format);
+	return INT2FIX(format->frameBufferSize());
+}
+
+/*
+ * set buffer size for clip
+ */
+static VALUE wiretap_format_set_buffer_size(VALUE self, VALUE value_){
+	VALUE value = rb_funcall(value_, to_i, 0);
+	
+	WireTapClipFormat* format;
+	Data_Get_Struct(self, WireTapClipFormat, format);
+	format->setFrameBufferSize(NUM2INT(value));
+	return self;
+}
+
+/*
+ * Get frame rate per second of this clip
+ */
+static VALUE wiretap_format_get_rate(VALUE self) {
+	WireTapClipFormat* format;
+	Data_Get_Struct(self, WireTapClipFormat, format);
+	return rb_float_new(double(format->frameRate()));
+}
+
+/*
+ * Set frame rate per second of this clip
+ */
+static VALUE wiretap_format_set_rate(VALUE self, VALUE value_){
+	VALUE value = rb_funcall(value_, rb_intern("to_f"), 0);
+	WireTapClipFormat* format;
+	Data_Get_Struct(self, WireTapClipFormat, format);
+	format->setFrameRate(float(RFLOAT(value)->value));
+	return self;
+}
+
+/*
+ * Get pixel ratio of clip. It is not frame ratio, but pixel ratio
+ */
+static VALUE wiretap_format_get_ratio(VALUE self) {
+	WireTapClipFormat* format;
+	Data_Get_Struct(self, WireTapClipFormat, format);
+	return rb_float_new(double(format->pixelRatio()));
+}
+
+/*
+ * Set pixel ratio of clip. It is not frame ratio, but pixel ratio
+ */
+static VALUE wiretap_format_set_ratio(VALUE self, VALUE value_){
+	VALUE value = rb_funcall(value_, rb_intern("to_f"), 0);
+	WireTapClipFormat* format;
+	Data_Get_Struct(self, WireTapClipFormat, format);
+	format->setPixelRatio(float(RFLOAT(value)->value));
+	return self;
+}
+
+/*
+ * Get scan format of clip
+ */
+static VALUE wiretap_format_get_scan(VALUE self) {
+	WireTapClipFormat* format;
+	Data_Get_Struct(self, WireTapClipFormat, format);
+	return ID2SYM(rb_intern(format->scanFormatStr(format->scanFormat())));
+}
+
+/*
+ * Set scan format of clip
+ */
+static VALUE wiretap_format_set_scan(VALUE self, VALUE value){
+	WireTapClipFormat* format;
+	Data_Get_Struct(self, WireTapClipFormat, format);
+	VALUE scan = rb_funcall(value, rb_intern("to_s"), 0);
+	WireTapClipFormat::ScanFormat scan_format = format->strToScanFormat(CSTR(scan));
+	format->setScanFormat(scan_format);
+	return self;
+}
+
+/*
+ * Get format tag for clip
+ */
+static VALUE wiretap_format_get_tag(VALUE self) {
+	WireTapClipFormat* format;
+	Data_Get_Struct(self, WireTapClipFormat, format);
+	return ID2SYM(rb_intern(format->formatTag()));
+}
+
+/*
+ * Set format tag for clip
+ */
+static VALUE wiretap_format_set_tag(VALUE self, VALUE value){
+	WireTapClipFormat* format;
+	Data_Get_Struct(self, WireTapClipFormat, format);
+	VALUE tag = rb_funcall(value, rb_intern("to_s"), 0);
+	format->setFormatTag(CSTR(tag));
+	return self;
+}
+
+/*
+ * Get metadata for format
+ */
+static VALUE wiretap_format_get_meta(VALUE self) {
+	WireTapClipFormat* format;
+	Data_Get_Struct(self, WireTapClipFormat, format);
+	return rb_str_new2(format->metaData());
+}
+
+/*
+ * Set metadata for format
+ */
+static VALUE wiretap_format_set_meta(VALUE self, VALUE value){
+	WireTapClipFormat* format;
+	Data_Get_Struct(self, WireTapClipFormat, format);
+	VALUE meta = rb_funcall(value, rb_intern("to_s"), 0);
+	format->setMetaData(CSTR(meta));
+	return self;
+}
+
+/*
+ * Get meta tag of format
+ */
+static VALUE wiretap_format_get_meta_tag(VALUE self) {
+	WireTapClipFormat* format;
+	Data_Get_Struct(self, WireTapClipFormat, format);
+	return ID2SYM(rb_intern(format->metaDataTag()));
+}
+
+/*
+ * Set meta tag for format
+ */
+static VALUE wiretap_format_set_meta_tag(VALUE self, VALUE value){
+	WireTapClipFormat* format;
+	Data_Get_Struct(self, WireTapClipFormat, format);
+	VALUE meta = rb_funcall(value, rb_intern("to_s"), 0);
+	format->setMetaDataTag(CSTR(meta));
+	return self;
+}
+
+
+
+
+
+
+/*
+ * Get number of audio samples in audio clip
+ */
+static VALUE wiretap_format_get_samples(VALUE self) {
+	WireTapAudioFormat* format;
+	Data_Get_Struct(self, WireTapAudioFormat, format);
+	return INT2FIX(format->numSamples());
+}
+
+/*
+ * Set number of samples in audio clip
+ */
+static VALUE wiretap_format_set_samples(VALUE self, VALUE value) {
+	WireTapAudioFormat* format;
+	Data_Get_Struct(self, WireTapAudioFormat, format);
+	format->setNumSamples(NUM2INT(value));
+	return self;
+}
+
+/*
+ * Get bits per sample in audio clip
+ */
+static VALUE wiretap_format_get_bps(VALUE self) {
+	WireTapAudioFormat* format;
+	Data_Get_Struct(self, WireTapAudioFormat, format);
+	return INT2FIX(format->bitsPerSample());
+}
+
+/*
+ * Set bits per sample in audio clip
+ */
+static VALUE wiretap_format_set_bps(VALUE self, VALUE value) {
+	WireTapAudioFormat* format;
+	Data_Get_Struct(self, WireTapAudioFormat, format);
+	format->setBitsPerSample(NUM2INT(value));
+	return self;
+}
+
+/*
+ * Get sample rate of the clip
+ */
+static VALUE wiretap_format_get_sample_rate(VALUE self) {
+	WireTapAudioFormat* format;
+	Data_Get_Struct(self, WireTapAudioFormat, format);
+	return rb_float_new(double(format->sampleRate()));
+}
+
+/*
+ * Set sample rate of the clip
+ */
+static VALUE wiretap_format_set_sample_rate(VALUE self, VALUE value){
+	WireTapAudioFormat* format;
+	Data_Get_Struct(self, WireTapAudioFormat, format);
+	format->setSampleRate(float(RFLOAT(value)->value));
+	return self;
+}
+
+
+
+void Init_format() {
+	
+	cClipFormat = rb_define_class_under(mWiretap, "ClipFormat", rb_cObject);
+	rb_define_alloc_func(cClipFormat, wiretap_clipformat_allocate);
+	rb_define_method(cClipFormat, "width", VALUEFUNC(wiretap_format_get_width), 0); 
+	rb_define_method(cClipFormat, "width=", VALUEFUNC(wiretap_format_set_width), 1);
+	rb_define_method(cClipFormat, "height", VALUEFUNC(wiretap_format_get_height), 0); 
+	rb_define_method(cClipFormat, "height=", VALUEFUNC(wiretap_format_set_height), 1);
+	rb_define_method(cClipFormat, "bpp", VALUEFUNC(wiretap_format_get_bpp), 0); 
+	rb_define_method(cClipFormat, "bpp=", VALUEFUNC(wiretap_format_set_bpp), 1);
+	rb_define_method(cClipFormat, "channels", VALUEFUNC(wiretap_format_get_channels), 0);
+	rb_define_method(cClipFormat, "channels=", VALUEFUNC(wiretap_format_set_channels), 1);
+	rb_define_method(cClipFormat, "buffer_size", VALUEFUNC(wiretap_format_get_buffer_size), 0); 
+	rb_define_method(cClipFormat, "buffer_size=", VALUEFUNC(wiretap_format_set_buffer_size), 1);
+	rb_define_method(cClipFormat, "rate", VALUEFUNC(wiretap_format_get_rate), 0); 
+	rb_define_method(cClipFormat, "rate=", VALUEFUNC(wiretap_format_set_rate), 1);
+	rb_define_method(cClipFormat, "ratio", VALUEFUNC(wiretap_format_get_ratio), 0); 
+	rb_define_method(cClipFormat, "ratio=", VALUEFUNC(wiretap_format_set_ratio), 1);
+	rb_define_method(cClipFormat, "scan", VALUEFUNC(wiretap_format_get_scan), 0); 
+	rb_define_method(cClipFormat, "scan=", VALUEFUNC(wiretap_format_set_scan), 1);
+	rb_define_method(cClipFormat, "tag", VALUEFUNC(wiretap_format_get_tag), 0);
+	rb_define_method(cClipFormat, "tag=", VALUEFUNC(wiretap_format_set_tag), 1);
+	rb_define_method(cClipFormat, "meta", VALUEFUNC(wiretap_format_get_meta), 0); 
+	rb_define_method(cClipFormat, "meta=", VALUEFUNC(wiretap_format_set_meta), 1);
+	rb_define_method(cClipFormat, "meta_tag", VALUEFUNC(wiretap_format_get_meta_tag), 0); 
+	rb_define_method(cClipFormat, "meta_tag=", VALUEFUNC(wiretap_format_set_meta_tag), 1);	
+
+	cAudioFormat = rb_define_class_under(mWiretap, "AudioFormat", cClipFormat);
+	rb_define_alloc_func(cAudioFormat, wiretap_audioformat_allocate);
+	rb_define_method(cAudioFormat, "samples", VALUEFUNC(wiretap_format_get_samples), 0); 
+	rb_define_method(cAudioFormat, "samples=", VALUEFUNC(wiretap_format_set_samples), 1);
+	rb_define_method(cAudioFormat, "bps", VALUEFUNC(wiretap_format_get_bps), 0); 
+	rb_define_method(cAudioFormat, "bps=", VALUEFUNC(wiretap_format_set_bps), 1);
+	rb_define_method(cAudioFormat, "sample_rate", VALUEFUNC(wiretap_format_get_sample_rate), 0); 
+	rb_define_method(cAudioFormat, "sample_rate=", VALUEFUNC(wiretap_format_set_sample_rate), 1);
+
+}
+
+#undef ATTRIBUTE
+
+

data/ext/image.h

@@ -0,0 +1,27 @@
+#ifndef _WIRETAP_IMAGE_H_
+#define _WIRETAP_IMAGE_H_
+
+typedef struct _memstream {
+	unsigned char* ptr;
+} memstream;
+typedef struct _filestream {
+	int count;
+	unsigned char buffer[3];
+	FILE* f;
+} filestream;
+
+void write_to_memory(void *stream, unsigned char channel);
+void write_to_file(void* file, unsigned char channel);
+void write_to_file_bgr(void* file, unsigned char channel);
+typedef void (*bitstream_writer)(void *data, unsigned char channel);
+void write_string_value(char* value, bitstream_writer writer, void *data);
+
+
+bool wiretap_write_frame_bmp(int width, int height, int bpp, unsigned char* frame, FILE* f);
+bool wiretap_write_frame_sgi(int width, int height, int bpp, unsigned char* frame, FILE* f);
+
+int wiretap_write_image_data(int width, int height, int bpp, const unsigned char* frame, bitstream_writer writer, void *data);
+
+#endif /* _WIRETAP_IMAGE_H_ */
+
+

data/ext/image_format.cpp

@@ -0,0 +1,194 @@
+#include "wiretap.h"
+
+enum {IMAGE_BMP, IMAGE_SGI};
+int img_format = IMAGE_SGI;
+
+
+
+static int wiretap_reduce_frame_bits24(const unsigned char* frame, int byte_length, bitstream_writer writer, void *data) {
+	for(int index = 0; index < byte_length; index++) {
+		writer(data, frame[index]);
+	}
+	return 0;
+}
+
+
+static int wiretap_reduce_frame_bits32(const unsigned char* frame, int byte_length, bitstream_writer writer, void *data) {
+	for(int index = 0; index < byte_length; index += 4) {
+		writer(data, frame[0]);
+
+		writer(data, frame[1] << 2 | frame[2] >> 6);
+
+		writer(data, frame[2] << 4 | frame[3] >> 4);
+		frame += 4;
+	}
+	return 0;
+}
+
+
+
+static int wiretap_reduce_frame_bits36(const unsigned char* frame, int byte_length, bitstream_writer writer, void *data) {
+
+	int orphan_index = 0;
+	int orphan[3*6]; //куски, которые пропускаем
+
+	int pixelnum = 0;
+
+	for(int index = 0; index < byte_length*2/9; index++) {
+		if (pixelnum == 6) //7й пиксель собираем и пропущеных кусков
+		{
+			writer(data, ((orphan[4] & 0xF) << 4) + (orphan[2] & 0xF)); // R
+			writer(data, ((orphan[5] & 0xF) << 4) + (orphan[3] & 0xF)); // G
+			writer(data, ((orphan[10] & 0xF) << 4) + (orphan[8] & 0xF));// B
+
+			pixelnum++;
+			continue;
+		}
+
+		if (pixelnum == 7) //8й пиксель тоже собираем из пропущеных кусков
+		{
+			writer(data, ((orphan[11] & 0xF) << 4) + (orphan[9] & 0xF)); // R
+			writer(data, ((orphan[16] & 0xF) << 4) + (orphan[14] & 0xF));// G
+			writer(data, ((orphan[17] & 0xF) << 4) + (orphan[15] & 0xF));// B
+
+			pixelnum = 0;
+			orphan_index = 0;
+
+			continue;
+		}
+
+		writer(data, *frame++); // R
+		orphan[orphan_index++] = *frame++;
+		
+		writer(data, *frame++); // G
+		orphan[orphan_index++] = *frame++;
+		
+		writer(data, *frame++); // B
+		orphan[orphan_index++] = *frame++;
+
+		pixelnum++;
+	}
+	return 0;	
+}
+
+static int wiretap_reduce_frame_bits48(const unsigned char* frame, int byte_length, bitstream_writer writer, void* data) {
+	for(int index = 0; index < byte_length / 6; index++) {
+		writer(data, frame[index*6 + 0]);
+		writer(data, frame[index*6 + 2]);
+		writer(data, frame[index*6 + 4]);
+	}
+	return 0;
+}
+
+bool wiretap_write_image_frame(int width, int height, int bpp, unsigned char* frame, const char* filename) {
+	FILE *f = fopen(filename, "w+");
+	if(!f) {
+		THROW("Couldn't open file for dump: %s", filename);
+	}
+	
+	switch(img_format) {
+		case IMAGE_SGI: {
+			wiretap_write_frame_sgi(width, height, bpp, frame, f);
+			break;
+		}
+		case IMAGE_BMP: {
+			wiretap_write_frame_bmp(width, height, bpp, frame, f);
+			break;
+		}
+		default: {
+			fclose(f); 
+			THROW("Error. Wrong internal format provided: %d", img_format);
+		}
+	}
+	fclose(f);
+	return true;
+}
+
+
+void write_to_memory(void *stream, unsigned char channel) {
+	memstream* data = (memstream *)stream;
+	*data->ptr++ = channel;
+}
+
+void write_to_file(void *stream, unsigned char channel) {
+	filestream* data = (filestream *)stream;
+	fwrite(&channel, sizeof(channel), 1, data->f);
+}
+
+
+
+int reduced_frame_length(int byte_length, int bpp) {
+	const int bits_in_byte = 8;
+	const int bytes_in_pixel = 3;
+	// bpp/bits_in_byte  - number of bytes per pixel in income frame. Maybe 3 (in case of 24bpp) or 4.5 (in case of 36bpp)
+	// byte_length/bytes_per_pixel - number of pixels in income frame
+	// number_pixels * 3 - number of pixels in outgoing frame
+	// 3 * byte_length/bpp * bits_in_byte
+	return (bytes_in_pixel * byte_length * bits_in_byte) / bpp;
+}
+
+
+
+int wiretap_write_image_data(int width, int height, int bpp, const unsigned char* frame, bitstream_writer writer, void *data) {
+	switch(bpp) {
+		case 24: {
+			wiretap_reduce_frame_bits24(frame, width*height*3, writer, data);
+			break;
+		}
+		case 30:
+		case 32: {
+			wiretap_reduce_frame_bits32(frame, width*height*4, writer, data);
+			break;
+		}
+		case 36: {
+			wiretap_reduce_frame_bits36(frame, width*height*9/2, writer, data);
+			break;
+		}
+		case 48: {
+			wiretap_reduce_frame_bits48(frame, width*height*6, writer, data);
+			break;
+		}
+		default: {
+			THROW("Unsupported color depth: %d", bpp);
+		}
+	}
+	return 0;
+}
+
+
+VALUE wiretap_dump_image_data(VALUE self, VALUE width, VALUE height, VALUE bpp, VALUE raw, VALUE filename) {
+	wiretap_write_image_frame(NUM2INT(width), NUM2INT(height), NUM2INT(bpp), (unsigned char* )STR(raw), CSTR(filename));
+	return Qtrue;
+}
+
+VALUE wiretap_set_dump_image_format(VALUE self, VALUE format) {
+	ID bmp, sgi, format_id;
+	bmp = rb_intern("bmp");
+	sgi = rb_intern("sgi");
+	Check_Type(format, T_SYMBOL);
+	format_id = SYM2ID(format);
+	
+	if(format_id == bmp ) {
+		img_format = IMAGE_BMP;
+	} else if(format_id == sgi) {
+		img_format = IMAGE_SGI;
+	} else {
+		THROW("Unknown format");
+	}
+	return self;
+}
+
+VALUE wiretap_get_dump_image_format(VALUE self) {
+	switch(img_format) {
+		case IMAGE_SGI: return ID2SYM(rb_intern("sgi"));
+		case IMAGE_BMP: return ID2SYM(rb_intern("bmp"));
+	}
+	return Qnil;
+}
+
+
+void Init_image_format() {
+	rb_define_singleton_method(mWiretap, "dump_image_data", VALUEFUNC(wiretap_dump_image_data), 5);
+	rb_define_singleton_method(mWiretap, "dump_image_format=", VALUEFUNC(wiretap_set_dump_image_format), 1);
+	rb_define_singleton_method(mWiretap, "dump_image_format", VALUEFUNC(wiretap_get_dump_image_format), 0);
+}