i2_bot/lib/getters/possibility.go

// i2_bot – Instinct PokememBro Bot
// Copyright (c) 2017 Vladimir "fat0troll" Hodakov

package getters

import (
	// stdlib
	"log"
)

func (g *Getters) PossibilityRequiredPokeballs(location int, grade int, lvl int) (float64, int) {
	var base_possibility float64 = 0.00
	var required_pokeballs int = 0
	var percentile = 0.00

	if lvl > 3 {
		switch {
		case grade == (lvl + 1):
			base_possibility = 0.05
		case grade == lvl:
			base_possibility = 0.5
		case grade == (lvl - 1):
			base_possibility = 0.3
		case grade == (lvl - 2):
			base_possibility = 0.1
		case grade == (lvl - 3):
			base_possibility = 0.05
		default:
			base_possibility = 0.00
		}
	} else if lvl == 3 {
		switch grade {
		case 4:
			base_possibility = 0.05
		case 3:
			base_possibility = 0.5
		case 2:
			base_possibility = 0.3
		case 1:
			base_possibility = 0.15
		default:
			base_possibility = 0.00
		}
	} else if lvl == 2 {
		switch grade {
		case 3:
			base_possibility = 0.05
		case 2:
			base_possibility = 0.70
		case 1:
			base_possibility = 0.25
		default:
			base_possibility = 0.00
		}
	} else if lvl == 1 {
		switch grade {
		case 2:
			base_possibility = 0.80
		case 1:
			base_possibility = 0.20
		default:
			base_possibility = 0.00
		}
	}

	var number_of_pokememes int = 0

	err := c.Db.Get(&number_of_pokememes, c.Db.Rebind("SELECT count(*) FROM pokememes p, pokememes_locations pl WHERE p.grade = ? AND pl.location_id = ? AND pl.pokememe_id = p.id;"), grade, location)
	if err != nil {
		log.Println(err)
	}

	if base_possibility != 0 && number_of_pokememes != 0 {
		percentile = base_possibility * 100.0 / float64(number_of_pokememes)
		required_pokeballs = int(100.0 / percentile)
	}

	return percentile, required_pokeballs
}