fix compile error

bigo-sg · Jun 20, 2022 · ee8f4ac · ee8f4ac
1 parent d3bff54
commit ee8f4ac
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/src/Core/BlockInfo.cpp b/src/Core/BlockInfo.cpp
@@ -65,6 +65,13 @@ void BlockMissingValues::setBit(size_t column_idx, size_t row_idx)
     mask[row_idx] = true;
 }
 
+void BlockMissingValues::setBits(size_t column_idx, size_t rows)
+{
+    RowsBitMask & mask = rows_mask_by_column_id[column_idx];
+    mask.resize(rows);
+    std::fill(mask.begin(), mask.end(), true);
+}
+
 const BlockMissingValues::RowsBitMask & BlockMissingValues::getDefaultsBitmask(size_t column_idx) const
 {
     static RowsBitMask none;

diff --git a/src/Core/BlockInfo.h b/src/Core/BlockInfo.h
@@ -57,6 +57,8 @@ class BlockMissingValues
     /// Check that we have to replace default value at least in one of columns
     bool hasDefaultBits(size_t column_idx) const;
     void setBit(size_t column_idx, size_t row_idx);
+    /// Set bits for all rows in a single column.
+    void setBits(size_t column_idx, size_t rows);
     bool empty() const { return rows_mask_by_column_id.empty(); }
     size_t size() const { return rows_mask_by_column_id.size(); }
     void clear() { rows_mask_by_column_id.clear(); }