downloadpeerlisttab.java

JMule是一个基于Java开发

/*
 *  JMule - Java file sharing client
 *  Copyright (C) 2007-2008 JMule team ( [e-mail] / http://jmule.org )
 *
 *  Any parts of this program derived from other projects, or contributed
 *  by third-party developers are copyrighted by their respective authors.
 *
 *  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 2
 *  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, write to the Free Software
 *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 *
 */
package org.jmule.ui.swt.maintabs.transfers;

import java.util.LinkedList;
import java.util.List;

import org.eclipse.swt.SWT;
import org.eclipse.swt.custom.CTabFolder;
import org.eclipse.swt.custom.CTabItem;
import org.eclipse.swt.events.SelectionAdapter;
import org.eclipse.swt.events.SelectionEvent;
import org.eclipse.swt.graphics.Image;
import org.eclipse.swt.layout.GridData;
import org.eclipse.swt.layout.GridLayout;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Menu;
import org.eclipse.swt.widgets.MenuItem;
import org.jmule.core.JMIterable;
import org.jmule.core.JMuleCoreFactory;
import org.jmule.core.downloadmanager.DownloadSession;
import org.jmule.core.downloadmanager.PeerDownloadStatus;
import org.jmule.core.edonkey.impl.Peer;
import org.jmule.core.uploadmanager.UploadManager;
import org.jmule.core.uploadmanager.UploadSession;
import org.jmule.countrylocator.CountryLocator;
import org.jmule.ui.FlagPack.FlagSize;
import org.jmule.ui.localizer.Localizer;
import org.jmule.ui.localizer._;
import org.jmule.ui.swt.Refreshable;
import org.jmule.ui.swt.SWTConstants;
import org.jmule.ui.swt.SWTImageRepository;
import org.jmule.ui.swt.SWTPreferences;
import org.jmule.ui.swt.common.CountryFlagPainter;
import org.jmule.ui.swt.tables.JMTable;
import org.jmule.ui.swt.tables.TableItemCountryFlag;
import org.jmule.ui.utils.PeerInfoFormatter;
import org.jmule.ui.utils.SpeedFormatter;
import org.jmule.util.Misc;

/**
 * Created on Aug 7, 2008
 * @author binary256
 * @version $Revision: 1.4 $
 * Last changed by $Author: binary256_ $ on $Date: 2008/10/16 18:20:01 $
 */
public class DownloadPeerListTab extends CTabItem implements Refreshable {

	private UploadManager upload_manager = JMuleCoreFactory.getSingleton().getUploadManager();
	
	private DownloadSession download_session;
	private DownloadPeerList peers_table;
	private List<Peer> shown_peers = new LinkedList<Peer>();
	private Menu popup_menu;
	
	public DownloadPeerListTab(CTabFolder tabFolder, DownloadSession downloadSession) {
		super(tabFolder, SWT.NONE);
		download_session = downloadSession;
		setText(_._("downloadinfowindow.tab.peerlist.title"));		
		
		Composite content = new Composite(tabFolder,SWT.NONE);
		setControl(content);
		content.setLayout(new GridLayout(1,true));
		peers_table =  new DownloadPeerList(content);
		peers_table.setLayoutData(new GridData(GridData.FILL_BOTH));
		
		popup_menu = new Menu(peers_table);
		
		MenuItem column_setup = new MenuItem(popup_menu,SWT.NONE);
		column_setup.setText(_._("downloadinfowindow.tab.peerlist.menu.column_setup"));
		column_setup.setImage(SWTImageRepository.getImage("columns_setup.png"));
		column_setup.addSelectionListener(new SelectionAdapter() {
			public void widgetSelected(SelectionEvent event) {
				peers_table.showColumnEditorWindow();
			}
		});
	}

	public void refresh() {
		if (isDisposed()) return ;
		
		JMIterable<Peer> upload_peers = null;
		UploadSession upload_session = null;
		
		if (upload_manager.hasUpload(download_session.getFileHash())) {
			upload_session = upload_manager.getUpload(download_session.getFileHash());
			upload_peers = upload_session.getPeers();
		}
		
		JMIterable<Peer> download_peers = download_session.getPeers();
		
		for(Peer peer : shown_peers) {
			boolean remove = false;
			if (!download_session.hasPeer(peer))
				if (upload_session == null)
					remove = true;
				else
					remove = !upload_session.hasPeer(peer);
			if (remove)
				peers_table.removeRow(peer);
		}

		for(Peer peer : download_peers) {
			if (!shown_peers.contains(peer)){
				shown_peers.add(peer);
				peers_table.addRow(peer);
			}
			peers_table.updateRow(peer);
		}
		if (upload_peers != null)
			for(Peer peer : upload_peers) {
				if (!shown_peers.contains(peer)){
					shown_peers.add(peer);
					peers_table.addRow(peer);
				}
				peers_table.updateRow(peer);
			}
		
	}
	
