diplomska_naloga/src/DnaStore.cpp
2025-01-23 14:16:49 +01:00

141 lines
3.6 KiB
C++

#include <cstdio>
#include <ctime>
#include "sys.hpp"
#include "DnaStore.hpp"
#include <raylib.h>
#define ID_FILE_NAME "ID.bin"
#define DATA_FILE_NAME "DATA.bin"
#define VECTOR_FILE_NAME "VECTOR.bin"
void DnaStore::load(DnaManagerData *data)
{
if (sys::fileExists(ID_FILE_NAME))
{
sys::loadDataFromFile(ID_FILE_NAME, &data->id, sizeof(uint128));
}
else
{
data->id = mrand::getState(time(nullptr));
sys::saveDataToFile(ID_FILE_NAME, &data->id, sizeof(uint128));
}
if (sys::fileExists(DATA_FILE_NAME))
{
const char *filename = sys::transformFilePath(DATA_FILE_NAME);
FILE *file = fopen(filename, "rb");
if (file == NULL)
return;
fread(&data->randSeed, 1, sizeof(uint128), file);
fread(&data->generation, 1, sizeof(int), file);
fread(&data->showed, 1, sizeof(int), file);
data->queued = data->showed;
int size, tmp;
fread(&size, sizeof(size), 1, file);
for (int i = 0; i < size; i++)
{
fread(&tmp, sizeof(tmp), 1, file);
if (tmp >= NUM_PER_GEN) // out of bounds error prevention if buffer was in the past biiger then its now
continue;
data->liked.push_back(tmp);
}
fread(&size, sizeof(size), 1, file);
for (int i = 0; i < size; i++)
{
fread(&tmp, sizeof(tmp), 1, file);
if (tmp >= NUM_PER_GEN) // out of bounds error prevention if buffer was in the past biiger then its now
continue;
data->disliked.push_back(tmp);
}
fclose(file);
}
else
{
data->randSeed = data->id;
data->queued = 0;
data->showed = 0;
data->generation = 0;
}
data->vector.resize(NUM_PER_GEN);
if (sys::fileExists(VECTOR_FILE_NAME))
{
sys::loadDataFromFile(VECTOR_FILE_NAME, data->vector.data(), sizeof(Dna) * NUM_PER_GEN);
}
else
{
for (std::size_t i = 0; i < NUM_PER_GEN; i++)
{
DNA::newDna(&data->vector[i], &data->randSeed);
}
saveVec(data); // v primeru da prvic ne gres cez celo generacijo se vec ne shrani in bo drugacen ker se bo regeneriral z drugim rand seed
saveData(data); // save ce rand seed se je spremenil in v premeru user ugasne brez da like se rand seed ne shrani in se resetira v ID
}
}
void DnaStore::saveData(DnaManagerData *data)
{
const char *filename = sys::transformFilePath(DATA_FILE_NAME);
FILE *file = fopen(filename, "wb");
if (file == NULL)
return;
fwrite(&data->randSeed, 1, sizeof(uint128), file);
fwrite(&data->generation, 1, sizeof(int), file);
fwrite(&data->showed, 1, sizeof(int), file);
int tmp;
tmp = data->liked.size();
fwrite(&tmp, sizeof(tmp), 1, file);
fwrite(data->liked.data(), sizeof(int) * tmp, 1, file);
tmp = data->disliked.size();
fwrite(&tmp, sizeof(tmp), 1, file);
fwrite(data->disliked.data(), sizeof(tmp) * tmp, 1, file);
fclose(file);
}
void DnaStore::saveVec(DnaManagerData *data)
{
const char *filename = sys::transformFilePath(VECTOR_FILE_NAME);
FILE *file = fopen(filename, "wb");
if (file == NULL)
return;
fwrite(data->vector.data(), sizeof(Dna) * NUM_PER_GEN, 1, file);
fclose(file);
}
void DnaStore::saveGen(DnaManagerData *data)
{
std::vector<NetUnit> gen;
gen.resize(NUM_PER_GEN);
for (std::size_t i = 0; i < NUM_PER_GEN; i++)
{
gen[i].hash = ComputeCRC32((unsigned char *)&data->vector[i], sizeof(Dna));
gen[i].liked = Liked::tbd;
gen[i].index = i;
}
for (auto &&i : data->liked)
{
gen[i].liked = Liked::yes;
}
for (auto &&i : data->disliked)
{
gen[i].liked = Liked::no;
}
const char *fileName = TextFormat("gen%04d.bin", data->generation);
sys::saveDataToFile(fileName, gen.data(), sizeof(NetUnit) * NUM_PER_GEN);
}