diff --git a/toolchains/arduino/org.eclipse.cdt.arduino-feature/feature.xml b/toolchains/arduino/org.eclipse.cdt.arduino-feature/feature.xml index a8abff53292..f7e3e4a7817 100644 --- a/toolchains/arduino/org.eclipse.cdt.arduino-feature/feature.xml +++ b/toolchains/arduino/org.eclipse.cdt.arduino-feature/feature.xml @@ -26,6 +26,7 @@ + checkedLibs; private class ContentProvider implements ITreeContentProvider { private LibraryIndex index; @@ -188,12 +190,62 @@ public class LibrariesPropertyPage extends PropertyPage { @Override protected TreeViewer doCreateTreeViewer(Composite parent, int style) { - return new ContainerCheckedTreeViewer(parent, style); + CheckboxTreeViewer viewer = new CheckboxTreeViewer(parent, style); + viewer.setCheckStateProvider(new ICheckStateProvider() { + @Override + public boolean isGrayed(Object element) { + if (element instanceof String) { + for (ArduinoLibrary lib : checkedLibs) { + if (element.equals(lib.getCategory())) { + return true; + } + } + } + return false; + } + + @Override + public boolean isChecked(Object element) { + if (element instanceof ArduinoLibrary) { + return checkedLibs.contains(element); + } else if (element instanceof String) { + for (ArduinoLibrary lib : checkedLibs) { + if (element.equals(lib.getCategory())) { + return true; + } + } + } + + return false; + } + }); + viewer.addCheckStateListener(new ICheckStateListener() { + @Override + public void checkStateChanged(CheckStateChangedEvent event) { + Object element = event.getElement(); + if (element instanceof ArduinoLibrary) { + if (event.getChecked()) { + checkedLibs.add((ArduinoLibrary) element); + } else { + checkedLibs.remove(element); + } + } else if (element instanceof String) { + if (!event.getChecked()) { + for (ArduinoLibrary lib : new ArrayList<>(checkedLibs)) { + if (element.equals(lib.getCategory())) { + checkedLibs.remove(lib); + } + } + } + } + } + }); + return viewer; } }; filteredTree.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true)); - ContainerCheckedTreeViewer viewer = (ContainerCheckedTreeViewer) filteredTree.getViewer(); + TreeViewer viewer = filteredTree.getViewer(); Tree tree = viewer.getTree(); tree.setHeaderVisible(true); @@ -208,18 +260,14 @@ public class LibrariesPropertyPage extends PropertyPage { viewer.setLabelProvider(new LabelProvider()); try { - viewer.setInput(manager.getLibraryIndex()); - // Set the check states for currently selected libraries IProject project = getElement().getAdapter(IProject.class); - Collection libraries = manager.getLibraries(project); - for (ArduinoLibrary lib : libraries) { - viewer.setChecked(lib, true); - } + checkedLibs = new HashSet<>(manager.getLibraries(project)); + viewer.setInput(manager.getLibraryIndex()); } catch (CoreException e) { Activator.log(e); } - return comp; + return comp; } private IProject getProject() { @@ -232,17 +280,8 @@ public class LibrariesPropertyPage extends PropertyPage { @Override public boolean performOk() { - List libs = new ArrayList<>(); - for (TreeItem categoryItem : filteredTree.getViewer().getTree().getItems()) { - for (TreeItem libItem : categoryItem.getItems()) { - ArduinoLibrary lib = (ArduinoLibrary) libItem.getData(); - if (libItem.getChecked()) { - libs.add(lib); - } - } - } try { - manager.setLibraries(getProject(), libs); + manager.setLibraries(getProject(), checkedLibs); } catch (CoreException e) { Activator.log(e); }