import FreeCAD, FreeCADGui, Part
import frameCmd
from os.path import *
from os import listdir
from pipeFeatures import pypeType
class AnyThing(pypeType): #object):
'''Class for object TType="Anything"
AnyThing(obj,name="Thing", fileName="", ports="0:0:0")
obj: the "App::FeaturePython object"
TType (string): name of the thing
FileName (string): a valid .STEP .IGES or .brep
ports: a string in the form "x0:y0:z0/x1:y1:z1/..." representing the position of ports 0,1....
'''
def __init__(self, obj,name="valve",fileName='ballDN15.stp',ports='0:0:0'):
#obj.Proxy = self
super(AnyThing,self).__init__(obj)
# define common properties
obj.PType="Any"
# define specific properties
obj.addProperty("App::PropertyString","FileName","AnyThing","The file of the shape (inside ./shapes)").FileName=fileName
portslist=list()
if ports:
for port in ports.split('/'):
portslist.append(FreeCAD.Vector([float(i) for i in port.split(":")]))
obj.Ports=portslist
if fileName:
s=Part.Shape()
path=join(dirname(abspath(__file__)),"shapes",fileName)
if exists(path):
s.read(path)
obj.Shape=s
else:
FreeCAD.Console.PrintError("%s file doesn't exist" %fileName)
def onChanged(self, fp, prop):
pass
def execute(self, fp):
super(AnyThing,self).execute(fp) # perform common operations
def makeThing(n='Valvola', fn='ballDN15.stp', p='0:0:0', pos=None, Z=None):
'''
makeThing(n,fn,p,pos,Z)
n = name
fn = file name
p = ports string (e.g. "0:0:0/0:0:69")
pos = position Vector
Z = orientation Vector
'''
if pos==None:
pos=FreeCAD.Vector(0,0,0)
if Z==None:
Z=FreeCAD.Vector(0,0,1)
a=FreeCAD.ActiveDocument.addObject("Part::FeaturePython",n)
AnyThing(a, name=n, fileName=fn, ports=p)
a.ViewObject.Proxy=0
a.Placement.Base=pos
rot=FreeCAD.Rotation(FreeCAD.Vector(0,0,1),Z)
a.Placement.Rotation=rot.multiply(a.Placement.Rotation)
return a
from frameForms import prototypeDialog
class shapeDialog(prototypeDialog):
'dialog for any()'
def __init__(self):
super(shapeDialog,self).__init__('shapes.ui')
self.shapesDir=join(dirname(abspath(__file__)),'shapes')
self.cwd=self.shapesDir
self.pipeDictList=list()
self.filesListed=list()
self.lastThing=None
for n in listdir(self.cwd):
absFile=join(self.cwd,n)
if isdir(absFile): self.form.comboDirs.addItem(n)
elif isfile(absFile)and n.upper().split(extsep)[-1] in ['STEP','STP','IGES','IGS','BREP','BRP']: self.form.listFiles.addItem(n)
self.form.listFiles.setCurrentRow(0)
self.form.comboDirs.currentIndexChanged.connect(self.fillFiles)
self.form.listFiles.itemClicked.connect(self.checkListed)
def accept(self):
fname=self.form.listFiles.currentItem().text()
row=None
pos=None
Z=None
if self.form.comboDirs.currentText()=='<shapes>': sdir=''
else: sdir=self.form.comboDirs.currentText()
if fname in self.filesListed:
for r in self.pipeDictList:
if r["fileName"]==fname:
row=r
break
name=row["name"]
ports=row["ports"]
else:
name=fname.split('.')[0]
ports='0:0:0'
selex=FreeCADGui.Selection.getSelectionEx()
if selex:
vxs=[v for sx in selex for so in sx.SubObjects for v in so.Vertexes]
cedges=[e for e in frameCmd.edges() if e.curvatureAt(0)!=0]
faces=frameCmd.faces()
if faces:
x=(faces[0].ParameterRange[0]+faces[0].ParameterRange[1])/2
y=(faces[0].ParameterRange[2]+faces[0].ParameterRange[3])/2
pos=faces[0].valueAt(x,y)
Z=faces[0].normalAt(x,y)
elif cedges:
pos=cedges[0].centerOfCurvatureAt(0)
Z=cedges[0].tangentAt(0).cross(cedges[0].normalAt(0))
elif vxs:
pos=vxs[0].Point
self.lastThing=makeThing(name,join(sdir,fname),ports,pos,Z)
if row:
self.lastThing.Kv=float(row["Kv"])
self.lastThing.PSize=row["DN"]
self.lastThing.PRating=row["PN"]
def fillFiles(self):
self.form.listFiles.clear()
self.form.labListed.setText('-')
if self.form.comboDirs.currentText()!='<shapes>':
self.cwd=join(self.shapesDir,self.form.comboDirs.currentText())
else:
self.cwd=self.shapesDir
for n in listdir(self.cwd):
if isfile(join(self.cwd,n)) and n.upper().split(extsep)[-1] in ['STEP','STP','IGES','IGS','BREP','BRP']:
self.form.listFiles.addItem(n)
if n[:4]=='Any_' and n[-4:]=='.csv':
f=open(join(self.cwd,n),'r')
from csv import DictReader
reader=DictReader(f,delimiter=';')
self.pipeDictList=[DNx for DNx in reader]
f.close()
self.filesListed=[row['fileName'] for row in self.pipeDictList]
self.form.listFiles.setCurrentRow(0)
def checkListed(self):
if self.form.listFiles.currentItem().text() in self.filesListed:
self.form.labListed.setText('*** LISTED ***')
else:
self.form.labListed.setText('(unlisted)')
