mbk/smogon-stats
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

viability ceiling, remove unused fields

Browse source
This commit is contained in:
mehbark 2025-03-24 18:17:37 -04:00
parent 2f661e09c5
commit d14bc6ca45
1 changed files with 15 additions and 13 deletions

28
src/main.rs
View file

@ -1,8 +1,6 @@
// 2 much noise
#![allow(unused)]
// https://www.smogon.com/stats/2025-02/chaos/gen9ou-0.json
// https://www.smogon.com/stats/2025-02/chaos/
use std::{collections::HashMap, env, error::Error, fmt, fs::File, path::PathBuf};
use std::{collections::HashMap, error::Error, fs::File, path::PathBuf};
use clap::Parser;
use rusqlite::Connection;
@ -18,13 +16,14 @@ fn main() -> Result<(), Box<dyn Error>> {
log::info!("opening connection");
let mut conn = Connection::open(output_file)?;
log::info!("creating tables");
// TODO: checks and counters (actually very cool)
// TODO: checks and counters, specifically KO or switch
conn.execute_batch(
"
BEGIN;
CREATE TABLE mon (
name STRING,
usage REAL NOT NULL
usage REAL NOT NULL,
viability_ceiling REAL NOT NULL
);
CREATE TABLE ability (
mon STRING,
@ -58,7 +57,12 @@ fn main() -> Result<(), Box<dyn Error>> {
log::info!("parsing stats");
let input_file = File::open(input_file)?;
let stats: Stats = serde_json::from_reader(input_file)?;
log::info!("{:#?}", stats.info);
log::info!(
"meta: {}: {} battles, cutoff: {}",
stats.info.metagame,
stats.info.battle_count,
stats.info.cutoff
);
for (i, (mon, data)) in stats.data.iter().enumerate() {
log::debug!("Processing mon #{i} ({mon})…");
@ -72,7 +76,10 @@ fn main() -> Result<(), Box<dyn Error>> {
// HACK: we get the weighted mon count by summing the ability usage
// there's is 1000% a better way to do this
let mon_count: f32 = data.abilities.values().sum();
tx.execute("INSERT INTO mon VALUES (?1, ?2)", (mon, data.usage))?;
tx.execute(
"INSERT INTO mon VALUES (?1, ?2, ?3)",
(mon, data.usage, data.viability_ceiling[1]),
)?;
for (ability, count) in &data.abilities {
tx.execute(
@ -160,8 +167,6 @@ struct Stats {
struct Info {
metagame: Box<str>,
cutoff: f64,
#[serde(rename = "cutoff deviation")]
cutoff_deviation: i64,
#[serde(rename = "number of battles")]
battle_count: i64,
}
@ -171,13 +176,10 @@ type Counts = HashMap<Box<str>, f32>;
#[derive(Debug, Deserialize)]
#[serde(rename_all = "PascalCase")]
struct Data {
#[serde(rename = "Raw count")]
count: f32,
#[serde(rename = "Viability Ceiling")]
viability_ceiling: Box<[u32]>,
abilities: Counts,
items: Counts,
spreads: Counts,
moves: Counts,
#[serde(rename = "Tera Types")]
tera: HashMap<Type, f32>,