#include "acidcam/ac.h"

static constexpr int MAX = 8;
static ac::MatrixCollection<MAX> collection;

extern "C" void init() {
    std::cout << "pic_adjust1 initialized..\n";
    srand(static_cast<unsigned int>(time(0)));
}

extern "C" void rls() {
    ac::release_all_objects();
}

extern "C" void clear() {
    std::cout << "Clear called..\n";
    ac::release_all_objects();
}

extern "C" void proc(cv::Mat  &frame) {
    if(collection.empty()) {
        srand(static_cast<unsigned int>(time(0)));
        collection.shiftFrames(frame);
    }
    else
        collection.shiftFrames(frame);
    
    static bool flash = true;
    if(flash) {
        static int dir = 1;
        static int offset = 0;
        int size_y = 0;
        
        size_y = rand()%frame.rows;
        
        for(int z = 0; z < frame.rows/2; ++z) {
            for(int i = 0; i < frame.cols; ++i) {
                cv::Vec3b &pixel = frame.at<cv::Vec3b>(z, i);
                int cy = AC_GetFZ(frame.rows-1, z, size_y);
                if(cy >= 0 && cy < frame.rows) {
                    cv::Vec3b &pix = collection.frames[offset].at<cv::Vec3b>(cy, i);
                    pixel = pix;
                }
            }
            size_y ++;
        }
        
        for(int z = frame.rows/2; z < frame.rows; ++z) {
            for(int i = 0; i < frame.cols; ++i) {
                cv::Vec3b &pixel = frame.at<cv::Vec3b>(z, i);
                int cy = AC_GetFZ(frame.rows-1, z, size_y);
                
                if(cy >= 0 && cy < frame.rows) {
                    cv::Vec3b &pix = collection.frames[offset].at<cv::Vec3b>(cy, i);
                    pixel = pix;
                }
            }
            size_y --;
        }
        
        if(++offset > (MAX-1)) {
            offset = 0;
        }
        
    } else {
        static int dir = 1;
        static int offset = 0;
        static int div = 2;
        static int size_y = frame.rows/16;
        
        for(int z = 0; z < frame.rows; ++z) {
            for(int i = 0; i < frame.cols; ++i) {
                cv::Vec3b &pixel = frame.at<cv::Vec3b>(z, i);
                int cy = AC_GetFZ(frame.rows-1, z, size_y);
                if(cy >= 0 && cy < frame.rows && i >= 0 && i < frame.cols) {
                    cv::Vec3b &pix = collection.frames[offset].at<cv::Vec3b>(cy, i);
                    pixel = pix;
                }
            }
            size_y ++;
            if(size_y > frame.rows*2)
                size_y = frame.rows/16;
        }
        
        if(++offset > (MAX-1)) {
            offset = 0;
        }
    }

    flash = (flash == true) ? false : true;
}