[Home] DMelt member access. Java source code: org.statcato.dialogs.calc.probdist.GeometricDistributionDialog
/*
* GeometricDistributionDialog.java
*
* Created on May 16, 2008, 9:06 AM
*/
package org.statcato.dialogs.calc.probdist;
import org.statcato.*;
import org.statcato.spreadsheet.*;
import org.statcato.utils.HelperFunctions;
import org.statcato.statistics.*;
import javax.swing.*;
import java.util.*;
/**
* A dialog for geometric probability distribution.
* Provides options to compute the probability density, cumulative
* probability, and inverse cumulative probability.
*
* @author Margaret Yau
* @version %I%, %G%
* @since 1.0
*/
public class GeometricDistributionDialog extends StatcatoDialog {
/** Creates new form GeometricDistributionDialog */
public GeometricDistributionDialog(java.awt.Frame parent, boolean modal,
Statcato app) {
super(parent, modal);
this.app = app;
this.ParentSpreadsheet = app.getSpreadsheet();
initComponents();
ButtonGroup group0 = new ButtonGroup();
group0.add(Type1RadioButton);
group0.add(Type2RadioButton);
ButtonGroup group1 = new ButtonGroup();
group1.add(ProbDensityRadioButton);
group1.add(CumProbRadioButton);
group1.add(InvCumProbRadioButton);
ButtonGroup group2 = new ButtonGroup();
group2.add(ConstRadioButton);
group2.add(ColumnRadioButton);
ParentSpreadsheet.populateComboBox(ColumnComboBox);
getRootPane().setDefaultButton(OKButton);
setHelpFile("calc-geometric");
name = "Geometric Probability Distribution";
description = "For calculations on geometric probability distribution.";
helpStrings.add("Enter a number between 0 and 1 for the success probability.");
helpStrings.add("Choose probability density for P(x), cumulative " +
"probability for (P(<= x), and inverse cumulative probability " +
"for finding x given a such that P(<= x) = a.");
helpStrings.add("Inputs can be numbers within a column or a single " +
"constant.");
}
/**
* Updates elements on the dialog so that they have the most
* current values. Called by {@link #setVisible} to make
* sure the dialog displays current values when made visible again.
*
* @see #setVisible(boolean)
*/
@Override
public void updateElements() {
updateComboBox(ColumnComboBox);
}
/** This method is called from within the constructor to
* initialize the form.
* WARNING: Do NOT modify this code. The content of this method is
* always regenerated by the Form Editor.
*/
// //GEN-BEGIN:initComponents
private void initComponents() {
OKButton = new javax.swing.JButton();
CancelButton = new javax.swing.JButton();
jPanel1 = new javax.swing.JPanel();
Type1RadioButton = new javax.swing.JRadioButton();
pTextField = new javax.swing.JTextField();
ProbDensityRadioButton = new javax.swing.JRadioButton();
InvCumProbRadioButton = new javax.swing.JRadioButton();
jLabel4 = new javax.swing.JLabel();
Type2RadioButton = new javax.swing.JRadioButton();
jLabel2 = new javax.swing.JLabel();
CumProbRadioButton = new javax.swing.JRadioButton();
jLabel1 = new javax.swing.JLabel();
jPanel2 = new javax.swing.JPanel();
StoreTextField = new javax.swing.JTextField();
ColumnComboBox = new javax.swing.JComboBox();
jLabel6 = new javax.swing.JLabel();
jLabel7 = new javax.swing.JLabel();
ConstRadioButton = new javax.swing.JRadioButton();
ConstTextField = new javax.swing.JTextField();
jLabel5 = new javax.swing.JLabel();
ColumnRadioButton = new javax.swing.JRadioButton();
setDefaultCloseOperation(javax.swing.WindowConstants.DISPOSE_ON_CLOSE);
setTitle("Geometric Probability Distribution");
OKButton.setText("Compute");
OKButton.addActionListener(new java.awt.event.ActionListener() {
public void actionPerformed(java.awt.event.ActionEvent evt) {
OKButtonActionPerformed(evt);
}
});
CancelButton.setText("Close");
CancelButton.addActionListener(new java.awt.event.ActionListener() {
public void actionPerformed(java.awt.event.ActionEvent evt) {
CancelButtonActionPerformed(evt);
}
});
jPanel1.setBorder(javax.swing.BorderFactory.createTitledBorder("Distribution"));
Type1RadioButton.setSelected(true);
Type1RadioButton.setText("Model the number of trials needed to produce the first success");
ProbDensityRadioButton.setSelected(true);
ProbDensityRadioButton.setText("Probability density");
InvCumProbRadioButton.setText("Inverse cumulative probability");
jLabel4.setFont(new java.awt.Font("Tahoma", 1, 11));
jLabel4.setText("Compute:");
Type2RadioButton.setText("Model the number of failures before the first success");
jLabel2.setText("Success probability:");
CumProbRadioButton.setText("Cumulative probability");
jLabel1.setFont(new java.awt.Font("Tahoma", 1, 11));
jLabel1.setText("Distribution Parameters:");
javax.swing.GroupLayout jPanel1Layout = new javax.swing.GroupLayout(jPanel1);
jPanel1.setLayout(jPanel1Layout);
jPanel1Layout.setHorizontalGroup(
jPanel1Layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
.addGroup(jPanel1Layout.createSequentialGroup()
.addContainerGap()
.addGroup(jPanel1Layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
.addComponent(jLabel1)
.addGroup(jPanel1Layout.createSequentialGroup()
.addGap(9, 9, 9)
.addGroup(jPanel1Layout.createParallelGroup(javax.swing.GroupLayout.Alignment.TRAILING)
.addComponent(CumProbRadioButton, javax.swing.GroupLayout.Alignment.LEADING)
.addComponent(ProbDensityRadioButton, javax.swing.GroupLayout.Alignment.LEADING)
.addComponent(InvCumProbRadioButton, javax.swing.GroupLayout.Alignment.LEADING)))
.addGroup(jPanel1Layout.createSequentialGroup()
.addGap(18, 18, 18)
.addGroup(jPanel1Layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
.addGroup(jPanel1Layout.createSequentialGroup()
.addComponent(jLabel2)
.addGap(9, 9, 9)
.addComponent(pTextField, javax.swing.GroupLayout.PREFERRED_SIZE, 91, javax.swing.GroupLayout.PREFERRED_SIZE))
.addComponent(Type2RadioButton)
.addComponent(Type1RadioButton)))
.addComponent(jLabel4))
.addContainerGap())
);
jPanel1Layout.setVerticalGroup(
jPanel1Layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
.addGroup(jPanel1Layout.createSequentialGroup()
.addContainerGap()
.addComponent(jLabel1)
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
.addGroup(jPanel1Layout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
.addComponent(jLabel2)
.addComponent(pTextField, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE))
.addGap(21, 21, 21)
.addComponent(Type1RadioButton)
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
.addComponent(Type2RadioButton)
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
.addComponent(jLabel4)
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
.addComponent(ProbDensityRadioButton)
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
.addComponent(CumProbRadioButton)
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
.addComponent(InvCumProbRadioButton)
.addContainerGap())
);
jPanel2.setBorder(javax.swing.BorderFactory.createTitledBorder("Inputs and Outputs"));
jLabel6.setFont(new java.awt.Font("Tahoma", 1, 11));
jLabel6.setText("Store Results in: (optional)");
jLabel7.setText("(e.g. C1 for column label, or variable name)");
ConstRadioButton.setText("Constant:");
ConstRadioButton.addChangeListener(new javax.swing.event.ChangeListener() {
public void stateChanged(javax.swing.event.ChangeEvent evt) {
ConstRadioButtonStateChanged(evt);
}
});
ConstTextField.setEnabled(false);
jLabel5.setFont(new java.awt.Font("Tahoma", 1, 11));
jLabel5.setText("Input(s):");
ColumnRadioButton.setSelected(true);
ColumnRadioButton.setText("Column:");
ColumnRadioButton.addChangeListener(new javax.swing.event.ChangeListener() {
public void stateChanged(javax.swing.event.ChangeEvent evt) {
ColumnRadioButtonStateChanged(evt);
}
});
javax.swing.GroupLayout jPanel2Layout = new javax.swing.GroupLayout(jPanel2);
jPanel2.setLayout(jPanel2Layout);
jPanel2Layout.setHorizontalGroup(
jPanel2Layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
.addGroup(jPanel2Layout.createSequentialGroup()
.addContainerGap()
.addGroup(jPanel2Layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
.addGroup(jPanel2Layout.createSequentialGroup()
.addGap(9, 9, 9)
.addGroup(jPanel2Layout.createParallelGroup(javax.swing.GroupLayout.Alignment.TRAILING, false)
.addGroup(javax.swing.GroupLayout.Alignment.LEADING, jPanel2Layout.createSequentialGroup()
.addComponent(ConstRadioButton)
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
.addComponent(ConstTextField))
.addGroup(javax.swing.GroupLayout.Alignment.LEADING, jPanel2Layout.createSequentialGroup()
.addComponent(ColumnRadioButton)
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.UNRELATED)
.addComponent(ColumnComboBox, javax.swing.GroupLayout.PREFERRED_SIZE, 109, javax.swing.GroupLayout.PREFERRED_SIZE))))
.addGroup(jPanel2Layout.createSequentialGroup()
.addGap(11, 11, 11)
.addComponent(StoreTextField, javax.swing.GroupLayout.PREFERRED_SIZE, 105, javax.swing.GroupLayout.PREFERRED_SIZE)
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
.addComponent(jLabel7))
.addComponent(jLabel5)
.addComponent(jLabel6))
.addContainerGap(20, Short.MAX_VALUE))
);
jPanel2Layout.setVerticalGroup(
jPanel2Layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
.addGroup(jPanel2Layout.createSequentialGroup()
.addContainerGap()
.addComponent(jLabel5)
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
.addGroup(jPanel2Layout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
.addComponent(ColumnRadioButton)
.addComponent(ColumnComboBox, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE))
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
.addGroup(jPanel2Layout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
.addComponent(ConstRadioButton)
.addComponent(ConstTextField, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE))
.addGap(20, 20, 20)
.addComponent(jLabel6)
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
.addGroup(jPanel2Layout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
.addComponent(StoreTextField, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE)
.addComponent(jLabel7))
.addContainerGap())
);
javax.swing.GroupLayout layout = new javax.swing.GroupLayout(getContentPane());
getContentPane().setLayout(layout);
layout.setHorizontalGroup(
layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
.addGroup(layout.createSequentialGroup()
.addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
.addGroup(layout.createSequentialGroup()
.addContainerGap()
.addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING, false)
.addComponent(jPanel1, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, Short.MAX_VALUE)
.addComponent(jPanel2, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, Short.MAX_VALUE)))
.addGroup(layout.createSequentialGroup()
.addGap(99, 99, 99)
.addComponent(OKButton)
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.UNRELATED)
.addComponent(CancelButton)))
.addContainerGap(18, Short.MAX_VALUE))
);
layout.linkSize(javax.swing.SwingConstants.HORIZONTAL, new java.awt.Component[] {CancelButton, OKButton});
layout.setVerticalGroup(
layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
.addGroup(layout.createSequentialGroup()
.addContainerGap()
.addComponent(jPanel1, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE)
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
.addComponent(jPanel2, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE)
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.UNRELATED)
.addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
.addComponent(OKButton)
.addComponent(CancelButton))
.addGap(20, 20, 20))
);
pack();
}// //GEN-END:initComponents
private void ConstRadioButtonStateChanged(javax.swing.event.ChangeEvent evt) {//GEN-FIRST:event_ConstRadioButtonStateChanged
if (ConstRadioButton.isSelected()) {
ColumnComboBox.setEnabled(false);
ConstTextField.setEnabled(true);
} else {
ColumnComboBox.setEnabled(true);
ConstTextField.setEnabled(false);
}
}//GEN-LAST:event_ConstRadioButtonStateChanged
private void ColumnRadioButtonStateChanged(javax.swing.event.ChangeEvent evt) {//GEN-FIRST:event_ColumnRadioButtonStateChanged
if (ColumnRadioButton.isSelected()) {
ColumnComboBox.setEnabled(true);
ConstTextField.setEnabled(false);
} else {
ColumnComboBox.setEnabled(false);
ConstTextField.setEnabled(true);
}
}//GEN-LAST:event_ColumnRadioButtonStateChanged
private void OKButtonActionPerformed(java.awt.event.ActionEvent evt) {//GEN-FIRST:event_OKButtonActionPerformed
app.compoundEdit = new DialogEdit("geometric distribution");
double p = 0;
String heading = "Geometric Distribution: ";
String text = "";
try {
p = Double.parseDouble(pTextField.getText());
} catch (NumberFormatException e) {
app.showErrorDialog("Enter a valid number for the success probability.");
return;
}
if (p <= 0 || p >= 1) {
app.showErrorDialog("The success probability must be between 0 and 1.");
return;
}
heading += "p=" + p;
Vector ColumnVector = new Vector();
// get input
if (ColumnRadioButton.isSelected()) {
// get data for selected input column
String columnLabel = ColumnComboBox.getSelectedItem().toString();
if (columnLabel.equals("")) { // no input variable
app.showErrorDialog("Select the input column.");
return;
}
int column = ParentSpreadsheet.parseColumnNumber(columnLabel);
Vector StrColumnVector = ParentSpreadsheet.getColumn(column);
ColumnVector = HelperFunctions.ConvertInputVectorToDoubles(StrColumnVector);
if (ColumnVector == null) {
app.showErrorDialog("Invalid input column " + columnLabel +
": all data must be numbers.");
return;
}
text += "Input: " + columnLabel + "
";
} else {
// get input from constant text field
try {
double num = Double.parseDouble(ConstTextField.getText());
ColumnVector.addElement(num);
text += "Input: " + num + "
";
} catch (NumberFormatException e) {
app.showErrorDialog("Invalid constant input. Enter a valid number.");
return;
}
}
// compute results
GeometricProbabilityDistribution npd = null;
if (Type1RadioButton.isSelected())
npd = new GeometricProbabilityDistribution(p, GeometricProbabilityDistribution.TYPEI);
else
npd = new GeometricProbabilityDistribution(p, GeometricProbabilityDistribution.TYPEII);
Vector ResultVector = new Vector();
if (ProbDensityRadioButton.isSelected()) {
text += "Type: Probability density
";
text += "";
text += "| X | P(X) | ";
for (int i = 0; i < ColumnVector.size(); ++i) {
Double elm = ColumnVector.elementAt(i);
if (elm != null) {
double x = elm.doubleValue();
double density = npd.density(x);
text += "| " + x + " | " +
HelperFunctions.formatFloat(density, 6) + " | ";
ResultVector.addElement(density+"");
}
}
text += " ";
} else if (CumProbRadioButton.isSelected()) {
text += "Type: Cumulative probability
";
text += "";
text += "| X | P(<<=X) | ";
for (int i = 0; i < ColumnVector.size(); ++i) {
Double elm = ColumnVector.elementAt(i);
if (elm != null) {
double x = elm.doubleValue();
double density = npd.cumulativeProbability(x);
text += "| " + x + " | " +
HelperFunctions.formatFloat(density, 6) + " | ";
ResultVector.addElement(density+"");
}
}
text += " ";
} else if (InvCumProbRadioButton.isSelected()) {
text += "Type: Inverse cumulative probability
";
text += "";
text += "| P(<<=X) | X | ";
for (int i = 0; i < ColumnVector.size(); ++i) {
Double elm = ColumnVector.elementAt(i);
if (elm != null) {
double area = elm.doubleValue();
Double x = npd.inverseCumulativeProbability(area);
if (x.equals(Double.NaN)) {
text += "| " + area + " | " +
"*" + " | ";
ResultVector.addElement(x+"");
} else {
text += "| " + area + " | " +
(int)x.doubleValue() + " | ";
ResultVector.addElement((int)x.doubleValue() + "");
}
}
}
text += " ";
}
String storeColumn = StoreTextField.getText();
if (!storeColumn.equals("")) {
// check if Column is valid
int storeColumnNum = ParentSpreadsheet.getColumnNumber(storeColumn);
if (storeColumnNum == -1) { // invalid column
app.showErrorDialog("Enter a valid column name (e.g. C1) or a valid variable name to store the results.");
return;
}
ParentSpreadsheet.setColumn(storeColumnNum, ResultVector);
}
app.addLogParagraph(heading, text);
app.compoundEdit.end();
app.addCompoundEdit(app.compoundEdit);
setVisible(false);
}//GEN-LAST:event_OKButtonActionPerformed
private void CancelButtonActionPerformed(java.awt.event.ActionEvent evt) {//GEN-FIRST:event_CancelButtonActionPerformed
setVisible(false);
}//GEN-LAST:event_CancelButtonActionPerformed
// Variables declaration - do not modify//GEN-BEGIN:variables
private javax.swing.JButton CancelButton;
private javax.swing.JComboBox ColumnComboBox;
private javax.swing.JRadioButton ColumnRadioButton;
private javax.swing.JRadioButton ConstRadioButton;
private javax.swing.JTextField ConstTextField;
private javax.swing.JRadioButton CumProbRadioButton;
private javax.swing.JRadioButton InvCumProbRadioButton;
private javax.swing.JButton OKButton;
private javax.swing.JRadioButton ProbDensityRadioButton;
private javax.swing.JTextField StoreTextField;
private javax.swing.JRadioButton Type1RadioButton;
private javax.swing.JRadioButton Type2RadioButton;
private javax.swing.JLabel jLabel1;
private javax.swing.JLabel jLabel2;
private javax.swing.JLabel jLabel4;
private javax.swing.JLabel jLabel5;
private javax.swing.JLabel jLabel6;
private javax.swing.JLabel jLabel7;
private javax.swing.JPanel jPanel1;
private javax.swing.JPanel jPanel2;
private javax.swing.JTextField pTextField;
// End of variables declaration//GEN-END:variables
}
|
Ads help maintain this website.