org.eclipse.papyrus.bpmn/custom-src/org/eclipse/papyrus/bpmn/BPMNProfile/impl/LaneCustom.java

/*****************************************************************************
 * Copyright (c) 2017 CEA LIST and others.
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *   CEA - Initial API and implementation
 *
 *****************************************************************************/
package org.eclipse.papyrus.bpmn.BPMNProfile.impl;

import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;

import org.apache.log4j.Logger;
import org.eclipse.papyrus.bpmn.BPMNProfile.BPMNProfileFactory;
import org.eclipse.papyrus.bpmn.BPMNProfile.BPMNProfilePackage;
import org.eclipse.papyrus.bpmn.BPMNProfile.Lane;
import org.eclipse.papyrus.bpmn.BPMNProfile.LaneSet;
import org.eclipse.uml2.uml.ActivityPartition;
import org.eclipse.uml2.uml.util.UMLUtil;
import org.eclipse.uml2.uml.util.UMLUtil.StereotypeApplicationHelper;

public class LaneCustom {
	private static final Logger log = Logger.getLogger(LaneCustom.class);

	private LaneCustom() {
	}

	public static LaneSet basicGetChildLaneSet(Lane lane) {
		LaneSet laneSet = null;
		ActivityPartition activityPartition = lane.getBase_ActivityPartition();
		if(activityPartition != null) {
			laneSet = UMLUtil.getStereotypeApplication(activityPartition, LaneSet.class);
		}
		
/*		if(activityPartition != null){
			List<ActivityPartition> subPartitions = activityPartition.getSubpartitions();
			if(subPartitions != null && !subPartitions.isEmpty()){
				laneSet = BPMNProfileFactory.eINSTANCE.createLaneSet();
				laneSet.setBase_ActivityPartition(activityPartition);
				List<Lane> lanes = subPartitions.stream().map(p -> UMLUtil.getStereotypeApplication(p, Lane.class)).filter(Objects::nonNull).collect(Collectors.toList());
				laneSet.getLanes().addAll(lanes);
				log.debug(lanes);
			}
		}*/
		return laneSet;
	}


	public static LaneSet basicGetLaneSet(Lane lane) {
		LaneSet laneSet = null;
		ActivityPartition activityPartition = lane.getBase_ActivityPartition();
		if(activityPartition != null) {
			ActivityPartition superPartition = activityPartition.getSuperPartition();
			if(superPartition != null){
				laneSet = UMLUtil.getStereotypeApplication(superPartition, LaneSet.class);
			}
		}
		return laneSet;
	}
}