ContinuousDistChart.java

/*
 * Class:        ContinuousDistChart
 * Description:  
 * Environment:  Java
 * Software:     SSJ 
 * Copyright (C) 2001  Pierre L'Ecuyer and Universite de Montreal
 * Organization: DIRO, Universite de Montreal
 * @author       Richard Simard
 * @since        May 2008
 *
 *
 * 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.
 *
 */
package umontreal.ssj.charts;
 
import umontreal.ssj.probdist.ContinuousDistribution;
 
import javax.swing.JFrame;

public class ContinuousDistChart {
   protected ContinuousDistribution dist;
   protected double a, b;
   protected int m;
   protected XYLineChart cdfChart;
   protected XYLineChart densityChart;
 
   private void init() {
      double[][] cdf = new double[2][m + 1];
      double[][] density = new double[2][m + 1];
      double h = (b - a) / m;
      double x;
      int coex = 0;
 
      try {
         for (int i = 0; i <= m; i++) {
            x = a + i * h;
            cdf[0][i] = x;
            cdf[1][i] = dist.cdf(x);
         }
         cdfChart = new XYLineChart("cdf: " + dist.toString(), "", "", cdf);
      } catch (UnsupportedOperationException e) {
         coex++;
         System.err.println(e);
//         e.printStackTrace();
      }
 
      try {
         for (int i = 0; i <= m; i++) {
            x = a + i * h;
            density[0][i] = x;
            density[1][i] = dist.density(x);
         }
         densityChart = new XYLineChart("density: " + dist.toString(), "", "", density);
      } catch (UnsupportedOperationException e) {
         System.err.println(e);
         if (coex == 1)
            throw e;
      }
      cdfChart.setprobFlag(true);
      densityChart.setprobFlag(true);
   }

   public ContinuousDistChart(ContinuousDistribution dist, double a, double b, int m) {
      this.dist = dist;
      this.a = a;
      this.b = b;
      this.m = m;
      init();
   }

   public JFrame viewCdf(int width, int height) {
      return cdfChart.view(width, height);
   }

   public JFrame viewDensity(int width, int height) {
      return densityChart.view(width, height);
   }

   public String toLatexCdf(int width, int height) {
      return cdfChart.toLatex(width, height);
   }

   public String toLatexDensity(int width, int height) {
      return densityChart.toLatex(width, height);
   }

   public void setParam(double a, double b, int m) {
      this.a = a;
      this.b = b;
      this.m = m;
      init();
   }
 
}