def parseConfig(config_file):
config_parser = ConfigParser.RawConfigParser()
config_parser.read(config_file)
data = { 'K' : 2,
'L' : 0,
'gt' : None,
'alpha' : None,
'beta' : None,
'gamma' : None }
if config_parser.sections() == []:
raise IOError("Invalid configuration file.")
if config_parser.has_section('Ground Truth'):
config.readVisualization(config_parser, 'Ground Truth', os.path.dirname(config_file), options)
config.readFilter(config_parser, 'Ground Truth', os.path.dirname(config_file), options)
config.readAlgorithm(config_parser, 'Ground Truth', os.path.dirname(config_file), options)
config.readMgsSamples(config_parser, 'Ground Truth', os.path.dirname(config_file), options)
data['gt'] = config.readVector(config_parser, 'Ground Truth', 'gt', float)
data['L'] = len(data['gt'])
data['alpha'], data['beta'], data['gamma'] = \
config.getParameters(config_parser, 'Ground Truth', os.path.dirname(config_file), data['K'], data['L'])
config.readStrategy(config_parser, 'Ground Truth', options)
result = loadResult()
if not result['states']:
result['states'] = config.readSeeds(config_parser, 'Ground Truth', 'seeds')
result = sample(result, data)
if config_parser.has_section('Experiment'):
config.readVisualization(config_parser, 'Experiment', os.path.dirname(config_file), options)
config.readFilter(config_parser, 'Experiment', os.path.dirname(config_file), options)
config.readAlgorithm(config_parser, 'Experiment', os.path.dirname(config_file), options)
config.readMgsSamples(config_parser, 'Experiment', os.path.dirname(config_file), options)
data['L'] = int(config_parser.get('Experiment', 'bins'))
data['alpha'], data['beta'], data['gamma'] = \
config.getParameters(config_parser, 'Experiment', os.path.dirname(config_file), data['K'], data['L'])
config.readStrategy(config_parser, 'Experiment', options)
result = loadResult()
result = sample(result, data)
if options['video']:
save_video()
if options['save']:
saveResult(result)
else:
if options['savefig']:
importMatplotlib('Agg')
from matplotlib.pyplot import savefig
vis.plotSampling(result, options, data)
# savefig(options['savefig'], bbox_inches='tight', pad_inches=0)
savefig(options['savefig'])
else:
importMatplotlib()
from matplotlib.pyplot import show
vis.plotSampling(result, options, data)
show()
def save_frame(result, data, utility, i):
importMatplotlib('Agg')
from matplotlib.pyplot import savefig
pyplot.clf()
bin_result = call_posterior(result['counts'], data, options)
bin_result['counts'] = result['counts']
bin_result['samples'] = result['samples']
bin_result['states'] = result['states']
bin_result['utility'] = utility
vis.plotSampling(bin_result, options, data)
savefig('%s_%03d.png' % (options['video'], i), bbox_inches='tight', pad_inches=0)
