a
This commit is contained in:
84
main.cpp
84
main.cpp
@@ -1,10 +1,76 @@
|
||||
#include <iostream>
|
||||
#include <string>
|
||||
#include "util/simple_httpserver.hpp"
|
||||
#include "util/grid2.hpp"
|
||||
#include "util/bmpwriter.hpp"
|
||||
#include "util/jxlwriter.hpp"
|
||||
#include "util/timing_decorator.hpp"
|
||||
|
||||
// Function to convert hex color string to Vec4
|
||||
Vec4 hexToVec4(const std::string& hex) {
|
||||
if (hex.length() != 6) {
|
||||
return Vec4(0, 0, 0, 1); // Default to black if invalid
|
||||
}
|
||||
|
||||
int r, g, b;
|
||||
sscanf(hex.c_str(), "%02x%02x%02x", &r, &g, &b);
|
||||
|
||||
return Vec4(r / 255.0f, g / 255.0f, b / 255.0f, 1.0f);
|
||||
}
|
||||
|
||||
// Generate gradient image
|
||||
bool generateGradientImage(const std::string& filename, int width = 512, int height = 512) {
|
||||
#TIME_FUNCTION;
|
||||
const int POINTS_PER_DIM = 256;
|
||||
|
||||
Grid2 grid;
|
||||
|
||||
// Define our target colors at specific positions
|
||||
Vec4 white = hexToVec4("ffffff"); // Top-left corner (1,1)
|
||||
Vec4 red = hexToVec4("ff0000"); // Top-right corner (1,-1)
|
||||
Vec4 green = hexToVec4("00ff00"); // Center (0,0)
|
||||
Vec4 blue = hexToVec4("0000ff"); // Bottom-left corner (-1,-1)
|
||||
Vec4 black = hexToVec4("000000"); // Bottom-right corner (-1,1)
|
||||
|
||||
// Create gradient points
|
||||
for (int y = 0; y < POINTS_PER_DIM; ++y) {
|
||||
for (int x = 0; x < POINTS_PER_DIM; ++x) {
|
||||
// Normalize coordinates to [-1, 1]
|
||||
float nx = (static_cast<float>(x) / (POINTS_PER_DIM - 1)) * 2.0f - 1.0f;
|
||||
float ny = (static_cast<float>(y) / (POINTS_PER_DIM - 1)) * 2.0f - 1.0f;
|
||||
|
||||
// Create position
|
||||
Vec2 pos(nx, ny);
|
||||
|
||||
// Convert to [0,1] range for interpolation
|
||||
float u = (nx + 1.0f) / 2.0f;
|
||||
float v = (ny + 1.0f) / 2.0f;
|
||||
|
||||
// Bilinear interpolation between corners
|
||||
Vec4 top = white * (1.0f - u) + red * u;
|
||||
Vec4 bottom = blue * (1.0f - u) + black * u;
|
||||
Vec4 cornerColor = top * (1.0f - v) + bottom * v;
|
||||
|
||||
// Calculate distance from center (0,0)
|
||||
float distFromCenter = std::sqrt(nx * nx + ny * ny) / std::sqrt(2.0f);
|
||||
|
||||
Vec4 color = green * (1.0f - distFromCenter) + cornerColor * distFromCenter;
|
||||
|
||||
grid.addPoint(pos, color);
|
||||
}
|
||||
}
|
||||
|
||||
// Render to RGB image
|
||||
std::vector<uint8_t> imageData = grid.renderToRGB(width, height);
|
||||
|
||||
// Save as JXL
|
||||
return JXLWriter::saveJXL(filename, imageData, width, height);
|
||||
}
|
||||
|
||||
int main(int argc, char* argv[]) {
|
||||
// Check command line arguments
|
||||
int port = 8080;
|
||||
std::string webRoot = "web";
|
||||
|
||||
for (int i = 1; i < argc; ++i) {
|
||||
std::string arg = argv[i];
|
||||
@@ -12,16 +78,30 @@ int main(int argc, char* argv[]) {
|
||||
if (i + 1 < argc) {
|
||||
port = std::stoi(argv[++i]);
|
||||
}
|
||||
} else if (arg == "--webroot" || arg == "-w") {
|
||||
if (i + 1 < argc) {
|
||||
webRoot = argv[++i];
|
||||
}
|
||||
} else if (arg == "--help" || arg == "-h") {
|
||||
std::cout << "Usage: " << argv[0] << " [options]" << std::endl;
|
||||
std::cout << "Options:" << std::endl;
|
||||
std::cout << " -p, --port PORT Set server port (default: 8080)" << std::endl;
|
||||
std::cout << " -w, --webroot DIR Set web root directory (default: web)" << std::endl;
|
||||
std::cout << " -h, --help Show this help message" << std::endl;
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
|
||||
SimpleHTTPServer server(port);
|
||||
// Generate gradient image before starting server
|
||||
std::cout << "Generating gradient image..." << std::endl;
|
||||
if (generateGradientImage(webRoot + "/gradient.jxl")) {
|
||||
std::cout << "Gradient image generated successfully" << std::endl;
|
||||
} else {
|
||||
std::cerr << "Failed to generate gradient image" << std::endl;
|
||||
return 1;
|
||||
}
|
||||
|
||||
SimpleHTTPServer server(port, webRoot);
|
||||
|
||||
if (!server.start()) {
|
||||
std::cerr << "Failed to start server on port " << port << std::endl;
|
||||
@@ -29,7 +109,7 @@ int main(int argc, char* argv[]) {
|
||||
}
|
||||
|
||||
std::cout << "Server running on http://localhost:" << port << std::endl;
|
||||
std::cout << "Open this URL in your web browser to view the dynamic gradient" << std::endl;
|
||||
std::cout << "Web root: " << webRoot << std::endl;
|
||||
std::cout << "Press Ctrl+C to stop the server" << std::endl;
|
||||
|
||||
server.handleRequests();
|
||||
|
||||
Reference in New Issue
Block a user