continued working on the pipes part

This commit is contained in:
ItsMAX0112 2024-11-04 03:28:41 +05:30
parent 0cc2998922
commit 04e67b26b7
2 changed files with 26 additions and 54 deletions

View File

@ -6,103 +6,83 @@
using namespace std; using namespace std;
struct image_t *S1_smoothen(struct image_t *input_image) float *S1_smoothen(struct image_t *input_image, float *smooth)
{ {
// cout << input_image->width << input_image->height << " " << input_image->image_pixels << endl;
int width = input_image->width; int width = input_image->width;
int height = input_image->height; int height = input_image->height;
// memory allocation long ind = 0;
struct image_t *smoother = new struct image_t;
smoother->height = height;
smoother->width = width;
smoother->image_pixels = new uint8_t **[height];
for (int i = 0; i < height; i++) for (int i = 0; i < height; i++)
{ {
smoother->image_pixels[i] = new uint8_t *[width];
for (int j = 0; j < width; j++) for (int j = 0; j < width; j++)
smoother->image_pixels[i][j] = new uint8_t[3];
}
for (int i = 1; i < height - 1; i++)
{
for (int j = 1; j < width - 1; j++)
{ {
for (int k = 0; k < 3; k++) for (int k = 0; k < 3; k++)
{ {
// image->image_pixels[i][j][k] = val;
// edge cases // edge cases
if (i == 0 && j == 0) if (i == 0 && j == 0)
{ // i-1 and j - 1 doesnt exist { // i-1 and j - 1 doesnt exist
smoother->image_pixels[i][j][k] = (input_image->image_pixels[i][j][k] / 4 + input_image->image_pixels[i][j + 1][k] / 4 + input_image->image_pixels[i + 1][j][k] / 4 + input_image->image_pixels[i + 1][j + 1][k] / 4); smooth[ind] = (float)(input_image->image_pixels[i][j][k] / 4 + input_image->image_pixels[i][j + 1][k] / 4 + input_image->image_pixels[i + 1][j][k] / 4 + input_image->image_pixels[i + 1][j + 1][k] / 4);
} }
else if (i == height - 1 && j == 0) else if (i == height - 1 && j == 0)
{ // i+1 and j-1 does not exist { // i+1 and j-1 does not exist
smoother->image_pixels[i][j][k] = (input_image->image_pixels[i - 1][j][k] / 4 + input_image->image_pixels[i - 1][j + 1][k] / 4 + input_image->image_pixels[i][j][k] / 4 + input_image->image_pixels[i][j + 1][k] / 4); smooth[ind] = (float)(input_image->image_pixels[i - 1][j][k] / 4 + input_image->image_pixels[i - 1][j + 1][k] / 4 + input_image->image_pixels[i][j][k] / 4 + input_image->image_pixels[i][j + 1][k] / 4);
} }
else if (i == 0 && j == width - 1) else if (i == 0 && j == width - 1)
{ // i-1 and j+1 does not exist { // i-1 and j+1 does not exist
smoother->image_pixels[i][j][k] = (input_image->image_pixels[i][j - 1][k] / 4 + input_image->image_pixels[i][j][k] / 4 + input_image->image_pixels[i + 1][j - 1][k] / 4 + input_image->image_pixels[i + 1][j][k] / 4); smooth[ind] = (float)(input_image->image_pixels[i][j - 1][k] / 4 + input_image->image_pixels[i][j][k] / 4 + input_image->image_pixels[i + 1][j - 1][k] / 4 + input_image->image_pixels[i + 1][j][k] / 4);
} }
else if (i == height - 1 && j == width - 1) else if (i == height - 1 && j == width - 1)
{ // i+1 and j+1 does not exist { // i+1 and j+1 does not exist
smoother->image_pixels[i][j][k] = (input_image->image_pixels[i - 1][j - 1][k] / 4 + input_image->image_pixels[i - 1][j][k] / 4 + input_image->image_pixels[i][j - 1][k] / 4 + input_image->image_pixels[i][j][k] / 4); smooth[ind] = (float)(input_image->image_pixels[i - 1][j - 1][k] / 4 + input_image->image_pixels[i - 1][j][k] / 4 + input_image->image_pixels[i][j - 1][k] / 4 + input_image->image_pixels[i][j][k] / 4);
} }
else if (i == 0) else if (i == 0)
{ // i - 1 does not exist { // i - 1 does not exist
smoother->image_pixels[i][j][k] = (input_image->image_pixels[i][j - 1][k] / 6 + input_image->image_pixels[i][j][k] / 6 + input_image->image_pixels[i][j + 1][k] / 6 + input_image->image_pixels[i + 1][j - 1][k] / 6 + input_image->image_pixels[i + 1][j][k] / 6 + input_image->image_pixels[i + 1][j + 1][k] / 6); smooth[ind] = (float)(input_image->image_pixels[i][j - 1][k] / 6 + input_image->image_pixels[i][j][k] / 6 + input_image->image_pixels[i][j + 1][k] / 6 + input_image->image_pixels[i + 1][j - 1][k] / 6 + input_image->image_pixels[i + 1][j][k] / 6 + input_image->image_pixels[i + 1][j + 1][k] / 6);
} }
else if (j == 0) else if (j == 0)
{ // j -1 does not exist { // j -1 does not exist
smoother->image_pixels[i][j][k] = (input_image->image_pixels[i - 1][j][k] / 6 + input_image->image_pixels[i - 1][j + 1][k] / 6 + input_image->image_pixels[i][j][k] / 6 + input_image->image_pixels[i][j + 1][k] / 6 + input_image->image_pixels[i + 1][j][k] / 6 + input_image->image_pixels[i + 1][j + 1][k] / 6); smooth[ind] = (float)(input_image->image_pixels[i - 1][j][k] / 6 + input_image->image_pixels[i - 1][j + 1][k] / 6 + input_image->image_pixels[i][j][k] / 6 + input_image->image_pixels[i][j + 1][k] / 6 + input_image->image_pixels[i + 1][j][k] / 6 + input_image->image_pixels[i + 1][j + 1][k] / 6);
} }
else if (i == height - 1) else if (i == height - 1)
{ // i+1 does not exist { // i+1 does not exist
smoother->image_pixels[i][j][k] = (input_image->image_pixels[i - 1][j - 1][k] / 6 + input_image->image_pixels[i - 1][j][k] / 6 + input_image->image_pixels[i - 1][j + 1][k] / 6 + input_image->image_pixels[i][j - 1][k] / 6 + input_image->image_pixels[i][j][k] / 6 + input_image->image_pixels[i][j + 1][k] / 6); smooth[ind] = (float)(input_image->image_pixels[i - 1][j - 1][k] / 6 + input_image->image_pixels[i - 1][j][k] / 6 + input_image->image_pixels[i - 1][j + 1][k] / 6 + input_image->image_pixels[i][j - 1][k] / 6 + input_image->image_pixels[i][j][k] / 6 + input_image->image_pixels[i][j + 1][k] / 6);
} }
else if (j == width - 1) else if (j == width - 1)
{ // j + 1 does not exist { // j + 1 does not exist
smoother->image_pixels[i][j][k] = (input_image->image_pixels[i - 1][j - 1][k] / 6 + input_image->image_pixels[i - 1][j][k] / 6 + input_image->image_pixels[i][j - 1][k] / 6 + input_image->image_pixels[i][j][k] / 6 + input_image->image_pixels[i + 1][j - 1][k] / 6 + input_image->image_pixels[i + 1][j][k] / 6); smooth[ind] = (float)(input_image->image_pixels[i - 1][j - 1][k] / 6 + input_image->image_pixels[i - 1][j][k] / 6 + input_image->image_pixels[i][j - 1][k] / 6 + input_image->image_pixels[i][j][k] / 6 + input_image->image_pixels[i + 1][j - 1][k] / 6 + input_image->image_pixels[i + 1][j][k] / 6);
} }
else else
{ {
smoother->image_pixels[i][j][k] = (input_image->image_pixels[i - 1][j - 1][k] / 9 + input_image->image_pixels[i - 1][j][k] / 9 + input_image->image_pixels[i - 1][j + 1][k] / 9 + input_image->image_pixels[i][j - 1][k] / 9 + input_image->image_pixels[i][j][k] / 9 + input_image->image_pixels[i][j + 1][k] / 9 + input_image->image_pixels[i + 1][j - 1][k] / 9 + input_image->image_pixels[i + 1][j][k] / 9 + input_image->image_pixels[i + 1][j + 1][k] / 9); smooth[ind] = (float)(input_image->image_pixels[i - 1][j - 1][k] / 9 + input_image->image_pixels[i - 1][j][k] / 9 + input_image->image_pixels[i - 1][j + 1][k] / 9 + input_image->image_pixels[i][j - 1][k] / 9 + input_image->image_pixels[i][j][k] / 9 + input_image->image_pixels[i][j + 1][k] / 9 + input_image->image_pixels[i + 1][j - 1][k] / 9 + input_image->image_pixels[i + 1][j][k] / 9 + input_image->image_pixels[i + 1][j + 1][k] / 9);
}
ind++;
} }
} }
} }
} return smooth;
return smoother;
} }
struct image_t *S2_find_details(struct image_t *input_image, struct image_t *smoothened_image) float *S2_find_details(struct image_t *input_image, float *smoothened_image)
{ {
// TODO // TODO
int width = input_image->width; int width = input_image->width;
int height = input_image->height; int height = input_image->height;
struct image_t *details = new struct image_t; float *details = (float *)malloc(height * width * 3 * sizeof(float));
details->height = height; long ind = 0;
details->width = width;
details->image_pixels = new uint8_t **[height];
for (int i = 0; i < height; i++) for (int i = 0; i < height; i++)
{ {
details->image_pixels[i] = new uint8_t *[width];
for (int j = 0; j < width; j++) for (int j = 0; j < width; j++)
details->image_pixels[i][j] = new uint8_t[3];
}
for (int i = 1; i < height - 1; i++)
{
for (int j = 1; j < width - 1; j++)
{ {
for (int k = 0; k < 3; k++) for (int k = 0; k < 3; k++)
{ {
details->image_pixels[i][j][k] = max(0, input_image->image_pixels[i][j][k] - smoothened_image->image_pixels[i][j][k]); details[ind] = max((float)0, (float)input_image->image_pixels[i][j][k] - smoothened_image[ind]);
ind++;
} }
} }
} }
@ -110,33 +90,25 @@ struct image_t *S2_find_details(struct image_t *input_image, struct image_t *smo
return details; return details;
} }
struct image_t *S3_sharpen(struct image_t *input_image, struct image_t *details_image) float *S3_sharpen(struct image_t *input_image, float *details_image)
{ {
// TODO // TODO
int width = input_image->width; int width = input_image->width;
int height = input_image->height; int height = input_image->height;
int a = 1; float a = 1;
struct image_t *sharp = new struct image_t; float *sharp = (float *)malloc(height * width * 3 * sizeof(float));
sharp->height = height;
sharp->width = width;
sharp->image_pixels = new uint8_t **[height];
long ind = 0;
for (int i = 0; i < height; i++) for (int i = 0; i < height; i++)
{ {
sharp->image_pixels[i] = new uint8_t *[width];
for (int j = 0; j < width; j++) for (int j = 0; j < width; j++)
sharp->image_pixels[i][j] = new uint8_t[3];
}
for (int i = 1; i < height - 1; i++)
{
for (int j = 1; j < width - 1; j++)
{ {
for (int k = 0; k < 3; k++) for (int k = 0; k < 3; k++)
{ {
sharp->image_pixels[i][j][k] = min(255, input_image->image_pixels[i][j][k] + details_image->image_pixels[i][j][k]) * a; sharp[ind] = min((float)255, (float)input_image->image_pixels[i][j][k] + details_image[ind]);
ind++;
} }
} }
} }