exoplanet/GraphLineCurve.java at master · sanjanakale19/exoplanet · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
import org.jfree.chart.ChartFactory;
import org.jfree.chart.ChartPanel;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.StandardChartTheme;
import org.jfree.chart.axis.NumberAxis;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.chart.plot.XYPlot;
import org.jfree.chart.renderer.xy.StandardXYItemRenderer;
import org.jfree.chart.renderer.xy.XYItemRenderer;
import org.jfree.chart.ui.RectangleInsets;
import org.jfree.data.xy.XYDataset;
import org.jfree.data.xy.XYSeries;
import org.jfree.data.xy.XYSeriesCollection;
import org.jfree.ui.ApplicationFrame;
import org.jfree.ui.RefineryUtilities;

import javax.swing.*;
import java.awt.*;
import java.util.ArrayList;
import java.util.List;
import java.awt.Color;
import javax.swing.JPanel;
import org.jfree.chart.ChartFactory;
import org.jfree.chart.ChartPanel;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.StandardChartTheme;
import org.jfree.chart.axis.NumberAxis;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.chart.plot.XYPlot;
import org.jfree.chart.renderer.xy.StandardXYItemRenderer;
import org.jfree.chart.renderer.xy.XYItemRenderer;
import org.jfree.data.xy.XYDataset;
import org.jfree.data.xy.XYSeries;
import org.jfree.data.xy.XYSeriesCollection;
import org.jfree.ui.ApplicationFrame;
import org.jfree.chart.ui.RectangleInsets;
import org.jfree.ui.RefineryUtilities;

import java.util.ArrayList;
import java.util.List;

class GraphLineCurve extends ApplicationFrame {

    private static final long serialVersionUID = 1L;
    private ChartPanel panel;
    private final List<GraphParam> params;

    static {
        ChartFactory.setChartTheme(new StandardChartTheme("JFree/Shadow", false));
    }

    ChartPanel getPanel() {
        return panel;
    }

    GraphLineCurve(String title, List<GraphParam> params) {
        super(title);
        this.params = params;
        panel = (ChartPanel) createDemoPanel();
        panel.setPreferredSize(new java.awt.Dimension(1500, 800));
        setContentPane(panel);
    }

    void graph() {
        pack();
        RefineryUtilities.centerFrameOnScreen(this);
        setVisible(true);
    }

    static GraphComplexCurve timeToFreqGraph(GraphLineCurve curve, List<Double> domain, String myTitle) {
        return new GraphComplexCurve(myTitle, domain, FourierTransform.toFrequency(toComplexList(curve.getParams())));
    }

    static List<ComplexNumber> toComplexList(List<GraphParam> list) {
        List<ComplexNumber> nums = new ArrayList<>();
        for (GraphParam param : list) {
            nums.add(new ComplexNumber(param.getFlux(), 0));
        }
        return nums;
    }

    private static JFreeChart createLineChart(XYDataset dataSet) {
        JFreeChart chart = ChartFactory.createXYLineChart("Corrected Flux vs. BJD Time",
                "BJD - 2454833", "Corrected Flux", dataSet, PlotOrientation.VERTICAL, true,
                true, false);

        chart.setBackgroundPaint(Color.white);

        XYPlot plot = (XYPlot) chart.getPlot();
        plot.setBackgroundPaint(Color.white);
        plot.setDomainGridlinePaint(Color.cyan);
        plot.setRangeGridlinePaint(Color.cyan);
        plot.setAxisOffset(new RectangleInsets(5.0, 5.0, 5.0, 5.0));
        plot.setDomainCrosshairVisible(true);
        plot.setRangeCrosshairVisible(true);

        XYItemRenderer r = plot.getRenderer();
        if (r instanceof StandardXYItemRenderer) {
            StandardXYItemRenderer renderer = (StandardXYItemRenderer) r;
            renderer.setBaseShapesVisible(true);
            renderer.setBaseShapesFilled(true);
            renderer.setDrawSeriesLineAsPath(true);
        }

        NumberAxis axis = (NumberAxis) plot.getDomainAxis();

        return chart;

    }

    private XYDataset createDataSet() {

        XYSeries s1 = new XYSeries(o -> 0);

        for (GraphParam param : params) {
            s1.add(param.getBJD(), param.getFlux());
        }
        XYSeriesCollection dataSet = new XYSeriesCollection();
        dataSet.addSeries(s1);

        return dataSet;
    }

    List<GraphParam> getParams() {
        return params;
    }

    private JPanel createDemoPanel() {
        JFreeChart chart = createLineChart(createDataSet());

        ChartPanel panel = new ChartPanel(chart);
        panel.setFillZoomRectangle(true);
        panel.setMouseWheelEnabled(true);
        return panel;
    }


}