	private class DownloadPeerList extends JMTable<Peer>{		
		
		public DownloadPeerList(Composite content) {
			super(content, SWT.NONE);
			int width;
			
			width = swt_preferences.getColumnWidth(SWTConstants.DOWNLOAD_PEER_LIST_NICKNAME_COLUMN_ID);
			addColumn(SWT.LEFT, SWTConstants.DOWNLOAD_PEER_LIST_NICKNAME_COLUMN_ID,   _._("downloadinfowindow.tab.peerlist.column.nickname"), "", width);
			
			width = swt_preferences.getColumnWidth(SWTConstants.DOWNLOAD_PEER_LIST_CC_COLUMN_ID);
			addColumn(SWT.CENTER,SWTConstants.DOWNLOAD_PEER_LIST_CC_COLUMN_ID, _._("downloadinfowindow.tab.peerlist.column.country_code"), _._("downloadinfowindow.tab.peerlist.column.country_code.desc"), width);
			if (CountryLocator.getInstance().isServiceDown())
				disableColumn(SWTConstants.DOWNLOAD_PEER_LIST_CC_COLUMN_ID);
			
			width = swt_preferences.getColumnWidth(SWTConstants.DOWNLOAD_PEER_LIST_FLAG_COLUMN_ID);
			addColumn(SWT.LEFT, SWTConstants.DOWNLOAD_PEER_LIST_FLAG_COLUMN_ID, 	  	_._("downloadinfowindow.tab.peerlist.column.flag"), "", width);
			if (CountryLocator.getInstance().isServiceDown())
				disableColumn(SWTConstants.DOWNLOAD_PEER_LIST_FLAG_COLUMN_ID);
			
			width = swt_preferences.getColumnWidth(SWTConstants.DOWNLOAD_PEER_LIST_IP_COLUMN_ID);
			addColumn(SWT.LEFT, SWTConstants.DOWNLOAD_PEER_LIST_IP_COLUMN_ID, 	  	 _._("downloadinfowindow.tab.peerlist.column.address"), "", width);
			
			width = swt_preferences.getColumnWidth(SWTConstants.DOWNLOAD_PEER_LIST_DOWN_SPEED_COLUMN_ID); 
			addColumn(SWT.RIGHT, SWTConstants.DOWNLOAD_PEER_LIST_DOWN_SPEED_COLUMN_ID, _._("downloadinfowindow.tab.peerlist.column.download_speed"), "", width);
			
			width = swt_preferences.getColumnWidth(SWTConstants.DOWNLOAD_PEER_LIST_UP_SPEED_COLUMN_ID);
			addColumn(SWT.RIGHT, SWTConstants.DOWNLOAD_PEER_LIST_UP_SPEED_COLUMN_ID,   _._("downloadinfowindow.tab.peerlist.column.upload_speed"),  "", width);
			
			width = swt_preferences.getColumnWidth(SWTConstants.DOWNLOAD_PEER_LIST_SOFTWARE_COLUMN_ID); 
			addColumn(SWT.LEFT, SWTConstants.DOWNLOAD_PEER_LIST_SOFTWARE_COLUMN_ID,   _._("downloadinfowindow.tab.peerlist.column.software"), "", width);
			
			width = swt_preferences.getColumnWidth(SWTConstants.DOWNLOAD_PEER_LIST_STATUS_COLUMN_ID);
			addColumn(SWT.LEFT, SWTConstants.DOWNLOAD_PEER_LIST_STATUS_COLUMN_ID,     _._("downloadinfowindow.tab.peerlist.column.status"), "", width);

			updateColumnVisibility();
			updateColumnOrder();
		}

		
		protected int compareObjects(Peer object1, Peer object2, int columnID,
				boolean order) {
			
			if (columnID==SWTConstants.DOWNLOAD_PEER_LIST_IP_COLUMN_ID)
				return Misc.compareAllObjects(object1,object2,"getAddress",order);
			
			if (columnID==SWTConstants.DOWNLOAD_PEER_LIST_NICKNAME_COLUMN_ID)
				return Misc.compareAllObjects(object1,object2,"getNickName",order);

			if (columnID==SWTConstants.DOWNLOAD_PEER_LIST_SOFTWARE_COLUMN_ID)
				return Misc.compareAllObjects(object1,object2,"getClientSoftware",order);
			
			if (columnID==SWTConstants.DOWNLOAD_PEER_LIST_DOWN_SPEED_COLUMN_ID)
				return Misc.compareAllObjects(object1,object2,"getDownloadSpeed",order);

			if (columnID==SWTConstants.DOWNLOAD_PEER_LIST_UP_SPEED_COLUMN_ID)
				return Misc.compareAllObjects(object1,object2,"getUploadSpeed",order);

			if (columnID == SWTConstants.DOWNLOAD_PEER_LIST_CC_COLUMN_ID) {
				String cc1 = CountryLocator.getInstance().getCountryCode(object1.getAddress());
				String cc2 = CountryLocator.getInstance().getCountryCode(object2.getAddress());
				int result = cc1.compareTo(cc2);
				if (order)
					return result;
				else
					return Misc.reverse(result);
			}
			
			if (columnID == SWTConstants.DOWNLOAD_PEER_LIST_FLAG_COLUMN_ID) {
				String country1 = CountryLocator.getInstance().getCountryName(object1.getAddress());
				String country2 = CountryLocator.getInstance().getCountryName(object2.getAddress());
				int result = country1.compareTo(country2);
				if (order)
					return result;
				else
					return Misc.reverse(result);
			}
			
			if (columnID==SWTConstants.DOWNLOAD_PEER_LIST_STATUS_COLUMN_ID) {
				int id1 = download_session.getPeerDownloadStatus(object1).getPeerStatus();
				int id2 = download_session.getPeerDownloadStatus(object2).getPeerStatus();
				if ((id1==PeerDownloadStatus.IN_QUEUE)&&(id2==PeerDownloadStatus.IN_QUEUE)) {
					id1 = download_session.getPeerDownloadStatus(object1).getQueueRank();
					id2 = download_session.getPeerDownloadStatus(object2).getQueueRank();
				}
				int result = 0;
				if (id1>id2) result = 1;
				if (id1<id2) result = -1;
				if (!order) return Misc.reverse(result);
				return result;
				
			}
			
			return 0;
		}

