silly code
parent
6747974360
commit
0cb0bdc672
|
|
@ -1,5 +1,5 @@
|
|||
cmake_minimum_required(VERSION 3.25)
|
||||
project(image-gp-2 VERSION 0.0.4)
|
||||
project(image-gp-2 VERSION 0.0.5)
|
||||
|
||||
option(ENABLE_ADDRSAN "Enable the address sanitizer" OFF)
|
||||
option(ENABLE_UBSAN "Enable the ub sanitizer" OFF)
|
||||
|
|
|
|||
|
|
@ -18,6 +18,7 @@
|
|||
|
||||
#ifndef GP_SYSTEM_H
|
||||
#define GP_SYSTEM_H
|
||||
#include <image_storage.h>
|
||||
#include <blt/std/types.h>
|
||||
|
||||
void setup_gp_system(blt::size_t population_size);
|
||||
|
|
@ -26,4 +27,6 @@ void run_step();
|
|||
|
||||
bool should_terminate();
|
||||
|
||||
image_storage_t& get_image(blt::size_t index);
|
||||
|
||||
#endif //GP_SYSTEM_H
|
||||
|
|
|
|||
|
|
@ -22,6 +22,7 @@
|
|||
#include <array>
|
||||
#include <atomic>
|
||||
#include <string>
|
||||
#include <blt/logging/logging.h>
|
||||
#include <blt/std/types.h>
|
||||
|
||||
using image_pixel_t = float;
|
||||
|
|
@ -49,12 +50,28 @@ struct image_storage_t
|
|||
}
|
||||
};
|
||||
|
||||
inline std::atomic_uint64_t g_allocated_nodes = 0;
|
||||
inline std::atomic_uint64_t g_deallocated_nodes = 0;
|
||||
|
||||
struct atomic_node_t
|
||||
{
|
||||
explicit atomic_node_t(image_storage_t* data): data(data)
|
||||
{
|
||||
++g_allocated_nodes;
|
||||
}
|
||||
|
||||
std::atomic<atomic_node_t*> next = nullptr;
|
||||
image_storage_t* data = nullptr;
|
||||
|
||||
~atomic_node_t()
|
||||
{
|
||||
++g_deallocated_nodes;
|
||||
}
|
||||
};
|
||||
|
||||
inline std::atomic_uint64_t g_allocated_blocks = 0;
|
||||
inline std::atomic_uint64_t g_deallocated_blocks = 0;
|
||||
|
||||
class atomic_list_t
|
||||
{
|
||||
public:
|
||||
|
|
@ -106,14 +123,15 @@ struct image_t
|
|||
delete front;
|
||||
} else
|
||||
data = new image_storage_t;
|
||||
++g_allocated_blocks;
|
||||
}
|
||||
|
||||
void drop()
|
||||
{
|
||||
const auto node = new atomic_node_t(); // NOLINT
|
||||
node->data = data;
|
||||
const auto node = new atomic_node_t(data); // NOLINT
|
||||
data = nullptr;
|
||||
g_image_list.push_back(node);
|
||||
++g_deallocated_blocks;
|
||||
}
|
||||
|
||||
[[nodiscard]] void* as_void_const() const
|
||||
|
|
|
|||
|
|
@ -1 +1 @@
|
|||
Subproject commit abb4cc26a4ab2f739e18bdc6dd52bbce24a1b0d5
|
||||
Subproject commit a742164bedd31cac6d828032ee1a4a18c64d0cac
|
||||
|
|
@ -30,11 +30,13 @@ gp_program program{
|
|||
}
|
||||
};
|
||||
|
||||
std::vector<image_storage_t> images;
|
||||
image_storage_t reference_image = image_storage_t::from_file("../silly.png");
|
||||
|
||||
void fitness_func(const tree_t& tree, fitness_t& fitness, const blt::size_t)
|
||||
void fitness_func(const tree_t& tree, fitness_t& fitness, const blt::size_t index)
|
||||
{
|
||||
auto image = tree.get_evaluation_ref<image_t>();
|
||||
std::memcpy(images[index].data.data(), image->get_data().data.data(), IMAGE_SIZE_BYTES);
|
||||
auto& data = image->get_data();
|
||||
for (blt::size_t x = 0; x < IMAGE_DIMENSIONS; ++x)
|
||||
{
|
||||
|
|
@ -47,13 +49,14 @@ void fitness_func(const tree_t& tree, fitness_t& fitness, const blt::size_t)
|
|||
const auto diff_r = data.get(x, y, 0) - reference_image.get(x, y, 0);
|
||||
const auto diff_g = data.get(x, y, 1) - reference_image.get(x, y, 1);
|
||||
const auto diff_b = data.get(x, y, 2) - reference_image.get(x, y, 2);
|
||||
fitness.raw_fitness += diff_r * diff_r * multiplier + diff_g * diff_g * multiplier + diff_b * diff_b * multiplier;
|
||||
const auto total = (diff_r + diff_g + diff_b) / 3;
|
||||
fitness.raw_fitness += (total * total) * multiplier;
|
||||
}
|
||||
}
|
||||
}
|
||||
fitness.raw_fitness /= static_cast<float>(IMAGE_DIMENSIONS * IMAGE_DIMENSIONS);
|
||||
fitness.raw_fitness /= static_cast<float>(IMAGE_SIZE_CHANNELS);
|
||||
fitness.standardized_fitness = fitness.raw_fitness;
|
||||
fitness.adjusted_fitness = fitness.standardized_fitness;
|
||||
fitness.adjusted_fitness = -fitness.standardized_fitness;
|
||||
}
|
||||
|
||||
void setup_operations()
|
||||
|
|
@ -108,7 +111,11 @@ void setup_operations()
|
|||
return std::exp(a);
|
||||
}, "exp_float");
|
||||
static operation_t op_log([](const float a) {
|
||||
return a <= 0.0f ? 0.0f : std::log(a);
|
||||
if (blt::f_equal(a, 0))
|
||||
return 0.0f;
|
||||
if (a < 0)
|
||||
return -std::log(-a);
|
||||
return std::log(a);
|
||||
}, "log_float");
|
||||
static auto lit = operation_t([]() {
|
||||
return program.get_random().get_float(-1.0f, 1.0f);
|
||||
|
|
@ -124,7 +131,9 @@ void setup_gp_system(const blt::size_t population_size)
|
|||
{
|
||||
prog_config_t config{};
|
||||
config.population_size = population_size;
|
||||
config.elites = 2;
|
||||
program.set_config(config);
|
||||
images.resize(population_size);
|
||||
setup_operations();
|
||||
static auto sel = select_tournament_t{};
|
||||
program.generate_initial_population(program.get_typesystem().get_type<image_t>().id());
|
||||
|
|
@ -151,3 +160,8 @@ bool should_terminate()
|
|||
{
|
||||
return program.should_terminate();
|
||||
}
|
||||
|
||||
image_storage_t& get_image(blt::size_t index)
|
||||
{
|
||||
return images[index];
|
||||
}
|
||||
|
|
|
|||
|
|
@ -16,8 +16,6 @@
|
|||
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/
|
||||
#include <image_storage.h>
|
||||
#define STB_IMAGE_IMPLEMENTATION
|
||||
#define STB_IMAGE_RESIZE_IMPLEMENTATION
|
||||
#include <stb_image.h>
|
||||
#include <stb_image_resize2.h>
|
||||
#include <blt/iterator/zip.h>
|
||||
|
|
@ -32,7 +30,7 @@ image_storage_t image_storage_t::from_file(const std::string& path)
|
|||
image_storage_t storage{};
|
||||
stbir_resize_float_linear(data, x, y, 0, storage.data.data(), IMAGE_DIMENSIONS, IMAGE_DIMENSIONS, 0, STBIR_RGBA);
|
||||
|
||||
STBI_FREE(data);
|
||||
stbi_image_free(data);
|
||||
return storage;
|
||||
}
|
||||
|
||||
|
|
|
|||
129
src/main.cpp
129
src/main.cpp
|
|
@ -5,18 +5,49 @@
|
|||
#include "blt/gfx/renderer/batch_2d_renderer.h"
|
||||
#include "blt/gfx/renderer/camera.h"
|
||||
#include <imgui.h>
|
||||
#include <thread>
|
||||
|
||||
blt::gfx::matrix_state_manager global_matrices;
|
||||
blt::gfx::resource_manager resources;
|
||||
blt::gfx::batch_renderer_2d renderer_2d(resources, global_matrices);
|
||||
blt::gfx::first_person_camera camera;
|
||||
|
||||
std::vector<blt::gfx::texture_gl2D*> gl_images;
|
||||
blt::size_t population_size = 64;
|
||||
|
||||
namespace im = ImGui;
|
||||
|
||||
std::atomic_bool run_generation = false;
|
||||
std::atomic_bool should_exit = false;
|
||||
|
||||
std::thread run_gp()
|
||||
{
|
||||
return std::thread{
|
||||
[]() {
|
||||
setup_gp_system(population_size);
|
||||
while (!should_terminate() && !should_exit)
|
||||
{
|
||||
if (run_generation)
|
||||
{
|
||||
run_step();
|
||||
run_generation = false;
|
||||
} else
|
||||
std::this_thread::sleep_for(std::chrono::milliseconds(10));
|
||||
}
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
void init(const blt::gfx::window_data&)
|
||||
{
|
||||
using namespace blt::gfx;
|
||||
|
||||
setup_gp_system(64);
|
||||
|
||||
for (blt::size_t i = 0; i < population_size; i++)
|
||||
{
|
||||
auto texture = new texture_gl2D(IMAGE_DIMENSIONS, IMAGE_DIMENSIONS, GL_RGBA8);
|
||||
gl_images.push_back(texture);
|
||||
resources.set(std::to_string(i), texture);
|
||||
}
|
||||
global_matrices.create_internals();
|
||||
resources.load_resources();
|
||||
renderer_2d.create();
|
||||
|
|
@ -30,11 +61,100 @@ void update(const blt::gfx::window_data& data)
|
|||
camera.update_view(global_matrices);
|
||||
global_matrices.update();
|
||||
|
||||
im::ShowDemoWindow();
|
||||
|
||||
ImGui::SetNextWindowPos(ImVec2(0, 0));
|
||||
ImGui::SetNextWindowSize(ImVec2(ImGui::GetIO().DisplaySize.x, ImGui::GetIO().DisplaySize.y));
|
||||
ImGui::Begin("MainWindow", nullptr,
|
||||
ImGuiWindowFlags_NoTitleBar | ImGuiWindowFlags_NoResize | ImGuiWindowFlags_NoMove | ImGuiWindowFlags_NoCollapse |
|
||||
ImGuiWindowFlags_NoBringToFrontOnFocus);
|
||||
|
||||
// Create the tab bar
|
||||
if (ImGui::BeginTabBar("MainTabs"))
|
||||
{
|
||||
// 1. Run GP tab
|
||||
if (ImGui::BeginTabItem("Run GP"))
|
||||
{
|
||||
// Left child - fixed width (250px)
|
||||
ImGui::BeginChild("ControlPanel", ImVec2(250, 0), true);
|
||||
{
|
||||
ImGui::Text("Control Panel");
|
||||
ImGui::Separator();
|
||||
if (ImGui::Button("Run Step"))
|
||||
{
|
||||
BLT_TRACE("Running step");
|
||||
run_generation = true;
|
||||
}
|
||||
}
|
||||
ImGui::EndChild();
|
||||
|
||||
// Right child - take the remaining space
|
||||
ImGui::SameLine();
|
||||
// ImGui::BeginChild("MainContent", ImVec2(0, 0), false, ImGuiWindowFlags_NoBackground);
|
||||
{}
|
||||
// ImGui::EndChild();
|
||||
|
||||
ImGui::EndTabItem();
|
||||
}
|
||||
|
||||
// 2. Explore Trees tab
|
||||
if (ImGui::BeginTabItem("Explore Trees"))
|
||||
{
|
||||
ImGui::Text("Here you can explore trees.");
|
||||
// Additional UI for exploring trees
|
||||
ImGui::EndTabItem();
|
||||
}
|
||||
|
||||
// 3. Statistics tab
|
||||
if (ImGui::BeginTabItem("Statistics"))
|
||||
{
|
||||
ImGui::Text("Here you can view statistics.");
|
||||
// Additional UI for statistical data
|
||||
ImGui::EndTabItem();
|
||||
}
|
||||
|
||||
ImGui::EndTabBar();
|
||||
}
|
||||
|
||||
ImGui::End(); // MainWindow
|
||||
|
||||
if (ImGui::Begin("Debug"))
|
||||
{
|
||||
const auto allocated_nodes = g_allocated_nodes.load(std::memory_order_relaxed);
|
||||
const auto deallocated_nodes = g_deallocated_nodes.load(std::memory_order_relaxed);
|
||||
const auto allocated_blocks = g_allocated_blocks.load(std::memory_order_relaxed);
|
||||
const auto deallocated_blocks = g_deallocated_blocks.load(std::memory_order_relaxed);
|
||||
ImGui::Text("Allocated Nodes / Deallocated Nodes: (%ld / %ld) (%lf%% / %ld)", allocated_nodes, deallocated_nodes,
|
||||
(static_cast<double>(deallocated_nodes) / static_cast<double>(allocated_nodes)) * 100.0, allocated_nodes - deallocated_nodes);
|
||||
ImGui::Text("Allocated Blocks / Deallocated Blocks: (%ld / %ld) (%lf%% / %ld)", allocated_blocks, deallocated_blocks,
|
||||
(static_cast<double>(deallocated_blocks) / static_cast<double>(allocated_blocks)) * 100.0, allocated_blocks - deallocated_blocks);
|
||||
}
|
||||
ImGui::End();
|
||||
|
||||
for (blt::size_t i = 0; i < population_size; i++)
|
||||
gl_images[i]->upload(get_image(i).data.data(), IMAGE_DIMENSIONS, IMAGE_DIMENSIONS, GL_RGBA, GL_FLOAT);
|
||||
|
||||
for (int i = 0; i < 10; i++)
|
||||
{
|
||||
for (int j = 0; j < 6; j++)
|
||||
{
|
||||
constexpr float padding_x = 32;
|
||||
constexpr float padding_y = 32;
|
||||
constexpr float img_width = 128;
|
||||
constexpr float img_height = 128;
|
||||
const float x = 256 + static_cast<float>(i) * img_width + padding_x * static_cast<float>(i) + img_width;
|
||||
const float y = static_cast<float>(data.height) - (16 + static_cast<float>(j) * img_height + padding_y * static_cast<float>(j) +
|
||||
img_height);
|
||||
renderer_2d.drawRectangle(blt::gfx::rectangle2d_t{x, y, img_width, img_height}, std::to_string(i * j));
|
||||
}
|
||||
}
|
||||
|
||||
renderer_2d.render(data.width, data.height);
|
||||
}
|
||||
|
||||
void destroy(const blt::gfx::window_data&)
|
||||
{
|
||||
gl_images.clear();
|
||||
global_matrices.cleanup();
|
||||
resources.cleanup();
|
||||
renderer_2d.cleanup();
|
||||
|
|
@ -43,5 +163,8 @@ void destroy(const blt::gfx::window_data&)
|
|||
|
||||
int main()
|
||||
{
|
||||
auto run_gp_thread = run_gp();
|
||||
blt::gfx::init(blt::gfx::window_data{"Image GP", init, update, destroy}.setSyncInterval(1));
|
||||
}
|
||||
should_exit = true;
|
||||
run_gp_thread.join();
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue