From 17a0f12203aaa644c2a5733f342ed2883d9664d7 Mon Sep 17 00:00:00 2001
From: Angel Farguell Caus <angel.farguell@gmail.com>
Date: Wed, 27 May 2020 10:38:03 -0600
Subject: [PATCH] adding bounding box creation in fire detections KML file

---
 JPSSD.py       | 33 ++++++++++++++++++++---
 contline.py    | 84 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 contlinesvm.py |  2 +-
 contour2kml.py | 12 +++------
 4 files changed, 118 insertions(+), 13 deletions(-)

diff --git a/JPSSD.py b/JPSSD.py
index 8a8c6a8..7175a62 100644
--- a/JPSSD.py
+++ b/JPSSD.py
@@ -1135,6 +1135,32 @@ def json2kml(d,kml_path,bounds,prods,opt='granule',minconf=80.):
 
             copyto('kmls/partial1.kml',kml)
 
+            # write bounding box
+            kml.write('<Folder>\n'
+                +'<name>Bounding Box</name>\n'
+                +'<Placemark>\n'
+                +'<name>Bounds</name>\n'
+                +'<description>Bounding box of the fire detections</description>\n'
+                +'<Style>\n'
+                +'<LabelStyle><color>00000000</color><scale>0.000000</scale></LabelStyle>\n'
+                +'<LineStyle><color>ff0000e6</color><width>4.000000</width></LineStyle>\n'
+                +'<PolyStyle><color>00f0f0f0</color><outline>1</outline></PolyStyle>\n'
+                +'</Style>\n'
+                +'<MultiGeometry>\n'
+                +'<Polygon>\n'
+                +'<extrude>0</extrude><altitudeMode>clampToGround</altitudeMode>\n'
+                +'<outerBoundaryIs><LinearRing><coordinates>\n'
+                +'%s,%s,0\n' % (bounds[0],bounds[2])
+                +'%s,%s,0\n' % (bounds[1],bounds[2])
+                +'%s,%s,0\n' % (bounds[1],bounds[3])
+                +'%s,%s,0\n' % (bounds[0],bounds[3])
+                +'%s,%s,0\n' % (bounds[0],bounds[2])
+                +'</coordinates></LinearRing></outerBoundaryIs>\n'
+                +'</Polygon>\n'
+                +'</MultiGeometry>\n'
+                +'</Placemark>\n'
+                +'</Folder>\n')
+
             # set some constants
             r = 6378   # Earth radius
             km_lat = 180/(np.pi*r)  # 1km in degrees latitude
@@ -1265,17 +1291,17 @@ def json2kml(d,kml_path,bounds,prods,opt='granule',minconf=80.):
                                     if conf < 80:
                                         kml.write('<Style>\n'+'<PolyStyle>\n'
                                                 +'<color>7000ffff</color>\n'
-                                                +'<outline>0</outline>\n'+'</PolyStyle>\n'
+                                                +'<outline>1</outline>\n'+'</PolyStyle>\n'
                                                 +'</Style>\n')
                                     elif conf < 90:
                                         kml.write('<Style>\n'+'<PolyStyle>\n'
                                                 +'<color>7000a5ff</color>\n'
-                                                +'<outline>0</outline>\n'+'</PolyStyle>\n'
+                                                +'<outline>1</outline>\n'+'</PolyStyle>\n'
                                                 +'</Style>\n')
                                     else:
                                         kml.write('<Style>\n'+'<PolyStyle>\n'
                                                 +'<color>700000ff</color>\n'
-                                                +'<outline>0</outline>\n'+'</PolyStyle>\n'
+                                                +'<outline>1</outline>\n'+'</PolyStyle>\n'
                                                 +'</Style>\n')
 
                                 kml.write('<Polygon>\n<outerBoundaryIs>\n<LinearRing>\n<coordinates>\n')
@@ -1297,7 +1323,6 @@ def json2kml(d,kml_path,bounds,prods,opt='granule',minconf=80.):
                         kml.write('</Folder>\n')
 
                 kml.write('</Folder>\n')
-
             kml.write('</Document>\n</kml>\n')
 
         print 'Created file %s' % kml_path
diff --git a/contline.py b/contline.py
index 37087a5..1f13a2d 100644
--- a/contline.py
+++ b/contline.py
@@ -9,7 +9,90 @@ from JPSSD import time_num2iso
 from contour2kml import contour2kml
 from scipy.ndimage import gaussian_filter
 
+def clean_polys(paths,bounds,plot=True):
+    npaths = []
+    maxd = .05
+    eps = 1e-5
+    dx = abs(bounds[1]-bounds[0])
+    dy = abs(bounds[3]-bounds[2])
+    xb = np.array(bounds[:2])
+    yb = np.array(bounds[2:])
+    for p in paths:
+        diff = np.abs(p[-1]-p[0])
+        if (diff>maxd).all() or (abs(diff[0]-dx)<eps or abs(diff[1]-dy)<eps):
+            x = [c[0] for c in p]
+            y = [c[1] for c in p]
+            xm = (x[0]+x[-1])/2
+            ym = (y[0]+y[-1])/2
+            xn = x[:]
+            yn = y[:]
+            x.append(p[0][0])
+            y.append(p[0][1])
+
+            if abs(diff[0]-dx)<eps:
+                ex = xn[-1]
+                ey = yb[np.argmin(abs(yb-xn[-1]))]
+                p = np.append(p,np.array([[ex,ey]]),axis=0)
+                ex = xn[0]
+                p = np.append(p,np.array([[ex,ey]]),axis=0)
+                #p = np.append(p,np.array([p[0]]),axis=0)
+                if False:
+                    plt.ion()
+                    plt.plot([xb[0],xb[0],xb[1],xb[1],xb[0]],[yb[0],yb[1],yb[1],yb[0],yb[0]],'k')
+                    plt.plot(x,y,'r')
+                    x = [c[0] for c in p]
+                    y = [c[1] for c in p]
+                    plt.plot(x,y,'g')
+                    plt.xlim(xb)
+                    plt.ylim(yb)
+                    plt.show()
+                    plt.pause(.001)
+                    plt.cla()
+            elif abs(diff[1]-dy)<eps:
+                #ex = xb[np.argmin(abs(xb-xn[-1]))]
+                ex = xb[0]
+                ey = yn[-1]
+                p = np.append(p,np.array([[ex,ey]]),axis=0)
+                ey = yn[0]
+                p = np.append(p,np.array([[ex,ey]]),axis=0)
+                #p = np.append(p,np.array([p[0]]),axis=0)
+                if plot:
+                    #plt.ion()
+                    plt.plot([xb[0],xb[0],xb[1],xb[1],xb[0]],[yb[0],yb[1],yb[1],yb[0],yb[0]],'k')
+                    plt.plot(x,y,'r')
+                    x = [c[0] for c in p]
+                    y = [c[1] for c in p]
+                    plt.plot(x,y,'g')
+                    plt.xlim(xb)
+                    plt.ylim(yb)
+                    plt.show()
+                    #plt.pause(.001)
+                    #plt.cla()
+            else:
+                ex = xb[np.argmin(abs(xm-xb))]
+                ey = yb[np.argmin(abs(ym-yb))]
+                p = np.append(p,np.array([[ex,ey]]),axis=0)
+                #p = np.append(p,np.array([p[0]]),axis=0)
+                if False:
+                    plt.ion()
+                    plt.plot([xb[0],xb[0],xb[1],xb[1],xb[0]],[yb[0],yb[1],yb[1],yb[0],yb[0]],'k')
+                    plt.plot(x,y,'r')
+                    x = [c[0] for c in p]
+                    y = [c[1] for c in p]
+                    plt.plot(xn,yn,'g')
+                    plt.xlim(xb)
+                    plt.ylim(yb)
+                    plt.show()
+                    plt.pause(.001)
+                    plt.cla()
+
+        else:
+            npaths.append(p)
+    return np.array(npaths)
+
 def get_contour_verts(xx, yy, zz, time_num_granules, contour_dt_hours=6, contour_dt_init=6, contour_dt_final=6, gauss_filter=True, plot_contours=False, col_repr=False, levels_gran=False):
+    bounds = (xx.min(),xx.max(),yy.min(),yy.max())
+    print bounds
     fig = plt.figure()
     # Computing the levels
     # Datetimes for the first and last granule
@@ -57,6 +140,7 @@ def get_contour_verts(xx, yy, zz, time_num_granules, contour_dt_hours=6, contour
                 # read all the vertices
                 paths.append(pp.vertices)
             contours.append(paths)
+            #contours.append(clean_polys(paths,bounds))
     else:
         # computing all the contours
         cn = plt.contour(xx,yy,zz,levels=levels)
diff --git a/contlinesvm.py b/contlinesvm.py
index 01cf857..20e1c88 100644
--- a/contlinesvm.py
+++ b/contlinesvm.py
@@ -25,7 +25,7 @@ zz=np.array(svm['Z'])*tscale+time_scale_num[0]
 print 'Computing the contours...'
 # Granules numeric times
 time_num_granules = svm['time_num_granules'][0]
-data = get_contour_verts(xx, yy, zz, time_num_granules, contour_dt_hours=6, contour_dt_init=24, contour_dt_final=12)
+data = get_contour_verts(xx, yy, zz, time_num_granules, contour_dt_hours=6, contour_dt_init=6, contour_dt_final=6)
 
 print 'Creating the KML file...'
 # Creating the KML file
diff --git a/contour2kml.py b/contour2kml.py
index b890926..cff5138 100644
--- a/contour2kml.py
+++ b/contour2kml.py
@@ -45,17 +45,13 @@ def contour2kml(data,kml_path):
                time_begin, time_begin, time_begin,idx))
            for polygon in c['polygons']:
                kml.write("""
-                                        <Polygon>
-                                                <outerBoundaryIs>
-                                                        <LinearRing>
-                                                                <coordinates>""")
+                                        <LineString>
+                                                  <coordinates>""")
                for segment in polygon:
                    kml.write("\n%s,%s,0" % tuple(segment))
                kml.write("""
-                                                                </coordinates>
-                                                        </LinearRing>
-                                                </outerBoundaryIs>
-                                        </Polygon>""")
+                                                  </coordinates>
+                                        </LineString>""")
            kml.write("""
                                 </MultiGeometry>
                         </Placemark>
-- 
2.11.4.GIT