Java中的旅行推销员可视化(Swing)

时间:2015-06-21 13:32:31

标签: java algorithm swing

出于练习目的,我挑战自己写一个解决TSP的程序,并逐步显示结果。

就目前而言,我的程序使用简单的最近邻算法。我希望我的程序灵活,所以当我添加一个新算法时,它也能够可视化结果,而不会弄乱显示逻辑。

我遇到的一个问题是 - 如何逐步显示解决方案?我通过创建多个部分解决方案,存储它们并逐个显示来解决它。我觉得它可以做得更好,但我的图形并不是很好,我希望能在这里获得一些线索。

这是一些代码: Point类代表一个城市。

class Point {
    private double x;
    private double y;

    public double getX() {
        return x;
    }
    public double getY() {
        return y;
    }

    public Point(double x, double y) {
        this.x = x;
        this.y = y;
    }

    public Point(){
        Random r = new Random();
        x=r.nextInt(1000);
        y=r.nextInt(650);
    }

    public double calculateDistanceToPoint(Point p) {
        double dist = Math.sqrt(Math.pow(this.x-p.x, 2) + Math.pow(this.y-p.y, 2));
        return round(dist,2);
    }

    private static double round(double value, int places) {
        if (places < 0) throw new IllegalArgumentException();

        BigDecimal bd = new BigDecimal(value);
        bd = bd.setScale(places, RoundingMode.HALF_UP);
        return bd.doubleValue();
    }
}

然后,正在进行计算的Solver类:

class Solver {
    //list of all points to visit
    private static ArrayList<Point> points = new ArrayList<>();

    //adjacency matrix
    private ArrayList<ArrayList<Double>> adjMatrix = new ArrayList<>();

    //found solution
    private static ArrayList<Point> solution = new ArrayList<>();

    //visited points
    private ArrayList<Integer> visitedPoints = new ArrayList<>();

    //used for visualisation
    private static Solution finalSolution = new Solution();

    public void clear() {
        points.clear();
        solution.clear();
        visitedPoints.clear();
        adjMatrix.clear();
        finalSolution.clear();
    }

    public void addPoint(Point p) {
        points.add(p);
    }

    public static ArrayList<Point> getPoints() {
        return Solver.points;
    }

    public void fillAdjacencyMatrix() {
        int iter_x;
        int iter_y;
        for (iter_x = 0; iter_x < points.size(); iter_x++) {
            ArrayList<Double> temp = new ArrayList<>();
            for (iter_y = 0; iter_y < points.size(); iter_y++) {
                if (iter_x == iter_y) {
                    temp.add(-1.0);
                } else {
                    temp.add(points.get(iter_x).calculateDistanceToPoint(points.get(iter_y)));
                }
            }
            adjMatrix.add(temp);
        }
    }

    private int getIndexOfMin(ArrayList<Double> arr) {
        Double min = Double.MAX_VALUE;
        int index = -2;
        for (int i = 0; i < arr.size(); i++) {
            Double val = arr.get(i);
            if (!(val == -1.0) && !visitedPoints.contains(i) && val < min) {
                min = val;
                index = i;
            }
        }
        return index;
    }

    public void solveUsingNN(int startingPoint) {
        int noOfVisited = 0;

        //find nearest point from the starting one
        int nearest = getIndexOfMin(adjMatrix.get(startingPoint));
        Solution sol = new Solution();

        //until we've visited all points
        while (noOfVisited!=points.size()) {
            //get next nearest point and add it to visited
            nearest = getIndexOfMin(adjMatrix.get(nearest));
            visitedPoints.add(nearest);

            //add this point to solution
            Point newPoint = points.get(nearest);
            solution.add(newPoint);

            //create a new frame for animation, containing all previous steps and recently added one
            SolutionStep ss = new SolutionStep();
            Point p;
            for (Point newPoint : solution) {
                p = new Point(newPoint.getX(), newPoint.getY());
                ss.addPoint(p);
            }
            sol.addStep(ss);
            noOfVisited++;
        }
        finalSolution=sol;
    }
}

然后,SolutionStep类:

class SolutionStep{
    public final ArrayList<Point> step = new ArrayList<>();
    public SolutionStep(){}

    public void addPoint(Point p){
        step.add(p);
    }
    public void draw(Graphics g) {
        Graphics2D g2 = (Graphics2D) g;
            for (int i = 0; i < step.size()-1; i++) {
                g2.draw(new Line2D.Double(step.get(i).getX(), step.get(i).getY(), step.get(i + 1).getX(), step.get(i + 1).getY()));
        }
    }
}

Solution,其中包含许多步骤。

public class Solution {
    private ArrayList<Point> points = new ArrayList<>();
    private static ArrayList<SolutionStep> playbackSolution = new ArrayList<>();
    private int noOfFrames;

