FloatMatrix.h

Go to the documentation of this file.

#ifndef __FloatMatrix_h__
#define __FloatMatrix_h__
 
#include <cstddef>
#include "basicmaths.h"
 
class FloatMatrix {
private:
#ifdef ARM_CORTEX
  arm_matrix_instance_f32 instance;
#else
  float* data;
  size_t rows;
  size_t columns;
#endif
public:
  FloatMatrix();
  FloatMatrix(float* data, size_t rows, size_t columns);
  FloatMatrix(const float* data, size_t rows, size_t columns);
 
  size_t getSize() const{
    return getRows()*getColumns();
  }
 
  size_t getRows() const{
#ifdef ARM_CORTEX
    return instance.numRows;
#else
    return rows;
#endif
  }
 
  size_t getColumns() const{
#ifdef ARM_CORTEX
    return instance.numCols;
#else
    return columns;
#endif
  }
 
  void clear(){
    setAll(0);
  }  
  
  void add(FloatMatrix operand2, FloatMatrix destination);
  
  void add(FloatMatrix operand2); //in-place
 
  /* /\** */
  /*  * Element-wise sum between matrix and array. */
  /*  * Sets each element in **destination** to the sum of the corresponding element of the matrix and **operand2** */
  /*  * @param[in] operand2 second operand for the sum */
  /*  * @param[out] destination the destination array */
  /* *\/ */
  /* void add(FloatArray operand2, FloatMatrix destination); */
  
  /* /\** */
  /*  * Element-wise sum between matrix and array. */
  /*  * Adds each element of **operand2** to the corresponding element of the matrix. */
  /*  * @param operand2 second operand for the sum */
  /* *\/ */
  /* void add(FloatMatrix operand2); //in-place */
 
  void add(float scalar);
    
  void multiply(FloatMatrix operand2, FloatMatrix destination);
  
  void multiply(FloatMatrix operand2); //in-place
  
  void multiply(float scalar);
 
  void setAll(float value);  
  
  operator float*(){
    return getData();
  }
 
  float* operator [](const int index){
    return &getData()[index*getColumns()];
  }
 
  float* getData(){
#ifdef ARM_CORTEX
    return instance.pData;
#else
    return data;
#endif
  }
 
  float getElement(int row, int col){
    return getData()[row*getColumns()+col];
  }
 
  void setElement(int row, int col, float value){
    getData()[row*getColumns()+col] = value;
  }
 
  void softmax(FloatMatrix destination);
 
  void softmax(){
    softmax(*this);
  }
 
  void sigmoid(FloatMatrix destination);
 
  void sigmoid(){
    sigmoid(*this);
  }
 
  void copyTo(FloatMatrix destination);
 
  void copyFrom(FloatMatrix source);
 
  static FloatMatrix create(size_t rows, size_t columns);
  
  static void destroy(FloatMatrix array);
};
 
#endif // __FloatMatrix_h__