Update of case_generation, support for multiple batches

Author: ebranlard

data/NREL5MW/Main_Onshore_OF2.fst renamed to data/NREL5MW/Main_Onshore_OF.fst

@@ -32,6 +32,14 @@ def handleRemoveReadonlyWin(func, path, exc_info):
         raise
 
 
+def forceCopyFile (sfile, dfile):
+    # ---- Handling error due to wrong mod
+    if os.path.isfile(dfile):
+        if not os.access(dfile, os.W_OK):
+            os.chmod(dfile, stat.S_IWUSR)
+    #print(sfile, ' > ', dfile)
+    shutil.copy2(sfile, dfile)
+
 def copyTree(src, dst):
     """ 
     Copy a directory to another one, overwritting files if necessary.
@@ -45,14 +53,6 @@ def forceMergeFlatDir(srcDir, dstDir):
             dstFile = os.path.join(dstDir, item)
             forceCopyFile(srcFile, dstFile)
 
-    def forceCopyFile (sfile, dfile):
-        # ---- Handling error due to wrong mod
-        if os.path.isfile(dfile):
-            if not os.access(dfile, os.W_OK):
-                os.chmod(dfile, stat.S_IWUSR)
-        #print(sfile, ' > ', dfile)
-        shutil.copy2(sfile, dfile)
-
     def isAFlatDir(sDir):
         for item in os.listdir(sDir):
             sItem = os.path.join(sDir, item)
@@ -292,35 +292,41 @@ def removeFASTOuputs(workDir):
 # --------------------------------------------------------------------------------}
 # --- Tools for template replacement 
 # --------------------------------------------------------------------------------{
-def paramsSteadyAero(p=dict()):
+def paramsSteadyAero(p=None):
+    p = dict() if p is None else p
     p['AeroFile|AFAeroMod']=1 # remove dynamic effects dynamic
     p['AeroFile|WakeMod']=1 # remove dynamic inflow dynamic
     p['AeroFile|TwrPotent']=0 # remove tower shadow
     return p
 
-def paramsNoGen(p=dict()):
+def paramsNoGen(p=None):
+    p = dict() if p is None else p
     p['EDFile|GenDOF' ]  = 'False'
     return p
 
-def paramsGen(p=dict()):
+def paramsGen(p=None):
+    p = dict() if p is None else p
     p['EDFile|GenDOF' ]  = 'True'
     return p
 
-def paramsNoController(p=dict()):
+def paramsNoController(p=None):
+    p = dict() if p is None else p
     p['ServoFile|PCMode']   = 0;
     p['ServoFile|VSContrl'] = 0;
     p['ServoFile|YCMode']   = 0;
     return p
 
-def paramsControllerDLL(p=dict()):
+def paramsControllerDLL(p=None):
+    p = dict() if p is None else p
     p['ServoFile|PCMode']   = 5;
     p['ServoFile|VSContrl'] = 5;
     p['ServoFile|YCMode']   = 5;
     p['EDFile|GenDOF']      = 'True';
     return p
 
 
-def paramsStiff(p=dict()):
+def paramsStiff(p=None):
+    p = dict() if p is None else p
     p['EDFile|FlapDOF1']  = 'False'
     p['EDFile|FlapDOF2']  = 'False'
     p['EDFile|EdgeDOF' ]  = 'False'
@@ -352,7 +358,7 @@ def iterify(x):
     RPM   = iterify(RPM)
     Pitch = iterify(Pitch)
     # --- If inputs are not flat but different vectors to length through, we flatten them (TODO: meshgrid and ravel?)
-    if not FlatInputs :
+    if not flatInputs :
         WS_flat    = []
         Pitch_flat = []
         RPM_flat   = []
@@ -386,7 +392,8 @@ def iterify(x):
         PARAMS.append(p)
     return PARAMS
 
-def paramsLinearTrim(p=dict()):
+def paramsLinearTrim(p=None):
+    p = dict() if p is None else p
 
     # Set a few DOFs, move this to main file
     p['Linearize']              = True
@@ -435,7 +442,8 @@ def paramsLinearTrim(p=dict()):
 # ---  
 # --------------------------------------------------------------------------------{
 def createStepWind(filename,WSstep=1,WSmin=3,WSmax=25,tstep=100,dt=0.5,tmin=0,tmax=999):
-    f = weio.FASTWndFile()
+    from pyFAST.input_output.fast_wind_file import FASTWndFile
+    f = FASTWndFile()
     Steps= np.arange(WSmin,WSmax+WSstep,WSstep)
     print(Steps)
     nCol = len(f.colNames)

pyFAST/case_generation/case_gen.py

@@ -14,7 +14,7 @@ def CPLambdaExample():
     """
     FAST_EXE  = os.path.join(MyDir, '../../../data/openfast.exe') # Location of a FAST exe (and dll)
     ref_dir   = os.path.join(MyDir, '../../../data/NREL5MW/')     # Folder where the fast input files are located (will be copied)
