Examples of cc.redberry.core.tensor.iterator.TreeTraverseIterator

Package cc.redberry.core.tensor.iterator

Examples of cc.redberry.core.tensor.iterator.TreeTraverseIterator

cc.redberry.core.tensor.iterator.TreeTraverseIterator

An iterator for tensors that allows the programmer to traverse the tensor tree structure, modify the tensor during iteration, and obtain information about iterator's current position in the tensor. A {@code TensorTreeIterator}has current element, so all methods are defined in terms of the cursor position. *

Example:

 Tensor tensor = Tensors.parse("Cos[a+b+Sin[x]]"); TensorIterator iterator = new TreeTraverseIterator(tensor); TraverseState state; while ((state = iterator.next()) != null) System.out.println(state + " " + iterator.depth() + " " + iterator.current());

This code will print:

 Entering   Cos[a+b+Sin[y*c+x]] Entering   a+b+Sin[y*c+x] Entering   a Leaving   a Entering   b Leaving   b Entering   Sin[y*c+x] Entering   y*c+x Entering   y*c Entering   y Leaving   y Entering   c Leaving   c Leaving   y*c Entering   x Leaving   x Leaving   y*c+x Leaving   Sin[y*c+x] Leaving   a+b+Sin[y*c+x] Leaving   Cos[a+b+Sin[y*c+x]]

@author Dmitry Bolotin @author Stanislav Poslavsky

        Tensor M = Tensors.parse("M");
        Tensor M8 = Tensors.parse("Power[M,8]");
        Tensor pT = Tensors.parse("pT");
        Tensor s = Tensors.parse("s");
        for (int i = 0; i < 1000; ++i) {
            TreeTraverseIterator iterator = new TreeTraverseIterator(target);


            while (iterator.next() != null)
                if (TensorUtils.equals(iterator.current(), pT))
                    iterator.set(M8);
                else if (TensorUtils.equals(iterator.current(), s))
                    iterator.set(AllToM);
            iterator.result();
        }
    }

View Full Code Here

        this.to = to;
    }


    @Override
    public Tensor transform(Tensor tensor) {
        TreeTraverseIterator iterator = new TreeTraverseIterator(tensor);
        TraverseState state;
        Tensor current;
        while ((state = iterator.next()) != null) {
            if (state != TraverseState.Leaving)
                continue;
            current = iterator.current();
            IndexMappingBuffer buffer = IndexMappings.getFirst(from, current);
            if (buffer != null) {
                Tensor newFrom = ApplyIndexMapping.applyIndexMapping(to, buffer);
                iterator.set(newFrom);
            }
        }
        return iterator.result();
    }

View Full Code Here

0 1 2

TOP

Related Classes of cc.redberry.core.tensor.iterator.TreeTraverseIterator

cc.redberry.core.performance.Substitution

cc.redberry.core.performance.SubstitutionBenchmark

cc.redberry.core.tensor.Tensor

cc.redberry.core.transformations.ApplyIndexMapping

cc.redberry.core.transformations.EliminateDueSymmetriesTransformation

cc.redberry.core.transformations.EliminateFromSymmetriesTransformation

cc.redberry.core.transformations.Expand

cc.redberry.core.transformations.RemoveDueToSymmetry

cc.redberry.core.transformations.substitutions.SubstitutionIterator

cc.redberry.core.transformations.Transformer

All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.