lib/write_xlsx/package/table.rb in write_xlsx-1.10.2 vs lib/write_xlsx/package/table.rb in write_xlsx-1.11.0

@@ -49,10 +49,11 @@
         set_the_table_and_autofilter_ranges
         set_the_autofilter_range
 
         add_the_table_columns
         write_the_cell_data_if_supplied
+        store_filter_cell_positions
       end
 
       def set_xml_writer(filename)
         @writer.set_xml_writer(filename)
       end
@@ -150,9 +151,17 @@
             token = data[i][j]
             @worksheet.write(row, col, token, @col_formats[j]) if token
             j += 1
           end
           i += 1
+        end
+      end
+
+      def store_filter_cell_positions
+        if ptrue?(@param[:autofilter])
+          (@col1..@col2).each do |col|
+            @worksheet.filter_cells["#{@row1}:#{col}"] = 1
+          end
         end
       end
 
       def prepare(id)
         @id = id