From 3dd8f101a9d9300e26c114897f927eba4cef1fb0 Mon Sep 17 00:00:00 2001 From: Tanishq Dubey Date: Sun, 16 Jun 2024 10:29:51 -0400 Subject: [PATCH] get proper main.cpp --- main.cpp | 103 +++++++++++++++++++++++++++++++++++++++++++------------ 1 file changed, 81 insertions(+), 22 deletions(-) diff --git a/main.cpp b/main.cpp index eb1c881..c06d690 100644 --- a/main.cpp +++ b/main.cpp @@ -37,6 +37,8 @@ #include +#include "lib/histogram.h" + struct Args : public argparse::Args { std::string &fpath = arg("path to the image"); }; @@ -212,38 +214,53 @@ void RotateImage(Texture t) { delete[] tempBuffer; } -Texture LoadTexture(const char * path) -{ +Texture LoadImage(const char * path) { const int channelCount = 4; int imageFileChannelCount; int width, height; image = (uint8_t *)stbi_load(path, &width, &height, &imageFileChannelCount, channelCount); - - if (image == NULL) - { + if (image == NULL) { fprintf(stderr, "%s\nFailed to open %s\n", stbi_failure_reason(), path); - return {nullptr,{0,0}}; } - auto exif = printExifData(path); + Texture t; + t.size = ImVec2((float)width,(float)height); + t.channels = channelCount; + t.exif = exif; + return t; +} - +Texture LoadTexture(Texture tin) { GLenum dataFormat = GL_RGBA; GLuint textureHandle; glGenTextures(1, &textureHandle); glBindTexture(GL_TEXTURE_2D, textureHandle); glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR); glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR); - glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA, width, height, 0, dataFormat, GL_UNSIGNED_BYTE, image); + glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA, tin.size.x, tin.size.y, 0, dataFormat, GL_UNSIGNED_BYTE, image); - Texture t; + Texture t = tin; t.texture = (void*)(uintptr_t)(textureHandle); - t.size = ImVec2((float)width,(float)height); - t.channels = channelCount; - t.exif = exif; + if (t.exif.ImageOrientation == "3") { + RotateImage(t); + ReloadTexture(t, t.size.x, t.size.y); + t = ReloadTexture(t, t.size.y, t.size.x); + RotateImage(t); + t = ReloadTexture(t, t.size.y, t.size.x); + } else if (t.exif.ImageOrientation == "6") { + RotateImage(t); + t = ReloadTexture(t, t.size.y, t.size.x); + } else if (t.exif.ImageOrientation == "8") { + RotateImage(t); + t = ReloadTexture(t, t.size.y, t.size.x); + RotateImage(t); + t = ReloadTexture(t, t.size.y, t.size.x); + RotateImage(t); + t = ReloadTexture(t, t.size.y, t.size.x); + } return t; } @@ -254,6 +271,19 @@ const int MAX_ANNOATED_TEXELS = 10000; // Main code int main(int argc, char* argv[]) { + Texture t; + + try { + auto args = argparse::parse(argc, argv, true); + t = LoadImage(args.fpath.c_str()); + if (t.texture == nullptr) { + std::cerr << "failed load image" << std::endl; + return -1; + } + } catch (const std::runtime_error &e) { + std::cerr << "failed to parse arguments: " << e.what() << std::endl; + return -1; + } // Setup SDL if (SDL_Init(SDL_INIT_VIDEO | SDL_INIT_TIMER | SDL_INIT_GAMECONTROLLER) != 0) { @@ -266,6 +296,7 @@ int main(int argc, char* argv[]) { bool AA_ENABLED = true; bool SHOW_HELP = false; bool SHOW_EXIF = false; + bool SHOW_HISTOGRAM = false; int MODE = 0; // Decide GL+GLSL versions @@ -306,9 +337,17 @@ int main(int argc, char* argv[]) { SDL_WindowFlags window_flags = (SDL_WindowFlags)(SDL_WINDOW_OPENGL | SDL_WINDOW_RESIZABLE | SDL_WINDOW_ALLOW_HIGHDPI); + + + int wh = 800; + int ww = 1280; + if (t.size.y > t.size.x) { + ww = 500; + wh = 1280; + } SDL_Window *window = SDL_CreateWindow("tview", SDL_WINDOWPOS_CENTERED, - SDL_WINDOWPOS_CENTERED, 1280, 720, window_flags); + SDL_WINDOWPOS_CENTERED, ww, wh, window_flags); if (window == nullptr) { printf("Error: SDL_CreateWindow(): %s\n", SDL_GetError()); return -1; @@ -334,16 +373,11 @@ int main(int argc, char* argv[]) { ImGui_ImplOpenGL3_Init(glsl_version); ImVec4 clear_color = ImVec4(0.45f, 0.55f, 0.60f, 1.00f); - Texture t; auto flags = ImGuiTexInspect::InspectorFlags_FillVertical | ImGuiTexInspect::InspectorFlags_FillHorizontal; - try { - auto args = argparse::parse(argc, argv, true); - t = LoadTexture(args.fpath.c_str()); - } catch (const std::runtime_error &e) { - std::cerr << "failed to parse arguments: " << e.what() << std::endl; - return -1; - } + t = LoadTexture(t); + Histogram histogram = Histogram(t.size.x, t.size.y, t.channels); + histogram.Load(image); // Main loop bool done = false; @@ -390,6 +424,9 @@ int main(int argc, char* argv[]) { case SDL_SCANCODE_E: SHOW_EXIF = !SHOW_EXIF; break; + case SDL_SCANCODE_C: + SHOW_HISTOGRAM = !SHOW_HISTOGRAM; + break; default: break; } @@ -497,6 +534,7 @@ int main(int argc, char* argv[]) { ImGui::Text("\tFloat Values"); ImGui::Separator(); ImGui::Text("h - show help popup"); + ImGui::Text("c - toggle color histogram"); ImGui::Text("e - toggle EXIF info"); ImGui::Separator(); ImGui::Text("q - quit"); @@ -596,6 +634,12 @@ int main(int argc, char* argv[]) { ImGui::TableSetColumnIndex(1); ImGui::Text("%sx%s", t.exif.ImageDimensionX.c_str(), t.exif.ImageDimensiony.c_str()); + ImGui::TableNextRow(); + ImGui::TableSetColumnIndex(0); + ImGui::Text("Orientation"); + ImGui::TableSetColumnIndex(1); + ImGui::Text("%s", t.exif.ImageOrientation.c_str()); + ImGui::EndTable(); } @@ -623,8 +667,23 @@ int main(int argc, char* argv[]) { ImGui::End(); } + + + if (SHOW_HISTOGRAM) { + ImGuiWindowClass topmost; + topmost.ClassId = ImHashStr("TopMost"); + topmost.ViewportFlagsOverrideSet = ImGuiViewportFlags_TopMost; + ImGui::SetNextWindowClass(&topmost); + ImGui::Begin("Histogram", NULL, ImGuiWindowFlags_NoCollapse | ImGuiWindowFlags_NoResize | ImGuiWindowFlags_AlwaysAutoResize | ImGuiWindowFlags_NoFocusOnAppearing); + histogram.Draw(); + ImGui::Separator(); + ImGui::Text("Press c to hide"); + + ImGui::End(); + } } + // Rendering ImGui::Render(); glViewport(0, 0, (int)io.DisplaySize.x, (int)io.DisplaySize.y);