bundles/org.simantics.spreadsheet.graph/src/org/simantics/spreadsheet/graph/CellsExtended.java

   1 /*******************************************************************************
   2  * Copyright (c) 2007, 2010 Association for Decentralized Information Management
   3  * in Industry THTH ry.
   4  * All rights reserved. This program and the accompanying materials
   5  * are made available under the terms of the Eclipse Public License v1.0
   6  * which accompanies this distribution, and is available at
   7  * http://www.eclipse.org/legal/epl-v10.html
   8  *
   9  * Contributors:
  10  *     VTT Technical Research Centre of Finland - initial API and implementation
  11  *******************************************************************************/
  12 package org.simantics.spreadsheet.graph;
  13
  14 import java.util.ArrayList;
  15 import java.util.Collection;
  16
  17 import org.simantics.db.ReadGraph;
  18 import org.simantics.db.exception.DatabaseException;
  19 import org.simantics.db.layer0.request.VariableRead;
  20 import org.simantics.db.layer0.variable.Variable;
  21 import org.simantics.utils.datastructures.Pair;
  22
  23 public class CellsExtended extends VariableRead<Collection<Pair<String, Variable>>> {
  24
  25     public CellsExtended(Variable variable) {
  26         super(variable);
  27     }
  28     @Override
  29     public Collection<Pair<String, Variable>> perform(ReadGraph graph) throws DatabaseException {
  30
  31         ArrayList<Pair<String, Variable>> cells = new ArrayList<>(graph.syncRequest(new Cells(variable)));
  32
  33         Variable lines = variable.getPossibleChild(graph, "Lines");
  34         if(lines != null) {
  35                 for(Variable line : graph.syncRequest(new Lines(lines))) {
  36                         cells.addAll(graph.syncRequest(new Cells(line)));
  37                 }
  38         }
  39
  40         return cells;
  41
  42     }
  43
  44 }