smftools 0.2.4__py3-none-any.whl → 0.3.0__py3-none-any.whl

smftools/cli/preprocess_adata.py

@@ -1,8 +1,15 @@
+from __future__ import annotations
+
 from pathlib import Path
 from typing import Optional, Tuple
 
 import anndata as ad
 
+from smftools.logging_utils import get_logger
+
+logger = get_logger(__name__)
+
+
 def preprocess_adata(
     config_path: str,
 ) -> Tuple[Optional[ad.AnnData], Optional[Path], Optional[ad.AnnData], Optional[Path]]:
@@ -29,8 +36,8 @@ def preprocess_adata(
         Path to preprocessed, duplicate-removed AnnData.
     """
     from ..readwrite import safe_read_h5ad
-    from .load_adata import load_adata
     from .helpers import get_adata_paths
+    from .load_adata import load_adata
 
     # 1) Ensure config is loaded and at least *some* AnnData stage exists
     loaded_adata, loaded_path, cfg = load_adata(config_path)
@@ -60,20 +67,27 @@ def preprocess_adata(
     # Case A: full redo of preprocessing
     # -----------------------------
     if getattr(cfg, "force_redo_preprocessing", False):
-        print("Forcing full redo of preprocessing workflow, starting from latest stage AnnData available.")
+        logger.info(
+            "Forcing full redo of preprocessing workflow, starting from latest stage AnnData available."
+        )
 
         if hmm_exists:
             adata = _load(hmm_path)
+            source_path = hmm_path
         elif spatial_exists:
             adata = _load(spatial_path)
+            source_path = spatial_path
         elif pp_dedup_exists:
             adata = _load(pp_dedup_path)
+            source_path = pp_dedup_path
         elif pp_exists:
             adata = _load(pp_path)
+            source_path = pp_path
         elif raw_exists:
             adata = _load(raw_path)
+            source_path = raw_path
         else:
-            print("Cannot redo preprocessing: no AnnData available at any stage.")
+            logger.error("Cannot redo preprocessing: no AnnData available at any stage.")
             return (None, None, None, None)
 
         pp_adata, pp_adata_path, pp_dedup_adata, pp_dedup_adata_path = preprocess_adata_core(
@@ -81,6 +95,8 @@ def preprocess_adata(
             cfg=cfg,
             pp_adata_path=pp_path,
             pp_dup_rem_adata_path=pp_dedup_path,
+            source_adata_path=source_path,
+            config_path=config_path,
         )
         return pp_adata, pp_adata_path, pp_dedup_adata, pp_dedup_adata_path
 
@@ -88,16 +104,18 @@ def preprocess_adata(
     # Case B: redo duplicate detection only
     # -----------------------------
     if getattr(cfg, "force_redo_flag_duplicate_reads", False):
-        print(
+        logger.info(
             "Forcing redo of duplicate detection workflow, starting from the preprocessed AnnData "
             "if available. Otherwise, will use the raw AnnData."
         )
         if pp_exists:
             adata = _load(pp_path)
+            source_path = pp_path
         elif raw_exists:
             adata = _load(raw_path)
+            source_path = raw_path
         else:
-            print(
+            logger.error(
                 "Cannot redo duplicate detection: no compatible AnnData available "
                 "(need at least raw or preprocessed)."
             )
@@ -108,6 +126,8 @@ def preprocess_adata(
             cfg=cfg,
             pp_adata_path=pp_path,
             pp_dup_rem_adata_path=pp_dedup_path,
+            source_adata_path=source_path,
+            config_path=config_path,
         )
         return pp_adata, pp_adata_path, pp_dedup_adata, pp_dedup_adata_path
 
@@ -117,43 +137,51 @@ def preprocess_adata(
 
     # If HMM exists, preprocessing is considered “done enough”
     if hmm_exists:
-        print(f"Skipping preprocessing. HMM AnnData found: {hmm_path}")
+        logger.debug(f"Skipping preprocessing. HMM AnnData found: {hmm_path}")
         return (None, None, None, None)
 
     # If spatial exists, also skip re-preprocessing by default
     if spatial_exists:
-        print(f"Skipping preprocessing. Spatial AnnData found: {spatial_path}")
+        logger.debug(f"Skipping preprocessing. Spatial AnnData found: {spatial_path}")
         return (None, None, None, None)
 
     # If pp_dedup exists, just return paths (no recomputation)
     if pp_dedup_exists:
-        print(f"Skipping preprocessing. Preprocessed deduplicated AnnData found: {pp_dedup_path}")
+        logger.debug(
+            f"Skipping preprocessing. Preprocessed deduplicated AnnData found: {pp_dedup_path}"
+        )
         return (None, pp_path, None, pp_dedup_path)
 
     # If pp exists but pp_dedup does not, load pp and run core
     if pp_exists:
-        print(f"Preprocessed AnnData found: {pp_path}")
+        logger.debug(f"Preprocessed AnnData found: {pp_path}")
         adata = _load(pp_path)
+        source_path = pp_path
         pp_adata, pp_adata_path, pp_dedup_adata, pp_dedup_adata_path = preprocess_adata_core(
             adata=adata,
             cfg=cfg,
             pp_adata_path=pp_path,
             pp_dup_rem_adata_path=pp_dedup_path,
+            source_adata_path=source_path,
+            config_path=config_path,
         )
         return pp_adata, pp_adata_path, pp_dedup_adata, pp_dedup_adata_path
 
     # Otherwise, fall back to raw (if available)
     if raw_exists:
         adata = _load(raw_path)
+        source_path = raw_path
         pp_adata, pp_adata_path, pp_dedup_adata, pp_dedup_adata_path = preprocess_adata_core(
             adata=adata,
             cfg=cfg,
             pp_adata_path=pp_path,
             pp_dup_rem_adata_path=pp_dedup_path,
+            source_adata_path=source_path,
+            config_path=config_path,
         )
         return pp_adata, pp_adata_path, pp_dedup_adata, pp_dedup_adata_path
 
-    print("No AnnData available at any stage for preprocessing.")
+    logger.error("No AnnData available at any stage for preprocessing.")
     return (None, None, None, None)
 
 
@@ -162,6 +190,8 @@ def preprocess_adata_core(
     cfg,
     pp_adata_path: Path,
     pp_dup_rem_adata_path: Path,
+    source_adata_path: Optional[Path] = None,
+    config_path: Optional[str] = None,
 ) -> Tuple[ad.AnnData, Path, ad.AnnData, Path]:
     """
     Core preprocessing pipeline.
@@ -190,31 +220,32 @@ def preprocess_adata_core(
     """
     from pathlib import Path
 
-    import numpy as np
-
-    from .helpers import write_gz_h5ad
-    from ..readwrite import make_dirs
+    from ..metadata import record_smftools_metadata
+    from ..plotting import plot_read_qc_histograms
     from ..preprocessing import (
-        load_sample_sheet,
-        filter_reads_on_length_quality_mapping,
-        clean_NaN,
-        calculate_coverage,
         append_base_context,
         append_binary_layer_by_base_context,
+        binarize_adata,
+        binarize_on_Youden,
+        calculate_complexity_II,
+        calculate_coverage,
+        calculate_position_Youden,
         calculate_read_modification_stats,
+        clean_NaN,
+        filter_reads_on_length_quality_mapping,
         filter_reads_on_modification_thresholds,
         flag_duplicate_reads,
-        calculate_complexity_II,
-        calculate_position_Youden,
-        binarize_on_Youden,
-        binarize_adata,
+        load_sample_sheet,
     )
-    from ..plotting import plot_read_qc_histograms
+    from ..readwrite import make_dirs
+    from .helpers import write_gz_h5ad
 
     ################################### 1) Load existing  ###################################
     # General config variable init - Necessary user passed inputs
-    smf_modality = cfg.smf_modality # needed for specifying if the data is conversion SMF or direct methylation detection SMF. Or deaminase smf Necessary.
-    output_directory = Path(cfg.output_directory)  # Path to the output directory to make for the analysis. Necessary.
+    smf_modality = cfg.smf_modality  # needed for specifying if the data is conversion SMF or direct methylation detection SMF. Or deaminase smf Necessary.
+    output_directory = Path(
+        cfg.output_directory
+    )  # Path to the output directory to make for the analysis. Necessary.
     make_dirs([output_directory])
 
     ######### Begin Preprocessing #########
@@ -222,172 +253,238 @@ def preprocess_adata_core(
 
     ## Load sample sheet metadata based on barcode mapping ##
     if getattr(cfg, "sample_sheet_path", None):
-        load_sample_sheet(adata, 
-                          cfg.sample_sheet_path, 
-                          mapping_key_column=cfg.sample_sheet_mapping_column, 
-                          as_category=True,
-                          force_reload=cfg.force_reload_sample_sheet)
+        load_sample_sheet(
+            adata,
+            cfg.sample_sheet_path,
+            mapping_key_column=cfg.sample_sheet_mapping_column,
+            as_category=True,
+            force_reload=cfg.force_reload_sample_sheet,
+        )
     else:
         pass
-    
+
     # Adding read length, read quality, reference length, mapped_length, and mapping quality metadata to adata object.
     pp_length_qc_dir = pp_dir / "01_Read_length_and_quality_QC_metrics"
 
     if pp_length_qc_dir.is_dir() and not cfg.force_redo_preprocessing:
-        print( f'{pp_length_qc_dir} already exists. Skipping read level QC plotting.')
+        logger.debug(f"{pp_length_qc_dir} already exists. Skipping read level QC plotting.")
     else:
         make_dirs([pp_dir, pp_length_qc_dir])
-        plot_read_qc_histograms(adata,
-                                pp_length_qc_dir, 
-                                cfg.obs_to_plot_pp_qc, 
-                                sample_key=cfg.sample_name_col_for_plotting, 
-                                rows_per_fig=cfg.rows_per_qc_histogram_grid)
+        plot_read_qc_histograms(
+            adata,
+            pp_length_qc_dir,
+            cfg.obs_to_plot_pp_qc,
+            sample_key=cfg.sample_name_col_for_plotting,
+            rows_per_fig=cfg.rows_per_qc_histogram_grid,
+        )
 
     # Filter on read length, read quality, reference length, mapped_length, and mapping quality metadata.
     print(adata.shape)
-    adata = filter_reads_on_length_quality_mapping(adata, 
-                                                         filter_on_coordinates=cfg.read_coord_filter,
-                                                         read_length=cfg.read_len_filter_thresholds,
-                                                         length_ratio=cfg.read_len_to_ref_ratio_filter_thresholds, 
-                                                         read_quality=cfg.read_quality_filter_thresholds,
-                                                         mapping_quality=cfg.read_mapping_quality_filter_thresholds,
-                                                         bypass=None,
-                                                         force_redo=None)
+    adata = filter_reads_on_length_quality_mapping(
+        adata,
+        filter_on_coordinates=cfg.read_coord_filter,
+        read_length=cfg.read_len_filter_thresholds,
+        length_ratio=cfg.read_len_to_ref_ratio_filter_thresholds,
+        read_quality=cfg.read_quality_filter_thresholds,
+        mapping_quality=cfg.read_mapping_quality_filter_thresholds,
+        bypass=None,
+        force_redo=None,
+    )
     print(adata.shape)
 
     pp_length_qc_dir = pp_dir / "02_Read_length_and_quality_QC_metrics_post_filtering"
 
     if pp_length_qc_dir.is_dir() and not cfg.force_redo_preprocessing:
-        print( f'{pp_length_qc_dir} already exists. Skipping read level QC plotting.')
+        logger.debug(f"{pp_length_qc_dir} already exists. Skipping read level QC plotting.")
     else:
         make_dirs([pp_dir, pp_length_qc_dir])
-        plot_read_qc_histograms(adata,
-                                pp_length_qc_dir, 
-                                cfg.obs_to_plot_pp_qc, 
-                                sample_key=cfg.sample_name_col_for_plotting, 
-                                rows_per_fig=cfg.rows_per_qc_histogram_grid)
-        
+        plot_read_qc_histograms(
+            adata,
+            pp_length_qc_dir,
+            cfg.obs_to_plot_pp_qc,
+            sample_key=cfg.sample_name_col_for_plotting,
+            rows_per_fig=cfg.rows_per_qc_histogram_grid,
+        )
+
     ############## Binarize direct modcall data and store in new layer. Clean nans and store as new layers with various nan replacement strategies ##########
-    if smf_modality == 'direct':
+    if smf_modality == "direct":
         native = True
         if cfg.fit_position_methylation_thresholds:
             pp_Youden_dir = pp_dir / "02B_Position_wide_Youden_threshold_performance"
             make_dirs([pp_Youden_dir])
             # Calculate positional methylation thresholds for mod calls
-            calculate_position_Youden(adata, 
-                                    positive_control_sample=cfg.positive_control_sample_methylation_fitting, 
-                                    negative_control_sample=cfg.negative_control_sample_methylation_fitting, 
-                                    J_threshold=cfg.fit_j_threshold, 
-                                    ref_column=cfg.reference_column, 
-                                    sample_column=cfg.sample_column,
-                                    infer_on_percentile=cfg.infer_on_percentile_sample_methylation_fitting, 
-                                    inference_variable=cfg.inference_variable_sample_methylation_fitting, 
-                                    save=True, 
-                                    output_directory=pp_Youden_dir
-                                    )
+            calculate_position_Youden(
+                adata,
+                positive_control_sample=cfg.positive_control_sample_methylation_fitting,
+                negative_control_sample=cfg.negative_control_sample_methylation_fitting,
+                J_threshold=cfg.fit_j_threshold,
+                ref_column=cfg.reference_column,
+                sample_column=cfg.sample_column,
+                infer_on_percentile=cfg.infer_on_percentile_sample_methylation_fitting,
+                inference_variable=cfg.inference_variable_sample_methylation_fitting,
+                save=True,
+                output_directory=pp_Youden_dir,
+            )
             # binarize the modcalls based on the determined thresholds
-            binarize_on_Youden(adata, 
-                            ref_column=cfg.reference_column,
-                            output_layer_name=cfg.output_binary_layer_name
-                            )
+            binarize_on_Youden(
+                adata,
+                ref_column=cfg.reference_column,
+                output_layer_name=cfg.output_binary_layer_name,
+            )
         else:
-            binarize_adata(adata, 
-                           source="X", 
-                           target_layer=cfg.output_binary_layer_name, 
-                           threshold=cfg.binarize_on_fixed_methlyation_threshold)
-
-        clean_NaN(adata, 
-                  layer=cfg.output_binary_layer_name,
-                  bypass=cfg.bypass_clean_nan, 
-                  force_redo=cfg.force_redo_clean_nan
-                  )
+            binarize_adata(
+                adata,
+                source="X",
+                target_layer=cfg.output_binary_layer_name,
+                threshold=cfg.binarize_on_fixed_methlyation_threshold,
+            )
+
+        clean_NaN(
+            adata,
+            layer=cfg.output_binary_layer_name,
+            bypass=cfg.bypass_clean_nan,
+            force_redo=cfg.force_redo_clean_nan,
+        )
     else:
         native = False
-        clean_NaN(adata, 
-                  bypass=cfg.bypass_clean_nan, 
-                  force_redo=cfg.force_redo_clean_nan
-                  )
-        
+        clean_NaN(adata, bypass=cfg.bypass_clean_nan, force_redo=cfg.force_redo_clean_nan)
+
     ############### Calculate positional coverage by reference set in dataset ###############
-    calculate_coverage(adata, 
-                       ref_column=cfg.reference_column, 
-                       position_nan_threshold=cfg.position_max_nan_threshold)
+    calculate_coverage(
+        adata,
+        ref_column=cfg.reference_column,
+        position_nan_threshold=cfg.position_max_nan_threshold,
+        smf_modality=smf_modality,
+        target_layer=cfg.output_binary_layer_name,
+    )
 
     ############### Add base context to each position for each Reference_strand and calculate read level methylation/deamination stats ###############
     # Additionally, store base_context level binary modification arrays in adata.obsm
-    append_base_context(adata, 
-                        ref_column=cfg.reference_column, 
-                        use_consensus=False, 
-                        native=native, 
-                        mod_target_bases=cfg.mod_target_bases,
-                        bypass=cfg.bypass_append_base_context,
-                        force_redo=cfg.force_redo_append_base_context)
-    
-    adata = append_binary_layer_by_base_context(adata, 
-                                                cfg.reference_column, 
-                                                smf_modality,
-                                                bypass=cfg.bypass_append_binary_layer_by_base_context,
-                                                force_redo=cfg.force_redo_append_binary_layer_by_base_context)
-
-    ############### Calculate read methylation/deamination statistics for specific base contexts defined above ###############
-    calculate_read_modification_stats(adata, 
-                                      cfg.reference_column, 
-                                      cfg.sample_column,
-                                      cfg.mod_target_bases,
-                                      bypass=cfg.bypass_calculate_read_modification_stats,
-                                      force_redo=cfg.force_redo_calculate_read_modification_stats)    
-    
+    append_base_context(
+        adata,
+        ref_column=cfg.reference_column,
+        use_consensus=False,
+        native=native,
+        mod_target_bases=cfg.mod_target_bases,
+        bypass=cfg.bypass_append_base_context,
+        force_redo=cfg.force_redo_append_base_context,
+    )
+
+    ############### Calculate read methylation/deamination statistics for specific base contexts defined by append_base_context ###############
+    calculate_read_modification_stats(
+        adata,
+        cfg.reference_column,
+        cfg.sample_column,
+        cfg.mod_target_bases,
+        bypass=cfg.bypass_calculate_read_modification_stats,
+        force_redo=cfg.force_redo_calculate_read_modification_stats,
+    )
+
     ### Make a dir for outputting sample level read modification metrics before filtering ###
     pp_meth_qc_dir = pp_dir / "03_read_modification_QC_metrics"
 
     if pp_meth_qc_dir.is_dir() and not cfg.force_redo_preprocessing:
-        print(f'{pp_meth_qc_dir} already exists. Skipping read level methylation QC plotting.')
+        logger.debug(
+            f"{pp_meth_qc_dir} already exists. Skipping read level methylation QC plotting."
+        )
     else:
         make_dirs([pp_dir, pp_meth_qc_dir])
-        obs_to_plot = ['Raw_modification_signal']
-        if any(base in cfg.mod_target_bases for base in ['GpC', 'CpG', 'C']):
-            obs_to_plot += ['Fraction_GpC_site_modified', 'Fraction_CpG_site_modified', 'Fraction_other_C_site_modified', 'Fraction_C_site_modified']
-        if 'A' in cfg.mod_target_bases:
-            obs_to_plot += ['Fraction_A_site_modified']
-        plot_read_qc_histograms(adata, 
-                                pp_meth_qc_dir, obs_to_plot, 
-                                sample_key=cfg.sample_name_col_for_plotting, 
-                                rows_per_fig=cfg.rows_per_qc_histogram_grid)
+        obs_to_plot = ["Raw_modification_signal"]
+        if any(base in cfg.mod_target_bases for base in ["GpC", "CpG", "C"]):
+            obs_to_plot += [
+                "Fraction_GpC_site_modified",
+                "Fraction_CpG_site_modified",
+                "Fraction_other_C_site_modified",
+                "Fraction_C_site_modified",
+            ]
+        if "A" in cfg.mod_target_bases:
+            obs_to_plot += ["Fraction_A_site_modified"]
+        plot_read_qc_histograms(
+            adata,
+            pp_meth_qc_dir,
+            obs_to_plot,
+            sample_key=cfg.sample_name_col_for_plotting,
+            rows_per_fig=cfg.rows_per_qc_histogram_grid,
+        )
 
     ##### Optionally filter reads on modification metrics
-    adata = filter_reads_on_modification_thresholds(adata, 
-                                                          smf_modality=smf_modality,
-                                                          mod_target_bases=cfg.mod_target_bases,
-                                                          gpc_thresholds=cfg.read_mod_filtering_gpc_thresholds, 
-                                                          cpg_thresholds=cfg.read_mod_filtering_cpg_thresholds,
-                                                          any_c_thresholds=cfg.read_mod_filtering_c_thresholds,
-                                                          a_thresholds=cfg.read_mod_filtering_a_thresholds,
-                                                          use_other_c_as_background=cfg.read_mod_filtering_use_other_c_as_background,
-                                                          min_valid_fraction_positions_in_read_vs_ref=cfg.min_valid_fraction_positions_in_read_vs_ref,
-                                                          bypass=cfg.bypass_filter_reads_on_modification_thresholds,
-                                                          force_redo=cfg.force_redo_filter_reads_on_modification_thresholds)
-    
+    adata = filter_reads_on_modification_thresholds(
+        adata,
+        smf_modality=smf_modality,
+        mod_target_bases=cfg.mod_target_bases,
+        gpc_thresholds=cfg.read_mod_filtering_gpc_thresholds,
+        cpg_thresholds=cfg.read_mod_filtering_cpg_thresholds,
+        any_c_thresholds=cfg.read_mod_filtering_c_thresholds,
+        a_thresholds=cfg.read_mod_filtering_a_thresholds,
+        use_other_c_as_background=cfg.read_mod_filtering_use_other_c_as_background,
+        min_valid_fraction_positions_in_read_vs_ref=cfg.min_valid_fraction_positions_in_read_vs_ref,
+        bypass=cfg.bypass_filter_reads_on_modification_thresholds,
+        force_redo=cfg.force_redo_filter_reads_on_modification_thresholds,
+    )
+
     pp_meth_qc_dir = pp_dir / "04_read_modification_QC_metrics_post_filtering"
-    
+
     if pp_meth_qc_dir.is_dir() and not cfg.force_redo_preprocessing:
-        print(f'{pp_meth_qc_dir} already exists. Skipping read level methylation QC plotting.')
+        logger.debug(
+            f"{pp_meth_qc_dir} already exists. Skipping read level methylation QC plotting."
+        )
     else:
         make_dirs([pp_dir, pp_meth_qc_dir])
-        obs_to_plot = ['Raw_modification_signal']
-        if any(base in cfg.mod_target_bases for base in ['GpC', 'CpG', 'C']):
-            obs_to_plot += ['Fraction_GpC_site_modified', 'Fraction_CpG_site_modified', 'Fraction_other_C_site_modified', 'Fraction_C_site_modified']
-        if 'A' in cfg.mod_target_bases:
-            obs_to_plot += ['Fraction_A_site_modified']
-        plot_read_qc_histograms(adata, 
-                                pp_meth_qc_dir, obs_to_plot, 
-                                sample_key=cfg.sample_name_col_for_plotting, 
-                                rows_per_fig=cfg.rows_per_qc_histogram_grid)
+        obs_to_plot = ["Raw_modification_signal"]
+        if any(base in cfg.mod_target_bases for base in ["GpC", "CpG", "C"]):
+            obs_to_plot += [
+                "Fraction_GpC_site_modified",
+                "Fraction_CpG_site_modified",
+                "Fraction_other_C_site_modified",
+                "Fraction_C_site_modified",
+            ]
+        if "A" in cfg.mod_target_bases:
+            obs_to_plot += ["Fraction_A_site_modified"]
+        plot_read_qc_histograms(
+            adata,
+            pp_meth_qc_dir,
+            obs_to_plot,
+            sample_key=cfg.sample_name_col_for_plotting,
+            rows_per_fig=cfg.rows_per_qc_histogram_grid,
+        )
+
+    ############### Calculate final positional coverage by reference set in dataset after filtering reads ###############
+    calculate_coverage(
+        adata,
+        ref_column=cfg.reference_column,
+        position_nan_threshold=cfg.position_max_nan_threshold,
+        smf_modality=smf_modality,
+        target_layer=cfg.output_binary_layer_name,
+        force_redo=True,
+    )
+
+    ############### Add base context to each position for each Reference_strand and calculate read level methylation/deamination stats after filtering reads ###############
+    # Additionally, store base_context level binary modification arrays in adata.obsm
+    append_base_context(
+        adata,
+        ref_column=cfg.reference_column,
+        use_consensus=False,
+        native=native,
+        mod_target_bases=cfg.mod_target_bases,
+        bypass=cfg.bypass_append_base_context,
+        force_redo=True,
+    )
+
+    # Add site type binary modification layers for valid coverage sites
+    adata = append_binary_layer_by_base_context(
+        adata,
+        cfg.reference_column,
+        smf_modality,
+        bypass=cfg.bypass_append_binary_layer_by_base_context,
+        force_redo=cfg.force_redo_append_binary_layer_by_base_context,
+        from_valid_sites_only=True,
+    )
 
     ############### Duplicate detection for conversion/deamination SMF ###############
-    if smf_modality != 'direct':
+    if smf_modality != "direct":
         references = adata.obs[cfg.reference_column].cat.categories
 
-        var_filters_sets =[]
+        var_filters_sets = []
         for ref in references:
             for site_type in cfg.duplicate_detection_site_types:
                 var_filters_sets += [[f"{ref}_{site_type}_site", f"position_in_{ref}"]]
@@ -397,27 +494,30 @@ def preprocess_adata_core(
         make_dirs([pp_dup_qc_dir])
 
         # Flag duplicate reads and plot duplicate detection QC
-        adata_unique, adata = flag_duplicate_reads(adata, 
-                                                    var_filters_sets, 
-                                                    distance_threshold=cfg.duplicate_detection_distance_threshold, 
-                                                    obs_reference_col=cfg.reference_column, 
-                                                    sample_col=cfg.sample_name_col_for_plotting,
-                                                    output_directory=pp_dup_qc_dir,
-                                                    metric_keys=cfg.hamming_vs_metric_keys,
-                                                    keep_best_metric=cfg.duplicate_detection_keep_best_metric,
-                                                    bypass=cfg.bypass_flag_duplicate_reads,
-                                                    force_redo=cfg.force_redo_flag_duplicate_reads,
-                                                    window_size=cfg.duplicate_detection_window_size_for_hamming_neighbors,
-                                                    min_overlap_positions=cfg.duplicate_detection_min_overlapping_positions,
-                                                    do_pca=cfg.duplicate_detection_do_pca,
-                                                    pca_n_components=50,
-                                                    pca_center=True,
-                                                    do_hierarchical=cfg.duplicate_detection_do_hierarchical,
-                                                    hierarchical_linkage=cfg.duplicate_detection_hierarchical_linkage,
-                                                    hierarchical_metric="euclidean",
-                                                    hierarchical_window=cfg.duplicate_detection_window_size_for_hamming_neighbors
-                                                    )
-        
+        adata_unique, adata = flag_duplicate_reads(
+            adata,
+            var_filters_sets,
+            distance_threshold=cfg.duplicate_detection_distance_threshold,
+            obs_reference_col=cfg.reference_column,
+            sample_col=cfg.sample_name_col_for_plotting,
+            output_directory=pp_dup_qc_dir,
+            metric_keys=cfg.hamming_vs_metric_keys,
+            keep_best_metric=cfg.duplicate_detection_keep_best_metric,
+            bypass=cfg.bypass_flag_duplicate_reads,
+            force_redo=cfg.force_redo_flag_duplicate_reads,
+            window_size=cfg.duplicate_detection_window_size_for_hamming_neighbors,
+            min_overlap_positions=cfg.duplicate_detection_min_overlapping_positions,
+            do_pca=cfg.duplicate_detection_do_pca,
+            pca_n_components=50,
+            pca_center=True,
+            do_hierarchical=cfg.duplicate_detection_do_hierarchical,
+            hierarchical_linkage=cfg.duplicate_detection_hierarchical_linkage,
+            hierarchical_metric="euclidean",
+            hierarchical_window=cfg.duplicate_detection_window_size_for_hamming_neighbors,
+            demux_types=("double", "already"),
+            demux_col="demux_type",
+        )
+
         # Use the flagged duplicate read groups and perform complexity analysis
         complexity_outs = pp_dup_qc_dir / "sample_complexity_analyses"
         make_dirs([complexity_outs])
@@ -426,15 +526,15 @@ def preprocess_adata_core(
             output_directory=complexity_outs,
             sample_col=cfg.sample_name_col_for_plotting,
             ref_col=cfg.reference_column,
-            cluster_col='sequence__merged_cluster_id',
+            cluster_col="sequence__merged_cluster_id",
             plot=True,
-            save_plot=True,   # set False to display instead
+            save_plot=True,  # set False to display instead
             n_boot=30,
             n_depths=12,
             random_state=42,
             csv_summary=True,
             bypass=cfg.bypass_complexity_analysis,
-            force_redo=cfg.force_redo_complexity_analysis
+            force_redo=cfg.force_redo_complexity_analysis,
         )
 
     else:
@@ -443,13 +543,29 @@ def preprocess_adata_core(
 
     ############################################### Save preprocessed adata with duplicate detection ###############################################
     if not pp_adata_path.exists() or cfg.force_redo_preprocessing:
-        print('Saving preprocessed adata.')
+        logger.info("Saving preprocessed adata.")
+        record_smftools_metadata(
+            adata,
+            step_name="preprocess",
+            cfg=cfg,
+            config_path=config_path,
+            input_paths=[source_adata_path] if source_adata_path else None,
+            output_path=pp_adata_path,
+        )
         write_gz_h5ad(adata, pp_adata_path)
 
     if not pp_dup_rem_adata_path.exists() or cfg.force_redo_preprocessing:
-        print('Saving preprocessed adata with duplicates removed.')
-        write_gz_h5ad(adata_unique, pp_dup_rem_adata_path) 
+        logger.info("Saving preprocessed adata with duplicates removed.")
+        record_smftools_metadata(
+            adata_unique,
+            step_name="preprocess",
+            cfg=cfg,
+            config_path=config_path,
+            input_paths=[pp_adata_path],
+            output_path=pp_dup_rem_adata_path,
+        )
+        write_gz_h5ad(adata_unique, pp_dup_rem_adata_path)
 
     ########################################################################################################################
 
-    return (adata, pp_adata_path, adata_unique, pp_dup_rem_adata_path)
+    return (adata, pp_adata_path, adata_unique, pp_dup_rem_adata_path)