Skip to content

Commit

Permalink
2.49.55
Browse files Browse the repository at this point in the history 
Updated viewport Placement tools
  • Loading branch information
@eirannejad
eirannejad committed Aug 12, 2016
1 parent 16f86de commit b7edd0b
Show file tree
Hide file tree
Showing 3 changed files with 175 additions and 93 deletions.
103 changes: 65 additions & 38 deletions pyRevit/Memory_copyViewPortPlacement.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@
import pickle
from collections import namedtuple

from Autodesk.Revit.DB import Transaction, Viewport, ViewSheet, XYZ
from Autodesk.Revit.DB import TransactionGroup, Transaction, Viewport, ViewSheet, ViewPlan, ViewDrafting, XYZ
from Autodesk.Revit.UI import TaskDialog

doc = __revit__.ActiveUIDocument.Document
Expand All @@ -38,12 +38,13 @@ def __init__(self):


Point = namedtuple('Point', ['X', 'Y','Z'])
originalviewtype = ''

selview = selvp = None
vpboundaryoffset = 0.01
activeSheet = uidoc.ActiveGraphicalView
transmatrix = TransformationMatrix()

revtransmatrix = TransformationMatrix()

def sheet_to_view_transform(sheetcoord):
global transmatrix
Expand All @@ -58,35 +59,44 @@ def sheet_to_view_transform(sheetcoord):

def set_tansform_matrix(selvp, selview):
# making sure the cropbox is active.
if not selview.CropBoxActive:
with Transaction(doc, 'Activate Crop Box') as t:
t.Start()
selview.CropBoxActive = True
t.Commit()

# get vp min max points in sheetUCS
ol = selvp.GetBoxOutline()
vptempmin = ol.MinimumPoint
vpmin = XYZ(vptempmin.X + vpboundaryoffset, vptempmin.Y + vpboundaryoffset, 0.0)
vptempmax = ol.MaximumPoint
vpmax = XYZ(vptempmax.X - vpboundaryoffset, vptempmax.Y - vpboundaryoffset, 0.0)

# get view min max points in modelUCS.
modelucsx = []
modelucsy = []
crsm = selview.GetCropRegionShapeManager()
cl = crsm.GetCropShape()[0]
for l in cl:
modelucsx.append(l.GetEndPoint(0).X)
modelucsy.append(l.GetEndPoint(0).Y)

cropmin = XYZ(min(modelucsx), min(modelucsy), 0.0)
cropmax = XYZ(max(modelucsx), max(modelucsy), 0.0)

transmatrix.sourcemin = vpmin
transmatrix.sourcemax = vpmax
transmatrix.destmin = cropmin
transmatrix.destmax = cropmax
cboxactive = selview.CropBoxActive
cboxvisible = selview.CropBoxVisible
with Transaction(doc, 'Activate and Read Cropbox Boundary') as t:
t.Start()
selview.CropBoxActive = True
selview.CropBoxVisible = False

# get view min max points in modelUCS.
modelucsx = []
modelucsy = []
crsm = selview.GetCropRegionShapeManager()
cl = crsm.GetCropShape()[0]
for l in cl:
modelucsx.append(l.GetEndPoint(0).X)
modelucsy.append(l.GetEndPoint(0).Y)
cropmin = XYZ(min(modelucsx), min(modelucsy), 0.0)
cropmax = XYZ(max(modelucsx), max(modelucsy), 0.0)

# get vp min max points in sheetUCS
ol = selvp.GetBoxOutline()
vptempmin = ol.MinimumPoint
vpmin = XYZ(vptempmin.X + vpboundaryoffset, vptempmin.Y + vpboundaryoffset, 0.0)
vptempmax = ol.MaximumPoint
vpmax = XYZ(vptempmax.X - vpboundaryoffset, vptempmax.Y - vpboundaryoffset, 0.0)

transmatrix.sourcemin = vpmin
transmatrix.sourcemax = vpmax
transmatrix.destmin = cropmin
transmatrix.destmax = cropmax

revtransmatrix.sourcemin = cropmin
revtransmatrix.sourcemax = cropmax
revtransmatrix.destmin = vpmin
revtransmatrix.destmax = vpmax

selview.CropBoxActive = cboxactive
selview.CropBoxVisible = cboxvisible
t.Commit()

usertemp = os.getenv('Temp')
prjname = op.splitext(op.basename(doc.PathName))[0]
Expand All @@ -95,18 +105,35 @@ def set_tansform_matrix(selvp, selview):
selected_ids = uidoc.Selection.GetElementIds()

