diff --git a/Analysis/plot_eddy_contours.py b/Analysis/plot_eddy_contours.py index 62d115e66809f8b6753f56082200114f8d326cba..0ad77095ea4ec19d0e4d9d44e33d637cb03c1fac 100755 --- a/Analysis/plot_eddy_contours.py +++ b/Analysis/plot_eddy_contours.py @@ -75,17 +75,21 @@ def snapshot(ax, ishape_list, readers, *, dashed = False, light = False, points = shape_rec_extr.shape.points[0] - if shape_rec_extr.record.cyclone == 0: - # Anti-cyclone - color = "red" + if "cyclone" in shape_rec_extr.record: + if shape_rec_extr.record.cyclone == 0: + # Anti-cyclone + color = "red" + else: + color = "blue" else: - color = "blue" + color = "green" lines = ax.plot(points[0], points[1], markersize = 10, color = color, fillstyle = "none", transform = src_crs) - if shape_rec_extr.record.valid == 1: + if "valid" not in shape_rec_extr.record \ + or shape_rec_extr.record.valid == 1: if light: lines[0].set_marker("+") else: @@ -107,7 +111,8 @@ def snapshot(ax, ishape_list, readers, *, dashed = False, light = False, transform = src_crs) if not light: - if shape_rec_extr.record.valid == 0: + if "valid" in shape_rec_extr.record \ + and shape_rec_extr.record.valid == 0: # Invalid outermost contour lines[0].set_marker("s") lines[0].set_fillstyle("none") @@ -122,11 +127,14 @@ def snapshot(ax, ishape_list, readers, *, dashed = False, light = False, if shape_m_s != None and shape_m_s.shapeType != shapefile.NULL: points = np.array(shape_m_s.points) - if shape_rec_extr.record.cyclone == 0: - # Anti-cyclone - color = "magenta" + if "cyclone" in shape_rec_extr.record: + if shape_rec_extr.record.cyclone == 0: + # Anti-cyclone + color = "magenta" + else: + color = "cyan" else: - color = "cyan" + color = "green" lines = ax.plot(points[:, 0], points[:, 1], color = color, transform = src_crs)