// %flair:license{
// This file is part of the Flair framework distributed under the
// CECILL-C License, Version 1.0.
// %flair:license}
//  created:    2018/01/30
//  filename:   Matrix.cpp
//
//  author:     Guillaume Sanahuja
//              Copyright Heudiasyc UMR UTC/CNRS 7253
//
//  version:    $Id: $
//
//  purpose:    Class defining a matrix
//
/*********************************************************************/

#include "Matrix.h"
#include "Matrix_impl.h"
#include <typeinfo>
#include <string.h>

//#include <iostream>

using std::string;

namespace flair {
namespace core {
/*! \class MatrixElement
*/
class MatrixElement : public IODataElement {
public:
  MatrixElement(const Matrix *matrix, string name, uint32_t row,
                  uint32_t col)
      : IODataElement(matrix, name) {
    this->matrix = matrix;
    this->row = row;
    this->col = col;
    if (row >= matrix->Rows() || col >= matrix->Cols()) {
      matrix->Err("index (%i,%i) out of bound (%i,%i)\n", row, col,
                  matrix->Rows() - 1, matrix->Cols() - 1);
      size = 0;
    } else {
      try {
        ScalarType const &scalarType = dynamic_cast<ScalarType const &>(
            matrix->GetDataType().GetElementDataType());
        size = scalarType.GetSize();
      } catch (std::bad_cast e) {
        matrix->Err("type not handled\n");
        size = 0;
      }
    }
  }

  ~MatrixElement() {}

  void CopyData(char *dst) const {
    if (typeid(matrix->GetDataType().GetElementDataType()) ==
        typeid(FloatType)) {
      float value = matrix->Value(row, col);
      memcpy(dst, &value, sizeof(value));
    } else if (typeid(matrix->GetDataType().GetElementDataType()) ==
               typeid(SignedIntegerType)) {
      switch (matrix->GetDataType().GetElementDataType().GetSize()) {
      case 1: {
        int8_t int8Value = matrix->Value(row, col);
        memcpy(dst, &int8Value, 1);
      } break;
      case 2: {
        int16_t int16Value = matrix->Value(row, col);
        memcpy(dst, &int16Value, 2);
      } break;
      }
    }
  }

  DataType const &GetDataType(void) const {
    return matrix->GetDataType().GetElementDataType();
  }

private:
  const Matrix *matrix;
  uint32_t row, col;
};

Matrix::Matrix(const Object *parent, uint32_t rows, uint32_t cols,
                   ScalarType const &elementDataType, string name, uint32_t n)
    : io_data(parent, name, n), dataType(rows, cols, elementDataType) {
  pimpl_ = new Matrix_impl(this, rows, cols, elementDataType, n);

  for (uint32_t i = 0; i < rows; i++) {
    for (uint32_t j = 0; j < cols; j++) {
      AppendLogDescription(pimpl_->descriptor->ElementName(i, j),
                           elementDataType);
      SetValue(i, j, 0);
    }
  }
}

Matrix::Matrix(const Object *parent, const cvmatrix_descriptor *descriptor,
                   ScalarType const &elementDataType, string name, uint32_t n)
    : io_data(parent, name, n),
      dataType(descriptor->Rows(), descriptor->Cols(), elementDataType) {
  pimpl_ = new Matrix_impl(this, descriptor, elementDataType, n);

  for (uint32_t i = 0; i < descriptor->Rows(); i++) {
    for (uint32_t j = 0; j < descriptor->Cols(); j++) {
      AppendLogDescription(descriptor->ElementName(i, j), elementDataType);
      SetValue(i, j, 0);
    }
  }
}

Matrix::~Matrix() { delete pimpl_; }

IODataElement *Matrix::Element(uint32_t row, uint32_t col) const {
  return new MatrixElement(this, Name(row, col), row, col);
}

IODataElement *Matrix::Element(uint32_t index) const {
  if (Rows() == 1) {
    return new MatrixElement(this, Name(0, index), 0, index);
  } else if (Cols() == 1) {
    return new MatrixElement(this, Name(index, 0), index, 0);
  } else {
    Err("matrix is not 1D\n");
    return nullptr;
  }
}

float Matrix::Value(uint32_t row, uint32_t col) const {
  float value;

  if (row >= (uint32_t)pimpl_->descriptor->Rows() ||
      col >= (uint32_t)pimpl_->descriptor->Cols()) {
    Warn("index (%i,%i) out of bound (%i,%i)\n", row, col,
         pimpl_->descriptor->Rows() - 1, pimpl_->descriptor->Cols() - 1);
    return 0;
  }

  GetMutex();
  value = pimpl_->datas[row][col];
  ReleaseMutex();

  return value;
}

float Matrix::ValueNoMutex(uint32_t row, uint32_t col) const {
  if (row >= (uint32_t)pimpl_->descriptor->Rows() ||
      col >= (uint32_t)pimpl_->descriptor->Cols()) {
    Warn("index (%i,%i) out of bound (%i,%i)\n", row, col,
         pimpl_->descriptor->Rows() - 1, pimpl_->descriptor->Cols() - 1);
    return 0;
  }

  return pimpl_->datas[row][col];
}

void Matrix::SetValue(uint32_t row, uint32_t col, float value) {
  if (row >= (uint32_t)pimpl_->descriptor->Rows() ||
      col >= (uint32_t)pimpl_->descriptor->Cols()) {
    Warn("index (%i,%i) out of bound (%i,%i)\n", row, col,
         pimpl_->descriptor->Rows() - 1, pimpl_->descriptor->Cols() - 1);
  } else {
    GetMutex();
    pimpl_->datas[row][col]=value;
    ReleaseMutex();
  }
}

void Matrix::SetValueNoMutex(uint32_t row, uint32_t col, float value) {
  if (row >= (uint32_t)pimpl_->descriptor->Rows() ||
      col >= (uint32_t)pimpl_->descriptor->Cols()) {
    Warn("index (%i,%i) out of bound (%i,%i)\n", row, col,
         pimpl_->descriptor->Rows() - 1, pimpl_->descriptor->Cols() - 1);
  } else {
    pimpl_->datas[row][col]=value;
  }
}

void Matrix::CopyDatas(char *dst) const {
  GetMutex();
  // printf("%f %x %i\n",cvGetReal2D(pimpl_->mat,0,0),dst,Size());
  memcpy(dst, pimpl_->datas, dataType.GetSize());
  ReleaseMutex();
}

uint32_t Matrix::Rows(void) const { return pimpl_->descriptor->Rows(); }

uint32_t Matrix::Cols(void) const { return pimpl_->descriptor->Cols(); }

string Matrix::Name(uint32_t row, uint32_t col) const {
  return pimpl_->descriptor->ElementName(row, col);
}

} // end namespace core
} // end namespace flair