EDIT

import java.util.Optional;

import eu.etaxonomy.cdm.model.common.IdentifiableEntity;

import eu.etaxonomy.cdm.model.name.SpecimenTypeDesignation;

import eu.etaxonomy.cdm.vaadin.model.registration.SpecimenTypeDesignationWorkingSetDTO;

    /**

     * @param baseEntityReference

     */

    public TypeDesignationWorkingSet getTypeDesignationWorkingSet(TypedEntityReference baseEntityReference) {

        return typeDesignationConverter != null ? typeDesignationConverter.getOrderdTypeDesignationWorkingSets().get(baseEntityReference) : null;

    }

    /**

     * @param baseEntityReference

     */

    public Set<TypeDesignationBase> getTypeDesignationsInWorkingSet(TypedEntityReference baseEntityReference) {

        Set<TypeDesignationBase> typeDesignations = new HashSet<>();

        TypeDesignationWorkingSet workingSet = getTypeDesignationWorkingSet(baseEntityReference);

        for(EntityReference ref :  workingSet.getTypeDesignations()){

            typeDesignations.add(findTypeDesignation(ref));

        }

        return typeDesignations;

    }

    public SpecimenTypeDesignationWorkingSetDTO getSpecimenTypeDesignationWorkingSetDTO(TypedEntityReference baseEntityReference) {

        Set<TypeDesignationBase> typeDesignations = getTypeDesignationsInWorkingSet(baseEntityReference);

        List<SpecimenTypeDesignation> specimenTypeDesignations = new ArrayList<>(typeDesignations.size());

        typeDesignations.forEach(td -> specimenTypeDesignations.add((SpecimenTypeDesignation)td));

        IdentifiableEntity<?> baseEntity = getTypeDesignationWorkingSet(baseEntityReference).getBaseEntity();

        SpecimenTypeDesignationWorkingSetDTO dto = new SpecimenTypeDesignationWorkingSetDTO(reg, baseEntity, specimenTypeDesignations);

        return dto;

    }

    /**

     *

     * @param workingSetId

     * @return the TypeDesignationWorkingSet in this DTO with the matching workingSetId or NULL

     */

    public TypeDesignationWorkingSet getTypeDesignationWorkingSet(int workingSetId) {

        Optional<TypeDesignationWorkingSet> workingSetOptional = getOrderdTypeDesignationWorkingSets().values().stream().filter(workingSet -> workingSet.getWorkingSetId() == workingSetId).findFirst();

        if(workingSetOptional.isPresent()){

            return workingSetOptional.get();

        }

        return null;

    }

    /**

     * @param ref

     * @return

     */

    private TypeDesignationBase findTypeDesignation(EntityReference ref) {

        return typeDesignationConverter != null ? typeDesignationConverter.findTypeDesignation(ref) : null;

    }