package com.imsl.test.example.stat.distributions;

import com.imsl.math.PrintMatrix;
import com.imsl.math.PrintMatrixFormat;
import com.imsl.stat.distributions.GeometricPD;
import java.text.DecimalFormat;
/**
 * <p>
 * Evaluates the geometric probability distribution.</p>
 * <p>
 * This example evaluates the geometric density, gradient, and hessian for a small
 * sample of data.</p>
 *
 * @see <a href="GeometricPDEx1.java">Code</a>
 * @see <a href="doc-files/GeometricPDEx1.out">Output</a>
 */

public class GeometricPDEx1 {

    public static void main(String[] args) {

        double[] x = {0, 1, 3, 5, 7, 9, 10, 12, 15};
        GeometricPD gpd = new GeometricPD();

        double[] values = new double[x.length];
        for(int i=0;i<x.length;i++){
            values[i] =  gpd.pdf(x[i], 0.5);
        }
       
        PrintMatrixFormat pmf = new PrintMatrixFormat();
        pmf.setNumberFormat(new DecimalFormat("0.00000000"));

        PrintMatrix pm = new PrintMatrix("Geometric(x, 0.5) pdf values at x=");
        pm.print(pmf, x);

        pm.setTitle("f(x; 0.5)=");
        pm.print(pmf, values);

        double[] gradient = gpd.getPDFGradient(x[3], 0.5);
        
        String title=String.format("Gradient of f(x;0.5) at x= %1.8f",x[3]);
        pm.setTitle(title);
        pm.print(pmf, gradient);

        double[][] hessian = gpd.getPDFHessian(x[3], 0.5);
        
        title=String.format("Hessian of f(x;0.5) at x= %1.8f",x[3]);
        pm.setTitle(title);
        pm.print(pmf, hessian);
    }
}