Home My Page Projects Code Snippets Project Openings diderot
Summary Activity Tracker Tasks SCM

SCM Repository

[diderot] Annotation of /branches/pure-cfg/src/include/Diderot/diderot.h
ViewVC logotype

Annotation of /branches/pure-cfg/src/include/Diderot/diderot.h

Parent Directory Parent Directory | Revision Log Revision Log


Revision 558 - (view) (download) (as text)

1 : jhr 438 /*! \file diderot.h
2 :     *
3 :     * \author John Reppy
4 : jhr 441 *
5 :     * This is the interface to the Diderot runtime for the C target. For now,
6 :     * we are targetting single-precision computations.
7 : jhr 438 */
8 :    
9 :     /*
10 :     * COPYRIGHT (c) 2010 The Diderot Project (http://diderot-language.cs.uchicago.edu)
11 :     * All rights reserved.
12 :     */
13 :    
14 :     #ifndef _DIDEROT_H_
15 :     #define _DIDEROT_H_
16 :    
17 : jhr 440 #define STATIC_INLINE static inline
18 :    
19 : jhr 438 #include <stdint.h>
20 :     #include <stdbool.h>
21 : jhr 558 #include <stdlib.h>
22 :     #include <math.h>
23 : jhr 438
24 :     /* library-call status */
25 :     typedef enum { DIDEROT_OK = 0, DIDEROT_FAIL = -1 } Status_t;
26 :    
27 :     /* SSE vector types */
28 : jhr 441 typedef float __m128 __attribute__ ((__vector_size__ (16), __may_alias__));
29 : jhr 545 typedef float vec2f __attribute__ ((vector_size (8)));
30 :     typedef float vec4f __attribute__ ((vector_size (16)));
31 :     typedef float vec8f __attribute__ ((vector_size (32)));
32 : jhr 438
33 : jhr 545 typedef float vec2d __attribute__ ((vector_size (16)));
34 :     typedef float vec4d __attribute__ ((vector_size (32)));
35 :     typedef float vec8d __attribute__ ((vector_size (64)));
36 : jhr 438
37 : jhr 545 typedef union { float r[2]; vec2f v; } union2f;
38 :     typedef union { float r[4]; vec4f v; } union4f;
39 :     typedef union { float r[8]; vec8f v; } union8f;
40 : jhr 438
41 : jhr 545 typedef union { double r[2]; vec2d v; } union2d;
42 :     typedef union { double r[4]; vec4d v; } union4d;
43 :     typedef union { double r[8]; vec8d v; } union8d;
44 :    
45 : jhr 558 typedef int32_t vec2i __attribute__ ((vector_size (8)));
46 :     typedef int32_t vec4i __attribute__ ((vector_size (16)));
47 :     typedef int32_t vec8i __attribute__ ((vector_size (32)));
48 :    
49 : jhr 545 /* vector lengths must be power of 2, but vec3 is useful, so we pad to 4 */
50 :     typedef vec4f vec3f;
51 :     typedef vec4d vec3d;
52 :    
53 : jhr 438 /* typedefs for Diderot types */
54 :     typedef int32_t Diderot_int_t;
55 :     typedef float Diderot_real_t;
56 : jhr 557 typedef union4f Diderot_vec3_t; // padded to fit in SSE register
57 :     typedef union4f Diderot_vec4_t;
58 : jhr 438 typedef const char *Diderot_string_t;
59 :    
60 : jhr 557 typedef Diderot_real_t Diderot_Mat2x2_t[4];
61 :     typedef Diderot_real_t Diderot_Mat3x3_t[9];
62 :     typedef Diderot_real_t Diderot_Mat4x4_t[16];
63 :    
64 : jhr 438 typedef struct { // wrapper for 1D image data
65 :     uint32_t dim; // dimension (== 1)
66 :     uint32_t size[1];
67 :     void *data;
68 : jhr 444 Diderot_Mat2x2_t m; // image to world-space transform
69 :     Diderot_Mat2x2_t mInv; // world to image-space transform (m inverse)
70 : jhr 438 } Diderot_image1D_t;
71 :    
72 :     typedef struct { // wrapper for 2D image data
73 :     uint32_t dim; // dimension (== 2)
74 :     uint32_t size[2]; // sizes (fast to slow)
75 :     void *data;
76 : jhr 444 Diderot_Mat3x3_t m; // image to world-space transform
77 :     Diderot_Mat3x3_t mInv; // world to image-space transform (m inverse)
78 :     Diderot_Mat3x3_t mInvT; // image to world-space transform for gradients
79 :     // (m inverse transpose)
80 : jhr 438 } Diderot_image2D_t;
81 :    
82 :     typedef struct { // wrapper for 3D image data
83 :     uint32_t dim; // dimension (== 3)
84 :     uint32_t size[3]; // sizes (fast to slow)
85 :     void *data;
86 : jhr 444 Diderot_Mat4x4_t m; // image to world-space transform
87 :     Diderot_Mat4x4_t mInv; // world to image-space transform (m inverse)
88 :     Diderot_Mat4x4_t mInvT; // image to world-space transform for gradients
89 :     // (m inverse transpose)
90 : jhr 438 } Diderot_image3D_t;
91 :    
92 :     /* Diderot library functions */
93 :    
94 :     /* load image data from Nrrd files */
95 : jhr 558 extern Status_t Diderot_LoadImage1D (Diderot_string_t name, Diderot_image1D_t *img);
96 :     extern Status_t Diderot_LoadImage2D (Diderot_string_t name, Diderot_image2D_t *img);
97 :     extern Status_t Diderot_LoadImage3D (Diderot_string_t name, Diderot_image3D_t *img);
98 : jhr 438
99 : jhr 439 /* functions to get input-parameter values */
100 : jhr 438 extern Status_t Diderot_InputString (const char *, const char **, bool);
101 :     extern Status_t Diderot_InputReal (const char *, Diderot_real_t *, bool);
102 : jhr 545 extern Status_t Diderot_InputVec3 (const char *, Diderot_vec3_t *, bool);
103 : jhr 438
104 : jhr 440 /* inline vector arithmetic functions */
105 : jhr 550 STATIC_INLINE vec3f Diderot_Vec3 (Diderot_real_t a, Diderot_real_t b, Diderot_real_t c)
106 : jhr 440 {
107 : jhr 545 return __extension__ (vec4f){ a, b, c, 0.0f };
108 : jhr 440 }
109 :    
110 : jhr 545 STATIC_INLINE vec4f Diderot_Vec4 (Diderot_real_t a, Diderot_real_t b, Diderot_real_t c, Diderot_real_t d)
111 : jhr 440 {
112 : jhr 545 return __extension__ (vec4f){ a, b, c, d };
113 : jhr 440 }
114 :    
115 : jhr 545 STATIC_INLINE vec4f Diderot_ScaleV3 (vec4f dstV, Diderot_real_t s, vec4f v)
116 : jhr 440 {
117 : jhr 441 return Diderot_Vec3(s, s, s) * v;
118 : jhr 440 }
119 :    
120 : jhr 441 // check if pos is inside the img, assuming that we have a border of width s.
121 :     //
122 :     STATIC_INLINE bool Diderot_Inside3D (Diderot_vec3_t pos, Diderot_image3D_t *img, int s)
123 :     {
124 :     // NOTE: there might be a vectorized way to do this compare!
125 : jhr 449 // cvtps2pi -- converts vector of floats to vector of int32_t values
126 :    
127 : jhr 557 return ((s <= pos.r[0]) && (pos.r[0] < (img->size[0] - s))
128 :     && (s <= pos.r[1]) && (pos.r[1] < (img->size[1] - s))
129 :     && (s <= pos.r[2]) && (pos.r[2] < (img->size[2] - s)));
130 : jhr 441 }
131 :    
132 : jhr 438 #endif /* !_DIDEROT_H_ */

root@smlnj-gforge.cs.uchicago.edu
ViewVC Help
Powered by ViewVC 1.0.0