-    main_file = 'Main_Onshore_OF2.fst'  # Main file in ref_dir, used as a template
+    main_file = 'Main_Onshore.fst'  # Main file in ref_dir, used as a template
 
     # --- Computing CP and CT matrices for range of lambda and pitches
     Lambda = np.linspace(0.1,10,3)

pyFAST/case_generation/examples/Example_CPLambdaPitch.py

@@ -17,7 +17,7 @@ def main():
     # --- Main Parameters
     ref_dir        = os.path.join(MyDir, '../../../data/NREL5MW/')  # Folder where the fast input files are located (will be copied)
     FAST_EXE       = os.path.join(MyDir, '../../../data/openfast.exe') # Location of a FAST exe (and dll)
-    main_file      = 'Main_Onshore_OF2.fst'          # Main file in ref_dir, used as a template
+    main_file      = 'Main_Onshore.fst'          # Main file in ref_dir, used as a template
     work_dir       = '_NREL5MW_ParametricExcel/'     # Output folder (will be created)
     parametricFile = 'ParametricExcel.xlsx'          # Excel file containing set of parameters
 

pyFAST/case_generation/examples/Example_ExcelFile.py

@@ -30,7 +30,7 @@ def ParametricExample():
     # --- Parameters for this script
     FAST_EXE  = os.path.join(MyDir, '../../../data/openfast.exe') # Location of a FAST exe (and dll)
     ref_dir   = os.path.join(MyDir, '../../../data/NREL5MW/')  # Folder where the fast input files are located (will be copied)
-    main_file = 'Main_Onshore_OF2.fst'  # Main file in ref_dir, used as a template
+    main_file = 'Main_Onshore.fst'  # Main file in ref_dir, used as a template
     work_dir  = '_NREL5MW_Parametric/'     # Output folder (will be created)
 
     # --- Defining the parametric study  (list of dictionnaries with keys as FAST parameters)
@@ -50,6 +50,12 @@ def ParametricExample():
             p['EDFile|BldFile(1)|AdjBlMs'] =1.1
             p['EDFile|BldFile(2)|AdjBlMs'] =1.1
             p['EDFile|BldFile(3)|AdjBlMs'] =1.1
+#         # Changing BeamDyn properties
+#         BP_ref = fBD['BeamProperties']
+#         BP     = BP_ref.copy()
+#         BP['K']= BP['K']*i
+#         p['BDBldFile(1)|BldFile|BeamProperties'] =BP
+
 
         p['EDFile|RotSpeed']       = rpm
         p['InflowFile|HWindSpeed'] = wsp

pyFAST/case_generation/examples/Example_Parametric.py

@@ -25,7 +25,7 @@ def PowerCurveParametricExample1():
     # --- Parameters for this script
     FAST_EXE  = os.path.join(MyDir, '../../../data/openfast.exe') # Location of a FAST exe (and dll)
     ref_dir   = os.path.join(MyDir, '../../../data/NREL5MW/')     # Folder where the fast input files are located (will be copied)
-    main_file = 'Main_Onshore_OF2.fst'               # Main file in ref_dir, used as a template
+    main_file = 'Main_Onshore.fst'               # Main file in ref_dir, used as a template
     work_dir  = '_NREL5MW_PowerCurveParametric/'     # Output folder (will be created)
 
     # --- Defining the parametric study  (list of dictionnaries with keys as FAST parameters)
