/************************************************************************

* Copyright(c) 2017  ZRJ

* All rights reserved.

*

* File:    video2frames.cpp

* Brief: 批量解帧视频文件

* Version: 1.0

* Author: ZRJ

* Email: happyamyhope@163.com

* Date:    2017/05/10

* History:

* 20170510: 批量解帧视频文件;

************************************************************************/

//-------------------------------------------------------------------------

//头文件

#include<highgui.h>

#include<cv.h>

#include<io.h>

#include<iostream>

#include<fstream>

#include <direct.h>//mkdir

using namespace std;

using namespace cv;

//调参

string filePath = "E:\\carriage_recognition\\video\\2017-05-02";

char picfilename[];

char save_path[];

//函数声明

void getFiles(string path, vector<string>& files)

{

    //文件句柄

    long   hFile = ;

    //文件信息

    struct _finddata_t fileinfo;

    string p;

    if ((hFile = (long)_findfirst(p.assign(path).append("\\*").c_str(), &fileinfo)) != -)

    {

        do

        {

            //如果是目录,迭代之

            //如果不是,加入列表

            if ((fileinfo.attrib &  _A_SUBDIR))

            {

                if (strcmp(fileinfo.name, ".") !=  && strcmp(fileinfo.name, "..") != )

                    getFiles(p.assign(path).append("\\").append(fileinfo.name), files);

            }

            else

            {

                files.push_back(p.assign(path).append("\\").append(fileinfo.name));

            }

        } while (_findnext(hFile, &fileinfo) == );

        _findclose(hFile);

    }

}

void mpf2frame(const char* videoPath, char* picfilename)

{

    CvCapture* capture = cvCaptureFromFile(videoPath);

    int fps = (int)cvGetCaptureProperty(capture, CV_CAP_PROP_FPS);

    long nFrame = (long)cvGetCaptureProperty(capture, CV_CAP_PROP_FRAME_COUNT); // 获取总帧数

    int width = (int)cvGetCaptureProperty(capture, CV_CAP_PROP_FRAME_WIDTH);

    int height = (int)cvGetCaptureProperty(capture, CV_CAP_PROP_FRAME_HEIGHT);

    IplImage* image = NULL;

    int i = ;

    if (capture)

    {

        while ()

        {

            image = cvQueryFrame(capture);

            if (image)

            {

                i++;

                if (i %  ==  && i <= nFrame )

                {

                    sprintf(save_path, "%s\\%d.jpg", picfilename, i);

                    cvSaveImage(save_path, image);

                }

                else if (i < nFrame)

                {

                }

                else if (i >= nFrame)

                {

                    return;

                }

            }

        }

    }

}

int main()

{

    ofstream fout("video_list.txt");

    vector<string> files;

    //获取该路径下的所有文件

    getFiles(filePath, files);

    int size = (int)files.size();

    for (int i = ; i < size; i++)

    {

        fout << files[i].c_str() << endl;

        cout << i << endl;

        const char* filename = files[i].c_str();

        sprintf(picfilename, "%s\\frame", filePath.c_str());

        _mkdir(picfilename);

        sprintf(picfilename, "%s\\%d", picfilename, i);

        _mkdir(picfilename);

        mpf2frame(filename, picfilename);

    }

    fout.close();

    return ;

}