File: cut-table.c

package info (click to toggle)
astrometry.net 0.93%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 19,372 kB
  • sloc: ansic: 163,192; python: 18,357; makefile: 1,522; sh: 138; cpp: 78; pascal: 67; awk: 56; perl: 9
file content (108 lines) | stat: -rw-r--r-- 2,644 bytes parent folder | download
1
2
3
4
5
6
7
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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
/*
 # This file is part of the Astrometry.net suite.
 # Licensed under a 3-clause BSD style license - see LICENSE
 */

#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <sys/types.h>
#include <sys/mman.h>

#include "fitstable.h"
#include "anqfits.h"
#include "ioutils.h"
#include "fitsioutils.h"
#include "errors.h"
#include "log.h"

int cut_table(const char* infn, const char* outfn, int N) {
    fitstable_t* in;
    //fitstable_t* out;
    FILE* fid;
    FILE* fin;
    int i, Next;

    in = fitstable_open(infn);
    if (!in) {
        ERROR("Failed to read input file %s", infn);
        return -1;
    }
    /*
     out = fitstable_open_for_writing(outfn);
     if (!out) {
     ERROR("Failed to out output file %s", outfn);
     return -1;
     }
     */
    fid = fopen(outfn, "wb");
    if (!fid) {
        ERROR("Failed to open output file %s", outfn);
        return -1;
    }

    fin = fopen(infn, "rb");
    if (!fin) {
        ERROR("Failed to open input file %s", infn);
        return -1;
    }

    /*
     fitstable_set_primary_header(out, fitstable_get_primary_header(in));
     if (fitstable_write_primary_header(out)) {
     ERROR("Failed to write primary header");
     return -1;
     }
     */

    if (qfits_header_dump(fitstable_get_primary_header(in), fid)) {
        ERROR("Failed to write primary header");
        return -1;
    }
	
    Next = fitstable_n_extensions(in);
    logverb("N extensions: %i\n", Next);
    for (i=1; i<Next; i++) {
        qfits_header* hdr = fitstable_get_header(in);
        int width, rows;

        width = qfits_header_getint(hdr, "NAXIS1", 0);
        rows = qfits_header_getint(hdr, "NAXIS2", 0);

        if (N < rows)
            rows = N;

        fits_header_mod_int(hdr, "NAXIS2", rows, "number of rows in table");
        if (qfits_header_dump(hdr, fid)) {
            ERROR("Failed to write HDU %i header", i);
            return -1;
        }

        if (rows * width) {
            int offset = in->table->col[0].off_beg;
            if (pipe_file_offset(fin, offset, (size_t)rows * (size_t)width, fid) ||
                fits_pad_file(fid)) {
                ERROR("Failed to write HDU %i data", i);
                return -1;
            }
        }

        if (i < Next-1)
            if (fitstable_open_next_extension(in)) {
                ERROR("Failed to open extension %i", i+1);
                return -1;
            }
    }
	
    if (fclose(fid)) {
        ERROR("Failed to close output file %s", outfn);
        return -1;
    }

    fclose(fin);
    fitstable_close(in);
    //fitstable_close(out);
    return 0;
}