sellportfoliochartjdialog.java

JStock是一个免费股市软件

/*
 * SellPortfolioChartJDialog.java
 *
 * Created on April 2, 2008, 10:05 AM
 */

package org.yccheok.jstock.gui;

import org.jfree.chart.ChartFactory;
import org.jfree.chart.ChartPanel;
import org.jfree.chart.JFreeChart;
import org.jfree.data.general.DefaultPieDataset;
import org.yccheok.jstock.engine.Symbol;
import org.yccheok.jstock.portfolio.Portfolio;
import org.yccheok.jstock.portfolio.Transaction;
import org.yccheok.jstock.portfolio.TransactionSummary;

/**
 *
 * @author  yccheok
 */
public class SellPortfolioChartJDialog extends javax.swing.JDialog {
    
    private static final String[] cNames = {
        "Net Gain %",
        "Net Loss %",
        "Gain %",
        "Loss %",
        "Net Gain Value",
        "Net Loss Value",
        "Gain Value",
        "Loss Value",
        "Net Selling Value",
        "Selling Value", 
        "Purchase Value",
        "Gain Price",
        "Lose Price",
        "Selling Price",
        "Purchase Price",        
        "Units",
        "Broker", 
        "Stamp Duty", 
        "Clearing Fee", 
    };
    
    /** Creates new form SellPortfolioChartJDialog */
    public SellPortfolioChartJDialog(java.awt.Frame parent, boolean modal, SellPortfolioTreeTableModel portfolioTreeTableModel) {
        super(parent, "Sell Summary", modal);
        initComponents();
        
        this.portfolioTreeTableModel = portfolioTreeTableModel;
        
        final JFreeChart freeChart = createChart(cNames[0]);
        
        chartPanel = new ChartPanel(freeChart, true, true, true, true, true);
        
        getContentPane().add(chartPanel, java.awt.BorderLayout.CENTER);        
    }
    
    /** This method is called from within the constructor to
     * initialize the form.
     * WARNING: Do NOT modify this code. The content of this method is
     * always regenerated by the Form Editor.
     */
    // <editor-fold defaultstate="collapsed" desc="Generated Code">//GEN-BEGIN:initComponents
    private void initComponents() {

        jPanel1 = new javax.swing.JPanel();
        jPanel2 = new javax.swing.JPanel();
        jComboBox1 = new javax.swing.JComboBox();

        setDefaultCloseOperation(javax.swing.WindowConstants.DISPOSE_ON_CLOSE);

        jPanel1.setLayout(new java.awt.BorderLayout());

        for(String cName : this.cNames) {
            this.jComboBox1.addItem(cName);
        }
        jComboBox1.addActionListener(new java.awt.event.ActionListener() {
            public void actionPerformed(java.awt.event.ActionEvent evt) {
                jComboBox1ActionPerformed(evt);
            }
        });
        jPanel2.add(jComboBox1);

        jPanel1.add(jPanel2, java.awt.BorderLayout.EAST);

        getContentPane().add(jPanel1, java.awt.BorderLayout.SOUTH);

        java.awt.Dimension screenSize = java.awt.Toolkit.getDefaultToolkit().getScreenSize();
        setBounds((screenSize.width-750)/2, (screenSize.height-600)/2, 750, 600);
    }// </editor-fold>//GEN-END:initComponents

