如何在用Java

时间:2016-09-14 12:05:07

标签: java image swing plot graph

我在程序中总共有500个数组列表作为HashMap,并希望为每个列表生成图形(总共500个图形)。我使用下面的代码绘制一个图表..

import java.awt.BasicStroke;
import java.awt.Color;
import java.awt.Dimension;
import java.awt.FontMetrics;
import java.awt.Graphics;
import java.awt.Graphics2D;
import java.awt.Point;
import java.awt.RenderingHints;
import java.awt.Stroke;
import java.util.ArrayList;
import java.util.List;
import javax.swing.*;

@SuppressWarnings("serial")
public class DrawGraph extends JPanel {
   private static final int PREF_W = 1000;
   private static final int PREF_H = 800;
   private static final int BORDER_GAP = 25;
   private static final int PADDING = 25;
   private static final Color GRAPH_COLOR = Color.green;
   private static final Color GRAPH_POINT_COLOR = new Color(100, 100, 100, 180);
   private static final Color GRID_COLOR = new Color(200, 200, 200, 200);
   private static final Stroke GRAPH_STROKE = new BasicStroke(3f);
   private static final int GRAPH_POINT_WIDTH = 4;
   private static final int Y_HATCH_CNT = 10;
   private List<Double> scores;

   public DrawGraph(List<Double> scores2) {
      this.scores = scores2;
   }

   @Override
   protected void paintComponent(Graphics g) {
      super.paintComponent(g);
      Graphics2D g2 = (Graphics2D)g;
      g2.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);

      double xScale = ((double) getWidth() - (2 * BORDER_GAP)-PADDING) / (scores.size() - 1);
      double yScale = ((double) getHeight() - (2 * BORDER_GAP)-PADDING) / (getMaxScore() - getMinScore());

      List<Point> graphPoints = new ArrayList<Point>();
      for (int i = 0; i < scores.size(); i++) {
         int x1 = (int) (i * xScale + BORDER_GAP + PADDING);
         int y1 = (int) ((getMaxScore() - scores.get(i)) * yScale + BORDER_GAP);
         graphPoints.add(new Point(x1, y1));
      }

      g2.setColor(Color.WHITE);
      g2.fillRect(BORDER_GAP + PADDING, BORDER_GAP, getWidth() - (2 * BORDER_GAP) - PADDING, getHeight() - 2 * BORDER_GAP - PADDING);
      g2.setColor(Color.BLACK);

      g2.drawLine(BORDER_GAP + PADDING, getHeight() - BORDER_GAP - PADDING, BORDER_GAP + PADDING, BORDER_GAP);
      g2.drawLine(BORDER_GAP + PADDING, getHeight() - BORDER_GAP - PADDING, getWidth() - BORDER_GAP, getHeight() - BORDER_GAP - PADDING);

      for (int i = 0; i < Y_HATCH_CNT + 1; i++) {
          int x0 = BORDER_GAP + PADDING;
          int x1 = GRAPH_POINT_WIDTH + BORDER_GAP + PADDING;
          int y0 = getHeight() - ((i * (getHeight() - BORDER_GAP * 2 - PADDING)) / Y_HATCH_CNT + BORDER_GAP + PADDING);
          int y1 = y0;
          if (scores.size() > 0) {
              g2.setColor(GRID_COLOR);
              g2.drawLine(BORDER_GAP + PADDING + 1 + GRAPH_POINT_WIDTH, y0, getWidth() - BORDER_GAP, y1);
              g2.setColor(Color.BLACK);
              String yLabel = ((int) ((getMinScore() + (getMaxScore() - getMinScore()) * ((i * 1.0) / Y_HATCH_CNT)) * 100)) / 100.0 + "";
              FontMetrics metrics = g2.getFontMetrics();
              int labelWidth = metrics.stringWidth(yLabel);
              g2.drawString(yLabel, x0 - labelWidth - 5, y0 + (metrics.getHeight() / 2) - 3);
          }
          g2.drawLine(x0, y0, x1, y1);
      }

      for (int i = 0; i < scores.size(); i++) {
          if (scores.size() > 1) {
              int x0 = i * (getWidth() - BORDER_GAP * 2 - PADDING) / (scores.size() - 1) + BORDER_GAP + PADDING;
              int x1 = x0;
              int y0 = getHeight() - BORDER_GAP - PADDING;
              int y1 = y0 - GRAPH_POINT_WIDTH;
              if ((i % ((int) ((scores.size() / 20.0)) + 1)) == 0) {
                  g2.setColor(GRID_COLOR);
                  g2.drawLine(x0, getHeight() - BORDER_GAP - PADDING - 1 - GRAPH_POINT_WIDTH, x1, BORDER_GAP);
                  g2.setColor(Color.BLACK);
                  String xLabel = i + "";
                  FontMetrics metrics = g2.getFontMetrics();
                  int labelWidth = metrics.stringWidth(xLabel);
                  g2.drawString(xLabel, x0 - labelWidth / 2, y0 + metrics.getHeight() + 3);
              }
              g2.drawLine(x0, y0, x1, y1);
          }
      }

      Stroke oldStroke = g2.getStroke();
      g2.setColor(GRAPH_COLOR);
      g2.setStroke(GRAPH_STROKE);
      for (int i = 0; i < graphPoints.size() - 1; i++) {
         int x1 = graphPoints.get(i).x;
         int y1 = graphPoints.get(i).y;
         int x2 = graphPoints.get(i + 1).x;
         int y2 = graphPoints.get(i + 1).y;
         g2.drawLine(x1, y1, x2, y2);         
      }

      g2.setStroke(oldStroke);      
      g2.setColor(GRAPH_POINT_COLOR);
      for (int i = 0; i < graphPoints.size(); i++) {
         int x = graphPoints.get(i).x - GRAPH_POINT_WIDTH / 2;
         int y = graphPoints.get(i).y - GRAPH_POINT_WIDTH / 2;;
         int ovalW = GRAPH_POINT_WIDTH;
         int ovalH = GRAPH_POINT_WIDTH;
         g2.fillOval(x, y, ovalW, ovalH);
      }
   }

   @Override
   public Dimension getPreferredSize() {
      return new Dimension(PREF_W, PREF_H);
   }

   private double getMinScore() {
       double minScore = Double.MAX_VALUE;
       for (Double score : scores) {
           minScore = Math.min(minScore, score);
       }
       return minScore;
   }

   private double getMaxScore() {
       double maxScore = Double.MIN_VALUE;
       for (Double score : scores) {
           maxScore = Math.max(maxScore, score);
       }
       return maxScore;
   }

   public static void createAndShowGui(ArrayList<Double> scores, String s) {
      DrawGraph mainPanel = new DrawGraph(scores);

      JFrame frame = new JFrame(s);
      frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
      frame.getContentPane().add(mainPanel);
      frame.pack();
      frame.setLocationRelativeTo(null);
      frame.setVisible(true);
   }

   public static void main(String args[]) {
       final ArrayList<Double> inputValues = new ArrayList<Double>();
       // Read input values from text document and store in ArrayList
       SwingUtilities.invokeLater(new Runnable() {
             public void run() {
                DrawGraph.createAndShowGui(inputValues, "MyGraphTitle");
             }
          });
   }
}

当我提供20个样本数组列表作为输入时,它绘制并正确显示20个图形。但是我想将图形保存为我的系统中的图像而不将其显示为单独的窗口,因为当绘制500个图形时我不想打开500个窗户。有没有选项可以在绘制图形后将图形保存为图像。?我在线检查了几种方法,但我不确定如何将保存为图像代码与使用2D图形的特定图形生成代码集成。

0 个答案:

没有答案