// Author: Mikhail PONIKAROV
#include <Model_Document.h>
-#include <ModelAPI_Feature.h>
#include <Model_Data.h>
-#include <Model_Object.h>
#include <Model_Application.h>
#include <Model_PluginManager.h>
#include <Model_Events.h>
+#include <Model_ResultPart.h>
+#include <Model_ResultConstruction.h>
+#include <Model_ResultBody.h>
#include <Events_Loop.h>
#include <Events_Error.h>
#include <TDataStd_ReferenceArray.hxx>
#include <TDataStd_HLabelArray1.hxx>
#include <TDataStd_Name.hxx>
+#include <TDF_Reference.hxx>
+#include <TDF_ChildIDIterator.hxx>
+#include <TDF_LabelMapHasher.hxx>
#include <climits>
#ifndef WIN32
static const int UNDO_LIMIT = 10; // number of possible undo operations
static const int TAG_GENERAL = 1; // general properties tag
-static const int TAG_OBJECTS = 2; // tag of the objects sub-tree (features)
+static const int TAG_OBJECTS = 2; // tag of the objects sub-tree (features, results)
static const int TAG_HISTORY = 3; // tag of the history sub-tree (python dump)
-using namespace std;
+// feature sub-labels
+static const int TAG_FEATURE_ARGUMENTS = 1; ///< where the arguments are located
+static const int TAG_FEATURE_RESULTS = 2; ///< where the results are located
/// Returns the file name of this document by the nameof directory and identifuer of a document
-static TCollection_ExtendedString DocFileName(const char* theFileName, const string& theID)
+static TCollection_ExtendedString DocFileName(const char* theFileName, const std::string& theID)
{
TCollection_ExtendedString aPath ((const Standard_CString)theFileName);
aPath += _separator_;
catch (Standard_Failure)
{
Handle(Standard_Failure) aFail = Standard_Failure::Caught();
- Events_Error::send(string("Exception in opening of document: ") + aFail->GetMessageString());
+ Events_Error::send(std::string("Exception in opening of document: ") + aFail->GetMessageString());
return false;
}
bool isError = aStatus != PCDM_RS_OK;
switch (aStatus)
{
case PCDM_RS_UnknownDocument:
- Events_Error::send(string("Can not open document: PCDM_RS_UnknownDocument")); break;
+ Events_Error::send(std::string("Can not open document: PCDM_RS_UnknownDocument")); break;
case PCDM_RS_AlreadyRetrieved:
- Events_Error::send(string("Can not open document: PCDM_RS_AlreadyRetrieved")); break;
+ Events_Error::send(std::string("Can not open document: PCDM_RS_AlreadyRetrieved")); break;
case PCDM_RS_AlreadyRetrievedAndModified:
- Events_Error::send(string("Can not open document: PCDM_RS_AlreadyRetrievedAndModified")); break;
+ Events_Error::send(
+ std::string("Can not open document: PCDM_RS_AlreadyRetrievedAndModified"));
+ break;
case PCDM_RS_NoDriver:
- Events_Error::send(string("Can not open document: PCDM_RS_NoDriver")); break;
+ Events_Error::send(std::string("Can not open document: PCDM_RS_NoDriver")); break;
case PCDM_RS_UnknownFileDriver:
- Events_Error::send(string("Can not open document: PCDM_RS_UnknownFileDriver")); break;
+ Events_Error::send(std::string("Can not open document: PCDM_RS_UnknownFileDriver")); break;
case PCDM_RS_OpenError:
- Events_Error::send(string("Can not open document: PCDM_RS_OpenError")); break;
+ Events_Error::send(std::string("Can not open document: PCDM_RS_OpenError")); break;
case PCDM_RS_NoVersion:
- Events_Error::send(string("Can not open document: PCDM_RS_NoVersion")); break;
+ Events_Error::send(std::string("Can not open document: PCDM_RS_NoVersion")); break;
case PCDM_RS_NoModel:
- Events_Error::send(string("Can not open document: PCDM_RS_NoModel")); break;
+ Events_Error::send(std::string("Can not open document: PCDM_RS_NoModel")); break;
case PCDM_RS_NoDocument:
- Events_Error::send(string("Can not open document: PCDM_RS_NoDocument")); break;
+ Events_Error::send(std::string("Can not open document: PCDM_RS_NoDocument")); break;
case PCDM_RS_FormatFailure:
- Events_Error::send(string("Can not open document: PCDM_RS_FormatFailure")); break;
+ Events_Error::send(std::string("Can not open document: PCDM_RS_FormatFailure")); break;
case PCDM_RS_TypeNotFoundInSchema:
- Events_Error::send(string("Can not open document: PCDM_RS_TypeNotFoundInSchema")); break;
+ Events_Error::send(std::string("Can not open document: PCDM_RS_TypeNotFoundInSchema"));
+ break;
case PCDM_RS_UnrecognizedFileFormat:
- Events_Error::send(string("Can not open document: PCDM_RS_UnrecognizedFileFormat")); break;
+ Events_Error::send(std::string("Can not open document: PCDM_RS_UnrecognizedFileFormat"));
+ break;
case PCDM_RS_MakeFailure:
- Events_Error::send(string("Can not open document: PCDM_RS_MakeFailure")); break;
+ Events_Error::send(std::string("Can not open document: PCDM_RS_MakeFailure")); break;
case PCDM_RS_PermissionDenied:
- Events_Error::send(string("Can not open document: PCDM_RS_PermissionDenied")); break;
+ Events_Error::send(std::string("Can not open document: PCDM_RS_PermissionDenied")); break;
case PCDM_RS_DriverFailure:
- Events_Error::send(string("Can not open document: PCDM_RS_DriverFailure")); break;
+ Events_Error::send(std::string("Can not open document: PCDM_RS_DriverFailure")); break;
default:
- Events_Error::send(string("Can not open document: unknown error")); break;
+ Events_Error::send(std::string("Can not open document: unknown error")); break;
}
}
if (!isError) {
}
catch (Standard_Failure) {
Handle(Standard_Failure) aFail = Standard_Failure::Caught();
- Events_Error::send(string("Exception in saving of document: ") + aFail->GetMessageString());
+ Events_Error::send(std::string("Exception in saving of document: ") + aFail->GetMessageString());
return false;
}
bool isDone = aStatus == PCDM_SS_OK || aStatus == PCDM_SS_No_Obj;
switch (aStatus)
{
case PCDM_SS_DriverFailure:
- Events_Error::send(string("Can not save document: PCDM_SS_DriverFailure"));
+ Events_Error::send(std::string("Can not save document: PCDM_SS_DriverFailure"));
break;
case PCDM_SS_WriteFailure:
- Events_Error::send(string("Can not save document: PCDM_SS_WriteFailure"));
+ Events_Error::send(std::string("Can not save document: PCDM_SS_WriteFailure"));
break;
case PCDM_SS_Failure:
default:
- Events_Error::send(string("Can not save document: PCDM_SS_Failure"));
+ Events_Error::send(std::string("Can not save document: PCDM_SS_Failure"));
break;
}
}
myTransactionsAfterSave = 0;
if (isDone) { // save also sub-documents if any
- set<string>::iterator aSubIter = mySubs.begin();
+ std::set<std::string>::iterator aSubIter = mySubs.begin();
for(; aSubIter != mySubs.end() && isDone; aSubIter++)
isDone = subDocument(*aSubIter)->save(theFileName);
}
aPM->setCurrentDocument(aPM->rootDocument());
}
// close all subs
- set<string>::iterator aSubIter = mySubs.begin();
+ std::set<std::string>::iterator aSubIter = mySubs.begin();
for(; aSubIter != mySubs.end(); aSubIter++)
subDocument(*aSubIter)->close();
mySubs.clear();
myDoc->NewCommand();
}
// new command for all subs
- set<string>::iterator aSubIter = mySubs.begin();
+ std::set<std::string>::iterator aSubIter = mySubs.begin();
for(; aSubIter != mySubs.end(); aSubIter++)
subDocument(*aSubIter)->startOperation();
}
void Model_Document::finishOperation()
{
// just to be sure that everybody knows that changes were performed
- Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_FEATURE_CREATED));
- Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_FEATURE_UPDATED));
- Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_FEATURE_DELETED));
+
+ if (!myDoc->HasOpenCommand() && myNestedNum != -1)
+ boost::static_pointer_cast<Model_PluginManager>(Model_PluginManager::get())->
+ setCheckTransactions(false); // for nested transaction commit
+ Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_CREATED));
+ Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
+ Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_TO_REDISPLAY));
+ Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_DELETED));
+ if (!myDoc->HasOpenCommand() && myNestedNum != -1)
+ boost::static_pointer_cast<Model_PluginManager>(Model_PluginManager::get())->
+ setCheckTransactions(true); // for nested transaction commit
if (myNestedNum != -1) // this nested transaction is owervritten
myNestedNum++;
}
// finish for all subs
- set<string>::iterator aSubIter = mySubs.begin();
+ std::set<std::string>::iterator aSubIter = mySubs.begin();
for(; aSubIter != mySubs.end(); aSubIter++)
subDocument(*aSubIter)->finishOperation();
}
}
synchronizeFeatures(true);
// abort for all subs
- set<string>::iterator aSubIter = mySubs.begin();
+ std::set<std::string>::iterator aSubIter = mySubs.begin();
for(; aSubIter != mySubs.end(); aSubIter++)
subDocument(*aSubIter)->abortOperation();
}
if (myDoc->GetAvailableUndos() > 0 && myNestedNum != 0 && myTransactionsAfterSave != 0 /* for omitting the first useless transaction */)
return true;
// check other subs contains operation that can be undoed
- set<string>::iterator aSubIter = mySubs.begin();
+ std::set<std::string>::iterator aSubIter = mySubs.begin();
for(; aSubIter != mySubs.end(); aSubIter++)
if (subDocument(*aSubIter)->canUndo())
return true;
myDoc->Undo();
synchronizeFeatures(true);
// undo for all subs
- set<string>::iterator aSubIter = mySubs.begin();
+ std::set<std::string>::iterator aSubIter = mySubs.begin();
for(; aSubIter != mySubs.end(); aSubIter++)
subDocument(*aSubIter)->undo();
}
if (myDoc->GetAvailableRedos() > 0)
return true;
// check other subs contains operation that can be redoed
- set<string>::iterator aSubIter = mySubs.begin();
+ std::set<std::string>::iterator aSubIter = mySubs.begin();
for(; aSubIter != mySubs.end(); aSubIter++)
if (subDocument(*aSubIter)->canRedo())
return true;
myTransactionsAfterSave++;
synchronizeFeatures(true);
// redo for all subs
- set<string>::iterator aSubIter = mySubs.begin();
+ std::set<std::string>::iterator aSubIter = mySubs.begin();
for(; aSubIter != mySubs.end(); aSubIter++)
subDocument(*aSubIter)->redo();
}
-boost::shared_ptr<ModelAPI_Feature> Model_Document::addFeature(string theID)
-{
- boost::shared_ptr<ModelAPI_Feature> aFeature =
- ModelAPI_PluginManager::get()->createFeature(theID);
- if (aFeature) {
- boost::dynamic_pointer_cast<Model_Document>(aFeature->documentToAdd())->addFeature(aFeature);
- } else {
- // TODO: generate error that feature is not created
- }
- return aFeature;
-}
-
/// Appenad to the array of references a new referenced label
static void AddToRefArray(TDF_Label& theArrayLab, TDF_Label& theReferenced) {
Handle(TDataStd_ReferenceArray) aRefs;
}
}
-void Model_Document::addFeature(const boost::shared_ptr<ModelAPI_Feature> theFeature)
+FeaturePtr Model_Document::addFeature(std::string theID)
{
- if (theFeature->isAction()) return; // do not add action to the data model
-
- boost::shared_ptr<ModelAPI_Document> aThis =
- Model_Application::getApplication()->getDocument(myID);
- TDF_Label aFeaturesLab = groupLabel(FEATURES_GROUP);
- TDF_Label aFeatureLab = aFeaturesLab.NewChild();
-
- // organize feature and data objects
- boost::shared_ptr<Model_Data> aData(new Model_Data);
- aData->setFeature(theFeature);
- aData->setLabel(aFeatureLab);
- theFeature->setDoc(aThis);
- theFeature->setData(aData);
- setUniqueName(theFeature);
- theFeature->initAttributes();
-
- // keep the feature ID to restore document later correctly
- TDataStd_Comment::Set(aFeatureLab, theFeature->getKind().c_str());
- myFeatures.push_back(theFeature);
- // store feature in the history of features array
- if (theFeature->isInHistory()) {
- AddToRefArray(aFeaturesLab, aFeatureLab);
- }
- // add featue to the group
- const std::string& aGroup = theFeature->getGroup();
- TDF_Label aGroupLab = groupLabel(aGroup);
- AddToRefArray(aGroupLab, aFeatureLab);
- // new name of this feature object by default equal to name of feature
- TDF_Label anObjLab = aGroupLab.NewChild();
- TCollection_ExtendedString aName(theFeature->data()->getName().c_str());
- TDataStd_Name::Set(anObjLab, aName);
- TDF_Label aGrLabChild = aGroupLab.FindChild(1);
- AddToRefArray(aGrLabChild, anObjLab); // reference to names is on the first sub
-
- // event: feature is added
- static Events_ID anEvent = Events_Loop::eventByName(EVENT_FEATURE_CREATED);
- Model_FeatureUpdatedMessage aMsg(theFeature, anEvent);
- Events_Loop::loop()->send(aMsg);
+ TDF_Label anEmptyLab;
+ FeaturePtr anEmptyFeature;
+ FeaturePtr aFeature = ModelAPI_PluginManager::get()->createFeature(theID);
+ boost::shared_ptr<Model_Document> aDocToAdd =
+ boost::dynamic_pointer_cast<Model_Document>(aFeature->documentToAdd());
+ if (aFeature) {
+ TDF_Label aFeatureLab;
+ if (!aFeature->isAction()) {// do not add action to the data model
+ TDF_Label aFeaturesLab = aDocToAdd->featuresLabel();
+ aFeatureLab = aFeaturesLab.NewChild();
+ aDocToAdd->initData(aFeature, aFeatureLab, TAG_FEATURE_ARGUMENTS);
+ // keep the feature ID to restore document later correctly
+ TDataStd_Comment::Set(aFeatureLab, aFeature->getKind().c_str());
+ aDocToAdd->myObjs.Bind(aFeatureLab, aFeature);
+ // store feature in the history of features array
+ if (aFeature->isInHistory()) {
+ AddToRefArray(aFeaturesLab, aFeatureLab);
+ }
+ }
+ if (!aFeature->isAction()) {// do not add action to the data model
+ // event: feature is added
+ static Events_ID anEvent = Events_Loop::eventByName(EVENT_OBJECT_CREATED);
+ ModelAPI_EventCreator::get()->sendUpdated(aFeature, anEvent);
+ }
+ }
+ return aFeature;
}
/// Appenad to the array of references a new referenced label.
return aResult;
}
-void Model_Document::removeFeature(boost::shared_ptr<ModelAPI_Feature> theFeature)
+void Model_Document::removeFeature(FeaturePtr theFeature)
{
boost::shared_ptr<Model_Data> aData = boost::static_pointer_cast<Model_Data>(theFeature->data());
- TDF_Label aFeatureLabel = aData->label();
- // remove the object
- TDF_Label aGroupLabel = groupLabel(theFeature->getGroup());
- int aRemovedIndex = RemoveFromRefArray(aGroupLabel, aFeatureLabel);
- RemoveFromRefArray(aGroupLabel.FindChild(1), TDF_Label(), aRemovedIndex);
- // remove feature from the myFeatures list
- std::vector<boost::shared_ptr<ModelAPI_Feature> >::iterator aFIter = myFeatures.begin();
- while(aFIter != myFeatures.end()) {
- if (*aFIter == theFeature) {
- aFIter = myFeatures.erase(aFIter);
- } else {
- aFIter++;
- }
- }
+ TDF_Label aFeatureLabel = aData->label().Father();
+ if (myObjs.IsBound(aFeatureLabel))
+ myObjs.UnBind(aFeatureLabel);
+ else return; // not found feature => do not remove
+
// erase all attributes under the label of feature
aFeatureLabel.ForgetAllAttributes();
// remove it from the references array
- RemoveFromRefArray(groupLabel(FEATURES_GROUP), aData->label());
+ RemoveFromRefArray(featuresLabel(), aData->label());
+
+ // event: feature is deleted
+ ModelAPI_EventCreator::get()->sendDeleted(theFeature->document(), ModelAPI_Feature::group());
+ // results of this feature must be redisplayed
+ static Events_ID EVENT_DISP = Events_Loop::loop()->eventByName(EVENT_OBJECT_TO_REDISPLAY);
+ const std::list<boost::shared_ptr<ModelAPI_Result> >& aResults = theFeature->results();
+ std::list<boost::shared_ptr<ModelAPI_Result> >::const_iterator aRIter = aResults.begin();
+ for(; aRIter != aResults.cend(); aRIter++) {
+ boost::shared_ptr<ModelAPI_Result> aRes = *aRIter;
+ aRes->setData(boost::shared_ptr<ModelAPI_Data>()); // deleted flag
+ ModelAPI_EventCreator::get()->sendUpdated(aRes, EVENT_DISP);
+ ModelAPI_EventCreator::get()->sendDeleted(theFeature->document(), aRes->groupName());
+ }
+}
- // event: feature is added
- Model_FeatureDeletedMessage aMsg(theFeature->document(), theFeature->getGroup());
- Events_Loop::loop()->send(aMsg);
+FeaturePtr Model_Document::feature(TDF_Label& theLabel)
+{
+ if (myObjs.IsBound(theLabel))
+ return myObjs.Find(theLabel);
+ return FeaturePtr(); // not found
}
-boost::shared_ptr<ModelAPI_Feature> Model_Document::feature(TDF_Label& theLabel)
+ObjectPtr Model_Document::object(TDF_Label theLabel)
{
- // iterate all features, may be optimized later by keeping labels-map
- vector<boost::shared_ptr<ModelAPI_Feature> >::iterator aFIter = myFeatures.begin();
- for(; aFIter != myFeatures.end(); aFIter++) {
- boost::shared_ptr<Model_Data> aData =
- boost::dynamic_pointer_cast<Model_Data>((*aFIter)->data());
- if (aData->label().IsEqual(theLabel))
- return *aFIter;
+ // try feature by label
+ FeaturePtr aFeature = feature(theLabel);
+ if (aFeature)
+ return feature(theLabel);
+ TDF_Label aFeatureLabel = theLabel.Father().Father(); // let's suppose it is result
+ aFeature = feature(aFeatureLabel);
+ if (aFeature) {
+ const std::list<boost::shared_ptr<ModelAPI_Result> >& aResults = aFeature->results();
+ std::list<boost::shared_ptr<ModelAPI_Result> >::const_iterator aRIter = aResults.cbegin();
+ for(; aRIter != aResults.cend(); aRIter++) {
+ boost::shared_ptr<Model_Data> aResData =
+ boost::dynamic_pointer_cast<Model_Data>((*aRIter)->data());
+ if (aResData->label().Father().IsEqual(theLabel))
+ return *aRIter;
+ }
}
- return boost::shared_ptr<ModelAPI_Feature>(); // not found
+ return FeaturePtr(); // not found
}
-boost::shared_ptr<ModelAPI_Document> Model_Document::subDocument(string theDocID)
+boost::shared_ptr<ModelAPI_Document> Model_Document::subDocument(std::string theDocID)
{
// just store sub-document identifier here to manage it later
if (mySubs.find(theDocID) == mySubs.end())
return Model_Application::getApplication()->getDocument(theDocID);
}
-boost::shared_ptr<ModelAPI_Feature> Model_Document::feature(
- const string& theGroupID, const int theIndex, const bool isOperation)
+ObjectPtr Model_Document::object(const std::string& theGroupID, const int theIndex)
{
- TDF_Label aGroupLab = groupLabel(theGroupID);
- Handle(TDataStd_ReferenceArray) aRefs;
- if (aGroupLab.FindAttribute(TDataStd_ReferenceArray::GetID(), aRefs)) {
- if (aRefs->Lower() <= theIndex && aRefs->Upper() >= theIndex) {
- TDF_Label aFeatureLab = aRefs->Value(theIndex);
- boost::shared_ptr<ModelAPI_Feature> aFeature = feature(aFeatureLab);
-
- if (theGroupID == FEATURES_GROUP || isOperation) { // just returns the feature from the history
- return aFeature;
- } else { // create a new object from the group to return it
- Handle(TDataStd_Name) aName; // name of the object
- if (aGroupLab.FindChild(1).FindAttribute(TDataStd_ReferenceArray::GetID(), aRefs))
- aRefs->Value(theIndex).FindAttribute(TDataStd_Name::GetID(), aName);
- boost::shared_ptr<Model_Object> anObj(new Model_Object(aFeature, aName));
- return anObj;
+ if (theGroupID == ModelAPI_Feature::group()) {
+ Handle(TDataStd_ReferenceArray) aRefs;
+ if (!featuresLabel().FindAttribute(TDataStd_ReferenceArray::GetID(), aRefs))
+ return ObjectPtr();
+ if (aRefs->Lower() > theIndex || aRefs->Upper() < theIndex)
+ return ObjectPtr();
+ TDF_Label aFeatureLabel = aRefs->Value(theIndex);
+ return feature(aFeatureLabel);
+ } else {
+ // comment must be in any feature: it is kind
+ int anIndex = 0;
+ TDF_ChildIDIterator aLabIter(featuresLabel(), TDataStd_Comment::GetID());
+ for(; aLabIter.More(); aLabIter.Next()) {
+ TDF_Label aFLabel = aLabIter.Value()->Label();
+ FeaturePtr aFeature = feature(aFLabel);
+ const std::list<boost::shared_ptr<ModelAPI_Result> >& aResults = aFeature->results();
+ std::list<boost::shared_ptr<ModelAPI_Result> >::const_iterator aRIter = aResults.begin();
+ for(; aRIter != aResults.cend(); aRIter++) {
+ if ((*aRIter)->isInHistory() && (*aRIter)->groupName() == theGroupID) {
+ if (anIndex == theIndex)
+ return *aRIter;
+ anIndex++;
+ }
}
}
}
-
// not found
- return boost::shared_ptr<ModelAPI_Feature>();
+ return ObjectPtr();
}
-int Model_Document::size(const string& theGroupID)
+int Model_Document::size(const std::string& theGroupID)
{
- Handle(TDataStd_ReferenceArray) aRefs;
- if (groupLabel(theGroupID).FindAttribute(TDataStd_ReferenceArray::GetID(), aRefs))
- return aRefs->Length();
+ int aResult = 0;
+ if (theGroupID == ModelAPI_Feature::group()) {
+ Handle(TDataStd_ReferenceArray) aRefs;
+ if (featuresLabel().FindAttribute(TDataStd_ReferenceArray::GetID(), aRefs))
+ return aRefs->Length();
+ } else {
+ // comment must be in any feature: it is kind
+ TDF_ChildIDIterator aLabIter(featuresLabel(), TDataStd_Comment::GetID());
+ for(; aLabIter.More(); aLabIter.Next()) {
+ TDF_Label aFLabel = aLabIter.Value()->Label();
+ FeaturePtr aFeature = feature(aFLabel);
+ const std::list<boost::shared_ptr<ModelAPI_Result> >& aResults = aFeature->results();
+ std::list<boost::shared_ptr<ModelAPI_Result> >::const_iterator aRIter = aResults.begin();
+ for(; aRIter != aResults.cend(); aRIter++) {
+ if ((*aRIter)->isInHistory() && (*aRIter)->groupName() == theGroupID) {
+ aResult++;
+ }
+ }
+ }
+ }
// group is not found
- return 0;
+ return aResult;
}
Model_Document::Model_Document(const std::string theID)
myDoc->CommitCommand();
}
-TDF_Label Model_Document::groupLabel(const string theGroup)
+TDF_Label Model_Document::featuresLabel()
{
- // searching for existing
- TCollection_ExtendedString aGroup(theGroup.c_str());
- TDF_ChildIDIterator aGroupIter(myDoc->Main().FindChild(TAG_OBJECTS), TDataStd_Comment::GetID());
- for(; aGroupIter.More(); aGroupIter.Next()) {
- Handle(TDataStd_Comment) aName = Handle(TDataStd_Comment)::DownCast(aGroupIter.Value());
- if (aName->Get() == aGroup)
- return aGroupIter.Value()->Label();
- }
- // create a new
- TDF_Label aNew = myDoc->Main().FindChild(TAG_OBJECTS).NewChild();
- TDataStd_Comment::Set(aNew, aGroup);
- return aNew;
+ return myDoc->Main().FindChild(TAG_OBJECTS);
}
-void Model_Document::setUniqueName(boost::shared_ptr<ModelAPI_Feature> theFeature)
+void Model_Document::setUniqueName(FeaturePtr theFeature)
{
- string aName; // result
- // iterate all features but also iterate group of this feature if object is not in history
- list<string> aGroups;
- aGroups.push_back(FEATURES_GROUP);
- if (!theFeature->isInHistory()) {
- aGroups.push_back(theFeature->getGroup());
- }
- for(list<string>::iterator aGIter = aGroups.begin(); aGIter != aGroups.end(); aGIter++) {
- // first count all objects of such kind to start with index = count + 1
- int a, aNumObjects = 0;
- int aSize = size(*aGIter);
- for(a = 0; a < aSize; a++) {
- if (feature(*aGIter, a)->getKind() == theFeature->getKind())
- aNumObjects++;
- }
- // generate candidate name
- stringstream aNameStream;
- aNameStream<<theFeature->getKind()<<"_"<<aNumObjects + 1;
- aName = aNameStream.str();
- // check this is unique, if not, increase index by 1
- for(a = 0; a < aSize;) {
- if (feature(*aGIter, a, true)->data()->getName() == aName) {
- aNumObjects++;
- stringstream aNameStream;
- aNameStream<<theFeature->getKind()<<"_"<<aNumObjects + 1;
- aName = aNameStream.str();
- // reinitialize iterator to make sure a new name is unique
- a = 0;
- } else a++;
+ if (!theFeature->data()->name().empty()) return; // not needed, name is already defined
+ std::string aName; // result
+ // first count all objects of such kind to start with index = count + 1
+ int aNumObjects = 0;
+ NCollection_DataMap<TDF_Label, FeaturePtr>::Iterator aFIter(myObjs);
+ for(; aFIter.More(); aFIter.Next()) {
+ if (aFIter.Value()->getKind() == theFeature->getKind())
+ aNumObjects++;
+ }
+ // generate candidate name
+ std::stringstream aNameStream;
+ aNameStream<<theFeature->getKind()<<"_"<<aNumObjects + 1;
+ aName = aNameStream.str();
+ // check this is unique, if not, increase index by 1
+ for(aFIter.Initialize(myObjs); aFIter.More(); ) {
+ FeaturePtr aFeature = aFIter.Value();
+ bool isSameName = aFeature->isInHistory() && aFeature->data()->name() == aName;
+ if (!isSameName) { // check also results to avoid same results names (actual for Parts)
+ const std::list<boost::shared_ptr<ModelAPI_Result> >& aResults = aFeature->results();
+ std::list<boost::shared_ptr<ModelAPI_Result> >::const_iterator aRIter = aResults.begin();
+ for(; aRIter != aResults.cend(); aRIter++) {
+ isSameName = (*aRIter)->isInHistory() && (*aRIter)->data()->name() == aName;
+ }
}
+ if (isSameName) {
+ aNumObjects++;
+ std::stringstream aNameStream;
+ aNameStream<<theFeature->getKind()<<"_"<<aNumObjects + 1;
+ aName = aNameStream.str();
+ // reinitialize iterator to make sure a new name is unique
+ aFIter.Initialize(myObjs);
+ } else aFIter.Next();
}
theFeature->data()->setName(aName);
}
-//! Returns the object by the feature
-boost::shared_ptr<ModelAPI_Feature> Model_Document::objectByFeature(
- const boost::shared_ptr<ModelAPI_Feature> theFeature)
-{
- for(int a = 0; a < size(theFeature->getGroup()); a++) {
- boost::shared_ptr<Model_Object> anObj =
- boost::dynamic_pointer_cast<Model_Object>(feature(theFeature->getGroup(), a));
- if (anObj && anObj->featureRef() == theFeature) {
- return anObj;
- }
+void Model_Document::initData(ObjectPtr theObj, TDF_Label theLab, const int theTag) {
+ boost::shared_ptr<ModelAPI_Document> aThis =
+ Model_Application::getApplication()->getDocument(myID);
+ boost::shared_ptr<Model_Data> aData(new Model_Data);
+ aData->setLabel(theLab.FindChild(theTag));
+ aData->setObject(theObj);
+ theObj->setDoc(aThis);
+ theObj->setData(aData);
+ FeaturePtr aFeature = boost::dynamic_pointer_cast<ModelAPI_Feature>(theObj);
+ if (aFeature) {
+ setUniqueName(aFeature); // must be before "initAttributes" because duplicate part uses name
+ aFeature->initAttributes();
}
- return boost::shared_ptr<ModelAPI_Feature>(); // not found
}
void Model_Document::synchronizeFeatures(const bool theMarkUpdated)
{
- boost::shared_ptr<ModelAPI_Document> aThis = Model_Application::getApplication()->getDocument(myID);
- // update features
- vector<boost::shared_ptr<ModelAPI_Feature> >::iterator aFIter = myFeatures.begin();
- // and in parallel iterate labels of features
- TDF_ChildIDIterator aFLabIter(groupLabel(FEATURES_GROUP), TDataStd_Comment::GetID());
- while(aFIter != myFeatures.end() || aFLabIter.More()) {
- static const int INFINITE_TAG = INT_MAX; // no label means that it exists somwhere in infinite
- int aFeatureTag = INFINITE_TAG;
- if (aFIter != myFeatures.end()) { // existing tag for feature
- boost::shared_ptr<Model_Data> aData = boost::dynamic_pointer_cast<Model_Data>((*aFIter)->data());
- aFeatureTag = aData->label().Tag();
- }
- int aDSTag = INFINITE_TAG;
- if (aFLabIter.More()) { // next label in DS is existing
- aDSTag = aFLabIter.Value()->Label().Tag();
- }
- if (aDSTag > aFeatureTag) { // feature is removed
- FeaturePtr aFeature = *aFIter;
- aFIter = myFeatures.erase(aFIter);
- // event: model is updated
- if (aFeature->isInHistory()) {
- Model_FeatureDeletedMessage aMsg1(aThis, FEATURES_GROUP);
- Events_Loop::loop()->send(aMsg1);
- }
- Model_FeatureDeletedMessage aMsg2(aThis, aFeature->getGroup());
- Events_Loop::loop()->send(aMsg2);
- } else if (aDSTag < aFeatureTag) { // a new feature is inserted
+ boost::shared_ptr<ModelAPI_Document> aThis =
+ Model_Application::getApplication()->getDocument(myID);
+ // update all objects by checking are they of labels or not
+ std::set<FeaturePtr> aCheckedFeatures;
+ TDF_ChildIDIterator aLabIter(featuresLabel(), TDataStd_Comment::GetID());
+ for(; aLabIter.More(); aLabIter.Next()) {
+ TDF_Label aFeatureLabel = aLabIter.Value()->Label();
+ if (!myObjs.IsBound(aFeatureLabel)) { // a new feature is inserted
// create a feature
- boost::shared_ptr<ModelAPI_Feature> aFeature = ModelAPI_PluginManager::get()->createFeature(
- TCollection_AsciiString(Handle(TDataStd_Comment)::DownCast(
- aFLabIter.Value())->Get()).ToCString());
-
- if (aFIter == myFeatures.end()) { // must be before "setData" to redo the sketch line correctly
- myFeatures.push_back(aFeature);
- aFIter = myFeatures.end();
- } else {
- aFIter++;
- myFeatures.insert(aFIter, aFeature);
- }
- boost::shared_ptr<Model_Data> aData(new Model_Data);
- TDF_Label aLab = aFLabIter.Value()->Label();
- aData->setLabel(aLab);
- aData->setFeature(aFeature);
- aFeature->setDoc(aThis);
- aFeature->setData(aData);
- aFeature->initAttributes();
+ FeaturePtr aNewObj = ModelAPI_PluginManager::get()->createFeature(
+ TCollection_AsciiString(Handle(TDataStd_Comment)::DownCast(aLabIter.Value())->Get())
+ .ToCString());
+ // this must be before "setData" to redo the sketch line correctly
+ myObjs.Bind(aFeatureLabel, aNewObj);
+ aCheckedFeatures.insert(aNewObj);
+ initData(aNewObj, aFeatureLabel, TAG_FEATURE_ARGUMENTS);
+ aNewObj->execute(); // to restore results list
// event: model is updated
- static Events_ID anEvent = Events_Loop::eventByName(EVENT_FEATURE_CREATED);
- Model_FeatureUpdatedMessage aMsg1(aFeature, anEvent);
- Events_Loop::loop()->send(aMsg1);
- boost::shared_ptr<ModelAPI_Feature> anObj = objectByFeature(aFeature);
- if (anObj) {
- Model_FeatureUpdatedMessage aMsg2(anObj, anEvent);
- Events_Loop::loop()->send(aMsg2);
- }
-
+ static Events_ID anEvent = Events_Loop::eventByName(EVENT_OBJECT_CREATED);
+ ModelAPI_EventCreator::get()->sendUpdated(aNewObj, anEvent);
// feature for this label is added, so go to the next label
- aFLabIter.Next();
} else { // nothing is changed, both iterators are incremented
+ aCheckedFeatures.insert(myObjs.Find(aFeatureLabel));
if (theMarkUpdated) {
- static Events_ID anEvent = Events_Loop::eventByName(EVENT_FEATURE_UPDATED);
- Model_FeatureUpdatedMessage aMsg(*aFIter, anEvent);
- Events_Loop::loop()->send(aMsg);
+ static Events_ID anEvent = Events_Loop::eventByName(EVENT_OBJECT_UPDATED);
+ ModelAPI_EventCreator::get()->sendUpdated(myObjs.Find(aFeatureLabel), anEvent);
}
- aFIter++;
- aFLabIter.Next();
}
}
+ // check all features are checked: if not => it was removed
+ NCollection_DataMap<TDF_Label, FeaturePtr>::Iterator aFIter(myObjs);
+ while(aFIter.More()) {
+ if (aCheckedFeatures.find(aFIter.Value()) == aCheckedFeatures.end()) {
+ FeaturePtr aFeature = aFIter.Value();
+ TDF_Label aLab = aFIter.Key();
+ aFIter.Next();
+ myObjs.UnBind(aLab);
+ // event: model is updated
+ if (aFeature->isInHistory()) {
+ ModelAPI_EventCreator::get()->sendDeleted(aThis, ModelAPI_Feature::group());
+ }
+ // results of this feature must be redisplayed (hided)
+ static Events_ID EVENT_DISP = Events_Loop::loop()->eventByName(EVENT_OBJECT_TO_REDISPLAY);
+ const std::list<boost::shared_ptr<ModelAPI_Result> >& aResults = aFeature->results();
+ std::list<boost::shared_ptr<ModelAPI_Result> >::const_iterator aRIter = aResults.begin();
+ for(; aRIter != aResults.cend(); aRIter++) {
+ boost::shared_ptr<ModelAPI_Result> aRes = *aRIter;
+ aRes->setData(boost::shared_ptr<ModelAPI_Data>()); // deleted flag
+ ModelAPI_EventCreator::get()->sendUpdated(aRes, EVENT_DISP);
+ ModelAPI_EventCreator::get()->sendDeleted(aThis, aRes->groupName());
+ }
+ } else aFIter.Next();
+ }
+
// after all updates, sends a message that groups of features were created or updated
boost::static_pointer_cast<Model_PluginManager>(Model_PluginManager::get())->
setCheckTransactions(false);
- Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_FEATURE_CREATED));
+ Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_CREATED));
if (theMarkUpdated)
- Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_FEATURE_UPDATED));
- Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_FEATURE_DELETED));
+ Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
+ Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_DELETED));
boost::static_pointer_cast<Model_PluginManager>(Model_PluginManager::get())->
setCheckTransactions(true);
}
+
+void Model_Document::storeResult(boost::shared_ptr<ModelAPI_Data> theFeatureData,
+ boost::shared_ptr<ModelAPI_Result> theResult, const int theResultIndex)
+{
+ boost::shared_ptr<ModelAPI_Document> aThis =
+ Model_Application::getApplication()->getDocument(myID);
+ theResult->setDoc(aThis);
+ initData(theResult, boost::dynamic_pointer_cast<Model_Data>(theFeatureData)->label().
+ Father().FindChild(TAG_FEATURE_RESULTS).FindChild(theResultIndex + 1), TAG_FEATURE_ARGUMENTS);
+ if (theResult->data()->name().empty()) { // if was not initialized, generate event and set a name
+ theResult->data()->setName(theFeatureData->name());
+ }
+}
+
+boost::shared_ptr<ModelAPI_ResultConstruction> Model_Document::createConstruction(
+ const boost::shared_ptr<ModelAPI_Data>& theFeatureData, const int theIndex)
+{
+ ObjectPtr anOldObject = object(boost::dynamic_pointer_cast<Model_Data>(theFeatureData)->
+ label().Father().FindChild(TAG_FEATURE_RESULTS).FindChild(theIndex + 1));
+ boost::shared_ptr<ModelAPI_ResultConstruction> aResult;
+ if (anOldObject) {
+ aResult = boost::dynamic_pointer_cast<ModelAPI_ResultConstruction>(anOldObject);
+ }
+ if (!aResult) {
+ aResult = boost::shared_ptr<ModelAPI_ResultConstruction>(new Model_ResultConstruction);
+ storeResult(theFeatureData, aResult);
+ }
+ return aResult;
+}
+
+boost::shared_ptr<ModelAPI_ResultBody> Model_Document::createBody(
+ const boost::shared_ptr<ModelAPI_Data>& theFeatureData, const int theIndex)
+{
+ ObjectPtr anOldObject = object(boost::dynamic_pointer_cast<Model_Data>(theFeatureData)->
+ label().Father().FindChild(TAG_FEATURE_RESULTS).FindChild(theIndex + 1));
+ boost::shared_ptr<ModelAPI_ResultBody> aResult;
+ if (anOldObject) {
+ aResult = boost::dynamic_pointer_cast<ModelAPI_ResultBody>(anOldObject);
+ }
+ if (!aResult) {
+ aResult = boost::shared_ptr<ModelAPI_ResultBody>(new Model_ResultBody);
+ storeResult(theFeatureData, aResult);
+ }
+ return aResult;
+}
+
+boost::shared_ptr<ModelAPI_ResultPart> Model_Document::createPart(
+ const boost::shared_ptr<ModelAPI_Data>& theFeatureData, const int theIndex)
+{
+ ObjectPtr anOldObject = object(boost::dynamic_pointer_cast<Model_Data>(theFeatureData)->
+ label().Father().FindChild(TAG_FEATURE_RESULTS).FindChild(theIndex + 1));
+ boost::shared_ptr<ModelAPI_ResultPart> aResult;
+ if (anOldObject) {
+ aResult = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(anOldObject);
+ }
+ if (!aResult) {
+ aResult = boost::shared_ptr<ModelAPI_ResultPart>(new Model_ResultPart);
+ storeResult(theFeatureData, aResult);
+ }
+ return aResult;
+}
+
+boost::shared_ptr<ModelAPI_Feature> Model_Document::feature(
+ const boost::shared_ptr<ModelAPI_Result>& theResult)
+{
+ boost::shared_ptr<Model_Data> aData = boost::dynamic_pointer_cast<Model_Data>(theResult->data());
+ if (aData) {
+ TDF_Label aFeatureLab = aData->label().Father().Father().Father();
+ return feature(aFeatureLab);
+ }
+ return FeaturePtr();
+}
+
+Standard_Integer HashCode(const TDF_Label& theLab,const Standard_Integer theUpper)
+{
+ return TDF_LabelMapHasher::HashCode(theLab, theUpper);
+
+}
+Standard_Boolean IsEqual(const TDF_Label& theLab1,const TDF_Label& theLab2)
+{
+ return TDF_LabelMapHasher::IsEqual(theLab1, theLab2);
+}
