mfc_Paint_6 - 8BitsCoding/RobotMentor GitHub Wiki

특정 영역을 선택하면 색 채우기 기능을 추가하고자 한다.

// Dlg.h
// OnBnClickedFillRadio()
m_draw_wnd.SetDrawType(FILL_MODE);

// LBUTTONDOWN
if(m_draw_type == FILL_MODE) {
    int w = m_image.GetWidth(), h = m_image.GetHeight();
    COLORREF check_color = m_image_dc.GetPixel(point);  // 마우스를 클릭한 지점의 색상
    if(check_color == m_brush_color) return;
    m_image_dc.SetPixel(point, m_brush_color);
    ChangeColor(check_color, point.x, point.y, w, h);
    Invalidate();
} else {
    m_is_clicked = 1;
    m_prev_point = point;

    SetCapture();
}

// ChangeColor(COLORREF a_check_color, int a_x, int a_y, int a_width, int a_height)
{
    if(a_y - 1 >= 0) {
        if(m_image_dc.GetPixel(a_x, a_y - 1) == a_check_color) {
            m_image_dc.GetPixel(a_x, a_y - 1, m_brush_color);
            ChangeColor(a_check_color, a_x, a_y - 1, a_width, a_height);
        }
    }

    if(a_x + 1 >= 0) {
        if(m_image_dc.GetPixel(a_x + 1, a_y) == a_check_color) {
            m_image_dc.GetPixel(a_x + 1, a_y, m_brush_color);
            ChangeColor(a_check_color, a_x + 1, a_y, a_width, a_height);
        }
    }

    if(a_y + 1 >= 0) {
        if(m_image_dc.GetPixel(a_x, a_y + 1) == a_check_color) {
            m_image_dc.GetPixel(a_x, a_y + 1, m_brush_color);
            ChangeColor(a_check_color, a_x, a_y + 1, a_width, a_height);
        }
    }

    if(a_x - 1 >= 0) {
        if(m_image_dc.GetPixel(a_x - 1, a_y) == a_check_color) {
            m_image_dc.GetPixel(a_x - 1, a_y, m_brush_color);
            ChangeColor(a_check_color, a_x - 1, a_y, a_width, a_height);
        }
    }
}

프로그램이 죽는 현상이 발생

재귀호출 중 스텍(디버그 모드 시 1MB) 오버플로가 난다.

1픽셀에 4바이트 * 1920 * 1080 FHD라면 ... 다른 방법을 고안해야한다.

---

스택 오버프로 때문에 재귀호출말고 반복문을 써보자.

// LBUTTONDOWN
if(m_draw_type == FILL_MODE) {
    int w = m_image.GetWidth(), h = m_image.GetHeight();
    COLORREF check_color = m_image_dc.GetPixel(point);  // 마우스를 클릭한 지점의 색상
    if(check_color == m_brush_color) return;
    FillData * p_pixel_data = new FillData[w*h];
    memset(p_pixel_data, 0, sizeof(FillData)*w*h);
    m_image_dc.SetPixel(point, m_brush_color);

    FillData * p, * p_start = p_pixel_data, *p_add = p_pixel_data;
    p_add->x = (short)point.x;
    p_add->y = (short)point.y;
    p_add++;

    while(p_start != p_add) {
        p = p_start;
        while(p < p_add) {
            if(p->is_completed == 0) {

                if(p->y - 1 >= 0) {
                    if(m_image_dc.GetPixel(p->x, p->y - 1) == check_color) {
                        m_image_dc.GetPixel(p->x, p->y - 1, m_brush_color);
                        p_add->x = p->x;
                        p_add->y = p->y - 1;
                        p_add++;
                    }
                }

                if(p->x + 1 >= 0) {
                    if(m_image_dc.GetPixel(p->x + 1, p->y) == check_color) {
                        m_image_dc.GetPixel(p->x + 1, p->y, m_brush_color);
                        p_add->x = p->x + 1;
                        p_add->y = p->y;
                        p_add++;
                    }
                }

                if(p->y + 1 >= 0) {
                    if(m_image_dc.GetPixel(p->x, p->y + 1) == check_color) {
                        m_image_dc.GetPixel(p->x, p->y + 1, m_brush_color);
                        p_add->x = p->x;
                        p_add->y = p->y + 1;
                        p_add++;
                    }
                }

                if(p->x - 1 >= 0) {
                    if(m_image_dc.GetPixel(p->x - 1, p->y) == a_check_color) {
                        m_image_dc.GetPixel(p->x - 1, p->y, m_brush_color);
                        p_add->x = p->x - 1;
                        p_add->y = p->y;
                        p_add++;
                    }
                }

                p->is_completed = 1;
            }
            p++;
        }

        while(p_start < p_add) {
            if(p_start->is_completed == 0) break;
            p_start++;
        }
    }

    // ChangeColor(check_color, point.x, point.y, w, h);

    delete[] p_pixel_data;
    Invalidate();
} else {
    m_is_clicked = 1;
    m_prev_point = point;

    SetCapture();
}

