/*
 * FuzzyRulePanel.java
 *
 * Created on July 7, 2002, 5:44 PM
 */

package uk.ac.leeds.ccg.projects.MedAction.FuzzyInference;

/**
 * a GUI to define rules for this model
 * @author  jianhui jin
 */
import javax.swing.*;
import java.io.*;
public class FuzzyRulePanel extends javax.swing.JPanel {
 
    public FuzzyRulePanel() {
        setFuzzyRuleModel(new FuzzyRuleModel("no name"));
        
        initComponents();
        //initialize the Variable and other's panel
        iniVariablePanel(variableComboBox.length);
        setBorder(new javax.swing.border.TitledBorder("No name" + " Fuzzy Rules"));
    }
    /**
     *
     * @param name
     */    
    public FuzzyRulePanel(String name) {
        setFuzzyRuleModel(new FuzzyRuleModel(name));
        
        initComponents();
        //initialize the Variable and other's panel
        iniVariablePanel(variableComboBox.length);
        setBorder(new javax.swing.border.TitledBorder(name + " Fuzzy Rules"));
    }
    
    /**
     *
     * @param model
     */    
    public FuzzyRulePanel( FuzzyRuleModel model ) {
        this.fuzzyRuleModel=model;
        initComponents();
        //initialize the Variable and other's panel
        iniVariablePanel(model.getFuzzyVariableName(model.getFuzzyVariable()).length);
        setBorder(new javax.swing.border.TitledBorder(model.getName() + " Fuzzy Rules"));
        FuzzyRule[] rules=model.getFuzzyRule();
        if(rules!=null&&rules.length!=0){
            for(int i=0;i<rules.length;i++) {
                this.ruleCombo.addItem("Rule " + Integer.toString(i+1));
            }
            this.ruleCombo.setSelectedIndex(0);
            
        }
        else {
            this.setPanel(model);
        }
    }
    
    
    
   
    /** This method is called from within the constructor to
     * initialize the form.
     */
    private void initComponents() {
        outVariableComboBox = new javax.swing.JComboBox();
        setNewRuleButton = new javax.swing.JButton();
        ruleCombo = new javax.swing.JComboBox();
        removeRuleButton = new javax.swing.JButton();
        ruleAdjustButton = new javax.swing.JButton();
        addVariableButton=new JButton();
        deleteVariableButton=new JButton();
        variableScrollPane = new javax.swing.JScrollPane();
        variablePanel = new javax.swing.JPanel();
        jLabel1 = new javax.swing.JLabel();
        jLabel2 = new javax.swing.JLabel();
        jLabel3 = new javax.swing.JLabel();
        jLabel4 = new javax.swing.JLabel();
        jLabel5 = new JLabel();
        confidenceTextField=new JTextField();
        
        setLayout(new org.netbeans.lib.awtextra.AbsoluteLayout());
        setBackground(new java.awt.Color(255, 255, 255));
        outVariableComboBox.setBackground(new java.awt.Color(0, 255, 255));
        add(outVariableComboBox, new org.netbeans.lib.awtextra.AbsoluteConstraints(210, 280, 80, 20));
        jLabel1.setText("IF");
        jLabel1.setFont(new java.awt.Font("Dialog", 1, 18));
        add(jLabel1, new org.netbeans.lib.awtextra.AbsoluteConstraints(20, 10, 20, 30));
        jLabel3.setText("THEN");
        jLabel3.setFont(new java.awt.Font("Dialog", 1, 18));
        add(jLabel3, new org.netbeans.lib.awtextra.AbsoluteConstraints(20, 270, -1, -1));
        
        jLabel2.setText("IS");
        add(jLabel2, new org.netbeans.lib.awtextra.AbsoluteConstraints(190, 280, -1, -1));
        
        jLabel4.setText("OUTVARIABLE");
        jLabel4.setForeground(new java.awt.Color(153, 255, 153));
        jLabel4.setFont(new java.awt.Font("Dialog", 1, 14));
        add(jLabel4, new org.netbeans.lib.awtextra.AbsoluteConstraints(80, 280, -1, -1));
        
        setNewRuleButton.setText("Add New Rule");
        setNewRuleButton.addActionListener(new java.awt.event.ActionListener() {
            public void actionPerformed(java.awt.event.ActionEvent evt) {
                setNewRuleButtonActionPerformed(evt);
            }
        });
        
        add(setNewRuleButton, new org.netbeans.lib.awtextra.AbsoluteConstraints(20, 320, -1, -1));
        
        ruleCombo.setBackground(new java.awt.Color(204, 204, 255));
        ruleCombo.addActionListener(new java.awt.event.ActionListener() {
            public void actionPerformed(java.awt.event.ActionEvent evt) {
                ruleComboActionPerformed(evt);
            }
        });
        
        add(ruleCombo, new org.netbeans.lib.awtextra.AbsoluteConstraints(155, 320, 80, 20));
        
        removeRuleButton.setText("Remove Rule");
        removeRuleButton.addActionListener(new java.awt.event.ActionListener() {
            public void actionPerformed(java.awt.event.ActionEvent evt) {
                removeRuleButtonActionPerformed(evt);
            }
        });
        
        add(removeRuleButton, new org.netbeans.lib.awtextra.AbsoluteConstraints(260, 320, 110, -1));
        
        ruleAdjustButton.setText("Set Rule");
        ruleAdjustButton.addActionListener(new java.awt.event.ActionListener() {
            public void actionPerformed(java.awt.event.ActionEvent evt) {
                ruleAdjustButtonActionPerformed(evt);
            }
        });
        
        add(ruleAdjustButton, new org.netbeans.lib.awtextra.AbsoluteConstraints(20, 350, 110, -1));
        
        
        addVariableButton.setText("Add Variable");
        addVariableButton.addActionListener(new java.awt.event.ActionListener() {
            public void actionPerformed(java.awt.event.ActionEvent evt) {
                addVariableButtonActionPerformed(evt);
            }
        });
        add(addVariableButton, new org.netbeans.lib.awtextra.AbsoluteConstraints(140, 350, -1, -1));
        
        
        deleteVariableButton.setText("Delete Variable");
        deleteVariableButton.addActionListener(new java.awt.event.ActionListener() {
            public void actionPerformed(java.awt.event.ActionEvent evt) {
                deleteVariableButtonActionPerformed(evt);
            }
        });
        add(deleteVariableButton, new org.netbeans.lib.awtextra.AbsoluteConstraints(260, 350, 110, -1));
        
        confidenceTextField.setToolTipText("value is 1 to 10");
        confidenceTextField.setText("1");
        confidenceTextField.setBackground(new java.awt.Color(255, 255, 255));
        confidenceTextField.addActionListener(new java.awt.event.ActionListener() {
            public void actionPerformed(java.awt.event.ActionEvent evt) {
                confidenceTextFieldActionPerformed(evt);
            }
        });
        
        add(confidenceTextField, new org.netbeans.lib.awtextra.AbsoluteConstraints(390, 280, 40, -1));
        
        jLabel5.setText("CONFIDENCE");
        add(jLabel5, new org.netbeans.lib.awtextra.AbsoluteConstraints(300, 280, 80, -1));
        
        
        
        
        
        
        variablePanel.setLayout(new org.netbeans.lib.awtextra.AbsoluteLayout());
        variablePanel.setBackground(new java.awt.Color(255, 255, 255));
        
        
    }
    