@@ -84,7 +84,7 @@ def PowerCurveParametricExample2():
     # --- Parameters for this script
     FAST_EXE  = os.path.join(MyDir, '../../../data/openfast.exe') # Location of a FAST exe (and dll)
     ref_dir   = os.path.join(MyDir, '../../../data/NREL5MW/')     # Folder where the fast input files are located (will be copied)
-    main_file = 'Main_Onshore_OF2.fst'                # Main file in ref_dir, used as a template
+    main_file = 'Main_Onshore_OF.fst'                # Main file in ref_dir, used as a template
     work_dir  = '_NREL5MW_PowerCurveParametric2/'     # Output folder (will be created)
     out_Ext   = '.outb' # Output extension
 

pyFAST/case_generation/examples/Example_PowerCurve_Parametric.py

@@ -139,20 +139,33 @@ def run_fast(input_file, fastExe=None, wait=True, showOutputs=False, showCommand
     return run_cmd(input_file, fastExe, wait=wait, showOutputs=showOutputs, showCommand=showCommand)
 
 
-def writeBatch(batchfile, fastfiles, fastExe=None):
+def writeBatch(batchfile, fastfiles, fastExe=None, nBatches=1):
     """ Write batch file, everything is written relative to the batch file"""
     if fastExe is None:
         fastExe=FAST_EXE
     fastExe_abs   = os.path.abspath(fastExe)
     batchfile_abs = os.path.abspath(batchfile)
     batchdir      = os.path.dirname(batchfile_abs)
     fastExe_rel   = os.path.relpath(fastExe_abs, batchdir)
-    with open(batchfile,'w') as f:
-        for ff in fastfiles:
-            ff_abs = os.path.abspath(ff)
-            ff_rel = os.path.relpath(ff_abs, batchdir)
-            l = fastExe_rel + ' '+ ff_rel
-            f.write("%s\n" % l)
+    def writeb(batchfile, fastfiles):
+        with open(batchfile,'w') as f:
+            for ff in fastfiles:
+                ff_abs = os.path.abspath(ff)
+                ff_rel = os.path.relpath(ff_abs, batchdir)
+                l = fastExe_rel + ' '+ ff_rel
+                f.write("%s\n" % l)
+    if nBatches==1:
+        writeb(batchfile, fastfiles)
+    else:
+        splits = np.array_split(fastfiles,nBatches)
+        base, ext = os.path.splitext(batchfile)
+        for i in np.arange(nBatches):
+            writeb(base+'_{:d}'.format(i+1) + ext, splits[i])
+
+
+
+
+
 
 def removeFASTOuputs(workDir):
     # Cleaning folder

pyFAST/case_generation/runner.py

@@ -21,7 +21,7 @@ def main():
 
     # --- Define output radial stations
     # Option 1 - Get all these locations automatically (recommended)
-    fstFile = os.path.join(MyDir,'../../../data/NREL5MW/Main_Onshore_OF2.fst') # must correspond to the one used to generate outputs
+    fstFile = os.path.join(MyDir,'../../../data/NREL5MW/Main_Onshore_OF.fst') # must correspond to the one used to generate outputs
     r_AD, r_ED, r_BD, IR_AD, IR_ED, IR_BD, R, r_hub, fst = postpro.FASTRadialOutputs(fstFile, df.columns.values)
 
     # Option 2 - Get ouputs locations for each module

pyFAST/input_output/examples/Example_RadialInterp.py

@@ -22,7 +22,7 @@ def main():
     # --- Step2 : Average data and extrat the radial stations
     # Averaging here is done over 1 period (avgParam=1, avgMethod='periods')
     # To get the output radial stations, a .fst file is needed
-    fstFile = os.path.join(MyDir,'../../../data/NREL5MW/Main_Onshore_OF2.fst')
+    fstFile = os.path.join(MyDir,'../../../data/NREL5MW/Main_Onshore_OF.fst')
     dfRad_ED, dfRad_AD, dfRad_BD = postpro.spanwisePostPro(FST_In=fstFile, avgMethod='periods', avgParam=1, df=df)
 
     # --- Step1&2 at once (when .outb and .fst are next to each other in same folder, with same name)