#ifndef CUDANET_I_LAYER_H
#define CUDANET_I_LAYER_H

namespace CUDANet::Layers {

/**
 * @brief Padding types
 *
 * SAME: Zero padding such that the output size is the same as the input
 * VALID: No padding
 *
 */
enum Padding { SAME, VALID };

/**
 * @brief Basic Sequential Layer
 * 
 */
class SequentialLayer {
  public:
    /**
     * @brief Destroy the Sequential Layer
     * 
     */
    virtual ~SequentialLayer() {};

    /**
     * @brief Forward propagation virtual function
     * 
     * @param input Device pointer to the input
     * @return float* Device pointer to the output
     */
    virtual float* forward(const float* input) = 0;
};

/**
 * @brief Base class for layers with weights and biases
 */
class WeightedLayer : public SequentialLayer {
  public:
    /**
     * @brief Destroy the ILayer object
     *
     */
    virtual ~WeightedLayer() {};

    /**
     * @brief Virtual function for forward pass
     *
     * @param input (Device) Pointer to the input
     * @return float* Device pointer to the output
     */
    virtual float* forward(const float* input) = 0;

    /**
     * @brief Virtual function for setting weights
     *
     * @param weights Pointer to the weights
     */
    virtual void setWeights(const float* weights) = 0;

    /**
     * @brief Virtual function for setting biases
     *
     * @param biases Pointer to the biases
     */
    virtual void setBiases(const float* biases) = 0;

  private:
    /**
     * @brief Initialize the weights
     */
    virtual void initializeWeights() = 0;

    /**
     * @brief Initialize the biases
     */
    virtual void initializeBiases() = 0;

    /**
     * @brief Copy the weights and biases to the device
     */
    virtual void toCuda() = 0;
};

}  // namespace CUDANet::Layers

#endif  // CUDANET_I_LAYERH