custom treatment of the commonBeam models generator folder

kbat · Dec 12, 2024 · 1a800ea · 1a800ea
1 parent a5b6a22
commit 1a800ea
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/mctools/common/CombLayer/paster/cl-add-variable.py b/mctools/common/CombLayer/paster/cl-add-variable.py
@@ -174,9 +174,12 @@ def main():
     h   = os.path.join(hDir,   args.className + ".h")
 
     if args.generator == "":
-        cxxGenDir = os.path.join('',*cxxDir.split('/')[:-1], "commonGenerator")
-        if not checkPaths([cxxGenDir],()):
-            cxxGenDir = cxxDir+"Var"
+        if args.model == "Model/MaxIV/commonBeam":
+            cxxGenDir = "Model/MaxIV/commonGenerator"
+        else:
+            cxxGenDir = os.path.join('',*cxxDir.split('/')[:-1], "commonGenerator")
+            if not checkPaths([cxxGenDir],()):
+                cxxGenDir = cxxDir+"Var"
 
         hGenDir = cxxGenDir + 'Inc'
         prefix = args.className