opensha-dev/src/main/java/scratch/kevin/ComcatMFDPlot.java at d6451a431bcddb3e38ec8139bb71a1c7aa5b9cae · opensha/opensha-dev · 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
package scratch.kevin;

import java.awt.Color;
import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Date;
import java.util.GregorianCalendar;
import java.util.List;

import org.jfree.data.Range;
import org.opensha.commons.data.comcat.ComcatAccessor;
import org.opensha.commons.data.comcat.ComcatRegion;
import org.opensha.commons.data.comcat.ComcatRegionAdapter;
import org.opensha.commons.data.function.DiscretizedFunc;
import org.opensha.commons.data.function.EvenlyDiscretizedFunc;
import org.opensha.commons.data.region.CaliforniaRegions;
import org.opensha.commons.geo.Location;
import org.opensha.commons.geo.Region;
import org.opensha.commons.gui.plot.HeadlessGraphPanel;
import org.opensha.commons.gui.plot.PlotCurveCharacterstics;
import org.opensha.commons.gui.plot.PlotLineType;
import org.opensha.commons.gui.plot.PlotSpec;
import org.opensha.commons.gui.plot.PlotSymbol;
import org.opensha.commons.gui.plot.PlotUtils;
import org.opensha.sha.earthquake.faultSysSolution.util.FaultSysTools;
import org.opensha.sha.earthquake.observedEarthquake.ObsEqkRupList;
import org.opensha.sha.earthquake.observedEarthquake.ObsEqkRupture;
import org.opensha.sha.earthquake.rupForecastImpl.nshm23.util.NSHM23_RegionLoader;
import org.opensha.sha.earthquake.rupForecastImpl.nshm23.util.NSHM23_RegionLoader.AnalysisRegions;
import org.opensha.sha.magdist.IncrementalMagFreqDist;

import net.mahdilamb.colormap.Colors;

public class ComcatMFDPlot {

	public static void main(String[] args) throws IOException {
		double minMag = 4d;
		double maxMag = 8d;
		int startYear = 1900;
//		int startYear = 1981;
		int endYear = 2025;
		Date startDate = new GregorianCalendar(startYear, 0, 1).getTime();
		Date endDate = new GregorianCalendar(endYear, 0, 1).getTime();

//		Region region = AnalysisRegions.CONUS_U3_RELM.load();
//		String title = "ComCat Events, CA, "+startYear+"-"+endYear;
//		String prefix = "ca_comcat_mfd_plot_"+startYear+"_"+endYear;

//		Region region = new CaliforniaRegions.RELM_SOCAL();
//		String title = "ComCat Events, Southern CA, "+startYear+"-"+endYear;
//		String prefix = "socal_comcat_mfd_plot_"+startYear+"_"+endYear;

		Region region = NSHM23_RegionLoader.loadFullConterminousWUS();
		String title = "ComCat Events, Western US, "+startYear+"-"+endYear;
		String prefix = "wus_comcat_mfd_plot_"+startYear+"_"+endYear;

		File outputDir = new File("/tmp");

		double minDepth = -10;
		double maxDepth = 200;

		EvenlyDiscretizedFunc refMFD = FaultSysTools.initEmptyMFD(minMag+0.01, maxMag-0.01);

		ComcatRegion cReg = new ComcatRegionAdapter(region);

		ComcatAccessor access = new ComcatAccessor();

		List<DiscretizedFunc> funcs = new ArrayList<>();
		List<PlotCurveCharacterstics> chars = new ArrayList<>();

		ObsEqkRupList events = access.fetchEventList(null, startDate.getTime(), endDate.getTime(), minDepth, maxDepth,
				cReg, false, false, minMag);

		System.out.println("Loaded "+events.size()+" events");

		IncrementalMagFreqDist mfd = new IncrementalMagFreqDist(refMFD.getMinX(), refMFD.size(), refMFD.getDelta());
		mfd.setName("Incremental");

		for (ObsEqkRupture rup : events)
			mfd.add(mfd.getClosestXIndex(rup.getMag()), 1d);

		funcs.add(mfd);
		chars.add(new PlotCurveCharacterstics(PlotSymbol.FILLED_CIRCLE, 4f, Colors.tab_red));

		EvenlyDiscretizedFunc cmlMFD = mfd.getCumRateDistWithOffset();
		cmlMFD.setName("Cumulative");
		funcs.add(cmlMFD);
		chars.add(new PlotCurveCharacterstics(PlotLineType.SOLID, 2f, Color.BLACK));

		PlotSpec spec = new PlotSpec(funcs, chars, title, "Magnitude", "Event Count");
		spec.setLegendInset(true);

		HeadlessGraphPanel gp = PlotUtils.initScreenHeadless();

		double maxY = cmlMFD.getY(0);

		maxY = Math.pow(10, Math.ceil(Math.log10(maxY)));

		gp.drawGraphPanel(spec, false, true, new Range(minMag, maxMag), new Range(0.9d, maxY));
//		gp.getYAxis().setTickLabelsVisible(false);

		PlotUtils.writePlots(outputDir, prefix, gp, 800, 750, true, true, false);
	}

}