File: Wm5LCPSolver.h

package info (click to toggle)
libwildmagic 5.17%2Bcleaned1-6
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, bullseye
  • size: 90,112 kB
  • sloc: cpp: 215,940; csh: 637; sh: 91; makefile: 39
file content (106 lines) | stat: -rw-r--r-- 3,419 bytes parent folder | download | duplicates (3)
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
// Geometric Tools, LLC
// Copyright (c) 1998-2014
// Distributed under the Boost Software License, Version 1.0.
// http://www.boost.org/LICENSE_1_0.txt
// http://www.geometrictools.com/License/Boost/LICENSE_1_0.txt
//
// File Version: 5.0.1 (2010/10/01)

#ifndef WM5LCPSOLVER_H
#define WM5LCPSOLVER_H

#include "Wm5PhysicsLIB.h"
#include "Wm5Vector3.h"

namespace Wm5
{

class WM5_PHYSICS_ITEM LCPSolver
{
public:
    // A class for solving the Linear Complementarity Problem (LCP)
    // w = Mz + q, w o z = 0, w >= 0, z >= 0.
    //
    // Input:
    //   'numEquations' is the number of equations.
    //   'M' is a positive semidefinite matrix of size 'numEquations'.
    //   'Q' is a vector of reals.
    //
    // Output:
    //   'Z' and 'W' are the solutions.
    //   'status' of the output is one of the status codes below.

    enum // status codes
    {
        SC_FOUND_SOLUTION,               // solution
        SC_FOUND_TRIVIAL_SOLUTION,       // solution (z = 0, w = q)
        SC_CANNOT_REMOVE_COMPLEMENTARY,  // no solution (unbounded)
        SC_EXCEEDED_MAX_RETRIES,         // no solution (round-off problems?)
    };

    LCPSolver (int numEquations, double** M, double* Q, double* Z,
        double* W, int& status, int maxRetries = 100,
        double zeroTolerance = 0.0, double ratioError = 0.0);

    // MaxRetries:  In theory, one iteration of the LCP solver should work.
    // Floating point round-off errors can cause the solver to fail.  When
    // this does, the solver is allowed to retry the algorithm with a
    // different order of input equations.

    // ZeroTolerance:  The search for a pivot equation uses a comparison of a
    // certain term to zero.  To deal with floating point round-off errors,
    // the comparison is based on a small tolerance about zero.

    // RatioError:  The solver computes constant coefficients, z coefficients,
    // and w coefficients.  If any of these are nearly zero, their values are
    // set to zero.  The decision is made based on a relative comparison of a
    // ratio to zero.

private:
    void AllocateEquations ();
    void DeallocateEquations ();
    bool InitializeEquations ();
    bool SelectEquation (int& equation);
    bool FindEquation (int& equation);
    bool EquationAlgorithm (int& equation);
    void Solve (char basicVariable, int basicVariableIndex);

    struct Equation
    {
        char Var;      // 'w' or 'z' are the only choices
        int VarIndex;  // index of the w or z variable
        double* C;     // constant coefficients
        double* W;     // coefficients of the w terms
        double* Z;     // coefficients of the z terms
    };

    int mNumEquations;
    double** mM;
    double* mQ;
    Equation* mEquations;
    char mNonBasicVariable;
    char mDepartingVariable;
    int mNonBasicVariableIndex;
    int mDepartingVariableIndex;
    double mFuzz;

    int mMaxRetries;
    double mZeroTolerance;
    double mRatioError;

#ifdef WM5_LCPSOLVER_LOG
    void OpenLog ();
    void CloseLog ();
    void PrintImmediateSolution ();
    void PrintCannotRemoveComplementary ();
    void PrintNoBasicVariable (char basicVariable, int basicVariableIndex);
    void PrintEquations ();
    void PrintResults ();
    std::ofstream mLog;
    int mPrintEquationsCalls;
#endif
};

}

#endif