fixed trex module formatting to account for CI hierarchy

Author: zeniheisser

epochX/cudacpp/CODEGEN/PLUGIN/CUDACPP_SA_OUTPUT/trex.py

@@ -8,26 +8,34 @@
 import re
 import sys
 import importlib.util
-SPEC_EXPORTCPP = importlib.util.find_spec('madgraph.iolibs.export_cpp')
-PLUGIN_export_cpp = importlib.util.module_from_spec(SPEC_EXPORTCPP)
-SPEC_EXPORTCPP.loader.exec_module(PLUGIN_export_cpp)
-sys.modules['PLUGIN.CUDACPP_OUTPUT.PLUGIN_export_cpp'] = PLUGIN_export_cpp # allow 'import PLUGIN.CUDACPP_OUTPUT.PLUGIN_export_cpp' in model_handling.py
-del SPEC_EXPORTCPP
-###print('id(export_cpp)=%s'%id(export_cpp))
-###print('id(PLUGIN_export_cpp)=%s'%id(PLUGIN_export_cpp))
 
-# AV - use template files from PLUGINDIR instead of MG5DIR
+# AV - PLUGIN_NAME can be one of PLUGIN/CUDACPP_OUTPUT or MG5aMC_PLUGIN/CUDACPP_OUTPUT
+PLUGIN_NAME = __name__.rsplit('.',1)[0]
+
+# AV - use templates for source code, scripts and Makefiles from PLUGINDIR instead of MG5DIR
 ###from madgraph import MG5DIR
 PLUGINDIR = os.path.dirname( __file__ )
 
-# AV - model_handling includes the custom FileWriter, ALOHAWriter, UFOModelConverter, OneProcessExporter and HelasCallWriter, plus additional patches
-import PLUGIN.CUDACPP_OUTPUT.model_handling as model_handling
-import PLUGIN.CUDACPP_OUTPUT.output as output
+__import__('%s.output'%PLUGIN_NAME)
+output = sys.modules['%s.output'%PLUGIN_NAME]
+__import__('%s.model_handling'%PLUGIN_NAME)
+model_handling = sys.modules['%s.model_handling'%PLUGIN_NAME]
+
+import importlib.util
+SPEC_EXPORTCPP = importlib.util.find_spec('madgraph.iolibs.export_cpp')
+PLUGIN_export_cpp = importlib.util.module_from_spec(SPEC_EXPORTCPP)
+SPEC_EXPORTCPP.loader.exec_module(PLUGIN_export_cpp)
+###sys.modules['PLUGIN.CUDACPP_OUTPUT.PLUGIN_export_cpp'] = PLUGIN_export_cpp # allow 'import PLUGIN.CUDACPP_OUTPUT.PLUGIN_export_cpp' in model_handling.py
+sys.modules['%s.PLUGIN_export_cpp'%PLUGIN_NAME] = PLUGIN_export_cpp # allow 'import <PLUGIN_NAME>.PLUGIN_export_cpp' in model_handling.py
+del SPEC_EXPORTCPP
 
 # AV - create a plugin-specific logger
 import logging
-logger = logging.getLogger('madgraph.PLUGIN.CUDACPP_OUTPUT.output')
-from madgraph import MG5DIR
+logger = logging.getLogger('madgraph.%s.model_handling'%PLUGIN_NAME)
+
+#------------------------------------------------------------------------------------
+
+
 #------------------------------------------------------------------------------------
 
 from os.path import join as pjoin