ref #8242 Refactor states to use DTOs

[taxeditor.git] / eu.etaxonomy.taxeditor.editor / src / main / java / eu / etaxonomy / taxeditor / editor / descriptiveDataSet / matrix / SpecimenColumnPropertyAccessor.java

diff --git a/eu.etaxonomy.taxeditor.editor/src/main/java/eu/etaxonomy/taxeditor/editor/descriptiveDataSet/matrix/SpecimenColumnPropertyAccessor.java b/eu.etaxonomy.taxeditor.editor/src/main/java/eu/etaxonomy/taxeditor/editor/descriptiveDataSet/matrix/SpecimenColumnPropertyAccessor.java
index 88283f4e20d9e3edf15c74a647a465e54f1c7666..cb3227a0ee7cb51a61978b84220e5a4221707685 100644 (file)
--- a/eu.etaxonomy.taxeditor.editor/src/main/java/eu/etaxonomy/taxeditor/editor/descriptiveDataSet/matrix/SpecimenColumnPropertyAccessor.java
+++ b/eu.etaxonomy.taxeditor.editor/src/main/java/eu/etaxonomy/taxeditor/editor/descriptiveDataSet/matrix/SpecimenColumnPropertyAccessor.java
@@ -8,20 +8,32 @@
 */
 package eu.etaxonomy.taxeditor.editor.descriptiveDataSet.matrix;
 
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.List;
+import java.util.UUID;
+import java.util.stream.Collectors;
+
 import org.eclipse.nebula.widgets.nattable.data.IColumnPropertyAccessor;
 
+import eu.etaxonomy.cdm.api.service.ITermService;
 import eu.etaxonomy.cdm.api.service.dto.RowWrapperDTO;
 import eu.etaxonomy.cdm.api.service.dto.SpecimenRowWrapperDTO;
+import eu.etaxonomy.cdm.api.service.dto.TaxonRowWrapperDTO;
 import eu.etaxonomy.cdm.model.description.Feature;
+import eu.etaxonomy.cdm.model.description.State;
+import eu.etaxonomy.cdm.model.term.DefinedTermBase;
+import eu.etaxonomy.cdm.persistence.dto.TermDto;
+import eu.etaxonomy.taxeditor.store.CdmStore;
 
 /**
- * Property accesor class which maps setting and getting data for
+ * Property accessor class which maps setting and getting data for
  * each row in the character matrix
  * @author pplitzner
  * @since Nov 26, 2017
  *
  */
-public class SpecimenColumnPropertyAccessor implements IColumnPropertyAccessor<RowWrapperDTO>{
+public class SpecimenColumnPropertyAccessor implements IColumnPropertyAccessor<Object>{
 
     private CharacterMatrix matrix;
 
@@ -29,12 +41,8 @@ public class SpecimenColumnPropertyAccessor implements IColumnPropertyAccessor<R
         this.matrix = matrix;
     }
 
-
-    /**
-     * {@inheritDoc}
-     */
     @Override
-    public Object getDataValue(RowWrapperDTO rowObject, int columnIndex) {
+    public Object getDataValue(Object rowObject, int columnIndex) {
         if(rowObject instanceof SpecimenRowWrapperDTO){
             SpecimenRowWrapperDTO rowWrapper = (SpecimenRowWrapperDTO) rowObject;
             switch (columnIndex) {
@@ -55,43 +63,53 @@ public class SpecimenColumnPropertyAccessor implements IColumnPropertyAccessor<R
             default:
                 break;
             }
-            Feature feature = matrix.getIndexToFeatureMap().get(columnIndex);
+            Feature feature = matrix.getIndexToCharacterMap().get(columnIndex);
             return rowWrapper.getDataValueForFeature(feature);
-        } else if (columnIndex == 0) {
+        } else if(rowObject instanceof TaxonRowWrapperDTO){
+            TaxonRowWrapperDTO taxonWrapper = (TaxonRowWrapperDTO)rowObject;
+            if(columnIndex==0){
+                return taxonWrapper.getDescription();
+            }
+            Feature feature = matrix.getIndexToCharacterMap().get(columnIndex);
+            return taxonWrapper.getDataValueForFeature(feature);
+
+        }
+        else if (columnIndex == 0) {
             return rowObject;
         }
         return null;
     }
 
-    /**
-     * {@inheritDoc}
-     */
     @Override
-    public void setDataValue(RowWrapperDTO rowObject, int columnIndex, Object newValue) {
-        RowWrapperDTO rowWrapper = rowObject;
-        Feature feature = matrix.getIndexToFeatureMap().get(columnIndex);
-        rowWrapper.setDataValueForFeature(feature, newValue);
+    public void setDataValue(Object rowObject, int columnIndex, Object newValue) {
+        if(rowObject instanceof RowWrapperDTO){
+            RowWrapperDTO rowWrapper = (RowWrapperDTO)rowObject;
+            Feature feature = matrix.getIndexToCharacterMap().get(columnIndex);
+            if(newValue instanceof Collection && ((Collection) newValue).stream().allMatch(o->o instanceof TermDto)){
+                List<TermDto> dtos = (List<TermDto>)newValue;
+                List<UUID> termUuids = dtos.stream().map(dto->dto.getUuid()).collect(Collectors.toList());
+                List<DefinedTermBase> terms = CdmStore.getService(ITermService.class).load(termUuids, null);
+                List<State> states = new ArrayList<>();
+                for (DefinedTermBase definedTermBase : terms) {
+                    if(definedTermBase instanceof State){
+                        states.add((State) definedTermBase);
+                    }
+                }
+                rowWrapper.setDataValueForFeature(feature, states);
+            }
+        }
     }
 
-    /**
-     * {@inheritDoc}
-     */
     @Override
     public int getColumnCount() {
         return matrix.getPropertyToLabelMap().size();
     }
 
-    /**
-     * {@inheritDoc}
-     */
     @Override
     public String getColumnProperty(int columnIndex) {
         return matrix.getPropertyToLabelMap().get(columnIndex);
     }
 
-    /**
-     * {@inheritDoc}
-     */
     @Override
     public int getColumnIndex(String propertyName){
         return matrix.getPropertyToLabelMap().indexOf(propertyName);