jpexs-decompiler/trunk/src/com/jpexs/asdec/abc/gui/ABCPanel.java

/*
 *  Copyright (C) 2010-2013 JPEXS
 * 
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 * 
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 * 
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.jpexs.asdec.abc.gui;

import com.jpexs.asdec.Main;
import com.jpexs.asdec.abc.ABC;
import com.jpexs.asdec.abc.gui.tablemodels.*;
import com.jpexs.asdec.gui.View;
import com.jpexs.asdec.tags.DoABCTag;
import java.awt.BorderLayout;
import java.awt.Component;
import java.awt.event.*;
import java.util.ArrayList;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.swing.*;
import javax.swing.border.BevelBorder;
import javax.swing.event.DocumentEvent;
import javax.swing.event.DocumentListener;
import javax.swing.table.*;
import jsyntaxpane.DefaultSyntaxKit;

public class ABCPanel extends JPanel implements ItemListener, ActionListener {

   public TraitsList navigator;
   public ClassesListTree classTree;
   public ABC abc;
   public List<DoABCTag> list;
   public JComboBox abcComboBox;
   public int listIndex = 0;
   public DecompiledEditorPane decompiledTextArea;
   public JScrollPane decompiledScrollPane;
   public JSplitPane splitPaneDecompiledVSDetail;
   public JSplitPane splitPaneTreeVSNavigator;
   public JSplitPane splitPaneTreeNavVSDecompiledDetail;
   private JTable constantTable;
   public JComboBox constantTypeList;
   public JLabel asmLabel = new JLabel("P-code source");
   public JLabel decLabel = new JLabel("ActionScript source");
   public DetailPanel detailPanel;
   public JTextField filterField = new JTextField("");

   private JTable autoResizeColWidth(JTable table, TableModel model) {
      table.setAutoResizeMode(JTable.AUTO_RESIZE_OFF);
      table.setModel(model);

      int margin = 5;

      for (int i = 0; i < table.getColumnCount(); i++) {
         int vColIndex = i;
         DefaultTableColumnModel colModel = (DefaultTableColumnModel) table.getColumnModel();
         TableColumn col = colModel.getColumn(vColIndex);
         int width;

         // Get width of column header
         TableCellRenderer renderer = col.getHeaderRenderer();

         if (renderer == null) {
            renderer = table.getTableHeader().getDefaultRenderer();
         }

         Component comp = renderer.getTableCellRendererComponent(table, col.getHeaderValue(), false, false, 0, 0);

         width = comp.getPreferredSize().width;

         // Get maximum width of column data
         for (int r = 0; r < table.getRowCount(); r++) {
            renderer = table.getCellRenderer(r, vColIndex);
            comp = renderer.getTableCellRendererComponent(table, table.getValueAt(r, vColIndex), false, false,
                    r, vColIndex);
            width = Math.max(width, comp.getPreferredSize().width);
         }

         // Add margin
         width += 2 * margin;

         // Set the width
         col.setPreferredWidth(width);
      }

      ((DefaultTableCellRenderer) table.getTableHeader().getDefaultRenderer()).setHorizontalAlignment(
              SwingConstants.LEFT);

      // table.setAutoCreateRowSorter(true);
      table.getTableHeader().setReorderingAllowed(false);

      return table;
   }

   public void setAbc(ABC abc) {
      this.abc = abc;
      updateConstList();
   }

   public void updateConstList() {
      switch (constantTypeList.getSelectedIndex()) {
         case 0:
            autoResizeColWidth(constantTable, new UIntTableModel(abc));
            break;
         case 1:
            autoResizeColWidth(constantTable, new IntTableModel(abc));
            break;
         case 2:
            autoResizeColWidth(constantTable, new DoubleTableModel(abc));
            break;
         case 3:
            autoResizeColWidth(constantTable, new DecimalTableModel(abc));
            break;
         case 4:
            autoResizeColWidth(constantTable, new StringTableModel(abc));
            break;
         case 5:
            autoResizeColWidth(constantTable, new NamespaceTableModel(abc));
            break;
         case 6:
            autoResizeColWidth(constantTable, new NamespaceSetTableModel(abc));
            break;
         case 7:
            autoResizeColWidth(constantTable, new MultinameTableModel(abc));
            break;
      }
      //DefaultTableColumnModel colModel  = (DefaultTableColumnModel) constantTable.getColumnModel();
      //colModel.getColumn(0).setMaxWidth(50);
   }

   public void switchAbc(int index) {
      listIndex = index;
      if (index == -1) {
         classTree.setDoABCTags(list);
      } else {
         List<DoABCTag> oneList = new ArrayList<DoABCTag>();
         oneList.add(list.get(index));
         this.abc = list.get(index).abc;
         classTree.setDoABCTags(oneList);
      }
      updateConstList();
   }

   public void initSplits() {
      splitPaneTreeVSNavigator.setDividerLocation(splitPaneTreeVSNavigator.getHeight() / 2);
      try {
         Thread.sleep(100);
      } catch (InterruptedException ex) {
         Logger.getLogger(ABCPanel.class.getName()).log(Level.SEVERE, null, ex);
      }
      splitPaneTreeNavVSDecompiledDetail.setDividerLocation(splitPaneTreeNavVSDecompiledDetail.getWidth() * 1 / 3);
      try {
         Thread.sleep(100);
      } catch (InterruptedException ex) {
         Logger.getLogger(ABCPanel.class.getName()).log(Level.SEVERE, null, ex);
      }
      splitPaneDecompiledVSDetail.setDividerLocation(splitPaneDecompiledVSDetail.getWidth() * 1 / 2);

   }

   public ABCPanel(List<DoABCTag> list) {


      DefaultSyntaxKit.initKit();

      this.list = list;
      if (list.size() > 0) {
         this.abc = list.get(listIndex).abc;
      }
      setLayout(new BorderLayout());





      
      decompiledTextArea = new DecompiledEditorPane(this);
      decompiledScrollPane = new JScrollPane(decompiledTextArea);      
      detailPanel = new DetailPanel(this);
      JPanel panB = new JPanel();
      panB.setLayout(new BorderLayout());
      panB.add(decompiledScrollPane, BorderLayout.CENTER);
      panB.add(decLabel, BorderLayout.NORTH);
      decLabel.setHorizontalAlignment(SwingConstants.CENTER);
      decLabel.setBorder(new BevelBorder(BevelBorder.RAISED));
      splitPaneDecompiledVSDetail = new JSplitPane(JSplitPane.HORIZONTAL_SPLIT,
              panB, detailPanel);
      splitPaneDecompiledVSDetail.setResizeWeight(0.5);
      splitPaneDecompiledVSDetail.setContinuousLayout(true);
      decompiledTextArea.setContentType("text/actionscript");

      JPanel pan2 = new JPanel();
      pan2.setLayout(new BorderLayout());
      pan2.add((abcComboBox = new JComboBox(new ABCComboBoxModel(list))), BorderLayout.NORTH);

      navigator = new TraitsList(this);
      navigator.setABC(list, abc);


      JPanel navPanel = new JPanel(new BorderLayout());
      JLabel traitsLabel = new JLabel("Traits");
      navPanel.add(traitsLabel, BorderLayout.NORTH);

      traitsLabel.setBorder(new BevelBorder(BevelBorder.RAISED));
      navPanel.add(new JScrollPane(navigator), BorderLayout.CENTER);

      Main.startWork("Building script tree...");

      filterField.setActionCommand("FILTERSCRIPT");
      filterField.addActionListener(this);


      filterField.getDocument().addDocumentListener(new DocumentListener() {
         @Override
         public void changedUpdate(DocumentEvent e) {
            warn();
         }

         @Override
         public void removeUpdate(DocumentEvent e) {
            warn();
         }

         @Override
         public void insertUpdate(DocumentEvent e) {
            warn();
         }

         public void warn() {
            doFilter();
         }
      });

      JPanel treePanel = new JPanel();
      treePanel.setLayout(new BorderLayout());
      treePanel.add(new JScrollPane(classTree = new ClassesListTree(list,this)), BorderLayout.CENTER);
      JPanel searchPanel = new JPanel();
      searchPanel.setLayout(new BorderLayout());
      searchPanel.add(filterField, BorderLayout.CENTER);
      JLabel picLabel = new JLabel(new ImageIcon(View.loadImage("com/jpexs/asdec/gui/graphics/search.png")));
      searchPanel.add(picLabel, BorderLayout.EAST);
      treePanel.add(searchPanel, BorderLayout.NORTH);

      splitPaneTreeVSNavigator = new JSplitPane(JSplitPane.VERTICAL_SPLIT,
              treePanel,
              navPanel);
      splitPaneTreeVSNavigator.setResizeWeight(0.5);
      splitPaneTreeVSNavigator.setContinuousLayout(true);
      Main.startWork("Creating window...");
      JTabbedPane tabbedPane = new JTabbedPane();
      tabbedPane.addTab("Scripts", splitPaneTreeVSNavigator);
      //tabbedPane.setTabPlacement(JTabbedPane.BOTTOM);

      pan2.add(tabbedPane, BorderLayout.CENTER);
      abcComboBox.addItemListener(this);


      splitPaneTreeNavVSDecompiledDetail = new JSplitPane(JSplitPane.HORIZONTAL_SPLIT,
              pan2,
              splitPaneDecompiledVSDetail);
      splitPaneTreeNavVSDecompiledDetail.setResizeWeight(0);
      splitPaneTreeNavVSDecompiledDetail.setContinuousLayout(true);
      //pan2.setPreferredSize(new Dimension(300, 200));





      add(splitPaneTreeNavVSDecompiledDetail, BorderLayout.CENTER);


      JPanel panConstants = new JPanel();
      panConstants.setLayout(new BorderLayout());
      constantTypeList = new JComboBox(new String[]{"UINT", "INT", "DOUBLE", "DECIMAL", "STRING", "NAMESPACE", "NAMESPACESET", "MULTINAME"});
      constantTable = new JTable();
      if (abc != null) {
         autoResizeColWidth(constantTable, new UIntTableModel(abc));
      }
      constantTable.setAutoCreateRowSorter(true);

      final List<DoABCTag> inlist = list;
      final ABCPanel t=this;
      constantTable.addMouseListener(new MouseAdapter() {
         @Override
         public void mouseClicked(MouseEvent e) {
            if (e.getClickCount() == 2) {
               if (constantTypeList.getSelectedIndex() == 7) { //MULTINAME
                  int rowIndex = constantTable.getSelectedRow();
                  if (rowIndex == -1) {
                     return;
                  }
                  int multinameIndex = constantTable.convertRowIndexToModel(rowIndex);
                  if (multinameIndex > 0) {
                     UsageFrame usageFrame = new UsageFrame(inlist, abc, multinameIndex,t);
                     usageFrame.setVisible(true);
                  }
               }
            }
         }
      });
      constantTypeList.addItemListener(this);
      panConstants.add(constantTypeList, BorderLayout.NORTH);
      panConstants.add(new JScrollPane(constantTable), BorderLayout.CENTER);
      tabbedPane.addTab("Constants", panConstants);
   }

   public void doFilter() {
      classTree.applyFilter(filterField.getText());
   }

   public void reload() {
      switchAbc(listIndex - 1);
   }

   public void itemStateChanged(ItemEvent e) {
      if (e.getSource() == abcComboBox) {
         int index = ((JComboBox) e.getSource()).getSelectedIndex();
         if (index == -1) {
            return;
         }
         switchAbc(index - 1);
      }
      if (e.getSource() == constantTypeList) {
         int index = ((JComboBox) e.getSource()).getSelectedIndex();
         if (index == -1) {
            return;
         }
         updateConstList();
      }
   }

   public void display() {
      setVisible(true);
   }

   @Override
   public void actionPerformed(ActionEvent e) {
      if (e.getActionCommand().equals("FILTERSCRIPT")) {
         doFilter();
      }
   }
}