change logic to remove column headers

w-ahmad · w-ahmad · commit 1780bb00bb3c · 2025-06-24T18:14:46.000+05:00
diff --git a/src/TableViewHeaderRow.cs b/src/TableViewHeaderRow.cs
@@ -181,7 +181,7 @@ private void RemoveHeaders(IEnumerable<TableViewColumn> columns)
         {
             foreach (var column in columns)
             {
-                var header = _headersStackPanel.Children.OfType<TableViewColumnHeader>().FirstOrDefault(x => x == column.HeaderControl);
+                var header = _headersStackPanel.Children.OfType<TableViewColumnHeader>().FirstOrDefault(x => x.Column == column);
                 if (header is not null)
                 {
                     _headersStackPanel.Children.Remove(header);

Original file line number	Diff line number	Diff line change
`@@ -181,7 +181,7 @@ private void RemoveHeaders(IEnumerable<TableViewColumn> columns)`
`181`	`181`	`{`
`182`	`182`	`foreach (var column in columns)`
`183`	`183`	`{`
`184`		`- var header = _headersStackPanel.Children.OfType<TableViewColumnHeader>().FirstOrDefault(x => x == column.HeaderControl);`
	`184`	`+ var header = _headersStackPanel.Children.OfType<TableViewColumnHeader>().FirstOrDefault(x => x.Column == column);`
`185`	`185`	`if (header is not null)`
`186`	`186`	`{`
`187`	`187`	`_headersStackPanel.Children.Remove(header);`