DiscreteDistIntChart.java

/*
 * Class:        DiscreteDistIntChart
 * 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.DiscreteDistributionInt;
 
import javax.swing.JFrame;
import java.awt.*;

public class DiscreteDistIntChart {
   protected DiscreteDistributionInt dist;
   protected int a, b;
   protected XYLineChart cdfChart;
   protected XYLineChart probChart;

   public DiscreteDistIntChart(DiscreteDistributionInt dist) {
      this.dist = dist;
      this.a = 0;
      this.b = 0;
   }

   public DiscreteDistIntChart(DiscreteDistributionInt dist, int a, int b) {
      this.dist = dist;
      if (a >= b)
         throw new IllegalArgumentException("a is bigger than b");
      this.a = a;
      this.b = b;
      init();
   }
 
   private void init() {
      int m = b - a + 1;
      double[][] cdf = new double[2][m];
      double[][] probability = new double[2][m];
 
      for (int i = 0; i < m; i++) {
         cdf[0][i] = i + a;
         cdf[1][i] = dist.cdf(i + a);
         probability[0][i] = i + a;
         probability[1][i] = dist.prob(i + a);
      }
 
      double[][] cdfFinal = new double[2][2 * (m - 1)];
      for (int i = 0; i < m - 1; i++) {
         cdfFinal[0][2 * i] = cdf[0][i];
         cdfFinal[0][2 * i + 1] = cdf[0][i + 1];
         cdfFinal[1][2 * i] = cdf[1][i];
         cdfFinal[1][2 * i + 1] = cdf[1][i];
      }
 
      cdfChart = new XYLineChart("cdf: " + dist.toString(), "", "", cdfFinal);
      probChart = new XYLineChart("probability: " + dist.toString(), "", "", probability);
      cdfChart.setprobFlag(true);
      probChart.setprobFlag(true);
 
      // Only for tikZ
      XYListSeriesCollection collec = cdfChart.getSeriesCollection();
      collec.setColor(0, Color.BLUE);
      collec.setPlotStyle(0, "thick");
      collec.setMarksType(0, "only marks");
 
      collec = probChart.getSeriesCollection();
      collec.setColor(0, Color.ORANGE);
      collec.setPlotStyle(0, "ycomb");
      collec.setMarksType(0, "*");
      collec.setDashPattern(0, "solid");
   }
 
   private void testParam() {
      if (a == 0 && b == 0) {
         double mean = dist.getMean();
         double sd = dist.getStandardDeviation();
         int xa = (int) Math.round(mean - 3.0 * sd);
         if (xa < dist.getXinf())
            xa = dist.getXinf();
         int xb = (int) Math.round(mean + 3.0 * sd);
         if (xb > dist.getXsup())
            xb = dist.getXsup();
         setParam(xa, xb);
      }
   }

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

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

   public JFrame viewProb(int width, int height, int a, int b) {
      setParam(a, b);
      return probChart.viewBar(width, height);
   }

   public JFrame viewProb(int width, int height) {
      testParam();
      return probChart.viewBar(width, height);
   }

   public void setParam(int a, int b) {
      if (a >= b)
         throw new IllegalArgumentException("a is bigger than b" + a + "  " + b);
      this.a = a;
      this.b = b;
      init();
   }

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

   public String toLatexProb(int width, int height) {
      testParam();
      return probChart.toLatex(width, height);
   }

   public XYLineChart getCdf() {
      return cdfChart;
   }

   public XYLineChart getProb() {
      return probChart;
   }
 
}