Sliced3DConfiguration.java

  1. package neureka.ndim.config.types.sliced;

  3. import neureka.ndim.config.types.D3C;

  5. public class Sliced3DConfiguration extends D3C //:= IMMUTABLE
  6. {
  7.     /**
  8.      *  The shape of the NDArray.
  9.      */
  10.     protected final int _shape1;
  11.     protected final int _shape2;
  12.     protected final int _shape3;
  13.     /**
  14.      *  The translation from a shape index (indices) to the index of the underlying data array.
  15.      */
  16.     private final int _stride1;
  17.     private final int _stride2;
  18.     private final int _stride3;
  19.     /**
  20.      *  The mapping of idx array.
  21.      */
  22.     private final int _indicesMap1;
  23.     private final int _indicesMap2; // Maps index integer to array like translation. Used to avoid distortion when slicing!
  24.     private final int _indicesMap3;
  25.     /**
  26.      *  Produces the steps of a tensor subset / slice
  27.      */
  28.     private final int _spread1;
  29.     private final int _spread2;
  30.     private final int _spread3;
  31.     /**
  32.      *  Defines the position of a subset / slice tensor within its parent!
  33.      */
  34.     private final int _offset1;
  35.     private final int _offset2;
  36.     private final int _offset3;


  39.     protected Sliced3DConfiguration(
  40.             int[] shape,
  41.             int[] strides,
  42.             int[] indicesMap,
  43.             int[] spread,
  44.             int[] offset
  45.     ) {
  46.         _shape1 = shape[ 0 ];
  47.         _shape2 = shape[ 1 ];
  48.         _shape3 = shape[ 2 ];
  49.         _stride1 = strides[ 0 ];
  50.         _stride2 = strides[ 1 ];
  51.         _stride3 = strides[ 2 ];
  52.         _indicesMap1 = indicesMap[ 0 ];
  53.         _indicesMap2 = indicesMap[ 1 ];
  54.         _indicesMap3 = indicesMap[ 2 ];
  55.         _spread1 = spread[ 0 ];
  56.         _spread2 = spread[ 1 ];
  57.         _spread3 = spread[ 2 ];
  58.         _offset1 = offset[ 0 ];
  59.         _offset2 = offset[ 1 ];
  60.         _offset3 = offset[ 2 ];
  61.     }

  63.     public static Sliced3DConfiguration construct(
  64.             int[] shape,
  65.             int[] strides,
  66.             int[] indicesMap,
  67.             int[] spread,
  68.             int[] offset
  69.     ) {
  70.         return _cached( new Sliced3DConfiguration(shape, strides, indicesMap, spread, offset) );
  71.     }

  73.     /** {@inheritDoc} */
  74.     @Override public final int rank() { return 3; }

  76.     /** {@inheritDoc} */
  77.     @Override public final int[] shape() { return new int[]{ _shape1, _shape2, _shape3 }; }

  79.     /** {@inheritDoc} */
  80.     @Override public final int shape( int i ) { return ( i==0 ? _shape1 : ( i==1 ? _shape2 : _shape3 ) ); }

  82.     /** {@inheritDoc} */
  83.     @Override public final int[] indicesMap() { return new int[]{ _indicesMap1, _indicesMap2, _indicesMap3 }; }

  85.     /** {@inheritDoc} */
  86.     @Override public final int indicesMap( int i ) { return ( i == 0 ? _indicesMap1 : ( i==1 ? _indicesMap2 : _indicesMap3 ) ); }

  88.     /** {@inheritDoc} */
  89.     @Override
  90.     public final int[] strides() { return new int[]{_stride1, _stride2, _stride3}; }

  92.     /** {@inheritDoc} */
  93.     @Override public final int strides(int i ) { return ( i == 0 ? _stride1 : ( i == 1 ? _stride2 : _stride3) ); }

  95.     /** {@inheritDoc} */
  96.     @Override public final int[] spread() { return new int[]{ _spread1, _spread2, _spread3 }; }

  98.     /** {@inheritDoc} */
  99.     @Override public final int spread( int i ) { return ( i == 0 ? _spread1 : ( i == 1 ? _spread2 : _spread3 ) ); }

  101.     /** {@inheritDoc} */
  102.     @Override public final int[] offset() { return new int[]{ _offset1, _offset2, _offset3 }; }

  104.     /** {@inheritDoc} */
  105.     @Override public final int offset( int i ) { return ( i == 0 ? _offset1 : ( i == 1 ? _offset2 : _offset3 ) ); }

  107.     /** {@inheritDoc} */
  108.     @Override
  109.     public final int indexOfIndex( int index ) {
  110.         int indices1, indices2, indices3;
  111.         indices1 = index / _indicesMap1;
  112.         index %= _indicesMap1;
  113.         indices2 = index / _indicesMap2;
  114.         index %= _indicesMap2;
  115.         indices3 = index / _indicesMap3;
  116.         return (indices1 * _spread1 + _offset1) * _stride1 +
  117.                 (indices2 * _spread2 + _offset2) * _stride2 +
  118.                 (indices3 * _spread3 + _offset3) * _stride3;
  119.     }

  121.     /** {@inheritDoc} */
  122.     @Override
  123.     public final int[] indicesOfIndex( int index ) {
  124.         int indices1, indices2, indices3;
  125.         indices1 = index / _indicesMap1;
  126.         index %= _indicesMap1;
  127.         indices2 = index / _indicesMap2;
  128.         index %= _indicesMap2;
  129.         indices3 = index / _indicesMap3;
  130.         return new int[]{indices1, indices2, indices3};
  131.     }

  133.     /** {@inheritDoc} */
  134.     @Override
  135.     public final int indexOfIndices(int[] indices) {
  136.         return (indices[ 0 ] * _spread1 + _offset1) * _stride1 +
  137.                     (indices[ 1 ] * _spread2 + _offset2) * _stride2 +
  138.                         (indices[ 2 ] * _spread3 + _offset3) * _stride3;
  139.     }

  141.     /** {@inheritDoc} */
  142.     @Override
  143.     public final int indexOfIndices(int d1, int d2, int d3 ) {
  144.         return (d1 * _spread1 + _offset1) * _stride1 +
  145.                 (d2 * _spread2 + _offset2) * _stride2 +
  146.                 (d3 * _spread3 + _offset3) * _stride3;
  147.     }

  149. }
Created with JaCoCo 0.8.11.202310140853