NeuralNetwork.hpp

#pragma once

#include "neuro_global.hpp"

#include "DataTransformation.hpp"
#include "NeuralLayer.hpp"

#include <QMetaType>
#include <QVector>

class NEUROSHARED_EXPORT NeuralNetwork: public DataTransformation
{
public:
    NeuralNetwork()=default;

    NeuralNetwork(const NeuralNetwork &)=default;
    NeuralNetwork(NeuralNetwork &&)=default;
    NeuralNetwork &operator=(const NeuralNetwork &)=default;
    NeuralNetwork &operator=(NeuralNetwork &&)=default;

    void swap(NeuralNetwork &neuralNetwork) noexcept;

    bool empty() const;
    int layerNumber() const;
    NeuralLayer &layer(const int index);
    const NeuralLayer &layer(const int index) const;
    NeuralLayer &inputLayer();
    const NeuralLayer &inputLayer() const;
    NeuralLayer &outputLayer();
    const NeuralLayer &outputLayer() const;
    void pushFront(const NeuralLayer &layer);
    void pushBack(const NeuralLayer &layer);
    void popFront();
    void popBack();
    void clear();

    void setActivationFunction(const ActivationFunctionPointer &activationFunction);

    int inputNumber() const override;
    int outputNumber() const override;

protected:
    DataVector transform_(const DataVector &input) const override;

private:
    inline friend QVector<NeuralLayer>::iterator begin(NeuralNetwork &neuralNetwork)
    {
        return neuralNetwork.m_layers.begin();
    }

    inline friend QVector<NeuralLayer>::const_iterator begin(const NeuralNetwork &neuralNetwork)
    {
        return neuralNetwork.m_layers.begin();
    }

    inline friend QVector<NeuralLayer>::iterator end(NeuralNetwork &neuralNetwork)
    {
        return neuralNetwork.m_layers.end();
    }

    inline friend QVector<NeuralLayer>::const_iterator end(const NeuralNetwork &neuralNetwork)
    {
        return neuralNetwork.m_layers.end();
    }

private:
    QVector<NeuralLayer> m_layers;
};

Q_DECLARE_METATYPE(NeuralNetwork)