/*
 *  Copyright (C) 2010-2014 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.decompiler.flash.gui;

import com.jpexs.decompiler.flash.SWF;
import com.jpexs.decompiler.flash.configuration.Configuration;
import com.jpexs.decompiler.flash.exporters.commonshape.Matrix;
import com.jpexs.decompiler.flash.tags.Tag;
import com.jpexs.decompiler.flash.tags.base.BoundedTag;
import com.jpexs.decompiler.flash.tags.base.CharacterTag;
import com.jpexs.decompiler.flash.tags.base.DrawableTag;
import com.jpexs.decompiler.flash.tags.base.ImageTag;
import com.jpexs.decompiler.flash.treeitems.FrameNodeItem;
import com.jpexs.decompiler.flash.treeitems.TreeItem;
import com.jpexs.decompiler.flash.types.ColorTransform;
import com.jpexs.decompiler.flash.types.RECT;
import com.jpexs.helpers.SerializableImage;
import java.awt.AlphaComposite;
import java.awt.BorderLayout;
import java.awt.Color;
import java.awt.Component;
import java.awt.Dimension;
import java.awt.Graphics;
import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.Rectangle;
import java.awt.event.ActionEvent;
import java.awt.event.MouseEvent;
import java.awt.event.MouseListener;
import java.awt.geom.AffineTransform;
import java.util.concurrent.Callable;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import javax.swing.AbstractAction;
import javax.swing.BorderFactory;
import javax.swing.JLabel;
import javax.swing.JMenuItem;
import javax.swing.JPanel;
import javax.swing.JPopupMenu;

/**
 *
 * @author JPEXS
 */
public class PreviewImage extends JPanel {

    private static final ExecutorService executor;
    private static final int PREVIEW_SIZE = 150;
    private static final int BORDER_SIZE = 5;
    private Image image;
    private boolean rendering;
    private final MainPanel mainPanel;
    private final TreeItem treeItem;

    static {
        executor = Executors.newFixedThreadPool(Configuration.parallelSpeedUp.get() ? Configuration.parallelThreadCount.get() : 1);
    }

    /**
     *
     * @param mainPanel
     * @param treeItem
     */
    public PreviewImage(final MainPanel mainPanel, final TreeItem treeItem) {
        this.mainPanel = mainPanel;
        this.treeItem = treeItem;
        Dimension dim = new Dimension(PREVIEW_SIZE + 2 * BORDER_SIZE, PREVIEW_SIZE + 2 * BORDER_SIZE);
        setMinimumSize(dim);
        setMaximumSize(dim);
        setPreferredSize(dim);
        setSize(dim);
        setLayout(null);
        setBorder(BorderFactory.createLineBorder(Color.black));

        if (treeItem instanceof Tag) {
            JPopupMenu contextMenu = new JPopupMenu();
            final JMenuItem removeMenuItem = new JMenuItem(mainPanel.translate("contextmenu.remove"));
            removeMenuItem.addActionListener(new AbstractAction() {

                @Override
                public void actionPerformed(ActionEvent e) {
                    Tag tag = (Tag) treeItem; 
                    tag.getSwf().removeTag(tag, false);
                    mainPanel.refreshTree();
                }
            });
            contextMenu.add(removeMenuItem);
            final JMenuItem removeWithDependenciesMenuItem = new JMenuItem(mainPanel.translate("contextmenu.removeWithDependencies"));
            removeWithDependenciesMenuItem.addActionListener(new AbstractAction() {

                @Override
                public void actionPerformed(ActionEvent e) {
                    Tag tag = (Tag) treeItem; 
                    tag.getSwf().removeTag(tag, true);
                    mainPanel.refreshTree();
                }
            });
            contextMenu.add(removeWithDependenciesMenuItem);
            this.setComponentPopupMenu(contextMenu);
        }

        this.addMouseListener(new MouseListener() {

            @Override
            public void mouseClicked(MouseEvent e) {

                if (e.getButton() == MouseEvent.BUTTON1 && e.getClickCount() >= 2) {
                    mainPanel.setTreeItem(treeItem);
                }
            }

            @Override
            public void mousePressed(MouseEvent e) {
            }

            @Override
            public void mouseReleased(MouseEvent e) {
            }

            @Override
            public void mouseEntered(MouseEvent e) {
            }

            @Override
            public void mouseExited(MouseEvent e) {
            }
        });
    }

