/*
* Redberry: symbolic tensor computations.
*
* Copyright (c) 2010-2012:
* Stanislav Poslavsky <stvlpos@mail.ru>
* Bolotin Dmitriy <bolotin.dmitriy@gmail.com>
*
* This file is part of Redberry.
*
* Redberry is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Redberry is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with Redberry. If not, see <http://www.gnu.org/licenses/>.
*/
package cc.redberry.transformation.collect;
import cc.redberry.core.tensor.Sum;
import cc.redberry.core.tensor.Tensor;
import cc.redberry.core.tensor.testing.TTest;
/**
*
* @author Dmitry Bolotin
* @author Stanislav Poslavsky
*/
public class CollecctEqualsInputPort implements CollectInputPort {
private final CollectInputPortImpl symbols = new CollectInputPortImpl(ScalarsSplitCriteria.INSTANCE);
private final CollectInputPortImpl equals = new CollectInputPortImpl(EqualsSplitCriteria.INSTANCE);
private boolean closed = false;
@Override
public void put(Tensor tensor) {
if (closed)
throw new RuntimeException("Port is closed.");
if (tensor == null) {
closed = true;
return;
}
if (tensor instanceof Sum) {
for (Tensor t : tensor)
put(t);
return;
}
if (TTest.testIsSymbol(tensor))
symbols.put(tensor);
else
equals.put(tensor);
}
public final Tensor result() {
symbols.put(null);
equals.put(null);
if (!symbols.initialized())
return equals.result();
if (!equals.initialized())
return symbols.result();
Tensor s1 = symbols.result();
Tensor s2 = equals.result();
return new Sum(s1, s2);
}
@Override
public CollectInputPort create() {
return new CollecctEqualsInputPort();
}
@Override
public boolean initialized() {
return symbols.initialized() || equals.initialized();
}
}