Morphological Operations: Dilation Solution

Morphological Operations: Dilation Solution Hackerrank

Morphological Operations: Dilation.cpp

@@ -0,0 +1,70 @@
+/*
+    RAMESH CHANDRA
+*/
+#include<bits/stdc++.h>
+using namespace std;
+
+void printImage(vector<string>&image){
+    for(int i=0;i<image.size();i++)
+        cout<<image[i]<<endl;
+}
+void dilatedPixel(pair<int,int> imagePixel,vector<string>&dilatedImage,vector<string>&mask,pair<int,int>centerOfMask){
+    int N = dilatedImage.size();
+    int M = dilatedImage[0].length();
+    int deltaI = centerOfMask.first;
+    int deltaJ = centerOfMask.second;
+
+    for(int i=imagePixel.first-deltaI,mask_i=0;mask_i<mask.size();mask_i++,i++){
+        for(int j=imagePixel.second-deltaJ,mask_j=0;mask_j<mask[mask_i].length();mask_j++,j++){
+            if(mask[mask_i][mask_j]=='0') //no operation required....as mask contain 0
+                continue;
+            if(i<0 || i>= N || j<0 || j>=M) //out of image
+                continue;
+
+            if(dilatedImage[i][j]=='0') //pixel contain atleast one zero where mask contain 1
+               dilatedImage[i][j]='1';
+        }
+    }
+}
+vector<string> dilation(vector<string>&image,vector<string>&mask,pair<int,int>centerOfMask){
+    vector<string> dilatedImage = image;
+
+    for(int i=0;i<image.size();i++){
+        for(int j=0;j<image[i].length();j++){
+
+            if(image[i][j]=='0')  //don't need to any thing
+                continue;
+            //check for this pixel
+            dilatedPixel({i,j},dilatedImage,mask,centerOfMask);
+        }
+    }
+    return dilatedImage;
+}
+int main() {
+
+    vector<string> image =  {
+                                "0000000000",
+                                "0111111100",
+                                "0000111100",
+                                "0000111100",
+                                "0001111100",
+                                "0000111100",
+                                "0001100000",
+                                "0000000000",
+                                "0000000000"
+                            };
+    vector<string> mask = {
+                            "111",
+                            "111",
+                            "111"
+                           };
+    //printImage(image);
+    vector<string> dilatedImage = dilation(image,mask,{1,1});
+    printImage(dilatedImage);
+    int count_one = 0;
+    for(int i=0;i<dilatedImage.size();i++)
+        for(int j=0;j<dilatedImage[i].length();j++)
+            count_one+=(dilatedImage[i][j]=='1');
+    cout<<count_one<<endl;
+    return 0;
+}