Subversion Repositories seema-scanner

#include <iostream>
#include <unistd.h>
#include "Codec.h"
#include "CodecGrayCode.h"
#include "CodecPhaseShift.h"

#include "Projector.h"

#include <opencv2/opencv.hpp>

int main(){

    Projector projector(1);
    unsigned int screenResX, screenResY;
    projector.getScreenRes(&screenResX, &screenResY);
    //unsigned int screenResX = 608, screenResY = 684;

    bool diamondPattern = true;

    // Codec for horizontal direction
    CodecGrayCode codecHorz(screenResX, screenResY, diamondPattern);

    // Codec for vertical direction (slight misuse of code)
    CodecGrayCode codecVert(screenResY, screenResX, diamondPattern);

    int NHorz = codecHorz.getNPatterns();
    int NVert = codecVert.getNPatterns();

    cv::namedWindow("Pattern");

    for(int i=0; i<NHorz; i++){
        cv::Mat pattern = codecHorz.getEncodingPattern(i);
        pattern = cv::repeat(pattern, screenResY/pattern.rows, screenResX/pattern.cols);
        cv::imshow("Pattern", pattern);
        projector.displayTexture(pattern.ptr(), pattern.cols, pattern.rows);
        // Treat pattern as a pseudo captured frame
        std::vector<cv::Mat> channels;
        cv::split(pattern, channels);
        codecHorz.setFrame(i, channels[0]);
        std::cout << "Horizontal Pattern " << i+1 << "/" << NHorz << std::endl << std::flush;
        cv::waitKey(10);
    }
    for(int i=0; i<NVert; i++){
        cv::Mat pattern = codecVert.getEncodingPattern(i);
        pattern = pattern.t();
        pattern = cv::repeat(pattern, screenResY/pattern.rows, screenResX/pattern.cols);
        cv::imshow("Pattern", pattern);
        projector.displayTexture(pattern.ptr(), pattern.cols, pattern.rows);
        // Treat pattern as a pseudo captured frame
        std::vector<cv::Mat> channels;
        cv::split(pattern, channels);
        codecVert.setFrame(i, channels[0]);
        std::cout << "Horizontal Pattern " << i+1 << "/" << NVert << std::endl << std::flush;
        cv::waitKey(10);
    }

    // Reconstruct up image
    cv::Mat up, mask, shading;
    codecHorz.decodeFrames(up, mask, shading);

    // Reconstruct vp image
    cv::Mat vp;
    codecVert.decodeFrames(vp, mask, shading);

    cv::namedWindow("up");
    cv::imshow("up", up/screenResX);
    cv::namedWindow("vp");
    cv::imshow("vp", vp/screenResY);
    cv::namedWindow("mask");
    cv::imshow("mask", mask);
    cv::namedWindow("shading");
    cv::imshow("shading", shading);

    cv::waitKey(0);
}