Constraint.cs 8.86 KB
Newer Older
BlackAngle233's avatar
BlackAngle233 committed
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
using System.Collections;
using System.Collections.Generic;
using UnityEngine;

public enum ConstraintType
{
    LineLength,
    LineEqual,
    LineNormal,
    LineNormalPlane,
    LineParallel,
    LineParallelPlane,
    LineCross,
    MidPoint,
    PointOnLine,
    PointInPlane,
    AngleEqual,
    PlaneParallelPlane,
    PlaneNormalPlane
}
public class Constraint
{
    public object Object1;
    public object Object2;

BlackAngle233's avatar
212    
BlackAngle233 committed
26
27
    public string name;

BlackAngle233's avatar
BlackAngle233 committed
28
29
30
31
32
33
34
35
36
37
    State state = State.unknown;

    ConstraintType constraintType;

    public Constraint(ConstraintType type, object obj1, object obj2)
    {
        Object1 = obj1;
        Object2 = obj2;
        constraintType = type;

BlackAngle233's avatar
BlackAngle233 committed
38
39
40
41
        Line l1 = null;
        Line l2 = null;
        float length = 0;

BlackAngle233's avatar
BlackAngle233 committed
42
43
44
        switch (type)
        {
            case ConstraintType.LineLength:
BlackAngle233's avatar
BlackAngle233 committed
45
46
                l1 = (Line)obj1;
                length = (float)obj2;
BlackAngle233's avatar
BlackAngle233 committed
47

BlackAngle233's avatar
BlackAngle233 committed
48
49
                l1.setConstraint(this);
                name = l1.name + "=" + length.ToString();
BlackAngle233's avatar
BlackAngle233 committed
50
51
                break;
            case ConstraintType.LineEqual:
BlackAngle233's avatar
BlackAngle233 committed
52
53
                l1 = (Line)obj1;
                l2 = (Line)obj2;
BlackAngle233's avatar
BlackAngle233 committed
54
55
56

                l1.setConstraint(this);
                l2.setConstraint(this);
BlackAngle233's avatar
212    
BlackAngle233 committed
57
                name = l1.name + "=" + l2.name;
BlackAngle233's avatar
BlackAngle233 committed
58
                break;
BlackAngle233's avatar
BlackAngle233 committed
59
60
61
62
63
64
65
66
            case ConstraintType.LineNormal:
                l1 = (Line)obj1;
                l2 = (Line)obj2;

                l1.setConstraint(this);
                l2.setConstraint(this);
                name = l1.name + "⊥" + l2.name;
                break;
BlackAngle233's avatar
BlackAngle233 committed
67
68
69
70
71
72
        }
    }

    public float getConstraintValue()
    {
        float result = 0;
BlackAngle233's avatar
BlackAngle233 committed
73
74
75
76
77
78

        Line l1 = null;
        Line l2 = null;

        float length = 0;

BlackAngle233's avatar
BlackAngle233 committed
79
80
81
        switch (constraintType)
        {
             case ConstraintType.LineLength:
BlackAngle233's avatar
BlackAngle233 committed
82
83
                l1 = (Line)Object1;
                length = (float)Object2;
BlackAngle233's avatar
BlackAngle233 committed
84

BlackAngle233's avatar
BlackAngle233 committed
85
                result = l1.getLength() - length;
BlackAngle233's avatar
BlackAngle233 committed
86
87
                break;
            case ConstraintType.LineEqual:
BlackAngle233's avatar
BlackAngle233 committed
88
89
                l1 = (Line)Object1;
                l2 = (Line)Object2;
BlackAngle233's avatar
BlackAngle233 committed
90
91
92
93
94
95
96
97
98
99
                
                if (l1.lineState > l2.lineState)
                {
                    result = l1.getLength() - l2.getLength();
                }
                else
                {
                    result = l2.getLength() - l1.getLength();
                }
                break;
BlackAngle233's avatar
BlackAngle233 committed
100
101
102
103
104
105
            case ConstraintType.LineNormal:
                l1 = (Line)Object1;
                l2 = (Line)Object2;
                
                result = Vector3.Dot(l1.direction, l2.direction);
                break;
BlackAngle233's avatar
BlackAngle233 committed
106
107
108
109
110
111
112
        }

        return result;
    }

