//HuffmanCode.java
/*
	程序說明：對圖像進行Huffman編碼
	
	程序修改:
	
	最后修改時間：2003-12
	
	作者：haibin
*/

import java.awt.*;
import java.awt.event.*;
import java.awt.image.*;
import javax.swing.*;

public class HuffmanCode extends Frame {
	Image im,tmp;
	int iw,ih;
	int[] pixels;
	boolean flag=false;
	
	//構造方法
	public HuffmanCode(){
		this.setTitle("Huffman編碼");
		Panel pdown;
		Button load,run,quit;
		//添加窗口監(jiān)聽事件
		addWindowListener(new WindowAdapter(){
			public void windowClosing(WindowEvent e){
				System.exit(0);
			}
		});
		
		pdown = new Panel();
		pdown.setBackground(Color.lightGray);
		
		load=new Button("裝載圖像");
		run = new Button("編碼(Huffman)");
		quit=new Button("退出");
		
		this.add(pdown,BorderLayout.SOUTH);
		
		pdown.add(load);
		pdown.add(run);
		pdown.add(quit);
		
		load.addActionListener(new ActionListener(){
			public void actionPerformed(ActionEvent e){
				jLoad_ActionPerformed(e);
			}
		});
		
		run.addActionListener(new ActionListener(){
			public void actionPerformed(ActionEvent e){
				jRun_ActionPerformed(e);
			}
		});
		
		quit.addActionListener(new ActionListener(){
			public void actionPerformed(ActionEvent e){
				jQuit_ActionPerformed(e);
			}
		});
	}
		
	public void jLoad_ActionPerformed(ActionEvent e){
		//利用MediaTracker跟蹤圖像的加載
		MediaTracker tracker = new MediaTracker(this);
		im=Toolkit.getDefaultToolkit().getImage("Miss.jpg");
		tracker.addImage(im,0);
	
		//等待圖像的完全加載
		try{
		tracker.waitForID(0);
 		}catch(InterruptedException e2){ e2.printStackTrace();}
 	
 		//獲取圖像的寬度iw和高度ih
		iw=im.getWidth(this);
		ih=im.getHeight(this);
		pixels=new int[iw*ih];
		
		try{
		PixelGrabber pg=new PixelGrabber(im,0,0,iw,ih,pixels,0,iw);
		pg.grabPixels();
		}catch (InterruptedException e3) {
			e3.printStackTrace();
		}
	
		//將數組中的象素產生一個圖像
		ImageProducer ip=new MemoryImageSource(iw,ih,pixels,0,iw);
		tmp=createImage(ip);
		flag=true;
		repaint();
	}
		
	public  void jRun_ActionPerformed(ActionEvent e){
		if(flag){
			
		//保存圖像的灰度
		int grey[]=new int [iw*ih];
		//獲得圖像的灰度值
		for(int i=0;i<iw*ih;i++)
		{
			grey[i]=pixels[i]&0xff;
		}	
		//對圖像進行Huffman編碼
		Huffman huffman=new Huffman(grey,iw,ih);
		System.out.println("iw:"+iw+"ih:"+ih);
		//h.test();
		huffman.huff();
		
		//圖像熵
		float entropy;
		
		//平均碼子長度
		float avgCode;
		
		//編碼效率
		float efficiency;
		
		//出現(xiàn)頻率
		float freq[]=new float[256];
		
		//Huffman編碼
		String sCode[] =new String[256];
		
		entropy=huffman.getEntropy();
		avgCode=huffman.getAvgCode();
		efficiency=huffman.getEfficiency();
		
		//出現(xiàn)頻率
		freq=huffman.getFreq();
		
		//Huffman編碼
		sCode=huffman.getCode();
		
		
		//調用HuffmanShow.java進行顯示
		HuffmanShow hs=new HuffmanShow(entropy,avgCode,efficiency);
		hs.setData(freq,sCode);
		hs.showTable();
		hs.show();

		repaint();
		}else{
	 	JOptionPane.showMessageDialog(null,"請先打開一幅圖片!",
               	 "Alert",JOptionPane.WARNING_MESSAGE);
               	}
	}
	
	public void jQuit_ActionPerformed(ActionEvent e)
	{
		//System.exit(0);
		JOptionPane op =new JOptionPane();
		int exit=op.showConfirmDialog(this,"你要退出嗎? ? ?","退出",JOptionPane.YES_NO_OPTION);
		
		if(exit==JOptionPane.YES_OPTION)
		{
			System.exit(0);
			
		}else{ }
	}
	
	//調用paint()方法，顯示圖像信息。
	public void paint(Graphics g){
		if(flag){
			g.drawImage(tmp,20,20,this);
		}else { }
	}
	
	//定義main方法，設置窗口的大小，顯示窗口
	public static void main(String[] args) {
		HuffmanCode hc = new HuffmanCode();
		hc.setSize(500,400);
		hc.show();
	} 
}