def call(self):
"""
Construct and call the homogenization engine accoring to options.
"""
options = self.options
opts = self.app_options
conf = self.problem.conf
coefs_name = opts.coefs
coef_info = conf.get_default_attr(opts.coefs, None,
'missing "%s" in problem description!'
% opts.coefs)
if options.detect_band_gaps:
# Compute band gaps coefficients and data.
keys = [key for key in coef_info if key.startswith('band_gaps')]
elif options.analyze_dispersion or options.phase_velocity:
# Insert incident wave direction to coefficients that need it.
for key, val in coef_info.iteritems():
coef_opts = val.get('options', None)
if coef_opts is None: continue
if (('incident_wave_dir' in coef_opts)
and (coef_opts['incident_wave_dir'] is None)):
coef_opts['incident_wave_dir'] = opts.incident_wave_dir
if options.analyze_dispersion:
# Compute dispersion coefficients and data.
keys = [key for key in coef_info
if key.startswith('dispersion')
or key.startswith('polarization_angles')]
else:
# Compute phase velocity and its requirements.
keys = [key for key in coef_info
if key.startswith('phase_velocity')]
else:
# Compute only the eigenvalue problems.
names = [req for req in conf.get(opts.requirements, [''])
if req.startswith('evp')]
coefs = {'dummy' : {'requires' : names,
'class' : CoefDummy,}}
conf.coefs_dummy = coefs
coefs_name = 'coefs_dummy'
keys = ['dummy']
he_options = Struct(coefs=coefs_name, requirements=opts.requirements,
compute_only=keys,
post_process_hook=self.post_process_hook)
volume = get_volume_from_options(opts, self.problem)
he = HomogenizationEngine(self.problem, options,
app_options=he_options,
volume=volume)
coefs = he()
coefs = Coefficients(**coefs.to_dict())
coefs.volume = volume
coefs_filename = op.join(opts.output_dir, opts.coefs_filename)
coefs.to_file_txt(coefs_filename + '.txt',
opts.tex_names,
opts.float_format)
bg_keys = [key for key in coefs.to_dict()
if key.startswith('band_gaps')
or key.startswith('dispersion')]
for ii, key in enumerate(bg_keys):
bg = coefs.get_default_attr(key)
log_save_name = bg.get_default_attr('log_save_name', None)
if log_save_name is not None:
filename = op.join(self.problem.output_dir, log_save_name)
bg.save_log(filename, opts.float_format, bg)
if options.plot:
if options.detect_band_gaps:
self.plot_band_gaps(coefs)
elif options.analyze_dispersion:
self.plot_dispersion(coefs)
elif options.phase_velocity:
keys = [key for key in coefs.to_dict()
if key.startswith('phase_velocity')]
for key in keys:
output('%s:' % key, coefs.get_default_attr(key))
return coefs
def call(self):
"""
Construct and call the homogenization engine accoring to options.
"""
options = self.options
opts = self.app_options
conf = self.problem.conf
coefs_name = opts.coefs
coef_info = conf.get_default_attr(
opts.coefs, None,
'missing "%s" in problem description!' % opts.coefs)
if options.detect_band_gaps:
# Compute band gaps coefficients and data.
keys = [key for key in coef_info if key.startswith('band_gaps')]
elif options.analyze_dispersion or options.phase_velocity:
# Insert incident wave direction to coefficients that need it.
for key, val in coef_info.iteritems():
coef_opts = val.get('options', None)
if coef_opts is None: continue
if (('incident_wave_dir' in coef_opts)
and (coef_opts['incident_wave_dir'] is None)):
coef_opts['incident_wave_dir'] = opts.incident_wave_dir
if options.analyze_dispersion:
# Compute dispersion coefficients and data.
keys = [
key for key in coef_info if key.startswith('dispersion')
or key.startswith('polarization_angles')
]
else:
# Compute phase velocity and its requirements.
keys = [
key for key in coef_info
if key.startswith('phase_velocity')
]
else:
# Compute only the eigenvalue problems.
names = [
req for req in conf.get(opts.requirements, [''])
if req.startswith('evp')
]
coefs = {
'dummy': {
'requires': names,
'class': CoefDummy,
}
}
conf.coefs_dummy = coefs
coefs_name = 'coefs_dummy'
keys = ['dummy']
he_options = Struct(coefs=coefs_name,
requirements=opts.requirements,
compute_only=keys,
post_process_hook=self.post_process_hook)
volume = get_volume_from_options(opts, self.problem)
he = HomogenizationEngine(self.problem,
options,
app_options=he_options,
volume=volume)
coefs = he()
coefs = Coefficients(**coefs.to_dict())
coefs.volume = volume
coefs_filename = op.join(opts.output_dir, opts.coefs_filename)
coefs.to_file_txt(coefs_filename + '.txt', opts.tex_names,
opts.float_format)
bg_keys = [
key for key in coefs.to_dict()
if key.startswith('band_gaps') or key.startswith('dispersion')
]
for ii, key in enumerate(bg_keys):
bg = coefs.get_default_attr(key)
log_save_name = bg.get_default_attr('log_save_name', None)
if log_save_name is not None:
filename = op.join(self.problem.output_dir, log_save_name)
bg.save_log(filename, opts.float_format, bg)
if options.plot:
if options.detect_band_gaps:
self.plot_band_gaps(coefs)
elif options.analyze_dispersion:
self.plot_dispersion(coefs)
elif options.phase_velocity:
keys = [
key for key in coefs.to_dict()
if key.startswith('phase_velocity')
]
for key in keys:
output('%s:' % key, coefs.get_default_attr(key))
return coefs
