Blame view

src/csv_reader.c 3.27 KB
296e73d0   jdequidt   ✚ fichier exemple...
1
2
3
4
5
6
  #include <stdio.h>
  #include <string.h>
  #include <stdbool.h>
  #include <stdlib.h>
  
  // Conditionals
39e273ac   fleroy2   sauvegarde fichie...
7
  const bool          IS_DEBUG            = true;
296e73d0   jdequidt   ✚ fichier exemple...
8
9
10
11
12
13
14
15
16
17
  
  // Constants
  const unsigned int  BUFFER_SIZE         = 2048;
  const unsigned int  FIELD_SIZE          = 20;
  const char          CSV_DELIMITERS[]    = ",";
  
  // Globals
  char**              CSV_HEADER_FIELDS;
  unsigned int        CSV_NB_FIELDS;
  
39e273ac   fleroy2   sauvegarde fichie...
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
  char *strtok_new(char * string, char const * delimiter)
  {
    static char *source = NULL;
    char *p, *ret = 0;
    if(string != NULL) source = string;
    if(source == NULL) return NULL;
    
    if( (p = strpbrk(source, delimiter)) != NULL)
      {
        *p = 0;
        ret = source;
        source = ++p;
      }
    return ret;
  }
  
  
296e73d0   jdequidt   ✚ fichier exemple...
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
  void display_header()
  {
      for(unsigned int i = 0; i < CSV_NB_FIELDS; i++)
      {
          printf("%d - %s\n", i, CSV_HEADER_FIELDS[i]);
      }
  }
  
  void read_csv_header(char * header_line)
  {
      int     line_length = strlen(header_line);
      int     nb_fields   = 0;
      char*   string_ptr  = header_line;
  
      // Count the occurrences of delimiters
      while (NULL != string_ptr)
      {
          nb_fields++;
          string_ptr = strpbrk(string_ptr, CSV_DELIMITERS);
          if (NULL != string_ptr)
          {
              string_ptr++;
          }
      }
  
      // Globals allocation
      CSV_NB_FIELDS       = nb_fields;
      CSV_HEADER_FIELDS   = malloc( nb_fields * sizeof(char*) );
  
      char* token         = strtok(header_line, CSV_DELIMITERS);                  // strtok init.
  
      // Re-read the line to get the header of the columns
      for (unsigned int i = 0; i < nb_fields; i++)
      {
          CSV_HEADER_FIELDS[i] = malloc( FIELD_SIZE * sizeof(char) );             // alloc
          memset(CSV_HEADER_FIELDS[i], 0, FIELD_SIZE);                            // 0 init.
          strcpy(CSV_HEADER_FIELDS[i], token);                                    // copy field in the structure
          token = strtok(NULL, CSV_DELIMITERS);                                   // loop to get a new field label
      }
  
      if (IS_DEBUG) display_header();
  }
  
  void read_csv_file(const char * filename)
  {
      FILE*   fp = fopen(filename, "r");
      char    buffer[BUFFER_SIZE];
  
      // Check if the file is really opened
      if (NULL == fp)
      {
          fprintf(stderr, "Unable to open file: %s\n", filename);
          return;
      }
  
      // 1st row is a header with field descriptions
      fgets(buffer, BUFFER_SIZE, fp);
      read_csv_header(buffer);
  
      // Remaining rows are the entries
      while ( NULL != fgets(buffer, BUFFER_SIZE, fp) )
      {
          char*           token;
          unsigned int    i = 0;
  
          // strtok init.
39e273ac   fleroy2   sauvegarde fichie...
101
          token = strtok_new(buffer, CSV_DELIMITERS);
296e73d0   jdequidt   ✚ fichier exemple...
102
103
104
  
          while (NULL != token)
          {
c3bfddce   jdequidt   🛠 fix du fichier...
105
106
107
108
109
110
              if (IS_DEBUG) printf("Field %d is %s\n", i++, token);
  
              // ...
              // you can strcpy the `token` string in your data structures
              // ...
  
39e273ac   fleroy2   sauvegarde fichie...
111
              token = strtok_new(NULL, CSV_DELIMITERS);
296e73d0   jdequidt   ✚ fichier exemple...
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
          }
      }
  
      fclose(fp);
  }
  
  void usage(const char * prog_name)
  {
      printf("Usage is %s your_csv_file\n\n", prog_name);
  }
  
  int main(int argc, char * argv[])
  {
      if (2 != argc)
      {
          usage(argv[0]);
          return 0;
      }
  
      read_csv_file(argv[1]);
      for(unsigned int i = 0; i < CSV_NB_FIELDS; i++)
      {
          free(CSV_HEADER_FIELDS[i]);
      }
      free(CSV_HEADER_FIELDS);
  
      return 0;
c3bfddce   jdequidt   🛠 fix du fichier...
139
  }