initial
This commit is contained in:
commit
d328bc9ce2
|
@ -0,0 +1,18 @@
|
||||||
|
*.bin
|
||||||
|
*.out
|
||||||
|
build/
|
||||||
|
*.exe
|
||||||
|
.gdb_history
|
||||||
|
|
||||||
|
*.so
|
||||||
|
*.bmp
|
||||||
|
*.d
|
||||||
|
|
||||||
|
test
|
||||||
|
.vscode/settings.json
|
||||||
|
|
||||||
|
lolautoaccept
|
||||||
|
lolautoaccept_strip
|
||||||
|
log.txt
|
||||||
|
*.png
|
||||||
|
|
|
@ -0,0 +1,3 @@
|
||||||
|
[submodule "thirdparty/Log"]
|
||||||
|
path = thirdparty/Log
|
||||||
|
url = https://git.okaestne.de/okaestne/Log.git
|
|
@ -0,0 +1,20 @@
|
||||||
|
{
|
||||||
|
"configurations": [
|
||||||
|
{
|
||||||
|
"name": "Linux",
|
||||||
|
"includePath": [
|
||||||
|
"${workspaceFolder}/src/",
|
||||||
|
"${workspaceFolder}/thirdparty/Log/",
|
||||||
|
"${workspaceFolder}/include/**",
|
||||||
|
|
||||||
|
"/usr/include/opencv4/**"
|
||||||
|
],
|
||||||
|
"defines": [],
|
||||||
|
"compilerPath": "/usr/bin/g++",
|
||||||
|
"cStandard": "c11",
|
||||||
|
"cppStandard": "c++17",
|
||||||
|
"intelliSenseMode": "gcc-x64"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"version": 4
|
||||||
|
}
|
|
@ -0,0 +1,50 @@
|
||||||
|
{
|
||||||
|
// Use IntelliSense to learn about possible attributes.
|
||||||
|
// Hover to view descriptions of existing attributes.
|
||||||
|
// For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
|
||||||
|
"version": "0.2.0",
|
||||||
|
"configurations": [
|
||||||
|
{
|
||||||
|
"name": "test Debuggen (gdb)",
|
||||||
|
"type": "cppdbg",
|
||||||
|
"request": "launch",
|
||||||
|
"program": "${workspaceFolder}/test",
|
||||||
|
"args": [],
|
||||||
|
"stopAtEntry": false,
|
||||||
|
"cwd": "${workspaceFolder}",
|
||||||
|
"environment": [],
|
||||||
|
"externalConsole": false,
|
||||||
|
"MIMode": "gdb",
|
||||||
|
"setupCommands": [
|
||||||
|
{
|
||||||
|
"description": "Automatische Strukturierung und Einrückung für \"gdb\" aktivieren",
|
||||||
|
"text": "-enable-pretty-printing",
|
||||||
|
"ignoreFailures": true
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"preLaunchTask": "make test",
|
||||||
|
"miDebuggerPath": "/usr/bin/gdb"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "Debuggen (gdb)",
|
||||||
|
"type": "cppdbg",
|
||||||
|
"request": "launch",
|
||||||
|
"program": "${workspaceFolder}/lolautoaccept",
|
||||||
|
"args": [],
|
||||||
|
"stopAtEntry": false,
|
||||||
|
"cwd": "${workspaceFolder}",
|
||||||
|
"environment": [],
|
||||||
|
"externalConsole": false,
|
||||||
|
"MIMode": "gdb",
|
||||||
|
"setupCommands": [
|
||||||
|
{
|
||||||
|
"description": "Automatische Strukturierung und Einrückung für \"gdb\" aktivieren",
|
||||||
|
"text": "-enable-pretty-printing",
|
||||||
|
"ignoreFailures": true
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"preLaunchTask": "make all",
|
||||||
|
"miDebuggerPath": "/usr/bin/gdb"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
|
@ -0,0 +1,45 @@
|
||||||
|
{
|
||||||
|
// See https://go.microsoft.com/fwlink/?LinkId=733558
|
||||||
|
// for the documentation about the tasks.json format
|
||||||
|
"version": "2.0.0",
|
||||||
|
"tasks": [
|
||||||
|
{
|
||||||
|
"label": "make all",
|
||||||
|
"type": "shell",
|
||||||
|
"command": "make -j all",
|
||||||
|
"group": {
|
||||||
|
"kind": "build",
|
||||||
|
"isDefault": true
|
||||||
|
},
|
||||||
|
"problemMatcher": [
|
||||||
|
"$gcc"
|
||||||
|
]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"label": "make clean",
|
||||||
|
"type": "shell",
|
||||||
|
"command": "make clean",
|
||||||
|
"problemMatcher": []
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"label": "leak check",
|
||||||
|
"type": "shell",
|
||||||
|
"command": "/usr/bin/valgrind --leak-check=full ${workspaceFolder}/lolautoaccept -s",
|
||||||
|
"problemMatcher": [],
|
||||||
|
"dependsOn": "make all"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"label": "leak check ALL",
|
||||||
|
"type": "shell",
|
||||||
|
"command": "/usr/bin/valgrind --leak-check=full --show-leak-kinds=all ${workspaceFolder}/lolautoaccept -s",
|
||||||
|
"problemMatcher": [],
|
||||||
|
"dependsOn": "make all"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"label": "make test",
|
||||||
|
"type": "shell",
|
||||||
|
"command": "make -j test",
|
||||||
|
"problemMatcher": ["$gcc"],
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
|
@ -0,0 +1,74 @@
|
||||||
|
# Author Yannis Gerlach
|
||||||
|
# Hochschule Osnabrück
|
||||||
|
# 13.11.2020
|
||||||
|
|
||||||
|
# `make clean all` nicht mit -j verwenden! -> race condition im make file
|
||||||
|
# statdessen: `make clean; make all -j` verwenden
|
||||||
|
|
||||||
|
NAME = lolautoaccept
|
||||||
|
NAMETEST = test
|
||||||
|
CFLAGS = -std=c++17 -O2 -g -pipe -Wall -Wextra -Wno-unused-parameter -Wpedantic -rdynamic -mfpmath=both -march=native -m64 -funroll-loops -mavx2 `pkg-config opencv4 --cflags`
|
||||||
|
CXX = g++
|
||||||
|
SRCF = src/
|
||||||
|
BUILDDIR = build/
|
||||||
|
TESTF = tests/
|
||||||
|
DEPF = $(BUILDDIR)deps/
|
||||||
|
INCF = ./include/
|
||||||
|
INCFS = $(shell find $(INCF) -type d)
|
||||||
|
|
||||||
|
LOGF = ./thirdparty/Log/
|
||||||
|
LOGO = $(LOGF)Log.o
|
||||||
|
|
||||||
|
INCLUDES = -I$(LOGF) $(addprefix -I, $(INCFS))
|
||||||
|
LDFLAGS = -lX11 -lXext `pkg-config opencv4 --libs`
|
||||||
|
|
||||||
|
SRCFILES = $(shell find $(SRCF) -name "*.cpp")
|
||||||
|
OBJFILES = $(patsubst $(SRCF)%, $(BUILDDIR)%, $(patsubst %.cpp, %.o, $(SRCFILES))) $(LOGO)
|
||||||
|
DEPFILES = $(wildcard $(DEPF)*.d)
|
||||||
|
|
||||||
|
SOURCEDIRS = $(shell find $(SRCF) -type d -printf "%p/\n")
|
||||||
|
BUILDDIRS = $(patsubst $(SRCF)%, $(BUILDDIR)%, $(SOURCEDIRS))
|
||||||
|
|
||||||
|
OBJFILESTEST = $(filter-out $(BUILDDIR)main.o, $(OBJFILES))
|
||||||
|
|
||||||
|
INCLUDES += $(addprefix -I, $(SOURCEDIRS))
|
||||||
|
|
||||||
|
all: $(NAME) runtest
|
||||||
|
|
||||||
|
$(NAME): $(BUILDDIRS) $(DEPF) $(OBJFILES)
|
||||||
|
@echo "Linking $@"
|
||||||
|
@$(CXX) $(CFLAGS) -o $@ $(filter %.o, $^) $(LDFLAGS)
|
||||||
|
|
||||||
|
$(BUILDDIR)%.o: $(SRCF)%.cpp
|
||||||
|
@echo "Compiling: $@"
|
||||||
|
@$(CXX) $(CFLAGS) $(INCLUDES) $< -MM -MT $@ > $(DEPF)$(subst /,_,$*).d
|
||||||
|
@$(CXX) -c -o $@ $(CFLAGS) $(INCLUDES) $<
|
||||||
|
|
||||||
|
$(NAME)_strip: $(NAME)
|
||||||
|
@echo "Strip $<"
|
||||||
|
@strip -o $@ $<
|
||||||
|
|
||||||
|
%/:
|
||||||
|
mkdir -p $@
|
||||||
|
|
||||||
|
clean-depends:
|
||||||
|
$(RM) -r $(DEPF)
|
||||||
|
|
||||||
|
$(LOGO):
|
||||||
|
$(MAKE) -C $(LOGF) all
|
||||||
|
|
||||||
|
clean:
|
||||||
|
$(RM) -r $(NAME) $(BUILDDIR) $(NAMETEST) $(NAME)_strip
|
||||||
|
$(MAKE) -C $(LOGF) $@
|
||||||
|
|
||||||
|
$(NAMETEST): $(BUILDDIRS) $(DEPF) $(TESTF)*.cpp $(OBJFILESTEST)
|
||||||
|
@echo "Compiling tests"
|
||||||
|
@$(CXX) -o $@ $(filter %.o, $^) $(filter %.cpp, $^) $(CFLAGS) -I$(SRCF) $(INCLUDES) $(LDFLAGS)
|
||||||
|
|
||||||
|
runtest: $(NAMETEST)
|
||||||
|
@echo "Running tests"
|
||||||
|
./$<
|
||||||
|
|
||||||
|
.PHONY: clean all $(NAMETEST) clean-depends runtest
|
||||||
|
|
||||||
|
include $(DEPFILES)
|
|
@ -0,0 +1,31 @@
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <opencv2/opencv.hpp>
|
||||||
|
|
||||||
|
#include <X11/Xlib.h>
|
||||||
|
#include <X11/Xutil.h>
|
||||||
|
|
||||||
|
#include <X11/extensions/XShm.h>
|
||||||
|
#include <sys/ipc.h>
|
||||||
|
#include <sys/shm.h>
|
||||||
|
|
||||||
|
class ScreenShot {
|
||||||
|
private:
|
||||||
|
Display* display = nullptr;
|
||||||
|
Window root;
|
||||||
|
XWindowAttributes window_attributes;
|
||||||
|
XImage* ximg = nullptr;
|
||||||
|
XShmSegmentInfo shminfo;
|
||||||
|
|
||||||
|
int x, y, width, height;
|
||||||
|
|
||||||
|
bool init;
|
||||||
|
|
||||||
|
public:
|
||||||
|
|
||||||
|
ScreenShot();
|
||||||
|
~ScreenShot();
|
||||||
|
|
||||||
|
void operator() (cv::Mat& cvimg);
|
||||||
|
|
||||||
|
};
|
|
@ -0,0 +1,82 @@
|
||||||
|
|
||||||
|
#include "Log.h"
|
||||||
|
|
||||||
|
#include "screen.h"
|
||||||
|
|
||||||
|
std::string type2str(int type) {
|
||||||
|
std::string r;
|
||||||
|
|
||||||
|
uchar depth = type & CV_MAT_DEPTH_MASK;
|
||||||
|
uchar chans = 1 + (type >> CV_CN_SHIFT);
|
||||||
|
|
||||||
|
switch ( depth ) {
|
||||||
|
case CV_8U: r = "8U"; break;
|
||||||
|
case CV_8S: r = "8S"; break;
|
||||||
|
case CV_16U: r = "16U"; break;
|
||||||
|
case CV_16S: r = "16S"; break;
|
||||||
|
case CV_32S: r = "32S"; break;
|
||||||
|
case CV_32F: r = "32F"; break;
|
||||||
|
case CV_64F: r = "64F"; break;
|
||||||
|
default: r = "User"; break;
|
||||||
|
}
|
||||||
|
|
||||||
|
r += "C";
|
||||||
|
r += (chans+'0');
|
||||||
|
|
||||||
|
return r;
|
||||||
|
}
|
||||||
|
|
||||||
|
int main(int argc, const char** argv) {
|
||||||
|
Log::init();
|
||||||
|
Log::setConsoleLogLevel(Log::Level::TRACE);
|
||||||
|
Log::addLogfile("log.txt", Log::Level::TRACE);
|
||||||
|
#if __unix__
|
||||||
|
Log::setColoredOutput(true);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
Log::info << "Hello, World!";
|
||||||
|
|
||||||
|
// load template
|
||||||
|
cv::Mat tem = cv::imread("imgs/Accept.png", cv::IMREAD_COLOR);
|
||||||
|
// cv::imshow("template", tem);
|
||||||
|
// cv::waitKey(0);
|
||||||
|
|
||||||
|
ScreenShot screen;
|
||||||
|
|
||||||
|
cv::Mat img;
|
||||||
|
screen(img);
|
||||||
|
// cv::imshow("img", img);
|
||||||
|
// cv::waitKey(0);
|
||||||
|
|
||||||
|
Log::info << "img " << img.type();
|
||||||
|
|
||||||
|
cv::Mat out;
|
||||||
|
int result_cols = img.cols - tem.cols + 1;
|
||||||
|
int result_rows = img.rows - tem.rows + 1;
|
||||||
|
out.create(result_rows, result_cols, CV_32FC1);
|
||||||
|
|
||||||
|
Log::info << "tem " << tem.type() << " " << type2str(tem.type());
|
||||||
|
// img.convertTo(img, tem.type());
|
||||||
|
|
||||||
|
cv::cvtColor(img, img, cv::COLOR_RGBA2RGB);
|
||||||
|
Log::info << "converted img: " << img.type() << " " << type2str(img.type());
|
||||||
|
|
||||||
|
cv::matchTemplate(img, tem, out, cv::TM_SQDIFF);
|
||||||
|
cv::normalize(out, out, 0, 1, cv::NORM_MINMAX, -1, cv::Mat());
|
||||||
|
|
||||||
|
double minVal; double maxVal;
|
||||||
|
cv::Point minLoc; cv::Point maxLoc; cv::Point matchLoc;
|
||||||
|
cv::minMaxLoc( out, &minVal, &maxVal, &minLoc, &maxLoc, cv::Mat() );
|
||||||
|
|
||||||
|
matchLoc = minLoc;
|
||||||
|
cv::rectangle(img, matchLoc, cv::Point( matchLoc.x + tem.cols , matchLoc.y + tem.rows ), cv::Scalar(255, 0, 0, 0), 2);
|
||||||
|
|
||||||
|
cv::imshow("result", img);
|
||||||
|
cv::waitKey(0);
|
||||||
|
|
||||||
|
time_t t = time(0);
|
||||||
|
cv::imwrite("debugimages/" + std::to_string(t) + ".png", img);
|
||||||
|
|
||||||
|
Log::stop();
|
||||||
|
return 0;
|
||||||
|
}
|
|
@ -0,0 +1,59 @@
|
||||||
|
// g++ screena.cpp -o screena -lX11 -lXext -Ofast -mfpmath=both -march=native -m64 -funroll-loops -mavx2 `pkg-config opencv --cflags --libs` && ./screena
|
||||||
|
|
||||||
|
// This includes most headers!
|
||||||
|
|
||||||
|
#include "screen.h"
|
||||||
|
|
||||||
|
// https://stackoverflow.com/a/39781697/4399859
|
||||||
|
|
||||||
|
ScreenShot::ScreenShot() {
|
||||||
|
|
||||||
|
display = XOpenDisplay(nullptr);
|
||||||
|
root = DefaultRootWindow(display);
|
||||||
|
|
||||||
|
int count_screens = ScreenCount(display);
|
||||||
|
if(count_screens == 0) {
|
||||||
|
puts("no screen info!");
|
||||||
|
}
|
||||||
|
for (int i = 0; i < count_screens; ++i) {
|
||||||
|
Screen* screen = ScreenOfDisplay(display, i);
|
||||||
|
printf("\tScreen %d: %dX%d\n", i + 1, screen->width, screen->height);
|
||||||
|
}
|
||||||
|
|
||||||
|
XGetWindowAttributes(display, root, &window_attributes);
|
||||||
|
Screen* screen = window_attributes.screen;
|
||||||
|
width = screen->width;
|
||||||
|
height = screen->height;
|
||||||
|
x = 0;
|
||||||
|
y = 0;
|
||||||
|
|
||||||
|
ximg = XShmCreateImage(display, DefaultVisualOfScreen(screen), DefaultDepthOfScreen(screen), ZPixmap, NULL, &shminfo, width, height);
|
||||||
|
|
||||||
|
shminfo.shmid = shmget(IPC_PRIVATE, ximg->bytes_per_line * ximg->height, IPC_CREAT|0777);
|
||||||
|
shminfo.shmaddr = ximg->data = (char*)shmat(shminfo.shmid, 0, 0);
|
||||||
|
shminfo.readOnly = false;
|
||||||
|
if(shminfo.shmid < 0)
|
||||||
|
puts("Fatal shminfo error!");
|
||||||
|
Status s1 = XShmAttach(display, &shminfo);
|
||||||
|
printf("XShmAttach() %s\n", s1 ? "success!" : "failure!");
|
||||||
|
|
||||||
|
init = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
void ScreenShot::operator() (cv::Mat& cv_img){
|
||||||
|
if(init)
|
||||||
|
init = false;
|
||||||
|
|
||||||
|
XShmGetImage(display, root, ximg, 0, 0, 0x00ffffff);
|
||||||
|
cv_img = cv::Mat(height, width, CV_8UC4, ximg->data);
|
||||||
|
}
|
||||||
|
|
||||||
|
ScreenShot::~ScreenShot(){
|
||||||
|
if(!init)
|
||||||
|
XDestroyImage(ximg);
|
||||||
|
|
||||||
|
XShmDetach(display, &shminfo);
|
||||||
|
shmdt(shminfo.shmaddr);
|
||||||
|
XCloseDisplay(display);
|
||||||
|
}
|
||||||
|
|
|
@ -0,0 +1,43 @@
|
||||||
|
#include <stdio.h>
|
||||||
|
#include "test.h"
|
||||||
|
|
||||||
|
#include <map>
|
||||||
|
#include <string>
|
||||||
|
#include <chrono>
|
||||||
|
|
||||||
|
#define RED "\033[1;91m"
|
||||||
|
#define GREEN "\033[1;92m"
|
||||||
|
#define AQUA "\033[1;36m"
|
||||||
|
#define RESET "\033[;1m"
|
||||||
|
|
||||||
|
extern std::map<std::string, test_t> tests;
|
||||||
|
|
||||||
|
void loadTests();
|
||||||
|
|
||||||
|
int main(int argc, char** argv) {
|
||||||
|
auto start = std::chrono::high_resolution_clock::now();
|
||||||
|
|
||||||
|
loadTests();
|
||||||
|
|
||||||
|
int failcount = 0;
|
||||||
|
int testcount = tests.size();
|
||||||
|
int testnumber = 0;
|
||||||
|
for(std::map<std::string, test_t>::iterator current = tests.begin(); current != tests.end(); ++current) {
|
||||||
|
printf("\033[1mRunning test: %d/%d " AQUA "%s " RESET, ++testnumber, testcount, current->first.c_str());
|
||||||
|
if((current->second)()) {
|
||||||
|
printf(GREEN "succeeded" RESET "!\n");
|
||||||
|
} else {
|
||||||
|
printf(RED "failed" RESET "\n");
|
||||||
|
failcount++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const char* color = (failcount > 0 ? RED : GREEN); // red or green
|
||||||
|
printf("%s%d" RESET "/%d failed\n", color, failcount, testcount);
|
||||||
|
|
||||||
|
auto end = std::chrono::high_resolution_clock::now();
|
||||||
|
std::chrono::duration<double> t = end - start;
|
||||||
|
printf("Testing took: %fms\n", (t.count() * 1000));
|
||||||
|
|
||||||
|
return failcount > 0;
|
||||||
|
}
|
|
@ -0,0 +1,6 @@
|
||||||
|
#include "test.h"
|
||||||
|
|
||||||
|
TEST(ABC) {
|
||||||
|
CMPASSERT(1, true);
|
||||||
|
|
||||||
|
} TESTEND
|
|
@ -0,0 +1,17 @@
|
||||||
|
#define TESTFAILED 0
|
||||||
|
#define TESTGOOD 1
|
||||||
|
|
||||||
|
#include <iostream>
|
||||||
|
|
||||||
|
#define TESTDATA "./tests/data/"
|
||||||
|
|
||||||
|
#define TESTNAME(NAME) test_##NAME
|
||||||
|
#define TESTFUNC(NAME) bool TESTNAME(NAME)()
|
||||||
|
#define TEST(NAME) TESTFUNC(NAME) {
|
||||||
|
#define TESTEND return TESTGOOD; }
|
||||||
|
|
||||||
|
#define ASSERT(BED, ERR) if(!(BED)) { std::cout << __FILE__ << ":" << __LINE__ << " " << ERR << std::endl; return TESTFAILED; }
|
||||||
|
#define CMPASSERTE(A, B, ERR) if( !((A) == (B))) { std::cout << __FILE__ << ":" << __LINE__ << " is: \"" << (A) << "\" should: \"" << (B) << "\""<< std::endl; return TESTFAILED; }
|
||||||
|
#define CMPASSERT(A, B) CMPASSERTE(A, B, "")
|
||||||
|
|
||||||
|
typedef bool (*test_t)();
|
|
@ -0,0 +1,14 @@
|
||||||
|
#include "test.h"
|
||||||
|
|
||||||
|
#include <map>
|
||||||
|
#include <string>
|
||||||
|
|
||||||
|
#define REGISTERTEST(NAME) tests.insert({#NAME, TESTNAME(NAME)})
|
||||||
|
|
||||||
|
TESTFUNC(ABC);
|
||||||
|
|
||||||
|
std::map<std::string, test_t> tests;
|
||||||
|
|
||||||
|
void loadTests() {
|
||||||
|
REGISTERTEST(ABC);
|
||||||
|
}
|
|
@ -0,0 +1 @@
|
||||||
|
Subproject commit 96d10d6e72ddaffd8b688db5ef2705e38aff3a75
|
Loading…
Reference in New Issue