From cded5612d60c93930dc26843bfa3c589450c1c5b Mon Sep 17 00:00:00 2001
From: Kevin Milner <kmilner@usc.edu>
Date: Thu, 14 Mar 2024 09:34:35 -0700
Subject: [PATCH] new merged solution util

---
 .../util/MergedSolutionCreator.java           | 110 ++++++++++++++++++
 1 file changed, 110 insertions(+)
 create mode 100644 src/main/java/org/opensha/sha/earthquake/faultSysSolution/util/MergedSolutionCreator.java

diff --git a/src/main/java/org/opensha/sha/earthquake/faultSysSolution/util/MergedSolutionCreator.java b/src/main/java/org/opensha/sha/earthquake/faultSysSolution/util/MergedSolutionCreator.java
new file mode 100644
index 000000000..ae0d650c7
--- /dev/null
+++ b/src/main/java/org/opensha/sha/earthquake/faultSysSolution/util/MergedSolutionCreator.java
@@ -0,0 +1,110 @@
+package org.opensha.sha.earthquake.faultSysSolution.util;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+import org.opensha.sha.earthquake.faultSysSolution.FaultSystemRupSet;
+import org.opensha.sha.earthquake.faultSysSolution.FaultSystemSolution;
+import org.opensha.sha.earthquake.faultSysSolution.modules.GridSourceProvider;
+import org.opensha.sha.faultSurface.FaultSection;
+
+import com.google.common.base.Preconditions;
+
+public class MergedSolutionCreator {
+	
+	/**
+	 * Mrege the given list of fault system solutions into a single solution. Does not combine any modules, and does
+	 * not yet merge gridded seismicity (TODO)
+	 * 
+	 * @param sols
+	 * @return
+	 */
+	public FaultSystemSolution merge(List<FaultSystemSolution> sols) {
+		return merge(sols.toArray(new FaultSystemSolution[0]));
+	}
+	
+	/**
+	 * Mrege the given list of fault system solutions into a single solution. Does not combine any modules, and does
+	 * not yet merge gridded seismicity (TODO)
+	 * 
+	 * @param sols
+	 * @return
+	 */
+	public FaultSystemSolution merge(FaultSystemSolution... sols) {
+		Preconditions.checkState(sols.length > 1, "Need at least 2 solutions to merge");
+		
+		int totNumSects = 0;
+		int totNumRups = 0;
+		for (int i=0; i<sols.length; i++) {
+			FaultSystemSolution sol = sols[i];
+			FaultSystemRupSet rupSet = sol.getRupSet();
+			System.out.println("RupSet "+i+" has "+rupSet.getNumSections()+" sects, "+rupSet.getNumRuptures()+" rups");
+			totNumSects += rupSet.getNumSections();
+			totNumRups += rupSet.getNumRuptures();
+			if (sol.hasAvailableModule(GridSourceProvider.class))
+				System.err.println("WARNING: this does not yet merge grid source providers"); // TODO
+		}
+		System.out.println("Total: "+totNumSects+" sects, "+totNumRups+" rups");
+		
+		List<FaultSection> mergedSects = new ArrayList<>(totNumSects);
+		List<List<Integer>> sectionForRups = new ArrayList<>(totNumSects);
+		double[] mags = new double[totNumRups];
+		double[] rakes = new double[totNumRups];
+		double[] rupAreas = new double[totNumRups];
+		double[] rupLengths = new double[totNumRups];
+		double[] rates = new double[totNumRups];
+		
+		int sectIndex = 0;
+		int rupIndex = 0;
+		
+		Map<Integer, String> prevParents = new HashMap<>();
+		
+		for (FaultSystemSolution sol : sols) {
+			FaultSystemRupSet rupSet = sol.getRupSet();
+			int[] sectMappings = new int[rupSet.getNumSections()];
+			System.out.println("Merging sol with "+rupSet.getNumSections()+" sects and "+rupSet.getNumRuptures()+" rups");
+			
+			Map<Integer, String> newParents = new HashMap<>();
+			for (int s=0; s<sectMappings.length; s++) {
+				FaultSection sect = rupSet.getFaultSectionData(s);
+				if (sect.getParentSectionId() >= 0)
+					newParents.put(sect.getParentSectionId(), sect.getParentSectionName());
+				sect = sect.clone();
+				sectMappings[s] = sectIndex;
+				sect.setSectionId(sectIndex);
+				mergedSects.add(sect);
+				
+				sectIndex++;
+			}
+			// see if there are any duplicate parent IDs
+			for (int parentID : newParents.keySet()) {
+				if (prevParents.containsKey(parentID))
+					System.err.println("WARNING: multiple solutions use the same parent section id ("+parentID+"): "
+							+prevParents.get(parentID)+" and "+newParents.get(parentID));
+				else
+					prevParents.put(parentID, newParents.get(parentID));
+			}
+			
+			for (int r=0; r<rupSet.getNumRuptures(); r++) {
+				List<Integer> prevSectIDs = rupSet.getSectionsIndicesForRup(r);
+				List<Integer> newSectIDs = new ArrayList<>(prevSectIDs.size());
+				for (int s : prevSectIDs)
+					newSectIDs.add(sectMappings[s]);
+				sectionForRups.add(newSectIDs);
+				mags[rupIndex] = rupSet.getMagForRup(r);
+				rakes[rupIndex] = rupSet.getAveRakeForRup(r);
+				rupAreas[rupIndex] = rupSet.getAreaForRup(r);
+				rupAreas[rupIndex] = rupSet.getLengthForRup(r);
+				rates[rupIndex] = sol.getRateForRup(r);
+				
+				rupIndex++;
+			}
+		}
+		
+		FaultSystemRupSet mergedRupSet = new FaultSystemRupSet(mergedSects, sectionForRups, mags, rakes, rupAreas, rupLengths);
+		return new FaultSystemSolution(mergedRupSet, rates);
+	}
+
+}