Skip to content

Commit e784a4a

Browse files
committed
Merge branch 'fixOutput'
2 parents adfe0e2 + 5526047 commit e784a4a

File tree

2 files changed

+8
-5
lines changed

2 files changed

+8
-5
lines changed

main.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,19 +18,21 @@ def loadJSON(fInput):
1818
name = data.get('name')
1919
lstTypes = data.get('types')
2020
lstFuncs = data.get('functions')
21+
strOutFolder = os.path.join(os.path.dirname(__file__), 'out')
22+
print "Output files to : %s"%(os.path.abspath(strOutFolder))
2123

2224
from dsFileBuilder import DSFileBuilder
23-
dsb = DSFileBuilder(name, lstTypes)
25+
dsb = DSFileBuilder(name, lstTypes, strOutFolder)
2426
dsb.buildDS()
2527
dicNumpyDS = dsb.getGeneratedNumpyDS()
2628

2729
from kernelFuncBuilder import KernelFuncBuilder
28-
kfb = KernelFuncBuilder(name, lstFuncs)
30+
kfb = KernelFuncBuilder(name, lstFuncs, strOutFolder)
2931
kfb.buildKF()
3032
dicKFuncDS = kfb.getGeneratedKFuncDS()
3133

3234
from oclPyObjGenerator import OCLPyObjGenerator
33-
opg = OCLPyObjGenerator(name, dicNumpyDS, dicKFuncDS)
35+
opg = OCLPyObjGenerator(name, dicNumpyDS, dicKFuncDS, strOutFolder)
3436
opg.generateOCLPyObj()
3537
a = opg.getObj()
3638
print a

oclPyObjGenerator.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -126,12 +126,13 @@ def __init__(self, strName, dicNumpyDS, dicKFuncDS, strFolder='out'):
126126

127127
def getObj(self):
128128
import pkgutil
129+
import imp
129130
modules = pkgutil.iter_modules(path=[self.strOutputFolder])
130131
instance = None
131132
for loader, mod_name, ispkg in modules:
132133
if mod_name == self.className:
133-
loaded_mod = __import__(self.strRelFolder+'.'+mod_name, fromlist=[mod_name])
134-
loaded_class = getattr(loaded_mod, self.className)
134+
module_src = imp.load_source(os.path.splitext(self.strFileName)[0], self.strFilePath)
135+
loaded_class = getattr(module_src, self.className)
135136
instance = loaded_class()
136137
break
137138
return instance

0 commit comments

Comments
 (0)