157 lines
5.1 KiB
Python
157 lines
5.1 KiB
Python
import math
|
||
|
||
mc_colors = [
|
||
['红色染料', '#B02E26', 'red'],
|
||
['绿色染料', '#5E7C16', 'green'],
|
||
['紫色染料', '#8932B8', 'purple'],
|
||
['青色染料', '#169C9C', 'cyan'],
|
||
['淡灰色染料', '#9D9D97', 'silver'],
|
||
['灰色染料', '#474F52', 'gray'],
|
||
['粉红色染料', '#F38BAA', 'pink'],
|
||
['黄绿色染料', '#80C71F', 'lime'],
|
||
['黄色染料', '#FED83D', 'yellow'],
|
||
['淡蓝色染料', '#3AB3DA', 'lightBlue'],
|
||
['品红色染料', '#C74EBD', 'magenta'],
|
||
['橙色染料', '#F9801D', 'orange'],
|
||
['黑色染料', '#1D1D21', 'black'],
|
||
['棕色染料', '#835432', 'brown'],
|
||
['蓝色染料', '#3C44AA', 'blue'],
|
||
['白色染料', '#F9FFFE', 'white']
|
||
]
|
||
|
||
def parse_mc_color(colors):
|
||
parsed_color = {}
|
||
for color in colors:
|
||
name, value, key = color
|
||
value = value.strip('#')
|
||
assert len(value) == 6
|
||
r = int(value[:2], base=16)
|
||
g = int(value[2:4], base=16)
|
||
b = int(value[4:], base=16)
|
||
parsed_color[key] = {
|
||
"name": name,
|
||
"key": key,
|
||
"value": [r, g, b]
|
||
}
|
||
return parsed_color
|
||
|
||
mc_colors = parse_mc_color(mc_colors)
|
||
|
||
def calculate_fused_color(counts):
|
||
num_colors = sum(counts.values())
|
||
if num_colors == 0:
|
||
return [160, 101, 64]
|
||
avg_color = []
|
||
# calculate total color.
|
||
for i in range(3):
|
||
this_total_color = sum([mc_colors[key]['value'][i]*count for key, count in counts.items()])
|
||
avg_color.append(this_total_color//num_colors)
|
||
# calculate average maximum.
|
||
average_maximum = 0
|
||
for key, count in counts.items():
|
||
average_maximum += max(mc_colors[key]['value']) * count
|
||
average_maximum //= num_colors
|
||
# calculate gain factor.
|
||
maximum_of_average = max(avg_color)
|
||
gain_factor = average_maximum / maximum_of_average
|
||
# generate result.
|
||
fused = [int(x * gain_factor) for x in avg_color]
|
||
return fused
|
||
|
||
def calculate_diff_color(c1, c2):
|
||
return sum([abs(x - y) for x, y in zip(c1, c2)])
|
||
|
||
def calculate_multipler(diff):
|
||
diff = 256 if diff > 256 else diff
|
||
multiplier = (1 - (0.50) * (256.0 - diff) / 256.0);
|
||
return multiplier * math.log(80) / math.log(125)
|
||
|
||
def export_recipe(recipe):
|
||
return ', '.join(['{0}:{1}'.format(mc_colors[key]['name'], recipe[key]) for key in recipe if recipe[key] > 0])
|
||
|
||
def find_best_color(luckycolor, limit=1):
|
||
min_diff = 10000
|
||
best_recipe = None
|
||
best_color = None
|
||
avaliable_colors = list(mc_colors.keys())
|
||
current_stat = [0] * len(avaliable_colors)
|
||
search_space = (limit+1)**len(current_stat)
|
||
print('正在搜索 {0} 种可能的组合'.format(search_space))
|
||
def get_next_stat(curent_stat):
|
||
digit = limit + 1
|
||
multipiler = 1
|
||
value = 0
|
||
# encode
|
||
for stat in current_stat[::-1]:
|
||
value += stat * multipiler
|
||
multipiler *= digit
|
||
# add
|
||
value += 1
|
||
# decode.
|
||
if value >= digit**len(current_stat):
|
||
return None
|
||
next_stat = []
|
||
while value > 0:
|
||
this_digit = value % digit
|
||
value = value // digit
|
||
next_stat.append(this_digit)
|
||
next_stat += [0] * (len(current_stat) - len(next_stat))
|
||
return next_stat[::-1]
|
||
searched = 0
|
||
while current_stat is not None:
|
||
this_color = {key:value for key, value in zip(avaliable_colors, current_stat)}
|
||
fused_color = calculate_fused_color(this_color)
|
||
diff_color = calculate_diff_color(luckycolor, fused_color)
|
||
if diff_color < min_diff:
|
||
min_diff = diff_color
|
||
best_recipe = this_color
|
||
best_color = fused_color
|
||
current_stat = get_next_stat(current_stat)
|
||
searched += 1
|
||
if searched % 10000 == 0:
|
||
print('[进度：{0}%]{1}/{2} 已完成, 当前最小误差为： {3}, 此时配方为： {4}'.format(
|
||
"%.2f"%(searched*100/search_space),
|
||
searched, search_space, min_diff,
|
||
export_recipe(best_recipe)
|
||
))
|
||
return min_diff, best_recipe, best_color
|
||
|
||
|
||
|
||
|
||
|
||
if __name__ == '__main__':
|
||
luckycolor = None
|
||
limit = 1
|
||
while luckycolor is None:
|
||
note = '\n'.join(
|
||
[
|
||
'请输入你的幸运色，用空格分隔。',
|
||
'如你的幸运色为100， 200， 100, 请输入 100 200 100。',
|
||
'默认每一种颜色采用的最大数量为1，若你希望用更多的颜色，可以在幸运色后追加一个数字表示每种颜色最大使用的颜色数量，如 100 200 100 2 表示每种颜色最多使用两个',
|
||
'不建议指定超过两个以上的颜色，所需的搜索空间可能过大导致无法在可接受的时间范围内计算完成，如一种颜色搜索空间为2^16=65536种组合，两种为3^16=43046721种组合。'
|
||
|
||
])
|
||
print(note)
|
||
try:
|
||
luckycolor_str = input('>>> ')
|
||
luckycolor = luckycolor_str.split()
|
||
luckycolor = [int(x) for x in luckycolor]
|
||
if len(luckycolor) < 3 or len(luckycolor) > 4:
|
||
raise ValueError
|
||
if len(luckycolor) == 4:
|
||
limit = luckycolor[3]
|
||
except Exception:
|
||
print('幸运色颜色格式错误qwq')
|
||
|
||
min_diff, best_recipe, best_color = find_best_color(luckycolor, limit=limit)
|
||
print('搜索完成！')
|
||
print('=================================')
|
||
print('你的幸运色为:', luckycolor)
|
||
print('最小颜色误差:', min_diff)
|
||
print('最佳配方:', export_recipe(best_recipe))
|
||
print('由此配方获得的幸运色:', best_color)
|
||
print('当穿着单件衣服时伤害减免为:', calculate_multipler(min_diff))
|
||
print('当穿着全套衣服时伤害减免为:', calculate_multipler(min_diff)**4)
|
||
print('=================================')
|