Desenhando Gráficos Simples com Java 2D (Cont.)

DesenhandoGraficos2.java
/* * DesenhandoGraficos2.java */ import java.awt.Color; import java.awt.Font; import java.awt.Graphics; import java.util.ArrayList; import java.util.List; import javax.swing.JPanel; /** * * @author Andreia */ public class DesenhandoGraficos2 extends JPanel { private int x_linhaVertical = 40; private int y_linhaVertical = 40; private int larguraColuna = 10;//largura da coluna private int margem = 10; private int altura = 210; // altura do grafico //* use na 'altura' um numero divisivel pela quantidade de linhas private int quantidadeLinhas = 7; private int espaco = 5; // espaco entre valores private String[] rotulos; private List<int[]> dadosList = new ArrayList<int[]>(); private List<Color> cores = new ArrayList<Color>(); private List<String> legendas = new ArrayList<String>(); private int tipoGrafico; // 0 - Colunas; 1 - Linha private String tituloGrafico; @Override public void paintComponent(Graphics g) { super.paintComponent(g); this.setBackground(Color.WHITE); desenhaLinhasDoGrafico(g); desenhaRotulos(g); desenhaLegenda(g); for (int i = 0; i < rotulos.length; i++) { for (int j = 0; j < dadosList.size(); j++) { int dado = dadosList.get(j)[i]; // seta cor da coluna ou linha g.setColor(cores.get(j)); // desenha grafico switch (tipoGrafico) { case 0: // grafico em colunas g.fillRect(i * espaco * 2 + (j + 1) * larguraColuna + (dadosList.size() * larguraColuna) * i + x_linhaVertical, (altura + y_linhaVertical) - dado, larguraColuna, dado); break; case 1: // grafico em linha if (i > 0) { g.drawLine(i * espaco + (dadosList.size() * larguraColuna) * i + x_linhaVertical, (altura + y_linhaVertical) - dadosList.get(j)[i - 1], (i + 1) * espaco + (dadosList.size() * larguraColuna) * (i + 1) + x_linhaVertical, (altura + y_linhaVertical) - dado); } // desenha marcadores na linha g.fillOval((i + 1) * espaco + dadosList.size() * larguraColuna + (dadosList.size() * larguraColuna) * i + x_linhaVertical - 3, (altura + y_linhaVertical) - dado - 3, 7, 7); break; } // fim do switch } // fim do for rotulos } // fim do for dados /* * desenha titulo do grafico */ g.setColor(Color.BLACK); g.setFont(new Font(Font.SANS_SERIF, Font.BOLD, 14)); g.drawString(tituloGrafico, margem, margem * 2); } // fim do metodo paintComponent private void desenhaLinhasDoGrafico(Graphics g) { int x2_linhaHorizontal = rotulos.length * espaco * 2 + larguraColuna + (dadosList.size() * larguraColuna) * rotulos.length + x_linhaVertical - 5; // comprimento da linha horizontal g.setColor(Color.BLACK); // desenha linha vertical g.drawLine(x_linhaVertical, y_linhaVertical, x_linhaVertical, y_linhaVertical + altura + 3); // desenha linhas horizontais int contador = altura; for (int i = 0; i <= quantidadeLinhas; i++) { int y_linhaHorizontal = y_linhaVertical + i * (altura / quantidadeLinhas); // desenha linha g.drawLine(x_linhaVertical - 3, y_linhaHorizontal, x2_linhaHorizontal, y_linhaHorizontal); // desenha numeros g.drawString("" + contador, margem, y_linhaHorizontal + 5); contador -= (altura / quantidadeLinhas); } // fim do for } // fim do metodo desenhaLinhasDoGrafico private void desenhaRotulos(Graphics g) { for (int i = 0; i < rotulos.length; i++) { g.setColor(Color.BLACK); // desenha rotulos g.drawString(rotulos[i], i * espaco * 2 + larguraColuna + (dadosList.size() * larguraColuna) * i + x_linhaVertical, altura + 60); // desenha tracos da linha horizontal g.drawLine((i * 2 + 1) * espaco + larguraColuna + (dadosList.size() * larguraColuna) * (i + 1) + x_linhaVertical, altura + y_linhaVertical, (i * 2 + 1) * espaco + larguraColuna + (dadosList.size() * larguraColuna) * (i + 1) + x_linhaVertical, altura + 45); } // fim do for } // fim do metodo desenhaRotulos /** * desenha legenda de cores */ private void desenhaLegenda(Graphics g) { for (int j = 0; j < dadosList.size(); j++) { g.setColor(Color.BLACK); // escreve legenda g.drawString(legendas.get(j), rotulos.length * espaco * 2 + larguraColuna + (dadosList.size() * larguraColuna) * rotulos.length + x_linhaVertical + 15, y_linhaVertical + j * 15 + 7); // seta cor da legenda g.setColor(cores.get(j)); // desenha um quadrado 7x7 g.fillRect(rotulos.length * espaco * 2 + larguraColuna + (dadosList.size() * larguraColuna) * rotulos.length + x_linhaVertical + 5, y_linhaVertical + j * 15, 7, 7); } // fim do for } // fim do metodo desenhaLegenda public void setTituloGrafico(String tituloGrafico) { this.tituloGrafico = tituloGrafico; } public void setRotulos(String[] rotulos) { this.rotulos = rotulos; } public void setDados(String titulo, int[] dados, Color cor) { legendas.add(titulo); cores.add(cor); dadosList.add(dados); } public void setTipoGrafico(int indice) { tipoGrafico = indice; repaint(); } // fim do metodo setTipoGrafico } // fim da classe DesenhandoGraficos2

TesteGraficos2.java
/* * TesteGraficos2.java */ import java.awt.BorderLayout; import java.awt.Color; import javax.swing.JComboBox; import javax.swing.JFrame; /** * * @author Andreia */ public class TesteGraficos2 { public static void main(String[] args) { // cria painel para desenhar grafico final DesenhandoGraficos2 desenhando = new DesenhandoGraficos2(); /* * Exemplo */ // seta titulo para o grafico desenhando.setTituloGrafico("Titulo do Grafico"); // rotulos aparecerao na parte inferior do grafico String[] rotulos = new String[]{"2008", "2009", "2010", "2011", "2012"}; desenhando.setRotulos(rotulos); // cria os dados para o grafico int[] dados1 = new int[rotulos.length]; int[] dados2 = new int[rotulos.length]; int[] dados3 = new int[rotulos.length]; dados1[0] = 23; // 2008 dados1[1] = 15; // 2009 dados1[2] = 55; // 2010 dados1[3] = 80; // 2011 dados1[4] = 73; // 2012 dados2[0] = 55; // 2008 dados2[1] = 37; // 2009 dados2[2] = 120;// 2010 dados2[3] = 135;// 2011 dados2[4] = 44; // 2012 dados3[0] = 79; // 2008 dados3[1] = 162;// 2009 dados3[2] = 204;// 2010 dados3[3] = 175;// 2011 dados3[4] = 90; // 2012 desenhando.setDados("Idoso", dados1, Color.GREEN); desenhando.setDados("Adulto", dados2, Color.YELLOW); desenhando.setDados("Crianca", dados3, Color.RED); /* */ // cria comboBox final JComboBox jComboBoxGrafico = new JComboBox(); jComboBoxGrafico.setModel(new javax.swing.DefaultComboBoxModel( new String[]{"Grafico em Colunas", "Grafico em Linha"})); jComboBoxGrafico.addActionListener(new java.awt.event.ActionListener() { @Override public void actionPerformed(java.awt.event.ActionEvent evt) { desenhando.setTipoGrafico(jComboBoxGrafico.getSelectedIndex()); } }); /* * cria janela */ JFrame frame = new JFrame(); frame.add(jComboBoxGrafico, BorderLayout.NORTH); frame.add(desenhando); frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE); frame.setSize(355, 355); frame.setVisible(true); } } // fim da classe TesteGraficos2