if selected_ids.Count == 1:
for vport_id in selected_ids:
vport_id = selected_ids[0]
try:
vport = doc.GetElement(vport_id)
except:
TaskDialog.Show('pyRevit', 'Select at least one viewport. No more, no less!')
if isinstance(vport, Viewport):
view = doc.GetElement(vport.ViewId)
if isinstance(vport, Viewport):
set_tansform_matrix(vport, view)
if view is not None and isinstance(view, ViewPlan):
with TransactionGroup(doc, 'Copy Viewport Location') as tg:
tg.Start()
set_tansform_matrix(vport, view)
center = vport.GetBoxCenter()
modelpoint = sheet_to_view_transform(center)
center_pt = Point(center.X, center.Y, center.Z)
model_pt = Point(modelpoint.X, modelpoint.Y, modelpoint.Z)
with open(datafile, 'wb') as fp:
originalviewtype = 'ViewPlan'
pickle.dump(originalviewtype, fp)
pickle.dump(center_pt, fp)
pickle.dump(model_pt, fp)
tg.Assimilate()
elif view is not None and isinstance(view, ViewDrafting):
center = vport.GetBoxCenter()
modelpoint = sheet_to_view_transform(center)
center_pt = Point(center.X, center.Y, center.Z)
model_pt = Point(modelpoint.X, modelpoint.Y, modelpoint.Z)
with open(datafile, 'wb') as fp:
originalviewtype = 'ViewDrafting'
pickle.dump(originalviewtype, fp)
pickle.dump(center_pt, fp)
pickle.dump(model_pt, fp)
break
else:
TaskDialog.Show('pyRevit', 'This tool only works with Plan, RCP, and Detail views and viewports.')
else:
TaskDialog.Show('pyRevit', 'Select at least one viewport. No more, no less!')
163 changes: 109 additions & 54 deletions pyRevit/Memory_pasteViewPortPlacement.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,9 @@

__window__.Close()

__doc__ = 'Paste a Viewport Placement from memory'
__doc__ = 'Paste the location of a viewport from memory onto the selected viewport. This relocates the selected ' \
'to the exact location of the original viewport.'

__author__ = 'Gui Talarico | [email protected]\nEhsan Iran-Nejad | [email protected]'
__version__ = '0.1.0'

Expand All @@ -21,13 +23,24 @@
import pickle
from collections import namedtuple

from Autodesk.Revit.DB import Transaction, Viewport, ViewSheet, XYZ
from Autodesk.Revit.DB import TransactionGroup, Transaction, Viewport, ViewPlan, ViewSheet, ViewDrafting, XYZ
from Autodesk.Revit.UI import TaskDialog

PINAFTERSET = False


doc = __revit__.ActiveUIDocument.Document
uidoc = __revit__.ActiveUIDocument


class OriginalIsViewDrafting(Exception):
pass


class OriginalIsViewPlan(Exception):
pass


class TransformationMatrix:
def __init__(self):
self.sourcemin = None
Expand All @@ -37,6 +50,7 @@ def __init__(self):


Point = namedtuple('Point', ['X', 'Y','Z'])
originalviewtype = ''

selview = selvp = None
vpboundaryoffset = 0.01
Expand Down Expand Up @@ -67,43 +81,45 @@ def view_to_sheet_transform(modelcoord):


def set_tansform_matrix(selvp, selview):
global transmatrix
global revtransmatrix
# making sure the cropbox is active.
if not selview.CropBoxActive:
with Transaction(doc, 'Activate Crop Box') as t:
t.Start()
selview.CropBoxActive = True
t.Commit()

# get vp min max points in sheetUCS
ol = selvp.GetBoxOutline()
vptempmin = ol.MinimumPoint
vpmin = XYZ(vptempmin.X + vpboundaryoffset, vptempmin.Y + vpboundaryoffset, 0.0)
vptempmax = ol.MaximumPoint
vpmax = XYZ(vptempmax.X - vpboundaryoffset, vptempmax.Y - vpboundaryoffset, 0.0)

# get view min max points in modelUCS.
modelucsx = []
modelucsy = []
crsm = selview.GetCropRegionShapeManager()
cl = crsm.GetCropShape()[0]
for l in cl:
modelucsx.append(l.GetEndPoint(0).X)
modelucsy.append(l.GetEndPoint(0).Y)

cropmin = XYZ(min(modelucsx), min(modelucsy), 0.0)
cropmax = XYZ(max(modelucsx), max(modelucsy), 0.0)

transmatrix.sourcemin = vpmin
transmatrix.sourcemax = vpmax
transmatrix.destmin = cropmin
transmatrix.destmax = cropmax

revtransmatrix.sourcemin = cropmin
revtransmatrix.sourcemax = cropmax
revtransmatrix.destmin = vpmin
revtransmatrix.destmax = vpmax
cboxactive = selview.CropBoxActive
cboxvisible = selview.CropBoxVisible
with Transaction(doc, 'Activate and Read Cropbox Boundary') as t:
t.Start()
selview.CropBoxActive = True
selview.CropBoxVisible = False

# get view min max points in modelUCS.
modelucsx = []
modelucsy = []
crsm = selview.GetCropRegionShapeManager()
cl = crsm.GetCropShape()[0]
for l in cl:
modelucsx.append(l.GetEndPoint(0).X)
modelucsy.append(l.GetEndPoint(0).Y)
cropmin = XYZ(min(modelucsx), min(modelucsy), 0.0)
cropmax = XYZ(max(modelucsx), max(modelucsy), 0.0)

