146 lines
4.1 KiB
Java
146 lines
4.1 KiB
Java
package View;
|
|
|
|
import Model.Line;
|
|
import org.jfree.chart.ChartFactory;
|
|
import org.jfree.chart.ChartPanel;
|
|
import org.jfree.chart.JFreeChart;
|
|
import org.jfree.chart.plot.PlotOrientation;
|
|
import org.jfree.chart.plot.XYPlot;
|
|
import org.jfree.chart.renderer.xy.XYLineAndShapeRenderer;
|
|
import org.jfree.data.xy.XYSeries;
|
|
import org.jfree.data.xy.XYSeriesCollection;
|
|
import org.jfree.util.ShapeUtilities;
|
|
|
|
import javax.swing.*;
|
|
import java.awt.*;
|
|
import java.util.ArrayList;
|
|
import java.util.Collections;
|
|
import java.util.LinkedList;
|
|
|
|
/**
|
|
* Implementierung verschiedener Algorithmen zur Berechnung von Ausgleichsgeraden.
|
|
*
|
|
* @Author: Armin Wolf
|
|
* @Email: a_wolf28@uni-muenster.de
|
|
* @Date: 29.05.2017.
|
|
*/
|
|
public class PlotDialog extends JPanel {
|
|
|
|
private JFreeChart chart;
|
|
private ChartPanel panel;
|
|
private XYSeriesCollection datapoints;
|
|
private XYSeries series;
|
|
private Double min;
|
|
private Double max;
|
|
private XYPlot xyPlot;
|
|
private int seriesCount;
|
|
|
|
private XYLineAndShapeRenderer renderer;
|
|
private Shape diamond;
|
|
|
|
public PlotDialog() {
|
|
super();
|
|
// this.setPreferredSize(new Dimension(1000, 1000));
|
|
// this.setMinimumSize(new Dimension(1000, 800));
|
|
this.setLayout(new BorderLayout());
|
|
seriesCount = 1;
|
|
}
|
|
|
|
public void createPlot(LinkedList<Line> points) {
|
|
if (!points.isEmpty()) {
|
|
try {
|
|
Thread thread = new Thread(() -> convertData(points));
|
|
thread.start();
|
|
thread.join();
|
|
} catch (InterruptedException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
//createScatterPlot
|
|
chart = ChartFactory.createXYLineChart("",
|
|
"X", "Y", datapoints, PlotOrientation.VERTICAL, true, true, false);
|
|
diamond = ShapeUtilities.createDiamond(2f);
|
|
chart.setBorderVisible(false);
|
|
chart.setAntiAlias(true);
|
|
chart.getPlot().setBackgroundPaint(Color.WHITE);
|
|
chart.setBorderVisible(false);
|
|
|
|
|
|
xyPlot = (XYPlot) chart.getPlot();
|
|
xyPlot.setDomainCrosshairVisible(true);
|
|
xyPlot.setRangeCrosshairVisible(true);
|
|
|
|
|
|
renderer = (XYLineAndShapeRenderer) xyPlot.getRenderer();
|
|
renderer.setSeriesLinesVisible(0, false);
|
|
renderer.setSeriesShapesVisible(0, true);
|
|
renderer.setSeriesLinesVisible(1, true);
|
|
renderer.setSeriesLinesVisible(1, true);
|
|
renderer.setSeriesPaint(0, Color.blue);
|
|
renderer.setSeriesShape(0, diamond);
|
|
renderer.setBaseSeriesVisible(true);
|
|
|
|
|
|
xyPlot.setDomainCrosshairVisible(true);
|
|
xyPlot.setRangeCrosshairVisible(true);
|
|
|
|
panel = new ChartPanel(chart);
|
|
panel.setMouseZoomable(false);
|
|
this.add(panel, BorderLayout.CENTER);
|
|
}
|
|
|
|
public void clear(){
|
|
if (datapoints != null)
|
|
datapoints.removeAllSeries();
|
|
}
|
|
|
|
public void addLineToPlot(double m, double b, Paint color, String name) {
|
|
|
|
XYSeries linesA = new XYSeries(name);
|
|
linesA.add(min.doubleValue()-10, min.doubleValue() * m + b);
|
|
linesA.add(max.doubleValue()+10, max.doubleValue() * m + b);
|
|
|
|
datapoints.addSeries(linesA);
|
|
|
|
renderer.setSeriesPaint(seriesCount, color);
|
|
renderer.setSeriesStroke(seriesCount, new BasicStroke(2.0f));
|
|
renderer.setBaseSeriesVisible(true);
|
|
renderer.setSeriesLinesVisible(seriesCount, true);
|
|
seriesCount++;
|
|
|
|
}
|
|
|
|
|
|
public void addLineToPlot(double m, double b, String name) {
|
|
|
|
XYSeries linesA = new XYSeries(name);
|
|
linesA.add(min.intValue()-10, min.intValue() * m + b);
|
|
linesA.add(max.intValue()+10, max.intValue() * m + b);
|
|
|
|
datapoints.addSeries(linesA);
|
|
seriesCount = xyPlot.getSeriesCount();
|
|
renderer.setSeriesPaint(seriesCount, Color.red);
|
|
renderer.setSeriesStroke(seriesCount, new BasicStroke(2.0f));
|
|
renderer.setBaseSeriesVisible(true);
|
|
renderer.setSeriesLinesVisible(seriesCount, true);
|
|
}
|
|
|
|
private void convertData(LinkedList<Line> points) {
|
|
|
|
datapoints = new XYSeriesCollection();
|
|
ArrayList<Double> coordinates = new ArrayList<>();
|
|
series = new XYSeries("points");
|
|
for (Line p : points) {
|
|
series.add(p.getM().doubleValue(), p.getB().doubleValue() * (-1));
|
|
coordinates.add(p.getM());
|
|
|
|
}
|
|
this.max = series.getMaxX();
|
|
this.min = series.getMinX();
|
|
datapoints.addSeries(series);
|
|
}
|
|
|
|
|
|
}
|
|
|