def angle_heapmaps(description, zextractor):
configuration_names = Configuration.names()
configurations = [Configuration.create_specific(name, 11, 4.5, "topology") for name in Configuration.names()]
#Get rid of configurations that aren't grids
configurations = filter(lambda c: isinstance(c.topology, Grid), configurations)
# Get rid of configurations that do not have 2 sources
configurations = filter(lambda c: len(c.source_ids) == 2, configurations)
grapher = configuration_heatmap.Grapher("results/Configurations", description, zextractor)
grapher.nokey = True
grapher.xaxis_label = "X Coordinate"
grapher.yaxis_label = "Y Coordinate"
grapher.cb_label = "Source Angle"
grapher.dgrid3d_dimensions = "11,11"
grapher.cbrange = (0, 180)
grapher.create(configurations)
summary_grapher = summary.GraphSummary(
os.path.join("results/Configurations", description),
'{}-{}'.format("results/Configurations", description)
)
summary_grapher.width_factor = None
summary_grapher.height = "7cm"
summary_grapher.run()
import simulator.Configuration as Configuration
from simulator.Topology import Grid
from data.graph import summary, configuration_heatmap
parser = argparse.ArgumentParser(description="Misc. Grapher", add_help=True)
parser.add_argument("--min-source-distance-heatmap", action='store_true', default=False)
parser.add_argument("--source-angle-estimate-heatmap", action='store_true', default=False)
parser.add_argument("--source-angle-meters-heatmap", action='store_true', default=False)
parser.add_argument("--source-angle-actual-heatmap", action='store_true', default=False)
parser.add_argument("--source-angle-det-heatmap", action='store_true', default=False)
parser.add_argument("--source-angle-tan-heatmap", action='store_true', default=False)
args = parser.parse_args(sys.argv[1:])
if args.min_source_distance_heatmap:
configuration_names = Configuration.names()
configurations = [Configuration.create_specific(name, 11, 4.5, "topology") for name in Configuration.names()]
#Get rid of configurations that aren't grids
configurations = filter(lambda c: isinstance(c.topology, Grid), configurations)
def zextractor(configuration, nid):
return min(configuration.node_source_distance(nid, src_id) for src_id in configuration.source_ids)
grapher = configuration_heatmap.Grapher("results/Configurations", "min-src-distance", zextractor)
grapher.nokey = True
grapher.xaxis_label = "X Coordinate"
grapher.yaxis_label = "Y Coordinate"
grapher.cb_label = "Minimum Source Distance (m)"