get proper main.cpp

This commit is contained in:
Tanishq Dubey 2024-06-16 10:29:51 -04:00
parent 01e85f5dc8
commit 3dd8f101a9

103
main.cpp
View File

@ -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 {
auto args = argparse::parse<Args>(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);