    private synchronized void renderImageTask(final TreeItem treeItem) {
        if (rendering) {
            return;
        }
        rendering = true;
        executor.submit(new Callable<Void>() {

            @Override
            public Void call() throws Exception {
                image = renderImage(treeItem.getSwf(), treeItem);
                View.execInEventDispatch(new Runnable() {

                    @Override
                    public void run() {
                        revalidate();
                        repaint();
                    }
                });
                return null;
            }

        });

    }

    private Image renderImage(SWF swf, TreeItem treeItem) {

        double scale = 1;
        int width = 0;
        int height = 0;
        SerializableImage imgSrc = null;
        Matrix m = new Matrix();
        if (treeItem instanceof FrameNodeItem) {
            FrameNodeItem fn = (FrameNodeItem) treeItem;
            RECT rect = swf.displayRect;
            imgSrc = SWF.frameToImageGet(swf.getTimeline(), fn.getFrame() - 1, 0, null, 0, rect, new Matrix(), new ColorTransform(), null, true);
            width = (imgSrc.getWidth());
            height = (imgSrc.getHeight());
        } else if (treeItem instanceof ImageTag) {
            imgSrc = ((ImageTag) treeItem).getImage();
            width = (imgSrc.getWidth());
            height = (imgSrc.getHeight());
        } else if (treeItem instanceof BoundedTag) {
            BoundedTag boundedTag = (BoundedTag) treeItem;
            RECT rect = boundedTag.getRect();
            width = (int) (rect.getWidth() / SWF.unitDivisor);
            height = (int) (rect.getHeight() / SWF.unitDivisor);
            m.translate(-rect.Xmin, -rect.Ymin);
        }

        int w1 = width;
        int h1 = height;
        int w2 = PREVIEW_SIZE;
        int h2 = PREVIEW_SIZE;

        int w;
        int h = h1 * w2 / w1;
        if (h > h2) {
            w = w1 * h2 / h1;
            h = h2;
        } else {
            w = w2;
        }

        scale = (double) w / (double) w1;
        if (w1 <= w2 && h1 <= h2) {
            scale = 1;
        }

        m = m.preConcatenate(Matrix.getScaleInstance(scale));
        width = (int) (scale * width);
        height = (int) (scale * height);
        if (width == 0 || height == 0) {
            return null;
        }

        SerializableImage image = new SerializableImage(width, height, SerializableImage.TYPE_INT_ARGB);
        image.fillTransparent();
        if (imgSrc == null) {
            DrawableTag drawable = (DrawableTag) treeItem;
            drawable.toImage(0, 0, 0, null, 0, image, m, new ColorTransform());
        } else {
            Graphics2D g = (Graphics2D) image.getGraphics();
            g.setTransform(m.toTransform());
            g.drawImage(imgSrc.getBufferedImage(), 0, 0, null);
        }
        return image.getBufferedImage();
    }

    public static Component createFolderPreviewImage(MainPanel mainPanel, TreeItem treeItem) {
        JPanel pan = new JPanel(new BorderLayout());
        PreviewImage imagePanel = new PreviewImage(mainPanel, treeItem);
        pan.add(imagePanel, BorderLayout.CENTER);
        String s;
        if (treeItem instanceof Tag) {
            s = ((Tag) treeItem).getTagName();
            if (treeItem instanceof CharacterTag) {
                s = s + " (" + ((CharacterTag) treeItem).getCharacterId() + ")";
            }
        } else {
            s = treeItem.toString();
        }
        JLabel lab = new JLabel(s);
        lab.setFont(lab.getFont().deriveFont(AffineTransform.getScaleInstance(0.8, 0.8)));
        pan.add(lab, BorderLayout.SOUTH);
        return pan;
    }

    @Override
    protected void paintComponent(Graphics g) {
        Graphics2D g2d = (Graphics2D) g;
        g2d.setPaint(View.transparentPaint);
        g2d.fill(new Rectangle(0, 0, getWidth(), getHeight()));
        g2d.setComposite(AlphaComposite.SrcOver);
        g2d.setPaint(View.swfBackgroundColor);
        g2d.fill(new Rectangle(0, 0, getWidth(), getHeight()));
        if (image != null) {
            int x = (getWidth() / 2) - (image.getWidth(this) / 2);
            int y = (getHeight() / 2) - (image.getHeight(this) / 2);
            g.drawImage(image, x, y, null);
        } else {
            renderImageTask(treeItem);
        }
    }
}