--- /dev/null
- part = model.activeDocument()
+ """compound of vertices Feature
+ Author: Nathalie Gore
+ """
+
+ from qtsalome import QMessageBox
+ from salome.shaper import model
+ from salome.shaper import geom
+ import ModelAPI
+
+ class compoundVertices(model.Feature):
+ """Import of Construction points
+ """
+
+ # Feature initializations
+
+ def __init__(self):
+ """x.__init__(...) initializes x; see x.__class__.__doc__ for signature"""
+ model.Feature.__init__(self)
+
+ @staticmethod
+ def ID():
+ """Return Id of the Feature."""
+ return "compoundVertices"
+
+ @staticmethod
+ def FILE_ID():
+ """Returns ID of the file select parameter."""
+ return "file_path"
+
+ @staticmethod
+ def SEPARATOR_ID():
+ """Returns ID of the separator parameter."""
+ return "separator"
+
+ def getKind(self):
+ """Override Feature.getKind()"""
+ return compoundVertices.ID()
+
+
+ # Initialization of the dialog panel
+
+ def initAttributes(self):
+ """Override Feature.initAttributes()"""
+ # Creating the input argument of the feature
+ self.data().addAttribute(self.FILE_ID(), ModelAPI.ModelAPI_AttributeString_typeId())
+ self.data().addAttribute(self.SEPARATOR_ID(), ModelAPI.ModelAPI_AttributeString_typeId())
+
+ self.lfeatures = []
+ self.folder = None
+ self.separator = " "
+
+ # Execution of the Import
+
+ def execute(self):
+ """F.execute() -- execute the Feature"""
+ # Retrieving the user input
+ apath = self.string(self.FILE_ID())
+ aseparator = self.string(self.SEPARATOR_ID()).value()
+ if aseparator:
+ self.separator = aseparator
+
+ filepath = apath.value()
+ if filepath != "" :
-
++ part = model.activeDocument()
+ if self.lfeatures :
+ for feature in self.lfeatures:
+ part.removeFeature(feature.feature())
+ self.lfeatures = []
+ model.removeFolder(self.folder)
+
+ from os.path import basename
+ filename = basename(filepath)
+ nameRes = "Points_" + filename
+
+ # Creating the construction points in the current document
+ lVertices = []
+
+ with open(filepath) as file:
+ for line in file:
+ coord = line.split(self.separator)
+ if len(coord) != 3:
+ #QMessageBox.warning( self, 'Error!', '3D coords waited!' )
+ return
+ x = float(coord[0]); y = float(coord[1]); z = float(coord[2]);
+ point = model.addPoint(part, x,y,z); point.execute(True); self.lfeatures.append(point)
+ vertex = model.addVertex(part, [point.result()]); vertex.execute(True); self.lfeatures.append(vertex)
+ lVertices.append(vertex.result())
+ file.close()
+ compound = model.addCompound(part, lVertices)
+ compound.execute(True); self.lfeatures.append(compound)
+ compound.result().setName(nameRes)
+ self.folder = model.addFolder(part, self.lfeatures[0], compound)
+ self.folder.setName(nameRes)
+ return
++
+ setError("The file does not exist")
+
+ def isMacro(self):
+ """Override Feature.initAttributes().
+ F.isMacro() -> True
+
+ compoundVertices feature is macro: removes itself on the creation transaction
+ finish.
+ """
+ return False
--- /dev/null
-<source>\r
- <workbench id="Macros" document="Part">\r
- <group id="Samples">\r
-\r
- <feature\r
- id="compoundVertices"\r
- title="Points set"\r
- tooltip="Import a set of construction points"\r
- icon="icons/Addons/import.png"\r
- helpfile="compoundVerticesFeature.html">\r
- <file_selector id="file_path" title="Import txt file (X Y Z)" path="">\r
- </file_selector>\r
- <stringvalue id="separator" label="Separator (optional): ">\r
- </stringvalue>\r
- </feature>\r
-\r
- </group>\r
- </workbench> \r
-</source>\r
++<source>
++ <workbench id="Macros" document="Part">
++ <group id="Samples">
++ <feature
++ id="compoundVertices"
++ title="Points set"
++ tooltip="Import a set of construction points"
++ icon="icons/Addons/import.png"
++ helpfile="compoundVerticesFeature.html">
++ <file_selector id="file_path" title="Import txt file (X Y Z)" path="">
++ </file_selector>
++ <stringvalue id="separator" label="Separator (optional): ">
++ </stringvalue>
++ </feature>
++ </group>
++ </workbench>
++</source>
--- /dev/null
-
+ """importParameters
+ Author: Nathalie Gore
+ """
+
+ from salome.shaper import model
+ from salome.shaper import geom
+ import ModelAPI
+
+ class importParameters(model.Feature):
+ """Import of Construction points
+ """
+
+ # Feature initializations
+
+ def __init__(self):
+ """x.__init__(...) initializes x; see x.__class__.__doc__ for signature"""
+ model.Feature.__init__(self)
+
+ @staticmethod
+ def ID():
+ """Return Id of the Feature."""
+ return "importParameters"
+
+ @staticmethod
+ def FILE_ID():
+ """Returns ID of the file select parameter."""
+ return "file_path"
+
+ def getKind(self):
+ """Override Feature.getKind()"""
+ return importParameters.ID()
+
+
+ # Initialization of the dialog panel
+
+ def initAttributes(self):
+ """Override Feature.initAttributes()"""
+ # Creating the input argument of the feature
+ self.data().addAttribute(self.FILE_ID(), ModelAPI.ModelAPI_AttributeString_typeId())
+
+ # Execution of the Import
+
+ def execute(self):
+ """F.execute() -- execute the Feature"""
+ # Retrieving the user input
+ apath = self.string(self.FILE_ID())
+ filepath = apath.value()
+ #print("filepath : ", filepath)
+ if filepath != "" :
+
+ # Creating the parameters in the current document
+ part = model.activeDocument()
+
+ with open(filepath) as file:
+ for line in file:
+ defParameters = line.replace("\n","").split(' ')
+ if len(defParameters) == 2 :
+ model.addParameter(part, defParameters[0], defParameters[1])
+ file.close()
+ return
-
-
++
+ setError("The file does not exist")
+
+ def isMacro(self):
+ """Override Feature.initAttributes().
+ F.isMacro() -> True
+
+ importParameters feature is macro: removes itself on the creation transaction
+ finish.
+ """
+ return True
--- /dev/null
-<source>\r
- <workbench id="Macros" document="Part">\r
- <group id="Samples">\r
-\r
- <feature id="importParameters"\r
- title="List of Parameters"\r
- tooltip="Import a set of parameters"\r
- icon="icons/Addons/parameters.png"\r
- helpfile="importParametersFeature.html">\r
- <file_selector id="file_path" title="Import file" path="">\r
- </file_selector>\r
- </feature>\r
-\r
- </group>\r
- </workbench> \r
-</source>\r
++<source>
++ <workbench id="Macros" document="Part">
++ <group id="Samples">
++ <feature id="importParameters"
++ title="List of Parameters"
++ tooltip="Import a set of parameters"
++ icon="icons/Addons/parameters.png"
++ helpfile="importParametersFeature.html">
++ <file_selector id="file_path" title="Import file" path="">
++ </file_selector>
++ </feature>
++ </group>
++ </workbench>
++</source>
