"Оптимизация" бинарного файла - C++ - Ответ 3650842

#include "stdafx.h"
#include <conio.h>
#include <stdlib.h>
#include <memory.h>
 
struct scan_info {
    char model[25]; // наименование модели
    int price; // цена
    double x_size; // горизонтальный размер области сканирования
    double y__size; // вертикальный размер области сканирования
    int optr; // оптическое разрешение
    int grey; // число градаций серого
} scan;
 
int ShowMenu();
void CreateFile(); 
void PrintFromFile();
void OptimizeFile(); 
 
int ShowMenu() {
    char ch = '\0';
    puts("1 : Create a file"); 
    puts("2 : View a file");
    puts("3 : Optimize a file");
    puts("4 : Exit");
    ch = _getch(); 
    return atoi(&ch); 
}
 
int main() {
    while (1) {  
        switch( ShowMenu() ) {  
            case 1: CreateFile(); break;
            case 2: PrintFromFile(); break; 
            case 3: OptimizeFile(); break; 
            case 4: return 0; 
            default: break;
        }
    }
}
const int min_records = 6; //минимальное число записей для ввода (нулевые учитываются)
 
void PrintArray (scan_info* si, int count) //вывод записей на экран из массива
{
    for (int i = 0; i < count; i++)
        printf ("\n%i. %s price: %i grey: %i", i+1, si[i].model, si[i].price, si[i].grey);
}
 
void PrintFromFile () //вывод записей на экран из файла
{
    FILE* data;
    char* db = "file.bin";
    data = fopen (db, "rb"); //открываем файл
    if (!data) { //ошибка, если не открылся
        printf ("Error");
        return;
    }
    short count; //читаем количество записей
    int k = fread (&count, sizeof(short), 1, data);
    if (k != 1) { //ошибка, если считалось неверное количество данных
        printf ("Error");
        return;
    }
    scan_info* si = new scan_info[count]; //выделяем место под массив
    k = fread (si, sizeof(scan_info), count, data); //считываем массив
    if (k != count) {
        printf ("Error");
        return;
    }
    for (int i = 0; i < count; i++) //выводим на экран
        printf ("\n%i. %s price: %i grey: %i", i+1, si[i].model, si[i].price, si[i].grey);
    fclose (data);
}
 
 
void OptimizeFile () //сжатие записей из файла
{ 
    FILE* data;
    char* db = "file.bin";
    data = fopen (db, "r+b"); //открываем файл
    if (!data) { //ошибка, если не открылся
        printf ("Error");
        return;
    }
    short count; //читаем количество записей
    int k = fread (&count, sizeof(short), 1, data);
    if (k != 1) { //ошибка, если считалось неверное количество данных
        printf ("Error");
        return;
    }
    scan_info* si = new scan_info[count]; //выделяем место под массив
    k = fread (si, sizeof(scan_info), count, data); //считываем массив
    if (k != count) {
        printf ("Error");
        return;
    }
 
    PrintArray (si, count); //выводим на экран для контроля
    fseek (data, sizeof(short), SEEK_SET); //переходим в начало файла, пропуская число записей
 
    short newcount = 0; //число реальных записей
    scan_info nul; //нулевая запись для сравнения
    memset (&nul, 0, sizeof(nul)); //заполняем нулями
 
    for (int i = 0; i < count; i++) { //просмотриваем массив
        if (memcmp (&nul, &si[i], sizeof(scan_info)) != 0) { //Если запись не идентична нулевой
            k = fwrite (&si[i], sizeof(scan_info), 1, data); //пишем запись в файл
            if (k != 1)  {
                printf ("Error");
                return;
            }
            newcount++; //наращиваем реальное количество
        }
    }
 
    fseek (data, 0, SEEK_SET); //переходим в начало файла
    k = fwrite (&newcount, sizeof(short), 1, data); //пишем реальное количество записей
    if (k != 1)  {
        printf ("Error");
        return;
    }
    fclose (data); //закрываем файл
}
 
void CreateFile () //создание файла
{
    short n = 0; //количество записей
    int pos;
    scan_info si, nul;
    memset (&nul, 0, sizeof(nul)); //нулевая запись для пропусков заполняется нулями
    FILE* data;
    char* db = "file.bin";
    data = fopen (db, "wb");
    if (!data) {
        printf ("Error");
        return;
    }
    int k = fwrite (&n, sizeof(short), 1, data); //пока пишем 0 в файл
    if (k != 1) {
        printf ("Error");
        return;
    }
    char c = 'y'; //признак выхода из цикла
    while (c=='y') { //вводим записи
        printf ("\nInput new record.\nInput model:\n");
        scanf ("%s", &si.model);
        printf ("\nInput price:\n");
        scanf ("%d", &si.price);
        printf ("\nInput horizontal size:\n");
        scanf ("%f", &si.x_size);
        printf ("\nInput vertical size:\n");
        scanf ("%f", &si.y__size);
        printf ("\nInput optical resolution:\n");
        scanf ("%d", &si.optr);
        printf ("\nInput shade of grey:\n");
        scanf ("%d", &si.grey);
        printf ("\nInput position to save (-1 for write to the end):\n");
        scanf ("%d", &pos); //дополнительно спрашиваем позицию для сохранения
 
        n++; //наращиваем количество
        if (pos < 0) pos = n; //если ввели отрицательную позицию, пишем в конец файла
        if (pos > n) { //если позиция дальше текущей
            fseek (data, (n-1)* sizeof(si) + sizeof(short), SEEK_SET);
//переставляем указатель в файле на текущее место
            while (pos > n) { //пишем столько нулевых записей, сколько позиций надо пропустить
                fwrite (&nul, sizeof(nul), 1, data);
                n++;
            }
        } else if (pos < n) { //если позиция раньше текущей
            n--; //не считаем эту запись - она займёт имеющееся место
            fseek (data, (pos-1)* sizeof(si) + sizeof(short), SEEK_SET);
//ставим указатель на нужное место
        } else //пишем на текущую позицию, ставим на неё указатель
            fseek (data, (n-1)* sizeof(si) + sizeof(short), SEEK_SET);
        
 
        k = fwrite (&si, sizeof(si), 1, data); //пишем запись в файл
        if (k != 1) {
            printf ("Error");
            return;
        }
        if (n >= min_records) { //если 6 записей уже введено, то предлагаем прервать ввод
            printf ("\nPress 'y', if you want to continue, or any other key, if you don't\n");
            c = getch(); //если будет введён не у, то выйдем из цикла
        }
    }
 
    fseek (data, 0, SEEK_SET); //идём в начало файла
    k = fwrite (&n, sizeof(short), 1, data); //пишем реальное количество записей
    if (k != 1) {
        printf ("Error");
        return;
    }
    fclose (data); //закрываем файл
}