opencv_videoio.cc

/*
 +----------------------------------------------------------------------+
 | PHP-OpenCV                                                           |
 +----------------------------------------------------------------------+
 | This source file is subject to version 2.0 of the Apache license,    |
 | that is bundled with this package in the file LICENSE, and is        |
 | available through the world-wide-web at the following url:           |
 | http://www.apache.org/licenses/LICENSE-2.0.html                      |
 | If you did not receive a copy of the Apache2.0 license and are unable|
 | to obtain it through the world-wide-web, please send a note to       |
 | hihozhou@gmail.com so we can mail you a copy immediately.            |
 +----------------------------------------------------------------------+
 | Author: HaiHao Zhou  <hihozhou@gmail.com>                            |
 +----------------------------------------------------------------------+
 */
#include "../../php_opencv.h"
#include "../../opencv_exception.h"
#include "core/opencv_mat.h"
#include "opencv_videoio.h"

using namespace cv;

zend_class_entry *opencv_video_capture_ce;

zend_object_handlers opencv_video_capture_object_handlers;

zend_object* opencv_video_capture_create_handler(zend_class_entry *type)
{
    size_t size = sizeof(opencv_video_capture_object);
    opencv_video_capture_object *obj = (opencv_video_capture_object *)ecalloc(1,size);
    memset(obj, 0, sizeof(opencv_video_capture_object));
    zend_object_std_init(&obj->std, type);
    object_properties_init(&obj->std, type);
    obj->std.ce = type;
    obj->std.handlers = &opencv_video_capture_object_handlers;
    return &obj->std;
}

PHP_METHOD(opencv_video_capture, __construct)
{
    zval *zval1 = NULL;
    if (zend_parse_parameters(ZEND_NUM_ARGS(), "|z", &zval1) == FAILURE) {
        RETURN_NULL();
    }
    opencv_video_capture_object *obj = Z_PHP_VIDEO_CAPTURE_P(getThis());
    char *error_message;
    VideoCapture *videoCapture;
    if(zval1 == NULL){
        videoCapture = new VideoCapture();
    }else{
        again:
        switch (Z_TYPE_P(zval1)) {
            case IS_LONG:
                try{
                    int long_val = (int)zval_get_long(zval1);
                    videoCapture = new VideoCapture(long_val);
                }catch (Exception e){
                    opencv_throw_exception(e.what());
                    RETURN_NULL();
                }
                break;
            case IS_DOUBLE:
                try{
                    int double_val = (int)zval_get_double(zval1);
                    videoCapture = new VideoCapture(double_val);
                }catch (Exception e){
                    opencv_throw_exception(e.what());
                    RETURN_NULL();
                }
                break;
            case IS_STRING:
                try{
                    String string_val=(String)ZSTR_VAL(zval_get_string(zval1));
                    videoCapture = new VideoCapture(string_val);
                }catch (Exception e){
                    opencv_throw_exception(e.what());
                    RETURN_NULL();
                }
                break;
            case IS_REFERENCE:
                zval1 = Z_REFVAL_P(zval1);
                goto again;
                break;
            default:
                opencv_throw_exception("Can't write file on unknow type.");
                RETURN_NULL();
                break;
        }
    }

    obj->videoCapture = videoCapture;
}

/**
 * 打开视频，当为参数为整形时候，打开指定设备，当是字符串时候，打开目录的文件
 * @param execute_data
 * @param return_value
 */
PHP_METHOD(opencv_video_capture, open)
{
    zval *zval1 = NULL;
    if (zend_parse_parameters(ZEND_NUM_ARGS(), "z", &zval1) == FAILURE) {
        RETURN_NULL();
    }
    opencv_video_capture_object *obj = Z_PHP_VIDEO_CAPTURE_P(getThis());
    bool return_bool;
    again:
    switch (Z_TYPE_P(zval1)) {
        case IS_LONG:
            try{
                int long_val = (int)zval_get_long(zval1);
                return_bool = obj->videoCapture->open(long_val);
            }catch (Exception e){
                opencv_throw_exception(e.what());
                RETURN_NULL();
            }
            break;
        case IS_DOUBLE:
            try{
                int double_val = (int)zval_get_double(zval1);
                return_bool = obj->videoCapture->open(double_val);
            }catch (Exception e){
                opencv_throw_exception(e.what());
                RETURN_NULL();
            }
            break;
        case IS_STRING:
            try{
                String string_val=(String)ZSTR_VAL(zval_get_string(zval1));
                return_bool = obj->videoCapture->open(string_val);
            }catch (Exception e){
                opencv_throw_exception(e.what());
                RETURN_NULL();
            }
            break;
        case IS_REFERENCE:
            //引用类型，获取引用真是的value，再判断一次
            zval1 = Z_REFVAL_P(zval1);
            goto again;
            break;
        default:
            opencv_throw_exception("Can't write file on unknow type.");
            RETURN_NULL();
            break;
    }
    RETURN_BOOL(return_bool);
}


