All files main.js

97.6% Statements 122/125
87.5% Branches 14/16
100% Functions 1/1
97.6% Lines 122/125

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 1261x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 6101x 6101x 6101x 6101x 6101x 6101x 6101x 6101x 6101x 6101x 6101x 2102x 104x 104x 1998x 1998x 2102x       2102x 2102x 2102x 6101x 9x 9x 9x 6101x 28x 6101x 40x 40x 3971x 3931x 3931x 3931x 3999x 6101x 1x 1x 1x 1x 1x  
/**
* @license Apache-2.0
*
* Copyright (c) 2025 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.
*
*
* ## Notice
*
* The original C++ code and copyright notice are from the [Boost library]{@link http://www.boost.org/doc/libs/1_88_0/boost/math/special_functions/trigamma.hpp}. The implementation follows the original but has been reformatted and modified for JavaScript.
*
* ```text
* (C) Copyright John Maddock 2006.
*
* Use, modification and distribution are subject to the
* Boost Software License, Version 1.0. (See accompanying file
* LICENSE or copy at http://www.boost.org/LICENSE_1_0.txt)
* ```
*/
 
'use strict';
 
// MODULES //
 
var f32 = require( '@stdlib/number/float64/base/to-float32' );
var sinpif = require( '@stdlib/math/base/special/sinpif' );
var absf = require( '@stdlib/math/base/special/absf' );
var floorf = require( '@stdlib/math/base/special/floorf' );
var PI_SQUARED = require( '@stdlib/constants/float32/pi-squared' );
var rateval12 = require( './rational_p12q12.js' );
var rateval24 = require( './rational_p24q24.js' );
var rateval4INF = require( './rational_p4infq4inf.js' );
 
 
// VARIABLES //
 
var YOFFSET12 = f32( 2.1093254089355469 );
var ZERO = f32( 0.0 );
var ONE = f32( 1.0 );
var TWO = f32( 2.0 );
var FOUR = f32( 4.0 );
 
 
// MAIN //
 
/**
* Evaluates the trigamma function for a single-precision floating-point number.
*
* @param {number} x - input value
* @returns {number} function value
*
* @example
* var v = trigammaf( -2.5 );
* // returns ~9.539
*
* @example
* var v = trigammaf( 1.0 );
* // returns ~1.645
*
* @example
* var v = trigammaf( 10.0 );
* // returns ~0.105
*
* @example
* var v = trigammaf( NaN );
* // returns NaN
*
* @example
* var v = trigammaf( -1.0 );
* // returns NaN
*/
function trigammaf( x ) {
	var result;
	var s;
	var y;
	var z;
 
	x = f32( x );
	result = ZERO;
 
	// Check for negative arguments and use reflection:
	if ( x <= ZERO ) {
		if ( floorf( x ) === x ) {
			return NaN;
		}
		// Reflect:
		z = f32( ONE - x );
		if ( z < ONE ) {
			result = f32( ONE / f32( z*z ) );
			z = f32( ONE + z );
		}
		s = ( absf( x ) < absf( z ) ) ? sinpif( x ) : sinpif( z );
		return f32( f32( result - trigammaf( z ) ) + f32( PI_SQUARED / f32( s*s ) ) );
	}
	if ( x < ONE ) {
		result = f32( ONE / f32( x*x ) );
		x = f32( x + ONE );
	}
	if ( x <= TWO ) {
		result = f32( result + f32( f32( YOFFSET12 + rateval12( x ) ) / f32( x*x ) ) );
	} else if ( x <= FOUR ) {
		y = f32( ONE / x );
		result = f32( result + f32( f32( ONE + rateval24( y ) ) / x ) );
	} else {
		y = f32( ONE / x );
		result = f32( result + f32( f32( ONE + rateval4INF( y ) ) / x ) );
	}
	return result;
}
 
 
// EXPORTS //
 
module.exports = trigammaf;