From e3616ecc59251857aa7880eb6d836900eff7e04c Mon Sep 17 00:00:00 2001
From: Maxim Poliakovski <maximumspatium@googlemail.com>
Date: Fri, 22 Oct 2021 13:00:17 +0200
Subject: [PATCH] Use main script's path to locate the database directory.

This is required when running the tool from another
project or IDE.
---
 PowerRomDasm.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/PowerRomDasm.py b/PowerRomDasm.py
index b622687..984dade 100644
--- a/PowerRomDasm.py
+++ b/PowerRomDasm.py
@@ -12,6 +12,7 @@ from ruamel.yaml import YAML
 from capstone import *
 from capstone.m68k import *
 
+import os
 import struct
 
 def bit_not(n, numbits=32):
@@ -317,9 +318,11 @@ if __name__ == "__main__":
         check_sum = struct.unpack('>I', rom_data[0:4])[0]
         print("ROM Checksum: %X" % check_sum)
 
+        my_path = os.path.dirname(os.path.realpath(__file__))
+
         db_name = 'ROMDB_' + '{:x}'.format(int(check_sum)).upper() + '.yaml'
 
-        with open('database/' + db_name, 'rb') as db_file:
+        with open(my_path + '/database/' + db_name, 'rb') as db_file:
             yaml = YAML()
             annot_db = yaml.load(db_file)