def vanilla():
from PIL import Image
target_path = "..\\img\\target_diamond.png"
target_path = "..\\img\\target_spade.png"
# target_path = "..\\img\\target_mickey.png"
# target_path = "..\\img\\target_heart.png"
# target_path = "..\\img\\target_forward.png"
# target_path = "..\\img\\target_recycle.png"
# target_img = Image.open(target_path).convert('L')
penalties = [Optimizer.penalty_name]
# energies = ["first moments (normalized)", "XOR comparison", "secondary moments (normalized)"]
# energies = ["XOR comparison"]
energies = ["uncovered pixels", "distance field"]
# energies = ["first moments (normalized)"]
# energies = ["secondary moments (normalized)"]
# energy_pairs = [energies, [1e-4,1,1e-4]]
weights = [1,1]
energy_pairs = [energies, weights]
rendis = Renderer_dispatcher()
rendis.target_image = Image.open(target_path)
rendis.start()
init_X_Y(640, 480)
plotter = plotting.Plotter(*get_fname("..\\res"))
optimizer = Optimizer(rendis)
plotting.attach_plotter(optimizer, plotter)
optimizer.set_target(target_path)
optimizer.set_method("CMA")
# optimizer.set_energy(["first moments (normalized)"], [1])
optimizer.set_energy(*energy_pairs)
# optimizer.set_energy(["XOR comparison"], [1])
gui = Minimalistic_GUI(optimizer)
gui.run()
def get_off_diag_eff(i, j, yldfile, verbose = 0):
eff = 0
cfeed = open(yldfile, 'r')
cfeedlines = cfeed.readlines()
fname_i = tools.get_fname(i)
fname_j = tools.get_fname(j)
idstring = '%s fakes %s:' % (fname_i, fname_j)
for st in cfeedlines:
if idstring in st:
use_this_value = False
a = float(st.split()[-2])
b = float(st.split()[3]) # signal MC yield/error on yield
if a > 0 and abs(1-b) > 3 and idstring in attr.used_crossfeeds:
eff = a
sys.stdout.write('crossfeed is using %s %s\n' % (idstring, a))
return eff
def get_off_diag_eff(i, j, yldfile, verbose=0):
eff = 0
cfeed = open(yldfile, 'r')
cfeedlines = cfeed.readlines()
fname_i = tools.get_fname(i)
fname_j = tools.get_fname(j)
idstring = '%s fakes %s:' % (fname_i, fname_j)
for st in cfeedlines:
if idstring in st:
use_this_value = False
a = float(st.split()[-2])
b = float(st.split()[3]) # signal MC yield/error on yield
if a > 0 and abs(1 - b) > 3 and idstring in attr.used_crossfeeds:
eff = a
sys.stdout.write('crossfeed is using %s %s\n' % (idstring, a))
return eff
def get_off_diag_err(i, j, yldfile, verbose = 0):
err = 0
cfeed = open(yldfile, 'r')
cfeedlines = cfeed.readlines()
fname_i = tools.get_fname(i)
fname_j = tools.get_fname(j)
idstring = '%s fakes %s:' % (fname_i, fname_j)
for st in cfeedlines:
if idstring in st:
use_this_value = False
a = float(st.split()[-2])
b = float(st.split()[3]) # signal MC yield/error on yield
c = float(st.split()[-1]) # efficiency error
if a > 0 and abs(1-b) > 3 and idstring in attr.used_crossfeeds:
err = c*math.sqrt(1-a)
if verbose > 0:
print 'err: crossfeed is using %s' % idstring
return err
def get_off_diag_err(i, j, yldfile, verbose=0):
err = 0
cfeed = open(yldfile, 'r')
cfeedlines = cfeed.readlines()
fname_i = tools.get_fname(i)
fname_j = tools.get_fname(j)
idstring = '%s fakes %s:' % (fname_i, fname_j)
for st in cfeedlines:
if idstring in st:
use_this_value = False
a = float(st.split()[-2])
b = float(st.split()[3]) # signal MC yield/error on yield
c = float(st.split()[-1]) # efficiency error
if a > 0 and abs(1 - b) > 3 and idstring in attr.used_crossfeeds:
err = c * math.sqrt(1 - a)
if verbose > 0:
print 'err: crossfeed is using %s' % idstring
return err
def all_single_energy_combos():
renderer = Renderer()
renderer.start()
renderer.wait_till_init()
x_0 = renderer.get_param()
for method_name, energy_name in product(Optimizer.method_dic, Optimizer.energy_dic):
print method_name, energy_name
renderer.set_param(x_0)
optimizer = Optimizer(None)
plotting.attach_plotter(optimizer, plotting.Plotter(*get_fname("..\\res")))
optimizer.set_target("C:\\Users\\cxz\\Pictures\\target.png")
optimizer.set_method(method_name)
optimizer.set_energy([energy_name], [1])
optimizer.run()
def _on_play_pause(self):
# not started
if self._optimizer == None or not self._optimizer.is_alive():
self._optimizer = Optimizer(self.rendis)
plotter = plotting.Plotter(*get_fname("..\\res"))
plotting.attach_plotter(self._optimizer, plotter)
# configuring the optimizer by feeding in optimizing-method and energy function
if self.target_path != None:
self._optimizer.set_target(self.target_path)
else:
msg_box = QMessageBox()
msg_box.setText("no target image is selected")
msg_box.exec_()
return
self._optimizer.set_method(self.optim_combo.currentText())
try:
self._optimizer.set_energy(*self._get_energy_pairs())
except RuntimeWarning as rtwng:
msg_box = QMessageBox()
msg_box.setText("no energy function is selected")
msg_box.exec_()
return
self._optimizer.set_finished_callback(self._on_optim_done)
param_lock = Lock()
self._optimizer.set_iter_callback(self._on_iter_callback, param_lock)
param_updater = Lock_listener(param_lock, self._on_param_updated)
if self.armijo_check.isEnabled():
self._optimizer.line_search_first = self.armijo_check.checkState()
self._optimizer.start()
self.play_pause_button.setText("PAUSE")
self.stop_button.setEnabled(True)
self.is_on_going = True
else:
# started already
self.is_on_going = not self.is_on_going
self._optimizer.switch()
if self.is_on_going: # optimizing, to pause
self.play_pause_button.setText("PLAY")
else: # pausing, to continue
self.play_pause_button.setText("PAUSE")
def get_default_plotted_optimizer(renderer):
optimizer = Optimizer(renderer)
plotting.attach_plotter(optimizer, plotting.Plotter(*get_fname("..\\res")))
return optimizer
