Tools to manage 2D+t

src/rsml/hirros.py

@@ -13,26 +13,36 @@
 import rsml
 
 
-def walk(dir: str, recursive=True):
-    """
+def walk(dir: str, recursive=True, force=False):
+    """ Traverse a set of directories and return rsml files.
+
+    Notes : rsml files are defined by 61_graph*.rsml
     """
     dir = Path(dir)
-    fns = None
+    fns = []
     for rsml_file in ['61_graph_expertized.rsml', '61_graph.rsml']:
-        fns = dir.glob(rsml_file)
-        if not fns and recursive:
-            fns = dir.glob('*/%s'%rsml_file)
-        if fns:
-            break
+        if fns and force:
+            continue
+        rsmls = dir.glob(rsml_file)
+        if rsmls : 
+            fns.extend(rsmls)
+        if recursive:
+            rsmls = dir.glob('*/%s'%rsml_file)
+            if rsmls : 
+                fns.extend(rsmls)
+
     if not fns:
         print("ERROR: no rsml files found")
 
     final_fns = []
     for fn in fns:
-        if (fn.parent/'80_graph_analysis.xlsx').exists():
-            continue
-        else:
-            final_fns.append(fn)
+        if not force:
+            if (fn.parent/'80_graph_analysis.xlsx').exists():
+                continue
+            elif (fn.parent/'80_graph_expertized_analysis.xlsx').exists():
+                continue
+        final_fns.append(fn)
+
     return final_fns
 
 def read(fn):
@@ -100,8 +110,7 @@ def length_and_number(secondary):
     return total_length.tolist(), total_number.tolist()
 
 
-def write_xls(xls_file, obs, primaries, secondaries):
-
+def dataframes(obs, primaries, secondaries):
     data = [obs]
     index=['time(h)']
     for i, p in enumerate(primaries):
@@ -120,8 +129,6 @@ def write_xls(xls_file, obs, primaries, secondaries):
 
     dfs= []
 
-
-
     for sec in secondaries:
 
         data = [obs]
@@ -144,23 +151,68 @@ def write_xls(xls_file, obs, primaries, secondaries):
         df = pd.DataFrame(data, index = index)
         dfs.append(df)
 
-    #print(dfp)
+    return dfp, dfs
+
+def write_xls(xls_file, dfp, dfs):
+
 
     if xls_file:
         with pd.ExcelWriter(xls_file, engine="xlsxwriter") as writer:  
             dfp.to_excel(writer, 
                         sheet_name='Prim',
                         float_format="%.2f", 
                         header = False)
+            workbook  = writer.book
+            bold_format = workbook.add_format({'bold': True})
+
+            worksheet = writer.sheets['Prim']
+            worksheet.set_row(0, None, bold_format)
+
             for i, df in enumerate(dfs):
                 df.to_excel(writer, 
                             sheet_name='RP%d'%(i+1),
                             float_format="%.2f", 
                             header = False)
+                worksheet = writer.sheets['RP%d'%(i+1)]
+                worksheet.set_row(0, None, bold_format)
 
+def write_xls_all(xls_file, fns, primaries, secondaries):
 
-def run(fn):
-    g = read(fn)
+
+    with pd.ExcelWriter(xls_file, engine="xlsxwriter") as writer:  
+        startrow = 0
+        for i, dfp in enumerate(primaries):
+
+            dfp.to_excel(writer, 
+                        sheet_name='Prim',
+                        float_format="%.2f", 
+                        header = False,
+                        startrow=startrow+1)
+            workbook  = writer.book
+            bold_format = workbook.add_format({'bold': True})
+
+            worksheet = writer.sheets['Prim']
+            worksheet.set_row(startrow+1, None, bold_format)
+            worksheet.write(startrow, 0, fns[i])
+            startrow += len(dfp)+3
+
+        startrow = 0
+        for i, dfs in enumerate(secondaries): 
+            name = fns[i]
+            for j, df in enumerate(dfs):
+                df.to_excel(writer, 
+                            sheet_name='Lateral',
+                            float_format="%.2f", 
+                            header = False,
+                            startrow=startrow+1)
+                worksheet = writer.sheets['Lateral']
+                worksheet.set_row(startrow+1, None, bold_format)
+                worksheet.write(startrow, 0, name)
+                worksheet.write(startrow, 1, 'Plant %d'%(j+1))
+
+                startrow += len(df)+3
+
+def process(g):
     obs = times(g)
 
     plant_ids = g.vertices(scale=1)
@@ -173,42 +225,94 @@ def run(fn):
         s2 = secondary(g, pid, obs)
         secondaries.append(s2)
 
+    df_primary, df_secondaries = dataframes(obs, prims, secondaries)
+    return df_primary, df_secondaries
+
+def run(fn):
+    g = read(fn)
+    dfp, dfs =  process(g)
+
     # Write in xlsx
-    xlsx_file = fn.parent/'80_graph_analysis.xlsx'
-    write_xls(xlsx_file, 
-              obs, primaries=prims, secondaries=secondaries)
+    if 'expertized' in fn:
+        xlsx_file = fn.parent/'80_graph_expertized_analysis.xlsx'
+    else: 
+        xlsx_file = fn.parent/'80_graph_analysis.xlsx'
+    write_xls(xlsx_file, dfp, dfs)
 
     print('WRITE %s'%xlsx_file)
 
+def run_all(fns):
+
+    prims = []
+    secondaries = []
+
+    for fn in fns:
+        g = read(fn)
+        dfp, dfs =  process(g)
+        prims.append(dfp)
+        secondaries.append(dfs)
+
+    write_xls_all('gxe_results.xlsx', fns, prims, secondaries)
+    print('WRITE gxe_results.xlsx')
+
+def merge_and_compute_primaries(dataframes):
+    """Mean of Mean
+    or mean of total Length and total number"""
+    pass
+
+def merge_and_compute_secondaries(dataframes):
+    """Mean of Mean
+    or mean of total Length and total number"""
+    pass
+
 def main():
 
     parser = argparse.ArgumentParser()
     parser.add_argument('-d', default='.', help='directory to process')
     parser.add_argument('-r', default=True, type=bool, help='traverse the directories recursively')
+    parser.add_argument('-f',  help='text file containing the list of directories to process')
 
     args = parser.parse_args()
 
     dir = args.d
     recursive = args.r
-
-    fns = walk(dir=dir, recursive=recursive)
-    if not fns:
-        return
-
-    if len(fns) == 1:
-        fn = fns[0]
-        print('Process file %s'%(fns[0]))
-        run(fn)
-
+    gxe_file = args.f
+
+    fns = []
+    if gxe_file:
+        print('GxE file to process', gxe_file)
+        dirs = []
+        with open(gxe_file, 'r') as gxe:
+            dirs = [d.strip() for d in gxe]
+            dirs = [d for d in dirs if Path(d).exists()]
+
+        for d in dirs: 
+            rsml_files = walk(dir=d, recursive=recursive, force=True)
+            fns.extend(rsml_files)
+
+        print('Process files %s'%(' '.join(fns)))
+        run_all(fns)
     else:
-        for fn in fns:
+        fns = walk(dir=dir, recursive=recursive)
+
+        if not fns:
+            return
+
+        if len(fns) == 1:
+            fn = fns[0]
             print('Process file %s'%(fns[0]))
             run(fn)
 
+        else:
+            for fn in fns:
+                print('Process file %s'%(fns[0]))
+                run(fn)
+
 
 
 if __name__=='__main__':
-    g = read('set_de_5/230403VS004')
+    fns = walk('set_de_5/230403VS004')
+    g = read(fns[0])
     obs = times(g)
 
     plant_ids = g.vertices(scale=1)