import datetime
import json
import os
import random
import numpy
from PIL import Image, PngImagePlugin, ImageDraw, ImageFont
from utils.message_util import MessageBuild
from .gacha_info import init_user_info, user_info, save_user_info
RES_PATH = os.path.join(os.path.dirname(os.path.dirname(__file__)), 'res', 'gacha_res')
font_path = os.path.join(RES_PATH, 'zh-cn.ttf')
count_font = ImageFont.truetype(font_path, 35)
time_font = ImageFont.truetype(font_path, 20)
with open(os.path.join(RES_PATH, 'type.json'), 'r', encoding="utf-8") as fp:
type_json = json.load(fp)
cache_img = {}
cache_item = {}
def random_int():
return numpy.random.randint(low=0, high=10000, size=None, dtype='l')
# 抽卡概率来自https://www.bilibili.com/read/cv10468091
# 角色抽卡概率
def character_probability(rank, count):
ret = 0
count += 1
if rank == 5 and count <= 73:
ret = 60
elif rank == 5 and count >= 74:
ret = 60 + 600 * (count - 73)
elif rank == 4 and count <= 8:
ret = 510
elif rank == 4 and count >= 9:
ret = 510 + 5100 * (count - 8)
return ret
# 武器抽卡概率
def weapon_probability(rank, count):
ret = 0
count += 1
if rank == 5 and count <= 62:
ret = 70
elif rank == 5 and count <= 73:
ret = 70 + 700 * (count - 62)
elif rank == 5 and count >= 74:
ret = 7770 + 350 * (count - 73)
elif rank == 4 and count <= 7:
ret = 600
elif rank == 4 and count == 8:
ret = 6600
elif rank == 4 and count >= 9:
ret = 6600 + 3000 * (count - 8)
return ret
def get_pool_type(gacha_type):
if gacha_type == 301 or gacha_type == 400:
return 'role'
if gacha_type == 200:
return 'permanent'
return 'weapon'
def get_rank(uid, pool_str):
value = random_int()
if pool_str == 'weapon':
index_5 = weapon_probability(5, user_info[uid]["gacha_list"][("gacha_5_%s" % pool_str)])
index_4 = weapon_probability(4, user_info[uid]["gacha_list"][("gacha_4_%s" % pool_str)]) + index_5
else:
index_5 = character_probability(5, user_info[uid]["gacha_list"][("gacha_5_%s" % pool_str)])
index_4 = character_probability(4, user_info[uid]["gacha_list"][("gacha_4_%s" % pool_str)]) + index_5
if value <= index_5:
return 5
elif value <= index_4:
return 4
else:
return 3
def is_Up(uid, rank, pool_str):
if pool_str == 'permanent':
return False
elif pool_str == 'weapon':
return random_int() <= 7500 or user_info[uid]["gacha_list"]["is_up_%s_weapon" % rank]
else:
return random_int() <= 5000 or user_info[uid]["gacha_list"]["is_up_%s_role" % rank]
def once(uid, gacha_data):
role = {}
init_user_info(uid)
pool_str = get_pool_type(gacha_data['gacha_type'])
# 判定星级
rank = get_rank(uid, pool_str)
# 是否为up
if rank != 3:
is_up = is_Up(uid, rank, pool_str)
user_info[uid]["gacha_list"]["wish_total"] += 1
if rank == 3:
role = random.choice(gacha_data['r3_prob_list'])
user_info[uid]["gacha_list"][("gacha_4_%s" % pool_str)] += 1
user_info[uid]["gacha_list"][("gacha_5_%s" % pool_str)] += 1
role['count'] = 1
else:
if rank == 5 and pool_str == 'weapon' and user_info[uid]["gacha_list"]["dg_time"] == 2 and "pool_type" not in gacha_data:
role['item_name'] = user_info[uid]["gacha_list"]["dg_name"]
role['item_type'] = '武器'
role['rank'] = rank
elif is_up:
role = random.choice(gacha_data['r%s_up_items' % rank])
user_info[uid]["gacha_list"]["wish_%s_up" % rank] += 1
role['rank'] = rank
else:
while True:
role = random.choice(gacha_data['r%s_prob_list' % rank])
if role['is_up'] == 0:
break
if rank == 4:
user_info[uid]["gacha_list"][("gacha_5_%s" % pool_str)] += 1
elif rank == 5:
user_info[uid]["gacha_list"][("gacha_4_%s" % pool_str)] += 1
if pool_str == 'weapon' and "pool_type" not in gacha_data:
if user_info[uid]["gacha_list"]["dg_name"] == '':
role['dg_time'] = -1
elif role['item_name'] != user_info[uid]["gacha_list"]["dg_name"]:
user_info[uid]["gacha_list"]["dg_time"] += 1
role['dg_time'] = user_info[uid]["gacha_list"]["dg_time"]
else:
user_info[uid]["gacha_list"]["dg_name"] = ''
user_info[uid]["gacha_list"]["dg_time"] = 0
role['dg_time'] = 3
user_info[uid]["gacha_list"]["wish_%s" % rank] += 1
if gacha_data['gacha_type'] != 200:
user_info[uid]["gacha_list"][("is_up_%s_%s" % (rank, pool_str))] = not is_up
if role['item_type'] == '角色':
itemname = 'role'
else:
itemname = 'weapon'
if role['item_name'] not in user_info[uid]["role_list"]:
user_info[uid]["%s_list" % itemname][role['item_name']] = {}
user_info[uid]["%s_list" % itemname][role['item_name']]['数量'] = 1
user_info[uid]["%s_list" % itemname][role['item_name']]['出货'] = []
if rank == 5:
user_info[uid]["%s_list" % itemname][role['item_name']]['星级'] = '★★★★★'
user_info[uid]["%s_list" % itemname][role['item_name']]['出货'].append(
(user_info[uid]['gacha_list']['gacha_%s_%s' % (rank, pool_str)] + 1))
else:
user_info[uid]["%s_list" % itemname][role['item_name']]['星级'] = '★★★★'
else:
user_info[uid]["%s_list" % itemname][role['item_name']]['数量'] += 1
if rank == 5:
user_info[uid]["%s_list" % itemname][role['item_name']]['出货'].append(
(user_info[uid]['gacha_list']['gacha_%s_%s' % (rank, pool_str)] + 1))
role['count'] = user_info[uid]["gacha_list"]["gacha_%s_%s" % (rank, pool_str)] + 1
user_info[uid]["gacha_list"]["gacha_%s_%s" % (rank, pool_str)] = 0
save_user_info()
return role
async def create_item(rank, item_type, name, element, count, dg_time):
bg = Image.open(os.path.join(RES_PATH, f'{rank}_background.png')).resize((143, 845))
item_img = Image.open(os.path.join(RES_PATH, item_type, f'{name}.png'))
rank_img = Image.open(os.path.join(RES_PATH, f'{rank}_star.png')).resize((119, 30))
if item_type == '角色':
item_img = item_img.resize((item_img.size[0] + 12, item_img.size[1] + 45))
item_img.alpha_composite(rank_img, (4, 510))
item_type_icon = Image.open(os.path.join(RES_PATH, '元素', f'{element}.png')).resize((80, 80))
item_img.alpha_composite(item_type_icon, (25, 420))
bg.alpha_composite(item_img, (3, 125))
else:
bg.alpha_composite(item_img, (3, 240))
bg.alpha_composite(rank_img, (9, 635))
item_type_icon = type_json.get(name)
if item_type_icon:
item_type_icon = Image.open(os.path.join(RES_PATH, '类型', f'{item_type_icon}.png')).resize((100, 100))
bg.alpha_composite(item_type_icon, (18, 530))
if rank == 5 and count != -1:
draw = ImageDraw.Draw(bg)
if len(str(count)) == 2:
draw.text((19, 750), ('[' + str(count) + '抽]'), font=count_font, fill='white')
else:
draw.text((26, 750), ('[' + str(count) + '抽]'), font=count_font, fill='white')
if dg_time != -1:
if dg_time == 3:
draw.text((3, 785), ('定轨结束'), font=count_font, fill='white')
else:
draw.text((7, 785), ('定轨' + str(dg_time) + '/2'), font=count_font, fill='white')
return bg
async def ten(uid, gacha_data) -> PngImagePlugin.PngImageFile:
gacha_list = []
for i in range(0, 10):
if gacha_data['gacha_type'] == 'all_star':
role = random.choice(gacha_data['list'])
role['count'] = -1
role['rank'] = 5
else:
role = once(uid, gacha_data).copy()
gacha_list.append(role)
gacha_list.sort(key=lambda x: x["rank"], reverse=True)
img: PngImagePlugin.PngImageFile
img = Image.open(os.path.join(RES_PATH, 'background.png'))
i = 0
for wish in gacha_list:
i += 1
rank = wish['rank']
item_type = wish['item_type']
name = wish['item_name']
element = wish.get('item_attr') or type_json[name]
count = wish['count']
try:
dg_time = wish['dg_time']
except:
dg_time = -1
i_img = await create_item(rank, item_type, name, element, count, dg_time)
img.alpha_composite(i_img, (105 + (i_img.size[0] * i), 123))
img.thumbnail((1024, 768))
img2 = Image.new("RGB", img.size, (255, 255, 255))
img2.paste(img, mask=img.split()[3])
return img2
async def more_ten(uid, gacha_data, num, sd):
time_str = datetime.datetime.strftime(datetime.datetime.now(), '%m-%d %H:%M')
if num == 1:
img = await ten(uid, gacha_data)
else:
img = Image.new("RGB", (1024, 575 * num), (255, 255, 255))
for i in range(0, num):
item_img = await ten(uid, gacha_data)
img.paste(item_img, (0, 575 * i))
draw = ImageDraw.Draw(img)
draw.text((27, 575 * num - 30), ('@%s %s Created By LittlePaimon' % (str(sd.nickname), time_str)), font=time_font,
fill="#8E8E8E")
return MessageBuild.Image(img, quality=75)