package de.prob2.ui.states;

import java.util.HashSet;

import com.google.inject.Inject;
import com.google.inject.Singleton;
import com.sun.javafx.collections.ObservableSetWrapper;

import de.prob.model.representation.AbstractElement;
import javafx.collections.ObservableSet;

@Singleton
public class ClassBlacklist {
	private final ObservableSet<Class<? extends AbstractElement>> blacklist;
	private final ObservableSet<Class<? extends AbstractElement>> knownClasses;
	
	@Inject
	private ClassBlacklist() {
		super();
		this.blacklist = new ObservableSetWrapper<>(new HashSet<>());
		this.knownClasses = new ObservableSetWrapper<>(new HashSet<>());
	}
	
	public ObservableSet<Class<? extends AbstractElement>> getBlacklist() {
		return this.blacklist;
	}
	
	public ObservableSet<Class<? extends AbstractElement>> getKnownClasses() {
		return this.knownClasses;
	}
}

@@ -26,35 +26,8 @@
 s;%0A%0A
-import java.util.HashSet;%0A%0A
 impo
@@ -95,65 +95,8 @@
 on;%0A
-import com.sun.javafx.collections.ObservableSetWrapper;%0A%0A
 impo
@@ -144,16 +144,57 @@
 lement;%0A
+import javafx.collections.FXCollections;%0A
 import j
@@ -245,16 +245,22 @@
 %0Apublic 
+final 
 class Cl
@@ -436,16 +436,48 @@
 @Inject%0A
+%09@SuppressWarnings(%22unchecked%22)%0A
 %09private
@@ -530,117 +530,89 @@
 t = 
-new ObservableSetWrapper%3C%3E(new HashSet%3C%3E());%0A%09%09this.knownClasses = new ObservableSetWrapper%3C%3E(new HashSet%3C%3E()
+FXCollections.observableSet();%0A%09%09this.knownClasses = FXCollections.observableSet(
 );%0A%09

Replace use of com.sun ObservableSetWrapper with FXCollections