File: plotmatch.c

package info (click to toggle)
astrometry.net 0.98%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 19,428 kB
  • sloc: ansic: 165,791; python: 18,438; makefile: 1,560; sh: 157; cpp: 78; pascal: 67; awk: 56; perl: 9
file content (123 lines) | stat: -rw-r--r-- 3,394 bytes parent folder | download | duplicates (5)
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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
/*
 # This file is part of the Astrometry.net suite.
 # Licensed under a 3-clause BSD style license - see LICENSE
 */
#include <string.h>
#include <math.h>

#include "plotmatch.h"
#include "cairoutils.h"
#include "ioutils.h"
#include "log.h"
#include "errors.h"
#include "permutedsort.h"
#include "matchfile.h"

DEFINE_PLOTTER(match);

plotmatch_t* plot_match_get(plot_args_t* pargs) {
    return plotstuff_get_config(pargs, "match");
}

void* plot_match_init(plot_args_t* plotargs) {
    plotmatch_t* args = calloc(1, sizeof(plotmatch_t));
    args->matches = bl_new(16, sizeof(MatchObj));
    return args;
}

int plot_match_plot(const char* command,
                    cairo_t* cairo, plot_args_t* pargs, void* baton) {
    plotmatch_t* args = (plotmatch_t*)baton;
    int i;
    anbool failed = FALSE;

    plotstuff_builtin_apply(cairo, pargs);

    for (i=0; i<bl_size(args->matches); i++) {
        MatchObj* mo = bl_access(args->matches, i);
        double xy[DQMAX*2];
        double theta[DQMAX];
        int perm[DQMAX];
        double cx,cy;
        int j;
        double x,y;
        //int order[] = {0,2,1,3,4,5};
        cx = cy = 0;
        for (j=0; j<mo->dimquads; j++) {
            double ra,dec;
            //xyzarr2radecdeg(mo->quadxyz + order[j]*3, &ra, &dec);
            xyzarr2radecdeg(mo->quadxyz + j*3, &ra, &dec);
            if (!plotstuff_radec2xy(pargs, ra, dec, &x, &y)) {
                failed = TRUE;
                break;
            }
            xy[2*j + 0] = x;
            xy[2*j + 1] = y;
            cx += x;
            cy += y;
        }
        if (failed)
            continue;

        // (In the grand tradition of infinite copy-n-paste...)
        // Make the quad convex so Sam's eyes don't bleed.
        cx /= mo->dimquads;
        cy /= mo->dimquads;
        for (j=0; j<mo->dimquads; j++)
            theta[j] = atan2(xy[2*j + 1] - cy, xy[2*j + 0] - cx);
        permutation_init(perm, mo->dimquads);
        permuted_sort(theta, sizeof(double), compare_doubles_asc, perm, mo->dimquads);

        for (j=0; j<mo->dimquads; j++) {
            x = xy[2*perm[j]+0];
            y = xy[2*perm[j]+1];
            if (j == 0)
                cairo_move_to(cairo, x, y);
            else
                cairo_line_to(cairo, x, y);
        }
        cairo_close_path(cairo);
        cairo_stroke(cairo);
    }
    return 0;
}

int plot_match_add_match(plotmatch_t* args, const MatchObj* mo) {
    bl_append(args->matches, mo);
    return 0;
}

int plot_match_set_filename(plotmatch_t* args, const char* filename) {
    matchfile* mf = matchfile_open(filename);
    MatchObj* mo;
    if (!mf) {
        ERROR("Failed to open matchfile \"%s\"", filename);
        return -1;
    }
    while (1) {
        mo = matchfile_read_match(mf);
        if (!mo)
            break;
        plot_match_add_match(args, mo);
    }
    return 0;
}

int plot_match_command(const char* cmd, const char* cmdargs,
                       plot_args_t* pargs, void* baton) {
    plotmatch_t* args = (plotmatch_t*)baton;
    if (streq(cmd, "match_file")) {
        plot_match_set_filename(args, cmdargs);
    } else {
        ERROR("Did not understand command \"%s\"", cmd);
        return -1;
    }
    return 0;
}

void plot_match_free(plot_args_t* plotargs, void* baton) {
    plotmatch_t* args = (plotmatch_t*)baton;
    bl_free(args->matches);
    free(args);
}