129 lines
5.3 KiB
Python
129 lines
5.3 KiB
Python
# -*- coding: utf8 -*-
|
|
|
|
# ***************************************************************************
|
|
# * *
|
|
# * Copyright (c) 2017 Yorik van Havre <yorik@uncreated.net> *
|
|
# * *
|
|
# * 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 *
|
|
# * *
|
|
# ***************************************************************************
|
|
|
|
"""The BIM ImagePlane command"""
|
|
|
|
|
|
import FreeCAD
|
|
import FreeCADGui
|
|
|
|
QT_TRANSLATE_NOOP = FreeCAD.Qt.QT_TRANSLATE_NOOP
|
|
translate = FreeCAD.Qt.translate
|
|
|
|
|
|
class BIM_ImagePlane:
|
|
def GetResources(self):
|
|
return {
|
|
"Pixmap": "Image_CreateImagePlane.svg",
|
|
"MenuText": QT_TRANSLATE_NOOP("BIM_ImagePlane", "Image plane"),
|
|
"ToolTip": QT_TRANSLATE_NOOP(
|
|
"BIM_ImagePlane", "Creates a plane from an image"
|
|
),
|
|
}
|
|
|
|
def IsActive(self):
|
|
v = hasattr(FreeCADGui.getMainWindow().getActiveWindow(), "getSceneGraph")
|
|
return v
|
|
|
|
def Activated(self):
|
|
import FreeCADGui
|
|
from PySide import QtCore, QtGui
|
|
|
|
try:
|
|
import DraftTrackers
|
|
except Exception:
|
|
import draftguitools.gui_trackers as DraftTrackers
|
|
|
|
self.tracker = DraftTrackers.rectangleTracker()
|
|
self.basepoint = None
|
|
self.opposite = None
|
|
(filename, _filter) = QtGui.QFileDialog.getOpenFileName(
|
|
QtGui.QApplication.activeWindow(),
|
|
translate("BIM", "Select image"),
|
|
None,
|
|
translate("BIM", "Image file (*.png *.jpg *.bmp)"),
|
|
)
|
|
if filename:
|
|
self.filename = filename
|
|
im = QtGui.QImage(self.filename)
|
|
self.proportion = float(im.height()) / float(im.width())
|
|
if hasattr(FreeCADGui, "Snapper"):
|
|
FreeCADGui.Snapper.getPoint(
|
|
callback=self.PointCallback, movecallback=self.MoveCallback
|
|
)
|
|
|
|
def MoveCallback(self, point, snapinfo):
|
|
if point and self.basepoint and (point != self.basepoint):
|
|
chord = point.sub(self.basepoint)
|
|
length = DraftVecUtils.project(chord, self.tracker.u).Length
|
|
height = length * self.proportion
|
|
self.opposite = (
|
|
FreeCAD.Vector(self.tracker.u)
|
|
.multiply(length)
|
|
.add(FreeCAD.Vector(self.tracker.v).multiply(height))
|
|
)
|
|
self.opposite = self.basepoint.add(self.opposite)
|
|
self.tracker.update(self.opposite)
|
|
|
|
def PointCallback(self, point, snapinfo):
|
|
import FreeCADGui
|
|
import Image
|
|
|
|
if not point:
|
|
# cancelled
|
|
self.tracker.off()
|
|
return
|
|
elif not self.basepoint:
|
|
# this is our first clicked point, nothing to do just yet
|
|
self.basepoint = point
|
|
self.tracker.setorigin(point)
|
|
self.tracker.on()
|
|
FreeCADGui.Snapper.getPoint(
|
|
last=point, callback=self.PointCallback, movecallback=self.MoveCallback
|
|
)
|
|
else:
|
|
# this is our second point
|
|
self.tracker.off()
|
|
midpoint = self.basepoint.add(
|
|
self.opposite.sub(self.basepoint).multiply(0.5)
|
|
)
|
|
rotation = FreeCAD.DraftWorkingPlane.getRotation().Rotation
|
|
diagonal = self.opposite.sub(self.basepoint)
|
|
length = DraftVecUtils.project(diagonal, FreeCAD.DraftWorkingPlane.u).Length
|
|
height = DraftVecUtils.project(diagonal, FreeCAD.DraftWorkingPlane.v).Length
|
|
FreeCAD.ActiveDocument.openTransaction("Create image plane")
|
|
image = FreeCAD.activeDocument().addObject(
|
|
"Image::ImagePlane", "ImagePlane"
|
|
)
|
|
image.Label = os.path.splitext(os.path.basename(self.filename))[0]
|
|
image.ImageFile = self.filename
|
|
image.Placement = FreeCAD.Placement(midpoint, rotation)
|
|
image.XSize = length
|
|
image.YSize = height
|
|
FreeCAD.ActiveDocument.commitTransaction()
|
|
FreeCAD.ActiveDocument.recompute()
|
|
|
|
|
|
FreeCADGui.addCommand("BIM_ImagePlane", BIM_ImagePlane())
|