ss14_chemistry_site/calc.py

from db import *
from math import floor

# DEBUG
from icecream import ic
ic.disable()

db = read_db()

# Составные
recipe = {}

'''
def sround(num, parts):
  acc = [1,2,3,4,5, 10, 15, 20, 25, 30, 35, 40, 45, 50, 55, 60, 65, 70, 75, 80, 85, 90, 95, 100]
  amount = num*parts

  # Ловим частые повторения
  tries = 0

  # Пока не вывели хорошее число
  good = False
  while not good:
    num = floor(num)
    ic(good, ' ', num)
    st = 1 ; good = True
    # Перебираем различное количество частей
    while st < parts:
      # Если в допустимых значениях
      if num*st in acc:
        st += 1
        continue

      good = False
      # Перебираем допустимые значения
      for i in acc:
        if abs(i - num*st) <= 2:
          num = i
          break
      st += 1

    # Ловим частые повторения
    tries += 1
    if tries > 500:
      return sround1(num, parts)

  if num*parts > amount:
    return sround1(num, parts)

  return num
'''


def sround(num, parts):
  acc = [1,2,3,4,5]
  num = floor(num)

  if num == 0:
    return 1
  if num in acc:
    return num
  elif num%5 == 0:
    return num

  while num%5 != 0:
    num -= 1
  return num

# Поиск элемента в списке рецепта
def ll_find(ll, pat):
  find = False
  for i in range(len(ll)):
    if ll[i][0] == pat:
      return i
  return None

def calc(el, amount, main = False):
  global db, recipe
  if main:
    recipe = []

  comps = db[el][3:] # Получаем составные
  heat = db[el][2]
  out = db[el][0] #Количество на выходе

  # Считаем количество частей
  parts = 0
  for i in comps:
    parts += i[1]

  # Делаем поправку на выход
  while out < parts:
    # Предварительная часть
    part = sround(amount/parts, parts)
    # Если итоговый объём <= входного объёма
    if (parts+1)*part <= amount:
      parts += 1
    else:
      break

  # Считаем 1 часть
  part = sround(amount/parts, parts)

  # Динамический объём
  # Нужен когда у вещества вход 20 а выход 10
  #part_vol = 
  

  # Перебираем составные и делаем рекурсию
  for i in comps:
    ic(i)
    if i[0] in db:
      lpart = calc(i[0], part*i[1])
      # Если наша часть больше чем составная
      if lpart < part:
        part = lpart
  # lpart - количество составного в итоге
  # 50/3, часть = 16, ИТОГ: 16*3=48 <


  if heat and not main:
    recipe = [['heat']] + recipe
  if heat and main:
    recipe.append(['heat'])

  # Перебираем элементарные вещества
  for i in comps:
    if i[0] not in db:
      if i[0] == 'Плазма':
        if ll_find(recipe, 'Плазма') == None:
          recipe = [[i[0], 1]] + recipe
      else:
        if heat:
          recipe = [[i[0], part*i[1]]] + recipe
        else:
          recipe.append([i[0], part*i[1]])

  # Если нету плазмы и нагрева - соединяем вещества
  if main:
    if ll_find(recipe, 'Плазма') == None and ll_find(recipe, 'heat') == None:
      ic('START: ', recipe)
      new_recipe = []
      
      while recipe != []:
        ic("ORIG:", recipe)
        el = recipe[0]
        new_recipe.append(el)
        del recipe[0]

        # Текущий id
        id = ll_find(new_recipe, el[0])

        # Если есть ещё такой элемент
        while ll_find(recipe, el[0]):
          # Если это отметка для нагрева
          if el[0] == 'heat':
            break
          same_id = ll_find(recipe, el[0])
          ic('OLD: ', new_recipe[id][1])
          new_recipe[id][1] += recipe[same_id][1]
          ic('NEW: ', new_recipe[id][1] + recipe[same_id][1])
          # Удаляем этот элемент
          del recipe[same_id]

        ic("NEW:", new_recipe)


      recipe = new_recipe


  if main:
    ic(part)
    return [recipe, out*part]
  else:
    ic(recipe)
    ic(el)
    ic('PART: ', part)
    # >>>>>>>>>>>>>>> Возможен баг <<<<<<<<<<<<
    #return part*parts
    # Возвращаем объём=часть*выход
    return part*out

#print( calc("Лексорин", 100, True))
#print( calc("Бикаридин", 100, True))
#print( calc("Диловен", 100, True))
#print( calc("Эфедрин", 100, True))
#print( calc("Криоксадон", 100, True) )
#print( calc("Гидроксид", 100, True) )
#print( calc("Пунктураз", 100, True) )

#print( calc("Эпинефрин", 100, True) )

#print( calc("Гидроксид", 100, True) )
#print( calc("Ацетон", 50, True) )

#print( calc("Фенол", 50, True) )
#print(calc("Бензол", 50, True))
#OK print( calc("Гидроксид", 50, True) )

#print( calc("", 100, True) )
#print( calc("", 100, True) )
#print( calc("", 100, True) )