jpexs-decompiler/trunk/src/com/jpexs/asdec/action/gui/ActionPanel.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.action.gui;

import com.jpexs.asdec.Main;
import com.jpexs.asdec.SWF;
import com.jpexs.asdec.action.TagNode;
import com.jpexs.asdec.action.parser.ASMParser;
import com.jpexs.asdec.action.parser.ParseException;
import com.jpexs.asdec.helpers.Highlighting;
import com.jpexs.asdec.tags.Tag;
import com.jpexs.asdec.tags.base.ASMSource;
import java.awt.BorderLayout;
import java.awt.FlowLayout;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import javax.swing.*;
import javax.swing.border.BevelBorder;
import javax.swing.event.TreeSelectionEvent;
import javax.swing.event.TreeSelectionListener;
import javax.swing.tree.DefaultTreeCellRenderer;
import javax.swing.tree.TreePath;
import jsyntaxpane.DefaultSyntaxKit;

public class ActionPanel extends JPanel implements TreeSelectionListener, ActionListener {

   public JTree tagTree;
   public JEditorPane editor;
   public JEditorPane decompiledEditor;
   public List<Tag> list;
   public JSplitPane splitPane;
   public JSplitPane splitPane2;
   public JButton saveButton = new JButton("Save");
   public JButton saveHexButton = new JButton("Save hex");
   public JButton loadHexButton = new JButton("Load hex");
   public JLabel asmLabel = new JLabel("P-code source (editable)");
   public JLabel decLabel = new JLabel("ActionScript source");

   public ActionPanel(List<Tag> list) {
      this.list = list;
      DefaultSyntaxKit.initKit();
      editor = new JEditorPane();
      decompiledEditor = new JEditorPane();
      tagTree = new JTree(new TagTreeModel(list));

      DefaultTreeCellRenderer treeRenderer = new DefaultTreeCellRenderer();
      ClassLoader cldr = this.getClass().getClassLoader();
      java.net.URL imageURL = cldr.getResource("com/jpexs/asdec/gui/graphics/as16.png");
      ImageIcon leafIcon = new ImageIcon(imageURL);
      treeRenderer.setLeafIcon(leafIcon);
      tagTree.setCellRenderer(treeRenderer);

      JPanel panB = new JPanel();
      panB.setLayout(new BorderLayout());
      asmLabel.setHorizontalAlignment(SwingConstants.CENTER);
      asmLabel.setBorder(new BevelBorder(BevelBorder.RAISED));
      panB.add(asmLabel, BorderLayout.NORTH);
      panB.add(new JScrollPane(editor), BorderLayout.CENTER);

      JPanel buttonsPan = new JPanel();
      buttonsPan.setLayout(new FlowLayout());
      buttonsPan.add(saveButton);
      //buttonsPan.add(saveHexButton);
      //buttonsPan.add(loadHexButton);
      panB.add(buttonsPan, BorderLayout.SOUTH);

      saveHexButton.addActionListener(this);
      saveHexButton.setActionCommand("SAVEHEXACTION");
      loadHexButton.addActionListener(this);
      loadHexButton.setActionCommand("LOADHEXACTION");
      saveButton.addActionListener(this);
      saveButton.setActionCommand("SAVEACTION");

      JPanel panA = new JPanel();
      panA.setLayout(new BorderLayout());
      panA.add(new JScrollPane(decompiledEditor), BorderLayout.CENTER);
      panA.add(decLabel, BorderLayout.NORTH);
      decLabel.setHorizontalAlignment(SwingConstants.CENTER);
      decLabel.setBorder(new BevelBorder(BevelBorder.RAISED));





      setLayout(new BorderLayout());
      add(splitPane = new JSplitPane(JSplitPane.HORIZONTAL_SPLIT, new JScrollPane(tagTree), splitPane2 = new JSplitPane(JSplitPane.HORIZONTAL_SPLIT, panA, panB)), BorderLayout.CENTER);
      splitPane.setResizeWeight(0.5);
      splitPane2.setResizeWeight(0.5);
      editor.setContentType("text/flasm");
      decompiledEditor.setContentType("text/actionscript");
      tagTree.addTreeSelectionListener(this);

   }

   public void initSplits() {
      splitPane.setDividerLocation(getWidth() / 3);
      splitPane2.setDividerLocation(getWidth() / 3);
   }

   public void valueChanged(TreeSelectionEvent e) {
      if (Main.isWorking()) {
         return;
      }
      Object obj = tagTree.getLastSelectedPathComponent();
      if (obj instanceof TagNode) {
         obj = ((TagNode) obj).tag;
         if (obj instanceof ASMSource) {
            Main.startWork("Decompiling...");
            final ASMSource asm = (ASMSource) obj;
            (new Thread() {
               @Override
               public void run() {
                  editor.setText(asm.getASMSource(SWF.DEFAULT_VERSION));
                  if (Main.DO_DECOMPILE) {
                     List<com.jpexs.asdec.action.Action> as = asm.getActions(SWF.DEFAULT_VERSION);
                     com.jpexs.asdec.action.Action.setActionsAddresses(as, 0, SWF.DEFAULT_VERSION);

                     decompiledEditor.setText(Highlighting.stripHilights(com.jpexs.asdec.action.Action.actionsToSource(as, SWF.DEFAULT_VERSION)));
                  }
                  Main.stopWork();
               }
            }).start();
         }
      }
   }

   public void display() {
      setVisible(true);
      splitPane.setDividerLocation(0.5);
      splitPane2.setDividerLocation(0.5);
   }

   public List<TagNode> getSelectedNodes() {
      List<TagNode> ret = new ArrayList<TagNode>();
      TreePath tps[] = tagTree.getSelectionPaths();
      if (tps == null) {
         return ret;
      }
      for (TreePath tp : tps) {
         TagNode te = (TagNode) tp.getLastPathComponent();
         ret.add(te);
      }
      return ret;
   }

   public void actionPerformed(ActionEvent e) {
      if (e.getActionCommand().equals("SAVEACTION")) {
         TagNode ti = (TagNode) tagTree.getLastSelectedPathComponent();
         if (ti.tag instanceof ASMSource) {
            ASMSource dat = (ASMSource) ti.tag;
            try {
               dat.setActions(ASMParser.parse(new ByteArrayInputStream(editor.getText().getBytes()), SWF.DEFAULT_VERSION), SWF.DEFAULT_VERSION);
               valueChanged(null);
               JOptionPane.showMessageDialog(this, "Code successfully saved");
            } catch (IOException ex) {
            } catch (ParseException ex) {
               JOptionPane.showMessageDialog(this, "" + ex.text + " on line " + ex.line, "Error", JOptionPane.ERROR_MESSAGE);
            }
         }
      }
   }
}