【开发日志】2023.05 ZENO----Image Processing----ImageInRange

ImageInRange

 

 

 

 

 

 

 

 

 

 

struct ImageInRange : INode {
    virtual void apply() override {
        auto image = get_input<PrimitiveObject>("image");
        auto mode = get_input2<std::string>("mode");
        auto &ud = image->userData();
        int w = ud.get2<int>("w");
        int h = ud.get2<int>("h");
        vec3i lb = get_input2<vec3i>("low_threshold");
        vec3i ub = get_input2<vec3i>("high_threshold");

        auto A11 = std::make_shared<PrimitiveObject>();
        A11->verts.resize(image->size());
        A11->userData().set2("isImage", 1);
        A11->userData().set2("w", w);
        A11->userData().set2("h", h);
        A11->verts.add_attr<float>("alpha");
        for(int i = 0;i < image->size();i++){
            A11->verts.attr<float>("alpha")[i] = 1;
        }
        std::vector<float> &Alpha = A11->verts.attr<float>("alpha");
        if(image->verts.has_attr("alpha")){
            A11->verts.attr<float>("alpha") = image->verts.attr<float>("alpha");
        }

        if(mode == "Transparent"){
            for (int i = 0; i < h; i++) {
                for (int j = 0; j < w; j++) {
                    if(((lb[0] <= 255 * image->verts[i * w + j][0]) && (255 * image->verts[i * w + j][0] <= ub[0])) &&
                       ((lb[1] <= 255 * image->verts[i * w + j][1]) && (255 * image->verts[i * w + j][1] <= ub[1])) &&
                       ((lb[2] <= 255 * image->verts[i * w + j][2]) && (255 * image->verts[i * w + j][2] <= ub[2]))){
                        A11->verts[i * w + j] = image->verts[i * w + j];
                    }
                    else{

                        A11->verts.attr<float>("alpha")[i * w + j] = 0;
                    }
                }
            }
        }
        else if(mode == "Black"){
            for (int i = 0; i < h; i++) {
                for (int j = 0; j < w; j++) {
                    if(((lb[0] <= 255 * image->verts[i * w + j][0]) && (255 * image->verts[i * w + j][0] <= ub[0])) &&
                       ((lb[1] <= 255 * image->verts[i * w + j][1]) && (255 * image->verts[i * w + j][1] <= ub[1])) &&
                       ((lb[2] <= 255 * image->verts[i * w + j][2]) && (255 * image->verts[i * w + j][2] <= ub[2]))){
                        A11->verts[i * w + j] = image->verts[i * w + j];
                    }
                    else{
                        A11->verts[i * w + j] = {0,0,0};
                    }
                }
            }
        }
        else if(mode == "White"){
            for (int i = 0; i < h; i++) {
                for (int j = 0; j < w; j++) {
                    if(((lb[0] <= 255 * image->verts[i * w + j][0]) && (255 * image->verts[i * w + j][0] <= ub[0])) &&
                       ((lb[1] <= 255 * image->verts[i * w + j][1]) && (255 * image->verts[i * w + j][1] <= ub[1])) &&
                       ((lb[2] <= 255 * image->verts[i * w + j][2]) && (255 * image->verts[i * w + j][2] <= ub[2]))){
                        A11->verts[i * w + j] = image->verts[i * w + j];
                    }
                    else{
                        A11->verts[i * w + j] = {1,1,1};
                    }
                }
            }
        }
        set_output("image", A11);
    }
};
ZENDEFNODE(ImageInRange, {
    {
        {"image"},
        {"enum Transparent Black White", "mode", "Transparent"},
        {"vec3i", "high_threshold", "255,255,255"},
        {"vec3i", "low_threshold", "0,0,0"},
    },
    {
        {"image"},
    },
    {},
    { "comp" },
});

ImageInRange_black

 

 

 

 

 

 

struct ImageInRange : INode {
    virtual void apply() override {
        auto image = get_input<PrimitiveObject>("image");
        auto &ud = image->userData();
        int w = ud.get2<int>("w");
        int h = ud.get2<int>("h");
        vec3i lb = get_input2<vec3i>("low_threshold");
        vec3i ub = get_input2<vec3i>("high_threshold");

        for (int i = 0; i < h; i++) {
            for (int j = 0; j < w; j++) {
                if(((lb[0] <= 255 * image->verts[i * w + j][0]) && (255 * image->verts[i * w + j][0] <= ub[0])) &&
                ((lb[1] <= 255 * image->verts[i * w + j][1]) && (255 * image->verts[i * w + j][1] <= ub[1])) &&
                ((lb[2] <= 255 * image->verts[i * w + j][2]) && (255 * image->verts[i * w + j][2] <= ub[2]))){
                    image->verts[i * w + j] = {1,1,1};
                }
                else{
                    image->verts[i * w + j] = {0,0,0};
                }
            }
        }
        set_output("image", image);
    }
};

ZENDEFNODE(ImageInRange, {
    {
        {"image"},
        {"vec3i", "high_threshold", "255,255,255"},
        {"vec3i", "low_threshold", "0,0,0"},
    },
    {
        {"image"},
    },
    {},
    { "comp" },
});