diff options
Diffstat (limited to 'sci_gateway1/cpp/opencv_VideoReader.cpp')
-rw-r--r-- | sci_gateway1/cpp/opencv_VideoReader.cpp | 389 |
1 files changed, 389 insertions, 0 deletions
diff --git a/sci_gateway1/cpp/opencv_VideoReader.cpp b/sci_gateway1/cpp/opencv_VideoReader.cpp new file mode 100644 index 0000000..51207bf --- /dev/null +++ b/sci_gateway1/cpp/opencv_VideoReader.cpp @@ -0,0 +1,389 @@ +/*************************************************** +Author : Tanmay Chaudhari, Shashank Shekhar +**************************************************/ + +#include <numeric> +#include "opencv2/core/core.hpp" +#include "opencv2/highgui/highgui.hpp" +#include "opencv2/opencv.hpp" +#include <iostream> +using namespace cv; +using namespace std; +extern "C" +{ + #include "api_scilab.h" + #include "Scierror.h" + #include "BOOL.h" + #include <localization.h> + #include "sciprint.h" + #include "../common.h" + //#include "../common.cpp" + + Ptr<VideoCapture> capture; + double videoCurrentTime = 0; + double videoDuration = 0; + int videoFPS; + char *videoFileName = NULL; + bool videoObjectSt = false; + int opencv_VideoReader(char *fname, unsigned long fname_len) + { + + SciErr sciErr; + int *piAddr = NULL; + int *piAddrVal = NULL; + int *piLen = NULL; + int iRows, iCols = 0; + char ** pstData1 = NULL; + char ** pstData2 = NULL; + double frames; + double width; + double height; + double CT; + char full_path[PATH_MAX]; + bool stillLeft = false; + CheckInputArgument(pvApiCtx,1,3); + CheckOutputArgument(pvApiCtx,1,10); + int noOfarguments = *getNbInputArgument(pvApiCtx); + if(noOfarguments==2) + { + Scierror(999,"Invalid number of Arguments\n"); + return 0; + } + + sciErr = getVarAddressFromPosition(pvApiCtx,1,&piAddr); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + if(!isStringType(pvApiCtx, piAddr)) + { + Scierror(999,"Invalid Argument\n"); + return 0; + } + //first call to get rows and columns + sciErr = getMatrixOfString(pvApiCtx, piAddr, &iRows, &iCols, NULL, NULL); + if(sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + piLen = (int*)malloc(sizeof(int) * iRows * iCols); + //second call to retrieve length of each string + sciErr = getMatrixOfString(pvApiCtx, piAddr, &iRows, &iCols, piLen, NULL); + if(sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + pstData1 = (char**)malloc(sizeof(char*) * iRows * iCols); + for(int j = 0 ; j < iRows * iCols ; j++) + { + pstData1[j] = (char*)malloc(sizeof(char) * (piLen[j] + 1));//+ 1 for null termination + } + //third call to retrieve data + sciErr = getMatrixOfString(pvApiCtx, piAddr, &iRows, &iCols, piLen, pstData1); + if(sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + if(noOfarguments>1) + { + sciErr = getVarAddressFromPosition(pvApiCtx,2,&piAddr); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + if(!isStringType(pvApiCtx, piAddr)) + { + Scierror(999,"Invalid Argument\n"); + return 0; + } + //first call to get rows and columns + sciErr = getMatrixOfString(pvApiCtx, piAddr, &iRows, &iCols, NULL, NULL); + if(sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + piLen = (int*)malloc(sizeof(int) * iRows * iCols); + //second call to retrieve length of each string + sciErr = getMatrixOfString(pvApiCtx, piAddr, &iRows, &iCols, piLen, NULL); + if(sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + pstData2 = (char**)malloc(sizeof(char*) * iRows * iCols); + for(int j = 0 ; j < iRows * iCols ; j++) + { + pstData2[j] = (char*)malloc(sizeof(char) * (piLen[j] + 1));//+ 1 for null termination + } + //third call to retrieve data + sciErr = getMatrixOfString(pvApiCtx, piAddr, &iRows, &iCols, piLen, pstData2); + if(sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + if(strcmp(pstData2[0], "CurrentTime")==0) + { + sciErr = getVarAddressFromPosition(pvApiCtx,3,&piAddrVal); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + if(!isDoubleType(pvApiCtx, piAddrVal) || isVarComplex(pvApiCtx,piAddrVal) || !isScalar(pvApiCtx, piAddrVal)) + { + + Scierror(999," Invalid Value for Current Time.\n"); + return 0; + } + getScalarDouble(pvApiCtx, piAddrVal, &CT); + if(CT<0) + { + Scierror(999,"Invalid Input for CurrentTime.\n"); + return 0; + } + CT*=1000; + } + else + { + Scierror(999,"Invalid Argument\n"); + return 0; + } + } + + if(videoObjectSt==true && strcmp(videoFileName, pstData1[0])==0) + { + if(int(videoCurrentTime) >= int(videoDuration)) + { + stillLeft = false; + } + else if(int(videoCurrentTime) < int(videoDuration)) + stillLeft = true; + } + + if(!stillLeft) + { + capture = new VideoCapture(pstData1[0]); + if(!capture->isOpened()) + { + Scierror(999,"Could not open file\n"); + return 0; + } + } + + videoFileName = pstData1[0]; + videoFPS = capture->get(CV_CAP_PROP_FPS); + frames = capture->get(CV_CAP_PROP_FRAME_COUNT); + + if(noOfarguments!=1) + { + if(int(CT) >= int(videoDuration)) + { + Scierror(999,"Current Time should be less than Video Duration\n"); + return 0; + } + capture->set(CV_CAP_PROP_POS_MSEC, CT); + } + + videoObjectSt = true; + videoCurrentTime = capture->get(CV_CAP_PROP_POS_MSEC); + videoDuration = (1000/double(videoFPS))*double(frames); + width = capture->get(CV_CAP_PROP_FRAME_WIDTH); + height = capture->get(CV_CAP_PROP_FRAME_HEIGHT); + realpath(pstData1[0], full_path); + + createSingleString(pvApiCtx, nbInputArgument(pvApiCtx) + 1, pstData1[0]); + createScalarDouble(pvApiCtx,nbInputArgument(pvApiCtx) + 2, videoDuration); + createScalarDouble(pvApiCtx,nbInputArgument(pvApiCtx) + 3, videoFPS); + createScalarDouble(pvApiCtx, nbInputArgument(pvApiCtx)+4, videoCurrentTime); + createScalarDouble(pvApiCtx,nbInputArgument(pvApiCtx) + 5, frames); + createScalarDouble(pvApiCtx,nbInputArgument(pvApiCtx) + 6, width); + createScalarDouble(pvApiCtx,nbInputArgument(pvApiCtx) + 7, height); + createSingleString(pvApiCtx, nbInputArgument(pvApiCtx)+8, full_path); + + for(int i=0;i<=8;i++) + { + AssignOutputVariable(pvApiCtx, i) = nbInputArgument(pvApiCtx) + i; + } + + ReturnArguments(pvApiCtx); + return 0; + } + + int has_frame(char *fname, unsigned long fname_len) + { + SciErr sciErr; + + if(videoFileName==NULL) + { + Scierror(999,"No video has been read yet.\n"); + return 0; + } + + int *piAddr = NULL; + int *piLen = NULL; + int iRows, iCols = 0; + char ** pstData = NULL; + + CheckInputArgument(pvApiCtx,0,0); + CheckOutputArgument(pvApiCtx,0,1); + + if(videoObjectSt == false) + { + Scierror(999,"Invalid Video File. You need to read the video using VideoReader.\n"); + return 0; + } + + if(videoCurrentTime>videoDuration) + createScalarBoolean(pvApiCtx,nbInputArgument(pvApiCtx) + 1, false); + else if(int(videoCurrentTime) == int(videoDuration)) + createScalarBoolean(pvApiCtx,nbInputArgument(pvApiCtx) + 1, false); + else + createScalarBoolean(pvApiCtx,nbInputArgument(pvApiCtx) + 1, true); + + AssignOutputVariable(pvApiCtx, 1) = nbInputArgument(pvApiCtx) + 1; + ReturnArguments(pvApiCtx); + + return 0; + } + + int read_frame(char *fname, unsigned long fname_len) + { + SciErr sciErr; + + if(videoFileName==NULL) + { + Scierror(999,"No video has been read yet.\n"); + return 0; + } + + if(int(videoCurrentTime)>=int(videoDuration)) + { + Scierror(999,"No more frames left to be read\n"); + return 0; + } + //sciprint("Current Time of the Video is : %f\n", videoCurrentTime); + + int *piAddr = NULL; + int *piLen = NULL; + int iRows, iCols = 0; + char ** pstData = NULL; + int *singleFrameList = NULL; + unsigned char *red = NULL; + unsigned char *green = NULL; + unsigned char *blue = NULL; + + CheckInputArgument(pvApiCtx,0,0); + CheckOutputArgument(pvApiCtx,0,1); + + if(videoObjectSt == false) + { + Scierror(999,"Invalid Video File. You need to read the video using VideoReader.\n"); + return 0; + } + +//--------------------------------------- Processing Frame at the particular frame ----------------------------------- + + Mat frame; + capture->set(CV_CAP_PROP_POS_MSEC, videoCurrentTime); + if(!capture->read(frame)) + { + Scierror(999, "Can't read Frame"); + return 0; + } +//-------------------------------------------------------------------------------------------------------------------- + + if(frame.channels()==1) + { + + sciErr = createList(pvApiCtx, nbInputArgument(pvApiCtx) + 1, 1, &singleFrameList); + if(sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + red = (unsigned char *)malloc(sizeof(unsigned char)*frame.rows*frame.cols); + + for(int k=0;k<frame.rows;k++) + for(int p=0;p<frame.cols;p++) + red[k+frame.rows*p]=frame.at<uchar>(k, p); + + sciErr = createMatrixOfUnsignedInteger8InList(pvApiCtx, nbInputArgument(pvApiCtx) + 1, singleFrameList, 1, frame.rows, frame.cols, red); + if(sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + free(red); + } + else + { + + sciErr = createList(pvApiCtx, nbInputArgument(pvApiCtx) + 1, 3, &singleFrameList); + if(sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + red = (unsigned char *)malloc(sizeof(unsigned char)*frame.rows*frame.cols); + green = (unsigned char *)malloc(sizeof(unsigned char)*frame.rows*frame.cols); + blue = (unsigned char *)malloc(sizeof(unsigned char)*frame.rows*frame.cols); + + for(int k=0;k<frame.rows;k++) + { + for(int p=0;p<frame.cols;p++) + { + Vec3b intensity = frame.at<Vec3b>(k, p); + red[k+frame.rows*p]=intensity.val[2]; + green[k+frame.rows*p]=intensity.val[1]; + blue[k+frame.rows*p]=intensity.val[0]; + } + } + + sciErr = createMatrixOfUnsignedInteger8InList(pvApiCtx, nbInputArgument(pvApiCtx) + 1, singleFrameList, 1, frame.rows, frame.cols, red); + if(sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + sciErr = createMatrixOfUnsignedInteger8InList(pvApiCtx, nbInputArgument(pvApiCtx) + 1, singleFrameList, 2, frame.rows, frame.cols, green); + if(sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + sciErr = createMatrixOfUnsignedInteger8InList(pvApiCtx, nbInputArgument(pvApiCtx) + 1, singleFrameList, 3, frame.rows, frame.cols, blue); + if(sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + free(red); + free(green); + free(blue); + } + + videoCurrentTime+=(1000/double(videoFPS)); + AssignOutputVariable(pvApiCtx, 1) = nbInputArgument(pvApiCtx) + 1; + + ReturnArguments(pvApiCtx); + + return 0; + } +} + + + |