src/SketchSolver/PlaneGCSSolver/PlaneGCSSolver_Solver.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
   2
   3 // File:    PlaneGCSSolver_Solver.cpp
   4 // Created: 14 Dec 2014
   5 // Author:  Artem ZHIDKOV
   6
   7 #include <PlaneGCSSolver_Solver.h>
   8 #include <Events_LongOp.h>
   9
  10
  11 PlaneGCSSolver_Solver::PlaneGCSSolver_Solver()
  12   : myEquationSystem(new GCS::System),
  13     myConfCollected(false),
  14     myDOF(0)
  15 {
  16 }
  17
  18 PlaneGCSSolver_Solver::~PlaneGCSSolver_Solver()
  19 {
  20   clear();
  21 }
  22
  23 void PlaneGCSSolver_Solver::clear()
  24 {
  25   myEquationSystem->clear();
  26   myParameters.clear();
  27   myConstraints.clear();
  28   myConflictingIDs.clear();
  29   myDOF = 0;
  30 }
  31
  32 void PlaneGCSSolver_Solver::addConstraint(GCSConstraintPtr theConstraint)
  33 {
  34   myEquationSystem->addConstraint(theConstraint.get());
  35   myConstraints[theConstraint->getTag()].insert(theConstraint);
  36   myDOF = -1;
  37 }
  38
  39 void PlaneGCSSolver_Solver::removeConstraint(ConstraintID theID)
  40 {
  41   myConstraints.erase(theID);
  42   if (myConstraints.empty()) {
  43     myEquationSystem->clear();
  44     myDOF = (int)myParameters.size();
  45   } else {
  46     myEquationSystem->clearByTag(theID);
  47     myDOF = -1;
  48   }
  49 }
  50
  51 double* PlaneGCSSolver_Solver::createParameter()
  52 {
  53   double* aResult = new double(0);
  54   myParameters.push_back(aResult);
  55   if (myDOF >= 0)
  56     ++myDOF;
  57   return aResult;
  58 }
  59
  60 void PlaneGCSSolver_Solver::removeParameters(const GCS::SET_pD& theParams)
  61 {
  62   for (int i = (int)myParameters.size() - 1; i >= 0; --i)
  63     if (theParams.find(myParameters[i]) != theParams.end()) {
  64       myParameters.erase(myParameters.begin() + i);
  65       --myDOF;
  66     }
  67 }
  68
  69 PlaneGCSSolver_Solver::SolveStatus PlaneGCSSolver_Solver::solve()
  70 {
  71   // clear list of conflicting constraints
  72   if (myConfCollected) {
  73     myConflictingIDs.clear();
  74     myConfCollected = false;
  75   }
  76
  77   if (myParameters.empty())
  78     return STATUS_INCONSISTENT;
  79
  80   Events_LongOp::start(this);
  81   // solve equations
  82   GCS::SolveStatus aResult = (GCS::SolveStatus)myEquationSystem->solve(myParameters);
  83   Events_LongOp::end(this);
  84
  85   SolveStatus aStatus;
  86   if (aResult == GCS::Success) {
  87     myEquationSystem->applySolution();
  88     if (myDOF < 0)
  89       myDOF = myEquationSystem->dofsNumber();
  90     aStatus = STATUS_OK;
  91   } else
  92     aStatus = STATUS_FAILED;
  93
  94   return aStatus;
  95 }
  96
  97 void PlaneGCSSolver_Solver::undo()
  98 {
  99   myEquationSystem->undoSolution();
 100 }
 101
 102 bool PlaneGCSSolver_Solver::isConflicting(const ConstraintID& theConstraint) const
 103 {
 104   if (!myConfCollected)
 105     const_cast<PlaneGCSSolver_Solver*>(this)->collectConflicting();
 106   return myConflictingIDs.find((int)theConstraint) != myConflictingIDs.end();
 107 }
 108
 109 void PlaneGCSSolver_Solver::collectConflicting()
 110 {
 111   GCS::VEC_I aConflict;
 112   myEquationSystem->getConflicting(aConflict);
 113   myConflictingIDs.insert(aConflict.begin(), aConflict.end());
 114
 115   myEquationSystem->getRedundant(aConflict);
 116   myConflictingIDs.insert(aConflict.begin(), aConflict.end());
 117
 118   myConfCollected = true;
 119 }
 120
 121 int PlaneGCSSolver_Solver::dof()
 122 {
 123   if (myDOF < 0 && !myConstraints.empty())
 124     solve();
 125   return myDOF;
 126 }