Merge from trunk.

[bertos.git] / wizard / const.py

diff --git a/wizard/const.py b/wizard/const.py
index c397b3b8efce6a48eed0ba874348a33d086bbc9b..697cd4637c0db2cb44b3d500e5f14203ca5d69c9 100644 (file)
--- a/wizard/const.py
+++ b/wizard/const.py
@@ -33,6 +33,14 @@
 # Author: Lorenzo Berni <duplo@develer.com>
 #
 
+import os, sys
+
+_tmp = sys.argv[0]
+if os.path.islink(_tmp):
+    _tmp = os.readlink(_tmp)
+DATA_DIR = os.path.dirname(os.path.abspath(_tmp))
+del _tmp
+
 CPU_DEF = {
     "CPU_NAME": "",
     "CPU_DIR": "",
@@ -65,6 +73,8 @@ EXTENSION_FILTER = (
     ".cxx",
     ".h",
     ".c++",
+    ".ld",
+    ".S",
     ".mk",
     "Makefile",
 )