def encode_report(rpt, rpt_path):
rpt_dict = {}
#write parcel json files
parcels = spatial.read_shapefile(sg.config.parcels_shapefile)
parcels = parcels[['PARCELID', 'coords']] #omit 'ADDRESS', 'OWNER1'
flooded = rpt.alt_report.parcel_flooding #proposed flooding condition
flooded = pd.merge(flooded, parcels, right_on='PARCELID', left_index=True)
rpt_dict['parcels'] = spatial.write_geojson(flooded, geomtype='polygon')
#non null delta category parcels
delta_parcels = rpt.flood_comparison.loc[pd.notnull(
rpt.flood_comparison.Category)]
delta_parcels = pd.merge(delta_parcels,
parcels,
right_on='PARCELID',
left_index=True)
rpt_dict['delta_parcels'] = spatial.write_geojson(delta_parcels,
geomtype='polygon')
#encode conduit and nodes data into geojson
# rpt_dict['conduits'] = spatial.write_geojson(rpt.alt_report.model.conduits())
rpt_dict['new_conduits'] = spatial.write_geojson(rpt.newconduits)
# rpt_dict['nodes'] = spatial.write_geojson(rpt.model.nodes(), geomtype='point')
#write summary stats
rpt_dict.update(rpt.summary_dict)
with open(rpt_path, 'w') as f:
f.write(json.dumps(rpt_dict))
def encode_report(rpt, rpt_path):
rpt_dict = {}
#write parcel json files
parcels = spatial.read_shapefile(sg.config.parcels_shapefile)
parcels = parcels[['PARCELID', 'coords']] #omit 'ADDRESS', 'OWNER1'
flooded = rpt.alt_report.parcel_flooding #proposed flooding condition
flooded = pd.merge(flooded, parcels, right_on='PARCELID', left_index=True)
rpt_dict['parcels'] = spatial.write_geojson(flooded, geomtype='polygon')
#non null delta category parcels
delta_parcels = rpt.flood_comparison.loc[pd.notnull(rpt.flood_comparison.Category)]
delta_parcels = pd.merge(delta_parcels, parcels, right_on='PARCELID', left_index=True)
rpt_dict['delta_parcels'] = spatial.write_geojson(delta_parcels, geomtype='polygon')
#encode conduit and nodes data into geojson
# rpt_dict['conduits'] = spatial.write_geojson(rpt.alt_report.model.conduits())
rpt_dict['new_conduits'] = spatial.write_geojson(rpt.newconduits)
# rpt_dict['nodes'] = spatial.write_geojson(rpt.model.nodes(), geomtype='point')
#write summary stats
rpt_dict.update(rpt.summary_dict)
with open(rpt_path, 'w') as f:
f.write(json.dumps(rpt_dict))
def write(self, rpt_dir):
#write cost per sewer segment spreadsheet
self.newconduits.to_csv(os.path.join(rpt_dir,'cost_estimate.csv'))
self.flood_comparison.to_csv(os.path.join(rpt_dir,'parcel_flood_comparison.csv'))
#write parcel json files
parcels = spatial.read_shapefile(sg.config.parcels_shapefile)
parcels = parcels[['PARCELID', 'ADDRESS', 'OWNER1', 'coords']]
flooded = self.flood_comparison
flooded = flooded.loc[flooded.Category.notnull()] #parcels with significant flood delta
flooded = pd.merge(flooded, parcels, right_on='PARCELID', left_index=True)
colors = flooded.apply(lambda row:'#%02x%02x%02x' % drawing.parcel_draw_color(row, style='delta'), axis=1)
flooded = flooded.assign(fill=colors)
geoparcelpath = os.path.join(rpt_dir,'delta_parcels.json')
spatial.write_geojson(flooded, filename=geoparcelpath, geomtype='polygon')
#write new conduit json, shapefiles
shpdir = os.path.join(os.path.dirname(rpt_dir), 'shapefiles')
if not os.path.exists(shpdir):os.mkdir(shpdir)
geocondpath = os.path.join(rpt_dir,'new_conduits.json')
shpcondpath = os.path.join(shpdir, self.alt_report.model.inp.name + '_new_conduits.shp')
spatial.write_geojson(self.newconduits, filename=geocondpath)
spatial.write_shapefile(self.newconduits, filename=shpcondpath)
#write node and conduit report csvs
self.alt_report.model.nodes().to_csv(os.path.join(rpt_dir,'nodes.csv'))
self.alt_report.model.conduits().to_csv(os.path.join(rpt_dir,'conduits.csv'))
#write a html map
with open (geocondpath, 'r') as f:
geo_conduits = geojson.loads(f.read())
proposed_flooded = self.alt_report.parcel_flooding
proposed_flooded = pd.merge(proposed_flooded, parcels, right_on='PARCELID', left_index=True)
geo_parcels = spatial.write_geojson(proposed_flooded)
# with open (geoparcelpath, 'r') as f:
# geo_parcels = geojson.loads(f.read())
with open(BETTER_BASEMAP_PATH, 'r') as bm:
filename = os.path.join(os.path.dirname(geocondpath), self.alt_report.model.name + '.html')
with open(filename, 'wb') as newmap:
for line in bm:
if '//INSERT GEOJSON HERE ~~~~~' in line:
newmap.write('conduits = {};\n'.format(geojson.dumps(geo_conduits)))
newmap.write('nodes = {};\n'.format(0))
newmap.write('parcels = {};\n'.format(geojson.dumps(geo_parcels)))
else:
newmap.write(line)
def write(self, rpt_dir):
#write cost per sewer segment spreadsheet
self.newconduits.to_csv(os.path.join(rpt_dir,'cost_estimate.csv'))
self.flood_comparison.to_csv(os.path.join(rpt_dir,'parcel_flood_comparison.csv'))
#write parcel json files
parcels = spatial.read_shapefile(sg.config.parcels_shapefile)
parcels = parcels[['PARCELID', 'ADDRESS', 'OWNER1', 'coords']]
flooded = self.flood_comparison
flooded = flooded.loc[flooded.Category.notnull()] #parcels with significant flood delta
flooded = pd.merge(flooded, parcels, right_on='PARCELID', left_index=True)
colors = flooded.apply(lambda row:'#%02x%02x%02x' % drawing.parcel_draw_color(row, style='delta'), axis=1)
flooded = flooded.assign(fill=colors)
geoparcelpath = os.path.join(rpt_dir,'delta_parcels.json')
spatial.write_geojson(flooded, filename=geoparcelpath, geomtype='polygon')
#write new conduit json, shapefiles
shpdir = os.path.join(os.path.dirname(rpt_dir), 'shapefiles')
if not os.path.exists(shpdir):os.mkdir(shpdir)
geocondpath = os.path.join(rpt_dir,'new_conduits.json')
shpcondpath = os.path.join(shpdir, self.alt_report.model.inp.name + '_new_conduits.shp')
spatial.write_geojson(self.newconduits, filename=geocondpath)
spatial.write_shapefile(self.newconduits, filename=shpcondpath)
#write node and conduit report csvs
self.alt_report.model.nodes().to_csv(os.path.join(rpt_dir,'nodes.csv'))
self.alt_report.model.conduits().to_csv(os.path.join(rpt_dir,'conduits.csv'))
#write a html map
with open (geocondpath, 'r') as f:
geo_conduits = geojson.loads(f.read())
proposed_flooded = self.alt_report.parcel_flooding
proposed_flooded = pd.merge(proposed_flooded, parcels, right_on='PARCELID', left_index=True)
geo_parcels = spatial.write_geojson(proposed_flooded)
# with open (geoparcelpath, 'r') as f:
# geo_parcels = geojson.loads(f.read())
with open(BETTER_BASEMAP_PATH, 'r') as bm:
filename = os.path.join(os.path.dirname(geocondpath), self.alt_report.model.name + '.html')
with open(filename, 'wb') as newmap:
for line in bm:
if '//INSERT GEOJSON HERE ~~~~~' in line:
newmap.write('conduits = {};\n'.format(geojson.dumps(geo_conduits)))
newmap.write('nodes = {};\n'.format(0))
newmap.write('parcels = {};\n'.format(geojson.dumps(geo_parcels)))
else:
newmap.write(line)
def geojson(self):
"""
Return a GeoJSON representation of the group
:return: GeoJSON string
"""
return write_geojson(self.dataframe, geomtype=self.geomtype)