def dp_dist_stats(config, parser): #@UnusedVariable
""" Computes statistics for images distances for different plan steps. """
parser.add_option("-o",
"--output",
default='out/dp-dist-stats',
help="Output directory")
parser.add_option("-d",
"--distances",
default='*',
help="Comma-separated list of distances. Can use *.")
parser.add_option("-s",
"--streams",
default='*',
help="Comma-separated list of streams. Can use *.")
parser.add_option("-r",
"--repeat",
default=1,
type='int',
help="Repeat many times.")
parser.add_option("-c",
"--command",
help="Command to pass to compmake for batch mode")
options = parser.parse_options()
distances = natsorted(config.distances.expand_names(options.distances))
streams = natsorted(config.streams.expand_names(options.streams))
logger.info('Using distances: %s' % distances)
logger.info('Using streams: %s' % streams)
id_comb = ','.join(streams) + '-' + ','.join(distances)
outdir = os.path.join(options.output, id_comb)
storage = os.path.join(outdir, 'compmake')
use_filesystem(storage)
read_rc_files()
rm = ReportManager(os.path.join(outdir, "reports"))
create_diststats_jobs(config=config,
distances=distances,
streams=streams,
rm=rm,
maxd=10)
rm.create_index_job()
if options.command:
return batch_command(options.command)
else:
compmake_console()
return 0
def define_jobs_context(self, context):
distances_library = get_conftools_uncertain_image_distances()
distances = distances_library.expand_names(self.options.distances)
distances = natsorted(distances)
streams_library = get_conftools_streams()
streams = streams_library.expand_names(self.options.streams)
streams = natsorted(streams)
# id_comb = ','.join(streams) + '-' + ','.join(distances)
create_diststats_jobs(context, distances=distances, streams=streams,
maxd=self.options.maxd)
def define_jobs_context(self, context):
distances_library = get_conftools_uncertain_image_distances()
distances = distances_library.expand_names(self.options.distances)
distances = natsorted(distances)
streams_library = get_conftools_streams()
streams = streams_library.expand_names(self.options.streams)
streams = natsorted(streams)
# id_comb = ','.join(streams) + '-' + ','.join(distances)
create_diststats_jobs(context,
distances=distances,
streams=streams,
maxd=self.options.maxd)
def get_all_episodes(self):
""" Returns the list of all episodes, both the already present
and the simulated. """
eps = []
eps.extend(self.id_episodes)
eps.extend(self.done_before)
return natsorted(set(eps))
def get_episodes_for_robot(self, id_robot, id_agent=None):
''' Returns a list of all episodes for the given robot (and
agent if it is given). ''' # TODO: implement this
episodes = []
for stream in self.get_streams_for_robot(id_robot):
if not id_agent:
episodes.extend(stream.get_id_episodes())
else:
if id_agent in stream.get_id_agents():
episodes.extend(stream.get_id_episodes())
return natsorted(episodes)
def dp_dist_stats(config, parser): #@UnusedVariable
""" Computes statistics for images distances for different plan steps. """
parser.add_option("-o", "--output", default='out/dp-dist-stats',
help="Output directory")
parser.add_option("-d", "--distances", default='*',
help="Comma-separated list of distances. Can use *.")
parser.add_option("-s", "--streams", default='*',
help="Comma-separated list of streams. Can use *.")
parser.add_option("-r", "--repeat", default=1, type='int',
help="Repeat many times.")
parser.add_option("-c", "--command",
help="Command to pass to compmake for batch mode")
options = parser.parse_options()
distances = natsorted(config.distances.expand_names(options.distances))
streams = natsorted(config.streams.expand_names(options.streams))
logger.info('Using distances: %s' % distances)
logger.info('Using streams: %s' % streams)
id_comb = ','.join(streams) + '-' + ','.join(distances)
outdir = os.path.join(options.output, id_comb)
storage = os.path.join(outdir, 'compmake')
use_filesystem(storage)
read_rc_files()
rm = ReportManager(os.path.join(outdir, "reports"))
create_diststats_jobs(config=config, distances=distances, streams=streams,
rm=rm, maxd=10)
rm.create_index_job()
if options.command:
return batch_command(options.command)
else:
compmake_console()
return 0
def define_jobs_context(self, context):
discdds_library = get_conftools_discdds()
discdds = discdds_library.expand_names(self.options.discdds)
discdds = natsorted(discdds)
for c, id_discdds in iterate_context_names(context, discdds):
options = self.options
params = dict(tolerance=options.tolerance,
collapse_threshold=options.collapse_threshold,
debug_it=options.debug_it,
max_it=options.num_iterations,
min_visibility=options.min_visibility)
r = c.comp_config(report_dds_fill, id_discdds, **params)
c.add_report(r, 'dds_fill', dds=id_discdds, **params)
def define_jobs_context(self, context):
discdds_library = get_conftools_discdds()
discdds = discdds_library.expand_names(self.options.discdds)
discdds = natsorted(discdds)
for c, id_discdds in iterate_context_names(context, discdds):
options = self.options
params = dict(tolerance=options.tolerance,
collapse_threshold=options.collapse_threshold,
debug_it=options.debug_it,
max_it=options.num_iterations,
min_visibility=options.min_visibility)
r = c.comp_config(report_dds_fill, id_discdds, **params)
c.add_report(r, 'dds_fill', dds=id_discdds, **params)
def get_id_episodes(self):
""" Returns the list of episodes simulated. """
return natsorted(self.id_episodes)
def get_id_episodes(self):
''' Returns the list of episodes simulated. '''
return natsorted(self.id_episodes)