		protected void addRow(Peer object) {
			super.addRow(object);
			
			if (!CountryLocator.getInstance().isServiceDown()) {
				
				Image image = SWTImageRepository.getFlagByAddress(object.getAddress(),FlagSize.S25x15);
				
				CountryFlagPainter painter = new CountryFlagPainter(image);
				TableItemCountryFlag table_item_painter = new TableItemCountryFlag(SWTPreferences.getDefaultColumnOrder(SWTConstants.DOWNLOAD_PEER_LIST_FLAG_COLUMN_ID),painter);
				addCustumControl(getItemCount()-1, table_item_painter);
		}
			
		}
		
		public void updateRow(Peer object) {
			String country_code = CountryLocator.getInstance().getCountryCode(object.getAddress());
			setRowText(object, SWTConstants.DOWNLOAD_PEER_LIST_CC_COLUMN_ID, country_code);
			
			setRowText(object, SWTConstants.DOWNLOAD_PEER_LIST_IP_COLUMN_ID, object.getAddress()+":"+object.getPort());
			setRowText(object, SWTConstants.DOWNLOAD_PEER_LIST_NICKNAME_COLUMN_ID, object.getNickName());
			setRowText(object, SWTConstants.DOWNLOAD_PEER_LIST_SOFTWARE_COLUMN_ID, PeerInfoFormatter.formatPeerSoftware(object));

			setRowText(object, SWTConstants.DOWNLOAD_PEER_LIST_DOWN_SPEED_COLUMN_ID,SpeedFormatter.formatSpeed(object.getDownloadSpeed()));
			setRowText(object, SWTConstants.DOWNLOAD_PEER_LIST_UP_SPEED_COLUMN_ID,SpeedFormatter.formatSpeed(object.getUploadSpeed()));
			
			if (download_session.hasPeer(object)) {
				setRowText(object, SWTConstants.DOWNLOAD_PEER_LIST_STATUS_COLUMN_ID, PeerInfoFormatter.formatPeerStatus(download_session.getPeerDownloadStatus(object)));
			} else {
				UploadSession upload_session = null;
				if (upload_manager.hasUpload(download_session.getFileHash()))
					upload_session = upload_manager.getUpload(download_session.getFileHash());
				if (upload_session != null) {				
					int rank = upload_session.getPeerPosition(object);
					String str="";
					if (rank!=0) {
						str = Localizer.getString("downloadinfowindow.tab.peerlist.column.status.upload_queue",rank+"");
					} else {
						str = _._("downloadinfowindow.tab.peerlist.column.status.uploading");
					}
					
					setRowText(object, SWTConstants.DOWNLOAD_PEER_LIST_STATUS_COLUMN_ID, str);
				} 
			}
				
		}

		protected Menu getPopUpMenu() {
			return popup_menu;
		}
	}

}