package org.djunits.value.vfloat.vector.data;
   
   import java.util.stream.IntStream;
   
   import org.djunits.value.storage.StorageType;
   import org.djunits.value.vfloat.function.FloatFunction;
   import org.djunits.value.vfloat.function.FloatFunction2;
   
   /**
   * Stores dense 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 FloatVectorDataDense extends FloatVectorData
  {
      /** */
      private static final long serialVersionUID = 1L;
  
      /**
       * Create a vector with dense data.
       * @param vectorSI float[]; the data to store
       */
      public FloatVectorDataDense(final float[] vectorSI)
      {
          super(StorageType.DENSE);
          this.vectorSI = new float[vectorSI.length];
          System.arraycopy(vectorSI, 0, this.vectorSI, 0, vectorSI.length);
      }
  
      @Override
      public final int cardinality()
      {
          // this does not copy the data. See http://stackoverflow.com/questions/23106093/how-to-get-a-stream-from-a-float
          return (int) IntStream.range(0, this.vectorSI.length).parallel().mapToDouble(i -> this.vectorSI[i])
                  .filter(d -> d != 0.0).count();
      }
  
      @Override
      public final FloatVectorDataDense assign(final FloatFunction floatFunction)
      {
          IntStream.range(0, size()).parallel().forEach(i -> this.vectorSI[i] = floatFunction.apply(this.vectorSI[i]));
          return this;
      }
  
      @Override
      public final FloatVectorDataDense assign(final FloatFunction2 floatFunction2, final FloatVectorData right)
      {
          if (right.isDense())
          {
              IntStream.range(0, size()).parallel()
                      .forEach(i -> this.vectorSI[i] = floatFunction2.apply(this.vectorSI[i], right.vectorSI[i]));
          }
          else
          { // right is sparse
              IntStream.range(0, size()).parallel()
                      .forEach(i -> this.vectorSI[i] = floatFunction2.apply(this.vectorSI[i], right.getSI(i)));
          }
          return this;
      }
  
      @Override
      public final FloatVectorDataDense toDense()
      {
          return this;
      }
  
      @Override
      public final FloatVectorDataSparse toSparse()
      {
          return FloatVectorDataSparse.instantiate(this.vectorSI);
      }
  
      @Override
      public final int size()
      {
          return this.vectorSI.length;
      }
  
      @Override
      public final float getSI(final int index)
      {
          return this.vectorSI[index];
      }
  
      @Override
      public final void setSI(final int index, final float valueSI)
      {
          this.vectorSI[index] = valueSI;
      }
  
      @Override
      public final float[] getDenseVectorSI()
      {
          return this.vectorSI.clone();
      }
 
     @Override
     public final FloatVectorDataDense copy()
     {
         float[] vCopy = new float[this.vectorSI.length];
         System.arraycopy(this.vectorSI, 0, vCopy, 0, this.vectorSI.length);
         return new FloatVectorDataDense(vCopy);
     }
 
     @Override
     public final FloatVectorDataDense plus(final FloatVectorData right)
     {
         checkSizes(right);
         float[] out = new float[size()];
         IntStream.range(0, size()).parallel().forEach(i -> out[i] = getSI(i) + right.getSI(i));
         return new FloatVectorDataDense(out);
     }
 
     @Override
     public final FloatVectorDataDense minus(final FloatVectorData right)
     {
         checkSizes(right);
         float[] out = new float[size()];
         IntStream.range(0, size()).parallel().forEach(i -> out[i] = getSI(i) - right.getSI(i));
         return new FloatVectorDataDense(out);
     }
 
     @Override
     public final FloatVectorData times(final FloatVectorData right)
     {
         if (right.isSparse())
         {
             // result shall be sparse
             return right.times(this);
         }
         // Both are dense
         checkSizes(right);
         return this.copy().multiplyBy(right);
     }
 
     @Override
     public final FloatVectorData divide(final FloatVectorData right)
     {
         checkSizes(right);
         return this.copy().divideBy(right);
     }
 
 }