backport nonref_flags elision

chrchang · Jul 4, 2024 · 2bbee44 · 2bbee44
1 parent adfee42
commit 2bbee44
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/2.0/plink2_data.cc b/2.0/plink2_data.cc
@@ -6388,6 +6388,16 @@ PglErr MakePgenRobust(const uintptr_t* sample_include, const uint32_t* new_sampl
           nonref_flags_write = nullptr;
           nonref_flags_storage = 1;
         }
+      } else {
+        if (nonref_flags_write[0] & 1) {
+          if (AllBitsAreOne(nonref_flags_write, raw_variant_ct)) {
+            nonref_flags_write = nullptr;
+            nonref_flags_storage = 2;
+          }
+        } else if (AllWordsAreZero(nonref_flags_write, BitCtToWordCt(raw_variant_ct))) {
+          nonref_flags_write = nullptr;
+          nonref_flags_storage = 1;
+        }
       }
       snprintf(outname_end, kMaxOutfnameExtBlen, ".pgen");
       uintptr_t spgw_alloc_cacheline_ct;
@@ -7517,6 +7527,16 @@ PglErr MakePlink2NoVsort(const uintptr_t* sample_include, const PedigreeIdInfo*
           nonref_flags_write = nullptr;
           nonref_flags_storage = 1;
         }
+      } else {
+        if (nonref_flags_write[0] & 1) {
+          if (AllBitsAreOne(nonref_flags_write, raw_variant_ct)) {
+            nonref_flags_write = nullptr;
+            nonref_flags_storage = 2;
+          }
+        } else if (AllWordsAreZero(nonref_flags_write, BitCtToWordCt(raw_variant_ct))) {
+          nonref_flags_write = nullptr;
+          nonref_flags_storage = 1;
+        }
       }
       ctx.pwcs = &(mpgwp->pwcs[0]);
       ctx.thread_write_genovecs = nullptr;