get proper main.cpp

main.cpp

@@ -37,6 +37,8 @@
 
 #include <exiv2/exiv2.hpp>
 
+#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<Args>(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 {
+  t = LoadTexture(t);
-    auto args = argparse::parse<Args>(argc, argv, true);
+  Histogram histogram = Histogram(t.size.x, t.size.y, t.channels);
-    t = LoadTexture(args.fpath.c_str());
+  histogram.Load(image);
-  } catch (const std::runtime_error &e) {
-    std::cerr << "failed to parse arguments: " << e.what() << std::endl;
-    return -1;
-  }
 
   // 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);