summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/main.glade268
-rw-r--r--src/main.rs173
2 files changed, 441 insertions, 0 deletions
diff --git a/src/main.glade b/src/main.glade
new file mode 100644
index 0000000..4d28906
--- /dev/null
+++ b/src/main.glade
@@ -0,0 +1,268 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!-- Generated with glade 3.40.0 -->
+<interface>
+ <requires lib="gtk+" version="3.24"/>
+ <object class="GtkAdjustment" id="adjDraws">
+ <property name="upper">100</property>
+ <property name="step-increment">1</property>
+ <property name="page-increment">10</property>
+ </object>
+ <object class="GtkAdjustment" id="adjLots">
+ <property name="upper">10000</property>
+ <property name="step-increment">1</property>
+ <property name="page-increment">10</property>
+ </object>
+ <object class="GtkTreeStore" id="ts">
+ <columns>
+ <!-- column-name name -->
+ <column type="gchararray"/>
+ <!-- column-name lots -->
+ <column type="guint"/>
+ </columns>
+ </object>
+ <object class="GtkApplicationWindow" id="window">
+ <property name="can-focus">False</property>
+ <property name="title" translatable="yes">RRaffle</property>
+ <property name="resizable">False</property>
+ <child>
+ <object class="GtkBox">
+ <property name="visible">True</property>
+ <property name="can-focus">False</property>
+ <child>
+ <!-- n-columns=3 n-rows=7 -->
+ <object class="GtkGrid">
+ <property name="visible">True</property>
+ <property name="can-focus">False</property>
+ <child>
+ <object class="GtkLabel" id="lbl_csv">
+ <property name="visible">True</property>
+ <property name="can-focus">False</property>
+ <property name="margin-start">5</property>
+ <property name="margin-end">5</property>
+ <property name="margin-top">5</property>
+ <property name="margin-bottom">5</property>
+ <property name="label" translatable="yes">Choose CSV</property>
+ </object>
+ <packing>
+ <property name="left-attach">0</property>
+ <property name="top-attach">0</property>
+ </packing>
+ </child>
+ <child>
+ <object class="GtkFileChooserButton" id="file_choose_csv">
+ <property name="visible">True</property>
+ <property name="can-focus">False</property>
+ <property name="margin-start">5</property>
+ <property name="margin-end">5</property>
+ <property name="margin-top">5</property>
+ <property name="margin-bottom">5</property>
+ <property name="title" translatable="yes"/>
+ </object>
+ <packing>
+ <property name="left-attach">1</property>
+ <property name="top-attach">0</property>
+ </packing>
+ </child>
+ <child>
+ <object class="GtkButton" id="btn_load">
+ <property name="label" translatable="yes">Load</property>
+ <property name="visible">True</property>
+ <property name="can-focus">True</property>
+ <property name="receives-default">True</property>
+ <property name="margin-start">5</property>
+ <property name="margin-end">5</property>
+ <property name="margin-top">5</property>
+ <property name="margin-bottom">5</property>
+ </object>
+ <packing>
+ <property name="left-attach">2</property>
+ <property name="top-attach">0</property>
+ </packing>
+ </child>
+ <child>
+ <object class="GtkLabel" id="lbl_max_wins">
+ <property name="visible">True</property>
+ <property name="can-focus">False</property>
+ <property name="margin-start">5</property>
+ <property name="margin-end">5</property>
+ <property name="margin-top">5</property>
+ <property name="margin-bottom">5</property>
+ <property name="label" translatable="yes">Max Wins</property>
+ </object>
+ <packing>
+ <property name="left-attach">0</property>
+ <property name="top-attach">1</property>
+ </packing>
+ </child>
+ <child>
+ <object class="GtkSpinButton" id="sp_max_wins">
+ <property name="visible">True</property>
+ <property name="can-focus">True</property>
+ <property name="margin-start">5</property>
+ <property name="margin-end">5</property>
+ <property name="margin-top">5</property>
+ <property name="margin-bottom">5</property>
+ <property name="adjustment">adjDraws</property>
+ <property name="climb-rate">1</property>
+ </object>
+ <packing>
+ <property name="left-attach">1</property>
+ <property name="top-attach">1</property>
+ <property name="width">2</property>
+ </packing>
+ </child>
+ <child>
+ <object class="GtkButton" id="btn_draw_single">
+ <property name="label" translatable="yes">Draw Single</property>
+ <property name="visible">True</property>
+ <property name="can-focus">True</property>
+ <property name="receives-default">True</property>
+ <property name="margin-start">5</property>
+ <property name="margin-end">5</property>
+ <property name="margin-top">5</property>
+ <property name="margin-bottom">5</property>
+ </object>
+ <packing>
+ <property name="left-attach">0</property>
+ <property name="top-attach">2</property>
+ <property name="width">3</property>
+ </packing>
+ </child>
+ <child>
+ <object class="GtkButton" id="btn_reset">
+ <property name="label" translatable="yes">Reset Draws</property>
+ <property name="visible">True</property>
+ <property name="can-focus">True</property>
+ <property name="receives-default">True</property>
+ <property name="margin-start">5</property>
+ <property name="margin-end">5</property>
+ <property name="margin-top">5</property>
+ <property name="margin-bottom">5</property>
+ </object>
+ <packing>
+ <property name="left-attach">0</property>
+ <property name="top-attach">4</property>
+ <property name="width">3</property>
+ </packing>
+ </child>
+ <child>
+ <object class="GtkLabel" id="lbl_num_draws">
+ <property name="visible">True</property>
+ <property name="can-focus">False</property>
+ <property name="margin-start">5</property>
+ <property name="margin-end">5</property>
+ <property name="margin-top">5</property>
+ <property name="margin-bottom">5</property>
+ <property name="label" translatable="yes">Draws</property>
+ </object>
+ <packing>
+ <property name="left-attach">0</property>
+ <property name="top-attach">5</property>
+ </packing>
+ </child>
+ <child>
+ <object class="GtkButton" id="btn_draw_multiple">
+ <property name="label" translatable="yes">Draw Multiple</property>
+ <property name="visible">True</property>
+ <property name="can-focus">True</property>
+ <property name="receives-default">True</property>
+ <property name="margin-start">5</property>
+ <property name="margin-end">5</property>
+ <property name="margin-top">5</property>
+ <property name="margin-bottom">5</property>
+ </object>
+ <packing>
+ <property name="left-attach">0</property>
+ <property name="top-attach">6</property>
+ <property name="width">3</property>
+ </packing>
+ </child>
+ <child>
+ <object class="GtkSpinButton" id="sp_num_draws">
+ <property name="visible">True</property>
+ <property name="can-focus">True</property>
+ <property name="margin-start">5</property>
+ <property name="margin-end">5</property>
+ <property name="margin-top">5</property>
+ <property name="margin-bottom">5</property>
+ <property name="adjustment">adjDraws</property>
+ </object>
+ <packing>
+ <property name="left-attach">1</property>
+ <property name="top-attach">5</property>
+ <property name="width">2</property>
+ </packing>
+ </child>
+ <child>
+ <object class="GtkLabel" id="lbl_round">
+ <property name="visible">True</property>
+ <property name="can-focus">False</property>
+ <property name="halign">start</property>
+ <property name="margin-start">5</property>
+ <property name="margin-end">5</property>
+ <property name="margin-top">5</property>
+ <property name="margin-bottom">5</property>
+ <property name="label" translatable="yes">Draw Round: 1</property>
+ </object>
+ <packing>
+ <property name="left-attach">0</property>
+ <property name="top-attach">3</property>
+ <property name="width">3</property>
+ </packing>
+ </child>
+ </object>
+ <packing>
+ <property name="expand">False</property>
+ <property name="fill">True</property>
+ <property name="position">0</property>
+ </packing>
+ </child>
+ <child>
+ <object class="GtkScrolledWindow">
+ <property name="visible">True</property>
+ <property name="can-focus">True</property>
+ <property name="shadow-type">in</property>
+ <property name="min-content-width">200</property>
+ <child>
+ <object class="GtkViewport">
+ <property name="visible">True</property>
+ <property name="can-focus">False</property>
+ <child>
+ <object class="GtkTreeView" id="tv">
+ <property name="visible">True</property>
+ <property name="can-focus">True</property>
+ <property name="model">ts</property>
+ <child internal-child="selection">
+ <object class="GtkTreeSelection"/>
+ </child>
+ <child>
+ <object class="GtkTreeViewColumn" id="cName">
+ <property name="title" translatable="yes">Name</property>
+ <child>
+ <object class="GtkCellRendererText" id="crName"/>
+ </child>
+ </object>
+ </child>
+ <child>
+ <object class="GtkTreeViewColumn" id="cLots">
+ <property name="title" translatable="yes">Lots</property>
+ <child>
+ <object class="GtkCellRendererSpin" id="crLots"/>
+ </child>
+ </object>
+ </child>
+ </object>
+ </child>
+ </object>
+ </child>
+ </object>
+ <packing>
+ <property name="expand">False</property>
+ <property name="fill">True</property>
+ <property name="position">1</property>
+ </packing>
+ </child>
+ </object>
+ </child>
+ </object>
+</interface>
diff --git a/src/main.rs b/src/main.rs
new file mode 100644
index 0000000..b92c136
--- /dev/null
+++ b/src/main.rs
@@ -0,0 +1,173 @@
+use gtk::prelude::*;
+use clap::Parser;
+use csv;
+use rand::Rng;
+use text_io;
+
+const APP_ID: &str = "com.encrox.rraffle";
+
+#[derive(Parser, Debug)]
+struct Cli {
+ /// CSV file with names and number of lots
+ #[arg(short, long)]
+ file: Option<std::path::PathBuf>,
+
+ /// Show GUI
+ #[arg(short, long)]
+ gui: bool,
+
+ /// Shuffle animation
+ #[arg(short, long)]
+ animation: Option<std::path::PathBuf>,
+
+ /// Number of draws
+ #[arg(short, long, default_value_t = 5)]
+ draws: u32,
+
+ /// Number of wins per entry
+ #[arg(short, long, default_value_t = 2)]
+ wins: u32,
+
+ /// Interactive when in CLI mode
+ #[arg(short, long)]
+ interactive: bool,
+}
+
+#[derive(Debug)]
+struct Entry {
+ name: String,
+ lots: u32,
+ wins: u32,
+ odds: f64
+}
+
+fn load_csv(entries : &mut Vec<Entry>, file: std::path::PathBuf) {
+ let reader = csv::Reader::from_path(file);
+ for record in reader.unwrap().records() {
+ //dbg!(record.unwrap().get(0).unwrap());
+ let rec = record.unwrap();
+ entries.push(Entry {
+ name: rec.get(0).unwrap().to_string(),
+ lots: rec.get(1).unwrap().parse::<u32>().unwrap(),
+ wins: 2,
+ odds: 0.0
+ });
+ }
+}
+
+fn max_wins(entries : &mut Vec<Entry>, wins : u32) {
+ for entry in entries {
+ entry.wins = wins;
+ }
+}
+
+fn calc_odds(entries : &mut Vec<Entry>) {
+ let mut sum : u32 = 0;
+ for entry in &mut *entries {
+ sum += if entry.wins > 0 { entry.lots } else { 0 };
+ }
+ for entry in entries {
+ entry.odds = if entry.wins > 0 { entry.lots as f64 / sum as f64 } else { 0.0 };
+ }
+}
+
+fn get_winner(entries : &mut Vec<Entry>) -> &Entry {
+ let mut rng = rand::thread_rng();
+ let rn = rng.gen::<f64>();
+ let mut sum : f64 = 0.0;
+ for entry in entries {
+ if rn >= sum && rn < sum + entry.odds {
+ entry.lots -= 1;
+ entry.wins -= 1;
+ return entry;
+ }
+ sum += entry.odds;
+ }
+ unreachable!();
+}
+
+fn main() {
+ let args = Cli::parse();
+ //dbg!(args);
+
+ if args.gui {
+ // Create a new application
+ let app = gtk::Application::builder().application_id(APP_ID).build();
+
+ // Connect to "activate" signal of `app`
+ app.connect_activate(build_ui);
+
+ // Run the application
+ app.run_with_args(&[""]);
+ } else {
+ if !args.file.is_none() {
+ let mut entries = Vec::<Entry>::new();
+ load_csv(&mut entries, args.file.unwrap());
+ max_wins(&mut entries, args.wins);
+ let mut round = 1;
+ 'outer: loop {
+ calc_odds(&mut entries);
+ println!("Draw round: {}", round);
+ println!("Entries: {:?}", &entries);
+ if args.interactive {
+ print!("Option (s = start, a = abort) [s/a]: ");
+ loop {
+ let option : char = text_io::read!();
+ match option {
+ 'a' => break 'outer,
+ 's' => break,
+ _ => println!("Unknown option: {}", option),
+ }
+ }
+ }
+ let winner = get_winner(&mut entries);
+ println!("Winner is: {}", winner.name);
+ if args.interactive {
+ print!("Option (n = next, a = abort) [n/a]: ");
+ loop {
+ let option : char = text_io::read!();
+ match option {
+ 'a' => break 'outer,
+ 'n' => break,
+ _ => println!("Unknown option: {}", option),
+ }
+ }
+ }
+ if round == args.draws && !args.interactive {
+ println!("Number of draws reached.");
+ break;
+ }
+ round += 1;
+ }
+ }
+ }
+}
+
+fn build_ui(app: &gtk::Application) {
+ let glade_src = include_str!("main.glade");
+ let builder = gtk::Builder::from_string(glade_src);
+
+ let window: gtk::ApplicationWindow = builder.object("window").expect("Couldn't get window");
+ let tv : gtk::TreeView = builder.object("tv").expect("Couldn't get tree view");
+ let btn_draw_single : gtk::Button = builder.object("btn_draw_single").expect("Couldn't get button btn_draw_single");
+ let btn_draw_multiple : gtk::Button = builder.object("btn_draw_multiple").expect("Couldn't get button btn_draw_multiple");
+ let file_choose_csv : gtk::FileChooserButton = builder.object("file_choose_csv").expect("Couldn't get button file_choose_csv");
+ let btn_load : gtk::Button = builder.object("btn_load").expect("Couldn't get button btn_load");
+ let lbl_round : gtk::Label = builder.object("lbl_round").expect("Couldn't get lbl_round");
+ let btn_reset : gtk::Button = builder.object("btn_reset").expect("Couldn't get btn_reset");
+ let sp_num_draws : gtk::SpinButton = builder.object("sp_num_draws").expect("Couldn't get sp_num_draws");
+ let ts : gtk::TreeStore = builder.object("ts").expect("Couldn't get tree store");
+
+ window.set_application(Some(app));
+
+ btn_load.connect_clicked(glib::clone!(@weak file_choose_csv, @weak ts => move |button| {
+ //let data = load_csv(file_choose_csv.filename().unwrap());
+ //for entry in data {
+ let iter = ts.insert(None, -1);
+ ts.set(&iter, &[(0, &glib::Value::from("Eva")), (1, &0.to_value())]);
+ //}
+ }));
+
+ // Present window
+ window.show_all();
+} \ No newline at end of file
generated by cgit v1.2.1 (git 2.18.0) at 2024-12-23 12:27:08 +0000