    private void confidenceTextFieldActionPerformed(java.awt.event.ActionEvent evt) {
        
        FuzzyRule[] r = fuzzyRuleModel.getFuzzyRule();
        if((r!=null)&&(r.length>0)){
            r[ruleCombo.getSelectedIndex()].setConfidence(Integer.parseInt(confidenceTextField.getText()));
        }// Add your handling code here:
    }
    
    private void iniVariablePanel(int length) {
        isLabel=new JLabel[length];
        variableComboBox=new JComboBox[length];
        variableMemberComboBox=new JComboBox[length];
        operatorComboBox=new JComboBox[length-1];
        variablePanel.removeAll();
        for(int i=0;i<length;i++) {
            isLabel[i]=new JLabel("IS");
            variablePanel.add(isLabel[i], new org.netbeans.lib.awtextra.AbsoluteConstraints(140, 20 + i * 40, -1, -1));
            
            
            variableComboBox[i]=new JComboBox();
            variableComboBox[i].setBackground(new java.awt.Color(0, 255, 255));
            variableComboBox[i].addActionListener(new java.awt.event.ActionListener() {
                public void actionPerformed(java.awt.event.ActionEvent evt) {
                    variableActionPerformed(evt);
                }
            });
            variablePanel.add(variableComboBox[i], new org.netbeans.lib.awtextra.AbsoluteConstraints(50, 20+i*40, 80, 20));
            
            
            
            variableMemberComboBox[i]=new JComboBox();
            variableMemberComboBox[i].setBackground(new java.awt.Color(255, 255, 204));
            variablePanel.add(variableMemberComboBox[i], new org.netbeans.lib.awtextra.AbsoluteConstraints(160, 20+i*40, 80, 20));
            
        }
        
        
        
        for(int i=0;i<operatorComboBox.length;i++) {
            operatorComboBox[i] = new JComboBox();
            operatorComboBox[i].setBackground(new java.awt.Color(255, 153, 153));
            setComboBox(operatorComboBox[i],fuzzyRuleModel.operatorName);
            variablePanel.add(operatorComboBox[i], new org.netbeans.lib.awtextra.AbsoluteConstraints(250, 20 + i*40, 80, 20));
        }
        
        
        
        variableScrollPane.setViewportView(variablePanel);
        this.remove(variableScrollPane);
        add(variableScrollPane, new org.netbeans.lib.awtextra.AbsoluteConstraints(70, 20, 350, 220));
        this.validate();
    }
    
    
    private void ruleAdjustButtonActionPerformed(java.awt.event.ActionEvent evt) {
        if((fuzzyRuleModel.getFuzzyRule()!=null)&&(fuzzyRuleModel.getFuzzyRule().length!=0)){
            String[] variableName=new String[variableComboBox.length];
            String[] memberName=new String[variableMemberComboBox.length];
            String[] operatorName=new String[operatorComboBox.length];
            for(int i=0;i<variableComboBox.length;i++) {
                variableName[i] = (String) variableComboBox[i].getSelectedItem();
                memberName[i]= (String) variableMemberComboBox[i].getSelectedItem();
            }
            
            for(int i=0;i<operatorComboBox.length;i++) {
                operatorName[i]=(String) operatorComboBox[i].getSelectedItem();
            }
            
            FuzzyRule newOne=new FuzzyRule(variableName,memberName, operatorName, (String) outVariableComboBox.getSelectedItem(),Integer.parseInt(confidenceTextField.getText()));
            
            FuzzyRule[] rules=fuzzyRuleModel.getFuzzyRule();
            // assign new one this position instead of old one
            rules[ruleCombo.getSelectedIndex()]=newOne;
            // reset the rules to model
            fuzzyRuleModel.setFuzzyRule(rules);
        }
        // Add your handling code here:
    }
    
