/*
 * cl_geo_map_handler.h - CL geometry map handler
 *
 *  Copyright (c) 2016 Intel Corporation
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 * Author: Wind Yuan <feng.yuan@intel.com>
 */

#ifndef XCAM_CL_GEO_MAP_HANDLER_H
#define XCAM_CL_GEO_MAP_HANDLER_H

#include <xcam_std.h>
#include <ocl/cl_image_handler.h>

namespace XCam {

struct GeoPos {
    double x;
    double y;

    GeoPos () : x(0), y(0) {}
};

class CLGeoMapKernel;
class GeoKernelParamCallback
{
    friend class CLGeoMapKernel;

public:
    GeoKernelParamCallback () {}
    virtual ~GeoKernelParamCallback () {}

protected:
    virtual SmartPtr<CLImage> get_geo_input_image (NV12PlaneIdx index) = 0;
    virtual SmartPtr<CLImage> get_geo_output_image (NV12PlaneIdx index) = 0;
    virtual SmartPtr<CLImage> get_geo_map_table () = 0;
    virtual void get_geo_equivalent_out_size (float &width, float &height) = 0;
    virtual void get_geo_pixel_out_size (float &width, float &height) = 0;

    virtual SmartPtr<CLImage> get_lsc_table () = 0;
    virtual float* get_lsc_gray_threshold() = 0;

private:
    XCAM_DEAD_COPY (GeoKernelParamCallback);
};

class CLGeoMapHandler;
class CLGeoMapKernel
    : public CLImageKernel
{
public:
    explicit CLGeoMapKernel (
        const SmartPtr<CLContext> &context,
        const SmartPtr<GeoKernelParamCallback> handler,
        bool need_lsc);

protected:
    virtual XCamReturn prepare_arguments (CLArgList &args, CLWorkSize &work_size);

private:
    SmartPtr<GeoKernelParamCallback>   _handler;
    bool                               _need_lsc;
};

class CLGeoMapHandler
    : public CLImageHandler
    , public GeoKernelParamCallback
{
public:
    explicit CLGeoMapHandler (const SmartPtr<CLContext> &context);
    void set_output_size (uint32_t width, uint32_t height) {
        _output_width = width;
        _output_height = height;
    }
    void get_output_size (uint32_t &width, uint32_t &height) const {
        width = _output_width;
        height = _output_height;
    }

    bool set_map_data (GeoPos *data, uint32_t width, uint32_t height);
    bool set_map_uint (float uint_x, float uint_y);
    void get_map_uint (float &uint_x, float &uint_y) {
        uint_x = _uint_x;
        uint_y = _uint_y;
    }

protected:
    // derived from GeoKernelParamCallback
    virtual SmartPtr<CLImage> get_geo_input_image (NV12PlaneIdx index) {
        XCAM_ASSERT (index < NV12PlaneMax);
        return _input [index];
    }
    virtual SmartPtr<CLImage> get_geo_output_image (NV12PlaneIdx index) {
        XCAM_ASSERT (index < NV12PlaneMax);
        return _output [index];
    }
    virtual SmartPtr<CLImage> get_geo_map_table () {
        XCAM_ASSERT (_geo_image.ptr ());
        return _geo_image;
    }
    virtual void get_geo_equivalent_out_size (float &width, float &height);
    virtual void get_geo_pixel_out_size (float &width, float &height);

    virtual SmartPtr<CLImage> get_lsc_table () {
        XCAM_ASSERT (false && "CLGeoMapHandler::lsc table is not supported");
        return NULL;
    }
    virtual float* get_lsc_gray_threshold () {
        XCAM_ASSERT (false && "CLGeoMapHandler::lsc gray threshold is not supported");
        return NULL;
    }

protected:
    virtual XCamReturn prepare_buffer_pool_video_info (
        const VideoBufferInfo &input,
        VideoBufferInfo &output);
    virtual XCamReturn prepare_parameters (SmartPtr<VideoBuffer> &input, SmartPtr<VideoBuffer> &output);
    virtual XCamReturn execute_done (SmartPtr<VideoBuffer> &output);

private:
    bool normalize_geo_map (uint32_t image_w, uint32_t image_h);
    bool check_geo_map_buf (uint32_t width, uint32_t height);

    XCAM_DEAD_COPY (CLGeoMapHandler);

private:
    uint32_t                         _output_width;
    uint32_t                         _output_height;
    uint32_t                         _map_width, _map_height;
    uint32_t                         _map_aligned_width;
    float                            _uint_x, _uint_y;
    SmartPtr<CLImage>                _input[NV12PlaneMax];
    SmartPtr<CLImage>                _output[NV12PlaneMax];
    SmartPtr<CLBuffer>               _geo_map;
    SmartPtr<CLImage>                _geo_image;
    bool                             _geo_map_normalized;
};

SmartPtr<CLImageKernel>
create_geo_map_kernel (
    const SmartPtr<CLContext> &context, SmartPtr<GeoKernelParamCallback> param_cb, bool need_lsc);

SmartPtr<CLImageHandler>
create_geo_map_handler (const SmartPtr<CLContext> &context, bool need_lsc = false);

}

#endif //XCAM_CL_GEO_MAP_HANDLER_H