Right-click menu for editing planetary events, and loading/saving them

Author: Akjosch

Diff for: MekHQ/src/mekhq/campaign/Campaign.java

@@ -3351,6 +3351,26 @@ public void writeToXml(PrintWriter pw1) {
             }
             retirementDefectionTracker.writeToXml(pw1, 1);
         }
+        
+        // Customised planetary events
+        pw1.println("\t<customPlanetaryEvents>");
+        for(Planet p : Planets.getInstance().getPlanets().values()) {
+            List<Planet.PlanetaryEvent> customEvents = new ArrayList<>();
+            for(Planet.PlanetaryEvent event : p.getEvents()) {
+                if(event.custom) {
+                    customEvents.add(event);
+                }
+            }
+            if(!customEvents.isEmpty()) {
+                pw1.println("\t\t<planet><id>" + p.getId() + "</id>");
+                for(Planet.PlanetaryEvent event : customEvents) {
+                    Planets.getInstance().writePlanetaryEvent(pw1, event);
+                    pw1.println();
+                }
+                pw1.println("\t\t</planet>");
+            }
+        }
+        pw1.println("\t</customPlanetaryEvents>");
 
         writeCustoms(pw1);
         // Okay, we're done.
@@ -3643,6 +3663,8 @@ else if (xn.equalsIgnoreCase("info")) {
                 	processLanceNodes(retVal, wn);
                 } else if (xn.equalsIgnoreCase("retirementDefectionTracker")) {
                 	retVal.retirementDefectionTracker = RetirementDefectionTracker.generateInstanceFromXML(wn, retVal);
+                } else if (xn.equalsIgnoreCase("customPlanetaryEvents")) {
+                    updatePlanetaryEventsFromXML(wn);
                 }
 
             } else {
@@ -4060,6 +4082,43 @@ else if (xn.equalsIgnoreCase("info")) {
         return retVal;
     }
 
+    private static void updatePlanetaryEventsFromXML(Node wn) {
+        Planets.reload(true);
+        NodeList wList = wn.getChildNodes();
+        for (int x = 0; x < wList.getLength(); x++) {
+            Node wn2 = wList.item(x);
+
+            // If it's not an element node, we ignore it.
+            if (wn2.getNodeType() != Node.ELEMENT_NODE) {
+                continue;
+            }
+
+            if (wn2.getNodeName().equalsIgnoreCase("planet")) {
+                NodeList planetNodes = wn2.getChildNodes();
+                String planetId = null;
+                List<Planet.PlanetaryEvent> events = new ArrayList<>();
+                for(int n = 0; n < planetNodes.getLength(); ++ n) {
+                    Node planetNode = planetNodes.item(n);
+                    if(planetNode.getNodeType() != Node.ELEMENT_NODE) {
+                        continue;
+                    }
+                    if(planetNode.getNodeName().equalsIgnoreCase("id")) {
+                        planetId = planetNode.getTextContent();
+                    } else if(planetNode.getNodeName().equalsIgnoreCase("event")) {
+                        Planet.PlanetaryEvent event = Planets.getInstance().readPlanetaryEvent(planetNode);
+                        if(null != event) {
+                            event.custom = true;
+                            events.add(event);
+                        }
+                    }
+                }
+                if(null != planetId) {
+                    Planets.getInstance().updatePlanetaryEvents(planetId, events, true);
+                }
+            }
+        }
+    }
+
     private static void fixIdReferences(Campaign retVal) {
         // set up translation hashes
         Hashtable<Integer, UUID> uHash = new Hashtable<Integer, UUID>();

Diff for: MekHQ/src/mekhq/gui/InterstellarMapPanel.java

@@ -20,6 +20,7 @@
 import java.awt.geom.Arc2D;
 import java.awt.geom.Line2D;
 import java.util.ArrayList;
+import java.util.List;
 import java.util.Set;
 
 import javax.swing.BorderFactory;
@@ -33,6 +34,8 @@
 import mekhq.campaign.JumpPath;
 import mekhq.campaign.universe.Faction;
 import mekhq.campaign.universe.Planet;
+import mekhq.campaign.universe.Planets;
+import mekhq.gui.dialog.NewPlanetaryEventDialog;
 
 
 /**
@@ -184,6 +187,18 @@ public void actionPerformed(ActionEvent ae) {
                         });
                     }
                     menuGM.add(item);
+                    item = new JMenuItem("Edit planetary events");
+                    item.setEnabled(selectedPlanet != null && campaign.isGM());
+                    if (selectedPlanet != null) {
+                        item.setText("Edit planetary events for " + selectedPlanet.getPrintableName(new DateTime(campaign.getCalendar())));
+                        item.addActionListener(new ActionListener() {
+                            @Override
+                            public void actionPerformed(ActionEvent ae) {
+                                openPlanetEventEditor(selectedPlanet);
+                            }
+                        });
+                    }
+                    menuGM.add(item);
                     popup.add(menuGM);
                 	popup.show(e.getComponent(), e.getX() + 10, e.getY() + 10);
                 }
@@ -495,6 +510,17 @@ private void changeSelectedPlanet(Planet p) {
     	hqview.refreshPlanetView();
     }
 
+    private void openPlanetEventEditor(Planet p) {
+        NewPlanetaryEventDialog editor = new NewPlanetaryEventDialog(null, campaign, selectedPlanet);
+        editor.setVisible(true);
+        List<Planet.PlanetaryEvent> result = editor.getChangedEvents();
+        if((null != result) && !result.isEmpty()) {
+            Planets.getInstance().updatePlanetaryEvents(p.getId(), result, true);
+            repaint();
+            hqview.refreshPlanetView();
+        }
+
+    }
     /**
      * All configuration behaviour of InterStellarMap are saved here.
      *