3 # Convert CSV parts list into vendor import format
5 # Copyright © 2015 Keith Packard <keithp@keithp.com>, GPL v3+
11 string[*] not_vendors;
12 string not_vendor_list;
19 void fatal(string format, poly args ...)
21 File::fprintf(stderr, format, args...);
25 string[*] read_line(file f) {
27 string line = fgets(f);
29 return String::parse_csv(line);
34 string[*] required_elements = {
45 bool has_header_member(string member) {
46 for (int i = 0; i < dim(header); i++)
47 if (header[i] == member)
52 bool has_vendor(string[*] vendors, string vendor) {
53 for (int i = 0; i < dim(vendors); i++)
54 if (vendors[i] == vendor)
59 void read_header(file f) {
60 header = read_line(f);
62 for (int i = 0; i < dim(required_elements); i++)
63 if (!has_header_member(required_elements[i]))
64 fatal("Missing header element \"%s\"\n", required_elements[i]);
67 string[string] read_entry(file f) {
68 string[*] elements = read_line(f);
69 string[string] entry = {};
71 if (dim(header) != dim(elements))
72 fatal("line %d: has %d instead of %d elements (%V)\n",
73 lineno, dim(elements), dim(header), elements);
75 for (int i = 0; i < dim(header); i++) {
76 if (elements[i] == "")
77 elements[i] = "unknown";
78 entry[header[i]] = elements[i];
83 string part_number(string[string] entry)
86 return entry["mfg_part_number"];
88 return entry["vendor_part_number"];
91 void process_seeed(string[string] entry)
93 if (entry["loadstatus"] != "smt")
96 static bool start = true;
98 printf("Part/Designator,Manufacturer Part Number/Seeed SKU, Quantity\n");
102 string[*] refdes = String::wordsplit(entry["refdes"], " \t");
105 for (int i = 0; i < dim(refdes); i++) {
106 printf("%s", refdes[i]);
107 if (i < dim(refdes) - 1)
112 printf(",%s,%s\n", entry["mfg_part_number"], entry["quantity"]);
115 void process_digikey(string[string] entry)
117 if (entry["loadstatus"] == "noload")
119 printf("%s,%s,%s %s\n",
126 void process_mouser(string[string] entry)
128 if (entry["loadstatus"] == "noload")
130 printf("%s|%s\n", part_number(entry), entry["quantity"]);
133 void process_other(string[string] entry) {
134 if (entry["loadstatus"] == "noload")
136 printf("%s,%s,%s,%s %s\n",
144 void process_file(file f) {
146 while (!File::end(f)) {
147 string[string] entry = read_entry(f);
148 string vendor = entry["vendor"];
149 if (!is_uninit(&vendors) && has_vendor(vendors, "seeed")) {
150 process_seeed(entry);
151 } else if ((!is_uninit(&vendors) && has_vendor(vendors, vendor)) ||
152 (!is_uninit(¬_vendors) && !has_vendor(not_vendors, vendor))) {
153 switch (entry["vendor"]) {
155 process_digikey(entry);
158 process_mouser(entry);
161 process_other(entry);
168 ParseArgs::argdesc argd = {
170 { .var = { .arg_flag = &mfg_part },
173 .desc = "Display manufacturer part number"},
174 { .var = { .arg_string = &vendor_list },
177 .expr_name = "vendors",
178 .desc = "Vendors to match"},
179 { .var = { .arg_string = ¬_vendor_list },
181 .name = "not-vendor",
182 .expr_name = "not-vendor",
183 .desc = "Vendors to exclude"},
189 ParseArgs::parseargs(&argd, &argv);
190 if (!is_uninit(&vendor_list))
191 vendors = String::parse_csv(vendor_list);
193 if (!is_uninit(¬_vendor_list))
194 not_vendors = String::parse_csv(not_vendor_list);
196 if (!is_uninit(&argi)) {
197 for (int i = argi; i < dim(argv); i++)
198 twixt(file f = File::open(argv[i], "r"); File::close(f))