neureka.framing.Relation Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of neureka Show documentation
A platform independent tensor library written in Java.
There is a newer version: 1.0.1
package neureka.framing;

import lombok.Getter;
import lombok.ToString;
import lombok.experimental.Accessors;
import neureka.Component;
import neureka.Tsr;

import java.lang.ref.WeakReference;
import java.util.function.Consumer;


@Accessors( prefix = {"_"} )
@ToString
public class Relation implements Component>
{
    @Getter
    private Tsr _parent;// Children need their parents. They shall not be garbage collected.

    private WeakReference>[] _children;// Children may be garbage collected if not needed.

    private int[][] _shapeRelations;

    @Override
    public void update( Tsr oldOwner, Tsr newOwner ) {
        if ( _parent != null) {
            Relation pr = _parent.find(Relation.class);
            for ( int i=0; i < pr._children.length; i++ ) {
                if (pr._children[ i ].get() == oldOwner) {
                    pr._children[ i ] = new WeakReference<>(newOwner);
                }
            }
        }
        if ( _children != null ) {
            for ( WeakReference> c : _children ) {
                Tsr t = c.get();
                if ( t != null ) {
                    Relation cr = (Relation) t.find( Relation.class );
                    if ( cr != null ) cr._parent = newOwner;
                }
            }
        }
    }


    public Relation addParent( Tsr parent ) {
        _parent = parent;
        return this;
    }


    public Relation addChild( Tsr child ) {
        if ( _children == null ) {
            _children = new WeakReference[]{ new WeakReference( child ) };
            _shapeRelations = new int[1][];
        } else {
            WeakReference>[] newChildren = new WeakReference[_children.length+1];
            int[][] newShapeRelations = new int[_children.length+1][];
            System.arraycopy( _children, 0, newChildren, 0, _children.length );
            System.arraycopy( _shapeRelations, 0, newShapeRelations, 0, _children.length );
            newChildren[_children.length] = new WeakReference(child);
            newShapeRelations[_children.length] = null;
            _children = newChildren;
            _shapeRelations = newShapeRelations;
        }
        return this;
    }

    public Relation addReshapeRelationFor( Tsr child, int[] reshape ) {
        for ( int i=0; i<_shapeRelations.length; i++ ) {
            Tsr c = _children[ i ].get();
            if ( c != null && c == child ) {
                _shapeRelations[ i ] = reshape;
            }
        }
        return this;
    }

    public int[] getReshapeRelationFor( Tsr child ) {
        for ( int i=0; i<_shapeRelations.length; i++ ) {
            Tsr c = _children[ i ].get();
            if ( c != null && c == child ) {
                return _shapeRelations[ i ];
            }
        }
        return null;
    }

    public Relation foreachChild( Consumer> action ) {
        if ( _children != null ) {
            for ( WeakReference> r : _children ) {
                Tsr c = r.get();
                if ( c != null ) {
                    action.accept( c );
                    Relation relation = (Relation) c.find( Relation.class );
                    if ( relation != null ) relation.foreachChild( action );
                }
            }
        }
        return this;
    }

    public Tsr findRootTensor() {
        if ( _parent == null ) return null;
        else if ( !_parent.has( Relation.class ) ) return null;
        else if ( !_parent.find( Relation.class ).hasParent() ) return _parent;
        else return _parent.find( Relation.class ).findRootTensor();
    }

    public boolean hasParent() {
        return _parent != null;
    }

    public boolean hasChildren() {
        return _children != null;
    }

    public int childCount() {
        return ( _children == null ) ? 0 : _children.length;
    }

    public Relation remove( Tsr child ) {
        //TODO!!
        return this;
    }

}