    private void ruleComboActionPerformed(java.awt.event.ActionEvent evt) {
        FuzzyRule[] temp=fuzzyRuleModel.getFuzzyRule();
        if((temp!=null)&&(temp.length>0)){
            
            int index=((JComboBox) evt.getSource()).getSelectedIndex();
            
            if(index < temp.length&& index >= 0){
                FuzzyRule selectedRule = temp[index];
                iniRuleComponents(selectedRule);
            }
        }
        // Add your handling code here:
    }
    private void iniRuleComponents(FuzzyRule rule){
        
        String[] vName=rule.getVariableName();
        String[] mName=rule.getMemberName();
        String[] opName=rule.getOperatorName();
        
        iniVariablePanel(vName.length);
        
        setPanel(this.fuzzyRuleModel);
        
        for(int i=0;i<vName.length;i++) {
            variableComboBox[i].setSelectedItem(vName[i]);
            variableMemberComboBox[i].setSelectedItem(mName[i]);
            
        }
        
        for(int i=0;i<operatorComboBox.length;i++) {
            operatorComboBox[i].setSelectedItem(opName[i]);
        }
        this.confidenceTextField.setText(Integer.toString(rule.getConfidence()));
        outVariableComboBox.setSelectedItem(rule.getOutMember());
    }
    