PHP_METHOD(opencv_video_capture, is_opened)
{
    opencv_video_capture_object *obj = Z_PHP_VIDEO_CAPTURE_P(getThis());
    bool is_opend = obj->videoCapture->isOpened();
    RETURN_BOOL(is_opend);
}

ZEND_BEGIN_ARG_INFO_EX(opencv_video_capture_read_arginfo, 0, 0, 1)
                ZEND_ARG_INFO(1, mat)
ZEND_END_ARG_INFO()

PHP_METHOD(opencv_video_capture, read)
{
    zval *mat_zval = NULL;
    if (zend_parse_parameters(ZEND_NUM_ARGS(), "z", &mat_zval) == FAILURE) {
        RETURN_NULL();
    }

    zval *mat_real_zval = Z_REFVAL_P(mat_zval);
    opencv_mat_object *real_object;
    Mat mat;
    if(Z_TYPE_P(mat_real_zval) == IS_OBJECT && Z_OBJCE_P(mat_real_zval)==opencv_mat_ce){
        real_object = Z_PHP_MAT_OBJ_P(mat_real_zval);
    }else{

        zval_ptr_dtor(mat_real_zval);
        zval instance;
        object_init_ex(&instance,opencv_mat_ce);
        ZVAL_COPY_VALUE(mat_real_zval, &instance);// Cover dst_real_zval by Mat object
        real_object = Z_PHP_MAT_OBJ_P(mat_real_zval);
    }
    opencv_video_capture_object *this_object = Z_PHP_VIDEO_CAPTURE_P(getThis());
    bool result = this_object->videoCapture->read(mat);
    if(result){
        real_object->mat = new Mat(mat);
        opencv_mat_update_property_by_c_mat(mat_real_zval, real_object->mat);
    }
    RETURN_BOOL(result);
}

/**
 * opencv_video_capture_methods[]
 */
const zend_function_entry opencv_video_capture_methods[] = {
        PHP_ME(opencv_video_capture, __construct, NULL, ZEND_ACC_PUBLIC|ZEND_ACC_CTOR)
        PHP_ME(opencv_video_capture, open, NULL, ZEND_ACC_PUBLIC)
        PHP_MALIAS(opencv_video_capture, isOpened ,is_opened, NULL, ZEND_ACC_PUBLIC)
        PHP_ME(opencv_video_capture, read, opencv_video_capture_read_arginfo, ZEND_ACC_PUBLIC)
        PHP_FE_END
};
/* }}} */

void opencv_video_capture_free_obj(zend_object *object)
{
    opencv_video_capture_object *obj;
    obj = get_video_capture_obj(object);
    delete obj->videoCapture;
    zend_object_std_dtor(object);
}

void opencv_video_capture_init(int module_number){
    zend_class_entry ce;
    INIT_NS_CLASS_ENTRY(ce,OPENCV_NS, "VideoCapture", opencv_video_capture_methods);
    opencv_video_capture_ce = zend_register_internal_class(&ce);

    opencv_video_capture_ce->create_object = opencv_video_capture_create_handler;
    memcpy(&opencv_video_capture_object_handlers,
           zend_get_std_object_handlers(), sizeof(zend_object_handlers));
    opencv_video_capture_object_handlers.clone_obj = NULL;
    opencv_video_capture_object_handlers.free_obj = opencv_video_capture_free_obj;
}

void opencv_videoio_init(int module_number){
    opencv_video_capture_init(module_number);
}