yggdrasil75/stupidsimcpp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

a

Browse Source
This commit is contained in:
Yggdrasil75
2025-11-06 09:15:30 -05:00
parent fc1e73d874
commit ce3014ce71
2 changed files with 116 additions and 101 deletions
Download Patch File Download Diff File Expand all files Collapse all files

84
main.cpp
View File

@@ -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();

133
util/simple_httpserver.hpp
View File

@@ -21,10 +21,6 @@
#include <arpa/inet.h>
#endif
#include "grid2.hpp"
#include "bmpwriter.hpp"
#include "jxlwriter.hpp"
class SimpleHTTPServer {
private:
int serverSocket;
@@ -32,66 +28,6 @@ private:
bool running;
std::string webRoot;
// 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) {
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);
}
// Read file content
std::string readFile(const std::string& filename) {
std::ifstream file(filename, std::ios::binary);
@@ -113,6 +49,7 @@ private:
if (filename.find(".png") != std::string::npos) return "image/png";
if (filename.find(".jpg") != std::string::npos || filename.find(".jpeg") != std::string::npos) return "image/jpeg";
if (filename.find(".json") != std::string::npos) return "application/json";
if (filename.find(".ico") != std::string::npos) return "image/x-icon";
return "text/plain";
}
@@ -134,16 +71,30 @@ private:
send(clientSocket, responseStr.c_str(), responseStr.length(), 0);
}
// Serve static file
void serveStaticFile(int clientSocket, const std::string& filepath) {
std::string fullPath = webRoot + "/" + filepath;
std::string content = readFile(fullPath);
// Extract file path from HTTP request
std::string getFilePath(const std::string& request) {
// Find the start of the path after "GET "
size_t start = request.find("GET ") + 4;
if (start == std::string::npos) return "";
if (!content.empty()) {
sendResponse(clientSocket, content, getContentType(filepath));
} else {
sendResponse(clientSocket, "404 Not Found: " + filepath, "text/plain", 404);
// Find the end of the path (space or ?)
size_t end = request.find(" ", start);
if (end == std::string::npos) end = request.find("?", start);
if (end == std::string::npos) return "";
std::string path = request.substr(start, end - start);
// Default to index.html for root path
if (path == "/") {
return "index.html";
}
// Remove leading slash
if (path.length() > 0 && path[0] == '/') {
path = path.substr(1);
}
return path;
}
public:
@@ -235,37 +186,21 @@ public:
if (bytesReceived > 0) {
std::string request(buffer);
std::cout << "Received request: " << request.substr(0, request.find('\n')) << std::endl;
std::string filePath = getFilePath(request);
// Handle different routes
if (request.find("GET / ") != std::string::npos || request.find("GET /index.html") != std::string::npos) {
serveStaticFile(clientSocket, "index.html");
} else if (request.find("GET /style.css") != std::string::npos) {
serveStaticFile(clientSocket, "style.css");
} else if (request.find("GET /script.js") != std::string::npos) {
serveStaticFile(clientSocket, "script.js");
} else if (request.find("GET /gradient.jxl") != std::string::npos) {
// Generate and serve the gradient image
if (generateGradientImage("output/gradient.jxl")) {
std::string imageContent = readFile("output/gradient.jxl");
if (!imageContent.empty()) {
sendResponse(clientSocket, imageContent, "image/jxl");
} else {
sendResponse(clientSocket, "Error generating image", "text/plain", 500);
}
if (!filePath.empty()) {
std::cout << "Serving: " << filePath << std::endl;
std::string fullPath = webRoot + "/" + filePath;
std::string content = readFile(fullPath);
if (!content.empty()) {
sendResponse(clientSocket, content, getContentType(filePath));
} else {
sendResponse(clientSocket, "Error generating image", "text/plain", 500);
}
} else if (request.find("GET /generate") != std::string::npos) {
// API endpoint to generate new gradient
std::string filename = "output/dynamic_gradient.jxl";
if (generateGradientImage(filename)) {
sendResponse(clientSocket, "{\"status\":\"success\",\"file\":\"" + filename + "\"}", "application/json");
} else {
sendResponse(clientSocket, "{\"status\":\"error\"}", "application/json", 500);
sendResponse(clientSocket, "404 Not Found: " + filePath, "text/plain", 404);
}
} else {
sendResponse(clientSocket, "404 Not Found", "text/plain", 404);
sendResponse(clientSocket, "400 Bad Request", "text/plain", 400);
}
}