//
// 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;
            int max_height;
            
            static size_t getPixelPosition(unsigned int x, unsigned int y){
                return x * CHANNELS + y * WIDTH * CHANNELS;
            }
            static double similarity(const unsigned char* pixels, unsigned int x, unsigned int y, int size);
            static double aroundSimilarity(const unsigned char* pixels, unsigned int x, unsigned int y, int size);
            
            void generateRandomTree(int n);
            
            Color execute_internal(double x, double y, int node);
        public:
            explicit GeneticTree(int max_height): max_height(max_height) {
                for (int i = 0; i < max_height; i++)
                    size *= 2;
                size += 1;
                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(0);
            }
            
            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);
            [[nodiscard]] int subtreeSize(int n) const;
            
            void deleteSubtree(int n);
            std::pair<GeneticNode**, size_t> moveSubtree(int n);
            void insertSubtree(int n, GeneticNode** tree, size_t size);
            void processImage(unsigned char* pixels);
            static double evaluate(const unsigned char* pixels);
            double evaluate();
            
            void deleteTree(){
                for (int i = 0; i < size; i++) {
                    delete nodes[i];
                    nodes[i] = nullptr;
                }
            }
            
            [[nodiscard]] inline int getSize() const {
                return size;
            }
            
            void mutate();
            
            void crossover(GeneticTree* other);
            
            static GeneticTree* breed(GeneticTree* parent1, GeneticTree* parent2);
            
            ~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;
            GeneticTree* last_tree = nullptr;
            GeneticTree* saved_tree = nullptr;
            
            void regenTreeDisplay();
            
            float renderProgress = 0;
        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