diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..88315da --- /dev/null +++ b/.gitignore @@ -0,0 +1,50 @@ +# Byte-compiled / optimized / DLL files +__pycache__/ +*.py[cod] + +# C extensions +*.so + +# Distribution / packaging +bin/ +build/ +develop-eggs/ +dist/ +eggs/ +lib/ +lib64/ +parts/ +sdist/ +var/ +*.egg-info/ +.installed.cfg +*.egg + +# Installer logs +pip-log.txt +pip-delete-this-directory.txt + +# Unit test / coverage reports +.tox/ +.coverage +.cache +nosetests.xml +coverage.xml + +# Translations +*.mo + +# Mr Developer +.mr.developer.cfg +.project +.pydevproject + +# Rope +.ropeproject + +# Django stuff: +*.log +*.pot + +# Sphinx documentation +docs/_build/ \ No newline at end of file diff --git a/.pylintrc b/.pylintrc new file mode 100644 index 0000000..39e11dd --- /dev/null +++ b/.pylintrc @@ -0,0 +1,570 @@ +[MAIN] + +# Specify a configuration file. +#rcfile= + +# Python code to execute, usually for sys.path manipulation such as +# pygtk.require(). +#init-hook= + +# Files or directories to be skipped. They should be base names, not +# paths. +ignore=CVS + +# Add files or directories matching the regex patterns to the ignore-list. The +# regex matches against paths and can be in Posix or Windows format. +ignore-paths= + +# Files or directories matching the regex patterns are skipped. The regex +# matches against base names, not paths. +ignore-patterns=^\.# + +# Pickle collected data for later comparisons. +persistent=yes + +# List of plugins (as comma separated values of python modules names) to load, +# usually to register additional checkers. +load-plugins= + pylint.extensions.check_elif, + pylint.extensions.bad_builtin, + pylint.extensions.docparams, + pylint.extensions.for_any_all, + pylint.extensions.set_membership, + pylint.extensions.code_style, + pylint.extensions.overlapping_exceptions, + pylint.extensions.typing, + pylint.extensions.redefined_variable_type, + pylint.extensions.comparison_placement, + pylint.extensions.mccabe, + +# Use multiple processes to speed up Pylint. Specifying 0 will auto-detect the +# number of processors available to use. +jobs=0 + +# When enabled, pylint would attempt to guess common misconfiguration and emit +# user-friendly hints instead of false-positive error messages. +suggestion-mode=yes + +# Allow loading of arbitrary C extensions. Extensions are imported into the +# active Python interpreter and may run arbitrary code. +unsafe-load-any-extension=no + +# A comma-separated list of package or module names from where C extensions may +# be loaded. Extensions are loading into the active Python interpreter and may +# run arbitrary code +extension-pkg-allow-list= + +# Minimum supported python version +py-version = 3.7.2 + +# Control the amount of potential inferred values when inferring a single +# object. This can help the performance when dealing with large functions or +# complex, nested conditions. +limit-inference-results=100 + +# Specify a score threshold to be exceeded before program exits with error. +fail-under=10.0 + +# Return non-zero exit code if any of these messages/categories are detected, +# even if score is above --fail-under value. Syntax same as enable. Messages +# specified are enabled, while categories only check already-enabled messages. +fail-on= + + +[MESSAGES CONTROL] + +# Only show warnings with the listed confidence levels. Leave empty to show +# all. Valid levels: HIGH, INFERENCE, INFERENCE_FAILURE, UNDEFINED +# confidence= + +# Enable the message, report, category or checker with the given id(s). You can +# either give multiple identifier separated by comma (,) or put this option +# multiple time (only on the command line, not in the configuration file where +# it should appear only once). See also the "--disable" option for examples. +enable= + use-symbolic-message-instead, + useless-suppression, + +# Disable the message, report, category or checker with the given id(s). You +# can either give multiple identifiers separated by comma (,) or put this +# option multiple times (only on the command line, not in the configuration +# file where it should appear only once).You can also use "--disable=all" to +# disable everything first and then re-enable specific checks. For example, if +# you want to run only the similarities checker, you can use "--disable=all +# --enable=similarities". If you want to run only the classes checker, but have +# no Warning level messages displayed, use"--disable=all --enable=classes +# --disable=W" + +disable= + attribute-defined-outside-init, + invalid-name, + missing-docstring, + protected-access, + too-few-public-methods, + # handled by black + format, + # We anticipate #3512 where it will become optional + fixme, + cyclic-import, + import-error, + no-member + + +[REPORTS] + +# Set the output format. Available formats are text, parseable, colorized, msvs +# (visual studio) and html. You can also give a reporter class, eg +# mypackage.mymodule.MyReporterClass. +output-format=text + +# Tells whether to display a full report or only the messages +reports=no + +# Python expression which should return a note less than 10 (10 is the highest +# note). You have access to the variables 'fatal', 'error', 'warning', 'refactor', 'convention' +# and 'info', which contain the number of messages in each category, as +# well as 'statement', which is the total number of statements analyzed. This +# score is used by the global evaluation report (RP0004). +evaluation=max(0, 0 if fatal else 10.0 - ((float(5 * error + warning + refactor + convention) / statement) * 10)) + +# Template used to display messages. This is a python new-style format string +# used to format the message information. See doc for all details +#msg-template= + +# Activate the evaluation score. +score=yes + + +[LOGGING] + +# Logging modules to check that the string format arguments are in logging +# function parameter format +logging-modules=logging + +# The type of string formatting that logging methods do. `old` means using % +# formatting, `new` is for `{}` formatting. +logging-format-style=old + + +[MISCELLANEOUS] + +# List of note tags to take in consideration, separated by a comma. +notes=FIXME,XXX,TODO + +# Regular expression of note tags to take in consideration. +#notes-rgx= + + +[SIMILARITIES] + +# Minimum lines number of a similarity. +min-similarity-lines=6 + +# Ignore comments when computing similarities. +ignore-comments=yes + +# Ignore docstrings when computing similarities. +ignore-docstrings=yes + +# Ignore imports when computing similarities. +ignore-imports=yes + +# Signatures are removed from the similarity computation +ignore-signatures=yes + + +[VARIABLES] + +# Tells whether we should check for unused import in __init__ files. +init-import=no + +# A regular expression matching the name of dummy variables (i.e. expectedly +# not used). +dummy-variables-rgx=_$|dummy + +# List of additional names supposed to be defined in builtins. Remember that +# you should avoid defining new builtins when possible. +additional-builtins= + +# List of strings which can identify a callback function by name. A callback +# name must start or end with one of those strings. +callbacks=cb_,_cb + +# Tells whether unused global variables should be treated as a violation. +allow-global-unused-variables=yes + +# List of names allowed to shadow builtins +allowed-redefined-builtins= + +# Argument names that match this expression will be ignored. Default to name +# with leading underscore. +ignored-argument-names=_.* + +# List of qualified module names which can have objects that can redefine +# builtins. +redefining-builtins-modules=six.moves,past.builtins,future.builtins,builtins,io + + +[FORMAT] + +# Maximum number of characters on a single line. +max-line-length=120 + +# Regexp for a line that is allowed to be longer than the limit. +ignore-long-lines=^\s*(# )??$ + +# Allow the body of an if to be on the same line as the test if there is no +# else. +single-line-if-stmt=no + +# Allow the body of a class to be on the same line as the declaration if body +# contains single statement. +single-line-class-stmt=no + +# Maximum number of lines in a module +max-module-lines=1000 + +# String used as indentation unit. This is usually " " (4 spaces) or "\t" (1 +# tab). +indent-string=' ' + +# Number of spaces of indent required inside a hanging or continued line. +indent-after-paren=4 + +# Expected format of line ending, e.g. empty (any line ending), LF or CRLF. +expected-line-ending-format= + + +[BASIC] + +# Good variable names which should always be accepted, separated by a comma +good-names=i,j,k,ex,Run,_ + +# Good variable names regexes, separated by a comma. If names match any regex, +# they will always be accepted +good-names-rgxs= + +# Bad variable names which should always be refused, separated by a comma +bad-names=foo,bar,baz,toto,tutu,tata + +# Bad variable names regexes, separated by a comma. If names match any regex, +# they will always be refused +bad-names-rgxs= + +# Colon-delimited sets of names that determine each other's naming style when +# the name regexes allow several styles. +name-group= + +# Include a hint for the correct naming format with invalid-name +include-naming-hint=no + +# Naming style matching correct function names. +function-naming-style=snake_case + +# Regular expression matching correct function names +function-rgx=[a-z_][a-z0-9_]{2,30}$ + +# Naming style matching correct variable names. +variable-naming-style=snake_case + +# Regular expression matching correct variable names +variable-rgx=[a-z_][a-z0-9_]{2,30}$ + +# Naming style matching correct constant names. +const-naming-style=UPPER_CASE + +# Regular expression matching correct constant names +const-rgx=(([A-Z_][A-Z0-9_]*)|(__.*__))$ + +# Naming style matching correct attribute names. +attr-naming-style=snake_case + +# Regular expression matching correct attribute names +attr-rgx=[a-z_][a-z0-9_]{2,}$ + +# Naming style matching correct argument names. +argument-naming-style=snake_case + +# Regular expression matching correct argument names +argument-rgx=[a-z_][a-z0-9_]{2,30}$ + +# Naming style matching correct class attribute names. +class-attribute-naming-style=any + +# Regular expression matching correct class attribute names +class-attribute-rgx=([A-Za-z_][A-Za-z0-9_]{2,30}|(__.*__))$ + +# Naming style matching correct class constant names. +class-const-naming-style=UPPER_CASE + +# Regular expression matching correct class constant names. Overrides class- +# const-naming-style. +#class-const-rgx= + +# Naming style matching correct inline iteration names. +inlinevar-naming-style=any + +# Regular expression matching correct inline iteration names +inlinevar-rgx=[A-Za-z_][A-Za-z0-9_]*$ + +# Naming style matching correct class names. +class-naming-style=PascalCase + +# Regular expression matching correct class names +class-rgx=[A-Z_][a-zA-Z0-9]+$ + + +# Naming style matching correct module names. +module-naming-style=snake_case + +# Regular expression matching correct module names +module-rgx=(([a-z_][a-z0-9_]*)|([A-Z][a-zA-Z0-9]+))$ + + +# Naming style matching correct method names. +method-naming-style=snake_case + +# Regular expression matching correct method names +method-rgx=[a-z_][a-z0-9_]{2,}$ + +# Regular expression which can overwrite the naming style set by typevar-naming-style. +#typevar-rgx= + +# Regular expression which should only match function or class names that do +# not require a docstring. Use ^(?!__init__$)_ to also check __init__. +no-docstring-rgx=__.*__ + +# Minimum line length for functions/classes that require docstrings, shorter +# ones are exempt. +docstring-min-length=-1 + +# List of decorators that define properties, such as abc.abstractproperty. +property-classes=abc.abstractproperty + + +[TYPECHECK] + +# Regex pattern to define which classes are considered mixins if ignore-mixin- +# members is set to 'yes' +mixin-class-rgx=.*MixIn + +# List of module names for which member attributes should not be checked +# (useful for modules/projects where namespaces are manipulated during runtime +# and thus existing member attributes cannot be deduced by static analysis). It +# supports qualified module names, as well as Unix pattern matching. +ignored-modules= + +# List of class names for which member attributes should not be checked (useful +# for classes with dynamically set attributes). This supports the use of +# qualified names. +ignored-classes=SQLObject, optparse.Values, thread._local, _thread._local + +# List of members which are set dynamically and missed by pylint inference +# system, and so shouldn't trigger E1101 when accessed. Python regular +# expressions are accepted. +generated-members=REQUEST,acl_users,aq_parent,argparse.Namespace + +# List of decorators that create context managers from functions, such as +# contextlib.contextmanager. +contextmanager-decorators=contextlib.contextmanager + +# Tells whether to warn about missing members when the owner of the attribute +# is inferred to be None. +ignore-none=yes + +# This flag controls whether pylint should warn about no-member and similar +# checks whenever an opaque object is returned when inferring. The inference +# can return multiple potential results while evaluating a Python object, but +# some branches might not be evaluated, which results in partial inference. In +# that case, it might be useful to still emit no-member and other checks for +# the rest of the inferred objects. +ignore-on-opaque-inference=yes + +# Show a hint with possible names when a member name was not found. The aspect +# of finding the hint is based on edit distance. +missing-member-hint=yes + +# The minimum edit distance a name should have in order to be considered a +# similar match for a missing member name. +missing-member-hint-distance=1 + +# The total number of similar names that should be taken in consideration when +# showing a hint for a missing member. +missing-member-max-choices=1 + +[SPELLING] + +# Spelling dictionary name. Available dictionaries: none. To make it working +# install python-enchant package. +spelling-dict= + +# List of comma separated words that should not be checked. +spelling-ignore-words= + +# List of comma separated words that should be considered directives if they +# appear and the beginning of a comment and should not be checked. +spelling-ignore-comment-directives=fmt: on,fmt: off,noqa:,noqa,nosec,isort:skip,mypy:,pragma:,# noinspection + +# A path to a file that contains private dictionary; one word per line. +spelling-private-dict-file=.pyenchant_pylint_custom_dict.txt + +# Tells whether to store unknown words to indicated private dictionary in +# --spelling-private-dict-file option instead of raising a message. +spelling-store-unknown-words=no + +# Limits count of emitted suggestions for spelling mistakes. +max-spelling-suggestions=2 + + +[DESIGN] + +# Maximum number of arguments for function / method +max-args=10 + +# Maximum number of locals for function / method body +max-locals=25 + +# Maximum number of return / yield for function / method body +max-returns=11 + +# Maximum number of branch for function / method body +max-branches=27 + +# Maximum number of statements in function / method body +max-statements=100 + +# Maximum number of parents for a class (see R0901). +max-parents=7 + +# List of qualified class names to ignore when counting class parents (see R0901). +ignored-parents= + +# Maximum number of attributes for a class (see R0902). +max-attributes=11 + +# Minimum number of public methods for a class (see R0903). +min-public-methods=2 + +# Maximum number of public methods for a class (see R0904). +max-public-methods=25 + +# Maximum number of boolean expressions in an if statement (see R0916). +max-bool-expr=5 + +# List of regular expressions of class ancestor names to +# ignore when counting public methods (see R0903). +exclude-too-few-public-methods= + +max-complexity=10 + +[CLASSES] + +# List of method names used to declare (i.e. assign) instance attributes. +defining-attr-methods=__init__,__new__,setUp,__post_init__ + +# List of valid names for the first argument in a class method. +valid-classmethod-first-arg=cls + +# List of valid names for the first argument in a metaclass class method. +valid-metaclass-classmethod-first-arg=mcs + +# List of member names, which should be excluded from the protected access +# warning. +exclude-protected=_asdict,_fields,_replace,_source,_make + +# Warn about protected attribute access inside special methods +check-protected-access-in-special-methods=no + +[IMPORTS] + +# List of modules that can be imported at any level, not just the top level +# one. +allow-any-import-level= + +# Allow wildcard imports from modules that define __all__. +allow-wildcard-with-all=no + +# Analyse import fallback blocks. This can be used to support both Python 2 and +# 3 compatible code, which means that the block might have code that exists +# only in one or another interpreter, leading to false positives when analysed. +analyse-fallback-blocks=no + +# Deprecated modules which should not be used, separated by a comma +deprecated-modules=regsub,TERMIOS,Bastion,rexec + +# Create a graph of every (i.e. internal and external) dependencies in the +# given file (report RP0402 must not be disabled) +import-graph= + +# Create a graph of external dependencies in the given file (report RP0402 must +# not be disabled) +ext-import-graph= + +# Create a graph of internal dependencies in the given file (report RP0402 must +# not be disabled) +int-import-graph= + +# Force import order to recognize a module as part of the standard +# compatibility libraries. +known-standard-library= + +# Force import order to recognize a module as part of a third party library. +known-third-party=enchant + +# Couples of modules and preferred modules, separated by a comma. +preferred-modules= + + +[EXCEPTIONS] + +# Exceptions that will emit a warning when being caught. Defaults to +# "Exception" +overgeneral-exceptions=Exception + + +[TYPING] + +# Set to ``no`` if the app / library does **NOT** need to support runtime +# introspection of type annotations. If you use type annotations +# **exclusively** for type checking of an application, you're probably fine. +# For libraries, evaluate if some users what to access the type hints at +# runtime first, e.g., through ``typing.get_type_hints``. Applies to Python +# versions 3.7 - 3.9 +runtime-typing = no + + +[DEPRECATED_BUILTINS] + +# List of builtins function names that should not be used, separated by a comma +bad-functions=map,input + + +[REFACTORING] + +# Maximum number of nested blocks for function / method body +max-nested-blocks=5 + +# Complete name of functions that never returns. When checking for +# inconsistent-return-statements if a never returning function is called then +# it will be considered as an explicit return statement and no message will be +# printed. +never-returning-functions=sys.exit,argparse.parse_error + + +[STRING] + +# This flag controls whether inconsistent-quotes generates a warning when the +# character used as a quote delimiter is used inconsistently within a module. +check-quote-consistency=no + +# This flag controls whether the implicit-str-concat should generate a warning +# on implicit string concatenation in sequences defined over several lines. +check-str-concat-over-line-jumps=no + + +[CODE_STYLE] + +# Max line length for which to sill emit suggestions. Used to prevent optional +# suggestions which would get split by a code formatter (e.g., black). Will +# default to the setting for ``max-line-length``. +#max-line-length-suggestions= diff --git a/bots/fishing/FishingAgent.py b/bots/fishing/FishingAgent.py new file mode 100644 index 0000000..93a7017 --- /dev/null +++ b/bots/fishing/FishingAgent.py @@ -0,0 +1,80 @@ +"""Module providing a fishing agent.""" + +import os +from threading import Thread +import time +import cv2 as cv +import numpy as np +import pyautogui + +class FishingAgent: + def __init__(self, mainAgent) -> None: + self.mainAgent = mainAgent + + # interpolate here_path to get the path to the fishing target image + assetPath = os.path.join( + os.path.dirname(os.path.realpath(__file__)), + "assets" + ) + assetPath = os.path.dirname(os.path.realpath(__file__)) + print(assetPath) + self.fishingBobber = cv.imread( + os.path.join( + assetPath, + "assets", "Bobber.png" + ) + ) + self.fishingThread = None + + def agentLog(self, log: str): + print("Fishing Agent: " + log) + + def castLure(self): + self.agentLog("Casting Lure.") + # pyautogui.press('1') + time.sleep(1) + self.findLure() + + def findLure(self): + if self.mainAgent.currImg is not None: + self.agentLog("Searching for Bobber.") + cv.imshow("Bobber", self.fishingBobber) + + lureLoc = cv.matchTemplate(self.mainAgent.currImg, self.fishingBobber, cv.TM_CCORR_NORMED) + lureLocArray = np.array(lureLoc) + + minVal, maxVal, minLoc, maxLoc = cv.minMaxLoc(lureLocArray) + self.moveToLure(maxLoc) + else: + print("No curr img") + + def moveToLure(self, maxLoc): + if maxLoc: + self.agentLog("Moving to Bobber.") + pyautogui.moveTo(maxLoc) + + def watchLure(self): + self.agentLog("Waiting for fish...") + + def pullLine(self): + self.agentLog("Pulling line!") + + def run(self): + if self.mainAgent.currImg is None: + print("Image capture not found! Did you start the screen capture thread?") + return + print("Starting fishing thread in 10 seconds...") + time.sleep(3) + + # print("Switching to fishing hotbar (hotbar 4)") + # pyautogui.keyDown('shift') + # pyautogui.press('4') + # pyautogui.keyUp('shift') + # time.sleep(1) + + self.fishingThread = Thread( + target=self.castLure, + args=(), + name="fishing thread", + daemon=True) + self.fishingThread.start() \ No newline at end of file diff --git a/bots/fishing/assets/Bobber.png b/bots/fishing/assets/Bobber.png new file mode 100644 index 0000000..07f0568 Binary files /dev/null and b/bots/fishing/assets/Bobber.png differ diff --git a/bots/fishing/assets/FishIcon.png b/bots/fishing/assets/FishIcon.png new file mode 100644 index 0000000..832155b Binary files /dev/null and b/bots/fishing/assets/FishIcon.png differ diff --git a/bots/fishing/assets/FishIconScreen.png b/bots/fishing/assets/FishIconScreen.png new file mode 100644 index 0000000..fe1af3f Binary files /dev/null and b/bots/fishing/assets/FishIconScreen.png differ diff --git a/bots/fishing/assets/Test3.png b/bots/fishing/assets/Test3.png new file mode 100644 index 0000000..89e02b8 Binary files /dev/null and b/bots/fishing/assets/Test3.png differ diff --git a/bots/fishing/assets/pecheTest2.png b/bots/fishing/assets/pecheTest2.png new file mode 100644 index 0000000..30b5dee Binary files /dev/null and b/bots/fishing/assets/pecheTest2.png differ diff --git a/bots/fishing/assets/test.png b/bots/fishing/assets/test.png new file mode 100644 index 0000000..f947c6f Binary files /dev/null and b/bots/fishing/assets/test.png differ diff --git a/core/MainAgent.py b/core/MainAgent.py new file mode 100644 index 0000000..7d25ddd --- /dev/null +++ b/core/MainAgent.py @@ -0,0 +1,18 @@ +from threading import Thread +from core.Monitor import Monitor + +class MainAgent: + def __init__(self) -> None: + self.monitor = Monitor() + self.screenThread = None + + def startScreenCaptureThread(self): + self.screenThread = Thread(target=self.monitor.updateScreen, args=(self,), name="Update screen thread", daemon=True) + self.screenThread.start() + print("Main Agent: Thread started") + + def stopScreenCaptureThread(self): + if (self.screenThread): + self.screenThread.terminate() + print("Main Agent: Thread terminated") + diff --git a/core/Monitor.py b/core/Monitor.py new file mode 100644 index 0000000..0c9a66c --- /dev/null +++ b/core/Monitor.py @@ -0,0 +1,39 @@ +import time +import pyautogui +import numpy as np +import cv2 as cv +from screeninfo import get_monitors + +FPS_REPORT_DELAY = 3 + +class Monitor: + def __init__(self): + self.monitor = None + self.screenshot = None + self.getMonitor() + + def getMonitor(self): + for monitor in get_monitors(): + if (monitor.is_primary): + self.monitor = monitor + + def updateScreen(self): + print("Starting computer vision screen update...") + print("Detected display resolution: " + str(self.monitor.width) + " x " + str(self.monitor.height)) + + loopTime = time.time() + fpsPrintTime = time.time() + while True: + newScreenshot = pyautogui.screenshot() + newScreenshot = cv.cvtColor(np.array(newScreenshot), cv.COLOR_RGB2BGR) + + cv.imwrite("in_memory_to_disk.png", newScreenshot) + self.screenshot = newScreenshot + + currTime = time.time() + if currTime - fpsPrintTime >= FPS_REPORT_DELAY: + print('FPS: {}'.format(1 / (currTime - loopTime))) + fpsPrintTime = currTime + loopTime = currTime + + \ No newline at end of file diff --git a/core/cli/CliAgent.py b/core/cli/CliAgent.py new file mode 100644 index 0000000..a91e5f3 --- /dev/null +++ b/core/cli/CliAgent.py @@ -0,0 +1,43 @@ +import cv2 as cv + +class CLIAgent: + def __init__(self, mainAgent) -> None: + self.isRunning = False + if (not mainAgent): + raise Exception("CLI: Main agent not found...") + self.mainAgent = mainAgent + + def printMenu(self): + print("Enter a command:") + print("\tH\tHelp.") + print("\tF\tStart the fishing agent.") + print("\tQ\tQuit.") + print("$>> ", end='') + + def run(self): + self.isRunning = True + + self.printMenu() + while self.isRunning: + userInput = input() + userInput = str.lower(userInput).strip() + + if (userInput == "f"): + print("Starting Fishing bot...") + self.mainAgent.startScreenCaptureThread() + break + # fishingAgent = FishingAgent(mainAgent) + # fishingAgent.run() + elif (userInput == "h"): + self.printMenu() + break + elif (userInput == "q"): + cv.destroyAllWindows() + print("Exiting application...") + self.isRunning = False + break + else: + print("Unknown Command.") + self.printMenu() + break + diff --git a/in_memory_to_disk.png b/in_memory_to_disk.png new file mode 100644 index 0000000..eb94cab Binary files /dev/null and b/in_memory_to_disk.png differ diff --git a/main.py b/main.py new file mode 100644 index 0000000..540dca3 --- /dev/null +++ b/main.py @@ -0,0 +1,9 @@ +from core.cli.CliAgent import CLIAgent +from core.MainAgent import MainAgent + +###### MAIN ######## +if __name__ == "__main__": + mainAgent = MainAgent() + cliAgent = CLIAgent(mainAgent) + + cliAgent.run() \ No newline at end of file diff --git a/tests/assets/Bobber.png b/tests/assets/Bobber.png new file mode 100644 index 0000000..07f0568 Binary files /dev/null and b/tests/assets/Bobber.png differ diff --git a/tests/assets/Bobber2.png b/tests/assets/Bobber2.png new file mode 100644 index 0000000..9bbe4b4 Binary files /dev/null and b/tests/assets/Bobber2.png differ diff --git a/tests/assets/FishIcon.png b/tests/assets/FishIcon.png new file mode 100644 index 0000000..832155b Binary files /dev/null and b/tests/assets/FishIcon.png differ diff --git a/tests/assets/FishIconScreen.png b/tests/assets/FishIconScreen.png new file mode 100644 index 0000000..fe1af3f Binary files /dev/null and b/tests/assets/FishIconScreen.png differ diff --git a/tests/assets/FishTest3.png b/tests/assets/FishTest3.png new file mode 100644 index 0000000..e63c9c8 Binary files /dev/null and b/tests/assets/FishTest3.png differ diff --git a/tests/assets/pecheTest2.png b/tests/assets/pecheTest2.png new file mode 100644 index 0000000..30b5dee Binary files /dev/null and b/tests/assets/pecheTest2.png differ diff --git a/tests/templateMatching.py b/tests/templateMatching.py new file mode 100644 index 0000000..cc11ab1 --- /dev/null +++ b/tests/templateMatching.py @@ -0,0 +1,48 @@ +import os +import cv2 as cv +from matplotlib import pyplot as plt + +# interpolate here_path to get the path to the fishing target image +herePath = os.path.dirname(os.path.realpath(__file__)) + +img = cv.imread(os.path.join(herePath, "assets", "Bobber2.png"), cv.IMREAD_GRAYSCALE) +assert img is not None, "file could not be read, check with os.path.exists()" +img2 = img.copy() +template = cv.imread(os.path.join(herePath, "assets", "FishTest3.png"), cv.IMREAD_GRAYSCALE) +assert template is not None, "file could not be read, check with os.path.exists()" +w, h = template.shape[::-1] + +# All the 6 methods for comparison in a list +methods = ['TM_CCOEFF', 'TM_CCOEFF_NORMED', 'TM_CCORR', + 'TM_CCORR_NORMED', 'TM_SQDIFF', 'TM_SQDIFF_NORMED'] + +for meth in methods: + img = img2.copy() + method = getattr(cv, meth) + + # Apply template Matching + res = cv.matchTemplate(img,template,method) + min_val, max_val, min_loc, max_loc = cv.minMaxLoc(res) + + # If the method is TM_SQDIFF or TM_SQDIFF_NORMED, take minimum + if method in [cv.TM_SQDIFF, cv.TM_SQDIFF_NORMED]: + top_left = min_loc + else: + top_left = max_loc + bottom_right = (top_left[0] + w, top_left[1] + h) + + cv.rectangle(img,top_left, bottom_right, 255, 2) + + plt.subplot(121) + plt.imshow(res,cmap = 'gray') + plt.title('Matching Result') + plt.xticks([]) + plt.yticks([]) + plt.subplot(122) + plt.imshow(img,cmap = 'gray') + plt.title('Detected Point') + plt.xticks([]) + plt.yticks([]) + plt.suptitle(meth) + + plt.show() \ No newline at end of file