    private void removeRuleButtonActionPerformed(java.awt.event.ActionEvent evt) {
        
        if((ruleCombo.getItemCount() >0)&&(fuzzyRuleModel.getFuzzyRule()!=null)&&(fuzzyRuleModel.getFuzzyRule().length > 0)){
            fuzzyRuleModel.removeRule(ruleCombo.getSelectedIndex());
            ruleCombo.removeItemAt(ruleCombo.getItemCount()-1);
            
            int temp=ruleCombo.getSelectedIndex();
            FuzzyRule[] rules=fuzzyRuleModel.getFuzzyRule();
            if(temp>=0) iniRuleComponents(rules[temp]);
        }
        // Add your handling code here:
    }
    
    private void setNewRuleButtonActionPerformed(java.awt.event.ActionEvent evt) {
        if(variableComboBox[0].getItemCount()<0) return;
        String[] variableName=new String[variableComboBox.length];
        String[] memberName=new String[variableMemberComboBox.length];
        String[] operatorName=new String[operatorComboBox.length];
        for(int i=0;i<variableComboBox.length;i++) {
            variableName[i] = (String) variableComboBox[i].getSelectedItem();
            memberName[i] = (String) variableMemberComboBox[i].getSelectedItem();
        }
        
        for(int i=0;i<operatorComboBox.length;i++) {
            operatorName[i]=(String) operatorComboBox[i].getSelectedItem();
        }
        
        FuzzyRule newOne=new FuzzyRule(variableName,memberName, operatorName, (String) outVariableComboBox.getSelectedItem(),Integer.parseInt(confidenceTextField.getText()));
        fuzzyRuleModel.setFuzzyRule(newOne);
        
        ruleCombo.removeAllItems();
        
        for(int i=0;i<fuzzyRuleModel.getFuzzyRule().length;i++)
            ruleCombo.addItem("Rule " + Integer.toString(i+1));
        
        ruleCombo.setSelectedIndex(ruleCombo.getItemCount()-1);
        
        // Add your handling code here:
        
    }
    
    private void variableActionPerformed(java.awt.event.ActionEvent evt) {
        FuzzyVariable[] fuzzyVariable=fuzzyRuleModel.getFuzzyVariable();
        JComboBox temp=(JComboBox) evt.getSource();
        JComboBox tempMember=null;
        // find the corresponding variableMemberComboBox
        for(int i=0;i<variableComboBox.length;i++) {
            if(variableComboBox[i].equals(temp))
            { tempMember=variableMemberComboBox[i];break;}
        }
        // find the selected fuzzyVariable, and assign all the member name of the selected variable to the corresponding variableMemberComboBox
        for(int i=0;i<fuzzyVariable.length;i++) {
            if(fuzzyVariable[i].getName().equals((String)temp.getSelectedItem())) {
                
                setComboBox(tempMember,fuzzyVariable[i].getMemberName());break;
            }
        }// Add your handling code here:
    }
    
    
    private void setComboBox(JComboBox box, String[] names) {
        box.removeAllItems();
        box.addItem("none");
        for(int i=0;i<names.length;i++)
            box.addItem(names[i]);
        
    }
    
    
    
    
    /**
     *
     * @param model
     */    
    public void setFuzzyRuleModel(FuzzyRuleModel model) {
        fuzzyRuleModel=model;
        this.removeAll();
        initComponents();
        this.iniVariablePanel(model.getFuzzyVariable().length);
        setBorder(new javax.swing.border.TitledBorder(model.getName() + " Fuzzy Rules"));
        this.setPanel(model);
    }
    
    /**
     *
     * @return
     */    
    public FuzzyRuleModel getFuzzyRuleModel() {
        return fuzzyRuleModel;
    }
    
