]> git.saurik.com Git - apple/icu.git/blob - icuSources/tools/tzcode/tz2icu.cpp
ICU-57131.0.1.tar.gz
[apple/icu.git] / icuSources / tools / tzcode / tz2icu.cpp
1
2 /*
3 **********************************************************************
4 * Copyright (c) 2003-2014, International Business Machines
5 * Corporation and others. All Rights Reserved.
6 **********************************************************************
7 * Author: Alan Liu
8 * Created: July 10 2003
9 * Since: ICU 2.8
10 **********************************************************************
11 */
12 #include "tzfile.h" // from Olson tzcode archive, copied to this dir
13
14 #ifdef WIN32
15
16 #include <windows.h>
17 #undef min // windows.h/STL conflict
18 #undef max // windows.h/STL conflict
19 // "identifier was truncated to 'number' characters" warning
20 #pragma warning(disable: 4786)
21
22 #else
23
24 #include <unistd.h>
25 #include <stdio.h>
26 #include <dirent.h>
27 #include <string.h>
28 #include <sys/stat.h>
29
30 #endif
31
32 #include <algorithm>
33 #include <cassert>
34 #include <ctime>
35 #include <fstream>
36 #include <iomanip>
37 #include <iostream>
38 #include <iterator>
39 #include <limits>
40 #include <map>
41 #include <set>
42 #include <sstream>
43 #include <sstream>
44 #include <stdexcept>
45 #include <string>
46 #include <vector>
47
48 #include "tz2icu.h"
49 #include "unicode/uversion.h"
50
51 using namespace std;
52
53 bool ICU44PLUS = TRUE;
54 string TZ_RESOURCE_NAME = ICU_TZ_RESOURCE;
55
56 //--------------------------------------------------------------------
57 // Time utilities
58 //--------------------------------------------------------------------
59
60 const int64_t SECS_PER_YEAR = 31536000; // 365 days
61 const int64_t SECS_PER_LEAP_YEAR = 31622400; // 366 days
62 const int64_t LOWEST_TIME32 = (int64_t)((int32_t)0x80000000);
63 const int64_t HIGHEST_TIME32 = (int64_t)((int32_t)0x7fffffff);
64
65 bool isLeap(int32_t y) {
66 return (y%4 == 0) && ((y%100 != 0) || (y%400 == 0)); // Gregorian
67 }
68
69 int64_t secsPerYear(int32_t y) {
70 return isLeap(y) ? SECS_PER_LEAP_YEAR : SECS_PER_YEAR;
71 }
72
73 /**
74 * Given a calendar year, return the GMT epoch seconds for midnight
75 * GMT of January 1 of that year. yearToSeconds(1970) == 0.
76 */
77 int64_t yearToSeconds(int32_t year) {
78 // inefficient but foolproof
79 int64_t s = 0;
80 int32_t y = 1970;
81 while (y < year) {
82 s += secsPerYear(y++);
83 }
84 while (y > year) {
85 s -= secsPerYear(--y);
86 }
87 return s;
88 }
89
90 /**
91 * Given 1970 GMT epoch seconds, return the calendar year containing
92 * that time. secondsToYear(0) == 1970.
93 */
94 int32_t secondsToYear(int64_t seconds) {
95 // inefficient but foolproof
96 int32_t y = 1970;
97 int64_t s = 0;
98 if (seconds >= 0) {
99 for (;;) {
100 s += secsPerYear(y++);
101 if (s > seconds) break;
102 }
103 --y;
104 } else {
105 for (;;) {
106 s -= secsPerYear(--y);
107 if (s <= seconds) break;
108 }
109 }
110 return y;
111 }
112
113 //--------------------------------------------------------------------
114 // Types
115 //--------------------------------------------------------------------
116
117 struct FinalZone;
118 struct FinalRule;
119 struct SimplifiedZoneType;
120
121 // A transition from one ZoneType to another
122 // Minimal size = 5 bytes (4+1)
123 struct Transition {
124 int64_t time; // seconds, 1970 epoch
125 int32_t type; // index into 'ZoneInfo.types' 0..255
126 Transition(int64_t _time, int32_t _type) {
127 time = _time;
128 type = _type;
129 }
130 };
131
132 // A behavior mode (what zic calls a 'type') of a time zone.
133 // Minimal size = 6 bytes (4+1+3bits)
134 // SEE: SimplifiedZoneType
135 struct ZoneType {
136 int64_t rawoffset; // raw seconds offset from GMT
137 int64_t dstoffset; // dst seconds offset from GMT
138
139 // We don't really need any of the following, but they are
140 // retained for possible future use. See SimplifiedZoneType.
141 int32_t abbr; // index into ZoneInfo.abbrs 0..n-1
142 bool isdst;
143 bool isstd;
144 bool isgmt;
145
146 ZoneType(const SimplifiedZoneType&); // used by optimizeTypeList
147
148 ZoneType() : rawoffset(-1), dstoffset(-1), abbr(-1) {}
149
150 // A restricted equality, of just the raw and dst offset
151 bool matches(const ZoneType& other) {
152 return rawoffset == other.rawoffset &&
153 dstoffset == other.dstoffset;
154 }
155 };
156
157 // A collection of transitions from one ZoneType to another, together
158 // with a list of the ZoneTypes. A ZoneInfo object may have a long
159 // list of transitions between a smaller list of ZoneTypes.
160 //
161 // This object represents the contents of a single zic-created
162 // zoneinfo file.
163 struct ZoneInfo {
164 vector<Transition> transitions;
165 vector<ZoneType> types;
166 vector<string> abbrs;
167
168 string finalRuleID;
169 int32_t finalOffset;
170 int32_t finalYear; // -1 if none
171
172 // If this is an alias, then all other fields are meaningless, and
173 // this field will point to the "real" zone 0..n-1.
174 int32_t aliasTo; // -1 if this is a "real" zone
175
176 // If there are aliases TO this zone, then the following set will
177 // contain their index numbers (each index >= 0).
178 set<int32_t> aliases;
179
180 ZoneInfo() : finalYear(-1), aliasTo(-1) {}
181
182 void mergeFinalData(const FinalZone& fz);
183
184 void optimizeTypeList();
185
186 // Set this zone to be an alias TO another zone.
187 void setAliasTo(int32_t index);
188
189 // Clear the list of aliases OF this zone.
190 void clearAliases();
191
192 // Add an alias to the list of aliases OF this zone.
193 void addAlias(int32_t index);
194
195 // Is this an alias to another zone?
196 bool isAlias() const {
197 return aliasTo >= 0;
198 }
199
200 // Retrieve alias list
201 const set<int32_t>& getAliases() const {
202 return aliases;
203 }
204
205 void print(ostream& os, const string& id) const;
206 };
207
208 void ZoneInfo::clearAliases() {
209 assert(aliasTo < 0);
210 aliases.clear();
211 }
212
213 void ZoneInfo::addAlias(int32_t index) {
214 assert(aliasTo < 0 && index >= 0 && aliases.find(index) == aliases.end());
215 aliases.insert(index);
216 }
217
218 void ZoneInfo::setAliasTo(int32_t index) {
219 assert(index >= 0);
220 assert(aliases.size() == 0);
221 aliasTo = index;
222 }
223
224 typedef map<string, ZoneInfo> ZoneMap;
225
226 typedef ZoneMap::const_iterator ZoneMapIter;
227
228 //--------------------------------------------------------------------
229 // ZONEINFO
230 //--------------------------------------------------------------------
231
232 // Global map holding all our ZoneInfo objects, indexed by id.
233 ZoneMap ZONEINFO;
234
235 //--------------------------------------------------------------------
236 // zoneinfo file parsing
237 //--------------------------------------------------------------------
238
239 // Read zic-coded 32-bit integer from file
240 int64_t readcoded(ifstream& file, int64_t minv=numeric_limits<int64_t>::min(),
241 int64_t maxv=numeric_limits<int64_t>::max()) {
242 unsigned char buf[4]; // must be UNSIGNED
243 int64_t val=0;
244 file.read((char*)buf, 4);
245 for(int32_t i=0,shift=24;i<4;++i,shift-=8) {
246 val |= buf[i] << shift;
247 }
248 if (val < minv || val > maxv) {
249 ostringstream os;
250 os << "coded value out-of-range: " << val << ", expected ["
251 << minv << ", " << maxv << "]";
252 throw out_of_range(os.str());
253 }
254 return val;
255 }
256
257 // Read zic-coded 64-bit integer from file
258 int64_t readcoded64(ifstream& file, int64_t minv=numeric_limits<int64_t>::min(),
259 int64_t maxv=numeric_limits<int64_t>::max()) {
260 unsigned char buf[8]; // must be UNSIGNED
261 int64_t val=0;
262 file.read((char*)buf, 8);
263 for(int32_t i=0,shift=56;i<8;++i,shift-=8) {
264 val |= (int64_t)buf[i] << shift;
265 }
266 if (val < minv || val > maxv) {
267 ostringstream os;
268 os << "coded value out-of-range: " << val << ", expected ["
269 << minv << ", " << maxv << "]";
270 throw out_of_range(os.str());
271 }
272 return val;
273 }
274
275 // Read a boolean value
276 bool readbool(ifstream& file) {
277 char c;
278 file.read(&c, 1);
279 if (c!=0 && c!=1) {
280 ostringstream os;
281 os << "boolean value out-of-range: " << (int32_t)c;
282 throw out_of_range(os.str());
283 }
284 return (c!=0);
285 }
286
287 /**
288 * Read the zoneinfo file structure (see tzfile.h) into a ZoneInfo
289 * @param file an already-open file stream
290 */
291 void readzoneinfo(ifstream& file, ZoneInfo& info, bool is64bitData) {
292 int32_t i;
293
294 // Check for TZ_ICU_MAGIC signature at file start. If we get a
295 // signature mismatch, it means we're trying to read a file which
296 // isn't a ICU-modified-zic-created zoneinfo file. Typically this
297 // means the user is passing in a "normal" zoneinfo directory, or
298 // a zoneinfo directory that is polluted with other files, or that
299 // the user passed in the wrong directory.
300 char buf[32];
301 file.read(buf, 4);
302 if (strncmp(buf, TZ_ICU_MAGIC, 4) != 0) {
303 throw invalid_argument("TZ_ICU_MAGIC signature missing");
304 }
305 // skip additional Olson byte version
306 file.read(buf, 1);
307 // if '\0', we have just one copy of data, if '2' or '3', there is additional
308 // 64 bit version at the end.
309 if(buf[0]!=0 && buf[0]!='2' && buf[0]!='3') {
310 throw invalid_argument("Bad Olson version info");
311 }
312
313 // Read reserved bytes. The first of these will be a version byte.
314 file.read(buf, 15);
315 if (*(ICUZoneinfoVersion*)&buf != TZ_ICU_VERSION) {
316 throw invalid_argument("File version mismatch");
317 }
318
319 // Read array sizes
320 int64_t isgmtcnt = readcoded(file, 0);
321 int64_t isdstcnt = readcoded(file, 0);
322 int64_t leapcnt = readcoded(file, 0);
323 int64_t timecnt = readcoded(file, 0);
324 int64_t typecnt = readcoded(file, 0);
325 int64_t charcnt = readcoded(file, 0);
326
327 // Confirm sizes that we assume to be equal. These assumptions
328 // are drawn from a reading of the zic source (2003a), so they
329 // should hold unless the zic source changes.
330 if (isgmtcnt != typecnt || isdstcnt != typecnt) {
331 throw invalid_argument("count mismatch between tzh_ttisgmtcnt, tzh_ttisdstcnt, tth_typecnt");
332 }
333
334 // Used temporarily to store transition times and types. We need
335 // to do this because the times and types are stored in two
336 // separate arrays.
337 vector<int64_t> transitionTimes(timecnt, -1); // temporary
338 vector<int32_t> transitionTypes(timecnt, -1); // temporary
339
340 // Read transition times
341 for (i=0; i<timecnt; ++i) {
342 if (is64bitData) {
343 transitionTimes[i] = readcoded64(file);
344 } else {
345 transitionTimes[i] = readcoded(file);
346 }
347 }
348
349 // Read transition types
350 for (i=0; i<timecnt; ++i) {
351 unsigned char c;
352 file.read((char*) &c, 1);
353 int32_t t = (int32_t) c;
354 if (t < 0 || t >= typecnt) {
355 ostringstream os;
356 os << "illegal type: " << t << ", expected [0, " << (typecnt-1) << "]";
357 throw out_of_range(os.str());
358 }
359 transitionTypes[i] = t;
360 }
361
362 // Build transitions vector out of corresponding times and types.
363 bool insertInitial = false;
364 if (is64bitData && !ICU44PLUS) {
365 if (timecnt > 0) {
366 int32_t minidx = -1;
367 for (i=0; i<timecnt; ++i) {
368 if (transitionTimes[i] < LOWEST_TIME32) {
369 if (minidx == -1 || transitionTimes[i] > transitionTimes[minidx]) {
370 // Preserve the latest transition before the 32bit minimum time
371 minidx = i;
372 }
373 } else if (transitionTimes[i] > HIGHEST_TIME32) {
374 // Skipping the rest of the transition data. We cannot put such
375 // transitions into zoneinfo.res, because data is limited to singed
376 // 32bit int by the ICU resource bundle.
377 break;
378 } else {
379 info.transitions.push_back(Transition(transitionTimes[i], transitionTypes[i]));
380 }
381 }
382
383 if (minidx != -1) {
384 // If there are any transitions before the 32bit minimum time,
385 // put the type information with the 32bit minimum time
386 vector<Transition>::iterator itr = info.transitions.begin();
387 info.transitions.insert(itr, Transition(LOWEST_TIME32, transitionTypes[minidx]));
388 } else {
389 // Otherwise, we need insert the initial type later
390 insertInitial = true;
391 }
392 }
393 } else {
394 for (i=0; i<timecnt; ++i) {
395 info.transitions.push_back(Transition(transitionTimes[i], transitionTypes[i]));
396 }
397 }
398
399 // Read types (except for the isdst and isgmt flags, which come later (why??))
400 for (i=0; i<typecnt; ++i) {
401 ZoneType type;
402
403 type.rawoffset = readcoded(file);
404 type.dstoffset = readcoded(file);
405 type.isdst = readbool(file);
406
407 unsigned char c;
408 file.read((char*) &c, 1);
409 type.abbr = (int32_t) c;
410
411 if (type.isdst != (type.dstoffset != 0)) {
412 throw invalid_argument("isdst does not reflect dstoffset");
413 }
414
415 info.types.push_back(type);
416 }
417
418 assert(info.types.size() == (unsigned) typecnt);
419
420 if (insertInitial) {
421 assert(timecnt > 0);
422 assert(typecnt > 0);
423
424 int32_t initialTypeIdx = -1;
425
426 // Check if the first type is not dst
427 if (info.types.at(0).dstoffset != 0) {
428 // Initial type's rawoffset is same with the rawoffset after the
429 // first transition, but no DST is observed.
430 int64_t rawoffset0 = (info.types.at(info.transitions.at(0).type)).rawoffset;
431 // Look for matching type
432 for (i=0; i<(int32_t)info.types.size(); ++i) {
433 if (info.types.at(i).rawoffset == rawoffset0
434 && info.types.at(i).dstoffset == 0) {
435 initialTypeIdx = i;
436 break;
437 }
438 }
439 } else {
440 initialTypeIdx = 0;
441 }
442 assert(initialTypeIdx >= 0);
443 // Add the initial type associated with the lowest int32 time
444 vector<Transition>::iterator itr = info.transitions.begin();
445 info.transitions.insert(itr, Transition(LOWEST_TIME32, initialTypeIdx));
446 }
447
448
449 // Read the abbreviation string
450 if (charcnt) {
451 // All abbreviations are concatenated together, with a 0 at
452 // the end of each abbr.
453 char* str = new char[charcnt + 8];
454 file.read(str, charcnt);
455
456 // Split abbreviations apart into individual strings. Record
457 // offset of each abbr in a vector.
458 vector<int32_t> abbroffset;
459 char *limit=str+charcnt;
460 for (char* p=str; p<limit; ++p) {
461 char* start = p;
462 while (*p != 0) ++p;
463 info.abbrs.push_back(string(start, p-start));
464 abbroffset.push_back(start-str);
465 }
466
467 // Remap all the abbrs. Old value is offset into concatenated
468 // raw abbr strings. New value is index into vector of
469 // strings. E.g., 0,5,10,14 => 0,1,2,3.
470
471 // Keep track of which abbreviations get used.
472 vector<bool> abbrseen(abbroffset.size(), false);
473
474 for (vector<ZoneType>::iterator it=info.types.begin();
475 it!=info.types.end();
476 ++it) {
477 vector<int32_t>::const_iterator x=
478 find(abbroffset.begin(), abbroffset.end(), it->abbr);
479 if (x==abbroffset.end()) {
480 // TODO: Modify code to add a new string to the end of
481 // the abbr list when a middle offset is given, e.g.,
482 // "abc*def*" where * == '\0', take offset of 1 and
483 // make the array "abc", "def", "bc", and translate 1
484 // => 2. NOT CRITICAL since we don't even use the
485 // abbr at this time.
486 #if 0
487 // TODO: Re-enable this warning if we start using
488 // the Olson abbr data, or if the above TODO is completed.
489 ostringstream os;
490 os << "Warning: unusual abbr offset " << it->abbr
491 << ", expected one of";
492 for (vector<int32_t>::const_iterator y=abbroffset.begin();
493 y!=abbroffset.end(); ++y) {
494 os << ' ' << *y;
495 }
496 cerr << os.str() << "; using 0" << endl;
497 #endif
498 it->abbr = 0;
499 } else {
500 int32_t index = x - abbroffset.begin();
501 it->abbr = index;
502 abbrseen[index] = true;
503 }
504 }
505
506 for (int32_t ii=0;ii<(int32_t) abbrseen.size();++ii) {
507 if (!abbrseen[ii]) {
508 cerr << "Warning: unused abbreviation: " << ii << endl;
509 }
510 }
511 }
512
513 // Read leap second info, if any.
514 // *** We discard leap second data. ***
515 for (i=0; i<leapcnt; ++i) {
516 readcoded(file); // transition time
517 readcoded(file); // total correction after above
518 }
519
520 // Read isstd flags
521 for (i=0; i<typecnt; ++i) info.types[i].isstd = readbool(file);
522
523 // Read isgmt flags
524 for (i=0; i<typecnt; ++i) info.types[i].isgmt = readbool(file);
525 }
526
527 //--------------------------------------------------------------------
528 // Directory and file reading
529 //--------------------------------------------------------------------
530
531 /**
532 * Process a single zoneinfo file, adding the data to ZONEINFO
533 * @param path the full path to the file, e.g., ".\zoneinfo\America\Los_Angeles"
534 * @param id the zone ID, e.g., "America/Los_Angeles"
535 */
536 void handleFile(string path, string id) {
537 // Check for duplicate id
538 if (ZONEINFO.find(id) != ZONEINFO.end()) {
539 ostringstream os;
540 os << "duplicate zone ID: " << id;
541 throw invalid_argument(os.str());
542 }
543
544 ifstream file(path.c_str(), ios::in | ios::binary);
545 if (!file) {
546 throw invalid_argument("can't open file");
547 }
548
549 // eat 32bit data part
550 ZoneInfo info;
551 readzoneinfo(file, info, false);
552
553 // Check for errors
554 if (!file) {
555 throw invalid_argument("read error");
556 }
557
558 // we only use 64bit part
559 ZoneInfo info64;
560 readzoneinfo(file, info64, true);
561
562 bool alldone = false;
563 int64_t eofPos = (int64_t) file.tellg();
564
565 // '\n' + <envvar string> + '\n' after the 64bit version data
566 char ch = file.get();
567 if (ch == 0x0a) {
568 bool invalidchar = false;
569 while (file.get(ch)) {
570 if (ch == 0x0a) {
571 break;
572 }
573 if (ch < 0x20) {
574 // must be printable ascii
575 invalidchar = true;
576 break;
577 }
578 }
579 if (!invalidchar) {
580 eofPos = (int64_t) file.tellg();
581 file.seekg(0, ios::end);
582 eofPos = eofPos - (int64_t) file.tellg();
583 if (eofPos == 0) {
584 alldone = true;
585 }
586 }
587 }
588 if (!alldone) {
589 ostringstream os;
590 os << (-eofPos) << " unprocessed bytes at end";
591 throw invalid_argument(os.str());
592 }
593
594 ZONEINFO[id] = info64;
595 }
596
597 /**
598 * Recursively scan the given directory, calling handleFile() for each
599 * file in the tree. The user should call with the root directory and
600 * a prefix of "". The function will call itself with non-empty
601 * prefix values.
602 */
603 #ifdef WIN32
604
605 void scandir(string dirname, string prefix="") {
606 HANDLE hList;
607 WIN32_FIND_DATA FileData;
608
609 // Get the first file
610 hList = FindFirstFile((dirname + "\\*").c_str(), &FileData);
611 if (hList == INVALID_HANDLE_VALUE) {
612 cerr << "Error: Invalid directory: " << dirname << endl;
613 exit(1);
614 }
615 for (;;) {
616 string name(FileData.cFileName);
617 string path(dirname + "\\" + name);
618 if (FileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
619 if (name != "." && name != "..") {
620 scandir(path, prefix + name + "/");
621 }
622 } else {
623 try {
624 string id = prefix + name;
625 handleFile(path, id);
626 } catch (const exception& e) {
627 cerr << "Error: While processing \"" << path << "\", "
628 << e.what() << endl;
629 exit(1);
630 }
631 }
632
633 if (!FindNextFile(hList, &FileData)) {
634 if (GetLastError() == ERROR_NO_MORE_FILES) {
635 break;
636 } // else...?
637 }
638 }
639 FindClose(hList);
640 }
641
642 #else
643
644 void scandir(string dir, string prefix="") {
645 DIR *dp;
646 struct dirent *dir_entry;
647 struct stat stat_info;
648 char pwd[512];
649 vector<string> subdirs;
650 vector<string> subfiles;
651
652 if ((dp = opendir(dir.c_str())) == NULL) {
653 cerr << "Error: Invalid directory: " << dir << endl;
654 exit(1);
655 }
656 if (!getcwd(pwd, sizeof(pwd))) {
657 cerr << "Error: Directory name too long" << endl;
658 exit(1);
659 }
660 chdir(dir.c_str());
661 while ((dir_entry = readdir(dp)) != NULL) {
662 string name = dir_entry->d_name;
663 string path = dir + "/" + name;
664 lstat(dir_entry->d_name,&stat_info);
665 if (S_ISDIR(stat_info.st_mode)) {
666 if (name != "." && name != "..") {
667 subdirs.push_back(path);
668 subdirs.push_back(prefix + name + "/");
669 // scandir(path, prefix + name + "/");
670 }
671 } else {
672 try {
673 string id = prefix + name;
674 subfiles.push_back(path);
675 subfiles.push_back(id);
676 // handleFile(path, id);
677 } catch (const exception& e) {
678 cerr << "Error: While processing \"" << path << "\", "
679 << e.what() << endl;
680 exit(1);
681 }
682 }
683 }
684 closedir(dp);
685 chdir(pwd);
686
687 for(int32_t i=0;i<(int32_t)subfiles.size();i+=2) {
688 try {
689 handleFile(subfiles[i], subfiles[i+1]);
690 } catch (const exception& e) {
691 cerr << "Error: While processing \"" << subfiles[i] << "\", "
692 << e.what() << endl;
693 exit(1);
694 }
695 }
696 for(int32_t i=0;i<(int32_t)subdirs.size();i+=2) {
697 scandir(subdirs[i], subdirs[i+1]);
698 }
699 }
700
701 #endif
702
703 //--------------------------------------------------------------------
704 // Final zone and rule info
705 //--------------------------------------------------------------------
706
707 /**
708 * Read and discard the current line.
709 */
710 void consumeLine(istream& in) {
711 int32_t c;
712 do {
713 c = in.get();
714 } while (c != EOF && c != '\n');
715 }
716
717 enum {
718 DOM = 0,
719 DOWGEQ = 1,
720 DOWLEQ = 2
721 };
722
723 const char* TIME_MODE[] = {"w", "s", "u"};
724
725 // Allow 29 days in February because zic outputs February 29
726 // for rules like "last Sunday in February".
727 const int32_t MONTH_LEN[] = {31,29,31,30,31,30,31,31,30,31,30,31};
728
729 const int32_t HOUR = 3600;
730
731 struct FinalZone {
732 int32_t offset; // raw offset
733 int32_t year; // takes effect for y >= year
734 string ruleid;
735 set<string> aliases;
736 FinalZone(int32_t _offset, int32_t _year, const string& _ruleid) :
737 offset(_offset), year(_year), ruleid(_ruleid) {
738 if (offset <= -16*HOUR || offset >= 16*HOUR) {
739 ostringstream os;
740 os << "Invalid input offset " << offset
741 << " for year " << year
742 << " and rule ID " << ruleid;
743 throw invalid_argument(os.str());
744 }
745 if (year < 1900) {
746 ostringstream os;
747 os << "Invalid input year " << year
748 << " with offset " << offset
749 << " and rule ID " << ruleid;
750 throw invalid_argument(os.str());
751 }
752 }
753 FinalZone() : offset(-1), year(-1) {}
754 void addLink(const string& alias) {
755 if (aliases.find(alias) != aliases.end()) {
756 ostringstream os;
757 os << "Duplicate alias " << alias;
758 throw invalid_argument(os.str());
759 }
760 aliases.insert(alias);
761 }
762 };
763
764 struct FinalRulePart {
765 int32_t mode;
766 int32_t month;
767 int32_t dom;
768 int32_t dow;
769 int32_t time;
770 int32_t offset; // dst offset, usually either 0 or 1:00
771
772 // Isstd and isgmt only have 3 valid states, corresponding to local
773 // wall time, local standard time, and GMT standard time.
774 // Here is how the isstd & isgmt flags are set by zic:
775 //| case 's': /* Standard */
776 //| rp->r_todisstd = TRUE;
777 //| rp->r_todisgmt = FALSE;
778 //| case 'w': /* Wall */
779 //| rp->r_todisstd = FALSE;
780 //| rp->r_todisgmt = FALSE;
781 //| case 'g': /* Greenwich */
782 //| case 'u': /* Universal */
783 //| case 'z': /* Zulu */
784 //| rp->r_todisstd = TRUE;
785 //| rp->r_todisgmt = TRUE;
786 bool isstd;
787 bool isgmt;
788
789 bool isset; // used during building; later ignored
790
791 FinalRulePart() : isset(false) {}
792 void set(const string& id,
793 const string& _mode,
794 int32_t _month,
795 int32_t _dom,
796 int32_t _dow,
797 int32_t _time,
798 bool _isstd,
799 bool _isgmt,
800 int32_t _offset) {
801 if (isset) {
802 throw invalid_argument("FinalRulePart set twice");
803 }
804 isset = true;
805 if (_mode == "DOWLEQ") {
806 mode = DOWLEQ;
807 } else if (_mode == "DOWGEQ") {
808 mode = DOWGEQ;
809 } else if (_mode == "DOM") {
810 mode = DOM;
811 } else {
812 throw invalid_argument("Unrecognized FinalRulePart mode");
813 }
814 month = _month;
815 dom = _dom;
816 dow = _dow;
817 time = _time;
818 isstd = _isstd;
819 isgmt = _isgmt;
820 offset = _offset;
821
822 ostringstream os;
823 if (month < 0 || month >= 12) {
824 os << "Invalid input month " << month;
825 }
826 if (dom < 1 || dom > MONTH_LEN[month]) {
827 os << "Invalid input day of month " << dom;
828 }
829 if (mode != DOM && (dow < 0 || dow >= 7)) {
830 os << "Invalid input day of week " << dow;
831 }
832 if (offset < 0 || offset > (2 * HOUR)) {
833 os << "Invalid input offset " << offset;
834 }
835 if (isgmt && !isstd) {
836 os << "Invalid input isgmt && !isstd";
837 }
838 if (!os.str().empty()) {
839 os << " for rule "
840 << id
841 << _mode
842 << month << dom << dow << time
843 << isstd << isgmt
844 << offset;
845 throw invalid_argument(os.str());
846 }
847 }
848
849 /**
850 * Return the time mode as an ICU SimpleTimeZone int from 0..2;
851 * see simpletz.h.
852 */
853 int32_t timemode() const {
854 if (isgmt) {
855 assert(isstd);
856 return 2; // gmt standard
857 }
858 if (isstd) {
859 return 1; // local standard
860 }
861 return 0; // local wall
862 }
863
864 // The SimpleTimeZone encoding method for rules is as follows:
865 // stz_dowim stz_dow
866 // DOM: dom 0
867 // DOWGEQ: dom -(dow+1)
868 // DOWLEQ: -dom -(dow+1)
869 // E.g., to encode Mon>=7, use stz_dowim=7, stz_dow=-2
870 // to encode Mon<=7, use stz_dowim=-7, stz_dow=-2
871 // to encode 7, use stz_dowim=7, stz_dow=0
872 // Note that for this program and for SimpleTimeZone, 0==Jan,
873 // but for this program 0==Sun while for SimpleTimeZone 1==Sun.
874
875 /**
876 * Return a "dowim" param suitable for SimpleTimeZone.
877 */
878 int32_t stz_dowim() const {
879 return (mode == DOWLEQ) ? -dom : dom;
880 }
881
882 /**
883 * Return a "dow" param suitable for SimpleTimeZone.
884 */
885 int32_t stz_dow() const {
886 return (mode == DOM) ? 0 : -(dow+1);
887 }
888 };
889
890 struct FinalRule {
891 FinalRulePart part[2];
892
893 bool isset() const {
894 return part[0].isset && part[1].isset;
895 }
896
897 void print(ostream& os) const;
898 };
899
900 map<string,FinalZone> finalZones;
901 map<string,FinalRule> finalRules;
902
903 map<string, set<string> > links;
904 map<string, string> reverseLinks;
905 map<string, string> linkSource; // id => "Olson link" or "ICU alias"
906
907 /**
908 * Predicate used to find FinalRule objects that do not have both
909 * sub-parts set (indicating an error in the input file).
910 */
911 bool isNotSet(const pair<const string,FinalRule>& p) {
912 return !p.second.isset();
913 }
914
915 /**
916 * Predicate used to find FinalZone objects that do not map to a known
917 * rule (indicating an error in the input file).
918 */
919 bool mapsToUnknownRule(const pair<const string,FinalZone>& p) {
920 return finalRules.find(p.second.ruleid) == finalRules.end();
921 }
922
923 /**
924 * This set is used to make sure each rule in finalRules is used at
925 * least once. First we populate it with all the rules from
926 * finalRules; then we remove all the rules referred to in
927 * finaleZones.
928 */
929 set<string> ruleIDset;
930
931 void insertRuleID(const pair<string,FinalRule>& p) {
932 ruleIDset.insert(p.first);
933 }
934
935 void eraseRuleID(const pair<string,FinalZone>& p) {
936 ruleIDset.erase(p.second.ruleid);
937 }
938
939 /**
940 * Populate finalZones and finalRules from the given istream.
941 */
942 void readFinalZonesAndRules(istream& in) {
943
944 for (;;) {
945 string token;
946 in >> token;
947 if (in.eof() || !in) {
948 break;
949 } else if (token == "zone") {
950 // zone Africa/Cairo 7200 1995 Egypt # zone Africa/Cairo, offset 7200, year >= 1995, rule Egypt (0)
951 string id, ruleid;
952 int32_t offset, year;
953 in >> id >> offset >> year >> ruleid;
954 consumeLine(in);
955 finalZones[id] = FinalZone(offset, year, ruleid);
956 } else if (token == "rule") {
957 // rule US DOWGEQ 3 1 0 7200 0 0 3600 # 52: US, file data/northamerica, line 119, mode DOWGEQ, April, dom 1, Sunday, time 7200, isstd 0, isgmt 0, offset 3600
958 // rule US DOWLEQ 9 31 0 7200 0 0 0 # 53: US, file data/northamerica, line 114, mode DOWLEQ, October, dom 31, Sunday, time 7200, isstd 0, isgmt 0, offset 0
959 string id, mode;
960 int32_t month, dom, dow, time, offset;
961 bool isstd, isgmt;
962 in >> id >> mode >> month >> dom >> dow >> time >> isstd >> isgmt >> offset;
963 consumeLine(in);
964 FinalRule& fr = finalRules[id];
965 int32_t p = fr.part[0].isset ? 1 : 0;
966 fr.part[p].set(id, mode, month, dom, dow, time, isstd, isgmt, offset);
967 } else if (token == "link") {
968 string fromid, toid; // fromid == "real" zone, toid == alias
969 in >> fromid >> toid;
970 // DO NOT consumeLine(in);
971 if (finalZones.find(toid) != finalZones.end()) {
972 throw invalid_argument("Bad link: `to' id is a \"real\" zone");
973 }
974
975 links[fromid].insert(toid);
976 reverseLinks[toid] = fromid;
977
978 linkSource[fromid] = "Olson link";
979 linkSource[toid] = "Olson link";
980 } else if (token.length() > 0 && token[0] == '#') {
981 consumeLine(in);
982 } else {
983 throw invalid_argument("Unrecognized keyword");
984 }
985 }
986
987 if (!in.eof() && !in) {
988 throw invalid_argument("Parse failure");
989 }
990
991 // Perform validity check: Each rule should have data for 2 parts.
992 if (count_if(finalRules.begin(), finalRules.end(), isNotSet) != 0) {
993 throw invalid_argument("One or more incomplete rule pairs");
994 }
995
996 // Perform validity check: Each zone should map to a known rule.
997 if (count_if(finalZones.begin(), finalZones.end(), mapsToUnknownRule) != 0) {
998 throw invalid_argument("One or more zones refers to an unknown rule");
999 }
1000
1001 // Perform validity check: Each rule should be referred to by a zone.
1002 ruleIDset.clear();
1003 for_each(finalRules.begin(), finalRules.end(), insertRuleID);
1004 for_each(finalZones.begin(), finalZones.end(), eraseRuleID);
1005 if (ruleIDset.size() != 0) {
1006 throw invalid_argument("Unused rules");
1007 }
1008 }
1009
1010 //--------------------------------------------------------------------
1011 // Resource bundle output
1012 //--------------------------------------------------------------------
1013
1014 // SEE olsontz.h FOR RESOURCE BUNDLE DATA LAYOUT
1015
1016 void ZoneInfo::print(ostream& os, const string& id) const {
1017 // Implement compressed format #2:
1018 os << " /* " << id << " */ ";
1019
1020 if (aliasTo >= 0) {
1021 assert(aliases.size() == 0);
1022 os << ":int { " << aliasTo << " } "; // No endl - save room for comment.
1023 return;
1024 }
1025
1026 if (ICU44PLUS) {
1027 os << ":table {" << endl;
1028 } else {
1029 os << ":array {" << endl;
1030 }
1031
1032 vector<Transition>::const_iterator trn;
1033 vector<ZoneType>::const_iterator typ;
1034
1035 bool first;
1036
1037 if (ICU44PLUS) {
1038 trn = transitions.begin();
1039
1040 // pre 32bit transitions
1041 if (trn != transitions.end() && trn->time < LOWEST_TIME32) {
1042 os << " transPre32:intvector { ";
1043 for (first = true; trn != transitions.end() && trn->time < LOWEST_TIME32; ++trn) {
1044 if (!first) {
1045 os<< ", ";
1046 }
1047 first = false;
1048 os << (int32_t)(trn->time >> 32) << ", " << (int32_t)(trn->time & 0x00000000ffffffff);
1049 }
1050 os << " }" << endl;
1051 }
1052
1053 // 32bit transtions
1054 if (trn != transitions.end() && trn->time < HIGHEST_TIME32) {
1055 os << " trans:intvector { ";
1056 for (first = true; trn != transitions.end() && trn->time < HIGHEST_TIME32; ++trn) {
1057 if (!first) {
1058 os << ", ";
1059 }
1060 first = false;
1061 os << trn->time;
1062 }
1063 os << " }" << endl;
1064 }
1065
1066 // post 32bit transitons
1067 if (trn != transitions.end()) {
1068 os << " transPost32:intvector { ";
1069 for (first = true; trn != transitions.end(); ++trn) {
1070 if (!first) {
1071 os<< ", ";
1072 }
1073 first = false;
1074 os << (int32_t)(trn->time >> 32) << ", " << (int32_t)(trn->time & 0x00000000ffffffff);
1075 }
1076 os << " }" << endl;
1077 }
1078 } else {
1079 os << " :intvector { ";
1080 for (trn = transitions.begin(), first = true; trn != transitions.end(); ++trn) {
1081 if (!first) os << ", ";
1082 first = false;
1083 os << trn->time;
1084 }
1085 os << " }" << endl;
1086 }
1087
1088
1089 first=true;
1090 if (ICU44PLUS) {
1091 os << " typeOffsets:intvector { ";
1092 } else {
1093 os << " :intvector { ";
1094 }
1095 for (typ = types.begin(); typ != types.end(); ++typ) {
1096 if (!first) os << ", ";
1097 first = false;
1098 os << typ->rawoffset << ", " << typ->dstoffset;
1099 }
1100 os << " }" << endl;
1101
1102 if (ICU44PLUS) {
1103 if (transitions.size() != 0) {
1104 os << " typeMap:bin { \"" << hex << setfill('0');
1105 for (trn = transitions.begin(); trn != transitions.end(); ++trn) {
1106 os << setw(2) << trn->type;
1107 }
1108 os << dec << "\" }" << endl;
1109 }
1110 } else {
1111 os << " :bin { \"" << hex << setfill('0');
1112 for (trn = transitions.begin(); trn != transitions.end(); ++trn) {
1113 os << setw(2) << trn->type;
1114 }
1115 os << dec << "\" }" << endl;
1116 }
1117
1118 // Final zone info, if any
1119 if (finalYear != -1) {
1120 if (ICU44PLUS) {
1121 os << " finalRule { \"" << finalRuleID << "\" }" << endl;
1122 os << " finalRaw:int { " << finalOffset << " }" << endl;
1123 os << " finalYear:int { " << finalYear << " }" << endl;
1124 } else {
1125 os << " \"" << finalRuleID << "\"" << endl;
1126 os << " :intvector { " << finalOffset << ", "
1127 << finalYear << " }" << endl;
1128 }
1129 }
1130
1131 // Alias list, if any
1132 if (aliases.size() != 0) {
1133 first = true;
1134 if (ICU44PLUS) {
1135 os << " links:intvector { ";
1136 } else {
1137 os << " :intvector { ";
1138 }
1139 for (set<int32_t>::const_iterator i=aliases.begin(); i!=aliases.end(); ++i) {
1140 if (!first) os << ", ";
1141 first = false;
1142 os << *i;
1143 }
1144 os << " }" << endl;
1145 }
1146
1147 os << " } "; // no trailing 'endl', so comments can be placed.
1148 }
1149
1150 inline ostream&
1151 operator<<(ostream& os, const ZoneMap& zoneinfo) {
1152 int32_t c = 0;
1153 for (ZoneMapIter it = zoneinfo.begin();
1154 it != zoneinfo.end();
1155 ++it) {
1156 if(c && !ICU44PLUS) os << ",";
1157 it->second.print(os, it->first);
1158 os << "//Z#" << c++ << endl;
1159 }
1160 return os;
1161 }
1162
1163 // print the string list
1164 ostream& printStringList( ostream& os, const ZoneMap& zoneinfo) {
1165 int32_t n = 0; // count
1166 int32_t col = 0; // column
1167 os << " Names {" << endl
1168 << " ";
1169 for (ZoneMapIter it = zoneinfo.begin();
1170 it != zoneinfo.end();
1171 ++it) {
1172 if(n) {
1173 os << ",";
1174 col ++;
1175 }
1176 const string& id = it->first;
1177 os << "\"" << id << "\"";
1178 col += id.length() + 2;
1179 if(col >= 50) {
1180 os << " // " << n << endl
1181 << " ";
1182 col = 0;
1183 }
1184 n++;
1185 }
1186 os << " // " << (n-1) << endl
1187 << " }" << endl;
1188
1189 return os;
1190 }
1191
1192 //--------------------------------------------------------------------
1193 // main
1194 //--------------------------------------------------------------------
1195
1196 // Unary predicate for finding transitions after a given time
1197 bool isAfter(const Transition t, int64_t thresh) {
1198 return t.time >= thresh;
1199 }
1200
1201 /**
1202 * A zone type that contains only the raw and dst offset. Used by the
1203 * optimizeTypeList() method.
1204 */
1205 struct SimplifiedZoneType {
1206 int64_t rawoffset;
1207 int64_t dstoffset;
1208 SimplifiedZoneType() : rawoffset(-1), dstoffset(-1) {}
1209 SimplifiedZoneType(const ZoneType& t) : rawoffset(t.rawoffset),
1210 dstoffset(t.dstoffset) {}
1211 bool operator<(const SimplifiedZoneType& t) const {
1212 return rawoffset < t.rawoffset ||
1213 (rawoffset == t.rawoffset &&
1214 dstoffset < t.dstoffset);
1215 }
1216 };
1217
1218 /**
1219 * Construct a ZoneType from a SimplifiedZoneType. Note that this
1220 * discards information; the new ZoneType will have meaningless
1221 * (empty) abbr, isdst, isstd, and isgmt flags; this is appropriate,
1222 * since ignoring these is how we do optimization (we have no use for
1223 * these in historical transitions).
1224 */
1225 ZoneType::ZoneType(const SimplifiedZoneType& t) :
1226 rawoffset(t.rawoffset), dstoffset(t.dstoffset),
1227 abbr(-1), isdst(false), isstd(false), isgmt(false) {}
1228
1229 /**
1230 * Optimize the type list to remove excess entries. The type list may
1231 * contain entries that are distinct only in terms of their dst, std,
1232 * or gmt flags. Since we don't care about those flags, we can reduce
1233 * the type list to a set of unique raw/dst offset pairs, and remap
1234 * the type indices in the transition list, which stores, for each
1235 * transition, a transition time and a type index.
1236 */
1237 void ZoneInfo::optimizeTypeList() {
1238 // Assemble set of unique types; only those in the `transitions'
1239 // list, since there may be unused types in the `types' list
1240 // corresponding to transitions that have been trimmed (during
1241 // merging of final data).
1242
1243 if (aliasTo >= 0) return; // Nothing to do for aliases
1244
1245 if (!ICU44PLUS) {
1246 // This is the old logic which has a bug, which occasionally removes
1247 // the type before the first transition. The problem was fixed
1248 // by inserting the dummy transition indirectly.
1249
1250 // If there are zero transitions and one type, then leave that as-is.
1251 if (transitions.size() == 0) {
1252 if (types.size() != 1) {
1253 cerr << "Error: transition count = 0, type count = " << types.size() << endl;
1254 }
1255 return;
1256 }
1257
1258 set<SimplifiedZoneType> simpleset;
1259 for (vector<Transition>::const_iterator i=transitions.begin();
1260 i!=transitions.end(); ++i) {
1261 assert(i->type < (int32_t)types.size());
1262 simpleset.insert(types[i->type]);
1263 }
1264
1265 // Map types to integer indices
1266 map<SimplifiedZoneType,int32_t> simplemap;
1267 int32_t n=0;
1268 for (set<SimplifiedZoneType>::const_iterator i=simpleset.begin();
1269 i!=simpleset.end(); ++i) {
1270 simplemap[*i] = n++;
1271 }
1272
1273 // Remap transitions
1274 for (vector<Transition>::iterator i=transitions.begin();
1275 i!=transitions.end(); ++i) {
1276 assert(i->type < (int32_t)types.size());
1277 ZoneType oldtype = types[i->type];
1278 SimplifiedZoneType newtype(oldtype);
1279 assert(simplemap.find(newtype) != simplemap.end());
1280 i->type = simplemap[newtype];
1281 }
1282
1283 // Replace type list
1284 types.clear();
1285 copy(simpleset.begin(), simpleset.end(), back_inserter(types));
1286
1287 } else {
1288 if (types.size() > 1) {
1289 // Note: localtime uses the very first non-dst type as initial offsets.
1290 // If all types are DSTs, the very first type is treated as the initial offsets.
1291
1292 // Decide a type used as the initial offsets. ICU put the type at index 0.
1293 ZoneType initialType = types[0];
1294 for (vector<ZoneType>::const_iterator i=types.begin(); i!=types.end(); ++i) {
1295 if (i->dstoffset == 0) {
1296 initialType = *i;
1297 break;
1298 }
1299 }
1300
1301 SimplifiedZoneType initialSimplifiedType(initialType);
1302
1303 // create a set of unique types, but ignoring fields which we're not interested in
1304 set<SimplifiedZoneType> simpleset;
1305 simpleset.insert(initialSimplifiedType);
1306 for (vector<Transition>::const_iterator i=transitions.begin(); i!=transitions.end(); ++i) {
1307 assert(i->type < (int32_t)types.size());
1308 simpleset.insert(types[i->type]);
1309 }
1310
1311 // Map types to integer indices, however, keeping the first type at offset 0
1312 map<SimplifiedZoneType,int32_t> simplemap;
1313 simplemap[initialSimplifiedType] = 0;
1314 int32_t n = 1;
1315 for (set<SimplifiedZoneType>::const_iterator i=simpleset.begin(); i!=simpleset.end(); ++i) {
1316 if (*i < initialSimplifiedType || initialSimplifiedType < *i) {
1317 simplemap[*i] = n++;
1318 }
1319 }
1320
1321 // Remap transitions
1322 for (vector<Transition>::iterator i=transitions.begin();
1323 i!=transitions.end(); ++i) {
1324 assert(i->type < (int32_t)types.size());
1325 ZoneType oldtype = types[i->type];
1326 SimplifiedZoneType newtype(oldtype);
1327 assert(simplemap.find(newtype) != simplemap.end());
1328 i->type = simplemap[newtype];
1329 }
1330
1331 // Replace type list
1332 types.clear();
1333 types.push_back(initialSimplifiedType);
1334 for (set<SimplifiedZoneType>::const_iterator i=simpleset.begin(); i!=simpleset.end(); ++i) {
1335 if (*i < initialSimplifiedType || initialSimplifiedType < *i) {
1336 types.push_back(*i);
1337 }
1338 }
1339
1340 // Reiterating transitions to remove any transitions which
1341 // do not actually change the raw/dst offsets
1342 int32_t prevTypeIdx = 0;
1343 for (vector<Transition>::iterator i=transitions.begin(); i!=transitions.end();) {
1344 if (i->type == prevTypeIdx) {
1345 // this is not a time transition, probably just name change
1346 // e.g. America/Resolute after 2006 in 2010b
1347 transitions.erase(i);
1348 } else {
1349 prevTypeIdx = i->type;
1350 i++;
1351 }
1352 }
1353 }
1354 }
1355
1356 }
1357
1358 /**
1359 * Merge final zone data into this zone.
1360 */
1361 void ZoneInfo::mergeFinalData(const FinalZone& fz) {
1362 int32_t year = fz.year;
1363 int64_t seconds = yearToSeconds(year);
1364
1365 if (!ICU44PLUS) {
1366 if (seconds > HIGHEST_TIME32) {
1367 // Avoid transitions beyond signed 32bit max second.
1368 // This may result incorrect offset computation around
1369 // HIGHEST_TIME32. This is a limitation of ICU
1370 // before 4.4.
1371 seconds = HIGHEST_TIME32;
1372 }
1373 }
1374
1375 vector<Transition>::iterator it =
1376 find_if(transitions.begin(), transitions.end(),
1377 bind2nd(ptr_fun(isAfter), seconds));
1378 transitions.erase(it, transitions.end());
1379
1380 if (finalYear != -1) {
1381 throw invalid_argument("Final zone already merged in");
1382 }
1383 finalYear = fz.year;
1384 finalOffset = fz.offset;
1385 finalRuleID = fz.ruleid;
1386 }
1387
1388 /**
1389 * Merge the data from the given final zone into the core zone data by
1390 * calling the ZoneInfo member function mergeFinalData.
1391 */
1392 void mergeOne(const string& zoneid, const FinalZone& fz) {
1393 if (ZONEINFO.find(zoneid) == ZONEINFO.end()) {
1394 throw invalid_argument("Unrecognized final zone ID");
1395 }
1396 ZONEINFO[zoneid].mergeFinalData(fz);
1397 }
1398
1399 /**
1400 * Visitor function that merges the final zone data into the main zone
1401 * data structures. It calls mergeOne for each final zone and its
1402 * list of aliases.
1403 */
1404 void mergeFinalZone(const pair<string,FinalZone>& p) {
1405 const string& id = p.first;
1406 const FinalZone& fz = p.second;
1407
1408 mergeOne(id, fz);
1409 }
1410
1411 /**
1412 * Print this rule in resource bundle format to os. ID and enclosing
1413 * braces handled elsewhere.
1414 */
1415 void FinalRule::print(ostream& os) const {
1416 // First print the rule part that enters DST; then the rule part
1417 // that exits it.
1418 int32_t whichpart = (part[0].offset != 0) ? 0 : 1;
1419 assert(part[whichpart].offset != 0);
1420 assert(part[1-whichpart].offset == 0);
1421
1422 os << " ";
1423 for (int32_t i=0; i<2; ++i) {
1424 const FinalRulePart& p = part[whichpart];
1425 whichpart = 1-whichpart;
1426 os << p.month << ", " << p.stz_dowim() << ", " << p.stz_dow() << ", "
1427 << p.time << ", " << p.timemode() << ", ";
1428 }
1429 os << part[whichpart].offset << endl;
1430 }
1431
1432 int main(int argc, char *argv[]) {
1433 string rootpath, zonetab, version;
1434 bool validArgs = FALSE;
1435
1436 if (argc == 4 || argc == 5) {
1437 validArgs = TRUE;
1438 rootpath = argv[1];
1439 zonetab = argv[2];
1440 version = argv[3];
1441 if (argc == 5) {
1442 if (strcmp(argv[4], "--old") == 0) {
1443 ICU44PLUS = FALSE;
1444 TZ_RESOURCE_NAME = ICU_TZ_RESOURCE_OLD;
1445 } else {
1446 validArgs = FALSE;
1447 }
1448 }
1449 }
1450 if (!validArgs) {
1451 cout << "Usage: tz2icu <dir> <cmap> <tzver> [--old]" << endl
1452 << " <dir> path to zoneinfo file tree generated by" << endl
1453 << " ICU-patched version of zic" << endl
1454 << " <cmap> country map, from tzdata archive," << endl
1455 << " typically named \"zone.tab\"" << endl
1456 << " <tzver> version string, such as \"2003e\"" << endl
1457 << " --old generating resource format before ICU4.4" << endl;
1458 exit(1);
1459 }
1460
1461 cout << "Olson data version: " << version << endl;
1462 cout << "ICU 4.4+ format: " << (ICU44PLUS ? "Yes" : "No") << endl;
1463
1464 try {
1465 ifstream finals(ICU_ZONE_FILE);
1466 if (finals) {
1467 readFinalZonesAndRules(finals);
1468
1469 cout << "Finished reading " << finalZones.size()
1470 << " final zones and " << finalRules.size()
1471 << " final rules from " ICU_ZONE_FILE << endl;
1472 } else {
1473 cerr << "Error: Unable to open " ICU_ZONE_FILE << endl;
1474 return 1;
1475 }
1476 } catch (const exception& error) {
1477 cerr << "Error: While reading " ICU_ZONE_FILE ": " << error.what() << endl;
1478 return 1;
1479 }
1480
1481 try {
1482 // Recursively scan all files below the given path, accumulating
1483 // their data into ZONEINFO. All files must be TZif files. Any
1484 // failure along the way will result in a call to exit(1).
1485 scandir(rootpath);
1486 } catch (const exception& error) {
1487 cerr << "Error: While scanning " << rootpath << ": " << error.what() << endl;
1488 return 1;
1489 }
1490
1491 cout << "Finished reading " << ZONEINFO.size() << " zoneinfo files ["
1492 << (ZONEINFO.begin())->first << ".."
1493 << (--ZONEINFO.end())->first << "]" << endl;
1494
1495 try {
1496 for_each(finalZones.begin(), finalZones.end(), mergeFinalZone);
1497 } catch (const exception& error) {
1498 cerr << "Error: While merging final zone data: " << error.what() << endl;
1499 return 1;
1500 }
1501
1502 // Process links (including ICU aliases). For each link set we have
1503 // a canonical ID (e.g., America/Los_Angeles) and a set of one or more
1504 // aliases (e.g., PST, PST8PDT, ...).
1505
1506 // 1. Add all aliases as zone objects in ZONEINFO
1507 for (map<string,set<string> >::const_iterator i = links.begin();
1508 i!=links.end(); ++i) {
1509 const string& olson = i->first;
1510 const set<string>& aliases = i->second;
1511 if (ZONEINFO.find(olson) == ZONEINFO.end()) {
1512 cerr << "Error: Invalid " << linkSource[olson] << " to non-existent \""
1513 << olson << "\"" << endl;
1514 return 1;
1515 }
1516 for (set<string>::const_iterator j=aliases.begin();
1517 j!=aliases.end(); ++j) {
1518 ZONEINFO[*j] = ZoneInfo();
1519 }
1520 }
1521
1522 // 2. Create a mapping from zones to index numbers 0..n-1.
1523 map<string,int32_t> zoneIDs;
1524 vector<string> zoneIDlist;
1525 int32_t z=0;
1526 for (ZoneMap::iterator i=ZONEINFO.begin(); i!=ZONEINFO.end(); ++i) {
1527 zoneIDs[i->first] = z++;
1528 zoneIDlist.push_back(i->first);
1529 }
1530 assert(z == (int32_t) ZONEINFO.size());
1531
1532 // 3. Merge aliases. Sometimes aliases link to other aliases; we
1533 // resolve these into simplest possible sets.
1534 map<string,set<string> > links2;
1535 map<string,string> reverse2;
1536 for (map<string,set<string> >::const_iterator i = links.begin();
1537 i!=links.end(); ++i) {
1538 string olson = i->first;
1539 while (reverseLinks.find(olson) != reverseLinks.end()) {
1540 olson = reverseLinks[olson];
1541 }
1542 for (set<string>::const_iterator j=i->second.begin(); j!=i->second.end(); ++j) {
1543 links2[olson].insert(*j);
1544 reverse2[*j] = olson;
1545 }
1546 }
1547 links = links2;
1548 reverseLinks = reverse2;
1549
1550 if (false) { // Debugging: Emit link map
1551 for (map<string,set<string> >::const_iterator i = links.begin();
1552 i!=links.end(); ++i) {
1553 cout << i->first << ": ";
1554 for (set<string>::const_iterator j=i->second.begin(); j!=i->second.end(); ++j) {
1555 cout << *j << ", ";
1556 }
1557 cout << endl;
1558 }
1559 }
1560
1561 // 4. Update aliases
1562 for (map<string,set<string> >::const_iterator i = links.begin();
1563 i!=links.end(); ++i) {
1564 const string& olson = i->first;
1565 const set<string>& aliases = i->second;
1566 ZONEINFO[olson].clearAliases();
1567 ZONEINFO[olson].addAlias(zoneIDs[olson]);
1568 for (set<string>::const_iterator j=aliases.begin();
1569 j!=aliases.end(); ++j) {
1570 assert(zoneIDs.find(olson) != zoneIDs.end());
1571 assert(zoneIDs.find(*j) != zoneIDs.end());
1572 assert(ZONEINFO.find(*j) != ZONEINFO.end());
1573 ZONEINFO[*j].setAliasTo(zoneIDs[olson]);
1574 ZONEINFO[olson].addAlias(zoneIDs[*j]);
1575 }
1576 }
1577
1578 // Once merging of final data is complete, we can optimize the type list
1579 for (ZoneMap::iterator i=ZONEINFO.begin(); i!=ZONEINFO.end(); ++i) {
1580 i->second.optimizeTypeList();
1581 }
1582
1583 // Create the country map
1584 map<string, string> icuRegions; // ICU's custom zone -> country override
1585 map<string, set<string> > countryMap; // country -> set of zones
1586 map<string, string> reverseCountryMap; // zone -> country
1587
1588 try {
1589 // Read icuregions file to collect ICU's own zone-region mapping data.
1590 ifstream frg(ICU_REGIONS);
1591 if (frg) {
1592 string line;
1593 while (getline(frg, line)) {
1594 if (line[0] == '#') continue;
1595
1596 string zone, country;
1597 istringstream is(line);
1598 is >> zone >> country;
1599 if (zone.size() == 0) continue;
1600 if (country.size() < 2) {
1601 cerr << "Error: Can't parse " << line << " in " << ICU_REGIONS << endl;
1602 return 1;
1603 }
1604 icuRegions[zone] = country;
1605 }
1606 } else {
1607 cout << "No custom region map [icuregions]" << endl;
1608 }
1609 } catch (const exception& error) {
1610 cerr << "Error: While reading " << ICU_REGIONS << ": " << error.what() << endl;
1611 return 1;
1612 }
1613
1614 try {
1615 ifstream f(zonetab.c_str());
1616 if (!f) {
1617 cerr << "Error: Unable to open " << zonetab << endl;
1618 return 1;
1619 }
1620 int32_t n = 0;
1621 string line;
1622 while (getline(f, line)) {
1623 string::size_type lb = line.find('#');
1624 if (lb != string::npos) {
1625 line.resize(lb); // trim comments
1626 }
1627 string country, coord, zone;
1628 istringstream is(line);
1629 is >> country >> coord >> zone;
1630 if (country.size() == 0) continue;
1631 if (country.size() != 2 || zone.size() < 1) {
1632 cerr << "Error: Can't parse " << line << " in " << zonetab << endl;
1633 return 1;
1634 }
1635 if (ZONEINFO.find(zone) == ZONEINFO.end()) {
1636 cerr << "Error: Country maps to invalid zone " << zone
1637 << " in " << zonetab << endl;
1638 return 1;
1639 }
1640 if (icuRegions.find(zone) != icuRegions.end()) {
1641 // Custom override
1642 string customCountry = icuRegions[zone];
1643 cout << "Region Mapping: custom override for " << zone
1644 << " " << country << " -> " << customCountry << endl;
1645 country = customCountry;
1646 }
1647 countryMap[country].insert(zone);
1648 reverseCountryMap[zone] = country;
1649 //cerr << (n+1) << ": " << country << " <=> " << zone << endl;
1650 ++n;
1651 }
1652 cout << "Finished reading " << n
1653 << " country entries from " << zonetab << endl;
1654 } catch (const exception& error) {
1655 cerr << "Error: While reading " << zonetab << ": " << error.what() << endl;
1656 return 1;
1657 }
1658
1659 // Merge ICU's own zone-region mapping data
1660 for (map<string,string>::const_iterator i = icuRegions.begin();
1661 i != icuRegions.end(); ++i) {
1662 const string& zid(i->first);
1663 if (reverseCountryMap.find(zid) != reverseCountryMap.end()) {
1664 continue;
1665 }
1666 cout << "Region Mapping: custom data zone=" << zid
1667 << ", region=" << i->second << endl;
1668
1669 reverseCountryMap[zid] = i->second;
1670 countryMap[i->second].insert(zid);
1671 }
1672
1673 // Merge ICU aliases into country map. Don't merge any alias
1674 // that already has a country map, since that doesn't make sense.
1675 // E.g. "Link Europe/Oslo Arctic/Longyearbyen" doesn't mean we
1676 // should cross-map the countries between these two zones.
1677 for (map<string,set<string> >::const_iterator i = links.begin();
1678 i!=links.end(); ++i) {
1679 const string& olson(i->first);
1680 if (reverseCountryMap.find(olson) == reverseCountryMap.end()) {
1681 continue;
1682 }
1683 string c = reverseCountryMap[olson];
1684 const set<string>& aliases(i->second);
1685 for (set<string>::const_iterator j=aliases.begin();
1686 j != aliases.end(); ++j) {
1687 if (reverseCountryMap.find(*j) == reverseCountryMap.end()) {
1688 countryMap[c].insert(*j);
1689 reverseCountryMap[*j] = c;
1690 //cerr << "Aliased country: " << c << " <=> " << *j << endl;
1691 }
1692 }
1693 }
1694
1695 // Create a pseudo-country containing all zones belonging to no country
1696 set<string> nocountry;
1697 for (ZoneMap::iterator i=ZONEINFO.begin(); i!=ZONEINFO.end(); ++i) {
1698 if (reverseCountryMap.find(i->first) == reverseCountryMap.end()) {
1699 nocountry.insert(i->first);
1700 }
1701 }
1702 countryMap[""] = nocountry;
1703
1704 // Get local time & year for below
1705 time_t sec;
1706 time(&sec);
1707 struct tm* now = localtime(&sec);
1708 int32_t thisYear = now->tm_year + 1900;
1709
1710 string filename = TZ_RESOURCE_NAME + ".txt";
1711 // Write out a resource-bundle source file containing data for
1712 // all zones.
1713 ofstream file(filename.c_str());
1714 if (file) {
1715 file << "//---------------------------------------------------------" << endl
1716 << "// Copyright (C) 2003";
1717 if (thisYear > 2003) {
1718 file << "-" << thisYear;
1719 }
1720 file << ", International Business Machines" << endl
1721 << "// Corporation and others. All Rights Reserved." << endl
1722 << "//---------------------------------------------------------" << endl
1723 << "// Build tool: tz2icu" << endl
1724 << "// Build date: " << asctime(now) /* << endl -- asctime emits CR */
1725 << "// tz database: ftp://ftp.iana.org/tz/" << endl
1726 << "// tz version: " << version << endl
1727 << "// ICU version: " << U_ICU_VERSION << endl
1728 << "//---------------------------------------------------------" << endl
1729 << "// >> !!! >> THIS IS A MACHINE-GENERATED FILE << !!! <<" << endl
1730 << "// >> !!! >>> DO NOT EDIT <<< !!! <<" << endl
1731 << "//---------------------------------------------------------" << endl
1732 << endl
1733 << TZ_RESOURCE_NAME << ":table(nofallback) {" << endl
1734 << " TZVersion { \"" << version << "\" }" << endl
1735 << " Zones:array { " << endl
1736 << ZONEINFO // Zones (the actual data)
1737 << " }" << endl;
1738
1739 // Names correspond to the Zones list, used for binary searching.
1740 printStringList ( file, ZONEINFO ); // print the Names list
1741
1742 // Final Rules are used if requested by the zone
1743 file << " Rules { " << endl;
1744 // Emit final rules
1745 int32_t frc = 0;
1746 for(map<string,FinalRule>::iterator i=finalRules.begin();
1747 i!=finalRules.end(); ++i) {
1748 const string& id = i->first;
1749 const FinalRule& r = i->second;
1750 file << " " << id << ":intvector {" << endl;
1751 r.print(file);
1752 file << " } //_#" << frc++ << endl;
1753 }
1754 file << " }" << endl;
1755
1756 // Emit country (region) map.
1757 if (ICU44PLUS) {
1758 file << " Regions:array {" << endl;
1759 int32_t zn = 0;
1760 for (ZoneMap::iterator i=ZONEINFO.begin(); i!=ZONEINFO.end(); ++i) {
1761 map<string, string>::iterator cit = reverseCountryMap.find(i->first);
1762 if (cit == reverseCountryMap.end()) {
1763 file << " \"001\",";
1764 } else {
1765 file << " \"" << cit->second << "\", ";
1766 }
1767 file << "//Z#" << zn++ << " " << i->first << endl;
1768 }
1769 file << " }" << endl;
1770 } else {
1771 file << " Regions { " << endl;
1772 int32_t rc = 0;
1773 for (map<string, set<string> >::const_iterator i=countryMap.begin();
1774 i != countryMap.end(); ++i) {
1775 string country = i->first;
1776 const set<string>& zones(i->second);
1777 file << " ";
1778 if(country[0]==0) {
1779 file << "Default";
1780 }
1781 file << country << ":intvector { ";
1782 bool first = true;
1783 for (set<string>::const_iterator j=zones.begin();
1784 j != zones.end(); ++j) {
1785 if (!first) file << ", ";
1786 first = false;
1787 if (zoneIDs.find(*j) == zoneIDs.end()) {
1788 cerr << "Error: Nonexistent zone in country map: " << *j << endl;
1789 return 1;
1790 }
1791 file << zoneIDs[*j]; // emit the zone's index number
1792 }
1793 file << " } //R#" << rc++ << endl;
1794 }
1795 file << " }" << endl;
1796 }
1797
1798 file << "}" << endl;
1799 }
1800
1801 file.close();
1802
1803 if (file) { // recheck error bit
1804 cout << "Finished writing " << TZ_RESOURCE_NAME << ".txt" << endl;
1805 } else {
1806 cerr << "Error: Unable to open/write to " << TZ_RESOURCE_NAME << ".txt" << endl;
1807 return 1;
1808 }
1809 }
1810 //eof