Examples with BuildableTSet edu.iu.dsc.tws.tset.sets.BuildableTSet used on opensource projects

Example 1 with BuildableTSet

use of edu.iu.dsc.tws.tset.sets.BuildableTSet in project beam by apache.

the class BeamBatchWorker method setupTSets.

/**
 * Extract the sideInput TSets and the Leaves from the graph.
 *
 * @param env BatchTSetEnvironment which contains the tSet Graph
 * @param sideInputIds the id's of the side Inputs
 * @param leaveIds the id's of the leaf
 */
private void setupTSets(BatchTSetEnvironment env, Map<String, String> sideInputIds, Set<String> leaveIds) {
    sideInputDataSets = new LinkedHashMap<>();
    leaves = new HashSet<>();
    // reset sources, so that the graph does not have two source objects
    // created during deserialization
    Set<BuildableTSet> newSources = new HashSet<>();
    for (BuildableTSet source : env.getGraph().getSources()) {
        newSources.add((BuildableTSet) env.getGraph().getNodeById(source.getId()));
    }
    env.getGraph().setSources(newSources);
    for (Map.Entry<String, String> entry : sideInputIds.entrySet()) {
        BatchTSet curr = (BatchTSet) env.getGraph().getNodeById(entry.getValue());
        sideInputDataSets.put(entry.getKey(), curr);
    }
    for (String leaveId : leaveIds) {
        leaves.add((TSet) env.getGraph().getNodeById(leaveId));
    }
}