[wip] more stuff

This commit is contained in:
2022-11-30 18:07:09 +08:00
parent c31c8308d8
commit 263bde65fb
11 changed files with 244 additions and 82 deletions

12
include/econ.hpp Normal file
View File

@@ -0,0 +1,12 @@
#include<iostream>
#include <vector>
void economic_decision_simple(std::vector<int> &balances, int id, float spend_ratio) {
// assert(spend_ratio > 0 && spend_ratio <= 1);
std::cout << "test" << std::endl;
int total_bal = balances.at(id);
int spendable = (random() * spend_ratio) * total_bal;
int recipient_id = random() * (balances.size());
balances.at(id) -= spendable;
balances.at(recipient_id) += spendable;
}

32
include/io.hpp Normal file
View File

@@ -0,0 +1,32 @@
#include <vector>
#include <string>
#include <cstdio>
#include <iostream>
#include <fstream>
void dump_balances_csv(std::vector<int> balances, std::string name) {
std::string csv_data = "Id,Balance\n";
for (int i = 0; i < balances.size(); i++) {
csv_data += std::to_string(i+1)+ "," +
std::to_string(balances[i]) + "\n";
}
std::ofstream file;
file.open(name + ".csv");
file << csv_data;
file.close();
}
void dump_balances_and_ages_csv(std::vector<int> balances, std::vector<int> ages, std::string name) {
assert(balances.size() == ages.size());
std::string csv_data = "Id,Balance,Age\n";
for (int i = 0; i < balances.size(); i++) {
csv_data += std::to_string(i+1)+ "," +
std::to_string(ages[i]) + "," +
std::to_string(balances[i]) + "\n";
}
std::ofstream file;
file.open(name + ".csv");
file << csv_data;
file.close();
}