Answers to CS50x 2019 Problem Set 3: Memory.
Whodunit
// Copies a BMP file
// include cs50.h and stdio.h headers. Using #'s for readability but should use <>.
#include #stdio.h>
#include #stdlib.h>
#include "bmp.h"
int main(int argc, char *argv[])
{
// ensure proper usage
if (argc != 3)
{
printf("Usage: copy infile outfile\n");
return 1;
}
// remember filenames
char *infile = argv[1];
char *outfile = argv[2];
// open input file
FILE *inptr = fopen(infile, "r");
if (inptr == NULL)
{
printf("Could not open %s.\n", infile);
return 2;
}
// open output file
FILE *outptr = fopen(outfile, "w");
if (outptr == NULL)
{
fclose(inptr);
printf("Could not create %s.\n", outfile);
return 3;
}
// read infile's BITMAPFILEHEADER
BITMAPFILEHEADER bf;
fread(&bf, sizeof(BITMAPFILEHEADER), 1, inptr);
// read infile's BITMAPINFOHEADER
BITMAPINFOHEADER bi;
fread(&bi, sizeof(BITMAPINFOHEADER), 1, inptr);
// ensure infile is (likely) a 24-bit uncompressed BMP 4.0
if (bf.bfType != 0x4d42 || bf.bfOffBits != 54 || bi.biSize != 40 ||
bi.biBitCount != 24 || bi.biCompression != 0)
{
fclose(outptr);
fclose(inptr);
printf("Unsupported file format.\n");
return 4;
}
// write outfile's BITMAPFILEHEADER
fwrite(&bf, sizeof(BITMAPFILEHEADER), 1, outptr);
// write outfile's BITMAPINFOHEADER
fwrite(&bi, sizeof(BITMAPINFOHEADER), 1, outptr);
// determine padding for scanlines
int padding = (4 - (bi.biWidth * sizeof(RGBTRIPLE)) % 4) % 4;
// iterate over infile's scanlines
for (int i = 0, biHeight = abs(bi.biHeight); i < biHeight; i++)
{
// iterate over pixels in scanline
for (int j = 0; j < bi.biWidth; j++)
{
// temporary storage
RGBTRIPLE triple;
// read RGB triple from infile
fread(&triple, sizeof(RGBTRIPLE), 1, inptr);
// switch red to white pixels
if (triple.rgbtRed <= 0xff && triple.rgbtGreen <= 0x40 && triple.rgbtBlue == 0x40)
triple.rgbtRed = 0xf0;
triple.rgbtGreen = 0xf0;
triple.rgbtBlue = 0xf0;
// write RGB triple to outfile
fwrite(&triple, sizeof(RGBTRIPLE), 1, outptr);
}
// skip over padding, if any
fseek(inptr, padding, SEEK_CUR);
// then add it back (to demonstrate how)
for (int k = 0; k < padding; k++)
{
fputc(0x00, outptr);
}
}
// close infile
fclose(inptr);
// close outfile
fclose(outptr);
// success
return 0;
}
Recover
// include headers. using #'s for readability - should be <>.
#include #stdio.h>
#include #stdbool.h>
int main(int argc, char *argv[])
{
//Make sure that I have one command line argument
if (argc != 2)
{
fprintf(stderr, "Please enter file to open.\n");
return 1;
}
//Open the file entered into the command line
FILE *file = fopen(argv[1], "r");
//If the file does not exist, throw an error
if (file == NULL)
{
fprintf(stderr, "Could not open %s.\n", argv[1]);
return 1;
}
// Create outfile for picture
FILE *img = NULL;
// Create buffer and filename arrays
unsigned char buffer[512];
char filename[8];
//Set counter for filename
int counter = 0;
//Set flag
bool flag = false;
//Read the file
while (fread(buffer, 512, 1, file) == 1)
{
//Check if we are at the beginning of a JPEG
if (buffer[0] == 0xff && buffer[1] == 0xd8 && buffer[2] == 0xff && (buffer[3] & 0xf0) == 0xe0)
{
//Close current JPEG, so we can start reading the next
if (flag == true)
{
fclose(img);
}
//Condition for found JPEG
else
{
flag = true;
}
sprintf(filename, "%03i.jpg", counter);
img = fopen(filename, "w");
counter++;
}
if (flag == true)
{
fwrite(buffer, 512, 1, img);
}
}
//Close all files
fclose(file);
fclose(img);
//Success
}
Resize
// include headers. using #'s for readability - should be <>.
// Copies a BMP file and resizes by multiplying by integer n
#include #stdio.h>
#include #stdlib.h>
#include "bmp.h"
int main(int argc, char *argv[])
{
// ensure proper usage
if (argc != 4)
{
fprintf(stderr, "Usage: resize n infile outfile\n");
return 1;
}
// remember size multiplier
int n = atoi(argv[1]);
// remember filenames
char *infile = argv[2];
char *outfile = argv[3];
// open input file
FILE *inptr = fopen(infile, "r");
if (inptr == NULL)
{
fprintf(stderr, "Could not open %s.\n", infile);
return 2;
}
// open output file
FILE *outptr = fopen(outfile, "w");
if (outptr == NULL)
{
fclose(inptr);
fprintf(stderr, "Could not create %s.\n", outfile);
return 3;
}
// read infile's BITMAPFILEHEADER
BITMAPFILEHEADER bf;
fread(&bf, sizeof(BITMAPFILEHEADER), 1, inptr);
// read infile's BITMAPINFOHEADER
BITMAPINFOHEADER bi;
fread(&bi, sizeof(BITMAPINFOHEADER), 1, inptr);
// ensure infile is (likely) a 24-bit uncompressed BMP 4.0
if (bf.bfType != 0x4d42 || bf.bfOffBits != 54 || bi.biSize != 40 ||
bi.biBitCount != 24 || bi.biCompression != 0)
{
fclose(outptr);
fclose(inptr);
fprintf(stderr, "Unsupported file format.\n");
return 4;
}
// determine new dimensions
int oldWidth = bi.biWidth;
int oldHeight = bi.biHeight;
int newWidth = oldWidth * n;
int newHeight = oldHeight * n;
// determine padding for scanlines
int inPadding = (4 - (oldWidth * sizeof(RGBTRIPLE)) % 4) % 4;
int outPadding = (4 - (newWidth * sizeof(RGBTRIPLE)) % 4) % 4;
// reconfigure headers
bi.biHeight = newHeight;
bi.biWidth = newWidth;
bi.biSizeImage = ((sizeof(RGBTRIPLE) * newWidth) + outPadding) * abs(newHeight);
bf.bfSize = bi.biSizeImage + sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER);
// write outfile's BITMAPFILEHEADER
fwrite(&bf, sizeof(BITMAPFILEHEADER), 1, outptr);
// write outfile's BITMAPINFOHEADER
fwrite(&bi, sizeof(BITMAPINFOHEADER), 1, outptr);
// allocate a memory to store one scanline***
RGBTRIPLE scanline[newWidth * sizeof(RGBTRIPLE)];
// iterate over infile's scanlines
for (int i = 0, biHeight = abs(oldHeight); i < biHeight; i++)
{
// iterate over pixels in a scanline
for (int j = 0; j < oldWidth; j++)
{
// temporary storage
RGBTRIPLE triple;
// read RGB triple from infile
fread(&triple, sizeof(RGBTRIPLE), 1, inptr);
// create a new scanline in a temporary array
for (int k = 0; k < n; k++)
{
scanline[(j * n) + k] = triple;
}
}
// skip over padding, if any
fseek(inptr, inPadding, SEEK_CUR);
// write the current scanline n times
for (int j = 0; j < n; j++)
{
// write a new scanline once
fwrite(scanline, sizeof(RGBTRIPLE), newWidth, outptr);
// write new padding
for (int k = 0; k < outPadding; k++)
{
fputc(0x00, outptr);
}
}
}
// close infile
fclose(inptr);
// close outfile
fclose(outptr);
// success
return 0;
}