    public Gradient getGradient()
    {
BlackAngle233's avatar
BlackAngle233 committed
113
114
115
116
117
        Line l1 = null;
        Line l2 = null;

        float length = 0;

BlackAngle233's avatar
BlackAngle233 committed
118
119
120
        switch (constraintType)
        {
            case ConstraintType.LineLength:
BlackAngle233's avatar
BlackAngle233 committed
121
122
                l1 = (Line)Object1;
                length = (float)Object2;
BlackAngle233's avatar
BlackAngle233 committed
123

BlackAngle233's avatar
BlackAngle233 committed
124
                return l1.getGradient();
BlackAngle233's avatar
BlackAngle233 committed
125
            case ConstraintType.LineEqual:
BlackAngle233's avatar
BlackAngle233 committed
126
127
                l1 = (Line)Object1;
                l2 = (Line)Object2;
BlackAngle233's avatar
BlackAngle233 committed
128
129
130
131
132
133
134
135
                if(l1.lineState > l2.lineState)
                {
                    return l1.getGradient();
                }
                else
                {
                    return l2.getGradient();
                }
BlackAngle233's avatar
BlackAngle233 committed
136
137
138
139
140
141
142
143
144
145
146
            case ConstraintType.LineNormal:
                l1 = (Line)Object1;
                l2 = (Line)Object2;
                if(l1.lineState > l2.lineState)
                {
                    return new Gradient(l2.getUnknownPoint() ,l1.direction);
                }
                else
                {
                    return new Gradient(l1.getUnknownPoint() ,l2.direction);
                }
BlackAngle233's avatar
BlackAngle233 committed
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
        }
        return null;
    }

    public void setState(State s)
    {
        state = s;
    }

    public State getState()
    {
        return state;
    }

    public void checkState()
    {
BlackAngle233's avatar
BlackAngle233 committed
163
164
165
166
167
        Line l1 = null;
        Line l2 = null;

        float length = 0;

BlackAngle233's avatar
BlackAngle233 committed
168
169
170
        switch (constraintType)
        {
            case ConstraintType.LineLength:
BlackAngle233's avatar
BlackAngle233 committed
171
172
                l1 = (Line)Object1;
                length = (float)Object2;
BlackAngle233's avatar
BlackAngle233 committed
173
                
BlackAngle233's avatar
BlackAngle233 committed
174
                setState(l1.lineState);
BlackAngle233's avatar
BlackAngle233 committed
175
176
                break;
            case ConstraintType.LineEqual:
BlackAngle233's avatar
BlackAngle233 committed
177
178
179
            case ConstraintType.LineNormal:
                l1 = (Line)Object1;
                l2 = (Line)Object2;
BlackAngle233's avatar
BlackAngle233 committed
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
                
                if(l1.lineState == State.confirm && l2.lineState == State.confirm)
                {
                    setState(State.confirm);
                } 
                else if((l1.lineState == State.confirm && l2.lineState == State.halfconfirm) || (l2.lineState == State.confirm && l1.lineState == State.halfconfirm))
                {
                    setState(State.halfconfirm);
                }
                else
                {
                    setState(State.unknown);
                }
                break;
        }
    }

