src/IFSelect/IFSelect_SelectExplore.cxx

   1 #include <IFSelect_SelectExplore.ixx>
   2 #include <TColStd_IndexedMapOfTransient.hxx>
   3 #include <stdio.h>
   4
   5
   6 IFSelect_SelectExplore::IFSelect_SelectExplore (const Standard_Integer level)
   7 : thelevel (level)    {  }
   8
   9
  10     Standard_Integer  IFSelect_SelectExplore::Level () const
  11       {  return thelevel;  }
  12
  13
  14     Interface_EntityIterator  IFSelect_SelectExplore::RootResult
  15   (const Interface_Graph& G) const
  16 {
  17 //  Attention, voila comme on procede
  18 //  On a une IndexedMapOfTransient en entree (entites deja traitees/a traiter)
  19 //    Elle est initialisee par InputResult
  20 //  Et une map en sortie (resultats pris) -> le resultat sera unique
  21 //  En entree, un curseur d entite courante
  22 //  Pour chaque entite, on appelle Explore. 3 cas possibles :
  23 //    retour False, on passe
  24 //    retour True et liste vide, on prend cette entite sans aller plus loin
  25 //    retour True et liste non vide, on ne prend pas cette entite mais on
  26 //      considere son resultat.
  27 //      Si dernier niveau, on le prend en entier. Sinon, il alimente l entree
  28
  29   Standard_Integer nb = G.Size();
  30   TColStd_IndexedMapOfTransient entrees (nb);
  31   TColStd_IndexedMapOfTransient result  (nb);
  32 //  Initialisation
  33   Standard_Integer i, j, level = 1, ilev = 0;
  34   Interface_EntityIterator input; input = InputResult(G);
  35   for (input.Start(); input.More(); input.Next())
  36     i = entrees.Add (input.Value());
  37   ilev = entrees.Extent();
  38
  39 // Exploration
  40   for (i = 1; i <= nb; i ++) {
  41     if (i > entrees.Extent()) break;
  42     if (i > ilev)  {
  43       level ++;
  44       if (level > thelevel && thelevel > 0) break;
  45       ilev = entrees.Extent();
  46     }
  47     Handle(Standard_Transient) ent = entrees.FindKey(i);
  48     if (ent.IsNull()) continue;
  49     Interface_EntityIterator exp;
  50     if (!Explore (level,ent,G,exp)) continue;
  51
  52 //  On prend en compte : entite a prendre directement ?
  53 //  reprendre liste en entree (niveau pas atteint) ou resultat (niveau atteint)
  54     if (exp.NbEntities() == 0) {
  55       j = result.Add (ent);
  56       continue;
  57     } else if (level == thelevel) {
  58       for (exp.Start(); exp.More(); exp.Next()) j = result.Add (exp.Value());
  59     } else {
  60       for (exp.Start(); exp.More(); exp.Next()) j = entrees.Add (exp.Value());
  61     }
  62   }
  63
  64 //  On recolte le resultat
  65   Interface_EntityIterator res;
  66   nb = result.Extent();
  67   for (j = 1; j <= nb; j ++) res.AddItem (result.FindKey(j));
  68   return res;
  69 }
  70
  71
  72     TCollection_AsciiString  IFSelect_SelectExplore::Label () const
  73 {
  74   TCollection_AsciiString labl;
  75   if (thelevel == 0) labl.AssignCat("(Recursive)");
  76   else if (thelevel > 0) {
  77     char lab[30];
  78     sprintf (lab,"(Level %d)",thelevel);
  79     labl.AssignCat(lab);
  80   }
  81   labl.AssignCat(ExploreLabel());
  82   return labl;
  83 }