Files
create/src/Mod/Ship/shipOutlineDraw/Preview.py
2012-08-01 19:32:43 -03:00

147 lines
6.5 KiB
Python

#***************************************************************************
#* *
#* Copyright (c) 2011, 2012 *
#* Jose Luis Cercos Pita <jlcercos@gmail.com> *
#* *
#* This program is free software; you can redistribute it and/or modify *
#* it under the terms of the GNU Lesser General Public License (LGPL) *
#* as published by the Free Software Foundation; either version 2 of *
#* the License, or (at your option) any later version. *
#* for detail see the LICENCE text file. *
#* *
#* This program is distributed in the hope that it will be useful, *
#* but WITHOUT ANY WARRANTY; without even the implied warranty of *
#* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
#* GNU Library General Public License for more details. *
#* *
#* You should have received a copy of the GNU Library General Public *
#* License along with this program; if not, write to the Free Software *
#* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 *
#* USA *
#* *
#***************************************************************************
# FreeCAD modules
import FreeCAD,FreeCADGui
from FreeCAD import Base, Vector
import Part
# FreeCADShip modules
from shipUtils import Paths, Translator
class Preview(object):
def __init__(self):
""" Constructor.
"""
self.obj = None
self.reinit()
def reinit(self):
""" Reinitializate drawer.
"""
self.clean()
def update(self, L, B, T, sectionsL, sectionsB, sectionsT, shape):
""" Update the 3D view printing annotations.
@param L Ship Lpp.
@param B Ship beam.
@param T Ship draft.
@param sectionsL Transversal sections.
@param sectionsB Longitudinal sections.
@param sectionsT Water lines.
@param shape Ship surfaces shell
@return Sections object. None if errors happens.
"""
FreeCAD.Console.PrintMessage(Translator.translate('Computing sections...\n'))
# Destroy all previous entities
self.clean()
# Receive data
nL = len(sectionsL)
nB = len(sectionsB)
nT = len(sectionsT)
if not (nL or nB or nT):
return None
# Found sections
sections = []
for i in range(0,nL):
pos = sectionsL[i]
# Cut ship
section = shape.slice(Vector(1.0,0.0,0.0), pos)
for j in range(0,len(section)):
edges = section[j].Edges
# We have 3 cases,
# * when section is before midship, then only starboard section will be considered
# * When section is midship, then all section must be preserved
# * When section is after midship, then only board will be considered
if pos > 0.01:
# Look for edges at the wrong side and delete it
for k in range(len(edges)-1,-1,-1):
edge = edges[k]
bbox = edge.BoundBox
if bbox.YMin < -0.001:
del edges[k]
elif pos < -0.01:
# Look for edges at the wrong side and delete it
for k in range(len(edges)-1,-1,-1):
edge = edges[k]
bbox = edge.BoundBox
if bbox.YMax > 0.001:
del edges[k]
sections.extend(edges)
for i in range(0,nB):
pos = sectionsB[i]
section = shape.slice(Vector(0.0,1.0,0.0), pos)
for j in range(0,len(section)):
edges = section[j].Edges
# Longitudinal sections will preserve board and starboard ever. Since we take from one side,
# we nned to mirror it.
section[j] = section[j].mirror(Vector(0.0, 0.0, 0.0),Vector(0.0, 1.0, 0.0))
edges2 = section[j].Edges
sections.extend(edges)
sections.extend(edges2)
for i in range(0,nT):
pos = sectionsT[i]
section = shape.slice(Vector(0.0,0.0,1.0), pos)
for j in range(0,len(section)):
edges = section[j].Edges
# We have 3 cases,
# * when section is below draft, then only board section will be considered
# * When section is draft, then all section must be preserved
# * When section is above draft, then only starboard will be considered
if pos > T + 0.01:
# Look for edges at the wrong side and delete it
for k in range(len(edges)-1,-1,-1):
edge = edges[k]
bbox = edge.BoundBox
if bbox.YMax > 0.001:
del edges[k]
elif pos < T - 0.01:
# Look for edges at the wrong side and delete it
for k in range(len(edges)-1,-1,-1):
edge = edges[k]
bbox = edge.BoundBox
if bbox.YMin < -0.001:
del edges[k]
sections.extend(edges)
# Convert all BSplines into a shape
if not sections:
msg = Translator.translate('Any valid ship section found')
FreeCAD.Console.PrintWarning(msg)
return
obj = sections[0]
for i in range(1,len(sections)):
obj = obj.oldFuse(sections[i]) # Only group the edges, don't try to build more complex entities
# Create the representable object
Part.show(obj)
objs = FreeCAD.ActiveDocument.Objects
self.obj = objs[len(objs)-1]
self.obj.Label = 'OutlineDraw'
return self.obj
def clean(self):
""" Erase all annotations from screen.
"""
if not self.obj:
return
FreeCAD.ActiveDocument.removeObject(self.obj.Name)
self.obj = None