开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 图形图像 > OpenGL > 查看源码
packdb.cpp
资源名称：celestia-1.6.0.tar.gz　[点击查看]
上传用户：center1979
上传日期：2022-07-26
资源大小：50633k
文件大小：10k
源码类别：

OpenGL

开发平台：

Visual C++

packdb.cpp：源码内容
  1. // packdb.cpp
  2. //
  3. // Copyright (C) 2001, Chris Laurel <claurel@shatters.net>
  4. //
  5. // This program is free software; you can redistribute it and/or
  6. // modify it under the terms of the GNU General Public License
  7. // as published by the Free Software Foundation; either version 2
  8. // of the License, or (at your option) any later version.
  10. #include <stdlib.h>
  11. #include <stdio.h>
  12. #include <string.h>
  14. #define SPECTRAL_O 0
  15. #define SPECTRAL_B 1
  16. #define SPECTRAL_A 2
  17. #define SPECTRAL_F 3
  18. #define SPECTRAL_G 4
  19. #define SPECTRAL_K 5
  20. #define SPECTRAL_M 6
  21. #define SPECTRAL_R 7
  22. #define SPECTRAL_S 8
  23. #define SPECTRAL_N 9
  24. #define SPECTRAL_WC 10
  25. #define SPECTRAL_WN 11
  27. // Stellar remnants
  28. #define SPECTRAL_WHITE_DWARF  16
  29. #define SPECTRAL_NEUTRON_STAR 32
  31. #define SPECTRAL_UNKNOWN  255
  33. #define LUM_Ia0    0
  34. #define LUM_Ia     1
  35. #define LUM_Ib     2
  36. #define LUM_II     3
  37. #define LUM_III    4
  38. #define LUM_IV     5
  39. #define LUM_V      6
  40. #define LUM_VI     7
  42. #define ID_NONE -1
  44. #define HD_CATALOG          0x00000000
  45. #define HIPPARCOS_CATALOG   0x10000000
  47. #define BINWRITE(fp, n) fwrite(&(n), sizeof(n), 1, (fp))
  50. typedef struct {
  51.     char colorType;
  52.     char subType;
  53.     char luminosity;
  54.     float colorIndex;
  55. } SpectralType;
  57. typedef struct {
  58.     int HIP;          /* HIPPARCOS catalogue number */
  59.     int HD;           /* HD catalogue number */
  60.     float appMag;     /* Apparent magnitude  */
  61.     float RA;         /* 0 -- 24 hours       */
  62.     float dec;        /* -90 -- +90 degrees  */
  63.     float parallax;   /* in milliarcseconds  */
  64.     char spectral[13];
  65.     unsigned char parallaxError;
  66. } Star;
  68. typedef struct {
  69.     int HD;
  70.     char *commonName;
  71.     char *altName;
  72. } HDNameEnt;
  74. /* Hardcoded file names */
  75. #define HIPPARCOS_MAIN_DB "hip_main.dat"
  76. #define COMMON_NAMES_DB   "hdnames.dat"
  78. HDNameEnt *hdNames;
  79. int nHDNames;
  80. Star *Stars;
  81. int nStars;
  84. int CompareHDNameEnt(const void *a, const void *b)
  85. {
  86.     int hda = ((HDNameEnt *) a)->HD;
  87.     int hdb = ((HDNameEnt *) b)->HD;
  89.     if (hda <  hdb)
  90. return -1;
  91.     else if (hda > hdb)
  92. return 1;
  93.     else
  94. return 0;
  95. }
  98. unsigned short PackSpectralType(char *spectralType)
  99. {
  100.     unsigned short packed = 0;
  101.     unsigned short letter;
  102.     unsigned short number;
  103.     unsigned short luminosity = LUM_V;
  104.     int i = 0;
  106.     // Subdwarves (luminosity class VI) are prefixed with sd
  107.     if (spectralType[i] == 's' && spectralType[i + 1] == 'd')
  108.     {
  109.         luminosity = LUM_VI;
  110.         i += 2;
  111.     }
  113.     switch (spectralType[i])
  114.     {
  115.       case 'O':
  116.         letter = SPECTRAL_O;
  117.         break;
  118.       case 'B':
  119.         letter = SPECTRAL_B;
  120.         break;
  121.       case 'A':
  122.         letter = SPECTRAL_A;
  123.         break;
  124.       case 'F':
  125.         letter = SPECTRAL_F;
  126.         break;
  127.       case 'G':
  128.         letter = SPECTRAL_G;
  129.         break;
  130.       case 'K':
  131.         letter = SPECTRAL_K;
  132.         break;
  133.       case 'M':      
  134.         letter = SPECTRAL_M;
  135.         break;
  136.       case 'R':
  137.         letter = SPECTRAL_R;
  138.         break;
  139.       case 'N':
  140.         letter = SPECTRAL_N;
  141.         break;
  142.       case 'S':
  143.         letter = SPECTRAL_S;
  144.         break;
  145.       case 'W':
  146. i++;
  147. if (spectralType[i] == 'C')
  148.       letter = SPECTRAL_WC;
  149. else if (spectralType[i] == 'N')
  150.     letter = SPECTRAL_WN;
  151. else
  152.     i--;
  153. break;
  154.       case 'D':
  155.         letter = SPECTRAL_WHITE_DWARF;
  156. break;
  157.       default:
  158.         letter = SPECTRAL_UNKNOWN;
  159.         break;
  160.     }
  162.     if (letter == SPECTRAL_WHITE_DWARF)
  163.         return letter << 8;
  165.     i++;
  166.     if (spectralType[i] >= '0' && spectralType[i] <= '9')
  167.         number = spectralType[i] - '0';
  168.     else
  169.         number = 0;
  171.     if (luminosity != LUM_VI)
  172.     {
  173.         i++;
  174.         luminosity = LUM_V;
  175.         while (i < 13 && spectralType[i] != '') {
  176.             if (spectralType[i] == 'I') {
  177.                 if (spectralType[i + 1] == 'I') {
  178.                     if (spectralType[i + 2] == 'I') {
  179.                         luminosity = LUM_III;
  180.                     } else {
  181.                         luminosity = LUM_II;
  182.                     }
  183.                 } else if (spectralType[i + 1] == 'V') {
  184.                     luminosity = LUM_IV;
  185.                 } else if (spectralType[i + 1] == 'a') {
  186.                     if (spectralType[i + 2] == '0')
  187.                         luminosity = LUM_Ia0;
  188.                     else
  189.                         luminosity = LUM_Ia;
  190.                 } else if (spectralType[i + 1] == 'b') {
  191.                     luminosity = LUM_Ib;
  192.                 } else {
  193.                     luminosity = LUM_Ib;
  194.                 }
  195.                 break;
  196.             } else if (spectralType[i] == 'V') {
  197.                 if (spectralType[i + 1] == 'I')
  198.                     luminosity = LUM_VI;
  199.                 else
  200.                     luminosity = LUM_V;
  201.                 break;
  202.             }
  203.             i++;
  204.         }
  205.     }
  207.     return (letter << 8) | (number << 4) | luminosity;
  208. }
  211. HDNameEnt *ReadCommonNames(FILE *fp)
  212. {
  213.     char buf[256];
  214.     int i;
  216.     hdNames = (HDNameEnt *) malloc(sizeof(HDNameEnt) * 3000);
  217.     if (hdNames == NULL)
  218. return NULL;
  220.     for (i = 0; ; i++) {
  221. char *name1, *name2, len;
  222. if (fgets(buf, 256, fp) == NULL)
  223.     break;
  225. /* Strip trailing newline */
  226. len = strlen(buf);
  227. if (len > 0 && buf[len - 1] == 'n')
  228.     buf[len - 1] = '';
  230. name1 = index(buf, ':');
  231. if (name1 == NULL)
  232.     break;
  233. name1[0] = '';
  234. name1++;
  235. name2 = index(name1, ':');
  236. if (name2 == NULL)
  237.     break;
  238. name2[0] = '';
  239. name2++;
  241. {
  242.     int hd;
  243.     if (sscanf(buf, "%d", &hd) != 1)
  244. break;
  245.     hdNames[i].HD = hd;
  246.     if (name1[0] != '') {
  247.         hdNames[i].commonName = (char *) malloc(strlen(name1) + 1);
  248. strcpy(hdNames[i].commonName, name1);
  249.     } else if (name2[0] != '') {
  250. hdNames[i].commonName = (char *) malloc(strlen(name2) + 1);
  251. strcpy(hdNames[i].commonName, name2);
  252.     }
  253. }
  254.     }
  256.     nHDNames = i;
  257.     qsort(hdNames, nHDNames, sizeof(HDNameEnt), CompareHDNameEnt);
  259.     return hdNames;
  260. }
  263. char *LookupName(int HD)
  264. {
  265.     HDNameEnt key;
  266.     HDNameEnt *found;
  268.     key.HD = HD;
  269.     found = (HDNameEnt *) bsearch((void *) &key, (void *) hdNames,
  270.   nHDNames, sizeof(HDNameEnt),
  271.   CompareHDNameEnt);
  272.     if (found == NULL)
  273. return NULL;
  274.     else
  275. return found->commonName;
  276. }
  279. void WriteStar(FILE *fp, Star *star)
  280. {
  281.     unsigned short spectralType = PackSpectralType(star->spectral);
  282.     short appMag = (short) (star->appMag * 256);
  283.     unsigned int catalog_no;
  285.     if (star->HD != ID_NONE)
  286.         catalog_no = star->HD | HD_CATALOG;
  287.     else
  288.         catalog_no = star->HIP | HIPPARCOS_CATALOG;
  289.     BINWRITE(fp, catalog_no);
  290.     BINWRITE(fp, star->RA);
  291.     BINWRITE(fp, star->dec);
  292.     BINWRITE(fp, star->parallax);
  293.     BINWRITE(fp, appMag);
  294.     BINWRITE(fp, spectralType);
  295.     BINWRITE(fp, star->parallaxError);
  296. }
  298. #define HIPPARCOS_RECORD_LENGTH 512
  300. Star *ReadHipparcosCatalog(FILE *fp)
  301. {
  302.     char buf[HIPPARCOS_RECORD_LENGTH];
  303.     int i;
  304.     int maxStars = 120000;
  305.     Star *stars;
  306.     int nBright = 0, nGood = 0;
  307.     char nameBuf[20];
  309.     fprintf(stderr, "Attempting to allocate %d bytesn", maxStars * sizeof(Star));
  310.     stars = (Star *) malloc(maxStars * sizeof(Star));
  311.     if (stars == NULL) {
  312.         fprintf(stderr, "Unable to allocate memory for stars.n");
  313. return NULL;
  314.     }
  316.     for (i = 0; ; i++) {
  317. int hh, mm, deg;
  318. float seconds;
  319. float parallaxError;
  320. char degSign;
  322. if (fgets(buf, HIPPARCOS_RECORD_LENGTH, fp) == NULL)
  323.     break;
  324. sscanf(buf + 2, "%d", &stars[i].HIP);
  325. if (sscanf(buf + 390, "%d", &stars[i].HD) != 1)
  326.     stars[i].HD = ID_NONE;
  327. sscanf(buf + 41, "%f", &stars[i].appMag);
  328. sscanf(buf + 79, "%f", &stars[i].parallax);
  329. sscanf(buf + 17, "%d %d %f", &hh, &mm, &seconds);
  330. stars[i].RA = hh + (float) mm / 60.0f + (float) seconds / 3600.0f;
  331. sscanf(buf + 29, "%c%d %d %f", &degSign, &deg, &mm, &seconds);
  332. stars[i].dec = deg + (float) mm / 60.0f + (float) seconds / 3600.0f;
  333. if (degSign == '-')
  334.     stars[i].dec = -stars[i].dec;
  335. sscanf(buf + 435, "%12s", &stars[i].spectral);
  336. sscanf(buf + 119, "%f", &parallaxError);
  337. if (stars[i].parallax <= 0 || parallaxError / stars[i].parallax > 1)
  338.         {
  339.     stars[i].parallaxError = (unsigned char) 255;
  340. }
  341. else
  342.         {
  343.     stars[i].parallaxError =
  344. (unsigned char) (parallaxError / stars[i].parallax * 200);
  345. }
  347. if (/* stars[i].appMag < 4.0f */
  348.             stars[i].parallax > 0 && 3260 / stars[i].parallax < 20) {
  349.     nBright++;
  350. #if 0
  351.     if (parallaxError / stars[i].parallax > 0.25f ||
  352. parallaxError / stars[i].parallax < 0.0f) {
  353. #endif
  354.     if (0) {
  355. char *name = LookupName(stars[i].HD);
  357. if (name == NULL) {
  358.     if (stars[i].HD != ID_NONE) {
  359. sprintf(nameBuf, "HD%d", stars[i].HD);
  360. name = nameBuf;
  361.     } else {
  362. sprintf(nameBuf, "HIP%d", stars[i].HIP);
  363. name = nameBuf;
  364.     }
  365. }
  366. printf("%-20s %5.2f %6.2f %3d%% %12s %5.2f %5.2fn",
  367.        name,
  368.        stars[i].appMag,
  369.        3260.0f / stars[i].parallax,
  370.        (int) (100.0f * parallaxError / stars[i].parallax),
  371.        stars[i].spectral,
  372.        stars[i].RA, stars[i].dec);
  373.     } else {
  374. nGood++;
  375.     }
  376. }
  377.     }
  379.     nStars = i;
  380.     printf("Stars: %d, Bright: %d, Good: %dn", nStars, nBright, nGood);
  382.     return stars;
  383. }
  386. int main(int argc, char *argv[])
  387. {
  388.     FILE *fp;
  390.     fp = fopen(COMMON_NAMES_DB, "r");
  391.     if (fp == NULL) {
  392. fprintf(stderr, "Error opening %sn", COMMON_NAMES_DB);
  393. return 1;
  394.     }
  395.     hdNames = ReadCommonNames(fp);
  396.     fclose(fp);
  397.     fp = NULL;
  398.     if (hdNames == NULL) {
  399. fprintf(stderr, "Error reading names file.n");
  400. return 1;
  401.     }
  403.     fp = fopen(HIPPARCOS_MAIN_DB, "r");
  404.     if (fp == NULL) {
  405. fprintf(stderr, "Error opening %sn", HIPPARCOS_MAIN_DB);
  406. return 1;
  407.     }
  408.     Stars = ReadHipparcosCatalog(fp);
  409.     fclose(fp);
  410.     if (Stars == NULL) {
  411. fprintf(stderr, "Error reading HIPPARCOS database.");
  412. return 1;
  413.     }
  414. #if 0
  415.     {
  416. int i;
  418. for (i = 0; i < nStars; i++) {
  419.     if (Stars[i].spectral[0] == 'O') {
  420. char *name = LookupName(Stars[i].HD);
  421. if (name != NULL)
  422.     printf("%s ", name);
  423. printf("%6d %6.3f %6.3f %sn",
  424.        Stars[i].HD, Stars[i].RA, Stars[i].dec, Stars[i].spectral);
  425.     }
  426. }
  427.     }
  428. #endif
  430.     {
  431.         int i;
  433.         FILE *fp = fopen("out", "wb");
  434.         if (fp == NULL) {
  435.    fprintf(stderr, "Error opening output file.n");
  436.            exit(1);
  437.         }
  439. BINWRITE(fp, nStars);
  440.         for (i = 0; i < nStars; i++)
  441.            WriteStar(fp, &Stars[i]);
  442.         
  443.         fclose(fp);
  444.     }
  445.     printf("Stars in catalog = %dn", nStars);
  446. }