handle directory name changes

[myplc.git] / plc-kml.py

diff --git a/plc-kml.py b/plc-kml.py
index 643fc0d..2a1ecf9 100755 (executable)
--- a/plc-kml.py
+++ b/plc-kml.py
@@ -1,15 +1,23 @@
 #!/usr/bin/env plcsh
+#
+# $Id$
 
 # this script generates a kml file, located under the default location below
 # you should crontab this job from your myplc image
 # you can then use the googlemap.js javascript for creating your applet
 # more on this at http://svn.planet-lab.org/wiki/GooglemapSetup
+# 
+# kml reference can be found at
+# http://code.google.com/apis/kml/documentation/kmlreference.html
+#
 
 import sys
 
-default_output       = "/var/www/html/sites/sites.kml"
-default_local_icon   = "sites/google-local.png"
-default_foreign_icon = "sites/google-foreign.png"
+default_output           = "/var/www/html/sites/sites.kml"
+default_local_icon       = "sites/google-local.png"
+default_foreign_icon     = "sites/google-foreign.png"
+default_local_builtin    = "palette-4.png"
+default_foreign_builtin  = "palette-3.png"
 
 class KmlMap:
 
@@ -28,22 +36,41 @@ class KmlMap:
     def write(self,string):
         self.output.write(string.encode("UTF-8"))
 
+    # mention local last 
+    @staticmethod
+    def site_compare (s1,s2):
+        p1 = p2 = 0
+        if s1['peer_id']: p1=s1['peer_id']
+        if s2['peer_id']: p2=s2['peer_id']
+        return p2-p1
+
     def refresh (self):
         self.open()
         self.write_header()
         # cache peers 
         peers = GetPeers({},['peer_id','peername'])
-        for site in GetSites({'enabled':True,'is_public':True}):
+        all_sites = GetSites({'enabled':True,'is_public':True})
+        all_sites.sort(KmlMap.site_compare)
+        for site in all_sites:
             self.write_site(site,peers)
         self.write_footer()
         self.close()
 
+# initial placement is for europe - dunno how to tune that yet
     def write_header (self):
         self.write("""<?xml version="1.0" encoding="UTF-8"?>
 <kml xmlns="http://earth.google.com/kml/2.2">
 <Document>
 <name> PlanetLab Sites </name>
-<description> This map shows all sites knows to the PlanetLab testbed. </description>
+<LookAt>
+<longitude>9.180821112577378</longitude>
+<latitude>44.43275321178062</latitude>
+<altitude>0</altitude>
+<range>5782133.196489797</range>
+<tilt>0</tilt>
+<heading>-7.767386340832667</heading>
+</LookAt>
+<description> All the sites known to the PlanetLab testbed. </description>
 """)
 
     def write_footer (self):
@@ -106,15 +133,19 @@ class KmlMap:
         # close description
         description +='</ul>'
 
+        if not self.options.labels:
+            name=""
+            description=""
+
         # STYLE
         if self.options.use_google_icons:
             if not peer_id:
                 # local sites
-                iconfile="palette-4.png"
+                iconfile=default_local_builtin
                 xyspec="<x>128</x><y>0</y><w>32</w><h>32</h>"
             else:
                 # remote
-                iconfile="palette-3.png"
+                iconfile=default_foreign_builtin
                 xyspec="<x>160</x><y>0</y><w>32</w><h>32</h>"
             iconurl="root://icons/%(iconfile)s"%locals()
         else:
@@ -127,9 +158,17 @@ class KmlMap:
 
         iconspec="<href>%(iconurl)s</href>%(xyspec)s"%locals()
 
+        # set the camera 50km high
         template="""<Placemark>
 <Style><IconStyle><Icon>%(iconspec)s</Icon></IconStyle></Style>
 <name><![CDATA[%(name)s]]></name>
+<LookAt>
+  <latitude>%(latitude)f</latitude>
+  <longitude>%(longitude)f</longitude>
+  <altitude>0</altitude>
+  <altitudeMode>relativeToGround</altitudeMode>              
+  <range>50000.</range> 
+</LookAt>
 <description><![CDATA[%(description)s]]></description>
 <Point> <coordinates>%(longitude)f,%(latitude)f,0</coordinates> </Point>
 </Placemark>
@@ -144,6 +183,9 @@ def main () :
     parser.add_option("-o","--output",action="store",dest="output",
                       default=default_output,
                       help="output file - default is %s"%default_output)
+    parser.add_option("-n","--no-label",action="store_false",dest="labels",
+                      default=True,
+                      help="outputs only geographic positions, no labels")
     parser.add_option("-c","--custom",action="store_false",dest="use_google_icons",
                       default=True,
                       help="use locally customized icons rather than the google-provided defaults")
@@ -153,6 +195,7 @@ def main () :
     parser.add_option("-f","--foreign",action="store",dest="foreign_icon",
                       default=default_foreign_icon,
                       help="set icon url to use for foreign sites marker -- default is %s"%default_foreign_icon)
+
     (options, args) = parser.parse_args()
     if len(args) != 0:
         parser.print_help()