# get vp min max points in sheetUCS
ol = selvp.GetBoxOutline()
vptempmin = ol.MinimumPoint
vpmin = XYZ(vptempmin.X + vpboundaryoffset, vptempmin.Y + vpboundaryoffset, 0.0)
vptempmax = ol.MaximumPoint
vpmax = XYZ(vptempmax.X - vpboundaryoffset, vptempmax.Y - vpboundaryoffset, 0.0)

transmatrix.sourcemin = vpmin
transmatrix.sourcemax = vpmax
transmatrix.destmin = cropmin
transmatrix.destmax = cropmax

revtransmatrix.sourcemin = cropmin
revtransmatrix.sourcemax = cropmax
revtransmatrix.destmin = vpmin
revtransmatrix.destmax = vpmax

selview.CropBoxActive = cboxactive
selview.CropBoxVisible = cboxvisible
t.Commit()

usertemp = os.getenv('Temp')
prjname = op.splitext(op.basename(doc.PathName))[0]
Expand All @@ -112,27 +128,66 @@ def set_tansform_matrix(selvp, selview):
selected_ids = uidoc.Selection.GetElementIds()

if selected_ids.Count == 1:
for element_id in selected_ids:
vport = doc.GetElement(element_id)
vport_id = selected_ids[0]
try:
vport = doc.GetElement(vport_id)
except:
TaskDialog.Show('pyRevit', 'Select at least one viewport. No more, no less!')
if isinstance(vport, Viewport):
view = doc.GetElement(vport.ViewId)
if isinstance(vport, Viewport):
set_tansform_matrix(vport, view)
if view is not None and isinstance(view, ViewPlan):
with TransactionGroup(doc, 'Paste Viewport Location') as tg:
tg.Start()
set_tansform_matrix(vport, view)
try:
with open(datafile, 'rb') as fp:
originalviewtype = pickle.load(fp)
if originalviewtype == 'ViewPlan':
savedcen_pt = pickle.load(fp)
savedmdl_pt = pickle.load(fp)
else:
raise OriginalIsViewDrafting
except IOError:
TaskDialog.Show('pyRevit',
'Could not find saved viewport placement.\nCopy a Viewport Placement first.')
except OriginalIsViewDrafting:
TaskDialog.Show('pyRevit',
'Viewport placement info is from a drafting view and can not be applied here.')
else:
savedcenter_pt = XYZ(savedcen_pt.X, savedcen_pt.Y, savedcen_pt.Z)
savedmodel_pt = XYZ(savedmdl_pt.X, savedmdl_pt.Y, savedmdl_pt.Z)
with Transaction(doc, 'Apply Viewport Placement') as t:
t.Start()
center = vport.GetBoxCenter()
centerdiff = view_to_sheet_transform(savedmodel_pt) - center
vport.SetBoxCenter(savedcenter_pt - centerdiff)
if PINAFTERSET:
vport.Pinned = True
t.Commit()
tg.Assimilate()
elif view is not None and isinstance(view, ViewDrafting):
try:
with open(datafile, 'rb') as fp:
savedcen_pt = pickle.load(fp)
savedmdl_pt = pickle.load(fp)
originalviewtype = pickle.load(fp)
if originalviewtype == 'ViewDrafting':
savedcen_pt = pickle.load(fp)
else:
raise OriginalIsViewPlan
except IOError:
TaskDialog.Show('pyRevit', 'Could not find saved viewport placement.\nCopy a Viewport Placement first.')
TaskDialog.Show('pyRevit',
'Could not find saved viewport placement.\nCopy a Viewport Placement first.')
except OriginalIsViewPlan:
TaskDialog.Show('pyRevit',
'Viewport placement info is from a model view and can not be applied here.')
else:
savedcenter_pt = XYZ(savedcen_pt.X, savedcen_pt.Y, savedcen_pt.Z)
savedmodel_pt = XYZ(savedmdl_pt.X, savedmdl_pt.Y, savedmdl_pt.Z)
t = Transaction(doc, 'Paste Viewport Placement')
t.Start()
center = vport.GetBoxCenter()
centerdiff = view_to_sheet_transform(savedmodel_pt) - center
vport.SetBoxCenter(savedcenter_pt - centerdiff)
# vport.Pinned = True
t.Commit()

with Transaction(doc, 'Apply Viewport Placement') as t:
t.Start()
vport.SetBoxCenter(savedcenter_pt)
if PINAFTERSET:
vport.Pinned = True
t.Commit()
else:
TaskDialog.Show('pyRevit', 'This tool only works with Plan, RCP, and Detail views and viewports.')
else:
TaskDialog.Show('pyRevit', 'Select at least one viewport. No more, no less!')
2 changes: 1 addition & 1 deletion pyRevit/Settings_aboutPyRevit.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@

print('''
About pyRevit...
Version 2.49.53
Version 2.49.55
This packages is prepared by Ehsan Iran-Nejad ([email protected]) and is licensed under GNU GENERAL PUBLIC LICENSE.
See full license here: https://github.com/eirannejad/pyRevit/blob/master/LICENSE.
Expand Down

0 comments on commit b7edd0b

Please sign in to comment.