gwint/image.cpp

91 lines
2.1 KiB
C++

#include <opencv2/highgui.hpp>
#include <opencv2/imgcodecs.hpp>
#include <opencv2/imgproc.hpp>
#include <nlohmann/json.hpp>
#include <iostream>
#include <fstream>
#include <string>
using namespace cv;
using nlohmann::json;
using std::string;
#define H 800
#define W 300
#define M 30
#define R 60
Mat source_image = Mat::zeros(Size(H,W),CV_8UC1);
void render(int i, int j, int y_s, int x_s, Mat *img) {
if(i >= y_s && i <= y_s+R-1 && j >= x_s && j <= x_s+R-1) {
source_image.at<uchar>(i,j)=img->at<uchar>(i-y_s,j-x_s);
}
}
int main(int argc, char** argv) {
// FACTION
int fac;
if(argc <= 1) {
std::cout << "0:U,1:SK,2:M,3:N,4:S";
std::cin >> fac;
}
else fac = std::atoi(argv[1]);
std::ifstream ff("cards/factions.json");
json factions = json::parse(ff);
string faction = factions[fac];
// NAME
string name;
if(argc <= 2) std::cin >> name;
else name = argv[2];
std::ifstream f("cards/"+faction+"/"+name+".json");
json data = json::parse(f);
// LOAD IMAGES
Mat i_look = imread("cards/"+faction+"/images/"+name+".png",IMREAD_GRAYSCALE);
Mat i_faction = imread("cards/images/factions/"+faction+".png",IMREAD_GRAYSCALE);
Mat i_line = imread("cards/images/lines/"+string(data["line"])+".png",IMREAD_GRAYSCALE);
// RENDER IMAGES
std::vector<Mat> i_traits;
for(auto trait : data["traits"]) {
i_traits.push_back(imread("cards/images/traits/"+string(trait)+".png",IMREAD_GRAYSCALE));
}
for(int i=0;i<source_image.rows;i++)
for(int j=0;j<source_image.cols;j++)
source_image.at<uchar>(i,j)=i_look.at<uchar>(i,j);
// RENDER POWER AND TRAITS
string bruh = std::to_string((int)data["power"]);
for(int i=0;i<source_image.rows;i++) {
for(int j=0;j<source_image.cols;j++) {
// Vlevo nahoře sílu
putText(
source_image,
bruh,
Point(M,M),
FONT_HERSHEY_COMPLEX_SMALL,
1,
Scalar(0,0,0),
2
);
render(i,j, M,W/2-R/2, &i_faction);
render(i,j, M,W-M-R, &i_line);
// Traits
for(int k = 0; k < i_traits.size(); k++)
render(i,j, H-M-R,W-M-R*(k+1), &i_traits[k]);
}
}
imshow("Result",source_image);
waitKey(0);
return 0;
}