    /**
     *
     * @param model
     */    
    public void setPanel(FuzzyRuleModel model) {
        
        
        if((model!=null)&&(model.getFuzzyVariable()!=null)) {
            this.fuzzyRuleModel=model;
            FuzzyRule[] rules=this.fuzzyRuleModel.getFuzzyRule();
            
            FuzzyVariable[] fuzzyVariable=fuzzyRuleModel.getFuzzyVariable();
            FuzzyVariable outVariable=fuzzyRuleModel.getOutVariable();
            for(int i=0;i<variableComboBox.length;i++) {
                setComboBox(variableComboBox[i],fuzzyRuleModel.getFuzzyVariableName(fuzzyRuleModel.getFuzzyVariable()));
            }
            setComboBox(outVariableComboBox,outVariable.getMemberName());
        }
        
    }
    
    
    private void addVariableButtonActionPerformed(java.awt.event.ActionEvent evt) {
        // assign old array to temp arrays
        JComboBox[] oldV=variableComboBox;
        JComboBox[] oldM=variableMemberComboBox;
        JComboBox[] oldO=operatorComboBox;
        JLabel[] oldIs=isLabel;
        // construct new arrays with old length increasing one
        variableComboBox=new JComboBox[oldV.length+1];
        variableMemberComboBox=new JComboBox[oldM.length+1];
        isLabel=new JLabel[oldIs.length+1];
        operatorComboBox=new JComboBox[oldO.length+1];
        
        // assign old to new without the final one
        for(int i=0;i<oldV.length;i++) {
            variableComboBox[i] = oldV[i];
            variableMemberComboBox[i]=oldM[i];
            isLabel[i]=oldIs[i];
        }
        // assign old to new without the final one
        for(int i=0;i<oldO.length;i++) {
            operatorComboBox[i]=oldO[i];
        }
        
        
        // construct new object to the array with the positon of length -2 since it is right before the last one
        variableComboBox[variableComboBox.length-1] = new JComboBox();
        variableMemberComboBox[variableMemberComboBox.length-1]=new JComboBox();
        isLabel[isLabel.length-1]=new JLabel("IS");
        operatorComboBox[operatorComboBox.length-1]=new JComboBox();
        
        
        
        // reposition all the components in the variablePanel
        this.variablePanel.removeAll();
        this.iniVariablePanel(this.variableComboBox.length);
        
        this.setPanel(this.fuzzyRuleModel);
        
        //this.setPanel();
        // Add your handling code here:
    }
    
    private void deleteVariableButtonActionPerformed(java.awt.event.ActionEvent evt) {
        if(variableComboBox.length >= 2){
            // construct new arrays with old length increasing one
            variableComboBox=new JComboBox[variableComboBox.length-1];
            variableMemberComboBox=new JComboBox[variableMemberComboBox.length-1];
            isLabel=new JLabel[isLabel.length-1];
            operatorComboBox=new JComboBox[operatorComboBox.length-1];
            
            // assign old to new without the final one
            for(int i=0;i<isLabel.length;i++) {
                variableComboBox[i] = new JComboBox();
                variableMemberComboBox[i]=new JComboBox();
                isLabel[i]=new JLabel("IS");
            }
            // assign old to new without the final one
            for(int i=0;i<operatorComboBox.length;i++) {
                operatorComboBox[i]=new JComboBox();
            }
            
            
            
            
            
            // reposition all the components in the variablePanel
            this.variablePanel.removeAll();
            this.iniVariablePanel(this.variableComboBox.length);
            this.setPanel(this.fuzzyRuleModel);
            
            
        }
    }
    
    /**
     *
     * @param fuzzyRuleModel
     * @param out
     * @throws IOException
     */    
    public static void save(FuzzyRuleModel fuzzyRuleModel, FileOutputStream out) throws IOException {
        
        ObjectOutputStream objectStream=new ObjectOutputStream(out);
        objectStream.writeObject(fuzzyRuleModel);
        objectStream.close();
        
    }
    
    /**
     *
     * @param in
     * @throws IOException
     * @throws ClassNotFoundException
     * @return
     */    
    public static Object load(FileInputStream in) throws IOException, ClassNotFoundException {
        ObjectInputStream objectStream=new ObjectInputStream(in);
        
        return objectStream.readObject();
        
        
    }
    
    // Variables declaration - do not modify
    private JPanel variablePanel;
    private JScrollPane variableScrollPane;
    
    private JButton removeRuleButton;
    private JButton ruleAdjustButton;
    private JButton setNewRuleButton;
    private JButton addVariableButton;
    private JButton deleteVariableButton;
    private JTextField confidenceTextField;
    private JLabel jLabel4;
    private JLabel jLabel3;
    private JLabel jLabel2;
    private JLabel jLabel1;
    private JLabel jLabel5;
    private JComboBox ruleCombo;
    private JComboBox outVariableComboBox;
    
    private FuzzyRuleModel fuzzyRuleModel=null;
    
    
    private JLabel[] isLabel=new JLabel[4];
    private JComboBox[] variableComboBox=new JComboBox[4];
    private JComboBox[] variableMemberComboBox=new JComboBox[4];
    private JComboBox[] operatorComboBox=new JComboBox[3];
}