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 export_to_shapefile(self, shpdir, prj=None):
"""
export the model data into a shapefile. element_type dictates which type
of data will be included.
default projection is PA State Plane - untested on other cases
"""
# CREATE THE CONDUIT shp
conds = self.conduits()
conds_path = os.path.join(shpdir, self.inp.name + '_conduits.shp')
spatial.write_shapefile(conds, conds_path, prj=prj)
# CREATE THE NODE shp
nodes = self.nodes()
nodes_path = os.path.join(shpdir, self.inp.name + '_nodes.shp')
spatial.write_shapefile(nodes, nodes_path, geomtype='point', prj=prj)
def export_to_shapefile(self, shpdir, prj=None):
"""
export the model data into a shapefile. element_type dictates which type
of data will be included.
default projection is PA State Plane - untested on other cases
"""
# CREATE THE CONDUIT shp
conds = self.conduits()
conds_path = os.path.join(shpdir, self.inp.name + '_conduits.shp')
spatial.write_shapefile(conds, conds_path, prj=prj)
# CREATE THE NODE shp
nodes = self.nodes()
nodes_path = os.path.join(shpdir, self.inp.name + '_nodes.shp')
spatial.write_shapefile(nodes, nodes_path, geomtype='point', prj=prj)