edlib 0.0.6 → 0.0.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 41a9c8bdf279436f3ca2da4840a62d875c62ab5630c0411c7751446dd7c6db48
-  data.tar.gz: 62436d4433d308d71e76ec5095a57d85c1da88a3a6addf1531cb00e65ed37d72
+  metadata.gz: 72ca7381d858e17e8ba5e2641d072cb4f391410b97ef9138504efa6851020373
+  data.tar.gz: 180e15f4a09149af7031536817a635be580d55afd25896341cb44cf54a9902be
 SHA512:
-  metadata.gz: 9afed6c6ab4e8a59906a98441c464f093ad45a4b92464b0efba67166d925cf0243e4e484b063264940a2afb4867a181deb7dc55733cfb8ae0f082b472f47f286
-  data.tar.gz: c472407d2aa613f12c75c554d525c3adafa21f689921c82e9fac3d43395ce3759f5240fafa06fe4c6985bac7d6ba5bf42944c746e55c0584b067ba57d8b455fb
+  metadata.gz: 5f53003d5e12d9d55b40201f435a278b4a99d6e01569cf0f5e96a263822058257d3dd1b65c7c0f48d39a92177c271131d7752bf982c68893316e5dccbdeefc9f
+  data.tar.gz: 27e1ec9237c5f86fe8775100dbda7e2b9851071a9043cace94b927a65e372afa0b7cbbca741d51266b1f2a0493917eeaf1b4b60da92600def529b2e4ac724edb

data/ext/edlib/edlibext.c

@@ -1,11 +1,30 @@
 #include "ruby.h"
 #include "edlibext.h"
 
+#define ALIGNER_GET_(name) \
+	static VALUE \
+	aligner_get_##name(VALUE self) \
+	{ \
+	    EdlibAlignConfig *config = aligner_get_config(self); \
+		return get_##name(config); \
+	}
+
+#define ALIGNER_SET_(name) \
+	static VALUE \
+	aligner_set_##name(VALUE self, VALUE value) \
+	{ \
+	    EdlibAlignConfig *config = aligner_get_config(self); \
+		return set_##name(config, value); \
+	}
+
 VALUE mEdlib;
 VALUE cAligner;
 
 // Aligner class
 
+static size_t aligner_config_memsize(const void *ptr);
+static void aligner_config_free(void *ptr);
+
 static const rb_data_type_t config_type = {
 	.wrap_struct_name = "RbAlignConfig",
 	.function = {
@@ -73,12 +92,7 @@ get_k(EdlibAlignConfig *config)
 	return INT2NUM(config->k);
 }
 
-static VALUE
-aligner_get_k(VALUE self)
-{
-	EdlibAlignConfig *config = aligner_get_config(self);
-	return get_k(config);
-}
+ALIGNER_GET_(k)
 
 static VALUE
 set_k(EdlibAlignConfig *config, VALUE k)
@@ -87,35 +101,25 @@ set_k(EdlibAlignConfig *config, VALUE k)
 	return k;
 }
 
-static VALUE
-aligner_set_k(VALUE self, VALUE k)
-{
-	EdlibAlignConfig *config = aligner_get_config(self);
-	return set_k(config, k);
-}
+ALIGNER_SET_(k)
 
 static VALUE
 get_mode(EdlibAlignConfig *config)
 {
 	switch (config->mode)
 	{
-	case 0:
+	case EDLIB_MODE_NW:
 		return rb_str_new2("NW");
-	case 1:
+	case EDLIB_MODE_SHW:
 		return rb_str_new2("SHW");
-	case 2:
+	case EDLIB_MODE_HW:
 		return rb_str_new2("HW");
 	default:
 		return Qnil;
 	}
 }
 
-static VALUE
-aligner_get_mode(VALUE self)
-{
-	EdlibAlignConfig *config = aligner_get_config(self);
-	return get_mode(config);
-}
+ALIGNER_GET_(mode)
 
 static VALUE
 set_mode(EdlibAlignConfig *config, VALUE mode)
