149 lines
5.9 KiB
Python
149 lines
5.9 KiB
Python
#***************************************************************************
|
|
#* *
|
|
#* Copyright (c) 2011, 2016 *
|
|
#* 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 *
|
|
#* *
|
|
#***************************************************************************
|
|
|
|
import os
|
|
from PySide import QtGui, QtCore
|
|
import FreeCAD
|
|
import FreeCADGui
|
|
from FreeCAD import Base
|
|
import Spreadsheet
|
|
|
|
class Plot(object):
|
|
def __init__(self, x, y, disp, xcb, ship):
|
|
""" Constructor. performs the plot and shows it.
|
|
@param x X coordinates.
|
|
@param y Transversal computed areas.
|
|
@param disp Ship displacement.
|
|
@param xcb Buoyancy center length.
|
|
@param ship Active ship instance.
|
|
"""
|
|
self.plot(x, y, disp, xcb, ship)
|
|
self.spreadSheet(x, y, ship)
|
|
|
|
def plot(self, x, y, disp, xcb, ship):
|
|
""" Perform the areas curve plot.
|
|
@param x X coordinates.
|
|
@param y Transversal areas.
|
|
@param disp Ship displacement.
|
|
@param xcb Buoyancy center length.
|
|
@param ship Active ship instance.
|
|
@return True if error happens.
|
|
"""
|
|
try:
|
|
import Plot
|
|
plt = Plot.figure('Areas curve')
|
|
except ImportError:
|
|
msg = QtGui.QApplication.translate(
|
|
"ship_console",
|
|
"Plot module is disabled, so I cannot perform the plot",
|
|
None)
|
|
FreeCAD.Console.PrintWarning(msg + '\n')
|
|
return True
|
|
# Plot areas curve
|
|
areas = Plot.plot(x, y, 'Transversal areas')
|
|
areas.line.set_linestyle('-')
|
|
areas.line.set_linewidth(2.0)
|
|
areas.line.set_color((0.0, 0.0, 0.0))
|
|
# Get perpendiculars data
|
|
Lpp = ship.Length.getValueAs('m').Value
|
|
FPx = 0.5 * Lpp
|
|
APx = -0.5 * Lpp
|
|
maxArea = max(y)
|
|
# Plot perpendiculars
|
|
FP = Plot.plot([FPx, FPx], [0.0, maxArea])
|
|
FP.line.set_linestyle('-')
|
|
FP.line.set_linewidth(1.0)
|
|
FP.line.set_color((0.0, 0.0, 0.0))
|
|
AP = Plot.plot([APx, APx], [0.0, maxArea])
|
|
AP.line.set_linestyle('-')
|
|
AP.line.set_linewidth(1.0)
|
|
AP.line.set_color((0.0, 0.0, 0.0))
|
|
# Add annotations for prependiculars
|
|
ax = Plot.axes()
|
|
ax.annotate('AP', xy=(APx + 0.01 * Lpp, 0.01 * maxArea), size=15)
|
|
ax.annotate('AP', xy=(APx + 0.01 * Lpp, 0.95 * maxArea), size=15)
|
|
ax.annotate('FP', xy=(FPx + 0.01 * Lpp, 0.01 * maxArea), size=15)
|
|
ax.annotate('FP', xy=(FPx + 0.01 * Lpp, 0.95 * maxArea), size=15)
|
|
# Add some additional data
|
|
addInfo = ("$XCB = {0} \\; \\mathrm{{m}}$\n"
|
|
"$Area_{{max}} = {1} \\; \\mathrm{{m}}^2$\n"
|
|
"$\\bigtriangleup = {2} \\; \\mathrm{{tons}}$".format(
|
|
xcb.getValueAs("m").Value,
|
|
maxArea,
|
|
disp.getValueAs("kg").Value / 1000.0))
|
|
ax.text(0.0,
|
|
0.01 * maxArea,
|
|
addInfo,
|
|
verticalalignment='bottom',
|
|
horizontalalignment='center',
|
|
fontsize=20)
|
|
# Write axes titles
|
|
Plot.xlabel(r'$x \; \mathrm{m}$')
|
|
Plot.ylabel(r'$Area \; \mathrm{m}^2$')
|
|
ax.xaxis.label.set_fontsize(20)
|
|
ax.yaxis.label.set_fontsize(20)
|
|
# Show grid
|
|
Plot.grid(True)
|
|
# End
|
|
plt.update()
|
|
return False
|
|
|
|
def spreadSheet(self, x, y, ship):
|
|
""" Write the output data file.
|
|
@param x X coordinates.
|
|
@param y Transversal areas.
|
|
@param ship Active ship instance.
|
|
"""
|
|
s = FreeCAD.activeDocument().addObject('Spreadsheet::Sheet',
|
|
'Areas curve')
|
|
|
|
# Print the header
|
|
s.set("A1", "x [m]")
|
|
s.set("B1", "area [m^2]")
|
|
s.set("C1", "FP x")
|
|
s.set("D1", "FP y")
|
|
s.set("E1", "AP x")
|
|
s.set("F1", "AP y")
|
|
|
|
# Print the perpendiculars data
|
|
Lpp = ship.Length.getValueAs('m').Value
|
|
FPx = 0.5 * Lpp
|
|
APx = -0.5 * Lpp
|
|
maxArea = max(y)
|
|
s.set("C2", str(FPx))
|
|
s.set("D2", str(0.0))
|
|
s.set("C3", str(FPx))
|
|
s.set("D3", str(maxArea))
|
|
s.set("E2", str(APx))
|
|
s.set("F2", str(0.0))
|
|
s.set("E3", str(APx))
|
|
s.set("F3", str(maxArea))
|
|
|
|
# Print the data
|
|
for i in range(len(x)):
|
|
s.set("A{}".format(i + 2), str(x[i]))
|
|
s.set("B{}".format(i + 2), str(y[i]))
|
|
|
|
# Recompute
|
|
FreeCAD.activeDocument().recompute()
|