【开发日志】2023.05 ZENO----Image Processing----ImageExtractColor、ImageDelColor

EndlessDaydream

已于 2023-05-30 15:22:40 修改

阅读量126

点赞数 1

分类专栏：开发日志 ZENO 文章标签：笔记

于 2023-05-30 15:21:10 首次发布

本文链接：https://blog.csdn.net/Angelloveyatou/article/details/130871113

版权

ZENO 同时被 2 个专栏收录

52 篇文章 0 订阅

订阅专栏

开发日志

49 篇文章 0 订阅

订阅专栏

该代码定义了两个图像处理节点，ImageExtractColor和ImageDelColor，用于根据颜色阈值改变图像像素。它们接受输入的颜色阈值范围和背景类型（透明、黑色或白色），并相应地调整图像像素。如果背景设置为透明，节点会设置超出阈值范围的像素的透明度为0；如果背景是黑色或白色，则会将这些像素替换为对应的颜色。

摘要由CSDN通过智能技术生成

struct ImageExtractColor : INode {
    virtual void apply() override {
        auto image = get_input<PrimitiveObject>("image");
        auto background = get_input2<std::string>("background");
        auto &ud = image->userData();
        int w = ud.get2<int>("w");
        int h = ud.get2<int>("h");
        vec3f up = get_input2<vec3f>("high_threshold");
        vec3f low = get_input2<vec3f>("low_threshold");
        float upr = up[0]/255, upg = up[1]/255, upb = up[2]/255;
        float lr = low[0]/255,lg = low[1]/255,lb = low[2]/255;
        zeno::log_info("up:{}, {}, {}",upr,upg,upb);
        zeno::log_info("low:{}, {}, {}",lr,lg,lb);
        if(background == "transparent"){
            if(!image->has_attr("alpha")){
                image->verts.add_attr<float>("alpha");
                for(int i = 0; i < image->verts.size();i++){
                    image->verts.attr<float>("alpha")[i] = 1;
                }
            }
            for (auto i = 0; i < image->verts.size(); i++) {
                if(((upr < image->verts[i][0]) || (image->verts[i][0] < lr)) ||
                   ((upg < image->verts[i][1]) || (image->verts[i][1] < lg)) ||
                   ((upb < image->verts[i][2]) || (image->verts[i][2] < lb))){
                    image->verts.attr<float>("alpha")[i] = 0;
                }
            }
        }
        else if(background == "black"){
            for (auto i = 0; i < image->verts.size(); i++) {
                if(((upr < image->verts[i][0]) || (image->verts[i][0] < lr)) ||
                   ((upg < image->verts[i][1]) || (image->verts[i][1] < lg)) ||
                   ((upb < image->verts[i][2]) || (image->verts[i][2] < lb))){
                    image->verts[i] = {0,0,0};

                }
            }
        }
        else if(background == "white"){
            for (auto i = 0; i < image->verts.size(); i++) {
                if(((upr < image->verts[i][0]) || (image->verts[i][0] < lr)) ||
                   ((upg < image->verts[i][1]) || (image->verts[i][1] < lg)) ||
                   ((upb < image->verts[i][2]) || (image->verts[i][2] < lb))){
                    image->verts[i] = {1,1,1};
                }
            }
        }
        set_output("image", image);
    }
};
ZENDEFNODE(ImageExtractColor, {
    {
        {"image"},
        {"vec3f", "high_threshold", "255,255,255"},
        {"vec3f", "low_threshold", "0,0,0"},
        {"enum transparent black white", "background", "transparent"},
    },
    {
        {"image"},
    },
    {},
    { "image" },
});

struct ImageDelColor: INode {
    void apply() override {
        auto image = get_input<PrimitiveObject>("image");
        auto background = get_input2<std::string>("background");
        auto &ud = image->userData();
        int w = ud.get2<int>("w");
        int h = ud.get2<int>("h");
        vec3f up = get_input2<vec3f>("high_threshold");
        vec3f low = get_input2<vec3f>("low_threshold");
        float upr = up[0]/255, upg = up[1]/255, upb = up[2]/255;
        float lr = low[0]/255,lg = low[1]/255,lb = low[2]/255;
        if(background == "transparent"){
            if(!image->has_attr("alpha")){
                image->verts.add_attr<float>("alpha");
                for(int i = 0; i < image->verts.size();i++){
                    image->verts.attr<float>("alpha")[i] = 1;
                }
            }
            for (auto i = 0; i < image->verts.size(); i++) {
                if(((lr <= image->verts[i][0]) && (image->verts[i][0] <= upr)) &&
                   ((lg <= image->verts[i][1]) && (image->verts[i][1] <= upg)) &&
                   ((lb <= image->verts[i][2]) && (image->verts[i][2] <= upb))){
                    image->verts.attr<float>("alpha")[i] = 0;
                }
            }
        }
        else if(background == "black"){
            for (auto i = 0; i < image->verts.size(); i++) {
                if(((lr <= image->verts[i][0]) && (image->verts[i][0] <= upr)) &&
                   ((lg <= image->verts[i][1]) && (image->verts[i][1] <= upg)) &&
                   ((lb <= image->verts[i][2]) && (image->verts[i][2] <= upb))){
                    image->verts[i] = {0,0,0};
                }
            }
        }
        else if(background == "white"){
            for (auto i = 0; i < image->verts.size(); i++) {
                if(((lr <= image->verts[i][0]) && (image->verts[i][0] <= upr)) &&
                   ((lg <= image->verts[i][1]) && (image->verts[i][1] <= upg)) &&
                   ((lb <= image->verts[i][2]) && (image->verts[i][2] <= upb))){
                    image->verts[i] = {1,1,1};
                }
            }
        }
        set_output("image", image);
    }
};

ZENDEFNODE(ImageDelColor, {
    {
        {"image"},
        {"vec3f", "high_threshold", "255,255,255"},
        {"vec3f", "low_threshold", "0,0,0"},
        {"enum transparent black white", "background", "transparent"},
    },
    {
        {"image"},
    },
    {},
    {"image"},
});

EndlessDaydream

关注

1
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
【开发日志】2023.05 ZENO----Image Processing----ImageExtractColor、ImageDelColor

ZENO----Image Processing----ImageExtractColor、ImageDelColor
复制链接

扫一扫

专栏目录