cleanup

[taxeditor.git] / eu.etaxonomy.taxeditor.store / src / main / java / eu / etaxonomy / taxeditor / ui / section / feature / FeatureStateWizard.java

diff --git a/eu.etaxonomy.taxeditor.store/src/main/java/eu/etaxonomy/taxeditor/ui/section/feature/FeatureStateWizard.java b/eu.etaxonomy.taxeditor.store/src/main/java/eu/etaxonomy/taxeditor/ui/section/feature/FeatureStateWizard.java
index 047831607920cbf12d44efc79481685fa0da9081..88be253f90434ab53676b61f04e9422928b43c2b 100644 (file)
--- a/eu.etaxonomy.taxeditor.store/src/main/java/eu/etaxonomy/taxeditor/ui/section/feature/FeatureStateWizard.java
+++ b/eu.etaxonomy.taxeditor.store/src/main/java/eu/etaxonomy/taxeditor/ui/section/feature/FeatureStateWizard.java
@@ -10,9 +10,9 @@ package eu.etaxonomy.taxeditor.ui.section.feature;
 
 import org.eclipse.jface.wizard.Wizard;
 
-import eu.etaxonomy.cdm.model.description.Feature;
 import eu.etaxonomy.cdm.model.description.FeatureState;
-import eu.etaxonomy.cdm.model.term.TermNode;
+import eu.etaxonomy.cdm.persistence.dto.FeatureStateDto;
+import eu.etaxonomy.cdm.persistence.dto.TermNodeDto;
 import eu.etaxonomy.taxeditor.ui.element.CdmFormFactory;
 
 /**
@@ -25,20 +25,18 @@ public class FeatureStateWizard extends Wizard {
     private static final String CREATE_FEATURE_STATE = "Create Feature State";
     private FeatureStateWizardPage page;
     private CdmFormFactory formFactory;
-    private String comboLabel;
-    private TermNode<? extends Feature> termNode;
+    private TermNodeDto termNode;
 
-    public FeatureStateWizard(String comboLabel, TermNode<? extends Feature> termNode, CdmFormFactory formFactory) {
+    public FeatureStateWizard(TermNodeDto termNode, CdmFormFactory formFactory) {
         super();
         this.formFactory = formFactory;
         this.termNode = termNode;
-        this.comboLabel = comboLabel;
         setWindowTitle(CREATE_FEATURE_STATE);
     }
 
     @Override
     public void addPages() {
-        page = new FeatureStateWizardPage(CREATE_FEATURE_STATE, comboLabel, termNode, formFactory);
+        page = new FeatureStateWizardPage(CREATE_FEATURE_STATE, termNode, formFactory);
         addPage(page);
     }
 
@@ -50,5 +48,9 @@ public class FeatureStateWizard extends Wizard {
     FeatureState getFeatureState(){
         return page.getFeatureState();
     }
+    
+    FeatureStateDto getFeatureStateDto(){
+        return page.getFeatureStateDto();
+    }
 
 }