Method GetCVKernelBody

compiler/lib/KernelGen/GeneralIntrinsic/CvtColor.cpp:470–504

}

std::string CvtColorKernel::GetCVKernelBody(TContext* context) const {
    auto kernel_sig = GetCVKernelSignature(context);
    auto mode = context->getAttrStr("mode");
    auto gen_func = g_cvt_reg.get_func(mode);
    auto code = gen_func(context);
    std::stringstream writer;
    writer << R"(
        #include <limits.h>
        #include <string.h>
        #include "gi_int.h"
        #include "tinycv_c.h"
    )";

    std::string body_temp = R"(
        ${aux_func}
        void ${kernel_sig}{
            uint8_t * sptr = src->data;
            uint8_t * dptr = dst->data;
            size_t rows = src->rows;
            size_t cols = src->cols;
            size_t hw = rows * cols;
            size_t ch = src->channels;
            
            ${cvt_code}
            
        }
    )";

    writer << StringTemplate::StringTemplateArgs()
                      .add("aux_func", code.aux_func)
                      .add("cvt_code", code.cvt_code)
                      .add("kernel_sig", kernel_sig)
                      .render(body_temp);
    return writer.str();
}

// vim: syntax=cpp.doxygen