change: include_whole_cell -> add_whole_cell

Also calculates cell_growth_rate.

src/wela/dataloader.py

@@ -239,16 +239,20 @@ class dataloader:
             # add missing signals to h5 files
             add_bud_fluorescence(self.h5dirpath / dataname, missing_signals)
 
-    def include_whole_cell(self, fl_channels=["GFP"]):
+    def add_whole_cell(self, fl_channels=[]):
         """Create whole-cell signals from mother and bud compartments."""
-        # cell volume and area
-        for signal in ["volume", "area"]:
-            self.df[f"cell_{signal}"] = np.nansum(
-                [self.df[signal].values, self.df[f"bud_{signal}"].values],
-                axis=0,
-            )
-            # replace zeros created by np.nansum with nan
-            self.df.loc[self.df[signal].isna(), f"cell_{signal}"] = np.nan
+        # cell volume, area, and growth rate
+        for signal in ["volume", "area", "growth_rate"]:
+            if (
+                signal in self.df.columns
+                and f"bud_{signal}" in self.df.columns
+            ):
+                self.df[f"cell_{signal}"] = np.nansum(
+                    [self.df[signal].values, self.df[f"bud_{signal}"].values],
+                    axis=0,
+                )
+                # replace zeros created by np.nansum with nan
+                self.df.loc[self.df[signal].isna(), f"cell_{signal}"] = np.nan
         # fluorescence
         for fl in fl_channels:
             for bg in ["", "_bgsub"]: