C / C ++更好的写这个方法?

时间:2014-03-15 15:31:26

标签: c++ c

我正在尝试编写易于调试的代码。是否有更好的方法来编写函数do_save()

int one;
int two[10];
char three;

...

unsigned char *state = (unsigned char*)malloc(1024);

...

void do_save()
{
    memcpy(state, &one, 4);
    memcpy(state+4, two, 40);
    memcpy(state+44, &three, 1);
}

当我说得更好时,我的意思是避免创建错误的一种方法,因为我弄乱了大小计数,特别是当有10或20个变量要保存在状态时。

4 个答案:

答案 0 :(得分:3)

使用结构。

int one;
int two[10];
char three;
typedef struct {
    int one;
    int two[10];
    char three;
} State;

...

State *state = new State; // C++ish
State *state = malloc(sizeof(State));  //Cish

...

void do_save(State* state) {
    state->one = &one;
    memcpy(state->two, &two, sizeof(state->two));
    state->three = three;
}

有了结构,你可以很容易地做很多事情。例如,您可以将当前状态和已保存状态分开,并使用等号完成保存/恢复。使用fread / fwrite可以轻松写入二进制文件。您可以根据需要将状态结构放在堆上或堆栈上。

#include <stdlib.h>
#include <stdio.h>

typedef struct {
    int one;
    int two[10];
    char three;
} State;

void print_state(State* state) {
    printf("%i ",   state->one);
    int i;
    for (i = 0; i < 10; ++i) {
        printf("%i ",   state->two[i]);
    }
    printf("%c\n",  state->three);
}

int main() {
    State* state = (State*)(malloc(sizeof(State)));
    State current_state;
    FILE* input = fopen("binary.data", "rb");
    if (input) {
        fread(&current_state, sizeof(State), 1, input);
        fclose(input);
    }

    print_state(&current_state);

    current_state.one = 1;
    int i;
    for (i = 0; i < 10; ++i) {
        current_state.two[i] = i + 1;
    }
    current_state.three = 'Z';
    *state = current_state;

    FILE* output = fopen("binary.data", "wb");
    fwrite(state, sizeof(State), 1, output);
    fclose(output);
        free(state);
}

答案 1 :(得分:2)

您可以使用变量为您完成工作。例如:

int sizetotal = 0;
int sizethis = 0;

sizethis = sizeof(one);
memcpy(state+sizetotal, &one, sizethis);
sizetotal += sizethis;

sizethis = sizeof(*two);
memcpy(state+sizetotal, two, sizethis);
sizetotal += sizethis;

sizethis = sizeof(three);
memcpy(state+sizetotal, &three, sizethis);
sizetotal += sizethis;

正如您所看到的那样,重复了3行 - 因此它可以放在宏或函数中。

当然,如果这是C ++,最好的办法是创建一个State对象并给它适当的方法。

答案 2 :(得分:1)

看起来你正在将连续的偏移从缓冲状态复制到其他一些位置。您可以通过简单地使用Struct - &gt;从连续偏移复制到连续偏移。像这样:

typedef struct {
    int one;
    int two[10];
    char three;
} Container;

Container  my_storage

...
unsigned char *state = (unsigned char*)malloc(1024);
...

void do_save(Container  *dest, char*src) {
   memcpy(src, state, sizeof(Container));    
}

答案 3 :(得分:1)

使用C ++:

    int one;
    std::array<int, 10> two;
    char three;

    std::vector<unsigned char> state(1024);

    auto start = state.begin();
    auto end = start + sizeof(one);
    std::copy(start, end, &one);

    start = ++end;
    end += two.size();
    std::copy(start, end, two.begin());

    start = ++end;
    end += sizeof(three);
    std::copy(start, end, &three);