Blame view

src/csv_reader.c 2.95 KB
296e73d0   jdequidt   ✚ fichier exemple...
1
2
3
4
5
6
  #include <stdio.h>
  #include <string.h>
  #include <stdbool.h>
  #include <stdlib.h>
  
  // Conditionals
280074a8   Corentin Danjou   ok
7
  const bool          IS_DEBUG            = true;
296e73d0   jdequidt   ✚ fichier exemple...
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
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
  
  // 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;
  
  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.
          token = strtok(buffer, CSV_DELIMITERS);
  
          while (NULL != token)
          {
280074a8   Corentin Danjou   ok
88
  	    if (IS_DEBUG) printf("Field %d is %s\n", i++, token);
c3bfddce   jdequidt   🛠 fix du fichier...
89
90
91
92
93
  
              // ...
              // you can strcpy the `token` string in your data structures
              // ...
  
296e73d0   jdequidt   ✚ fichier exemple...
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
              token = strtok(NULL, CSV_DELIMITERS);
          }
      }
  
      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...
122
  }