    public void calculateConstraint(Point target)
    {
BlackAngle233's avatar
BlackAngle233 committed
199
        Vector3 normal;
BlackAngle233's avatar
BlackAngle233 committed
200
201
202
        Vector3 direction;
        Point tmp;

BlackAngle233's avatar
BlackAngle233 committed
203
204
205
206
207
208
209
        Line l1 = null;
        Line l2 = null;

        float length = 0;
        float cos = 0;
        float sin = 0;

BlackAngle233's avatar
BlackAngle233 committed
210
211
212
        switch (constraintType)
        {
            case ConstraintType.LineLength:
BlackAngle233's avatar
BlackAngle233 committed
213
214
                l1 = (Line)Object1;
                length = (float)Object2;
BlackAngle233's avatar
BlackAngle233 committed
215

BlackAngle233's avatar
BlackAngle233 committed
216
                tmp = l1.getAnotherPoint(target);
BlackAngle233's avatar
BlackAngle233 committed
217
                
BlackAngle233's avatar
BlackAngle233 committed
218
                direction = l1.getDirection(target);
BlackAngle233's avatar
BlackAngle233 committed
219
220
221
222
223
                target.setPoint(tmp.add(direction * length));

                target.setState(State.confirm);
                break;
            case ConstraintType.LineEqual:
BlackAngle233's avatar
BlackAngle233 committed
224
225
                l1 = (Line)Object1;
                l2 = (Line)Object2;
BlackAngle233's avatar
BlackAngle233 committed
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241

                if(l1.lineState == State.confirm)
                {
                    tmp = l2.getAnotherPoint(target);

                    direction = l2.getDirection(target);
                    target.setPoint(tmp.add(direction * l1.getLength()));
                }
                else if(l2.lineState == State.confirm)
                {
                    tmp = l1.getAnotherPoint(target);

                    direction = l1.getDirection(target);
                    target.setPoint(tmp.add(direction * l2.getLength()));
                }

BlackAngle233's avatar
BlackAngle233 committed
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
                target.setState(State.confirm);
                break;
            case ConstraintType.LineNormal:
                l1 = (Line)Object1;
                l2 = (Line)Object2;

                if(l1.lineState == State.confirm)
                {
                    normal = l1.direction.normalized;

                    tmp = l2.getAnotherPoint(target);
                    direction = new Vector3(target.x - tmp.x, target.y - tmp.y, target.z - tmp.z);

                    sin = Vector3.Dot(l1.getVector3(), l2.getVector3()) / l1.getLength() / l2.getLength();
                    cos = Mathf.Sqrt(1 - cos * cos);

                    direction = direction * cos + Vector3.Cross(normal, direction) * sin + normal * Vector3.Dot(normal, direction) * (1 - cos);

                    target.setPoint(direction.x - tmp.x, direction.y - tmp.y, direction.z - tmp.z);
                }
                else if(l2.lineState == State.confirm)
                {
                    normal = l2.direction.normalized;

                    tmp = l1.getAnotherPoint(target);
                    direction = new Vector3(target.x - tmp.x, target.y - tmp.y, target.z - tmp.z);

                    sin = Vector3.Dot(l2.getVector3(), l1.getVector3()) / l2.getLength() / l1.getLength();
                    cos = Mathf.Sqrt(1 - cos * cos);

                    direction = direction * cos + Vector3.Cross(normal, direction) * sin + normal * Vector3.Dot(normal, direction) * (1 - cos);

                    target.setPoint(direction.x - tmp.x, direction.y - tmp.y, direction.z - tmp.z);
                }

BlackAngle233's avatar
BlackAngle233 committed
277
278
279
280
281
282
283
284
285
286
                target.setState(State.confirm);
                break;
        }
    }

    public void tryCalculateConstraint(Point target)
    {
        Gradient tmp;
        float value = 0;

BlackAngle233's avatar
BlackAngle233 committed
287
288
289
290
291
        Line l1 = null;
        Line l2 = null;

        float length = 0;

BlackAngle233's avatar
BlackAngle233 committed
292
293
294
        switch (constraintType)
        {
            case ConstraintType.LineLength:
BlackAngle233's avatar
BlackAngle233 committed
295
296
                l1 = (Line)Object1;
                length = (float)Object2;
BlackAngle233's avatar
BlackAngle233 committed
297
298
299
300
301
302
303
304

                tmp = getGradient();
                value = getConstraintValue();

                target += tmp.calculateGradient(value);

                break;
            case ConstraintType.LineEqual:
BlackAngle233's avatar
BlackAngle233 committed
305
306
                l1 = (Line)Object1;
                l2 = (Line)Object2;
BlackAngle233's avatar
BlackAngle233 committed
307
308
309
310
311
312
313

                tmp = getGradient();
                value = getConstraintValue();

                target += tmp.calculateGradient(value);

                break;
BlackAngle233's avatar
BlackAngle233 committed
314
315
316
317
318
319
320
321
322
323
324
            case ConstraintType.LineNormal:
                l1 = (Line)Object1;
                l2 = (Line)Object2;

                tmp = getGradient();
                value = getConstraintValue();

                target += tmp.calculateGradient(value);

                break;

BlackAngle233's avatar
BlackAngle233 committed
325
326
327
        }
    }
}