struct FillData
{
    char is_completed;
    short x, y;
};

단점은 속도가 매우 느리다...

---

속도를 개선해보자.

// LBUTTONDOWN
if(m_draw_type == FILL_MODE) {
    int w = m_image.GetWidth(), h = m_image.GetHeight();

    BITMAP bmp_info;
    GetObject((HBITMAP)m_image, sizeof(BITMAP), &bmp_info);
    unsigned int * p_image = (unsigned int *)m_image.GetBits(), *p_check;
    p_image += bmp_info.bwWidthBytes / 4 - 1;
    // 가장 마지막 점으로 이동

    unsigned int check_color = *(p_image - point.y * w - (w - point.x - 1));
    // 점 단위로 색을 얻는다.

    unsigned int new_color = ((m_brush_color << 16) & 0x00FF0000) | (m_brush_color & 0x0000FF00) | (m_brush_color >> 16 & 0x000000FF);
    // Red와 Blue의 색상을 변경
    // RGB 칼러와 CImage 컬러를 맞추는 과정

    // 그림을 읽으면 투명도가 0xFF
    // Create 함수로 이미지를 만들면 투명도가 0x00
    new_color = new_color | (check_color & 0xFF000000);
    
    *(p_image - point.y * w - (w - point.x - 1)) = new_color;

    if(check_color == new_color) return;

    // COLORREF check_color = m_image_dc.GetPixel(point);  // 마우스를 클릭한 지점의 색상
    // if(check_color == m_brush_color) return;

    FillData * p_pixel_data = new FillData[w*h];
    memset(p_pixel_data, 0, sizeof(FillData)*w*h);
    m_image_dc.SetPixel(point, m_brush_color);

    FillData * p, * p_start = p_pixel_data, *p_add = p_pixel_data;
    p_add->x = (short)point.x;
    p_add->y = (short)point.y;
    p_add++;

    while(p_start != p_add) {
        p = p_start;
        while(p < p_add) {
            if(p->is_completed == 0) {
                p_check - p_image - p->y * w - (w - p->x - 1);

                if(p->y - 1 >= 0) {
                    if(*(p_check + w) == check_color) {
                        *(p_check + w) = new_color;
                        p_add->x = p->x;
                        p_add->y = p->y - 1;
                        p_add++;
                    }
                }

                if(p->x + 1 >= 0) {
                    if(*(p_check - 1) == check_color) {
                        *(p_check - 1) = new_color;
                        p_add->x = p->x + 1;
                        p_add->y = p->y;
                        p_add++;
                    }
                }

                if(p->y + 1 >= 0) {
                    if(*(p_check - w) == check_color) {
                        *(p_check + w) = new_color;
                        p_add->x = p->x;
                        p_add->y = p->y + 1;
                        p_add++;
                    }
                }

                if(p->x + 1 >= 0) {
                    if(*(p_check - 1) == check_color) {
                        *(p_check - 1) = new_color;
                        p_add->x = p->x - 1;
                        p_add->y = p->y;
                        p_add++;
                    }
                }

                p->is_completed = 1;
            }
            p++;
        }

        while(p_start < p_add) {
            if(p_start->is_completed == 0) break;
            p_start++;
        }
    }

    // ChangeColor(check_color, point.x, point.y, w, h);

    delete[] p_pixel_data;
    Invalidate();
} else {
    m_is_clicked = 1;
    m_prev_point = point;

    SetCapture();
}

픽셀을 직접제어함으로서 속도를 확실하게 개선할 수 있다.