141 lines
3.8 KiB
C++
141 lines
3.8 KiB
C++
//
|
|
// Created by brett on 7/18/23.
|
|
//
|
|
|
|
#ifndef PARKSNREC_PROGRAM_V3_H
|
|
#define PARKSNREC_PROGRAM_V3_H
|
|
|
|
#include <genetic/v3/functions_v3.h>
|
|
#include "ImNodesEz.h"
|
|
|
|
namespace parks::genetic {
|
|
|
|
struct GeneticNode {
|
|
FunctionID op;
|
|
unsigned int pos{};
|
|
ParameterSet set;
|
|
|
|
GeneticNode(FunctionID op, unsigned int pos, ParameterSet set);
|
|
};
|
|
|
|
class GeneticTree {
|
|
private:
|
|
GeneticNode** nodes;
|
|
int size = 1;
|
|
|
|
void generateRandomTree();
|
|
|
|
Color execute_internal(double x, double y, int node);
|
|
public:
|
|
explicit GeneticTree(int max_height){
|
|
for (int i = 0; i < max_height; i++)
|
|
size *= 2;
|
|
nodes = new GeneticNode*[size];
|
|
|
|
for (int i = 0; i < size; i++)
|
|
nodes[i] = nullptr;
|
|
|
|
//nodes[0] = new GeneticNode(FunctionID::ADD, 0, functions[FunctionID::ADD].generateRandomParameters());
|
|
|
|
generateRandomTree();
|
|
}
|
|
|
|
Color execute(double x, double y);
|
|
|
|
static inline int left(int pos){
|
|
return 2 * (pos + 1);
|
|
}
|
|
|
|
static inline int right(int pos){
|
|
return 2 * (pos + 1) + 1;
|
|
}
|
|
|
|
static inline int parent(int pos){
|
|
if (pos <= 0)
|
|
return -1;
|
|
//if (pos % 2 == 0)
|
|
return pos / 2;
|
|
//return (pos + 1) / 2;
|
|
}
|
|
|
|
inline GeneticNode* node(int pos){
|
|
if (pos < 0 || pos >= size)
|
|
return nullptr;
|
|
return nodes[pos];
|
|
}
|
|
|
|
inline GeneticNode* leftNode(int pos){
|
|
return node(left(pos));
|
|
}
|
|
|
|
inline GeneticNode* rightNode(int pos){
|
|
if (pos < 0 || pos >= size)
|
|
return nullptr;
|
|
return node(right(pos));
|
|
}
|
|
|
|
static int height(int node);
|
|
|
|
void deleteTree(){
|
|
for (int i = 0; i < size; i++) {
|
|
delete nodes[i];
|
|
nodes[i] = nullptr;
|
|
}
|
|
}
|
|
|
|
[[nodiscard]] inline int getSize() const {
|
|
return size;
|
|
}
|
|
|
|
~GeneticTree(){
|
|
deleteTree();
|
|
delete[] nodes;
|
|
}
|
|
};
|
|
|
|
class Program {
|
|
private:
|
|
struct ImNode_t
|
|
{
|
|
int height{};
|
|
int index{};
|
|
FunctionID id;
|
|
ImVec2 pos{};
|
|
bool selected{};
|
|
ImNodes::Ez::SlotInfo inputs[1]{};
|
|
ImNodes::Ez::SlotInfo outputs[2]{};
|
|
};
|
|
std::vector<ImNode_t> treeNodes;
|
|
|
|
unsigned char pixels[WIDTH * HEIGHT * CHANNELS];
|
|
GeneticTree* tree;
|
|
|
|
static size_t getPixelPosition(unsigned int x, unsigned int y){
|
|
return x * CHANNELS + y * WIDTH * CHANNELS;
|
|
}
|
|
void regenTreeDisplay();
|
|
|
|
float renderProgress;
|
|
public:
|
|
Program() = default;
|
|
|
|
void run();
|
|
void draw();
|
|
|
|
[[nodiscard]] float getRenderProgress() const{
|
|
return renderProgress;
|
|
}
|
|
|
|
inline unsigned char* getPixels(){
|
|
return pixels;
|
|
}
|
|
|
|
~Program(){
|
|
delete tree;
|
|
}
|
|
};
|
|
|
|
}
|
|
|
|
#endif //PARKSNREC_PROGRAM_V3_H
|