@@ -127,18 +131,19 @@ set_mode(EdlibAlignConfig *config, VALUE mode)
 	switch (TYPE(mode))
 	{
 	case T_STRING:;
-		rb_funcall(mode, rb_intern("upcase!"), 0);
-		if (strcmp(RSTRING_PTR(mode), "NW") == 0)
+	    VALUE mode_str = rb_funcall(mode, rb_intern("upcase"), 0);
+	    char *mode_s = RSTRING_PTR(mode_str);
+		if (strcmp(mode_s, "NW") == 0)
 		{
-			config->mode = 0;
+			config->mode = EDLIB_MODE_NW;
 		}
-		else if (strcmp(RSTRING_PTR(mode), "SHW") == 0)
+		else if (strcmp(mode_s, "SHW") == 0)
 		{
-			config->mode = 1;
+			config->mode = EDLIB_MODE_SHW;
 		}
-		else if (strcmp(RSTRING_PTR(mode), "HW") == 0)
+		else if (strcmp(mode_s, "HW") == 0)
 		{
-			config->mode = 2;
+			config->mode = EDLIB_MODE_HW;
 		}
 		else
 		{
@@ -159,35 +164,25 @@ set_mode(EdlibAlignConfig *config, VALUE mode)
 	return mode;
 }
 
-static VALUE
-aligner_set_mode(VALUE self, VALUE mode)
-{
-	EdlibAlignConfig *config = aligner_get_config(self);
-	return set_mode(config, mode);
-}
+ALIGNER_SET_(mode)
 
 static VALUE
 get_task(EdlibAlignConfig *config)
 {
 	switch (config->task)
 	{
-	case 0:
+	case EDLIB_TASK_DISTANCE:
 		return rb_str_new2("DISTANCE");
-	case 1:
+	case EDLIB_TASK_LOC:
 		return rb_str_new2("LOC");
-	case 2:
+	case EDLIB_TASK_PATH:
 		return rb_str_new2("PATH");
 	default:
 		return Qnil;
 	}
 }
 
-static VALUE
-aligner_get_task(VALUE self)
-{
-	EdlibAlignConfig *config = aligner_get_config(self);
-	return get_task(config);
-}
+ALIGNER_GET_(task)
 
 static VALUE
 set_task(EdlibAlignConfig *config, VALUE task)
@@ -199,18 +194,19 @@ set_task(EdlibAlignConfig *config, VALUE task)
 	switch (TYPE(task))
 	{
 	case T_STRING:;
-		rb_funcall(task, rb_intern("upcase!"), 0);
-		if (strcmp(RSTRING_PTR(task), "DISTANCE") == 0)
+		VALUE task_str = rb_funcall(task, rb_intern("upcase"), 0);
+		char *task_s = RSTRING_PTR(task_str);
+		if (strcmp(task_s, "DISTANCE") == 0)
 		{
-			config->task = 0;
+			config->task = EDLIB_TASK_DISTANCE;
 		}
-		else if (strcmp(RSTRING_PTR(task), "LOC") == 0)
+		else if (strcmp(task_s, "LOC") == 0)
 		{
-			config->task = 1;
+			config->task = EDLIB_TASK_LOC;
 		}
-		else if (strcmp(RSTRING_PTR(task), "PATH") == 0)
+		else if (strcmp(task_s, "PATH") == 0)
 		{
-			config->task = 2;
+			config->task = EDLIB_TASK_PATH;
 		}
 		else
 		{
@@ -231,12 +227,7 @@ set_task(EdlibAlignConfig *config, VALUE task)
 	return task;
 }
 
-static VALUE
-aligner_set_task(VALUE self, VALUE task)
-{
-	EdlibAlignConfig *config = aligner_get_config(self);
-	return set_task(config, task);
-}
+ALIGNER_SET_(task)
 
 static VALUE
 get_additional_equalities(EdlibAlignConfig *config)
@@ -255,12 +246,7 @@ get_additional_equalities(EdlibAlignConfig *config)
 	return equalities;
 }
 
-static VALUE
-aligner_get_additional_equalities(VALUE self)
-{
-	EdlibAlignConfig *config = aligner_get_config(self);
-	return get_additional_equalities(config);
-}
+ALIGNER_GET_(additional_equalities)
 
 static VALUE
 set_additional_equalities(EdlibAlignConfig *config, EdlibEqualityPair *eqpairs, VALUE equalities)
@@ -455,5 +441,5 @@ void Init_edlibext(void)
 	rb_define_method(cAligner, "additional_equalities", aligner_get_additional_equalities, 0);
 	rb_define_method(cAligner, "additional_equalities=", aligner_set_additional_equalities, 1);
 	rb_define_method(cAligner, "config", aligner_config_hash, 0);
-	rb_define_method(cAligner, "align", aligner_align, 2);
+	rb_define_method(cAligner, "align_raw", aligner_align, 2);
 }

data/ext/edlib/edlibext.h

