/* * Oct 25, 2002 * * Document me! */package cdt.projects.tree.wizards.treatment;import java.awt.*;import java.awt.event.ActionEvent;import java.awt.event.ActionListener;import java.util.ArrayList;import java.util.Enumeration;import java.util.Iterator;import javax.swing.*;import javax.swing.border.Border;import javax.swing.border.TitledBorder;import javax.swing.event.ListSelectionEvent;import javax.swing.event.ListSelectionListener;import cdt.gui.HTMLTextPane;import cdt.projects.Project;import cdt.projects.tree.nodes.*;import cdt.wizard.WizardFrame;import cdt.Frame1;public class TreatmentWizard1 extends WizardFrame implements ActionListener {    /** The parent frame to this wizard. */    private JFrame parent;    /** The JList containing the treatments. */    private JList treatmentList;    /** Hashtable where the key is the diffDiag node, and the value is the finalDiag node. */    private ArrayList treatments;    /** JPanel containing the treatment information. */    private JPanel treatmentPanel;    /**     * The {@link cdt.gui.HTMLTextPane HTMLTextPane} containing     * the text of the differential diagnosis feedback.     */    private HTMLTextPane treatmentText;    /** The treatment folder in the project. */    private final Hold oldTreatmentFolder;    /**     * Creates the differential diagnosis wizard.     *     * @param parent The parent JFrame, in this case typically an instance of     *      {@link cdt.Frame1 Frame1}.     */    public TreatmentWizard1(JFrame parent) {        super(parent);        this.parent = parent;        this.treatments = new ArrayList();        this.oldTreatmentFolder = (Hold)this.data.get("oldTreatmentFolder");        Enumeration enum = oldTreatmentFolder.children();        while(enum.hasMoreElements()) {            Node treatmentNode = (Node)enum.nextElement();            if(treatmentNode instanceof Item) {                this.treatments.add(treatmentNode.copy());            }        }        init();    }    /**     * Sets up the layout of this page.     */    public void customize() {        showPreviousButton(true);        showFinishButton(true);        showNextButton(false);        setTitle("Treatment Wizard");        Dimension dim;        // Sets up a JList of the treatments        Iterator iter = this.treatments.iterator();        DefaultListModel listModel = new DefaultListModel();        while(iter.hasNext()) {            String diagnosisName = ((Node)iter.next()).getName();            listModel.addElement(diagnosisName);        }        this.treatmentList = new JList(listModel);        this.treatmentList.setSelectionMode(ListSelectionModel.SINGLE_SELECTION);        this.treatmentList.addListSelectionListener(new TreatmentWizard1.TreatmentSelectionListener());        // Puts the list into a JScrollPane        dim = new Dimension(120, 150);        JScrollPane scrollPane = new JScrollPane(this.treatmentList);        scrollPane.setMaximumSize(dim);        scrollPane.setPreferredSize(dim);        scrollPane.setMinimumSize(dim);        // Puts the diagnosis list's scrollpane in a JPanel with the [ + ] and [ - ] buttons        JPanel leftPanel = new JPanel();        leftPanel.setLayout(new BoxLayout(leftPanel, BoxLayout.Y_AXIS));        leftPanel.add(scrollPane);        leftPanel.add(Box.createVerticalStrut(5));        JButton plus = new JButton("Add Treatment");        plus.addActionListener(this);        dim = new Dimension(135, 30);        plus.setMaximumSize(dim);        plus.setMinimumSize(dim);        plus.setPreferredSize(dim);        JPanel temp = new JPanel();        temp.add(plus);        leftPanel.add(temp);        JButton minus = new JButton("Remove Treatment");        minus.addActionListener(this);        temp = new JPanel();        dim = new Dimension(135, 30);        minus.setMaximumSize(dim);        minus.setMinimumSize(dim);        minus.setPreferredSize(dim);        temp.add(minus);        leftPanel.add(temp);        // Creates the HTMLTextPane for the Differential Diagnosis text        this.treatmentText = new HTMLTextPane(this.parent, "");        this.treatmentText.setEditable(false);        this.treatmentText.setBackground(Color.lightGray);        dim = new Dimension(360, 120);        JScrollPane diffDiagScrollPane = new JScrollPane(this.treatmentText);        diffDiagScrollPane.setPreferredSize(dim);        diffDiagScrollPane.setMinimumSize(dim);        diffDiagScrollPane.setMaximumSize(dim);        Border b = BorderFactory.createLineBorder(Color.darkGray);        b = BorderFactory.createTitledBorder(b, "Treatment Text");        ((TitledBorder)b).setTitleJustification(TitledBorder.CENTER);        diffDiagScrollPane.setBorder(b);        // Creates the panel containing the check box, and the two HTMLTextPanes        this.treatmentPanel = new JPanel();        this.treatmentPanel.setLayout(new BoxLayout(this.treatmentPanel, BoxLayout.Y_AXIS));        this.treatmentPanel.add(diffDiagScrollPane);        setupBorder(this.treatmentPanel, "No Treatment Selected");        // Organizes the main panel        JPanel mainPanel = new JPanel();        mainPanel.setLayout(new BoxLayout(mainPanel, BoxLayout.X_AXIS));        mainPanel.add(leftPanel);        mainPanel.add(Box.createHorizontalStrut(5));        mainPanel.add(new JSeparator(JSeparator.VERTICAL));        mainPanel.add(Box.createHorizontalStrut(5));        mainPanel.add(this.treatmentPanel);        centerPanel.add(mainPanel);        pack();    }    /**     * Sets up an HTMLEditorKit with the appropriate border.     *     * @param c The component to put the border onto.     * @param title The title for the border.     */    private void setupBorder(JComponent c, String title) {        TitledBorder b = BorderFactory.createTitledBorder(title);        b.setTitleJustification(TitledBorder.CENTER);        c.setBorder(b);    }    /**     * Moves to the next page in the wizard.     */    public void nextAction() {    }    /**     * A button has been pressed.     */    public void actionPerformed(ActionEvent e) {        // Adds a diagnosis        if(e.getActionCommand().equals("Add Treatment")) {            String name = JOptionPane.showInputDialog(this, "Please enter name of a new treatment.");            if(null != name && false == name.equals("")) {                // Makes sure this name is not a duplicate                Iterator iter = treatments.iterator();                while(iter.hasNext()) {                    Node n = (Node)iter.next();                    if(n.getName().equals(name)) {                        cdt.ErrorD.ErrorDlg.ErrorMsg("There is already a treatment with this name.");                        return;                    }                }                // Disables any ChoiceSelectionListeners on the diagnosisList to prevent                // unwanted behavior                ListSelectionListener listListeners[] = treatmentList.getListSelectionListeners();                for(int i = 0; i < listListeners.length; ++i) {                    if(listListeners[i] instanceof TreatmentWizard1.TreatmentSelectionListener) {                        ((TreatmentWizard1.TreatmentSelectionListener)listListeners[i]).setActive(false);                    }                }                DefaultListModel listModel = (DefaultListModel)treatmentList.getModel();                listModel.addElement(name);                treatmentList.setModel(listModel);                // Adds a new diagnosis Item nodes to the treatments HashMap                treatments.add(new Item(name));                // Re-enables any ChoiceSelectionListeners that were disabled during the                // execution of this method.                for(int i = 0; i < listListeners.length; ++i) {                    if(listListeners[i] instanceof TreatmentWizard1.TreatmentSelectionListener) {                        ((TreatmentWizard1.TreatmentSelectionListener)listListeners[i]).setActive(true);                    }                }                treatmentList.setSelectedIndex(listModel.getSize()-1);            }        }        // Removes a diagnosis        else if(e.getActionCommand().equals("Remove Treatment")) {            final int selectedIndex = this.treatmentList.getSelectedIndex();            DefaultListModel listModel = new DefaultListModel();            ListModel oldList = treatmentList.getModel();            for(int i = 0; i < oldList.getSize(); ++i) {                if(i != selectedIndex) {                    listModel.addElement(oldList.getElementAt(i));                } else {                    // Removes the treatment from the ArrayList                    Iterator iter = treatments.iterator();                    while(iter.hasNext()) {                        Node n = (Node)iter.next();                        if(n.getName().equals(oldList.getElementAt(i))) {                            iter.remove();                            break;                        }                    }                }            }            treatmentList.setModel(listModel);            if(selectedIndex < listModel.size()) {                treatmentList.setSelectedIndex(selectedIndex);            } else {                treatmentText.setText("");                treatmentText.setEditable(false);                treatmentText.setBackground(Color.lightGray);                setupBorder(treatmentPanel, "No Treatment Selected");            }        }    }    /**     * Saves whatever data is in the currently open {@link cdt.gui.HTMLTextPane HTMLTextPane's}.     */    private final void saveTreatment() {        final int selectedChoice = this.treatmentList.getSelectedIndex();        if(-1 != selectedChoice) {            final String selectedTreatment = (String)treatmentList.getModel().getElementAt(selectedChoice);            // Saves the text for the treatment.            Iterator iter = treatments.iterator();            while(iter.hasNext()) {                Node n = (Node)iter.next();                if(n.getName().equals(selectedTreatment)) {                    n.setData(treatmentText.getText());                    break;                }            }        }    }    /**     * Creates the treatments and closes the wizard.     */    public void finishAction() {        saveTreatment();        if(null != oldTreatmentFolder) {            // Clears out the old treatment folder in preparation of inserting the treatments from the wizard            Enumeration enum = oldTreatmentFolder.children();            while(enum.hasMoreElements()) {                Node n = (Node)enum.nextElement();                if(n instanceof Item) {                    n.removeNode();                }            }            // Adds the new treatments into the treatments folder            Iterator iter = treatments.iterator();            while(iter.hasNext()) {                Node n = (Node)iter.next();                n.setFile(Node.foldername(n.getName()) + ".html");                oldTreatmentFolder.addNodeInto(n);            }            // Changes the wizard's name if necessary            Node n = Node.findChildNode(oldTreatmentFolder, "Create treatments");            if(null != n) {                n.setName("Modify treatments");            }        }        Project.currentProject.getTree().updateUI();        // Cleans up everything and closes the wizard        super.finishAction();    }    /**     * Changes the text in the editor according to what choice is selected.     */    private class TreatmentSelectionListener implements ListSelectionListener {        /** The currently 'selected' choice's index. */        private int currentSelection;        /** Whether this listener is currently listening. */        private boolean isActive;        /**         * Creates a ChoiceSelectionListener that will affect a JTextComponent.         */        public TreatmentSelectionListener() {            this.isActive = true;            this.currentSelection = -1;        }        public boolean isActive() {            return this.isActive;        }        public void setActive(boolean isActive) {            this.isActive = isActive;        }        /**         * Fired when an item in the registered list is selected.         *         * @param e The ListSelectionEvent that describes the selection.         */        public synchronized void valueChanged(ListSelectionEvent e) {            if(false == isActive() || e.getValueIsAdjusting()) { return; }            final JList treatmentList = (JList)e.getSource();            final int newIndex = treatmentList.getSelectedIndex();            if(newIndex == this.currentSelection ) {                // The selection hasn't changed, so don't do anything                return;            } else if(newIndex == -1) {                this.currentSelection = -1;                return;            }            final String nextSelectedTreatment = (String)treatmentList.getModel().getElementAt(newIndex);            // Saves the text from the previously selected answer            if(-1 != this.currentSelection) {                String selectedTreatment = (String)treatmentList.getModel().getElementAt(this.currentSelection);                // Saves the text for the differential diagnosis.                Iterator iter = treatments.iterator();                while(iter.hasNext()) {                    Node n = (Node)iter.next();                    if(n.getName().equals(selectedTreatment)) {                        n.setData(treatmentText.getText());                        break;                    }                }            }            // Makes sure that the answerText and the feedbackText are            // editable when an answer is selected from the list            if(false == treatmentText.isEditable()) {                treatmentText.setEditable(true);                treatmentText.setBackground(UIManager.getColor("controlLtHighlight"));            }            // Switches the text in the answerText and feedbackText to the            // text corresponding to the answer that was selected            Iterator iter = treatments.iterator();            while(iter.hasNext()) {                Node n = (Node)iter.next();                if(n.getName().equals(nextSelectedTreatment)) {                    treatmentText.setText(n.getData());                    break;                }            }            setupBorder(treatmentPanel, nextSelectedTreatment);            this.currentSelection = newIndex;        }    }}