source: trunk/VUT/GtpVisibilityPreprocessor/src/VssRay.h @ 428

Revision 428, 4.9 KB checked in by mattausch, 19 years ago (diff)

fixed vspkdtree rays bug

Line 
1#ifndef __VSS_RAY_H
2#define __VSS_RAY_H
3
4#include <vector>
5using namespace std;
6#include "Vector3.h"
7#include "Ray.h"
8
9class AxisAlignedBox3;
10class Intersectable;
11
12class VssRay {
13public:
14        // various flags
15  enum {
16    FPosDirX = 1,  // the direction of ray in X-axis is positive
17    FPosDirY = 2,  // the direction of ray in Y-axis is positive
18    FPosDirZ = 4   // the direction of ray in Z-axis is positive
19  };
20
21        static int mailID;
22  int mMailbox;
23       
24        // side of the ray - used for the ray classification
25        //  char mSide;
26       
27        // computed t
28  float mT;
29
30        // inverse of the ray size
31        float mInvSize;
32       
33        // counter of references to this ray
34  short mRefCount;
35
36        // various flags
37  char mFlags;
38       
39  Vector3 mOrigin;
40  Vector3 mTermination;
41       
42        /// Termination object for the ray
43        /// only the termination object is actually used
44        Intersectable *mOriginObject;
45        Intersectable *mTerminationObject;
46       
47        //////////////////////////////
48        VssRay(const Vector3 &origin,
49                                 const Vector3 &termination,
50                                 Intersectable *originObject,
51                                 Intersectable *terminationObject):
52                mMailbox(-1),
53    mOrigin(origin),
54    mTermination(termination),
55                mOriginObject(originObject),
56                mTerminationObject(terminationObject),
57                mRefCount(0),
58    mFlags(0)
59        {
60    Precompute();
61  }
62       
63        VssRay(const Ray &ray):
64                mRefCount(0),
65                mFlags(0),
66                mMailbox(-1),
67                mOriginObject(ray.sourceObject.mObject)
68        {
69                if (ray.sourceObject.mObject)
70                        mOrigin = ray.Extrap(ray.sourceObject.mT);
71                else
72                        mOrigin = ray.GetLoc();
73
74                Debug << "origin: " << mOrigin << endl;
75
76                if (!ray.intersections.empty())
77                {
78                        mTermination = ray.Extrap(ray.intersections[0].mT);
79                        mTerminationObject = ray.intersections[0].mObject;
80                }
81                else
82                {
83                        //mTermination = 1e6;
84                        mTermination = mOrigin + ray.GetDir() * 1e6;
85                        mTerminationObject = NULL;
86                }
87
88                Precompute();
89        }
90
91  void Precompute() {
92    mFlags = 0;
93                Vector3 dir = GetDir();
94
95#define BIDIRECTIONAL_RAY 0
96#if BIDIRECTIONAL_RAY
97                if (dir.y < 0) {
98                        // swap objects and poits       
99                        swap(mOriginObject, mTerminationObject);
100                        swap(mOrigin, mTermination);
101                        dir = -dir;
102                }
103#endif
104                if (dir.x > 0.0f) mFlags |= FPosDirX;
105    if (dir.y > 0.0f) mFlags |= FPosDirY;
106    if (dir.z > 0.0f) mFlags |= FPosDirZ;
107
108    mInvSize = 1.0f/Magnitude(GetDir());
109  }
110
111  void Mail() { mMailbox = mailID; }
112  static void NewMail() { mailID++; }
113  bool Mailed() const { return mMailbox == mailID; }
114
115  bool Mailed(const int mail) {
116    return mMailbox >= mailID + mail;
117  }
118
119        int HitCount() const {
120#if BIDIRECTIONAL_RAY
121                if (mOriginObject && mTerminationObject)
122                        return 2;
123                if (mOriginObject || mTerminationObject)
124                        return 1;
125                return 0;
126#else
127                return (mTerminationObject) ? 1 : 0;
128#endif
129        }
130       
131        Vector3 GetOrigin() const { return mOrigin; }
132  Vector3 GetTermination() const { return mTermination; }
133  Vector3 GetDir() const { return mTermination - mOrigin; }
134  //  Vector3 GetNormalizedDir() const { return Normalize(termination - mOrigin); }
135  Vector3 GetNormalizedDir() const { return (mTermination - mOrigin)*mInvSize; }
136
137        float GetDirParametrization(const int axis) const;
138       
139        float GetInvSize() const { return  mInvSize; }
140  float GetOrigin(const int axis) const { return mOrigin[axis]; }
141  float GetTermination(const int axis) const { return mTermination[axis]; }
142  float GetDir(const int axis) const { return mTermination[axis] - mOrigin[axis]; }
143  float GetNormalizedDir(const int axis) const {
144    return (mTermination[axis] - mOrigin[axis])*mInvSize;
145  }
146       
147  bool
148  ComputeMinMaxT(const AxisAlignedBox3 &box,
149                                                                 float &tmin,
150                                                                 float &tmax) const;
151       
152  bool
153  Intersects(const AxisAlignedBox3 &box,
154                                                 float &tmin,
155                                                 float &tmax) const;
156       
157  bool
158  IntersectsSphere(const Vector3 &center,
159                                                                         const float sqrRadius,
160                                                                         Vector3 &point,
161                                                                         float &t) const;
162       
163  void
164  Translate(const Vector3 &translation) {
165    mOrigin += translation;
166    mTermination += translation;
167  }
168
169        void SetupEndPoints(const Vector3 &origin,
170                                                                                        const Vector3 &termination)
171        {
172                mOrigin = origin;
173                mTermination = termination;
174                Precompute();
175        }
176                                                                                       
177  bool HasPosDir(const int axis) const { return mFlags & (1<<axis); }
178
179  char Flags() const { return mFlags;}
180  void SetFlags(char orFlag) { mFlags |= orFlag;}
181 
182  bool IsActive() const { return mRefCount>0; }
183       
184  // reference counting for leaf nodes
185  int RefCount() const { return mRefCount; }
186  int Ref() { return mRefCount++; }
187       
188  void ScheduleForRemoval() { if (mRefCount>0) mRefCount = -mRefCount; }
189  bool ScheduledForRemoval() const { return mRefCount<0; }
190  void Unref() {
191    if (mRefCount > 0)
192      mRefCount--;
193    else
194      if (mRefCount < 0)
195                                mRefCount++;
196      else {
197                                cerr<<"Trying to unref already deleted ray!"<<endl;
198                                exit(1);
199      }
200  }
201};
202
203
204
205typedef  vector<VssRay *> VssRayContainer;
206
207
208
209#endif
Note: See TracBrowser for help on using the repository browser.