@@ -9,34 +9,7 @@ typedef struct
     EdlibEqualityPair *equalityPairs;
 } RbAlignConfig;
 
-static VALUE aligner_config_allocate(VALUE klass);
-static size_t aligner_config_memsize(const void *ptr);
-static void aligner_config_free(void *ptr);
-
 static EdlibAlignConfig *aligner_get_config(VALUE self);
 static EdlibEqualityPair *aligner_get_equalityPairs(VALUE self);
 
-static VALUE get_k(EdlibAlignConfig *config);
-static VALUE aligner_get_k(VALUE self);
-static VALUE set_k(EdlibAlignConfig *config, VALUE k);
-static VALUE aligner_set_k(VALUE self, VALUE k);
-static VALUE get_mode(EdlibAlignConfig *config);
-static VALUE aligner_get_mode(VALUE self);
-static VALUE set_mode(EdlibAlignConfig *config, VALUE mode);
-static VALUE aligner_set_mode(VALUE self, VALUE mode);
-static VALUE get_task(EdlibAlignConfig *config);
-static VALUE aligner_get_task(VALUE self);
-static VALUE set_task(EdlibAlignConfig *config, VALUE task);
-static VALUE aligner_set_task(VALUE self, VALUE task);
-static VALUE get_additional_equalities(EdlibAlignConfig *config);
-static VALUE aligner_get_additional_equalities(VALUE self);
-static VALUE set_additional_equalities(EdlibAlignConfig *config, EdlibEqualityPair *eqpairs, VALUE equalities);
-static VALUE aligner_set_additional_equalities(VALUE self, VALUE equalities);
-
-static VALUE aligner_config_hash(VALUE self);
-
-static VALUE aligner_initialize_raw(VALUE self, VALUE k, VALUE mode, VALUE task, VALUE additional_equalities);
-static VALUE aligner_align(VALUE self, VALUE query, VALUE target);
-void Init_edlibext(void);
-
 #endif // EDLIBEXT_H

data/ext/edlib/extconf.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require 'mkmf'
 
 create_makefile('edlib/edlibext')

data/lib/edlib/version.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module Edlib
-  VERSION = '0.0.6'
+  VERSION = '0.0.7'
 end

data/lib/edlib.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require_relative 'edlib/edlibext'
 
 module Edlib
@@ -9,5 +11,68 @@ module Edlib
       task = task.upcase if task.is_a? String
       initialize_raw(k, mode, task, additional_equalities)
     end
+
+    def align(query, target, nice: false)
+      if nice
+        align_nice(query, target)
+      else
+        align_raw(query, target)
+      end
+    end
+
+    private
+
+    def align_nice(query, target)
+      result = align(query, target)
+      result.merge(nice(result, query, target))
+    end
+
+    def nice(result, query, target, gap_symbol: '-')
+      raise 'result does not have :locations and :cigar' unless result.key?(:locations) && result.key?(:cigar)
+
+      target_pos = result[:locations][0][0]
+      query_pos = 0
+      query_aln = String.new
+      match_aln = String.new
+      target_aln = String.new
+      cigar = result[:cigar]
+      cigar.scan(/(\d+)(\D)/).each do |num, op|
+        num = num.to_i
+        case op
+        when '='
+          target_aln << target[target_pos, num]
+          target_pos += num
+          query_aln << query[query_pos, num]
+          query_pos += num
+          match_aln << '|' * num
+        when 'X'
+          target_aln << target[target_pos, num]
+          target_pos += num
+          query_aln << query[query_pos, num]
+          query_pos += num
+          match_aln << '.' * num
+        when 'D'
+          target_aln << target[target_pos, num]
+          target_pos += num
+          query_aln << gap_symbol * num
+          query_pos += 0
+          match_aln << gap_symbol * num
+        when 'I'
+          target_aln << gap_symbol * num
+          target_pos += 0
+          query_aln << query[query_pos, num]
+          query_pos += num
+          match_aln << gap_symbol * num
+        else
+          raise "Unknown CIGAR operation: #{op}"
+        end
+      end
+
+      {
+        query_aligned: query_aln,
+        match_aligned: match_aln,
+        target_aligned: target_aln
+      }
+    end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: edlib
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.7
 platform: ruby
 authors:
 - kojix2
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-11-02 00:00:00.000000000 Z
+date: 2022-11-07 00:00:00.000000000 Z
 dependencies: []
 description: 'Lightweight, super fast C/C++ library for sequence alignment using edit
   (Levenshtein) distance. '