big_pie 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 45a9f7bdb31f20d220aba497b88d80cb0bd74b5395db56818b36c133619bb222
-  data.tar.gz: 583aa41083110f3860cafe0364638baac775b3f472c911c663c86ebb68b36050
+  metadata.gz: 2db6a2fd68ee7adbf9929df4c4cf9097b6053230871d5a73f65733ac1c230e8d
+  data.tar.gz: c472cae6043e622789fe46d1a927c2da2fd0ea884b9dc115ac6fdd558d2b123a
 SHA512:
-  metadata.gz: 817f9e39c9a346ab5e21d5dee883a6e1c62c2a9f0bce5ea046981ad4ffc126175339a2c5e44f035a654a635b11902535df5226735e0917c537b1bae65c4cd010
-  data.tar.gz: bbec218860433997d5bea3037b2cf103ab09a8599c7f40974a21a2d541059a036e955aecb097e7597236de7a8efa561773e6b504a1f8ad17d6a3d922e5ea2564
+  metadata.gz: d24697e37b8800e8607cfbd62811c21e56cc3ec994a37e45d1b15d50722df00fa9f83f964f29fe06c2610c787e48abd441c0cb1a7049c79e21a00ac9d58e1385
+  data.tar.gz: fed229285ca7eeaace2cc137003e3b8f3bbd7034909f4469e4350e44eccd5ae5351956cefe292d713aeffb6bbf4aba7bd7cc9473f8554712bac066d08c663646

data/ext/bigpie.c

@@ -2,29 +2,29 @@
 #include <inttypes.h>
 #include "ruby.h"
 
-#if defined(__GNUC__) && !defined(__clang__) && !defined(__INTEL_COMPILER)
+#ifdef __GNUC__
 	#pragma GCC optimize ("Ofast")
 	#pragma GCC diagnostic warning "-Wall"
-#elif defined(__clang__)
-	#pragma clang optimize on
-	#pragma clang diagnostic warning "-Wall"
-#elif defined(__INTEL_COMPILER)
-	#pragma intel optimization_level 3
 #endif
 
 VALUE calculatePi(volatile VALUE obj, volatile VALUE number) {
 	VALUE ary = rb_ary_new() ;
-	VALUE shove = rb_intern("<<") ;
-
 	VALUE _rb_one = INT2FIX(0) ;
-	if (rb_funcallv_public(number, rb_intern("<"), 1, &_rb_one)) return ary ;
+	VALUE block_given = rb_block_given_p() ;
+
+	if (rb_funcallv_public(number, rb_intern("<"), 1, &_rb_one)) {
+		if (block_given) {
+			return Qnil ;
+		} else {
+			return ary ;
+		}
+	}
 
+	VALUE shove = rb_intern("<<") ;
 	register uint64_t num = NUM2ULL(number) ;
 	register uint64_t index = 0 ;
-	register int8_t comp ;
 
-	mpz_t q, t, k, m, x, r ;
-	mpz_t temp1 ;
+	mpz_t q, t, k, m, x, r, temp1 ;
 
 	mpz_init(q) ;
 	mpz_init(t) ;
@@ -46,14 +46,16 @@ VALUE calculatePi(volatile VALUE obj, volatile VALUE number) {
 		mpz_addmul_ui(temp1, q, 4) ;
 		mpz_submul(temp1, m, t) ;
 
-		comp = mpz_cmp_ui(temp1, 0) ;
-
-		if(comp < 0) {
+		if(mpz_cmp_ui(temp1, 0) < 0) {
 			++index ;
 			VALUE m_to_ui = INT2FIX(mpz_get_ui(m)) ;
-			rb_funcallv_public(ary, shove, 1, &m_to_ui) ;
 
-			// r
+			if (block_given) {
+				rb_yield_values(1, m_to_ui) ;
+			} else {
+				rb_funcallv_public(ary, shove, 1, &m_to_ui) ;
+			}
+
 			mpz_submul(r, m, t) ;
 			mpz_mul_ui(r, r, 10) ;
 
@@ -62,7 +64,6 @@ VALUE calculatePi(volatile VALUE obj, volatile VALUE number) {
 		} else {
 			mpz_mul(t, t, x) ;
 
-			// r
 			mpz_addmul_ui(r, q, 2) ;
 			mpz_mul(r, r, x) ;
 
@@ -81,7 +82,11 @@ VALUE calculatePi(volatile VALUE obj, volatile VALUE number) {
 	mpz_clear(r) ;
 	mpz_clear(temp1) ;
 
-	return ary ;
+	if (block_given) {
+		return Qnil ;
+	} else {
+		return ary ;
+	}
 }
 
 void Init_bigpie() {

data/lib/big_pie.rb

@@ -1,7 +1 @@
-# Frozen_String_Literal: true
-
-module BigPie
-	VERSION = "0.1.2"
-end
-
 require 'bigpie'

data/lib/big_pie/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module BigPie
-	VERSION = "0.1.2"
+	VERSION = "0.1.3"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: big_pie
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - Sourav Goswami
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-01-03 00:00:00.000000000 Z
+date: 2021-01-05 00:00:00.000000000 Z
 dependencies: []
 description: Calculate N Digits of Pi, argument upto unsigned long long (generally
   2 ** 64 - 1)