package org.djunits.value.vfloat.vector.data;
   
   import java.util.Arrays;
   import java.util.stream.IntStream;
   
   import org.djunits.value.ValueRuntimeException;
   import org.djunits.value.storage.StorageType;
   import org.djunits.value.vfloat.function.FloatFunction;
   import org.djunits.value.vfloat.function.FloatFunction2;
  
  /**
   * Stores sparse data for a FloatVector and carries out basic operations.
   * <p>
   * Copyright (c) 2013-2024 Delft University of Technology, PO Box 5, 2600 AA, Delft, the Netherlands. All rights reserved. <br>
   * BSD-style license. See <a href="https://djunits.org/docs/license.html">DJUNITS License</a>.
   * </p>
   * @author <a href="https://www.tudelft.nl/averbraeck">Alexander Verbraeck</a>
   * @author <a href="https://www.tudelft.nl/staff/p.knoppers/">Peter Knoppers</a>
   */
  public class FloatVectorDataSparse extends FloatVectorData
  {
      /** */
      private static final long serialVersionUID = 1L;
  
      /** The index values of the Vector. */
      private int[] indices;
  
      /** The length of the vector (padded with 0 after highest index in indices). */
      private final int size;
  
      /**
       * Create a vector with sparse data.
       * @param vectorSI float[]; the data to store
       * @param indices int[]; the index values of the Vector
       * @param size int; the length of the vector (padded with 0 after highest index in indices)
       */
      public FloatVectorDataSparse(final float[] vectorSI, final int[] indices, final int size)
      {
          super(StorageType.SPARSE);
          this.vectorSI = vectorSI;
          this.indices = indices;
          this.size = size;
      }
  
      @Override
      public final int cardinality()
      {
          return this.indices.length;
      }
  
      @Override
      public FloatVectorData assign(final FloatFunction floatFunction)
      {
          if (floatFunction.apply(0f) != 0f)
          {
              // It is most unlikely that the result AND the left and right operands are efficiently stored in Sparse format
              FloatVectorDataSparse result = toDense().assign(floatFunction).toSparse();
              this.indices = result.indices;
              this.vectorSI = result.vectorSI;
              return this;
          }
          // The code below relies on the fact that doubleFunction.apply(0d) yields 0d
          int currentSize = size();
          if (currentSize > 16)
          {
              currentSize = 16;
          }
          int[] newIndices = new int[currentSize];
          float[] newValues = new float[currentSize];
          int nonZeroValues = 0;
          int ownIndex = 0;
          while (ownIndex < this.indices.length)
          {
              int index = this.indices[ownIndex];
              float value = floatFunction.apply(this.vectorSI[ownIndex]);
              ownIndex++;
              if (value != 0f)
              {
                  if (nonZeroValues >= currentSize)
                  {
                      // increase size of arrays
                      currentSize *= 2;
                      if (currentSize > size())
                      {
                          currentSize = size();
                      }
                      int[] newNewIndices = new int[currentSize];
                      System.arraycopy(newIndices, 0, newNewIndices, 0, newIndices.length);
                      newIndices = newNewIndices;
                      float[] newNewValues = new float[currentSize];
                      System.arraycopy(newValues, 0, newNewValues, 0, newValues.length);
                      newValues = newNewValues;
                  }
                  newIndices[nonZeroValues] = index;
                  newValues[nonZeroValues] = value;
                  nonZeroValues++;
              }
          }
          if (nonZeroValues < currentSize)
         {
             // reduce size of arrays
             int[] newNewIndices = new int[nonZeroValues];
             System.arraycopy(newIndices, 0, newNewIndices, 0, nonZeroValues);
             newIndices = newNewIndices;
             float[] newNewValues = new float[nonZeroValues];
             System.arraycopy(newValues, 0, newNewValues, 0, nonZeroValues);
             newValues = newNewValues;
         }
         this.indices = newIndices;
         this.vectorSI = newValues;
         return this;
     }
 
     @Override
     public final FloatVectorDataSparse assign(final FloatFunction2 floatFunction, final FloatVectorData right)
     {
         if (floatFunction.apply(0f, 0f) != 0f)
         {
             // It is most unlikely that the result AND the left and right operands are efficiently stored in Sparse format
             FloatVectorDataSparse result = toDense().assign(floatFunction, right).toSparse();
             this.indices = result.indices;
             this.vectorSI = result.vectorSI;
             return this;
         }
         // The code below relies on the fact that doubleFunction.apply(0d, 0d) yields 0d
         checkSizes(right);
         int currentSize = size();
         if (currentSize > 16)
         {
             currentSize = 16;
         }
         int[] newIndices = new int[currentSize];
         float[] newValues = new float[currentSize];
         int nonZeroValues = 0;
         int ownIndex = 0;
         int otherIndex = 0;
         if (right.isSparse())
         { // both are sparse, result must be sparse
             FloatVectorDataSparse other = (FloatVectorDataSparse) right;
             while (ownIndex < this.indices.length || otherIndex < other.indices.length)
             {
                 float value;
                 int index;
                 if (ownIndex < this.indices.length && otherIndex < other.indices.length)
                 { // neither we nor right has run out of values
                     if (this.indices[ownIndex] == other.indices[otherIndex])
                     {
                         value = floatFunction.apply(this.vectorSI[ownIndex], other.vectorSI[otherIndex]);
                         index = this.indices[ownIndex];
                         ownIndex++;
                         otherIndex++;
                     }
                     else if (this.indices[ownIndex] < other.indices[otherIndex])
                     {
                         // we have a non-zero; right has a zero
                         value = floatFunction.apply(this.vectorSI[ownIndex], 0f);
                         index = this.indices[ownIndex];
                         ownIndex++;
                     }
                     else
                     {
                         // we have a zero; right has a non-zero
                         value = floatFunction.apply(0f, other.vectorSI[otherIndex]);
                         index = other.indices[otherIndex];
                         otherIndex++;
                     }
                 }
                 else if (ownIndex < this.indices.length)
                 { // right has run out of values; we have not
                     value = floatFunction.apply(this.vectorSI[ownIndex], 0f);
                     index = this.indices[ownIndex];
                     ownIndex++;
                 }
                 else
                 { // we have run out of values; right has not
                     value = floatFunction.apply(0f, other.vectorSI[otherIndex]);
                     index = other.indices[otherIndex];
                     otherIndex++;
                 }
                 if (value != 0f)
                 {
                     if (nonZeroValues >= currentSize)
                     {
                         // increase size of arrays
                         currentSize *= 2;
                         if (currentSize > size())
                         {
                             currentSize = size();
                         }
                         int[] newNewIndices = new int[currentSize];
                         System.arraycopy(newIndices, 0, newNewIndices, 0, newIndices.length);
                         newIndices = newNewIndices;
                         float[] newNewValues = new float[currentSize];
                         System.arraycopy(newValues, 0, newNewValues, 0, newValues.length);
                         newValues = newNewValues;
                     }
                     newIndices[nonZeroValues] = index;
                     newValues[nonZeroValues] = value;
                     nonZeroValues++;
                 }
             }
         }
         else
         { // we are sparse; other is dense, result must be sparse
             FloatVectorDataDense other = (FloatVectorDataDense) right;
             while (otherIndex < right.vectorSI.length)
             {
                 float value;
                 int index = otherIndex;
                 if (ownIndex < this.indices.length)
                 { // neither we nor right has run out of values
                     if (this.indices[ownIndex] == otherIndex)
                     {
                         value = floatFunction.apply(this.vectorSI[ownIndex], other.vectorSI[otherIndex]);
                         ownIndex++;
                     }
                     else
                     {
                         // we have a zero; other has a value
                         value = floatFunction.apply(0f, other.vectorSI[otherIndex]);
                     }
                     otherIndex++;
                 }
                 else
                 { // we have run out of values; right has not
                     value = floatFunction.apply(0f, other.vectorSI[otherIndex]);
                     otherIndex++;
                 }
                 if (value != 0f)
                 {
                     if (nonZeroValues >= currentSize)
                     {
                         // increase size of arrays
                         currentSize *= 2;
                         if (currentSize > size())
                         {
                             currentSize = size();
                         }
                         int[] newNewIndices = new int[currentSize];
                         System.arraycopy(newIndices, 0, newNewIndices, 0, newIndices.length);
                         newIndices = newNewIndices;
                         float[] newNewValues = new float[currentSize];
                         System.arraycopy(newValues, 0, newNewValues, 0, newValues.length);
                         newValues = newNewValues;
                     }
                     newIndices[nonZeroValues] = index;
                     newValues[nonZeroValues] = value;
                     nonZeroValues++;
                 }
             }
         }
         if (nonZeroValues < currentSize)
         {
             // reduce size of arrays
             int[] newNewIndices = new int[nonZeroValues];
             System.arraycopy(newIndices, 0, newNewIndices, 0, nonZeroValues);
             newIndices = newNewIndices;
             float[] newNewValues = new float[nonZeroValues];
             System.arraycopy(newValues, 0, newNewValues, 0, nonZeroValues);
             newValues = newNewValues;
         }
         this.indices = newIndices;
         this.vectorSI = newValues;
         return this;
     }
 
     @Override
     public final FloatVectorDataDense toDense()
     {
         float[] denseSI = new float[this.size];
         for (int index = 0; index < this.vectorSI.length; index++)
         {
             denseSI[this.indices[index]] = this.vectorSI[index];
         }
         return new FloatVectorDataDense(denseSI);
     }
 
     @Override
     public final FloatVectorDataSparse toSparse()
     {
         return this;
     }
 
     @Override
     public final int size()
     {
         return this.size;
     }
 
     @Override
     public final float getSI(final int index)
     {
         int internalIndex = Arrays.binarySearch(this.indices, index);
         return internalIndex < 0 ? 0.0f : this.vectorSI[internalIndex];
     }
 
     @Override
     public final void setSI(final int index, final float valueSI)
     {
         int internalIndex = Arrays.binarySearch(this.indices, index);
         if (internalIndex >= 0)
         {
             if (valueSI == 0f)
             {
                 // remove room
                 int[] indicesNew = new int[this.indices.length - 1];
                 float[] vectorSINew = new float[this.vectorSI.length - 1];
                 System.arraycopy(this.indices, 0, indicesNew, 0, internalIndex);
                 System.arraycopy(this.vectorSI, 0, vectorSINew, 0, internalIndex);
                 System.arraycopy(this.indices, internalIndex + 1, indicesNew, internalIndex,
                         this.indices.length - internalIndex - 1);
                 System.arraycopy(this.vectorSI, internalIndex + 1, vectorSINew, internalIndex,
                         this.indices.length - internalIndex - 1);
                 this.indices = indicesNew;
                 this.vectorSI = vectorSINew;
             }
             else
             {
                 this.vectorSI[internalIndex] = valueSI;
             }
             return;
         }
         if (valueSI == 0f)
         {
             return;
         }
         // make room. TODO increase size in chunks
         internalIndex = -internalIndex - 1;
         int[] indicesNew = new int[this.indices.length + 1];
         float[] vectorSINew = new float[this.vectorSI.length + 1];
         System.arraycopy(this.indices, 0, indicesNew, 0, internalIndex);
         System.arraycopy(this.vectorSI, 0, vectorSINew, 0, internalIndex);
         System.arraycopy(this.indices, internalIndex, indicesNew, internalIndex + 1, this.indices.length - internalIndex);
         System.arraycopy(this.vectorSI, internalIndex, vectorSINew, internalIndex + 1, this.indices.length - internalIndex);
         indicesNew[internalIndex] = index;
         vectorSINew[internalIndex] = valueSI;
         this.indices = indicesNew;
         this.vectorSI = vectorSINew;
     }
 
     @Override
     public final float[] getDenseVectorSI()
     {
         return toDense().vectorSI;
     }
 
     @Override
     public final FloatVectorDataSparse copy()
     {
         float[] vCopy = new float[this.vectorSI.length];
         System.arraycopy(this.vectorSI, 0, vCopy, 0, this.vectorSI.length);
         int[] iCopy = new int[this.indices.length];
         System.arraycopy(this.indices, 0, iCopy, 0, this.indices.length);
         return new FloatVectorDataSparse(vCopy, iCopy, this.size);
     }
 
     /**
      * Instantiate a FloatVectorDataSparse from an array.
      * @param valuesSI float[]; the (SI) values to store
      * @return the FloatVectorDataSparse
      */
     public static FloatVectorDataSparse instantiate(final float[] valuesSI)
     {
         // determine number of non-null cells
         int length = (int) IntStream.range(0, valuesSI.length).parallel().mapToDouble(i -> valuesSI[i]).filter(f -> f != 0.0f)
                 .count();
         float[] sparseSI = new float[length];
         int[] indices = new int[length];
 
         // fill the sparse data structures. Cannot be parallelized because of stateful and sequence-sensitive count
         int count = 0;
         for (int i = 0; i < valuesSI.length; i++)
         {
             if (valuesSI[i] != 0.0)
             {
                 sparseSI[count] = valuesSI[i];
                 indices[count] = i;
                 count++;
             }
         }
         return new FloatVectorDataSparse(sparseSI, indices, valuesSI.length);
     }
 
     @Override
     public final FloatVectorData plus(final FloatVectorData right)
     {
         if (right.isDense())
         {
             return right.plus(this);
         }
         return this.copy().incrementBy(right);
     }
 
     @Override
     public final FloatVectorData minus(final FloatVectorData right)
     {
         if (right.isDense())
         {
             return this.toDense().decrementBy(right);
         }
         return this.copy().decrementBy(right);
     }
 
     @Override
     public final FloatVectorData times(final FloatVectorData right)
     {
         return this.copy().multiplyBy(right);
     }
 
     @Override
     public final FloatVectorData divide(final FloatVectorData right) throws ValueRuntimeException
     {
         if (right.isSparse())
         {
             // Sparse divided by sparse makes a dense
             return this.toDense().divideBy(right);
         }
         // Sparse divided by dense makes a sparse
         return this.copy().divideBy(right);
     }
 
     @Override
     public int hashCode()
     {
         return super.hashCode();
     }
 
     @Override
     @SuppressWarnings({"checkstyle:needbraces", "checkstyle:designforextension"})
     public boolean equals(final Object obj)
     {
         if (this == obj)
             return true;
         if (obj == null)
             return false;
         if (!(obj instanceof FloatVectorData))
             return false;
         FloatVectorData other = (FloatVectorData) obj;
         if (this.size() != other.size())
             return false;
         if (other instanceof FloatVectorDataDense)
             return super.equals(other);
         if (getClass() != obj.getClass())
             return false;
         // Both are sparse
         if (!Arrays.equals(this.indices, ((FloatVectorDataSparse) other).indices))
             return false;
         return Arrays.equals(this.vectorSI, ((FloatVectorDataSparse) other).vectorSI);
     }
 
 }