git.hungrycats.org Git - xscreensaver/blob - hacks/glx/tube.c

   1 /* tube, Copyright (c) 2001 Jamie Zawinski <jwz@jwz.org>
   2  * Utility functions to create tubes and cones in GL.
   3  *
   4  * Permission to use, copy, modify, distribute, and sell this software and its
   5  * documentation for any purpose is hereby granted without fee, provided that
   6  * the above copyright notice appear in all copies and that both that
   7  * copyright notice and this permission notice appear in supporting
   8  * documentation.  No representations are made about the suitability of this
   9  * software for any purpose.  It is provided "as is" without express or
  10  * implied warranty.
  11  */
  12
  13 #include "config.h"
  14 #include <stdlib.h>
  15 #include <math.h>
  16 #include <GL/glx.h>
  17 #include "tube.h"
  18
  19 static void
  20 unit_tube (int faces, Bool smooth, Bool wire)
  21 {
  22   int i;
  23   GLfloat step = M_PI * 2 / faces;
  24   GLfloat s2 = step/2;
  25   GLfloat th;
  26   GLfloat x, y, x0, y0;
  27   int z = 0;
  28
  29   /* side walls
  30    */
  31   glFrontFace(GL_CCW);
  32   glBegin (wire ? GL_LINES : (smooth ? GL_QUAD_STRIP : GL_QUADS));
  33
  34   th = 0;
  35   x = 1;
  36   y = 0;
  37
  38   if (!smooth)
  39     {
  40       x0 = cos (s2);
  41       y0 = sin (s2);
  42     }
  43
  44   if (smooth) faces++;
  45
  46   for (i = 0; i < faces; i++)
  47     {
  48       if (smooth)
  49         glNormal3f(x, 0, y);
  50       else
  51         glNormal3f(x0, 0, y0);
  52
  53       glVertex3f(x, 0, y);
  54       glVertex3f(x, 1, y);
  55
  56       th += step;
  57       x  = cos (th);
  58       y  = sin (th);
  59
  60       if (!smooth)
  61         {
  62           x0 = cos (th + s2);
  63           y0 = sin (th + s2);
  64
  65           glVertex3f(x, 1, y);
  66           glVertex3f(x, 0, y);
  67         }
  68     }
  69   glEnd();
  70
  71   /* End caps
  72    */
  73   for (z = 0; z <= 1; z++)
  74     {
  75       glFrontFace(z == 0 ? GL_CCW : GL_CW);
  76       glNormal3f(0, (z == 0 ? -1 : 1), 0);
  77       glBegin(wire ? GL_LINE_LOOP : GL_TRIANGLE_FAN);
  78       if (! wire) glVertex3f(0, z, 0);
  79       for (i = 0, th = 0; i <= faces; i++)
  80         {
  81           GLfloat x = cos (th);
  82           GLfloat y = sin (th);
  83           glVertex3f(x, z, y);
  84           th += step;
  85         }
  86       glEnd();
  87     }
  88 }
  89
  90
  91 static void
  92 unit_cone (int faces, Bool smooth, Bool wire)
  93 {
  94   int i;
  95   GLfloat step = M_PI * 2 / faces;
  96   GLfloat s2 = step/2;
  97   GLfloat th;
  98   GLfloat x, y, x0, y0;
  99
 100   /* side walls
 101    */
 102   glFrontFace(GL_CW);
 103   glBegin(wire ? GL_LINES : GL_TRIANGLES);
 104
 105   th = 0;
 106   x = 1;
 107   y = 0;
 108   x0 = cos (s2);
 109   y0 = sin (s2);
 110
 111   for (i = 0; i < faces; i++)
 112     {
 113       glNormal3f(x0, 0, y0);
 114       glVertex3f(0,  1, 0);
 115
 116       if (smooth) glNormal3f(x, 0, y);
 117       glVertex3f(x, 0, y);
 118
 119       th += step;
 120       x0 = cos (th + s2);
 121       y0 = sin (th + s2);
 122       x  = cos (th);
 123       y  = sin (th);
 124
 125       if (smooth) glNormal3f(x, 0, y);
 126       glVertex3f(x, 0, y);
 127     }
 128   glEnd();
 129
 130   /* End cap
 131    */
 132   glFrontFace(GL_CCW);
 133   glNormal3f(0, -1, 0);
 134   glBegin(wire ? GL_LINE_LOOP : GL_TRIANGLE_FAN);
 135   if (! wire) glVertex3f(0, 0, 0);
 136   for (i = 0, th = 0; i <= faces; i++)
 137     {
 138       GLfloat x = cos (th);
 139       GLfloat y = sin (th);
 140       glVertex3f(x, 0, y);
 141       th += step;
 142     }
 143   glEnd();
 144 }
 145
 146
 147 static void
 148 tube_1 (GLfloat x1, GLfloat y1, GLfloat z1,
 149         GLfloat x2, GLfloat y2, GLfloat z2,
 150         GLfloat diameter, GLfloat cap_size,
 151         int faces, Bool smooth, Bool wire,
 152         Bool cone_p)
 153 {
 154   GLfloat length, X, Y, Z;
 155
 156   if (diameter <= 0) abort();
 157
 158   X = (x2 - x1);
 159   Y = (y2 - y1);
 160   Z = (z2 - z1);
 161
 162   if (X == 0 && Y == 0 && Z == 0)
 163     return;
 164
 165   length = sqrt (X*X + Y*Y + Z*Z);
 166
 167   glPushMatrix();
 168
 169   glTranslatef(x1, y1, z1);
 170   glRotatef (-atan2 (X, Y)               * (180 / M_PI), 0, 0, 1);
 171   glRotatef ( atan2 (Z, sqrt(X*X + Y*Y)) * (180 / M_PI), 1, 0, 0);
 172   glScalef (diameter, length, diameter);
 173
 174   /* extend the endpoints of the tube by the cap size in both directions */
 175   if (cap_size != 0)
 176     {
 177       GLfloat c = cap_size/length;
 178       glTranslatef (0, -c, 0);
 179       glScalef (1, 1+c+c, 1);
 180     }
 181
 182   if (cone_p)
 183     unit_cone (faces, smooth, wire);
 184   else
 185     unit_tube (faces, smooth, wire);
 186
 187   glPopMatrix();
 188 }
 189
 190
 191 void
 192 tube (GLfloat x1, GLfloat y1, GLfloat z1,
 193       GLfloat x2, GLfloat y2, GLfloat z2,
 194       GLfloat diameter, GLfloat cap_size,
 195       int faces, Bool smooth, Bool wire)
 196 {
 197   tube_1 (x1, y1, z1, x2, y2, z2, diameter, cap_size, faces, smooth, wire,
 198           False);
 199 }
 200
 201
 202 void
 203 cone (GLfloat x1, GLfloat y1, GLfloat z1,
 204       GLfloat x2, GLfloat y2, GLfloat z2,
 205       GLfloat diameter, GLfloat cap_size,
 206       int faces, Bool smooth, Bool wire)
 207 {
 208   tube_1 (x1, y1, z1, x2, y2, z2, diameter, cap_size, faces, smooth, wire,
 209           True);
 210 }