Skip to content

Commit 7f0e049

Browse files
committed
feat(src): add the C source files
1 parent aaf7299 commit 7f0e049

File tree

4 files changed

+193
-0
lines changed
  • lib/node_modules/@stdlib/stats/base/dists/negative-binomial/mgf

4 files changed

+193
-0
lines changed
Lines changed: 43 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,43 @@
1+
/**
2+
* @license Apache-2.0
3+
*
4+
* Copyright (c) 2024 The Stdlib Authors.
5+
*
6+
* Licensed under the Apache License, Version 2.0 (the "License");
7+
* you may not use this file except in compliance with the License.
8+
* You may obtain a copy of the License at
9+
*
10+
* http://www.apache.org/licenses/LICENSE-2.0
11+
*
12+
* Unless required by applicable law or agreed to in writing, software
13+
* distributed under the License is distributed on an "AS IS" BASIS,
14+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15+
* See the License for the specific language governing permissions and
16+
* limitations under the License.
17+
*/
18+
19+
#ifndef STDLIB_STATS_BASE_DISTS_NEGATIVE_BINOMIAL_MGF_H
20+
#define STDLIB_STATS_BASE_DISTS_NEGATIVE_BINOMIAL_MGF_H
21+
22+
/*
23+
* If C++, prevent name mangling so that the compiler emits a binary file having undecorated names, thus mirroring the behavior of a C compiler.
24+
*/
25+
#ifdef __cplusplus
26+
extern "C" {
27+
#endif
28+
29+
/**
30+
* Evaluates the moment-generating function (MGF) for a negative binomial distribution.
31+
*
32+
* @param t input value
33+
* @param r number of successes until experiment is stopped
34+
* @param p success probability
35+
* @return evaluated MGF
36+
*/
37+
double stdlib_base_negative_binomial_mgf(const double t, const double r, const double p);
38+
39+
#ifdef __cplusplus
40+
}
41+
#endif
42+
43+
#endif // !STDLIB_STATS_BASE_DISTS_NEGATIVE_BINOMIAL_MGF_H
Lines changed: 70 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,70 @@
1+
#/
2+
# @license Apache-2.0
3+
#
4+
# Copyright (c) 2024 The Stdlib Authors.
5+
#
6+
# Licensed under the Apache License, Version 2.0 (the "License");
7+
# you may not use this file except in compliance with the License.
8+
# You may obtain a copy of the License at
9+
#
10+
# http://www.apache.org/licenses/LICENSE-2.0
11+
#
12+
# Unless required by applicable law or agreed to in writing, software
13+
# distributed under the License is distributed on an "AS IS" BASIS,
14+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15+
# See the License for the specific language governing permissions and
16+
# limitations under the License.
17+
#/
18+
19+
# VARIABLES #
20+
21+
ifndef VERBOSE
22+
QUIET := @
23+
else
24+
QUIET :=
25+
endif
26+
27+
# Determine the OS ([1][1], [2][2]).
28+
#
29+
# [1]: https://en.wikipedia.org/wiki/Uname#Examples
30+
# [2]: http://stackoverflow.com/a/27776822/2225624
31+
OS ?= $(shell uname)
32+
ifneq (, $(findstring MINGW,$(OS)))
33+
OS := WINNT
34+
else
35+
ifneq (, $(findstring MSYS,$(OS)))
36+
OS := WINNT
37+
else
38+
ifneq (, $(findstring CYGWIN,$(OS)))
39+
OS := WINNT
40+
else
41+
ifneq (, $(findstring Windows_NT,$(OS)))
42+
OS := WINNT
43+
endif
44+
endif
45+
endif
46+
endif
47+
48+
49+
# RULES #
50+
51+
#/
52+
# Removes generated files for building an add-on.
53+
#
54+
# @example
55+
# make clean-addon
56+
#/
57+
clean-addon:
58+
$(QUIET) -rm -f *.o *.node
59+
60+
.PHONY: clean-addon
61+
62+
#/
63+
# Removes generated files.
64+
#
65+
# @example
66+
# make clean
67+
#/
68+
clean: clean-addon
69+
70+
.PHONY: clean
Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
/**
2+
* @license Apache-2.0
3+
*
4+
* Copyright (c) 2024 The Stdlib Authors.
5+
*
6+
* Licensed under the Apache License, Version 2.0 (the "License");
7+
* you may not use this file except in compliance with the License.
8+
* You may obtain a copy of the License at
9+
*
10+
* http://www.apache.org/licenses/LICENSE-2.0
11+
*
12+
* Unless required by applicable law or agreed to in writing, software
13+
* distributed under the License is distributed on an "AS IS" BASIS,
14+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15+
* See the License for the specific language governing permissions and
16+
* limitations under the License.
17+
*/
18+
19+
#include "stdlib/stats/base/dists/negative-binomial/mgf.h"
20+
#include "stdlib/math/base/napi/ternary.h"
21+
22+
// cppcheck-suppress shadowFunction
23+
STDLIB_MATH_BASE_NAPI_MODULE_DDD_D( stdlib_base_negative_binomial_mgf )
Lines changed: 57 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,57 @@
1+
/**
2+
* @license Apache-2.0
3+
*
4+
* Copyright (c) 2024 The Stdlib Authors.
5+
*
6+
* Licensed under the Apache License, Version 2.0 (the "License");
7+
* you may not use this file except in compliance with the License.
8+
* You may obtain a copy of the License at
9+
*
10+
* http://www.apache.org/licenses/LICENSE-2.0
11+
*
12+
* Unless required by applicable law or agreed to in writing, software
13+
* distributed under the License is distributed on an "AS IS" BASIS,
14+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15+
* See the License for the specific language governing permissions and
16+
* limitations under the License.
17+
*/
18+
19+
#include "stdlib/math/base/assert/is_nan.h"
20+
#include "stdlib/math/base/special/exp.h"
21+
#include "stdlib/math/base/special/pow.h"
22+
#include "stdlib/math/base/special/ln.h"
23+
24+
/**
25+
* Evaluates the moment-generating function (MGF) for a negative binomial distribution.
26+
*
27+
* @param t input value
28+
* @param r number of successes until experiment is stopped
29+
* @param p success probability
30+
* @return evaluated MGF
31+
*
32+
* @example
33+
* double y = stdlib_base_negative_binomial_mgf( 0.05, 20.0, 0.8 );
34+
* // returns ~267.839
35+
*/
36+
double stdlib_base_negative_binomial_mgf(const double t, const double r, const double p) {
37+
// Check for invalid input values:
38+
if (
39+
stdlib_base_is_nan(t) ||
40+
stdlib_base_is_nan(r) ||
41+
stdlib_base_is_nan(p) ||
42+
r <= 0.0 ||
43+
p < 0.0 ||
44+
p > 1.0 ||
45+
t >= -stdlib_base_ln(p)
46+
) {
47+
return 0.0 / 0.0; // NaN
48+
}
49+
50+
// Compute the MGF value:
51+
double exp_t = stdlib_base_exp(t);
52+
double numerator = (1.0 - p) * exp_t;
53+
double denominator = 1.0 - (p * exp_t);
54+
double ratio = numerator / denominator;
55+
56+
return stdlib_base_pow(ratio, r);
57+
}

0 commit comments

Comments
 (0)