From 5801f58e84515642f2f3542b9318946c3eeb6ab2 Mon Sep 17 00:00:00 2001 From: Philipp Burckhardt Date: Fri, 28 Aug 2026 00:02:23 -0500 Subject: [PATCH] feat: add C implementation for `stats/base/dists/negative-binomial/pmf` Co-Authored-By: Claude Fable 5 --- .../pmf/benchmark/benchmark.native.js | 72 ++++++ .../pmf/benchmark/c/Makefile | 146 +++++++++++++ .../pmf/benchmark/c/benchmark.c | 142 ++++++++++++ .../dists/negative-binomial/pmf/binding.gyp | 170 +++++++++++++++ .../negative-binomial/pmf/examples/c/Makefile | 146 +++++++++++++ .../pmf/examples/c/example.c | 43 ++++ .../dists/negative-binomial/pmf/include.gypi | 53 +++++ .../stats/base/dists/negative-binomial/pmf.h | 38 ++++ .../dists/negative-binomial/pmf/lib/native.js | 88 ++++++++ .../dists/negative-binomial/pmf/manifest.json | 84 +++++++ .../dists/negative-binomial/pmf/package.json | 3 + .../dists/negative-binomial/pmf/src/Makefile | 70 ++++++ .../dists/negative-binomial/pmf/src/addon.c | 22 ++ .../dists/negative-binomial/pmf/src/main.c | 56 +++++ .../pmf/test/{test.pmf.js => test.main.js} | 0 .../negative-binomial/pmf/test/test.native.js | 205 ++++++++++++++++++ 16 files changed, 1338 insertions(+) create mode 100644 lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/benchmark/benchmark.native.js create mode 100644 lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/benchmark/c/Makefile create mode 100644 lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/benchmark/c/benchmark.c create mode 100644 lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/binding.gyp create mode 100644 lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/examples/c/Makefile create mode 100644 lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/examples/c/example.c create mode 100644 lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/include.gypi create mode 100644 lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/include/stdlib/stats/base/dists/negative-binomial/pmf.h create mode 100644 lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/lib/native.js create mode 100644 lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/manifest.json create mode 100644 lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/src/Makefile create mode 100644 lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/src/addon.c create mode 100644 lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/src/main.c rename lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/test/{test.pmf.js => test.main.js} (100%) create mode 100644 lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/test/test.native.js diff --git a/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/benchmark/benchmark.native.js b/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/benchmark/benchmark.native.js new file mode 100644 index 000000000000..7f081afbfc6e --- /dev/null +++ b/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/benchmark/benchmark.native.js @@ -0,0 +1,72 @@ +/** +* @license Apache-2.0 +* +* Copyright (c) 2026 The Stdlib Authors. +* +* Licensed under the Apache License, Version 2.0 (the "License"); +* you may not use this file except in compliance with the License. +* You may obtain a copy of the License at +* +* http://www.apache.org/licenses/LICENSE-2.0 +* +* Unless required by applicable law or agreed to in writing, software +* distributed under the License is distributed on an "AS IS" BASIS, +* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +* See the License for the specific language governing permissions and +* limitations under the License. +*/ + +'use strict'; + +// MODULES // + +var resolve = require( 'path' ).resolve; +var bench = require( '@stdlib/bench' ); +var tryRequire = require( '@stdlib/utils/try-require' ); +var uniform = require( '@stdlib/random/array/uniform' ); +var discreteUniform = require( '@stdlib/random/array/discrete-uniform' ); +var isnan = require( '@stdlib/math/base/assert/is-nan' ); +var EPS = require( '@stdlib/constants/float64/eps' ); +var format = require( '@stdlib/string/format' ); +var pkg = require( './../package.json' ).name; + + +// VARIABLES // + +var pmf = tryRequire( resolve( __dirname, './../lib/native.js' ) ); +var opts = { + 'skip': ( pmf instanceof Error ) +}; + + +// MAIN // + +bench( format( '%s::native', pkg ), opts, function benchmark( b ) { + var arrayOpts; + var r; + var p; + var x; + var y; + var i; + + arrayOpts = { + 'dtype': 'float64' + }; + x = discreteUniform( 100, 1, 100, arrayOpts ); + r = discreteUniform( 100, 1, 100, arrayOpts ); + p = uniform( 100, EPS, 1.0, arrayOpts ); + + b.tic(); + for ( i = 0; i < b.iterations; i++ ) { + y = pmf( x[ i % x.length ], r[ i % r.length ], p[ i % p.length ] ); + if ( isnan( y ) ) { + b.fail( 'should not return NaN' ); + } + } + b.toc(); + if ( isnan( y ) ) { + b.fail( 'should not return NaN' ); + } + b.pass( 'benchmark finished' ); + b.end(); +}); diff --git a/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/benchmark/c/Makefile b/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/benchmark/c/Makefile new file mode 100644 index 000000000000..979768abbcec --- /dev/null +++ b/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/benchmark/c/Makefile @@ -0,0 +1,146 @@ +#/ +# @license Apache-2.0 +# +# Copyright (c) 2026 The Stdlib Authors. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +#/ + +# VARIABLES # + +ifndef VERBOSE + QUIET := @ +else + QUIET := +endif + +# Determine the OS ([1][1], [2][2]). +# +# [1]: https://en.wikipedia.org/wiki/Uname#Examples +# [2]: http://stackoverflow.com/a/27776822/2225624 +OS ?= $(shell uname) +ifneq (, $(findstring MINGW,$(OS))) + OS := WINNT +else +ifneq (, $(findstring MSYS,$(OS))) + OS := WINNT +else +ifneq (, $(findstring CYGWIN,$(OS))) + OS := WINNT +else +ifneq (, $(findstring Windows_NT,$(OS))) + OS := WINNT +endif +endif +endif +endif + +# Define the program used for compiling C source files: +ifdef C_COMPILER + CC := $(C_COMPILER) +else + CC := gcc +endif + +# Define the command-line options when compiling C files: +CFLAGS ?= \ + -std=c99 \ + -O3 \ + -Wall \ + -pedantic + +# Determine whether to generate position independent code ([1][1], [2][2]). +# +# [1]: https://gcc.gnu.org/onlinedocs/gcc/Code-Gen-Options.html#Code-Gen-Options +# [2]: http://stackoverflow.com/questions/5311515/gcc-fpic-option +ifeq ($(OS), WINNT) + fPIC ?= +else + fPIC ?= -fPIC +endif + +# List of includes (e.g., `-I /foo/bar -I /beep/boop/include`): +INCLUDE ?= + +# List of source files: +SOURCE_FILES ?= + +# List of libraries (e.g., `-lopenblas -lpthread`): +LIBRARIES ?= + +# List of library paths (e.g., `-L /foo/bar -L /beep/boop`): +LIBPATH ?= + +# List of C targets: +c_targets := benchmark.out + + +# RULES # + +#/ +# Compiles source files. +# +# @param {string} [C_COMPILER] - C compiler (e.g., `gcc`) +# @param {string} [CFLAGS] - C compiler options +# @param {(string|void)} [fPIC] - compiler flag determining whether to generate position independent code (e.g., `-fPIC`) +# @param {string} [INCLUDE] - list of includes (e.g., `-I /foo/bar -I /beep/boop/include`) +# @param {string} [SOURCE_FILES] - list of source files +# @param {string} [LIBPATH] - list of library paths (e.g., `-L /foo/bar -L /beep/boop`) +# @param {string} [LIBRARIES] - list of libraries (e.g., `-lopenblas -lpthread`) +# +# @example +# make +# +# @example +# make all +#/ +all: $(c_targets) + +.PHONY: all + +#/ +# Compiles C source files. +# +# @private +# @param {string} CC - C compiler (e.g., `gcc`) +# @param {string} CFLAGS - C compiler options +# @param {(string|void)} fPIC - compiler flag determining whether to generate position independent code (e.g., `-fPIC`) +# @param {string} INCLUDE - list of includes (e.g., `-I /foo/bar`) +# @param {string} SOURCE_FILES - list of source files +# @param {string} LIBPATH - list of library paths (e.g., `-L /foo/bar`) +# @param {string} LIBRARIES - list of libraries (e.g., `-lopenblas`) +#/ +$(c_targets): %.out: %.c + $(QUIET) $(CC) $(CFLAGS) $(fPIC) $(INCLUDE) -o $@ $(SOURCE_FILES) $< $(LIBPATH) -lm $(LIBRARIES) + +#/ +# Runs compiled benchmarks. +# +# @example +# make run +#/ +run: $(c_targets) + $(QUIET) ./$< + +.PHONY: run + +#/ +# Removes generated files. +# +# @example +# make clean +#/ +clean: + $(QUIET) -rm -f *.o *.out + +.PHONY: clean diff --git a/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/benchmark/c/benchmark.c b/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/benchmark/c/benchmark.c new file mode 100644 index 000000000000..c20c8e8bbb60 --- /dev/null +++ b/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/benchmark/c/benchmark.c @@ -0,0 +1,142 @@ +/** +* @license Apache-2.0 +* +* Copyright (c) 2026 The Stdlib Authors. +* +* Licensed under the Apache License, Version 2.0 (the "License"); +* you may not use this file except in compliance with the License. +* You may obtain a copy of the License at +* +* http://www.apache.org/licenses/LICENSE-2.0 +* +* Unless required by applicable law or agreed to in writing, software +* distributed under the License is distributed on an "AS IS" BASIS, +* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +* See the License for the specific language governing permissions and +* limitations under the License. +*/ + +#include "stdlib/stats/base/dists/negative-binomial/pmf.h" +#include "stdlib/math/base/special/ceil.h" +#include +#include +#include +#include + +#define NAME "negative-binomial-pmf" +#define ITERATIONS 1000000 +#define REPEATS 3 + +/** +* Prints the TAP version. +*/ +static void print_version( void ) { + printf( "TAP version 13\n" ); +} + +/** +* Prints the TAP summary. +* +* @param total total number of tests +* @param passing total number of passing tests +*/ +static void print_summary( int total, int passing ) { + printf( "#\n" ); + printf( "1..%d\n", total ); // TAP plan + printf( "# total %d\n", total ); + printf( "# pass %d\n", passing ); + printf( "#\n" ); + printf( "# ok\n" ); +} + +/** +* Prints benchmarks results. +* +* @param elapsed elapsed time in seconds +*/ +static void print_results( double elapsed ) { + double rate = (double)ITERATIONS / elapsed; + printf( " ---\n" ); + printf( " iterations: %d\n", ITERATIONS ); + printf( " elapsed: %0.9f\n", elapsed ); + printf( " rate: %0.9f\n", rate ); + printf( " ...\n" ); +} + +/** +* Returns a clock time. +* +* @return clock time +*/ +static double tic( void ) { + struct timeval now; + gettimeofday( &now, NULL ); + return (double)now.tv_sec + (double)now.tv_usec/1.0e6; +} + +/** +* Generates a random number on the interval [min,max). +* +* @param min minimum value (inclusive) +* @param max maximum value (exclusive) +* @return random number +*/ +static double random_uniform( const double min, const double max ) { + double v = (double)rand() / ( (double)RAND_MAX + 1.0 ); + return min + ( v*(max-min) ); +} + +/** +* Runs a benchmark. +* +* @return elapsed time in seconds +*/ +static double benchmark( void ) { + double elapsed; + double x[ 100 ]; + double r[ 100 ]; + double p[ 100 ]; + double y; + double t; + int i; + + for ( i = 0; i < 100; i++ ) { + x[ i ] = stdlib_base_ceil( random_uniform( 0.0, 100.0 ) ); + r[ i ] = stdlib_base_ceil( random_uniform( 1.0, 100.0 ) ); + p[ i ] = random_uniform( 0.01, 1.0 ); + } + + t = tic(); + for ( i = 0; i < ITERATIONS; i++ ) { + y = stdlib_base_dists_negative_binomial_pmf( x[ i % 100 ], r[ i % 100 ], p[ i % 100 ] ); + if ( y != y ) { + printf( "should not return NaN\n" ); + break; + } + } + elapsed = tic() - t; + if ( y != y ) { + printf( "should not return NaN\n" ); + } + return elapsed; +} + +/** +* Main execution sequence. +*/ +int main( void ) { + double elapsed; + int i; + + // Use the current time to seed the random number generator: + srand( time( NULL ) ); + + print_version(); + for ( i = 0; i < REPEATS; i++ ) { + printf( "# c::%s\n", NAME ); + elapsed = benchmark(); + print_results( elapsed ); + printf( "ok %d benchmark finished\n", i+1 ); + } + print_summary( REPEATS, REPEATS ); +} diff --git a/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/binding.gyp b/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/binding.gyp new file mode 100644 index 000000000000..0d6508a12e99 --- /dev/null +++ b/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/binding.gyp @@ -0,0 +1,170 @@ +# @license Apache-2.0 +# +# Copyright (c) 2026 The Stdlib Authors. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +# A `.gyp` file for building a Node.js native add-on. +# +# [1]: https://gyp.gsrc.io/docs/InputFormatReference.md +# [2]: https://gyp.gsrc.io/docs/UserDocumentation.md +{ + # List of files to include in this file: + 'includes': [ + './include.gypi', + ], + + # Define variables to be used throughout the configuration for all targets: + 'variables': { + # Target name should match the add-on export name: + 'addon_target_name%': 'addon', + + # Set variables based on the host OS: + 'conditions': [ + [ + 'OS=="win"', + { + # Define the object file suffix: + 'obj': 'obj', + }, + { + # Define the object file suffix: + 'obj': 'o', + } + ], # end condition (OS=="win") + ], # end conditions + }, # end variables + + # Define compile targets: + 'targets': [ + + # Target to generate an add-on: + { + # The target name should match the add-on export name: + 'target_name': '<(addon_target_name)', + + # Define dependencies: + 'dependencies': [], + + # Define directories which contain relevant include headers: + 'include_dirs': [ + # Local include directory: + '<@(include_dirs)', + ], + + # List of source files: + 'sources': [ + '<@(src_files)', + ], + + # Settings which should be applied when a target's object files are used as linker input: + 'link_settings': { + # Define libraries: + 'libraries': [ + '<@(libraries)', + ], + + # Define library directories: + 'library_dirs': [ + '<@(library_dirs)', + ], + }, + + # C/C++ compiler flags: + 'cflags': [ + # Enable commonly used warning options: + '-Wall', + + # Aggressive optimization: + '-O3', + ], + + # C specific compiler flags: + 'cflags_c': [ + # Specify the C standard to which a program is expected to conform: + '-std=c99', + ], + + # C++ specific compiler flags: + 'cflags_cpp': [ + # Specify the C++ standard to which a program is expected to conform: + '-std=c++11', + ], + + # Linker flags: + 'ldflags': [], + + # Apply conditions based on the host OS: + 'conditions': [ + [ + 'OS=="mac"', + { + # Linker flags: + 'ldflags': [ + '-undefined dynamic_lookup', + '-Wl,-no-pie', + '-Wl,-search_paths_first', + ], + }, + ], # end condition (OS=="mac") + [ + 'OS!="win"', + { + # C/C++ flags: + 'cflags': [ + # Generate platform-independent code: + '-fPIC', + ], + }, + ], # end condition (OS!="win") + ], # end conditions + }, # end target <(addon_target_name) + + # Target to copy a generated add-on to a standard location: + { + 'target_name': 'copy_addon', + + # Declare that the output of this target is not linked: + 'type': 'none', + + # Define dependencies: + 'dependencies': [ + # Require that the add-on be generated before building this target: + '<(addon_target_name)', + ], + + # Define a list of actions: + 'actions': [ + { + 'action_name': 'copy_addon', + 'message': 'Copying addon...', + + # Explicitly list the inputs in the command-line invocation below: + 'inputs': [], + + # Declare the expected outputs: + 'outputs': [ + '<(addon_output_dir)/<(addon_target_name).node', + ], + + # Define the command-line invocation: + 'action': [ + 'cp', + '<(PRODUCT_DIR)/<(addon_target_name).node', + '<(addon_output_dir)/<(addon_target_name).node', + ], + }, + ], # end actions + }, # end target copy_addon + ], # end targets +} diff --git a/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/examples/c/Makefile b/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/examples/c/Makefile new file mode 100644 index 000000000000..c8f8e9a1517b --- /dev/null +++ b/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/examples/c/Makefile @@ -0,0 +1,146 @@ +#/ +# @license Apache-2.0 +# +# Copyright (c) 2026 The Stdlib Authors. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +#/ + +# VARIABLES # + +ifndef VERBOSE + QUIET := @ +else + QUIET := +endif + +# Determine the OS ([1][1], [2][2]). +# +# [1]: https://en.wikipedia.org/wiki/Uname#Examples +# [2]: http://stackoverflow.com/a/27776822/2225624 +OS ?= $(shell uname) +ifneq (, $(findstring MINGW,$(OS))) + OS := WINNT +else +ifneq (, $(findstring MSYS,$(OS))) + OS := WINNT +else +ifneq (, $(findstring CYGWIN,$(OS))) + OS := WINNT +else +ifneq (, $(findstring Windows_NT,$(OS))) + OS := WINNT +endif +endif +endif +endif + +# Define the program used for compiling C source files: +ifdef C_COMPILER + CC := $(C_COMPILER) +else + CC := gcc +endif + +# Define the command-line options when compiling C files: +CFLAGS ?= \ + -std=c99 \ + -O3 \ + -Wall \ + -pedantic + +# Determine whether to generate position independent code ([1][1], [2][2]). +# +# [1]: https://gcc.gnu.org/onlinedocs/gcc/Code-Gen-Options.html#Code-Gen-Options +# [2]: http://stackoverflow.com/questions/5311515/gcc-fpic-option +ifeq ($(OS), WINNT) + fPIC ?= +else + fPIC ?= -fPIC +endif + +# List of includes (e.g., `-I /foo/bar -I /beep/boop/include`): +INCLUDE ?= + +# List of source files: +SOURCE_FILES ?= + +# List of libraries (e.g., `-lopenblas -lpthread`): +LIBRARIES ?= + +# List of library paths (e.g., `-L /foo/bar -L /beep/boop`): +LIBPATH ?= + +# List of C targets: +c_targets := example.out + + +# RULES # + +#/ +# Compiles source files. +# +# @param {string} [C_COMPILER] - C compiler (e.g., `gcc`) +# @param {string} [CFLAGS] - C compiler options +# @param {(string|void)} [fPIC] - compiler flag determining whether to generate position independent code (e.g., `-fPIC`) +# @param {string} [INCLUDE] - list of includes (e.g., `-I /foo/bar -I /beep/boop/include`) +# @param {string} [SOURCE_FILES] - list of source files +# @param {string} [LIBPATH] - list of library paths (e.g., `-L /foo/bar -L /beep/boop`) +# @param {string} [LIBRARIES] - list of libraries (e.g., `-lopenblas -lpthread`) +# +# @example +# make +# +# @example +# make all +#/ +all: $(c_targets) + +.PHONY: all + +#/ +# Compiles C source files. +# +# @private +# @param {string} CC - C compiler (e.g., `gcc`) +# @param {string} CFLAGS - C compiler options +# @param {(string|void)} fPIC - compiler flag determining whether to generate position independent code (e.g., `-fPIC`) +# @param {string} INCLUDE - list of includes (e.g., `-I /foo/bar`) +# @param {string} SOURCE_FILES - list of source files +# @param {string} LIBPATH - list of library paths (e.g., `-L /foo/bar`) +# @param {string} LIBRARIES - list of libraries (e.g., `-lopenblas`) +#/ +$(c_targets): %.out: %.c + $(QUIET) $(CC) $(CFLAGS) $(fPIC) $(INCLUDE) -o $@ $(SOURCE_FILES) $< $(LIBPATH) -lm $(LIBRARIES) + +#/ +# Runs compiled examples. +# +# @example +# make run +#/ +run: $(c_targets) + $(QUIET) ./$< + +.PHONY: run + +#/ +# Removes generated files. +# +# @example +# make clean +#/ +clean: + $(QUIET) -rm -f *.o *.out + +.PHONY: clean diff --git a/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/examples/c/example.c b/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/examples/c/example.c new file mode 100644 index 000000000000..af055cb4d555 --- /dev/null +++ b/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/examples/c/example.c @@ -0,0 +1,43 @@ +/** +* @license Apache-2.0 +* +* Copyright (c) 2026 The Stdlib Authors. +* +* Licensed under the Apache License, Version 2.0 (the "License"); +* you may not use this file except in compliance with the License. +* You may obtain a copy of the License at +* +* http://www.apache.org/licenses/LICENSE-2.0 +* +* Unless required by applicable law or agreed to in writing, software +* distributed under the License is distributed on an "AS IS" BASIS, +* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +* See the License for the specific language governing permissions and +* limitations under the License. +*/ + +#include "stdlib/stats/base/dists/negative-binomial/pmf.h" +#include "stdlib/math/base/special/ceil.h" +#include +#include + +static double random_uniform( const double min, const double max ) { + double v = (double)rand() / ( (double)RAND_MAX + 1.0 ); + return min + ( v * (max - min) ); +} + +int main( void ) { + double x; + double r; + double p; + double y; + int i; + + for ( i = 0; i < 25; i++ ) { + x = stdlib_base_ceil( random_uniform( 0.0, 40.0 ) ); + r = stdlib_base_ceil( random_uniform( 1.0, 40.0 ) ); + p = random_uniform( 0.01, 1.0 ); + y = stdlib_base_dists_negative_binomial_pmf( x, r, p ); + printf( "x: %lf, r: %lf, p: %lf, P(X=x;r,p): %lf\n", x, r, p, y ); + } +} diff --git a/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/include.gypi b/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/include.gypi new file mode 100644 index 000000000000..bee8d41a2caf --- /dev/null +++ b/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/include.gypi @@ -0,0 +1,53 @@ +# @license Apache-2.0 +# +# Copyright (c) 2026 The Stdlib Authors. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +# A GYP include file for building a Node.js native add-on. +# +# Main documentation: +# +# [1]: https://gyp.gsrc.io/docs/InputFormatReference.md +# [2]: https://gyp.gsrc.io/docs/UserDocumentation.md +{ + # Define variables to be used throughout the configuration for all targets: + 'variables': { + # Source directory: + 'src_dir': './src', + + # Include directories: + 'include_dirs': [ + ' + +/** +* Evaluates the probability mass function (PMF) for a negative binomial distribution with number of successes until experiment is stopped `r` and success probability `p` at a value `x`. +* +* @param x input value +* @param r number of successes until experiment is stopped +* @param p success probability +* @return evaluated PMF +* +* @example +* double y = stdlib_base_dists_negative_binomial_pmf( 5.0, 20.0, 0.8 ); +* // returns ~0.157 +*/ +double stdlib_base_dists_negative_binomial_pmf( const double x, const double r, const double p ) { + double derivative; + double out; + if ( + stdlib_base_is_nan( x ) || + stdlib_base_is_nan( r ) || + stdlib_base_is_nan( p ) || + r <= 0.0 || + p <= 0.0 || + p > 1.0 + ) { + return 0.0 / 0.0; // NaN + } + if ( !stdlib_base_is_nonnegative_integer( x ) || p == 0.0 ) { + return 0.0; + } + // Compute the derivative of the regularized incomplete beta function: + stdlib_base_kernel_betainc( p, r, x + 1.0, true, false, &out, &derivative ); + return ( p / ( r + x ) ) * derivative; +} diff --git a/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/test/test.pmf.js b/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/test/test.main.js similarity index 100% rename from lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/test/test.pmf.js rename to lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/test/test.main.js diff --git a/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/test/test.native.js b/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/test/test.native.js new file mode 100644 index 000000000000..2715ed6d1cad --- /dev/null +++ b/lib/node_modules/@stdlib/stats/base/dists/negative-binomial/pmf/test/test.native.js @@ -0,0 +1,205 @@ +/** +* @license Apache-2.0 +* +* Copyright (c) 2026 The Stdlib Authors. +* +* Licensed under the Apache License, Version 2.0 (the "License"); +* you may not use this file except in compliance with the License. +* You may obtain a copy of the License at +* +* http://www.apache.org/licenses/LICENSE-2.0 +* +* Unless required by applicable law or agreed to in writing, software +* distributed under the License is distributed on an "AS IS" BASIS, +* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +* See the License for the specific language governing permissions and +* limitations under the License. +*/ + +'use strict'; + +// MODULES // + +var resolve = require( 'path' ).resolve; +var tape = require( 'tape' ); +var isAlmostSameValue = require( '@stdlib/assert/is-almost-same-value' ); +var isnan = require( '@stdlib/math/base/assert/is-nan' ); +var PINF = require( '@stdlib/constants/float64/pinf' ); +var NINF = require( '@stdlib/constants/float64/ninf' ); +var tryRequire = require( '@stdlib/utils/try-require' ); + + +// VARIABLES // + +var pmf = tryRequire( resolve( __dirname, './../lib/native.js' ) ); +var opts = { + 'skip': ( pmf instanceof Error ) +}; + + +// FIXTURES // + +var smallSmall = require( './fixtures/julia/small_small.json' ); +var smallHigh = require( './fixtures/julia/small_high.json' ); +var highSmall = require( './fixtures/julia/high_small.json' ); +var highHigh = require( './fixtures/julia/high_high.json' ); + + +// TESTS // + +tape( 'main export is a function', opts, function test( t ) { + t.ok( true, __filename ); + t.strictEqual( typeof pmf, 'function', 'main export is a function' ); + t.end(); +}); + +tape( 'if provided `NaN` for any parameter, the function returns `NaN`', opts, function test( t ) { + var y = pmf( NaN, 20.0, 0.5 ); + t.strictEqual( isnan( y ), true, 'returns expected value' ); + y = pmf( 0.0, NaN, 0.5 ); + t.strictEqual( isnan( y ), true, 'returns expected value' ); + y = pmf( 0.0, 20.0, NaN ); + t.strictEqual( isnan( y ), true, 'returns expected value' ); + t.end(); +}); + +tape( 'if provided a negative integer for `x` and a valid `r` and `p`, the function returns `0`', opts, function test( t ) { + var y = pmf( NINF, 20, 0.5 ); + t.strictEqual( y, 0.0, 'returns expected value' ); + + y = pmf( -20.0, 20, 0.5 ); + t.strictEqual( y, 0.0, 'returns expected value' ); + + y = pmf( -100.0, 20, 0.5 ); + t.strictEqual( y, 0.0, 'returns expected value' ); + + y = pmf( -1.0, 20, 0.5 ); + t.strictEqual( y, 0.0, 'returns expected value' ); + + t.end(); +}); + +tape( 'if provided a non-integer for `x` and a valid `r` and `p`, the function returns `0`', opts, function test( t ) { + var y = pmf( -1.5, 20.0, 0.5 ); + t.strictEqual( y, 0.0, 'returns expected value' ); + + y = pmf( -0.5, 20.0, 0.5 ); + t.strictEqual( y, 0.0, 'returns expected value' ); + + y = pmf( 1.5, 20.0, 0.5 ); + t.strictEqual( y, 0.0, 'returns expected value' ); + + y = pmf( 2.5, 20.0, 0.5 ); + t.strictEqual( y, 0.0, 'returns expected value' ); + + t.end(); +}); + +tape( 'if provided `r` which is not a positive number, the function returns `NaN`', opts, function test( t ) { + var y; + + y = pmf( 2.0, -0.5, 0.5 ); + t.strictEqual( isnan( y ), true, 'returns expected value' ); + + y = pmf( 0.0, -1.0, 0.5 ); + t.strictEqual( isnan( y ), true, 'returns expected value' ); + + y = pmf( 2.0, NINF, 0.5 ); + t.strictEqual( isnan( y ), true, 'returns expected value' ); + + t.end(); +}); + +tape( 'if provided a success probability `p` outside of `[0,1]`, the function returns `NaN`', opts, function test( t ) { + var y; + + y = pmf( 2.0, 20, -1.0 ); + t.strictEqual( isnan( y ), true, 'returns expected value' ); + + y = pmf( 0.0, 20, 1.5 ); + t.strictEqual( isnan( y ), true, 'returns expected value' ); + + y = pmf( 2.0, 20, NINF ); + t.strictEqual( isnan( y ), true, 'returns expected value' ); + + y = pmf( 2.0, 20, PINF ); + t.strictEqual( isnan( y ), true, 'returns expected value' ); + + t.end(); +}); + +tape( 'the function evaluates the pmf for `x` given large `r` and `p`', opts, function test( t ) { + var expected; + var i; + var r; + var p; + var x; + var y; + + expected = highHigh.expected; + x = highHigh.x; + r = highHigh.r; + p = highHigh.p; + for ( i = 0; i < x.length; i++ ) { + y = pmf( x[ i ], r[ i ], p[ i ] ); + t.strictEqual( isAlmostSameValue( y, expected[ i ], 655 ), true, 'returns expected value' ); + } + t.end(); +}); + +tape( 'the function evaluates the pmf for `x` given large parameter `r` and small `p`', opts, function test( t ) { + var expected; + var i; + var r; + var p; + var x; + var y; + + expected = highSmall.expected; + x = highSmall.x; + r = highSmall.r; + p = highSmall.p; + for ( i = 0; i < x.length; i++ ) { + y = pmf( x[ i ], r[ i ], p[ i ] ); + t.strictEqual( isAlmostSameValue( y, expected[ i ], 688 ), true, 'returns expected value' ); + } + t.end(); +}); + +tape( 'the function evaluates the pmf for `x` given small `r` and large `p`', opts, function test( t ) { + var expected; + var i; + var r; + var p; + var x; + var y; + + expected = smallHigh.expected; + x = smallHigh.x; + r = smallHigh.r; + p = smallHigh.p; + for ( i = 0; i < x.length; i++ ) { + y = pmf( x[ i ], r[ i ], p[ i ] ); + t.strictEqual( isAlmostSameValue( y, expected[ i ], 143 ), true, 'returns expected value' ); + } + t.end(); +}); + +tape( 'the function evaluates the pmf for `x` given small `r` and `p`', opts, function test( t ) { + var expected; + var i; + var r; + var p; + var x; + var y; + + expected = smallSmall.expected; + x = smallSmall.x; + r = smallSmall.r; + p = smallSmall.p; + for ( i = 0; i < x.length; i++ ) { + y = pmf( x[ i ], r[ i ], p[ i ] ); + t.strictEqual( isAlmostSameValue( y, expected[ i ], 110 ), true, 'returns expected value' ); + } + t.end(); +});