    private JFreeChart createChart(String name) {
        final Portfolio portfolio = (Portfolio)portfolioTreeTableModel.getRoot();
        final int count = portfolio.getChildCount();
        DefaultPieDataset data = new DefaultPieDataset();

        for(int i=0; i<count; i++) {
            TransactionSummary transactionSummary = (TransactionSummary)portfolio.getChildAt(i);
            
            if(transactionSummary.getChildCount() <= 0) continue;
            
            Transaction transaction = (Transaction)transactionSummary.getChildAt(0);
            final Symbol symbol = transaction.getContract().getStock().getSymbol();
            
            /* Should use reflection technology. */
            if(name.equals(cNames[0])) {
                data.setValue(symbol.toString(), portfolioTreeTableModel.getNetGainLossPercentage(transactionSummary));
            }
            if(name.equals(cNames[1])) {
                data.setValue(symbol.toString(), -portfolioTreeTableModel.getNetGainLossPercentage(transactionSummary));
            }            
            else if(name.equals(cNames[2])) {
                data.setValue(symbol.toString(), portfolioTreeTableModel.getGainLossPercentage(transactionSummary));                
            }
            else if(name.equals(cNames[3])) {
                data.setValue(symbol.toString(), -portfolioTreeTableModel.getGainLossPercentage(transactionSummary));                
            }            
            else if(name.equals(cNames[4])) {
                data.setValue(symbol.toString(), portfolioTreeTableModel.getNetGainLossValue(transactionSummary));                
            }
            else if(name.equals(cNames[5])) {
                data.setValue(symbol.toString(), -portfolioTreeTableModel.getNetGainLossValue(transactionSummary));                
            }
            else if(name.equals(cNames[6])) {
                data.setValue(symbol.toString(), portfolioTreeTableModel.getGainLossValue(transactionSummary));                
            }
            else if(name.equals(cNames[7])) {
                data.setValue(symbol.toString(), -portfolioTreeTableModel.getGainLossValue(transactionSummary));                
            }            
            else if(name.equals(cNames[8])) {
                data.setValue(symbol.toString(), transactionSummary.getNetTotal());                
            }
            else if(name.equals(cNames[9])) {
                data.setValue(symbol.toString(), transactionSummary.getTotal());                
            }
            else if(name.equals(cNames[10])) {
                data.setValue(symbol.toString(), transactionSummary.getReferenceTotal());                
            }
            else if(name.equals(cNames[11])) {
                data.setValue(symbol.toString(), portfolioTreeTableModel.getGainLossPrice(transactionSummary));                
            }
            else if(name.equals(cNames[12])) {
                data.setValue(symbol.toString(), -portfolioTreeTableModel.getGainLossPrice(transactionSummary));                
            }            
            else if(name.equals(cNames[13])) {
                data.setValue(symbol.toString(), portfolioTreeTableModel.getSellingPrice(transactionSummary));                
            }
            else if(name.equals(cNames[14])) {
                data.setValue(symbol.toString(), portfolioTreeTableModel.getPurchasePrice(transactionSummary));                
            }
            else if(name.equals(cNames[15])) {
                data.setValue(symbol.toString(), transactionSummary.getQuantity());                
            }
            else if(name.equals(cNames[16])) {
                data.setValue(symbol.toString(), transactionSummary.getCalculatedBroker());                
            }
            else if(name.equals(cNames[17])) {
                data.setValue(symbol.toString(), transactionSummary.getCalculatedStampDuty());                
            }
            else if(name.equals(cNames[18])) {
                data.setValue(symbol.toString(), transactionSummary.getCalculatdClearingFee());                
            }            
        }
        
        // create a chart...
        return ChartFactory.createPieChart(name, data, true, true, true);        
    }
    
    private void jComboBox1ActionPerformed(java.awt.event.ActionEvent evt) {//GEN-FIRST:event_jComboBox1ActionPerformed
        // TODO add your handling code here:
        String selected = ((javax.swing.JComboBox)evt.getSource()).getSelectedItem().toString();
        
        final JFreeChart freeChart = this.createChart(selected);
        chartPanel.setChart(freeChart);
    }//GEN-LAST:event_jComboBox1ActionPerformed
    
    private SellPortfolioTreeTableModel portfolioTreeTableModel;
    private ChartPanel chartPanel;
    
    // Variables declaration - do not modify//GEN-BEGIN:variables
    private javax.swing.JComboBox jComboBox1;
    private javax.swing.JPanel jPanel1;
    private javax.swing.JPanel jPanel2;
    // End of variables declaration//GEN-END:variables
    
}