    public Solution(ArrayList<SolutionStep> listOfSteps, int noOfFrames){
        this.noOfFrames=noOfFrames;
        playbackSolution=listOfSteps;
    }
    public Solution(){}

    public static ArrayList<SolutionStep> getPlayback(){
        return playbackSolution;
    }
    public void clear(){
        playbackSolution.clear();
    }

    public void addStep(SolutionStep solutionStep){
        playbackSolution.add(solutionStep);
    }

    public void draw(Graphics g) {
        int numberOfPoints;
        points = Solver.getPoints();
        Graphics2D g2 = (Graphics2D) g;
        //draw all points
        for (Point point : points) {
            g2.fill(new Rectangle2D.Double(point.getX(), point.getY(), 6, 6));
        }

        //draw next line
        for(int i = 0;i<noOfFrames;i++) {
            playbackSolution.get(i).draw(g);
        }

        //if we are at the final solution, draw a line from last point to the first
        if (noOfFrames == points.size()){
            numberOfPoints = points.size();
            Point first = playbackSolution.get(0).step.get(0);
            Point last = playbackSolution.get(numberOfPoints-1).step.get(numberOfPoints-1);
            g2.draw(new Line2D.Double(first.getX(), first.getY(), last.getX(), last.getY()));
        }
    }
}

最后,Visualisation

class Visualisation extends JFrame {
    private DrawingPanel contentPane;
    private int noOfPoints = 10;
    private int delay_time = 300;

    public Visualisation() {
        setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
        setBounds(100, 100, 1100, 700);
        contentPane = new DrawingPanel();
        setContentPane(contentPane);
        JButton start = new JButton("Start");
        start.addActionListener(new ActionListener() {
            @Override
            public void actionPerformed(ActionEvent actionEvent) {
                Solver s = new Solver();
                s.clear();
                contentPane.displayNoOfSteps = 0;
                for (int i=0;i<noOfPoints;i++) {
                    s.addPoint(new Point());
                }
                s.fillAdjacencyMatrix();
                s.solveUsingNN(0);
                new javax.swing.Timer(delay_time, new ActionListener(){
                    @Override
                    public void actionPerformed(ActionEvent e){
                            contentPane.repaint();
                    }
                }).start();
                contentPane.repaint();
            }
        });
        contentPane.add(start);
    }
}

DrawingPanel

class DrawingPanel extends JPanel{
    public int displayNoOfSteps = 1;

    DrawingPanel(){}
    @Override
    protected void paintComponent(Graphics g) {
        super.paintComponent(g);
        Solution sol = new Solution(Solution.getPlayback(),displayNoOfSteps);
        sol.draw(g);

        if (displayNoOfSteps< Solution.getPlayback().size())
        displayNoOfSteps++;
        }
    }

主要课程:

class Main {
    public static void main(String[] args){
        Visualisation frame = new Visualisation();
        frame.setVisible(true);
    }
} 
  1. 现在,在Visualisation课程中,我有一个Timer。此计时器在repaint()DrawingPanel ms调用delay_time,并且在每次迭代中都会增加要显示的步骤数。问题是,如果我运行一个模拟,然后再次点击Start,模拟运行得更快,并且在几次运行后,它几乎立即显示最后一步。我不知道出了什么问题。我该如何处理?

  2. 我在程序启动时遇到错误 -

    at Solution.draw(Solution.java:57)

    at DrawingPanel.paintComponent(Visualisation.java:53)

  3. 指的是行:

    playbackSolution.get(i).draw(g);
    

    sol.draw(g);
    

    但我还没画任何东西! repaint()位于ActionListener中的JButton。或者也许正在绘制JButton来电draw()?我怎样摆脱这个问题?

    1. 此外,我觉得我使用了太多静态字段和方法 - 但另一方面,创建例如Solver的实例然后是非更好的静态方法来获得解决方案?或者也许让Solver成为单身人士?无论如何都有一个例子。

    2. 正如我前面提到的,在编写更复杂的算法(如模拟退火)之前,我希望得到一些关于此代码的反馈,因此很容易保持良好的代码质量。我可以在此代码中更改哪些内容,以便更轻松地添加新功能?

1 个答案:

答案 0 :(得分:1)

有经验维护一个相当大的Java Swing应用程序,我永远不会自愿承诺在Swing中做一些新的东西。

我认为通过使用第三方工具可视化图形可以很好地解决这个特殊问题。 Graphviz是一个选项,但存在其他几种工具。请查看here以获取更多示例。

您所做的只是以可视化工具的符号生成图形。您可以使用节点名称来显示Salesman采用的路径:1,2,3等