Compare commits
603 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| f75afc2b37 | |||
| 34c922feb1 | |||
| cee05260c1 | |||
| add470699b | |||
| f5bcd805e8 | |||
| cf4bb14240 | |||
| 1ee1650fef | |||
| 70f5f672a6 | |||
| b329ec8c97 | |||
| b54cfe16e8 | |||
| 357859827c | |||
| f35457dff8 | |||
| f993c38ea9 | |||
| 624537cc5d | |||
| ebb7f1b3bf | |||
| ac2a95d968 | |||
| 2440c44f44 | |||
| 33dfa1229c | |||
| ea61bd06b8 | |||
| e071aca299 | |||
| 4a8a980034 | |||
| b05447dc30 | |||
| 7a36f9ec08 | |||
| 80e3971706 | |||
| dc770b7a79 | |||
| f8e7d391be | |||
| 3092ba3056 | |||
| 9d10006116 | |||
| 95f4d051d6 | |||
| c79ad8e87d | |||
| 9797dfbfc0 | |||
| 63b00a6257 | |||
| 36091845a6 | |||
| 89e1855e2c | |||
| b7f5725282 | |||
| dc8c8cedf4 | |||
| 31e8b838dd | |||
| baf7f32c2a | |||
| ad1d58bd4c | |||
| ec86688750 | |||
| 84098f4c42 | |||
| 77f9d73698 | |||
| 812e8dd2f3 | |||
| 5c19d9aa16 | |||
| 29d706e1b6 | |||
| 25d02673a7 | |||
| 6a4dc62622 | |||
| bb9a5ea604 | |||
| 2e995bd0fa | |||
| 38e6096ea9 | |||
| 30e3398cd4 | |||
| 3f7e98aff5 | |||
| 1ad8429928 | |||
| a6c820183b | |||
| 56b6c813c0 | |||
| 8ab110f7ec | |||
| 11ff018f18 | |||
| 227f63d266 | |||
| 9e8aaf4f3b | |||
| afcb760bbf | |||
| 58283439ab | |||
| 0c29a2ec0a | |||
| 4ec62d4168 | |||
| 8d596823ed | |||
| ccd62e3452 | |||
| 1bd08497e6 | |||
| d23cc5a738 | |||
| 3e2ac4b5b6 | |||
| 928bc15ff1 | |||
| e410e40060 | |||
| d1f2c938b1 | |||
| 388ec2bdfd | |||
| b057c6c0da | |||
| 40089bdbb8 | |||
| 49dd688219 | |||
| 6993e8cb83 | |||
| 709015b9d8 | |||
| c5e0dfa36b | |||
| 1f72741b62 | |||
| 0f8f7e02be | |||
| f3e188b4f8 | |||
| 346b0210a5 | |||
| 5d072bbb1d | |||
| fcc8527c9f | |||
| 2814dabe14 | |||
| 0f6150b272 | |||
| e3bc770369 | |||
| 1bb6ce7b63 | |||
| 6bf05f6ffd | |||
| 684c868cd0 | |||
| 27009a61e9 | |||
| 27ff32f584 | |||
| 223ede3ec7 | |||
| 3b2e609888 | |||
| f4caa55298 | |||
| 166d686b9d | |||
| f3f43eebd5 | |||
| 3c16b35232 | |||
| 86fb480a15 | |||
| abef8f85d9 | |||
| 33a7009585 | |||
| 0249ecb313 | |||
| bbf8a3b45e | |||
| f1eec55633 | |||
| f2bc9a9701 | |||
| 1eb662c6a7 | |||
| c08d4c398f | |||
| a81a44ca1d | |||
| c4c4431ed2 | |||
| fd23d40a5c | |||
| f2d6e03ad8 | |||
| d5c9097afa | |||
| 90cd0706ba | |||
| 2daf0de073 | |||
| 3653ece109 | |||
| c29e0bf8ee | |||
| 95a84eda53 | |||
| 8116402023 | |||
| f6238eb29c | |||
| 0a4c9ec380 | |||
| 4b8b1393ba | |||
| ee01b28645 | |||
| 06225a8aab | |||
| 1129d169cb | |||
| a787bfb8a3 | |||
| 700ec92ed2 | |||
| 7fbf0a827b | |||
| b6f659e91f | |||
| e9628cde77 | |||
| cd919353f7 | |||
| 6400138a3c | |||
| 9b8ddcc706 | |||
| ba3e0b275e | |||
| 45a68b3c5f | |||
| 49379a3e33 | |||
| b30ffdc7d1 | |||
| 2e53a2fea4 | |||
| 9dc51ee1dc | |||
| 58e7ac5138 | |||
| 6bc9bfdc28 | |||
| cce1f3b092 | |||
| ebcbe29398 | |||
| 59b6b21736 | |||
| 6c9823776a | |||
| c77e7642b7 | |||
| 9240fce897 | |||
| a87b7a830e | |||
| aad7943f1a | |||
| 589b309d91 | |||
| 95cb7c35de | |||
| aab3f1792a | |||
| 80a6e424e4 | |||
| f08cac36e7 | |||
| aa01b5ccf4 | |||
| da7d260542 | |||
| 64e0f65e29 | |||
| fedf036e2f | |||
| b9df9b09de | |||
| 3afdb916b3 | |||
| 8d7b0adea4 | |||
| 7a6d956ccf | |||
| d272743642 | |||
| c3ce77fe4d | |||
| 564aa58ebf | |||
| 48fe756fcd | |||
| 8e2f546cd6 | |||
| 3219172fed | |||
| 098d576a0b | |||
| 1950c412c2 | |||
| b646b2c7b1 | |||
| 19f22296b4 | |||
| 240fb6cbbe | |||
| 5ca6a3f2c2 | |||
| 06fef6ca91 | |||
| dcbe2d7814 | |||
| b66850b994 | |||
| 6a337ccdad | |||
| 92fd5387ef | |||
| 6205afdda5 | |||
| d060c3b8cf | |||
| 5dd3dd713a | |||
| 4fc1053af7 | |||
| 80204faedf | |||
| 06a47a2f9b | |||
| b65a7a3d4a | |||
| 7064574387 | |||
| 169ae4d32a | |||
| 4d44b41e2f | |||
| a1f835dc77 | |||
| e2172216a5 | |||
| 965c3e9c6e | |||
| 65e4812ba1 | |||
| 87df86f723 | |||
| fd32371be3 | |||
| 19c1334bb3 | |||
| 7a36450143 | |||
| d37fce644b | |||
| 00bcefba56 | |||
| d53bc064e1 | |||
| 50affbedf1 | |||
| f00aeecead | |||
| bbf2ee79be | |||
| abb1f042d4 | |||
| 04b4e02d05 | |||
| 144990db7a | |||
| a5592cdd7d | |||
| 249e3a0a84 | |||
| f869139d2a | |||
| 3954369db5 | |||
| 7165d99b76 | |||
| 8234db872d | |||
| 9acc84387f | |||
| 4eb1f2b773 | |||
| 4d60607a92 | |||
| 09f809fc8f | |||
| 9f6833c36e | |||
| 1cc8ed804a | |||
| eb0f036641 | |||
| f140c41af6 | |||
| 8a91634f55 | |||
| b025819307 | |||
| 3ec98d3dd2 | |||
| ba182d48c4 | |||
| f35853cff1 | |||
| d64ad50d32 | |||
| ab0b012f58 | |||
| dfa8ff2929 | |||
| 14b628cd3d | |||
| dfc1d4aaf7 | |||
| 8c39da370f | |||
| 532eba37a5 | |||
| 39207e5716 | |||
| 48ee5c1ffe | |||
| b41eb812ab | |||
| fece1a0151 | |||
| 3a9e03b982 | |||
| 96f0db5f63 | |||
| 56a43bca06 | |||
| 5f451f0143 | |||
| 4cda3ff143 | |||
| 167a691294 | |||
| 2f320d360e | |||
| 25ebdb8c3b | |||
| afd52e788f | |||
| 7997eedb05 | |||
| e98aecaa82 | |||
| 73e44c6b0e | |||
| c2b7752539 | |||
| 050fc61377 | |||
| 3ab4774dc8 | |||
| d9727005cf | |||
| f73fc604d0 | |||
| 4bd3dcaf61 | |||
| a2ecf1d20e | |||
| 74228b13dd | |||
| 5aa39fefa4 | |||
| 0205636b28 | |||
| 9869ed3b63 | |||
| 6a29e0c7cd | |||
| 011e1b90c1 | |||
| a1189e3878 | |||
| 4fb801b44c | |||
| 80eeb2de39 | |||
| 753760fee4 | |||
| 8fbb01dee4 | |||
| 41f2b93347 | |||
| 98a6f3251d | |||
| 8287ed2e57 | |||
| 3f5cf49971 | |||
| efc2c0b07c | |||
| 242625ebe3 | |||
| 5f5af2b15a | |||
| ff33b82013 | |||
| dbe059f507 | |||
| 9d19e52037 | |||
| f0495ab093 | |||
| e57dacc6ef | |||
| 69dbf2f117 | |||
| 3cf3867877 | |||
| 3577f3fd6a | |||
| 242af16e81 | |||
| 20f36ffe19 | |||
| b103d0da09 | |||
| 4add9c5d72 | |||
| aecb15e549 | |||
| bff8140559 | |||
| fe98f71be2 | |||
| f2008502f3 | |||
| 2981c1270b | |||
| 3011921df9 | |||
| 79ecba213a | |||
| 06db6653fb | |||
| 86200091dc | |||
| 4b2a678605 | |||
| 1ca2c03ec0 | |||
| 528b81d294 | |||
| d0dd57a731 | |||
| 25258fd39f | |||
| a884f7a6ea | |||
| 6e0df0d4e4 | |||
| 58ff3e667e | |||
| 3f5c933ee9 | |||
| d137c7b991 | |||
| 5eec02296b | |||
| 681acd81c8 | |||
| 09dd819504 | |||
| 211296c803 | |||
| 0fd5448342 | |||
| 3002d7a31b | |||
| 8eee1ede58 | |||
| 75763d0929 | |||
| 2f2a67a569 | |||
| 7246d239be | |||
| 4d1fe83c1e | |||
| 1097bba539 | |||
| eb25b6d821 | |||
| 80515fdc69 | |||
| 4a340ef1ae | |||
| b6a96802a8 | |||
| f2a0ffa9ee | |||
| 24479811f7 | |||
| 2153db9704 | |||
| dbcc820304 | |||
| 830632fa1e | |||
| 2f57ca96ca | |||
| 1b9ee91880 | |||
| 8fb7519e7b | |||
| 9dce548a05 | |||
| 487253babe | |||
| f245584428 | |||
| 1ac784d290 | |||
| 314a962014 | |||
| 6b07a0fff4 | |||
| d3208de754 | |||
| 4ba19231b7 | |||
| a93ac797fc | |||
| 4569b1e623 | |||
| ae3222683e | |||
| 780184c562 | |||
| 310b29292d | |||
| 8593200a48 | |||
| e53ede3349 | |||
| 892396f1e9 | |||
| c2ba9bc8f5 | |||
| 3f8ec1d259 | |||
| fd203bbfea | |||
| 5ac0c64c9e | |||
| db367d92cb | |||
| 289ed58812 | |||
| d1a65510ef | |||
| 4f58f77c92 | |||
| 10084b30da | |||
| 88f1365fa6 | |||
| f767f53c24 | |||
| 9322e4bfe8 | |||
| 31a9452ef0 | |||
| c69f70c7ac | |||
| 7ff0778c7e | |||
| 0cd08b5cef | |||
| 46eed71a56 | |||
| 623270117e | |||
| c5d081a765 | |||
| b2d9aba778 | |||
| 12b8bb20d5 | |||
| 6e6214133a | |||
| 2c35c4c1b2 | |||
| 80f9c54111 | |||
| 508cf6790f | |||
| 1c20949f58 | |||
| 88e17d66b8 | |||
| 625dd64f87 | |||
| b22d94606c | |||
| 309a657420 | |||
| 95d79391dd | |||
| 49ec641f7d | |||
| 3410e1613b | |||
| b013855c97 | |||
| 78f1a4e23e | |||
| bf4afe0180 | |||
| 7714e8ba28 | |||
| 5ff0965ee6 | |||
| 0add4fabec | |||
| 8df7559e76 | |||
| a1f5d0f21f | |||
| 1fe08136de | |||
| 7ef35cd4e1 | |||
| f9a75772ed | |||
| b85f805330 | |||
| 6af53c56b3 | |||
| 100dc9213b | |||
| 8068962cdf | |||
| aad9246463 | |||
| 0dbf691f16 | |||
| 9628ab5bdf | |||
| db195a4285 | |||
| 7d52ac9568 | |||
| 8c08bf87cc | |||
| 1c979f821c | |||
| 8589c7b467 | |||
| 64bf623ee4 | |||
| 36069e2908 | |||
| 54e129d0b4 | |||
| cffcaf74a1 | |||
| 1bd60a92b6 | |||
| 37fc1ff57f | |||
| 9bfc73a53d | |||
| 32f3415680 | |||
| ede16df1b1 | |||
| 0f067ce968 | |||
| 0e03226a55 | |||
| f5b436ea62 | |||
| 70de8b7a45 | |||
| 87dea86ba6 | |||
| d8eef2b51c | |||
| f452562fff | |||
| a559158931 | |||
| cc08df5c88 | |||
| 4aff5aaca9 | |||
| ea2243a14a | |||
| 15f88773d2 | |||
| e703116e78 | |||
| ae081295d5 | |||
| 2ae25f22be | |||
| 5c2ecc6f97 | |||
| 2ee6636dde | |||
| fdd05f7958 | |||
| 62e3e80f65 | |||
| 6953a34645 | |||
| 8d704b1034 | |||
| 816ff52669 | |||
| 105ef9b713 | |||
| a38f82a91a | |||
| 1388fb3c5a | |||
| 5d1be43263 | |||
| 1da76dadf8 | |||
| 188efd7ea5 | |||
| f5f5c513a6 | |||
| 70e0e1e727 | |||
| 9301186b63 | |||
| 0689cf34a1 | |||
| 8299ca84af | |||
| 5f1228091e | |||
| 7afb2ec18a | |||
| 758c6f48cd | |||
| 44ba054e07 | |||
| 0f2b208b90 | |||
| accdd00880 | |||
| de4ed8e55a | |||
| 5bd46de070 | |||
| fb23cc3eaf | |||
| fafa529df4 | |||
| 5ed98f7acf | |||
| e542a7d99b | |||
| 9635f04db8 | |||
| cd40e32b4e | |||
| 8319a73edf | |||
| aca7547f6c | |||
| c4061182f9 | |||
| d1dcf86357 | |||
| 70d52db17f | |||
| d2392b0d2b | |||
| 14bf10f1bb | |||
| 10d94e5d28 | |||
| dff6b3101e | |||
| 497b04a70b | |||
| 31137a9fd8 | |||
| f3f65a037a | |||
| a680bea021 | |||
| 1138cc12d4 | |||
| 13e995db53 | |||
| 572eeecb3a | |||
| f0d39dc39f | |||
| 74e84f803d | |||
| 79f09e5364 | |||
| 57f671dd60 | |||
| d862ace188 | |||
| fd46280130 | |||
| 0c68eb1a6a | |||
| 11e4860364 | |||
| f8474777c0 | |||
| 242476a43a | |||
| 75398aa830 | |||
| 9827c46988 | |||
| 8c7908b4ef | |||
| 76aaf2fd41 | |||
| a8b2500c0a | |||
| 3a863cd0dd | |||
| a57fbe6e3d | |||
| f346a3918c | |||
| e13a910700 | |||
| 71320a5acc | |||
| 6f409ee228 | |||
| 44f3c88c81 | |||
| 8b3aaefe8c | |||
| f63345e304 | |||
| f81752b41e | |||
| 482a3189d8 | |||
| c8714d0df8 | |||
| fe69091c5c | |||
| 457380bc3c | |||
| 6613e1a7a6 | |||
| 027df4f5d9 | |||
| 171105f827 | |||
| f2363586aa | |||
| 28a7a02ee5 | |||
| dce78ceeca | |||
| a5824702ab | |||
| bb8486c94a | |||
| d32fc23e14 | |||
| 3e85a018fc | |||
| dd2e222c84 | |||
| bcdabb1226 | |||
| 8c1a5b4463 | |||
| 4811329d9e | |||
| 6c3ab4baef | |||
| a7295e7b25 | |||
| fb8fda27c5 | |||
| 32b114274c | |||
| 02a9da050f | |||
| c6ba0ff86d | |||
| 0d386aa93d | |||
| d439f65463 | |||
| 345bae9463 | |||
| cb1f9c0480 | |||
| 2be74ce617 | |||
| 56675ef88d | |||
| 4c381bd809 | |||
| 8c3885ece8 | |||
| 15bd3f5070 | |||
| 6fd4571d34 | |||
| 5081177653 | |||
| 0d0ca2c811 | |||
| 230e27a162 | |||
| 669fd19c2e | |||
| 139f8ccb33 | |||
| 7f625bd468 | |||
| 4d07ba7637 | |||
| 7431b57e0e | |||
| 57c19ba3c5 | |||
| 8e3c2ece2f | |||
| cfc58d9456 | |||
| e74a6cebb1 | |||
| 5e01d5a976 | |||
| a2f7ab422f | |||
| 8c58d9d14c | |||
| 90e61b6dc1 | |||
| 5c479e4c0e | |||
| 97d20ad7b1 | |||
| 2800b84747 | |||
| 511d69314e | |||
| 24e7f02213 | |||
| c1d7947085 | |||
| 21281e5d77 | |||
| 29bbdc69a2 | |||
| efe6e6a4a0 | |||
| f036109020 | |||
| 86bc2dc590 | |||
| a34b6ad0c2 | |||
| e436949ef9 | |||
| 6d8f4a4a80 | |||
| dabb65427a | |||
| 57ece17e8b | |||
| 4362f737d0 | |||
| b55df58313 | |||
| afa6722253 | |||
| a3d4881578 | |||
| 1af158a5e0 | |||
| 47857a9db0 | |||
| 3b026e6027 | |||
| d572609f75 | |||
| 37032cc7aa | |||
| 6027325878 | |||
| 5ddb2aa052 | |||
| 67a18821cc | |||
| 2688a69286 | |||
| 56216a6137 | |||
| 319cbf8960 | |||
| d7ac6c0b95 | |||
| 201ca5f26e | |||
| 89528437b1 | |||
| 91bde24fe9 | |||
| 991b0f9ff1 | |||
| ee1715ff8a | |||
| 70ee09b9bb | |||
| 83dd62d03f | |||
| 94cc0ac3f7 | |||
| 36cb94d3d7 | |||
| c60baf78c5 | |||
| d72cfd3522 | |||
| a26618a4f7 | |||
| eaf370407d | |||
| a2b50fe5a1 | |||
| 7e62f76841 | |||
| fc804f16d3 | |||
| 6c7da24595 | |||
| b284d39328 | |||
| 907185c9bb | |||
| a189a2e1c0 | |||
| 1fad926275 | |||
| 99c147fe2f | |||
| e2adf710b3 | |||
| 9509344533 | |||
| 6fabc6cae6 |
+9
-299
@@ -1,307 +1,17 @@
|
||||
version: 2.1
|
||||
|
||||
orbs:
|
||||
win: circleci/windows@5.0.0
|
||||
|
||||
# Define the jobs we want to run for this project
|
||||
jobs:
|
||||
package-connector:
|
||||
build:
|
||||
docker:
|
||||
- image: cimg/python:3.11.0
|
||||
- image: cimg/base:2023.03
|
||||
steps:
|
||||
- checkout
|
||||
- run:
|
||||
name: Setup SEMVER value
|
||||
command: |
|
||||
SEMVER=$(if [ "${CIRCLE_TAG}" ]; then echo $CIRCLE_TAG; else echo "0.0.0"; fi;)
|
||||
echo $SEMVER > ./SEMVER
|
||||
python3 patch_version.py $SEMVER
|
||||
- run:
|
||||
name: install dependencies
|
||||
command: poetry install --only main
|
||||
- run:
|
||||
name: export package dependencies
|
||||
command: ./export_dependencies.sh
|
||||
- persist_to_workspace:
|
||||
root: ./
|
||||
paths:
|
||||
- bpy_speckle
|
||||
- patch_installer.py
|
||||
- SEMVER
|
||||
|
||||
build-connector-zip:
|
||||
docker:
|
||||
- image: cimg/python:3.11.0
|
||||
steps:
|
||||
- attach_workspace:
|
||||
at: ./
|
||||
- run: &restore_semver
|
||||
name: Restore Semver
|
||||
command: SEMVER=$(cat ./SEMVER) && echo $SEMVER
|
||||
- run:
|
||||
name: Package to Zip
|
||||
command: zip -r bpy_speckle.zip bpy_speckle
|
||||
- persist_to_workspace:
|
||||
root: ./
|
||||
paths:
|
||||
- bpy_speckle.zip
|
||||
|
||||
get-ci-tools: # Clones our ci tools and persists them to the workspace
|
||||
docker:
|
||||
- image: cimg/base:2021.01
|
||||
steps:
|
||||
- add_ssh_keys:
|
||||
fingerprints:
|
||||
- "77:64:03:93:c5:f3:1d:a6:fd:bd:fb:d1:05:56:ca:e9"
|
||||
- run:
|
||||
name: I know Github as a host
|
||||
command: |
|
||||
mkdir ~/.ssh
|
||||
touch ~/.ssh/known_hosts
|
||||
ssh-keyscan github.com >> ~/.ssh/known_hosts
|
||||
- run:
|
||||
name: Clone
|
||||
command: git clone git@github.com:specklesystems/speckle-sharp-ci-tools.git speckle-sharp-ci-tools
|
||||
- run:
|
||||
command: cd speckle-sharp-ci-tools
|
||||
- persist_to_workspace:
|
||||
root: ./
|
||||
paths:
|
||||
- speckle-sharp-ci-tools
|
||||
|
||||
build-installer-win:
|
||||
executor:
|
||||
name: win/default
|
||||
shell: cmd.exe
|
||||
steps:
|
||||
- attach_workspace:
|
||||
at: ./
|
||||
- run:
|
||||
name: Patch installer
|
||||
shell: powershell.exe
|
||||
command: python patch_installer.py (Get-Content -Raw SEMVER)
|
||||
- run:
|
||||
name: Create Innosetup signing cert
|
||||
shell: powershell.exe
|
||||
command: |
|
||||
echo $env:PFX_B64 > "speckle-sharp-ci-tools\SignTool\AEC Systems Ltd.txt"
|
||||
certutil -decode "speckle-sharp-ci-tools\SignTool\AEC Systems Ltd.txt" "speckle-sharp-ci-tools\SignTool\AEC Systems Ltd.pfx"
|
||||
- run:
|
||||
name: Installer
|
||||
shell: cmd.exe #does not work in powershell
|
||||
command: speckle-sharp-ci-tools\InnoSetup\ISCC.exe speckle-sharp-ci-tools\blender.iss /Sbyparam=$p
|
||||
- persist_to_workspace:
|
||||
root: ./
|
||||
paths:
|
||||
- speckle-sharp-ci-tools/Installers/blender/blender-*.exe
|
||||
|
||||
build-installer-mac:
|
||||
macos:
|
||||
xcode: 12.5.1
|
||||
parameters:
|
||||
runtime:
|
||||
type: string
|
||||
slug:
|
||||
type: string
|
||||
installer_path:
|
||||
type: string
|
||||
default: speckle-sharp-ci-tools/Mac/SpeckleBlenderInstall
|
||||
steps:
|
||||
- checkout
|
||||
- attach_workspace:
|
||||
at: ./
|
||||
- run:
|
||||
name: Install mono
|
||||
command: |
|
||||
HOMEBREW_NO_AUTO_UPDATE=1 brew install mono
|
||||
# Compress build files
|
||||
- run:
|
||||
name: Install dotnet
|
||||
command: curl -sSL https://dot.net/v1/dotnet-install.sh | bash /dev/stdin
|
||||
- run: *restore_semver
|
||||
- run:
|
||||
name: Copy connector files to installer
|
||||
command: |
|
||||
mkdir -p <<parameters.installer_path >>/.installationFiles/
|
||||
cp bpy_speckle.zip << parameters.installer_path >>/.installationFiles
|
||||
- run:
|
||||
name: Build Mac installer
|
||||
command: ~/.dotnet/dotnet publish << parameters.installer_path >>/SpeckleBlenderInstall.sln -r << parameters.runtime >> -c Release
|
||||
- run:
|
||||
name: Zip installer
|
||||
command: |
|
||||
SEMVER=$(cat ./SEMVER)
|
||||
echo $SEMVER
|
||||
mkdir -p speckle-sharp-ci-tools/Installers/blender
|
||||
(cd <<parameters.installer_path>>/bin/Release/net6.0/<< parameters.runtime >>/publish/ && zip -r - ./) > << parameters.slug >>-${SEMVER}.zip
|
||||
cp << parameters.slug >>-${SEMVER}.zip speckle-sharp-ci-tools/Installers/blender/
|
||||
|
||||
- persist_to_workspace:
|
||||
root: ./
|
||||
paths:
|
||||
- speckle-sharp-ci-tools/Installers/blender/<< parameters.slug >>*.zip
|
||||
|
||||
build-installer-manual:
|
||||
docker:
|
||||
- image: cimg/base:2021.01
|
||||
parameters:
|
||||
slug:
|
||||
type: string
|
||||
default: bpy_speckle
|
||||
steps:
|
||||
- attach_workspace:
|
||||
at: ./
|
||||
- run: *restore_semver
|
||||
- run:
|
||||
name: Copy zip with semver
|
||||
command: |
|
||||
SEMVER=$(cat ./SEMVER)
|
||||
mkdir -p speckle-sharp-ci-tools/Installers/blender
|
||||
cp bpy_speckle.zip speckle-sharp-ci-tools/Installers/blender/<< parameters.slug >>-${SEMVER}.zip
|
||||
- persist_to_workspace:
|
||||
root: ./
|
||||
paths:
|
||||
- speckle-sharp-ci-tools/Installers/blender/<< parameters.slug >>*.zip
|
||||
|
||||
deploy-connector:
|
||||
docker:
|
||||
- image: mcr.microsoft.com/dotnet/sdk:6.0
|
||||
parameters:
|
||||
file_slug:
|
||||
type: string
|
||||
os:
|
||||
type: string
|
||||
extension:
|
||||
type: string
|
||||
arch:
|
||||
type: string
|
||||
default: Any
|
||||
steps:
|
||||
- checkout
|
||||
- attach_workspace:
|
||||
at: ./
|
||||
- run:
|
||||
name: Install Manager Feed CLI
|
||||
command: dotnet tool install --global Speckle.Manager.Feed
|
||||
- run: *restore_semver
|
||||
- run:
|
||||
name: Upload new version
|
||||
# this is where the installer gets the semver baked into the file name
|
||||
command: |
|
||||
SEMVER=$(cat ./SEMVER)
|
||||
echo $SEMVER
|
||||
/root/.dotnet/tools/Speckle.Manager.Feed deploy \
|
||||
-s blender \
|
||||
-v ${SEMVER} \
|
||||
-u https://releases.speckle.dev/installers/blender/<< parameters.file_slug >>-${SEMVER}.<< parameters.extension >> \
|
||||
-o << parameters.os >> \
|
||||
-a << parameters.arch >> \
|
||||
-f speckle-sharp-ci-tools/Installers/blender/<< parameters.file_slug >>-${SEMVER}.<< parameters.extension >>
|
||||
- run: echo "so long and thanks for all the fish"
|
||||
|
||||
# Orchestrate our job run sequence
|
||||
workflows:
|
||||
build: # build the installers, but don't persist to workspace for deployment
|
||||
build_and_test:
|
||||
when:
|
||||
false
|
||||
jobs:
|
||||
- package-connector:
|
||||
filters: &build_filters
|
||||
tags:
|
||||
only: /([0-9]+)\.([0-9]+)\.([0-9]+)(?:-\w+)?(?:\.[0-9]+)?/
|
||||
|
||||
- build-connector-zip:
|
||||
requires:
|
||||
- package-connector
|
||||
filters: *build_filters
|
||||
|
||||
- get-ci-tools:
|
||||
filters: *build_filters
|
||||
|
||||
- build-installer-win:
|
||||
context: innosetup
|
||||
name: Windows Installer Build
|
||||
requires:
|
||||
- package-connector
|
||||
- get-ci-tools
|
||||
filters: *build_filters
|
||||
|
||||
- deploy-connector:
|
||||
context: do-spaces-speckle-releases
|
||||
name: deploy-windows
|
||||
file_slug: blender
|
||||
os: WIN
|
||||
arch: Any
|
||||
extension: exe
|
||||
requires:
|
||||
- Manual Installer Build
|
||||
- Windows Installer Build
|
||||
- Mac Intel Build
|
||||
- Mac ARM Build
|
||||
filters: &deploy_filters
|
||||
branches:
|
||||
ignore: /.*/
|
||||
tags:
|
||||
only: /([0-9]+)\.([0-9]+)\.([0-9]+)(?:-\w+)?(?:\.[0-9]+)?/
|
||||
|
||||
- build-installer-mac:
|
||||
name: Mac ARM Build
|
||||
slug: blender-mac-arm
|
||||
runtime: osx-arm64
|
||||
requires:
|
||||
- get-ci-tools
|
||||
- build-connector-zip
|
||||
filters: *build_filters
|
||||
|
||||
- deploy-connector:
|
||||
context: do-spaces-speckle-releases
|
||||
name: deploy-mac-arm
|
||||
file_slug: blender-mac-arm
|
||||
os: OSX
|
||||
arch: Arm
|
||||
extension: zip
|
||||
requires:
|
||||
- Manual Installer Build
|
||||
- Windows Installer Build
|
||||
- Mac Intel Build
|
||||
- Mac ARM Build
|
||||
filters: *deploy_filters
|
||||
|
||||
- build-installer-mac:
|
||||
name: Mac Intel Build
|
||||
slug: blender-mac-intel
|
||||
runtime: osx-x64
|
||||
requires:
|
||||
- get-ci-tools
|
||||
- build-connector-zip
|
||||
filters: *build_filters
|
||||
|
||||
- deploy-connector:
|
||||
context: do-spaces-speckle-releases
|
||||
name: deploy-mac-intel
|
||||
file_slug: blender-mac-intel
|
||||
os: OSX
|
||||
arch: Intel
|
||||
extension: zip
|
||||
requires:
|
||||
- Manual Installer Build
|
||||
- Windows Installer Build
|
||||
- Mac Intel Build
|
||||
- Mac ARM Build
|
||||
filters: *deploy_filters
|
||||
|
||||
- build-installer-manual:
|
||||
name: Manual Installer Build
|
||||
requires:
|
||||
- get-ci-tools
|
||||
- build-connector-zip
|
||||
filters: *build_filters
|
||||
|
||||
- deploy-connector:
|
||||
context: do-spaces-speckle-releases
|
||||
name: deploy-manual
|
||||
file_slug: bpy_speckle
|
||||
os: Any
|
||||
arch: Any
|
||||
extension: zip
|
||||
requires:
|
||||
- Manual Installer Build
|
||||
- Windows Installer Build
|
||||
- Mac Intel Build
|
||||
- Mac ARM Build
|
||||
filters: *deploy_filters
|
||||
- build
|
||||
|
||||
@@ -1,12 +0,0 @@
|
||||
name: Update issue Status
|
||||
|
||||
on:
|
||||
issues:
|
||||
types: [closed]
|
||||
|
||||
jobs:
|
||||
update_issue:
|
||||
uses: specklesystems/github-actions/.github/workflows/project-add-issue.yml@main
|
||||
secrets: inherit
|
||||
with:
|
||||
issue-id: ${{ github.event.issue.node_id }}
|
||||
@@ -1,12 +0,0 @@
|
||||
name: Move new issues into Project
|
||||
|
||||
on:
|
||||
issues:
|
||||
types: [opened]
|
||||
|
||||
jobs:
|
||||
track_issue:
|
||||
uses: specklesystems/github-actions/.github/workflows/project-add-issue.yml@main
|
||||
secrets: inherit
|
||||
with:
|
||||
issue-id: ${{ github.event.issue.node_id }}
|
||||
@@ -0,0 +1,31 @@
|
||||
name: "PR workflow"
|
||||
on:
|
||||
pull_request:
|
||||
branches:
|
||||
- "v3-dev"
|
||||
jobs:
|
||||
build:
|
||||
name: Pre-commit Checks
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
- name: Install uv and set the python version
|
||||
uses: astral-sh/setup-uv@v5
|
||||
with:
|
||||
python-version: "3.11"
|
||||
enable-cache: true
|
||||
cache-dependency-glob: "uv.lock"
|
||||
|
||||
- name: Install the project
|
||||
run: uv sync --all-extras --dev
|
||||
|
||||
- uses: actions/cache@v3
|
||||
with:
|
||||
path: ~/.cache/pre-commit/
|
||||
key: ${{ hashFiles('.pre-commit-config.yaml') }}
|
||||
|
||||
- name: Run pre-commit
|
||||
run: uv run pre-commit run --all-files
|
||||
|
||||
- name: Minimize uv cache
|
||||
run: uv cache prune --ci
|
||||
@@ -0,0 +1,96 @@
|
||||
name: "Release workflow"
|
||||
on:
|
||||
push:
|
||||
branches: ["main", "installer-test/**"]
|
||||
tags: ["v3.*.*"]
|
||||
|
||||
jobs:
|
||||
build:
|
||||
name: Build Zip
|
||||
runs-on: ubuntu-latest
|
||||
env:
|
||||
ZIP_NAME: "blender.zip"
|
||||
SEMVER: null
|
||||
FILE_VERSION: null
|
||||
outputs:
|
||||
semver: ${{ steps.set-version.outputs.semver }}
|
||||
fileVersion: ${{ steps.set-version.outputs.fileVersion }}
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
- name: 🐍 Install uv and set the python version
|
||||
uses: astral-sh/setup-uv@v5
|
||||
with:
|
||||
python-version: "3.11"
|
||||
enable-cache: true
|
||||
cache-dependency-glob: "uv.lock"
|
||||
|
||||
- id: set-version
|
||||
name: Set version to output
|
||||
run: |
|
||||
TAG=${{ github.ref_name }}
|
||||
if [[ "${{ github.ref }}" != refs/tags/* ]]; then
|
||||
TAG="v3.0.99.${{ github.run_number }}"
|
||||
fi
|
||||
SEMVER="${TAG#v}"
|
||||
FILE_VERSION=$(echo "$TAG" | sed -E 's/^v([0-9]+\.[0-9]+\.[0-9]+).*/\1/')
|
||||
FILE_VERSION="$FILE_VERSION.${{ github.run_number }}"
|
||||
|
||||
echo "semver=$SEMVER" >> "$GITHUB_OUTPUT"
|
||||
echo "fileVersion=$FILE_VERSION" >> "$GITHUB_OUTPUT"
|
||||
|
||||
echo $SEMVER
|
||||
echo $FILE_VERSION
|
||||
|
||||
- name: ✏ Patch Version
|
||||
run: python patch_version.py ${{ steps.set-version.outputs.fileVersion }}
|
||||
|
||||
- name: 🔄 UV Sync
|
||||
run: uv sync --all-extras --dev
|
||||
|
||||
- name: 📄 Export Package Dependencies
|
||||
run: ./export_dependencies.sh
|
||||
|
||||
- name: 🗃 Zip Package
|
||||
run: zip -r ${{env.ZIP_NAME}} bpy_speckle
|
||||
|
||||
- name: ⬆️ Upload artifacts
|
||||
uses: actions/upload-artifact@v4
|
||||
with:
|
||||
name: output-${{ steps.set-version.outputs.semver }}
|
||||
path: ${{env.ZIP_NAME}}
|
||||
if-no-files-found: error
|
||||
retention-days: 1
|
||||
compression-level: 0 # no compression
|
||||
|
||||
- name: 💾 Minimize uv cache
|
||||
run: uv cache prune --ci
|
||||
|
||||
deploy-installers:
|
||||
runs-on: ubuntu-latest
|
||||
needs: build
|
||||
env:
|
||||
IS_PUBLIC_RELEASE: ${{ github.ref_type == 'tag' }}
|
||||
steps:
|
||||
- name: 🔫 Trigger Build Installer(s)
|
||||
uses: the-actions-org/workflow-dispatch@v4.0.0
|
||||
with:
|
||||
workflow: Build Installers
|
||||
repo: specklesystems/connector-installers
|
||||
token: ${{ secrets.CONNECTORS_GH_TOKEN }}
|
||||
inputs: '{
|
||||
"run_id": "${{ github.run_id }}",
|
||||
"semver": "${{ needs.build.outputs.semver }}",
|
||||
"file_version": "${{ needs.build.outputs.fileVersion }}",
|
||||
"repo": "${{ github.repository }}",
|
||||
"is_public_release": ${{ env.IS_PUBLIC_RELEASE }}
|
||||
}'
|
||||
ref: main
|
||||
wait-for-completion: true
|
||||
wait-for-completion-interval: 10s
|
||||
wait-for-completion-timeout: 10m
|
||||
display-workflow-run-url: true
|
||||
display-workflow-run-url-interval: 10s
|
||||
|
||||
- uses: geekyeggo/delete-artifact@v5
|
||||
with:
|
||||
name: output-*
|
||||
+2
-1
@@ -13,4 +13,5 @@ Installers/
|
||||
modules/
|
||||
.tool-versions
|
||||
requirements.txt
|
||||
SEMVER
|
||||
SEMVER
|
||||
dui3/
|
||||
|
||||
@@ -0,0 +1,21 @@
|
||||
repos:
|
||||
- repo: local
|
||||
hooks:
|
||||
# Run the linter.
|
||||
- id: ruff
|
||||
name: ruff lint
|
||||
entry: uv run ruff check --force-exclude
|
||||
language: system
|
||||
types_or: [python, pyi]
|
||||
# Run the formatter.
|
||||
- id: ruff-format
|
||||
name: ruff format
|
||||
entry: uv run ruff format --force-exclude
|
||||
language: system
|
||||
types_or: [python, pyi]
|
||||
|
||||
- repo: https://github.com/pre-commit/pre-commit-hooks
|
||||
rev: v5.0.0
|
||||
hooks:
|
||||
- id: end-of-file-fixer
|
||||
- id: trailing-whitespace
|
||||
@@ -7,7 +7,7 @@
|
||||
</h3>
|
||||
<p align="center"><b>Speckle</b> is the data infrastructure for the AEC industry.</p><br/>
|
||||
|
||||
<p align="center"><a href="https://twitter.com/SpeckleSystems"><img src="https://img.shields.io/twitter/follow/SpeckleSystems?style=social" alt="Twitter Follow"></a> <a href="https://speckle.community"><img src="https://img.shields.io/discourse/users?server=https%3A%2F%2Fspeckle.community&style=flat-square&logo=discourse&logoColor=white" alt="Community forum users"></a> <a href="https://speckle.systems"><img src="https://img.shields.io/badge/https://-speckle.systems-royalblue?style=flat-square" alt="website"></a> <a href="https://speckle.guide/dev/"><img src="https://img.shields.io/badge/docs-speckle.guide-orange?style=flat-square&logo=read-the-docs&logoColor=white" alt="docs"></a></p>
|
||||
<p align="center"><a href="https://twitter.com/SpeckleSystems"><img src="https://img.shields.io/twitter/follow/SpeckleSystems?style=social" alt="Twitter Follow"></a> <a href="https://speckle.community"><img src="https://img.shields.io/discourse/users?server=https%3A%2F%2Fspeckle.community&style=flat-square&logo=discourse&logoColor=white" alt="Community forum users"></a> <a href="https://speckle.systems"><img src="https://img.shields.io/badge/https://-speckle.systems-royalblue?style=flat-square" alt="website"></a> <a href="https://docs.speckle.systems/dev/"><img src="https://img.shields.io/badge/docs-docs.speckle.systems-orange?style=flat-square&logo=read-the-docs&logoColor=white" alt="docs"></a></p>
|
||||
<p align="center"><a href="https://github.com/specklesystems/speckle-blender/"><img src="https://circleci.com/gh/specklesystems/speckle-blender.svg?style=svg&circle-token=76eabd350ea243575cbb258b746ed3f471f7ac29" alt="Speckle-Next"></a> </p>
|
||||
|
||||
# About Speckle
|
||||
@@ -25,58 +25,90 @@ What is Speckle? Check our ](https://speckle.xyz) ⇒ creating an account at our public server
|
||||
- [](https://marketplace.digitalocean.com/apps/speckle-server?refcode=947a2b5d7dc1) ⇒ deploying an instance in 1 click
|
||||
- [](https://app.speckle.systems) ⇒ creating an account at our public server
|
||||
|
||||
### Resources
|
||||
|
||||
- [](https://speckle.community) for help, feature requests or just to hang with other speckle enthusiasts, check out our community forum!
|
||||
- [](https://speckle.systems) our tutorials portal is full of resources to get you started using Speckle
|
||||
- [](https://speckle.guide/user/blender.html) reference on almost any end-user and developer functionality
|
||||
- [](https://docs.speckle.systems/connectors/blender) reference on almost any end-user and developer functionality
|
||||
|
||||
|
||||
# Repo structure
|
||||
# Blender Connector
|
||||
|
||||
The Speckle UI can be found in the 3d viewport toolbar (N), under the Speckle tab.
|
||||
|
||||
Head to the [**📚 documentation**](https://speckle.guide/user/blender.html) for more information.
|
||||
|
||||
## Disclaimer
|
||||
This code is WIP and as such should be used with extreme caution on non-sensitive projects.
|
||||
|
||||
## Installation
|
||||
|
||||
1. Place `bpy_speckle` folder in your `addons` folder. On Windows this is typically `%APPDATA%/Blender Foundation/Blender/2.80/scripts/addons`.
|
||||
2. Go to `Edit->Preferences` (Ctrl + Alt + U)
|
||||
3. Go to the `Add-ons` tab
|
||||
4. Find and enable `SpeckleBlender 2.0` in the `Scene` category. <!-- **If enabling for the first time, expect the UI to freeze for bit while it silently installs all the dependencies.** -->
|
||||
5. The Speckle UI can be found in the 3d viewport toolbar (N), under the `Speckle` tab.
|
||||
We officially support Blender 4.2 and newer, on Windows.
|
||||
|
||||
## Usage
|
||||
- Available user accounts are automatically detected and made available. To add user accounts use **Speckle Manager**.
|
||||
- Select the user from the dropdown list in the `Users` panel. This will populate the `Streams` list with available streams for the selected user.
|
||||
- Select a branch and commit from the dropdown menus.
|
||||
- Click on `Receive` to download the objects from the selected stream, branch, and commit. The stream objects will be loaded into a Blender Collection, named `<STREAM_NAME> [ <STREAM_BRANCH> @ <BRANCH_COMMIT> ]`. <!-- You can filter the stream by entering a query into the `Filter` field (i.e. `properties.weight>10` or `type="Mesh"`). -->
|
||||
- Click on `Open Stream in Web` to view the stream in your web browser.
|
||||
|
||||
## Caveats
|
||||
Once enabled in `Preferences -> Addons`,
|
||||
The Speckle connector UI can be found in the 3d viewport toolbar (N), under the `Speckle` tab.
|
||||
|
||||
- Mesh objects are supported. Breps are imported as meshes using their `displayValue` data.
|
||||
- Curves have limited support: `Polylines` are supported; `NurbsCurves` are supported, though they are not guaranteed to look the same; `Lines` are supported; `Arcs` are not supported, though they are very roughly approximated; `PolyCurves` are supported for linear / polyline segments and very approximate arc segments. These conversions are a point of focus for further development.
|
||||
- Available user accounts are automatically detected and made available.
|
||||
- Select the account from the dropdown list in the `Accounts` panel. This will populate the `Projects` list with available projects for the selected user account.
|
||||
- Select a model and version from the dropdown menus.
|
||||
- Click on `Load` to download and convert the objects from the selected model version. The objects will be linked into a Blender Collection.
|
||||
- Click on `Open Model in Web` to view the model in your web browser.
|
||||
|
||||
## Custom properties
|
||||
## Supported Elements
|
||||
|
||||
- **SpeckleBlender** will look for a `texture_coordinates` property and use that to create a UV layer for the imported object. These texture coordinates are a space-separated list of floats (`[u v u v u v etc...]`) that is encoded as a base64 blob. This is subject to change as **SpeckleBlender** develops.
|
||||
- If a `renderMaterial` property is found, **SpeckleBlender** will create a material named using the sub-property `renderMaterial.name`. If a material with that name already exists in Blender, **SpeckleBlender** will just assign that existing material to the object. This allows geometry to be updated without having to re-assign and re-create materials.
|
||||
- Vertex colors are supported. The `colors` list from Speckle meshes is translated to a vertex color layer.
|
||||
- Speckle properties will be imported as custom properties on Blender objects. Nested dictionaries are expanded to individual properties by flattening their key hierarchy. I.e. `propA:{'propB': {'propC':10, 'propD':'foobar'}}` is flattened to `propA.propB.propC = 10` and `propA.propB.propD = "foobar"`.
|
||||
The Blender Connector is still a work in progress and, as such, data sent from the Blender connector is a highly lossy exchange. Our connectors are ever evolving to facilitate more and more Speckle usecases. We welcome feedback, requests, edge cases, and contributions!
|
||||
|
||||
## Dependency Installation and Compatibility with Other Blender Addons
|
||||
|
||||
Upon first launch of the addon, the Speckle connector installs its SpecklePy dependencies in `%appdata%/Speckle/connector_installations` on Windows.
|
||||
|
||||
This is done through our [`installer.py`](https://github.com/specklesystems/speckle-blender/blob/main/bpy_speckle/installer.py). Through pip, we install the correct version of each dependency for your blender python version, host OS, and system architecture.
|
||||
As such, an internet connection is required for first launch of the connector.
|
||||
|
||||
Other blender addons may require dependencies that conflict with specklepy. In these cases, one or both addons may fail to load.
|
||||
If you suspect you're seeing a conflict, Please uninstall other third party addons one at a time to identify which addon is conflicting.
|
||||
|
||||
If you find an addon that conflicts, please try using a different version of that addon (newer or older).
|
||||
|
||||
If you can't find a version of an addon that works, please let us know on [our forums](https://speckle.community/) the name of the addon, the versions you've tried, the version of the Speckle connector you've tried, and your OS (win/mac/linux).
|
||||
|
||||
## Local Development
|
||||
|
||||
Pre-resquisits:
|
||||
|
||||
- [uv](https://docs.astral.sh/uv/getting-started/installation/)
|
||||
- A supported [blender](https://www.blender.org/download/) version
|
||||
- [Blender Development](https://marketplace.visualstudio.com/items?itemName=JacquesLucke.blender-development) extension for VS Code (recommended)
|
||||
|
||||
First time setup (or anytime you change pyproject.toml)
|
||||
Run the following commands
|
||||
|
||||
```sh
|
||||
uv sync
|
||||
./export_dependencies.sh
|
||||
```
|
||||
|
||||
🪟 To activate the environment in a terminal (Windows):
|
||||
|
||||
```powershell
|
||||
.venv\Scripts\activate
|
||||
```
|
||||
|
||||
🐧 To activate the environment in a terminal (Linux / macOS):
|
||||
|
||||
```sh
|
||||
source .venv/bin/activate.fish
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
To run the blender plugin, run the `>Blender: Start`
|
||||
from VS code (`ctrl + shift + p`)
|
||||
<img width="1469" height="379" alt="image" src="https://github.com/user-attachments/assets/9dc174a0-07fc-47c7-85d1-bd5a04d8f8c7" />
|
||||
|
||||
## Contributing
|
||||
|
||||
@@ -89,6 +121,3 @@ The Speckle Community hangs out on [the forum](https://discourse.speckle.works),
|
||||
## License
|
||||
|
||||
Unless otherwise described, the code in this repository is licensed under the Apache-2.0 License. Please note that some modules, extensions or code herein might be otherwise licensed. This is indicated either in the root of the containing folder under a different license file, or in the respective file's header. If you have any questions, don't hesitate to get in touch with us via [email](mailto:hello@speckle.systems).
|
||||
|
||||
## Notes
|
||||
Thanks to [Tom Svilans](http://tomsvilans.com) ([Github](https://github.com/tsvilans)) for the original v1 contribution!
|
||||
|
||||
+222
-80
@@ -1,113 +1,255 @@
|
||||
# This program is free software; you can redistribute it and/or modify
|
||||
# it under the terms of the GNU General Public License as published by
|
||||
# the Free Software Foundation; either version 3 of the License, or
|
||||
# (at your option) any later version.
|
||||
#
|
||||
# This program is distributed in the hope that it will be useful, but
|
||||
# WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
# MERCHANTIBILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||
# General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU General Public License
|
||||
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
# ruff: noqa
|
||||
import bpy
|
||||
from bpy_speckle.installer import ensure_dependencies
|
||||
from bpy.types import WindowManager
|
||||
from .connector.ui import icons
|
||||
|
||||
# Ensure dependencies
|
||||
from .installer import ensure_dependencies
|
||||
|
||||
ensure_dependencies(f"Blender {bpy.app.version[0]}.{bpy.app.version[1]}")
|
||||
|
||||
from specklepy.logging import metrics
|
||||
|
||||
from bpy_speckle.ui import *
|
||||
from bpy_speckle.properties import *
|
||||
from bpy_speckle.operators import *
|
||||
from bpy_speckle.callbacks import *
|
||||
from bpy.app.handlers import persistent
|
||||
|
||||
bl_info = {
|
||||
"name": "SpeckleBlender 2.0",
|
||||
"author": "Speckle Systems",
|
||||
"version": (0, 2, 0),
|
||||
"blender": (2, 92, 0),
|
||||
"name": "Speckle Connector",
|
||||
"author": "Speckle",
|
||||
"version": (3, 999, 999),
|
||||
"blender": (4, 2, 0),
|
||||
"location": "3d viewport toolbar (N), under the Speckle tab.",
|
||||
"description": "The Speckle Connector using specklepy 2.0!",
|
||||
"warning": "This add-on is WIP and should be used with caution",
|
||||
"wiki_url": "https://github.com/specklesystems/speckle-blender",
|
||||
"description": "Publish models to and load models from other AEC apps.",
|
||||
"wiki_url": "https://speckle.systems/connectors/blender",
|
||||
"category": "Scene",
|
||||
}
|
||||
|
||||
|
||||
# UI
|
||||
from .connector.ui.main_panel import SPECKLE_PT_main_panel
|
||||
from .connector.ui.update_panel import SPECKLE_PT_update_panel
|
||||
from .connector.ui.model_cards_panel import SPECKLE_PT_model_cards_panel
|
||||
from .connector.utils.account_manager import speckle_workspace
|
||||
from .connector.ui.project_selection_dialog import (
|
||||
SPECKLE_OT_project_selection_dialog,
|
||||
SPECKLE_UL_projects_list,
|
||||
)
|
||||
from .connector.ui.model_selection_dialog import (
|
||||
SPECKLE_OT_model_selection_dialog,
|
||||
SPECKLE_UL_models_list,
|
||||
)
|
||||
from .connector.ui.version_selection_dialog import (
|
||||
SPECKLE_OT_version_selection_dialog,
|
||||
SPECKLE_UL_versions_list,
|
||||
)
|
||||
from .connector.ui.selection_filter_dialog import SPECKLE_OT_selection_filter_dialog
|
||||
from .connector.utils.property_groups import (
|
||||
speckle_project,
|
||||
speckle_model,
|
||||
speckle_version,
|
||||
speckle_object,
|
||||
speckle_collection,
|
||||
speckle_model_card,
|
||||
)
|
||||
|
||||
"""
|
||||
Import SpeckleBlender classes
|
||||
"""
|
||||
# Operators
|
||||
from .connector.blender_operators.publish_button import SPECKLE_OT_publish
|
||||
from .connector.blender_operators.load_button import SPECKLE_OT_load
|
||||
from .connector.blender_operators.model_card_settings import (
|
||||
SPECKLE_OT_model_card_settings,
|
||||
SPECKLE_OT_view_in_browser,
|
||||
SPECKLE_OT_view_model_versions,
|
||||
SPECKLE_OT_delete_model_card,
|
||||
)
|
||||
from .connector.blender_operators.select_objects import SPECKLE_OT_select_objects
|
||||
from .connector.blender_operators.add_account_button import SPECKLE_OT_add_account
|
||||
from .connector.blender_operators.add_account_button import (
|
||||
SPECKLE_OT_show_auth_error,
|
||||
SPECKLE_OT_dismiss_popup,
|
||||
)
|
||||
from .connector.blender_operators.model_card_load_button import (
|
||||
SPECKLE_OT_load_model_card,
|
||||
)
|
||||
from .connector.blender_operators.model_card_publish_button import (
|
||||
SPECKLE_OT_publish_model_card,
|
||||
)
|
||||
from .connector.blender_operators.add_project_by_url import (
|
||||
SPECKLE_OT_add_project_by_url,
|
||||
)
|
||||
|
||||
"""
|
||||
Add load handler to initialize Speckle when
|
||||
loading a Blender file
|
||||
"""
|
||||
from .connector.blender_operators.create_project import SPECKLE_OT_create_project
|
||||
from .connector.blender_operators.create_model import SPECKLE_OT_create_model
|
||||
from .connector.blender_operators.version_check import SPECKLE_OT_version_check
|
||||
from .connector.blender_operators.update_button import SPECKLE_OT_update_button
|
||||
from .connector.utils.account_manager import (
|
||||
speckle_account,
|
||||
get_default_account_id,
|
||||
_client_cache,
|
||||
)
|
||||
|
||||
@persistent
|
||||
def load_handler(dummy):
|
||||
pass
|
||||
# Calling users_load is an expensive operation, one that force users to wait a good 10s every time blender loads.
|
||||
# Until we can do this non-blocking, we will make the user hit the refresh button each time.
|
||||
#bpy.ops.speckle.users_load()
|
||||
|
||||
# Instead, we shall just reset the user selection to an uninitiailised state
|
||||
bpy.ops.speckle.users_reset()
|
||||
|
||||
"""
|
||||
Permanent handle on callbacks
|
||||
"""
|
||||
|
||||
callbacks = {}
|
||||
|
||||
"""
|
||||
Add Speckle classes for registering
|
||||
"""
|
||||
|
||||
speckle_classes = []
|
||||
speckle_classes.extend(operator_classes)
|
||||
speckle_classes.extend(property_classes)
|
||||
speckle_classes.extend(ui_classes)
|
||||
# States
|
||||
from .connector.states.speckle_state import (
|
||||
register as register_speckle_state,
|
||||
unregister as unregister_speckle_state,
|
||||
)
|
||||
|
||||
|
||||
def register():
|
||||
from bpy.utils import register_class
|
||||
from .connector.ui.workspace_selection_dialog import (
|
||||
SPECKLE_OT_workspace_selection_dialog,
|
||||
SPECKLE_UL_workspaces_list,
|
||||
)
|
||||
|
||||
for cls in speckle_classes:
|
||||
register_class(cls)
|
||||
# Utils
|
||||
from .connector.ui.account_selection_dialog import (
|
||||
SPECKLE_OT_account_selection_dialog,
|
||||
SPECKLE_UL_accounts_list,
|
||||
)
|
||||
|
||||
metrics.set_host_app("blender", f"blender {bpy.app.version_string}")
|
||||
|
||||
"""
|
||||
Register all new properties
|
||||
"""
|
||||
def delayed_version_check():
|
||||
"""Timer function to check for updates after addon startup"""
|
||||
try:
|
||||
bpy.ops.speckle.version_check()
|
||||
except Exception as e:
|
||||
print(f"[Speckle] Failed to check for updates: {e}")
|
||||
|
||||
bpy.types.Scene.speckle = bpy.props.PointerProperty(type=SpeckleSceneSettings)
|
||||
bpy.types.Collection.speckle = bpy.props.PointerProperty(
|
||||
type=SpeckleCollectionSettings
|
||||
|
||||
def invoke_window_manager_properties():
|
||||
# Accounts
|
||||
WindowManager.speckle_accounts = bpy.props.CollectionProperty(type=speckle_account)
|
||||
WindowManager.selected_account_id = bpy.props.StringProperty()
|
||||
# Workspaces
|
||||
WindowManager.speckle_workspaces = bpy.props.CollectionProperty(
|
||||
type=speckle_workspace
|
||||
)
|
||||
bpy.types.Object.speckle = bpy.props.PointerProperty(type=SpeckleObjectSettings)
|
||||
WindowManager.selected_workspace = bpy.props.PointerProperty(type=speckle_workspace)
|
||||
WindowManager.can_create_project_in_workspace = bpy.props.BoolProperty()
|
||||
# Projects
|
||||
WindowManager.speckle_projects = bpy.props.CollectionProperty(type=speckle_project)
|
||||
WindowManager.selected_project_id = bpy.props.StringProperty()
|
||||
WindowManager.selected_project_name = bpy.props.StringProperty()
|
||||
# Models
|
||||
WindowManager.speckle_models = bpy.props.CollectionProperty(type=speckle_model)
|
||||
WindowManager.selected_model_id = bpy.props.StringProperty()
|
||||
WindowManager.selected_model_name = bpy.props.StringProperty()
|
||||
# Versions
|
||||
WindowManager.speckle_versions = bpy.props.CollectionProperty(type=speckle_version)
|
||||
WindowManager.selected_version_id = bpy.props.StringProperty()
|
||||
WindowManager.selected_version_load_option = bpy.props.StringProperty()
|
||||
# Send / Publish buttons
|
||||
WindowManager.ui_mode = bpy.props.EnumProperty( # type: ignore
|
||||
name="UI Mode",
|
||||
description="Publish or Load a model",
|
||||
items=[
|
||||
("PUBLISH", "Publish", "Publish a model to Speckle", "EXPORT", 0),
|
||||
("LOAD", "Load", "Load a model from Speckle", "IMPORT", 1),
|
||||
],
|
||||
default="PUBLISH",
|
||||
)
|
||||
# Objects
|
||||
WindowManager.speckle_objects = bpy.props.CollectionProperty(type=speckle_object)
|
||||
# Update checking
|
||||
WindowManager.update_available = bpy.props.BoolProperty(default=False)
|
||||
WindowManager.latest_version = bpy.props.StringProperty(default="")
|
||||
WindowManager.update_url = bpy.props.StringProperty(default="")
|
||||
|
||||
"""
|
||||
Add callbacks
|
||||
"""
|
||||
|
||||
# Callback for displaying the current user account on top of the 3d view
|
||||
# callbacks['view3d_status'] = ((
|
||||
# bpy.types.SpaceView3D.draw_handler_remove, # Function pointer for removal
|
||||
# bpy.types.SpaceView3D.draw_handler_add(draw_speckle_info, (None, None), 'WINDOW', 'POST_PIXEL'), # Add handler
|
||||
# 'WINDOW' # Callback space for removal
|
||||
# ))
|
||||
# Classes to load
|
||||
classes = (
|
||||
SPECKLE_PT_update_panel,
|
||||
SPECKLE_PT_main_panel,
|
||||
SPECKLE_PT_model_cards_panel,
|
||||
SPECKLE_OT_publish,
|
||||
SPECKLE_OT_load,
|
||||
SPECKLE_OT_project_selection_dialog,
|
||||
speckle_project,
|
||||
SPECKLE_UL_projects_list,
|
||||
speckle_workspace,
|
||||
SPECKLE_OT_model_selection_dialog,
|
||||
speckle_model,
|
||||
SPECKLE_UL_models_list,
|
||||
SPECKLE_OT_version_selection_dialog,
|
||||
speckle_version,
|
||||
SPECKLE_UL_versions_list,
|
||||
SPECKLE_OT_selection_filter_dialog,
|
||||
speckle_object,
|
||||
speckle_collection,
|
||||
speckle_model_card,
|
||||
SPECKLE_OT_model_card_settings,
|
||||
SPECKLE_OT_view_in_browser,
|
||||
SPECKLE_OT_view_model_versions,
|
||||
SPECKLE_OT_delete_model_card,
|
||||
SPECKLE_OT_select_objects,
|
||||
SPECKLE_OT_add_account,
|
||||
SPECKLE_OT_show_auth_error,
|
||||
SPECKLE_OT_dismiss_popup,
|
||||
SPECKLE_OT_load_model_card,
|
||||
SPECKLE_OT_publish_model_card,
|
||||
SPECKLE_OT_add_project_by_url,
|
||||
SPECKLE_OT_create_project,
|
||||
SPECKLE_OT_create_model,
|
||||
SPECKLE_OT_version_check,
|
||||
SPECKLE_OT_update_button,
|
||||
speckle_account,
|
||||
SPECKLE_UL_workspaces_list,
|
||||
SPECKLE_OT_workspace_selection_dialog,
|
||||
SPECKLE_OT_account_selection_dialog,
|
||||
SPECKLE_UL_accounts_list,
|
||||
)
|
||||
|
||||
bpy.app.handlers.load_post.append(load_handler)
|
||||
|
||||
# Register and Unregister
|
||||
def register():
|
||||
icons.load_icons()
|
||||
|
||||
for cls in classes:
|
||||
bpy.utils.register_class(cls)
|
||||
register_speckle_state() # Register SpeckleState
|
||||
|
||||
invoke_window_manager_properties()
|
||||
|
||||
# Pre-warm client cache for default account
|
||||
try:
|
||||
default_account_id = get_default_account_id()
|
||||
if default_account_id:
|
||||
print(
|
||||
f"[Speckle] Pre-warming client for default account: {default_account_id}"
|
||||
)
|
||||
_client_cache.get_client(default_account_id)
|
||||
print(
|
||||
f"[Speckle] Client pre-warming complete for account: {default_account_id}"
|
||||
)
|
||||
except Exception as e:
|
||||
print(f"[Speckle] Failed to pre-warm client: {e}")
|
||||
|
||||
# Use a timer to delay the version check
|
||||
bpy.app.timers.register(delayed_version_check, first_interval=2.0)
|
||||
|
||||
|
||||
def unregister():
|
||||
# Clear any pending timers to prevent duplicate calls
|
||||
if bpy.app.timers.is_registered(delayed_version_check):
|
||||
bpy.app.timers.unregister(delayed_version_check)
|
||||
|
||||
bpy.app.handlers.load_post.remove(load_handler)
|
||||
# Clean up authentication server
|
||||
from .connector.blender_operators.add_account_button import cleanup_auth_server
|
||||
|
||||
"""
|
||||
Remove callbacks
|
||||
"""
|
||||
cleanup_auth_server()
|
||||
|
||||
for cb in callbacks.values():
|
||||
cb[0](cb[1], cb[2])
|
||||
|
||||
from bpy.utils import unregister_class
|
||||
|
||||
for cls in reversed(speckle_classes):
|
||||
unregister_class(cls)
|
||||
icons.unload_icons()
|
||||
unregister_speckle_state() # Unregister SpeckleState
|
||||
_client_cache.clear()
|
||||
for cls in classes:
|
||||
bpy.utils.unregister_class(cls)
|
||||
|
||||
|
||||
# Run the register function when the script is executed
|
||||
if __name__ == "__main__":
|
||||
register()
|
||||
|
||||
@@ -0,0 +1,66 @@
|
||||
schema_version = "1.0.0"
|
||||
|
||||
# Example of manifest file for a Blender extension
|
||||
# Change the values according to your extension
|
||||
id = "speckle_blender_addon"
|
||||
version = "3.0.0"
|
||||
name = "Speckle Connector"
|
||||
tagline = "Load models from other AEC apps into Blender with Speckle."
|
||||
maintainer = "Speckle"
|
||||
# Supported types: "add-on", "theme"
|
||||
type = "add-on"
|
||||
|
||||
# Optional link to documentation, support, source files, etc
|
||||
website = "https://speckle.systems/connectors/blender"
|
||||
|
||||
# Optional list defined by Blender and server, see:
|
||||
# https://docs.blender.org/manual/en/dev/advanced/extensions/tags.html
|
||||
tags = ["Scene"]
|
||||
|
||||
blender_version_min = "4.2.0"
|
||||
# # Optional: Blender version that the extension does not support, earlier versions are supported.
|
||||
# # This can be omitted and defined later on the extensions platform if an issue is found.
|
||||
# blender_version_max = "5.1.0"
|
||||
|
||||
# License conforming to https://spdx.org/licenses/ (use "SPDX: prefix)
|
||||
# https://docs.blender.org/manual/en/dev/advanced/extensions/licenses.html
|
||||
license = ["SPDX:Apache-2.0"]
|
||||
# Optional: required by some licenses.
|
||||
copyright = ["2022-2025 AEC SYSTEMS LTD"]
|
||||
|
||||
# Optional list of supported platforms. If omitted, the extension will be available in all operating systems.
|
||||
# platforms = ["windows-x64", "macos-arm64", "linux-x64"]
|
||||
# Other supported platforms: "windows-arm64", "macos-x64"
|
||||
|
||||
# Optional: bundle 3rd party Python modules.
|
||||
# https://docs.blender.org/manual/en/dev/advanced/extensions/python_wheels.html
|
||||
# wheels = [
|
||||
# "./wheels/hexdump-3.3-py3-none-any.whl",
|
||||
# "./wheels/jsmin-3.0.1-py3-none-any.whl",
|
||||
# ]
|
||||
|
||||
# Optional: add-ons can list which resources they will require:
|
||||
# * files (for access of any filesystem operations)
|
||||
# * network (for internet access)
|
||||
# * clipboard (to read and/or write the system clipboard)
|
||||
# * camera (to capture photos and videos)
|
||||
# * microphone (to capture audio)
|
||||
# permissions = ["files", "network", "clipboard"]
|
||||
|
||||
|
||||
# If using network, remember to also check `bpy.app.online_access`
|
||||
# https://docs.blender.org/manual/en/dev/advanced/extensions/addons.html#internet-access
|
||||
#
|
||||
# For each permission it is important to also specify the reason why it is required.
|
||||
# Keep this a single short sentence without a period (.) at the end.
|
||||
# For longer explanations use the documentation or detail page.
|
||||
#
|
||||
[permissions]
|
||||
network = "Speckle Server comms, and PyPI for dependency management"
|
||||
files = "Data caching, Account Management, Python dependency management"
|
||||
clipboard = "Copy and paste URLs and Names (UI)"
|
||||
|
||||
# Optional: build settings.
|
||||
# https://docs.blender.org/manual/en/dev/advanced/extensions/command_line_arguments.html#command-line-args-extension-build
|
||||
[build]
|
||||
paths_exclude_pattern = ["__pycache__/", "/.vscode", "*.code-workspace"]
|
||||
@@ -1,2 +0,0 @@
|
||||
from .on_mesh_edit import scb_on_mesh_edit
|
||||
from .draw_speckle_info import draw_speckle_info
|
||||
@@ -1,23 +0,0 @@
|
||||
"""
|
||||
Drawing callback to display active Speckle user
|
||||
"""
|
||||
|
||||
import blf
|
||||
import bpy
|
||||
|
||||
|
||||
def draw_speckle_info(self, context):
|
||||
"""
|
||||
Draw active user info on the 3d viewport
|
||||
"""
|
||||
scn = bpy.context.scene
|
||||
if len(scn.speckle.users) > 0:
|
||||
user = scn.speckle.users[int(scn.speckle.active_user)]
|
||||
dpi = bpy.context.preferences.system.dpi
|
||||
|
||||
blf.position(0, 100, 50, 0)
|
||||
blf.size(0, 20, dpi)
|
||||
blf.draw(0, "Active Speckle user: {} ({})".format(user.name, user.email))
|
||||
blf.position(0, 100, 20, 0)
|
||||
blf.size(0, 16, dpi)
|
||||
blf.draw(0, "Server: {}".format(user.server))
|
||||
@@ -1,13 +0,0 @@
|
||||
import bpy
|
||||
from bpy.app.handlers import persistent
|
||||
|
||||
|
||||
@persistent
|
||||
def scb_on_mesh_edit(context):
|
||||
"""
|
||||
DEPRECATED
|
||||
Do something whenever a mesh is updated
|
||||
"""
|
||||
edit_obj = bpy.context.edit_object
|
||||
if edit_obj is not None and edit_obj.is_updated_data is True:
|
||||
print("Mesh edited: {}".format(edit_obj))
|
||||
@@ -1,7 +0,0 @@
|
||||
"""
|
||||
Permanent handle on all user clients
|
||||
"""
|
||||
from specklepy.api.client import SpeckleClient
|
||||
|
||||
|
||||
speckle_clients: list[SpeckleClient] = []
|
||||
@@ -0,0 +1,9 @@
|
||||
from ..blender_operators.load_button import SPECKLE_OT_load # noqa: F401
|
||||
from .model_card_load_button import SPECKLE_OT_load_model_card # noqa: F401
|
||||
from ..blender_operators.publish_button import SPECKLE_OT_publish # noqa: F401
|
||||
from ..blender_operators.model_card_settings import (
|
||||
SPECKLE_OT_model_card_settings, # noqa: F401
|
||||
SPECKLE_OT_view_in_browser, # noqa: F401
|
||||
SPECKLE_OT_view_model_versions, # noqa: F401
|
||||
SPECKLE_OT_delete_model_card, # noqa: F401
|
||||
)
|
||||
@@ -0,0 +1,259 @@
|
||||
import bpy
|
||||
import textwrap
|
||||
from bpy.types import Event, Context
|
||||
from typing import Optional
|
||||
from ..utils.authentication import (
|
||||
AuthenticationServer,
|
||||
SPECKLE_AUTH_PORT,
|
||||
)
|
||||
|
||||
|
||||
# Global auth server instance
|
||||
_auth_server = None
|
||||
|
||||
|
||||
class SPECKLE_OT_add_account(bpy.types.Operator):
|
||||
"""Operator for adding a new Speckle account."""
|
||||
|
||||
bl_idname = "speckle.add_account"
|
||||
bl_label = "Add New Account"
|
||||
bl_description = "Add a new account"
|
||||
|
||||
server_url: bpy.props.StringProperty( # type: ignore
|
||||
name="Server URL",
|
||||
description="Speckle server URL to connect to",
|
||||
default="https://app.speckle.systems",
|
||||
)
|
||||
|
||||
_timer = None
|
||||
_timeout_counter = 0
|
||||
_max_timeout = 300 # 5 minutes in seconds (300 checks at ~1 sec intervals)
|
||||
|
||||
def invoke(self, context: Context, event: Event) -> set[str]:
|
||||
return context.window_manager.invoke_props_dialog(self)
|
||||
|
||||
def draw(self, context: Context):
|
||||
layout = self.layout
|
||||
# Server URL textbox
|
||||
layout.prop(self, "server_url", text="Server URL")
|
||||
|
||||
def execute(self, context: Context) -> set[str]:
|
||||
print(f"[Add Account] Starting authentication for server: {self.server_url}")
|
||||
cleanup_auth_server()
|
||||
|
||||
# Try to start own auth server first - it will fail gracefully if port is in use
|
||||
global _auth_server
|
||||
_auth_server = AuthenticationServer(port=SPECKLE_AUTH_PORT)
|
||||
|
||||
if _auth_server.start():
|
||||
return self._initiate_own_server_flow(context)
|
||||
|
||||
# Server failed to start - port is in use
|
||||
_auth_server = None
|
||||
print(f"[Add Account] Port {SPECKLE_AUTH_PORT} is already in use")
|
||||
self.report(
|
||||
{"ERROR"},
|
||||
f"Port {SPECKLE_AUTH_PORT} is already in use. Please close any application using it and try again.",
|
||||
)
|
||||
return {"CANCELLED"}
|
||||
|
||||
def _initiate_own_server_flow(self, context: Context) -> set[str]:
|
||||
"""Start auth flow with our own server."""
|
||||
try:
|
||||
_auth_server.open_auth_url(self.server_url)
|
||||
self._start_modal_timer(context)
|
||||
return {"RUNNING_MODAL"}
|
||||
except Exception as e:
|
||||
print(f"[Add Account] Failed to open browser: {e}")
|
||||
self.report({"ERROR"}, f"Failed to open browser: {e}")
|
||||
cleanup_auth_server()
|
||||
return {"CANCELLED"}
|
||||
|
||||
def _start_modal_timer(self, context: Context):
|
||||
"""Start modal timer for auth polling."""
|
||||
self._timeout_counter = 0
|
||||
wm = context.window_manager
|
||||
self._timer = wm.event_timer_add(1.0, window=context.window)
|
||||
wm.modal_handler_add(self)
|
||||
|
||||
def modal(self, context: Context, event: Event) -> set[str]:
|
||||
global _auth_server
|
||||
|
||||
if event.type != "TIMER":
|
||||
return {"PASS_THROUGH"}
|
||||
|
||||
# Check for timeout
|
||||
self._timeout_counter += 1
|
||||
if self._timeout_counter >= self._max_timeout:
|
||||
print("[Add Account] Authentication timed out after 5 minutes")
|
||||
self._cleanup(context)
|
||||
self.report(
|
||||
{"WARNING"},
|
||||
"Authentication timed out after 5 minutes. Please try again.",
|
||||
)
|
||||
return {"CANCELLED"}
|
||||
|
||||
# Check for no active auth server
|
||||
if not _auth_server:
|
||||
print("[Add Account] No active auth server, cancelling")
|
||||
self._cleanup(context)
|
||||
return {"CANCELLED"}
|
||||
|
||||
# Check auth server completion
|
||||
if _auth_server.is_complete():
|
||||
return self._finish_auth(
|
||||
context,
|
||||
_auth_server.is_successful(),
|
||||
_auth_server.get_error_message(),
|
||||
"Auth server",
|
||||
)
|
||||
|
||||
# Still waiting
|
||||
return {"RUNNING_MODAL"}
|
||||
|
||||
def _finish_auth(
|
||||
self,
|
||||
context: Context,
|
||||
is_successful: bool,
|
||||
error_msg: Optional[str],
|
||||
auth_type: str,
|
||||
) -> set[str]:
|
||||
"""Complete authentication and cleanup."""
|
||||
print(
|
||||
f"[Add Account] {auth_type} authentication complete. Success: {is_successful}"
|
||||
)
|
||||
self._cleanup(context)
|
||||
return self._handle_auth_complete(context, is_successful, error_msg)
|
||||
|
||||
def _handle_auth_complete(
|
||||
self, context: Context, is_successful: bool, error_msg: Optional[str]
|
||||
) -> set[str]:
|
||||
"""Handle authentication completion and update UI state."""
|
||||
if is_successful:
|
||||
print("[Add Account] Account added successfully - refreshing UI")
|
||||
|
||||
# Import account management functions
|
||||
from ..utils.account_manager import get_account_enum_items, _client_cache
|
||||
from ..ui.account_selection_dialog import (
|
||||
update_workspaces_list,
|
||||
update_projects_list,
|
||||
)
|
||||
|
||||
# Get the newly added account (most recent one)
|
||||
accounts = get_account_enum_items()
|
||||
if accounts and accounts[0][0] != "NO_ACCOUNTS":
|
||||
new_account_id = accounts[-1][0] # Last account added
|
||||
|
||||
# Set as selected account
|
||||
context.window_manager.selected_account_id = new_account_id
|
||||
|
||||
# Clear client cache to force re-authentication
|
||||
_client_cache.clear()
|
||||
|
||||
# Refresh UI state
|
||||
try:
|
||||
update_workspaces_list(context)
|
||||
update_projects_list(context)
|
||||
except Exception as e:
|
||||
print(f"[Add Account] Error refreshing UI state: {e}")
|
||||
|
||||
self.report({"INFO"}, "Account added successfully and is now active!")
|
||||
else:
|
||||
self.report({"INFO"}, "Account added successfully!")
|
||||
|
||||
return {"FINISHED"}
|
||||
else:
|
||||
error_details = error_msg if error_msg else "Unknown error"
|
||||
print(f"[Add Account] Authentication failed: {error_details}")
|
||||
self.report({"ERROR"}, f"Authentication failed: {error_details}")
|
||||
|
||||
# Show persistent error popup with details
|
||||
# Store error in window manager for the popup operator
|
||||
context.window_manager["speckle_auth_error"] = error_details
|
||||
bpy.ops.speckle.show_auth_error("INVOKE_DEFAULT")
|
||||
|
||||
return {"CANCELLED"}
|
||||
|
||||
def _cleanup(self, context: Context):
|
||||
# Remove timer
|
||||
if self._timer is not None:
|
||||
context.window_manager.event_timer_remove(self._timer)
|
||||
self._timer = None
|
||||
|
||||
# Shutdown auth server/authenticator
|
||||
cleanup_auth_server()
|
||||
|
||||
|
||||
def cleanup_auth_server():
|
||||
"""Shutdown auth server on addon unload."""
|
||||
global _auth_server
|
||||
|
||||
if _auth_server is not None:
|
||||
try:
|
||||
_auth_server.shutdown()
|
||||
except Exception as e:
|
||||
print(f"[Add Account] Failed to cleanup auth server: {e}")
|
||||
print(f"[Add Account] Port {SPECKLE_AUTH_PORT} may still be occupied")
|
||||
_auth_server = None
|
||||
|
||||
|
||||
class SPECKLE_OT_show_auth_error(bpy.types.Operator):
|
||||
"""Show persistent error dialog for authentication failures."""
|
||||
|
||||
bl_idname = "speckle.show_auth_error"
|
||||
bl_label = "Authentication Error"
|
||||
bl_options = {"INTERNAL"}
|
||||
|
||||
def execute(self, context: Context) -> set[str]:
|
||||
# Clean up the temporary error message
|
||||
if "speckle_auth_error" in context.window_manager:
|
||||
del context.window_manager["speckle_auth_error"]
|
||||
return {"FINISHED"}
|
||||
|
||||
def invoke(self, context: Context, event: Event) -> set[str]:
|
||||
return context.window_manager.invoke_popup(self, width=450)
|
||||
|
||||
def draw(self, context: Context):
|
||||
layout = self.layout
|
||||
|
||||
# Error header
|
||||
box = layout.box()
|
||||
row = box.row()
|
||||
row.label(text="", icon="ERROR")
|
||||
row.label(text="Authentication Failed", icon="NONE")
|
||||
|
||||
layout.separator()
|
||||
|
||||
# Error details
|
||||
error_details = context.window_manager.get(
|
||||
"speckle_auth_error", "Unknown error"
|
||||
)
|
||||
col = layout.column(align=True)
|
||||
|
||||
# Wrap long error messages
|
||||
wrapper = textwrap.TextWrapper(width=60)
|
||||
for line in error_details.split("\n"):
|
||||
if line:
|
||||
for wrapped_line in wrapper.wrap(line):
|
||||
col.label(text=wrapped_line)
|
||||
else:
|
||||
col.label(text="")
|
||||
|
||||
layout.separator()
|
||||
|
||||
# Close button
|
||||
layout.operator("speckle.dismiss_popup", text="Close", icon="X")
|
||||
|
||||
|
||||
class SPECKLE_OT_dismiss_popup(bpy.types.Operator):
|
||||
"""Dismiss popup dialog."""
|
||||
|
||||
bl_idname = "speckle.dismiss_popup"
|
||||
bl_label = "Dismiss"
|
||||
bl_options = {"INTERNAL"}
|
||||
|
||||
def execute(self, context: Context) -> set[str]:
|
||||
# Clean up any temporary data
|
||||
if "speckle_auth_error" in context.window_manager:
|
||||
del context.window_manager["speckle_auth_error"]
|
||||
return {"FINISHED"}
|
||||
@@ -0,0 +1,75 @@
|
||||
import bpy
|
||||
from bpy.types import Context, Event, UILayout
|
||||
from ..utils.account_manager import (
|
||||
get_model_details_by_wrapper,
|
||||
get_project_from_url,
|
||||
can_load,
|
||||
)
|
||||
|
||||
|
||||
class SPECKLE_OT_add_project_by_url(bpy.types.Operator):
|
||||
"""
|
||||
operator for adding a Speckle project by URL
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.add_project_by_url"
|
||||
bl_label = "Add Project by URL"
|
||||
bl_description = "Add a project from a URL"
|
||||
|
||||
url: bpy.props.StringProperty( # type: ignore
|
||||
name="Project URL", description="Enter the Speckle project URL", default=""
|
||||
)
|
||||
|
||||
def execute(self, context: Context) -> set[str]:
|
||||
self.report({"INFO"}, f"Adding project from URL: {self.url}")
|
||||
|
||||
wm = context.window_manager
|
||||
|
||||
# Get project from URL
|
||||
wrapper, client, project, error_message = get_project_from_url(self.url)
|
||||
|
||||
if error_message:
|
||||
self.report({"ERROR"}, error_message)
|
||||
return {"CANCELLED"}
|
||||
|
||||
# Get model details from the wrapper
|
||||
(
|
||||
account_id,
|
||||
project_id,
|
||||
project_name,
|
||||
model_id,
|
||||
model_name,
|
||||
version_id,
|
||||
load_option,
|
||||
) = get_model_details_by_wrapper(wrapper)
|
||||
|
||||
# Check permissions
|
||||
can_load_permission, permission_error = can_load(client, project)
|
||||
if not can_load_permission:
|
||||
self.report({"ERROR"}, permission_error)
|
||||
return {"CANCELLED"}
|
||||
|
||||
# Update the window manager with the selected project/model/version
|
||||
wm.selected_account_id = account_id
|
||||
|
||||
if project_id:
|
||||
wm.selected_project_id = project_id
|
||||
wm.selected_project_name = project_name
|
||||
if model_id:
|
||||
wm.selected_model_id = model_id
|
||||
wm.selected_model_name = model_name
|
||||
if version_id:
|
||||
wm.selected_version_id = version_id
|
||||
wm.selected_version_id = version_id
|
||||
wm.selected_version_load_option = load_option
|
||||
|
||||
context.window.screen = context.window.screen
|
||||
context.area.tag_redraw()
|
||||
return {"FINISHED"}
|
||||
|
||||
def invoke(self, context: Context, event: Event) -> set[str]:
|
||||
return context.window_manager.invoke_props_dialog(self)
|
||||
|
||||
def draw(self, context: Context) -> None:
|
||||
layout: UILayout = self.layout
|
||||
layout.prop(self, "url", text="")
|
||||
@@ -0,0 +1,85 @@
|
||||
import bpy
|
||||
from bpy.types import Context, Event, UILayout
|
||||
from specklepy.core.api.inputs import CreateModelInput
|
||||
from typing import Tuple
|
||||
|
||||
from ..utils.account_manager import _client_cache, can_create_model
|
||||
|
||||
|
||||
class SPECKLE_OT_create_model(bpy.types.Operator):
|
||||
bl_idname = "speckle.create_model"
|
||||
bl_label = "Create Model"
|
||||
bl_description = "Create a new Speckle model"
|
||||
|
||||
_can_create: bool = True
|
||||
|
||||
model_name: bpy.props.StringProperty(name="Model Name") # type: ignore
|
||||
|
||||
@classmethod
|
||||
def description(cls, context: Context, properties) -> str:
|
||||
if not cls._can_create:
|
||||
return "Workspace limits have been reached"
|
||||
return "Create a new Speckle model"
|
||||
|
||||
def execute(self, context: Context) -> set[str]:
|
||||
wm = context.window_manager
|
||||
|
||||
authorized, auth_message = can_create_model(
|
||||
wm.selected_account_id, wm.selected_project_id
|
||||
)
|
||||
if not authorized:
|
||||
self.report({"ERROR"}, auth_message)
|
||||
return {"CANCELLED"}
|
||||
|
||||
if not self.model_name.strip():
|
||||
self.report({"ERROR"}, "Model name cannot be empty")
|
||||
return {"CANCELLED"}
|
||||
|
||||
try:
|
||||
model_id, model_name = create_model(
|
||||
wm.selected_account_id, wm.selected_project_id, self.model_name
|
||||
)
|
||||
wm.selected_model_id = model_id
|
||||
wm.selected_model_name = model_name
|
||||
self.report({"INFO"}, f"Created model: {model_name} -> ID: {model_id}")
|
||||
# Force redraw
|
||||
context.window.screen = context.window.screen
|
||||
context.area.tag_redraw()
|
||||
except Exception as e:
|
||||
self.report({"ERROR"}, f"Failed to create model: {str(e)}")
|
||||
return {"CANCELLED"}
|
||||
return {"FINISHED"}
|
||||
|
||||
def invoke(self, context: Context, event: Event) -> set[str]:
|
||||
return context.window_manager.invoke_props_dialog(self)
|
||||
|
||||
def draw(self, context: Context) -> None:
|
||||
layout: UILayout = self.layout
|
||||
layout.prop(self, "model_name")
|
||||
|
||||
|
||||
def register() -> None:
|
||||
bpy.utils.register_class(SPECKLE_OT_create_model)
|
||||
|
||||
|
||||
def unregister() -> None:
|
||||
bpy.utils.unregister_class(SPECKLE_OT_create_model)
|
||||
|
||||
|
||||
def create_model(account_id: str, project_id: str, model_name: str) -> Tuple[str, str]:
|
||||
try:
|
||||
# Get cached client
|
||||
client = _client_cache.get_client(account_id)
|
||||
if not client:
|
||||
raise ValueError(f"Could not get client for account: {account_id}")
|
||||
|
||||
model = client.model.create(
|
||||
input=CreateModelInput(
|
||||
name=model_name, description="", project_id=project_id
|
||||
)
|
||||
)
|
||||
return (model.id, model.name)
|
||||
except Exception as e:
|
||||
# Clear cache on error to prevent stale clients
|
||||
_client_cache.clear()
|
||||
raise e
|
||||
@@ -0,0 +1,75 @@
|
||||
import bpy
|
||||
from bpy.types import Context, Event, UILayout
|
||||
|
||||
from specklepy.core.api.inputs.project_inputs import WorkspaceProjectCreateInput
|
||||
from specklepy.core.api.enums import ProjectVisibility
|
||||
from typing import Tuple
|
||||
|
||||
from ..utils.account_manager import _client_cache
|
||||
|
||||
|
||||
class SPECKLE_OT_create_project(bpy.types.Operator):
|
||||
"""
|
||||
operator for adding a Speckle project by URL
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.create_project"
|
||||
bl_label = "Create Project"
|
||||
bl_description = "Create a new Speckle project"
|
||||
|
||||
project_name: bpy.props.StringProperty(name="Project Name") # type: ignore
|
||||
|
||||
def execute(self, context: Context) -> set[str]:
|
||||
wm = context.window_manager
|
||||
project_id, project_name = create_project(
|
||||
wm.selected_account_id,
|
||||
self.project_name,
|
||||
wm.selected_workspace.id,
|
||||
)
|
||||
wm.selected_project_id = project_id
|
||||
wm.selected_project_name = project_name
|
||||
self.report({"INFO"}, f"Created project: {project_name} -> ID: {project_id}")
|
||||
# Force redraw
|
||||
context.window.screen = context.window.screen
|
||||
context.area.tag_redraw()
|
||||
return {"FINISHED"}
|
||||
|
||||
def invoke(self, context: Context, event: Event) -> set[str]:
|
||||
return context.window_manager.invoke_props_dialog(self)
|
||||
|
||||
def draw(self, context: Context) -> None:
|
||||
layout: UILayout = self.layout
|
||||
layout.prop(self, "project_name")
|
||||
|
||||
|
||||
def register() -> None:
|
||||
bpy.utils.register_class(SPECKLE_OT_create_project)
|
||||
|
||||
|
||||
def unregister() -> None:
|
||||
bpy.utils.unregister_class(SPECKLE_OT_create_project)
|
||||
|
||||
|
||||
def create_project(
|
||||
account_id: str, project_name: str, workspace_id: str
|
||||
) -> Tuple[str, str]:
|
||||
try:
|
||||
# Get cached client
|
||||
client = _client_cache.get_client(account_id)
|
||||
if not client:
|
||||
raise Exception(f"Could not get client for account: {account_id}")
|
||||
project = client.project.create_in_workspace(
|
||||
input=WorkspaceProjectCreateInput(
|
||||
name=project_name,
|
||||
description="",
|
||||
visibility=ProjectVisibility("PUBLIC"),
|
||||
workspaceId=workspace_id,
|
||||
)
|
||||
)
|
||||
|
||||
return (project.id, project.name)
|
||||
except Exception as e:
|
||||
print(f"Failed to create project: {str(e)}")
|
||||
# Clear cache on error to prevent stale clients
|
||||
_client_cache.clear()
|
||||
raise
|
||||
@@ -0,0 +1,79 @@
|
||||
import bpy
|
||||
from typing import Set
|
||||
from bpy.types import Context, Event
|
||||
from ..operations.load_operation import load_operation
|
||||
from ..utils.account_manager import get_server_url_by_account_id
|
||||
from ..utils.model_card_utils import (
|
||||
update_model_card_objects,
|
||||
delete_model_card_objects,
|
||||
model_card_exists,
|
||||
)
|
||||
|
||||
|
||||
class SPECKLE_OT_load(bpy.types.Operator):
|
||||
bl_idname = "speckle.load"
|
||||
bl_label = "Load model"
|
||||
bl_description = "Load selection from Speckle"
|
||||
|
||||
instance_loading_mode: bpy.props.EnumProperty( # type: ignore
|
||||
name="Instance Loading",
|
||||
description="Choose how to load instances",
|
||||
items=[
|
||||
(
|
||||
"INSTANCE_PROXIES",
|
||||
"Collection Instances",
|
||||
"Load objects as collection instances",
|
||||
),
|
||||
(
|
||||
"LINKED_DUPLICATES",
|
||||
"Linked Duplicates",
|
||||
"Get objects as linked duplicates",
|
||||
),
|
||||
],
|
||||
default="INSTANCE_PROXIES",
|
||||
)
|
||||
|
||||
def draw(self, context: Context) -> None:
|
||||
layout = self.layout
|
||||
row = layout.row()
|
||||
row.label(text="Instance Loading:")
|
||||
row.prop(self, "instance_loading_mode", text="")
|
||||
|
||||
def invoke(self, context: Context, event: Event) -> Set[str]:
|
||||
return context.window_manager.invoke_props_dialog(self)
|
||||
|
||||
def execute(self, context: Context) -> Set[str]:
|
||||
wm = context.window_manager
|
||||
if model_card_exists(
|
||||
wm.selected_project_id, wm.selected_model_id, False, context
|
||||
):
|
||||
model_card = context.scene.speckle_state.get_model_card_by_id(
|
||||
f"{wm.ui_mode}-{wm.selected_project_id}-{wm.selected_model_id}"
|
||||
)
|
||||
delete_model_card_objects(model_card, context)
|
||||
else:
|
||||
model_card = context.scene.speckle_state.model_cards.add()
|
||||
model_card.account_id = wm.selected_account_id
|
||||
model_card.server_url = get_server_url_by_account_id(wm.selected_account_id)
|
||||
model_card.project_id = wm.selected_project_id
|
||||
model_card.project_name = wm.selected_project_name
|
||||
model_card.model_id = wm.selected_model_id
|
||||
model_card.model_name = wm.selected_model_name
|
||||
model_card.is_publish = False
|
||||
model_card.load_option = wm.selected_version_load_option
|
||||
model_card.version_id = wm.selected_version_id
|
||||
model_card.instance_loading_mode = self.instance_loading_mode
|
||||
|
||||
converted_objects = load_operation(context, self.instance_loading_mode)
|
||||
update_model_card_objects(model_card, converted_objects)
|
||||
|
||||
# Clear selected model details from Window Manager
|
||||
wm.selected_account_id = ""
|
||||
wm.selected_project_id = ""
|
||||
wm.selected_project_name = ""
|
||||
wm.selected_model_id = ""
|
||||
wm.selected_model_name = ""
|
||||
wm.selected_version_load_option = ""
|
||||
wm.selected_version_id = ""
|
||||
|
||||
return {"FINISHED"}
|
||||
@@ -0,0 +1,81 @@
|
||||
import bpy
|
||||
from typing import Set
|
||||
from bpy.types import Context
|
||||
from ..utils.version_manager import get_latest_version
|
||||
from ..operations.load_operation import load_operation
|
||||
from ..utils.model_card_utils import (
|
||||
delete_model_card_objects,
|
||||
update_model_card_objects,
|
||||
collect_objects_with_properties,
|
||||
)
|
||||
|
||||
|
||||
class SPECKLE_OT_load_model_card(bpy.types.Operator):
|
||||
bl_idname = "speckle.model_card_load"
|
||||
bl_label = "Load Latest from Speckle"
|
||||
bl_description = "Depending on the load option, loads the latest or a specific version from Speckle"
|
||||
|
||||
model_card_id: bpy.props.StringProperty(name="Model Card ID", default="") # type: ignore
|
||||
|
||||
def execute(self, context: Context) -> Set[str]:
|
||||
wm = context.window_manager
|
||||
|
||||
# Get the model card
|
||||
model_card = context.scene.speckle_state.get_model_card_by_id(
|
||||
self.model_card_id
|
||||
)
|
||||
if model_card is None:
|
||||
self.report({"ERROR"}, "Model card not found")
|
||||
return {"CANCELLED"}
|
||||
|
||||
old_properties = collect_objects_with_properties(model_card)
|
||||
delete_model_card_objects(model_card, context)
|
||||
|
||||
# set wm
|
||||
wm.selected_account_id = model_card.account_id
|
||||
wm.selected_project_id = model_card.project_id
|
||||
wm.selected_model_name = model_card.model_name
|
||||
|
||||
# if load option is set to "LATEST"
|
||||
if model_card.load_option == "LATEST":
|
||||
# get latest version from speckle
|
||||
latest_version_id, message, timestamp = get_latest_version(
|
||||
model_card.account_id, model_card.project_id, model_card.model_id
|
||||
)
|
||||
# set version id in wm
|
||||
wm.selected_version_id = latest_version_id
|
||||
|
||||
# load latest version
|
||||
converted_objects = load_operation(
|
||||
context, model_card.instance_loading_mode
|
||||
)
|
||||
# update model card details
|
||||
update_model_card_objects(model_card, converted_objects, old_properties)
|
||||
model_card.version_id = latest_version_id
|
||||
|
||||
else:
|
||||
# set version id in wm
|
||||
wm.selected_version_id = model_card.version_id
|
||||
|
||||
# load version id
|
||||
converted_objects = load_operation(
|
||||
context, model_card.instance_loading_mode
|
||||
)
|
||||
if not converted_objects:
|
||||
self.report({"ERROR"}, "Load operation failed")
|
||||
return {"CANCELLED"}
|
||||
# update model card details
|
||||
update_model_card_objects(model_card, converted_objects, old_properties)
|
||||
|
||||
# Clear selected model details from Window Manager
|
||||
wm.selected_account_id = ""
|
||||
wm.selected_project_id = ""
|
||||
wm.selected_version_id = ""
|
||||
wm.selected_model_name = ""
|
||||
|
||||
self.report(
|
||||
{"INFO"},
|
||||
f"{len(converted_objects)} objects loaded from Speckle. Model: {model_card.model_name}, Version: {model_card.version_id}",
|
||||
)
|
||||
|
||||
return {"FINISHED"}
|
||||
@@ -0,0 +1,81 @@
|
||||
import bpy
|
||||
from typing import Set
|
||||
from bpy.types import Context, Event
|
||||
from ..operations.publish_operation import publish_operation
|
||||
from ..utils.account_manager import can_create_version
|
||||
|
||||
|
||||
class SPECKLE_OT_publish_model_card(bpy.types.Operator):
|
||||
bl_idname = "speckle.model_card_publish"
|
||||
bl_label = "Publish model"
|
||||
bl_description = "Publish tracked objects to Speckle"
|
||||
|
||||
model_card_id: bpy.props.StringProperty(name="Model Card ID", default="") # type: ignore
|
||||
version_message: bpy.props.StringProperty(name="Version Message", default="") # type: ignore
|
||||
|
||||
def draw(self, context: Context) -> None:
|
||||
layout = self.layout
|
||||
layout.prop(self, "version_message")
|
||||
|
||||
def invoke(self, context: Context, event: Event) -> Set[str]:
|
||||
return context.window_manager.invoke_props_dialog(self)
|
||||
|
||||
def execute(self, context: Context) -> Set[str]:
|
||||
wm = context.window_manager
|
||||
|
||||
# Get the model card
|
||||
model_card = context.scene.speckle_state.get_model_card_by_id(
|
||||
self.model_card_id
|
||||
)
|
||||
|
||||
# On-demand permission check
|
||||
authorized, auth_message = can_create_version(
|
||||
model_card.account_id, model_card.project_id, model_card.model_id
|
||||
)
|
||||
if not authorized:
|
||||
self.report({"ERROR"}, auth_message)
|
||||
return {"CANCELLED"}
|
||||
|
||||
# set wm
|
||||
wm.selected_account_id = model_card.account_id
|
||||
wm.selected_project_id = model_card.project_id
|
||||
wm.selected_model_id = model_card.model_id
|
||||
|
||||
# get model card objects
|
||||
objects_to_convert = []
|
||||
for speckle_obj in model_card.objects:
|
||||
blender_obj = bpy.data.objects.get(speckle_obj.name)
|
||||
if blender_obj:
|
||||
objects_to_convert.append(blender_obj)
|
||||
else:
|
||||
self.report(
|
||||
{"WARNING"}, f"Object '{speckle_obj.name}' not found, skipping"
|
||||
)
|
||||
|
||||
if not objects_to_convert:
|
||||
self.report({"ERROR"}, "No objects to publish")
|
||||
return {"CANCELLED"}
|
||||
|
||||
# publish to speckle
|
||||
success, message, version_id = publish_operation(
|
||||
context,
|
||||
objects_to_convert,
|
||||
self.version_message,
|
||||
model_card.apply_modifiers,
|
||||
)
|
||||
|
||||
if not success:
|
||||
self.report({"ERROR"}, message)
|
||||
return {"CANCELLED"}
|
||||
|
||||
model_card.version_id = version_id
|
||||
model_card.is_publish = True
|
||||
|
||||
# Clear selected model details from Window Manager
|
||||
wm.selected_account_id = ""
|
||||
wm.selected_project_id = ""
|
||||
wm.selected_model_id = ""
|
||||
|
||||
self.report({"INFO"}, message)
|
||||
|
||||
return {"FINISHED"}
|
||||
@@ -0,0 +1,127 @@
|
||||
import bpy
|
||||
import webbrowser
|
||||
from typing import Set
|
||||
from bpy.types import Event, Context, UILayout
|
||||
|
||||
|
||||
class SPECKLE_OT_model_card_settings(bpy.types.Operator):
|
||||
"""
|
||||
manages settings and actions for a Speckle model card
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.model_card_settings"
|
||||
bl_label = "Model Card Settings"
|
||||
bl_description = "More options for the model card"
|
||||
model_card_id: bpy.props.StringProperty(name="Model Card ID", default="") # type:ignore
|
||||
|
||||
def execute(self, context: Context) -> Set[str]:
|
||||
return {"FINISHED"}
|
||||
|
||||
def draw(self, context: Context) -> None:
|
||||
layout: UILayout = self.layout
|
||||
layout.operator(
|
||||
"speckle.view_in_browser", text="View in Browser"
|
||||
).model_card_id = self.model_card_id
|
||||
layout.operator(
|
||||
"speckle.view_model_versions", text="View Model Versions"
|
||||
).model_card_id = self.model_card_id
|
||||
layout.separator()
|
||||
row = layout.row()
|
||||
# add a button for deleting the model card
|
||||
row.alert = True
|
||||
delete_op = row.operator(
|
||||
"speckle.delete_model_card", text="Delete Model Card", icon="TRASH"
|
||||
)
|
||||
delete_op.model_card_id = self.model_card_id
|
||||
|
||||
def invoke(self, context: Context, event: Event) -> Set[str]:
|
||||
wm = context.window_manager
|
||||
return wm.invoke_popup(self)
|
||||
|
||||
|
||||
class SPECKLE_OT_view_in_browser(bpy.types.Operator):
|
||||
"""
|
||||
opens the current model in the Speckle web viewer
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.view_in_browser"
|
||||
bl_label = "View in Browser"
|
||||
bl_description = "View the model in the browser"
|
||||
|
||||
model_card_id: bpy.props.StringProperty() # type: ignore
|
||||
|
||||
def execute(self, context: Context) -> Set[str]:
|
||||
model_card = context.scene.speckle_state.get_model_card_by_id(
|
||||
self.model_card_id
|
||||
)
|
||||
if model_card is None:
|
||||
self.report({"ERROR"}, "Model card not found")
|
||||
return {"CANCELLED"}
|
||||
|
||||
url = f"{model_card.server_url}/projects/{model_card.project_id}/models/{model_card.model_id}"
|
||||
webbrowser.open(url)
|
||||
self.report({"INFO"}, f"Viewing in the browser: {url}")
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
class SPECKLE_OT_view_model_versions(bpy.types.Operator):
|
||||
"""
|
||||
opens the model's version history in the Speckle web app
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.view_model_versions"
|
||||
bl_label = "View Model Versions"
|
||||
bl_description = "View the model versions in the browser"
|
||||
|
||||
model_card_id: bpy.props.StringProperty() # type: ignore
|
||||
|
||||
def execute(self, context: Context) -> Set[str]:
|
||||
model_card = context.scene.speckle_state.get_model_card_by_id(
|
||||
self.model_card_id
|
||||
)
|
||||
if model_card is None:
|
||||
self.report({"ERROR"}, "Model card not found")
|
||||
return {"CANCELLED"}
|
||||
|
||||
url = f"{model_card.server_url}/projects/{model_card.project_id}/models/{model_card.model_id}/versions"
|
||||
webbrowser.open(url)
|
||||
|
||||
self.report({"INFO"}, "Viewing model's versions in the browser")
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
class SPECKLE_OT_delete_model_card(bpy.types.Operator):
|
||||
"""
|
||||
deletes a Speckle model card from the Blender UI
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.delete_model_card"
|
||||
bl_label = "Delete Model Card"
|
||||
bl_description = "Delete this model card"
|
||||
|
||||
model_card_id: bpy.props.StringProperty() # type: ignore
|
||||
|
||||
def execute(self, context: Context) -> Set[str]:
|
||||
model_card = context.scene.speckle_state.get_model_card_by_id(
|
||||
self.model_card_id
|
||||
)
|
||||
if model_card is None:
|
||||
self.report({"ERROR"}, "Model card not found")
|
||||
return {"CANCELLED"}
|
||||
|
||||
model_name = model_card.model_name
|
||||
|
||||
# find the index of the model card and remove it
|
||||
for i, card in enumerate(context.scene.speckle_state.model_cards):
|
||||
if card.get_model_card_id() == self.model_card_id:
|
||||
context.scene.speckle_state.model_cards.remove(i)
|
||||
break
|
||||
|
||||
self.report({"INFO"}, f"Model card '{model_name}' has been deleted")
|
||||
context.window.screen = context.window.screen
|
||||
|
||||
context.area.tag_redraw()
|
||||
return {"FINISHED"}
|
||||
|
||||
def invoke(self, context: Context, event: Event) -> Set[str]:
|
||||
return self.execute(context)
|
||||
@@ -0,0 +1,122 @@
|
||||
import bpy
|
||||
from bpy.types import Context
|
||||
from bpy.types import Event
|
||||
from typing import Set
|
||||
|
||||
from ..operations.publish_operation import publish_operation
|
||||
from ..utils.account_manager import get_server_url_by_account_id, can_create_version
|
||||
from ..utils.model_card_utils import model_card_exists, update_model_card_objects
|
||||
|
||||
|
||||
class SPECKLE_OT_publish(bpy.types.Operator):
|
||||
bl_idname = "speckle.publish"
|
||||
bl_label = "Publish to Speckle"
|
||||
bl_description = "Publish selected objects to Speckle"
|
||||
|
||||
version_message: bpy.props.StringProperty(name="Version Message") # type: ignore
|
||||
apply_modifiers: bpy.props.BoolProperty( # type: ignore
|
||||
name="Apply Modifiers",
|
||||
description="Apply all modifiers to objects before conversion",
|
||||
default=True,
|
||||
)
|
||||
|
||||
def draw(self, context: Context) -> None:
|
||||
layout = self.layout
|
||||
layout.prop(self, "version_message")
|
||||
layout.prop(self, "apply_modifiers")
|
||||
|
||||
def invoke(self, context: Context, event: Event) -> Set[str]:
|
||||
return context.window_manager.invoke_props_dialog(self)
|
||||
|
||||
def execute(self, context: Context) -> Set[str]:
|
||||
wm = context.window_manager
|
||||
|
||||
# check if we have stored objects from selection dialog
|
||||
if not wm.speckle_objects:
|
||||
self.report(
|
||||
{"ERROR"},
|
||||
"No objects selected to publish. Please use 'Select Objects' first.",
|
||||
)
|
||||
return {"CANCELLED"}
|
||||
|
||||
account_id = getattr(wm, "selected_account_id", "")
|
||||
project_id = getattr(wm, "selected_project_id", "")
|
||||
model_id = getattr(wm, "selected_model_id", "")
|
||||
|
||||
if not account_id:
|
||||
self.report({"ERROR"}, "No account selected")
|
||||
return {"CANCELLED"}
|
||||
|
||||
if not project_id:
|
||||
self.report({"ERROR"}, "No project selected")
|
||||
return {"CANCELLED"}
|
||||
|
||||
if not model_id:
|
||||
self.report({"ERROR"}, "No model selected")
|
||||
return {"CANCELLED"}
|
||||
|
||||
authorized, auth_message = can_create_version(account_id, project_id, model_id)
|
||||
if not authorized:
|
||||
self.report({"ERROR"}, auth_message)
|
||||
return {"CANCELLED"}
|
||||
|
||||
objects_to_convert = []
|
||||
for speckle_obj in wm.speckle_objects:
|
||||
blender_obj = bpy.data.objects.get(speckle_obj.name)
|
||||
if blender_obj:
|
||||
objects_to_convert.append(blender_obj)
|
||||
else:
|
||||
self.report(
|
||||
{"WARNING"}, f"Object '{speckle_obj.name}' not found, skipping"
|
||||
)
|
||||
|
||||
if not objects_to_convert:
|
||||
self.report({"ERROR"}, "None of the selected objects could be found")
|
||||
return {"CANCELLED"}
|
||||
|
||||
success, message, version_id = publish_operation(
|
||||
context, objects_to_convert, self.version_message, self.apply_modifiers
|
||||
)
|
||||
|
||||
if not success:
|
||||
self.report({"ERROR"}, message)
|
||||
return {"CANCELLED"}
|
||||
|
||||
# create model card if operation was successful
|
||||
if hasattr(context.scene, "speckle_state") and hasattr(
|
||||
context.scene.speckle_state, "model_cards"
|
||||
):
|
||||
if model_card_exists(
|
||||
wm.selected_project_id, wm.selected_model_id, True, context
|
||||
):
|
||||
model_card = context.scene.speckle_state.get_model_card_by_id(
|
||||
f"{wm.ui_mode}-{wm.selected_project_id}-{wm.selected_model_id}"
|
||||
)
|
||||
else:
|
||||
model_card = context.scene.speckle_state.model_cards.add()
|
||||
|
||||
model_card.account_id = account_id
|
||||
model_card.server_url = get_server_url_by_account_id(account_id)
|
||||
model_card.project_id = project_id
|
||||
model_card.project_name = getattr(wm, "selected_project_name", "")
|
||||
model_card.model_id = model_id
|
||||
model_card.model_name = getattr(wm, "selected_model_name", "")
|
||||
model_card.is_publish = True
|
||||
model_card.load_option = "SPECIFIC" # published versions are specific
|
||||
model_card.version_id = version_id
|
||||
model_card.apply_modifiers = self.apply_modifiers
|
||||
update_model_card_objects(model_card, objects_to_convert)
|
||||
|
||||
# clear selected model details from Window Manager
|
||||
wm.selected_account_id = ""
|
||||
wm.selected_project_id = ""
|
||||
wm.selected_project_name = ""
|
||||
wm.selected_model_id = ""
|
||||
wm.selected_model_name = ""
|
||||
wm.selected_version_load_option = ""
|
||||
wm.selected_version_id = ""
|
||||
wm.speckle_objects.clear()
|
||||
|
||||
self.report({"INFO"}, message)
|
||||
context.area.tag_redraw()
|
||||
return {"FINISHED"}
|
||||
@@ -0,0 +1,34 @@
|
||||
from bpy.types import Operator
|
||||
from bpy.props import StringProperty
|
||||
from ..utils.model_card_utils import select_model_card_objects, zoom_to_selected_objects
|
||||
|
||||
|
||||
class SPECKLE_OT_select_objects(Operator):
|
||||
"""
|
||||
select all objects imported from this Speckle model
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.select_objects"
|
||||
bl_label = "Select Objects"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
bl_description = (
|
||||
"Selects and zooms extents to objects loaded from this Speckle model"
|
||||
)
|
||||
|
||||
model_card_id: StringProperty(
|
||||
name="Model Card ID", description="ID of the model card", default=""
|
||||
) # type: ignore
|
||||
|
||||
def execute(self, context):
|
||||
model_card = context.scene.speckle_state.get_model_card_by_id(
|
||||
self.model_card_id
|
||||
)
|
||||
if model_card is None:
|
||||
self.report({"ERROR"}, "Model card not found")
|
||||
return {"CANCELLED"}
|
||||
|
||||
select_model_card_objects(model_card, context)
|
||||
zoom_to_selected_objects(context)
|
||||
|
||||
self.report({"INFO"}, f"Selected {len(context.selected_objects)} objects")
|
||||
return {"FINISHED"}
|
||||
@@ -0,0 +1,27 @@
|
||||
import bpy
|
||||
import webbrowser
|
||||
from bpy.types import Context
|
||||
|
||||
|
||||
class SPECKLE_OT_update_button(bpy.types.Operator):
|
||||
"""Operator for opening the download URL for the latest Speckle Blender connector"""
|
||||
|
||||
bl_idname = "speckle.update_button"
|
||||
bl_label = "Update"
|
||||
bl_description = "Download the latest version of the Speckle Blender connector"
|
||||
|
||||
def execute(self, context: Context) -> set[str]:
|
||||
wm = context.window_manager
|
||||
|
||||
if not wm.update_url:
|
||||
self.report({"ERROR"}, "No update URL available")
|
||||
return {"CANCELLED"}
|
||||
|
||||
try:
|
||||
webbrowser.open(wm.update_url)
|
||||
self.report({"INFO"}, f"Opening download page for v{wm.latest_version}")
|
||||
except Exception as e:
|
||||
self.report({"ERROR"}, f"Failed to open download page: {str(e)}")
|
||||
return {"CANCELLED"}
|
||||
|
||||
return {"FINISHED"}
|
||||
@@ -0,0 +1,51 @@
|
||||
import bpy
|
||||
from bpy.types import Context
|
||||
from specklepy.core.api.connector_versions import get_latest_version
|
||||
|
||||
# Get current version from bl_info
|
||||
from ... import bl_info
|
||||
|
||||
|
||||
class SPECKLE_OT_version_check(bpy.types.Operator):
|
||||
"""Operator for checking if a newer version of the Speckle Blender connector is available"""
|
||||
|
||||
bl_idname = "speckle.version_check"
|
||||
bl_label = "Check for Updates"
|
||||
bl_description = (
|
||||
"Check if a newer version of the Speckle Blender connector is available"
|
||||
)
|
||||
|
||||
def execute(self, context: Context) -> set[str]:
|
||||
wm = context.window_manager
|
||||
|
||||
# Reset previous state
|
||||
wm.update_available = False
|
||||
wm.latest_version = ""
|
||||
wm.update_url = ""
|
||||
|
||||
try:
|
||||
current_version = bl_info["version"]
|
||||
current_version_str = (
|
||||
f"{current_version[0]}.{current_version[1]}.{current_version[2]}"
|
||||
)
|
||||
|
||||
# Get latest version info
|
||||
latest_version_info = get_latest_version("blender", False)
|
||||
latest_version_str = latest_version_info.number # semantic version string
|
||||
|
||||
# Compare versions - if they're different, show update
|
||||
if latest_version_str != current_version_str:
|
||||
wm.update_available = True
|
||||
wm.latest_version = latest_version_str
|
||||
wm.update_url = str(
|
||||
latest_version_info.url
|
||||
) # Convert HttpUrl to string
|
||||
self.report({"INFO"}, f"Update available: v{latest_version_str}")
|
||||
else:
|
||||
self.report({"INFO"}, "You have the latest version")
|
||||
|
||||
except Exception as e:
|
||||
error_msg = f"Failed to check for updates: {str(e)}"
|
||||
self.report({"ERROR"}, error_msg)
|
||||
|
||||
return {"FINISHED"}
|
||||
@@ -0,0 +1,2 @@
|
||||
from ..operations.load_operation import load_operation # noqa: F401
|
||||
from ..operations.publish_operation import publish_operation # noqa: F401
|
||||
@@ -0,0 +1,310 @@
|
||||
from typing import Dict, Union
|
||||
|
||||
import bpy
|
||||
from bpy.types import Context
|
||||
from specklepy.core.api import host_applications, operations
|
||||
from specklepy.core.api.inputs.version_inputs import MarkReceivedVersionInput
|
||||
from specklepy.logging import metrics
|
||||
from specklepy.objects.graph_traversal.default_traversal import (
|
||||
create_default_traversal_function,
|
||||
)
|
||||
from specklepy.objects.models.collections.collection import Collection as SCollection
|
||||
from specklepy.transports.server import ServerTransport
|
||||
|
||||
from ... import bl_info
|
||||
from ...converter.to_native import (
|
||||
convert_to_native,
|
||||
find_instance_definitions,
|
||||
instance_definition_proxy_to_native,
|
||||
render_material_proxy_to_native,
|
||||
)
|
||||
from ...converter.utils import (
|
||||
build_object_id_map,
|
||||
get_project_workspace_id,
|
||||
)
|
||||
from ..utils.account_manager import _client_cache
|
||||
from ..utils.get_ascendants import get_ascendants
|
||||
|
||||
|
||||
def load_operation(
|
||||
context: Context, instance_loading_mode: str = "INSTANCE_PROXIES"
|
||||
) -> Dict[str, Union[bpy.types.Collection, bpy.types.Object]]:
|
||||
"""
|
||||
load objects from Speckle and maintain hierarchy.
|
||||
"""
|
||||
|
||||
wm = context.window_manager
|
||||
accountId: str = wm.selected_account_id # type: ignore
|
||||
projectId: str = wm.selected_project_id # type: ignore
|
||||
versionId: str = wm.selected_version_id # type: ignore
|
||||
|
||||
# get cached client
|
||||
client = _client_cache.get_client(accountId)
|
||||
if not client:
|
||||
print("No Speckle client found")
|
||||
return {}
|
||||
|
||||
print(f"Using client for account: {accountId}")
|
||||
|
||||
transport = ServerTransport(stream_id=projectId, client=client)
|
||||
|
||||
version = client.version.get(versionId, projectId)
|
||||
obj_id = version.referenced_object
|
||||
if not obj_id:
|
||||
raise ValueError("Unable to receive version beyond workspaces limit")
|
||||
|
||||
version_data = operations.receive(obj_id, transport)
|
||||
|
||||
metrics.set_host_app("blender")
|
||||
client.version.received(
|
||||
MarkReceivedVersionInput(
|
||||
version_id=version.id,
|
||||
project_id=projectId,
|
||||
source_application="blender",
|
||||
)
|
||||
)
|
||||
|
||||
metrics.track(
|
||||
metrics.RECEIVE,
|
||||
client.account,
|
||||
{
|
||||
"ui": "dui3",
|
||||
"hostAppVersion": ".".join(map(str, bl_info["blender"])),
|
||||
"core_version": ".".join(map(str, bl_info["version"])),
|
||||
"sourceHostApp": host_applications.get_host_app_from_string(
|
||||
version.source_application
|
||||
).slug,
|
||||
"isMultiplayer": version.author_user.id != client.account.userInfo.id,
|
||||
"workspace_id": get_project_workspace_id(client, wm.selected_project_id),
|
||||
},
|
||||
)
|
||||
|
||||
# Build object ID map once
|
||||
object_id_map = build_object_id_map(version_data)
|
||||
|
||||
# Create material mapping first
|
||||
material_mapping = render_material_proxy_to_native(version_data)
|
||||
|
||||
definition_collections, definition_objects = instance_definition_proxy_to_native(
|
||||
version_data,
|
||||
material_mapping,
|
||||
instance_loading_mode=instance_loading_mode,
|
||||
object_id_map=object_id_map,
|
||||
)
|
||||
|
||||
definitions_root_collection = None
|
||||
if definition_collections:
|
||||
definitions_root_collection = bpy.data.collections.new("InstanceDefinitions")
|
||||
|
||||
for collection in definition_collections.values():
|
||||
definitions_root_collection.children.link(collection)
|
||||
|
||||
definition_object_ids = set()
|
||||
for definition in find_instance_definitions(version_data).values():
|
||||
definition_object_ids.update(definition.objects)
|
||||
for obj_id in definition.objects:
|
||||
# Use ID map
|
||||
found_obj = object_id_map.get(obj_id)
|
||||
if found_obj:
|
||||
if hasattr(found_obj, "id"):
|
||||
definition_object_ids.add(found_obj.id)
|
||||
if hasattr(found_obj, "applicationId"):
|
||||
definition_object_ids.add(found_obj.applicationId)
|
||||
|
||||
traversal_function = create_default_traversal_function()
|
||||
|
||||
root_collection_name = f"{wm.selected_model_name} - {wm.selected_version_id}"
|
||||
root_collection = bpy.data.collections.new(root_collection_name)
|
||||
context.scene.collection.children.link(root_collection)
|
||||
|
||||
context.window_manager.progress_begin(0, 100)
|
||||
|
||||
converted_objects = definition_objects.copy()
|
||||
|
||||
created_collections = {}
|
||||
created_collections[root_collection_name] = root_collection
|
||||
|
||||
collection_hierarchy = {}
|
||||
all_objects = {}
|
||||
|
||||
speckle_root_id = None
|
||||
|
||||
for traversal_item in traversal_function.traverse(version_data):
|
||||
speckle_obj = traversal_item.current
|
||||
|
||||
# Skip objects that are part of instance definitions
|
||||
if speckle_obj.id in definition_object_ids or (
|
||||
hasattr(speckle_obj, "applicationId")
|
||||
and speckle_obj.applicationId in definition_object_ids
|
||||
):
|
||||
continue
|
||||
|
||||
all_objects[speckle_obj.id] = speckle_obj
|
||||
|
||||
# get all ascendants in order (current to root)
|
||||
ascendants = list(get_ascendants(traversal_item))
|
||||
parent_ascendants = ascendants[1:] if len(ascendants) > 1 else []
|
||||
|
||||
if isinstance(speckle_obj, SCollection):
|
||||
if not parent_ascendants and speckle_root_id is None:
|
||||
speckle_root_id = speckle_obj.id
|
||||
|
||||
collection_name = getattr(
|
||||
speckle_obj, "name", f"Collection_{speckle_obj.id}"
|
||||
)
|
||||
|
||||
parent_id = None
|
||||
for parent in parent_ascendants:
|
||||
if isinstance(parent, SCollection) and hasattr(parent, "id"):
|
||||
parent_id = parent.id
|
||||
break
|
||||
|
||||
collection_hierarchy[speckle_obj.id] = {
|
||||
"id": speckle_obj.id,
|
||||
"name": collection_name,
|
||||
"parent_id": parent_id,
|
||||
"applicationId": getattr(speckle_obj, "applicationId", ""),
|
||||
"blender_collection": None,
|
||||
"full_path": [collection_name],
|
||||
}
|
||||
|
||||
if parent_id in collection_hierarchy:
|
||||
collection_hierarchy[speckle_obj.id]["full_path"] = (
|
||||
collection_hierarchy[parent_id]["full_path"] + [collection_name]
|
||||
)
|
||||
|
||||
else:
|
||||
pass
|
||||
|
||||
def get_collection_depth(coll_id):
|
||||
parent_id = collection_hierarchy[coll_id]["parent_id"]
|
||||
if parent_id is None:
|
||||
return 0
|
||||
if parent_id not in collection_hierarchy:
|
||||
return 0
|
||||
return 1 + get_collection_depth(parent_id)
|
||||
|
||||
sorted_collections = sorted(
|
||||
collection_hierarchy.keys(),
|
||||
key=lambda coll_id: (
|
||||
get_collection_depth(coll_id),
|
||||
collection_hierarchy[coll_id]["name"],
|
||||
),
|
||||
)
|
||||
|
||||
if speckle_root_id and speckle_root_id in collection_hierarchy:
|
||||
collection_hierarchy[speckle_root_id]["blender_collection"] = root_collection
|
||||
converted_objects[speckle_root_id] = root_collection
|
||||
|
||||
# create collections in depth order (skip the root that's already mapped)
|
||||
for coll_id in sorted_collections:
|
||||
if coll_id == speckle_root_id:
|
||||
continue
|
||||
|
||||
coll_info = collection_hierarchy[coll_id]
|
||||
coll_name = coll_info["name"]
|
||||
parent_id = coll_info["parent_id"]
|
||||
full_path = coll_info["full_path"]
|
||||
|
||||
collection_key = tuple(full_path)
|
||||
|
||||
parent_collection = root_collection
|
||||
if parent_id and parent_id in collection_hierarchy:
|
||||
parent_info = collection_hierarchy[parent_id]
|
||||
if parent_info["blender_collection"]:
|
||||
parent_collection = parent_info["blender_collection"]
|
||||
|
||||
if collection_key in created_collections:
|
||||
print(f"Collection already exists: {coll_name}")
|
||||
blender_collection = created_collections[collection_key]
|
||||
else:
|
||||
blender_collection = bpy.data.collections.new(coll_name)
|
||||
if coll_info.get("applicationId"):
|
||||
blender_collection["applicationId"] = coll_info["applicationId"]
|
||||
parent_collection.children.link(blender_collection)
|
||||
created_collections[collection_key] = blender_collection
|
||||
|
||||
coll_info["blender_collection"] = blender_collection
|
||||
converted_objects[coll_id] = blender_collection
|
||||
|
||||
conversion_count = 0
|
||||
for traversal_item in traversal_function.traverse(version_data):
|
||||
speckle_obj = traversal_item.current
|
||||
|
||||
if isinstance(speckle_obj, SCollection):
|
||||
continue
|
||||
|
||||
if not hasattr(speckle_obj, "id"):
|
||||
print("Skipping object without ID")
|
||||
continue
|
||||
|
||||
# Skip objects that are part of instance definitions
|
||||
if speckle_obj.id in definition_object_ids or (
|
||||
hasattr(speckle_obj, "applicationId")
|
||||
and speckle_obj.applicationId in definition_object_ids
|
||||
):
|
||||
continue
|
||||
|
||||
if speckle_obj.id in converted_objects:
|
||||
continue
|
||||
|
||||
try:
|
||||
target_collection = root_collection
|
||||
ascendants = list(get_ascendants(traversal_item))
|
||||
|
||||
for parent in ascendants[1:] if len(ascendants) > 1 else []:
|
||||
if isinstance(parent, SCollection) and hasattr(parent, "id"):
|
||||
parent_id = parent.id
|
||||
if parent_id in collection_hierarchy:
|
||||
coll_info = collection_hierarchy[parent_id]
|
||||
if coll_info["blender_collection"]:
|
||||
target_collection = coll_info["blender_collection"]
|
||||
break
|
||||
|
||||
blender_obj = convert_to_native(
|
||||
speckle_obj,
|
||||
material_mapping,
|
||||
definition_collections=definition_collections,
|
||||
root_collection=target_collection,
|
||||
instance_loading_mode=instance_loading_mode,
|
||||
)
|
||||
|
||||
if blender_obj is None:
|
||||
continue
|
||||
|
||||
converted_objects[speckle_obj.id] = blender_obj
|
||||
if hasattr(speckle_obj, "applicationId"):
|
||||
converted_objects[speckle_obj.applicationId] = blender_obj
|
||||
|
||||
if not isinstance(blender_obj, bpy.types.Collection):
|
||||
try:
|
||||
already_linked = False
|
||||
for coll in bpy.data.collections:
|
||||
if blender_obj.name in coll.objects:
|
||||
already_linked = True
|
||||
|
||||
if not already_linked:
|
||||
target_collection.objects.link(blender_obj)
|
||||
|
||||
except RuntimeError as e:
|
||||
print(f"Error linking object to collection: {e}")
|
||||
|
||||
except Exception as e:
|
||||
print(f"Error converting {speckle_obj.speckle_type}: {str(e)}")
|
||||
import traceback
|
||||
|
||||
traceback.print_exc()
|
||||
|
||||
conversion_count += 1
|
||||
if conversion_count % 10 == 0:
|
||||
context.window_manager.progress_update(min(conversion_count, 100))
|
||||
|
||||
context.window_manager.progress_end()
|
||||
|
||||
for area in context.screen.areas:
|
||||
if area.type == "OUTLINER":
|
||||
area.tag_redraw()
|
||||
|
||||
print(f"\nLoad process completed. Imported {len(converted_objects)} objects.")
|
||||
|
||||
return converted_objects
|
||||
@@ -0,0 +1,452 @@
|
||||
import bpy
|
||||
from bpy.types import Context, Collection as BlenderCollection
|
||||
from typing import List, Optional, Dict, Tuple
|
||||
|
||||
from specklepy.objects import Base
|
||||
from specklepy.objects.models.collections.collection import Collection
|
||||
from specklepy.core.api import operations
|
||||
from specklepy.transports.server import ServerTransport
|
||||
from specklepy.core.api.inputs.version_inputs import CreateVersionInput
|
||||
from specklepy.objects.models.units import Units
|
||||
from specklepy.logging.exceptions import GraphQLException, WorkspacePermissionException
|
||||
from specklepy.core.api.inputs.model_ingestion_inputs import (
|
||||
ModelIngestionCreateInput,
|
||||
ModelIngestionStartProcessingInput,
|
||||
ModelIngestionSuccessInput,
|
||||
ModelIngestionFailedInput,
|
||||
SourceDataInput,
|
||||
)
|
||||
|
||||
from ...converter.to_speckle import convert_to_speckle
|
||||
from ...converter.to_speckle.material_to_speckle import (
|
||||
add_render_material_proxies_to_base,
|
||||
)
|
||||
from ...converter.utils import get_project_workspace_id
|
||||
from ..utils.account_manager import _client_cache
|
||||
from specklepy.logging import metrics
|
||||
from ... import bl_info
|
||||
|
||||
|
||||
def _check_use_model_ingestion_send(client, project_id: str, model_id: str) -> bool:
|
||||
"""Check if the server supports model ingestion and the user is authorized."""
|
||||
try:
|
||||
result = client.model.can_create_model_ingestion(project_id, model_id)
|
||||
result.ensure_authorised()
|
||||
return True
|
||||
except GraphQLException:
|
||||
return False
|
||||
|
||||
|
||||
def _build_source_data() -> SourceDataInput:
|
||||
"""Build data input for model ingestion."""
|
||||
file_name = bpy.path.basename(bpy.data.filepath)
|
||||
if not file_name:
|
||||
file_name = "Untitled.blend"
|
||||
|
||||
file_size_bytes: Optional[int] = None
|
||||
if bpy.data.filepath:
|
||||
import os
|
||||
|
||||
try:
|
||||
file_size_bytes = os.path.getsize(bpy.data.filepath)
|
||||
except OSError:
|
||||
pass
|
||||
|
||||
blender_version = ".".join(map(str, bl_info["blender"]))
|
||||
return SourceDataInput(
|
||||
source_application_slug="blender",
|
||||
source_application_version=blender_version,
|
||||
file_name=file_name,
|
||||
file_size_bytes=file_size_bytes,
|
||||
)
|
||||
|
||||
|
||||
def _send_via_ingestion(
|
||||
client,
|
||||
project_id: str,
|
||||
model_id: str,
|
||||
obj_id: str,
|
||||
version_message: str,
|
||||
) -> str:
|
||||
"""Send via the model ingestion. Returns version_id."""
|
||||
source_data = _build_source_data()
|
||||
|
||||
create_input = ModelIngestionCreateInput(
|
||||
project_id=project_id,
|
||||
model_id=model_id,
|
||||
source_data=source_data,
|
||||
progress_message="Model ingestion created",
|
||||
)
|
||||
ingestion = client.model_ingestion.create(create_input)
|
||||
ingestion_id = ingestion.id
|
||||
|
||||
try:
|
||||
start_input = ModelIngestionStartProcessingInput(
|
||||
project_id=project_id,
|
||||
ingestion_id=ingestion_id,
|
||||
progress_message="Processing model ingestion",
|
||||
source_data=source_data,
|
||||
)
|
||||
client.model_ingestion.start_processing(start_input)
|
||||
|
||||
success_input = ModelIngestionSuccessInput(
|
||||
project_id=project_id,
|
||||
ingestion_id=ingestion_id,
|
||||
root_object_id=obj_id,
|
||||
version_message=version_message,
|
||||
)
|
||||
version_id = client.model_ingestion.complete(success_input)
|
||||
return version_id
|
||||
except Exception:
|
||||
try:
|
||||
fail_input = ModelIngestionFailedInput(
|
||||
project_id=project_id,
|
||||
ingestion_id=ingestion_id,
|
||||
error_reason="Failed during processing",
|
||||
)
|
||||
client.model_ingestion.fail_with_error(fail_input)
|
||||
except Exception:
|
||||
pass
|
||||
raise
|
||||
|
||||
|
||||
def _send_via_version_create(
|
||||
client,
|
||||
project_id: str,
|
||||
model_id: str,
|
||||
obj_id: str,
|
||||
version_message: str,
|
||||
) -> str:
|
||||
"""Send via the legacy version.create() flow. Returns version_id."""
|
||||
version_input = CreateVersionInput(
|
||||
objectId=obj_id,
|
||||
modelId=model_id,
|
||||
projectId=project_id,
|
||||
message=version_message,
|
||||
sourceApplication="blender",
|
||||
)
|
||||
version = client.version.create(version_input)
|
||||
return version.id
|
||||
|
||||
|
||||
def publish_operation(
|
||||
context: Context,
|
||||
objects_to_convert: List,
|
||||
version_message: str = "",
|
||||
apply_modifiers: bool = True,
|
||||
) -> Tuple[bool, str, Optional[str]]:
|
||||
"""
|
||||
publish objects to speckle
|
||||
"""
|
||||
wm = context.window_manager
|
||||
|
||||
try:
|
||||
# get cached client
|
||||
client = _client_cache.get_client(wm.selected_account_id)
|
||||
if not client:
|
||||
return False, "No Speckle client found", None
|
||||
|
||||
project_id = wm.selected_project_id
|
||||
model_id = wm.selected_model_id
|
||||
|
||||
# check ingestion support before sending data (fail fast on permission errors)
|
||||
use_ingestion = _check_use_model_ingestion_send(client, project_id, model_id)
|
||||
|
||||
transport = ServerTransport(stream_id=project_id, client=client)
|
||||
|
||||
# build collection hierarchy and convert objects
|
||||
root_collection = build_collection_hierarchy(
|
||||
context, objects_to_convert, apply_modifiers
|
||||
)
|
||||
|
||||
if not root_collection:
|
||||
return False, "No objects could be converted to Speckle format", None
|
||||
|
||||
# add material proxies
|
||||
add_render_material_proxies_to_base(root_collection, objects_to_convert)
|
||||
|
||||
obj_id = operations.send(root_collection, [transport])
|
||||
|
||||
if use_ingestion:
|
||||
version_id = _send_via_ingestion(
|
||||
client, project_id, model_id, obj_id, version_message
|
||||
)
|
||||
else:
|
||||
version_id = _send_via_version_create(
|
||||
client, project_id, model_id, obj_id, version_message
|
||||
)
|
||||
|
||||
# Get account for metrics tracking
|
||||
from specklepy.core.api.credentials import get_local_accounts
|
||||
|
||||
account = next(
|
||||
(acc for acc in get_local_accounts() if acc.id == wm.selected_account_id),
|
||||
None,
|
||||
)
|
||||
|
||||
if account:
|
||||
# track metrics
|
||||
metrics.set_host_app("blender")
|
||||
metrics.track(
|
||||
metrics.SEND,
|
||||
account,
|
||||
{
|
||||
"ui": "dui3",
|
||||
"hostAppVersion": ".".join(map(str, bl_info["blender"])),
|
||||
"core_version": ".".join(map(str, bl_info["version"])),
|
||||
"workspace_id": get_project_workspace_id(client, project_id),
|
||||
},
|
||||
)
|
||||
|
||||
# count total objects for success message
|
||||
total_objects = count_objects_in_collection(root_collection)
|
||||
|
||||
return (
|
||||
True,
|
||||
f"Successfully published {total_objects} objects with hierarchy to Speckle",
|
||||
version_id,
|
||||
)
|
||||
|
||||
except WorkspacePermissionException as e:
|
||||
return False, f"Permission denied: {str(e)}", None
|
||||
|
||||
except Exception as e:
|
||||
import traceback
|
||||
|
||||
traceback.print_exc()
|
||||
# Clear cache on error to prevent stale clients
|
||||
_client_cache.clear()
|
||||
return False, f"Failed to publish: {str(e)}", None
|
||||
|
||||
|
||||
def build_collection_hierarchy(
|
||||
context: Context, objects_to_convert: List, apply_modifiers: bool = True
|
||||
) -> Optional[Collection]:
|
||||
"""
|
||||
build a speckle collection hierarchy that mimicks blender's collection structure
|
||||
"""
|
||||
# set name for root collection
|
||||
file_name = bpy.path.basename(bpy.data.filepath)
|
||||
collection_name = file_name if file_name else "Untitled.blend"
|
||||
|
||||
collection_data = analyze_collection_structure(objects_to_convert)
|
||||
|
||||
if not collection_data["objects"] and not collection_data["collections"]:
|
||||
return None
|
||||
|
||||
converted_objects = convert_selected_objects(
|
||||
context, objects_to_convert, apply_modifiers
|
||||
)
|
||||
if not converted_objects:
|
||||
return None
|
||||
|
||||
# create the root Speckle collection
|
||||
root_collection = Collection(name=collection_name)
|
||||
root_collection.units = get_scene_units(context.scene).value
|
||||
root_collection["version"] = 3
|
||||
|
||||
# maps Blender collection to Speckle collection
|
||||
collection_mapping = {} #
|
||||
|
||||
# create Speckle collections for each blender collection
|
||||
for blender_coll in collection_data["collections"]:
|
||||
speckle_coll = Collection(name=blender_coll.name)
|
||||
speckle_coll.units = root_collection.units
|
||||
collection_mapping[blender_coll] = speckle_coll
|
||||
|
||||
for blender_coll in collection_data["collections"]:
|
||||
speckle_coll = collection_mapping[blender_coll]
|
||||
|
||||
parent_coll = find_parent_collection(
|
||||
blender_coll, collection_data["collections"]
|
||||
)
|
||||
|
||||
if parent_coll and parent_coll in collection_mapping:
|
||||
parent_speckle_coll = collection_mapping[parent_coll]
|
||||
parent_speckle_coll.elements.append(speckle_coll)
|
||||
else:
|
||||
root_collection.elements.append(speckle_coll)
|
||||
|
||||
# assign objects to their collections
|
||||
object_mapping = {}
|
||||
for i, blender_obj in enumerate(objects_to_convert):
|
||||
if i < len(converted_objects) and converted_objects[i] is not None:
|
||||
object_mapping[blender_obj] = converted_objects[i]
|
||||
|
||||
for blender_obj, speckle_obj in object_mapping.items():
|
||||
placed = False
|
||||
|
||||
target_collection = find_target_collection_for_object(
|
||||
blender_obj, collection_data["collections"]
|
||||
)
|
||||
|
||||
if target_collection and target_collection in collection_mapping:
|
||||
collection_mapping[target_collection].elements.append(speckle_obj)
|
||||
placed = True
|
||||
|
||||
# if not placed in any subcollection, add to root
|
||||
if not placed:
|
||||
root_collection.elements.append(speckle_obj)
|
||||
|
||||
return root_collection
|
||||
|
||||
|
||||
def analyze_collection_structure(objects: List) -> Dict:
|
||||
"""
|
||||
analyze the collection structure of the given objects
|
||||
"""
|
||||
collections_set = set()
|
||||
objects_collections = {}
|
||||
|
||||
direct_collections = set()
|
||||
for obj in objects:
|
||||
obj_collections = []
|
||||
for collection in bpy.data.collections:
|
||||
if obj.name in collection.objects:
|
||||
direct_collections.add(collection)
|
||||
obj_collections.append(collection)
|
||||
objects_collections[obj] = obj_collections
|
||||
|
||||
# find all ancestor collections
|
||||
def find_all_ancestors(collection):
|
||||
"""recursively find all ancestor collections"""
|
||||
ancestors = set()
|
||||
|
||||
for potential_parent in bpy.data.collections:
|
||||
if collection.name in potential_parent.children:
|
||||
ancestors.add(potential_parent)
|
||||
# Recursively find ancestors of the parent
|
||||
ancestors.update(find_all_ancestors(potential_parent))
|
||||
|
||||
return ancestors
|
||||
|
||||
for collection in direct_collections:
|
||||
collections_set.add(collection)
|
||||
ancestors = find_all_ancestors(collection)
|
||||
collections_set.update(ancestors)
|
||||
|
||||
collections_list = list(collections_set)
|
||||
collections_list.sort(key=lambda c: get_collection_depth(c))
|
||||
|
||||
return {
|
||||
"collections": collections_list,
|
||||
"objects": objects,
|
||||
"object_collections": objects_collections,
|
||||
}
|
||||
|
||||
|
||||
def get_collection_depth(collection: BlenderCollection) -> int:
|
||||
"""
|
||||
get the depth of a collection in the hierarchy
|
||||
"""
|
||||
depth = 0
|
||||
for scene in bpy.data.scenes:
|
||||
if collection.name in scene.collection.children:
|
||||
return depth
|
||||
|
||||
for parent_coll in bpy.data.collections:
|
||||
if collection.name in parent_coll.children:
|
||||
return get_collection_depth(parent_coll) + 1
|
||||
|
||||
return depth
|
||||
|
||||
|
||||
def find_parent_collection(
|
||||
collection: BlenderCollection, all_collections: List[BlenderCollection]
|
||||
) -> Optional[BlenderCollection]:
|
||||
"""
|
||||
find the parent collection
|
||||
"""
|
||||
for potential_parent in all_collections:
|
||||
if collection.name in potential_parent.children:
|
||||
return potential_parent
|
||||
return None
|
||||
|
||||
|
||||
def find_target_collection_for_object(
|
||||
obj, collections: List[BlenderCollection]
|
||||
) -> Optional[BlenderCollection]:
|
||||
"""
|
||||
find the deepest collection that contains this object
|
||||
"""
|
||||
target_collection = None
|
||||
max_depth = -1
|
||||
|
||||
for collection in collections:
|
||||
if obj.name in collection.objects:
|
||||
depth = get_collection_depth(collection)
|
||||
if depth > max_depth:
|
||||
max_depth = depth
|
||||
target_collection = collection
|
||||
|
||||
return target_collection
|
||||
|
||||
|
||||
def convert_selected_objects(
|
||||
context: Context, objects_to_convert: List, apply_modifiers: bool = True
|
||||
) -> List[Optional[Base]]:
|
||||
"""
|
||||
convert selected objects to Speckle format with proper units
|
||||
"""
|
||||
scene = context.scene
|
||||
units = get_scene_units(scene)
|
||||
scale_factor = scene.unit_settings.scale_length
|
||||
|
||||
speckle_objects = []
|
||||
for obj in objects_to_convert:
|
||||
if not obj or obj.type not in ["MESH", "CURVE", "EMPTY"]:
|
||||
speckle_objects.append(None)
|
||||
continue
|
||||
|
||||
speckle_obj = convert_to_speckle(
|
||||
obj, scale_factor, units.value, apply_modifiers
|
||||
)
|
||||
speckle_objects.append(speckle_obj)
|
||||
|
||||
return speckle_objects
|
||||
|
||||
|
||||
def get_scene_units(scene) -> Units:
|
||||
"""
|
||||
get units from Blender's unit system
|
||||
"""
|
||||
unit_settings = scene.unit_settings
|
||||
|
||||
if unit_settings.system == "METRIC":
|
||||
if unit_settings.length_unit == "METERS":
|
||||
return Units.m
|
||||
elif unit_settings.length_unit == "CENTIMETERS":
|
||||
return Units.cm
|
||||
elif unit_settings.length_unit == "MILLIMETERS":
|
||||
return Units.mm
|
||||
elif unit_settings.length_unit == "KILOMETERS":
|
||||
return Units.km
|
||||
else:
|
||||
return Units.m
|
||||
elif unit_settings.system == "IMPERIAL":
|
||||
if unit_settings.length_unit == "FEET":
|
||||
return Units.feet
|
||||
elif unit_settings.length_unit == "INCHES":
|
||||
return Units.inches
|
||||
elif unit_settings.length_unit == "YARDS":
|
||||
return Units.yards
|
||||
elif unit_settings.length_unit == "MILES":
|
||||
return Units.miles
|
||||
else:
|
||||
return Units.feet
|
||||
else:
|
||||
return Units.m # default to meters
|
||||
|
||||
|
||||
def count_objects_in_collection(collection: Collection) -> int:
|
||||
"""
|
||||
recursively count all objects in a collection and its sub-collections
|
||||
"""
|
||||
count = 0
|
||||
if hasattr(collection, "elements"):
|
||||
for element in collection.elements:
|
||||
if isinstance(element, Collection):
|
||||
count += count_objects_in_collection(element)
|
||||
else:
|
||||
count += 1
|
||||
return count
|
||||
@@ -0,0 +1,31 @@
|
||||
import bpy
|
||||
from bpy.props import CollectionProperty
|
||||
from bpy.types import PropertyGroup
|
||||
from typing import Optional
|
||||
|
||||
from ..utils.property_groups import speckle_model_card
|
||||
|
||||
|
||||
class SpeckleState(PropertyGroup):
|
||||
"""
|
||||
manages the state of the Speckle addon in Blender
|
||||
"""
|
||||
|
||||
model_cards: CollectionProperty(type=speckle_model_card) # type: ignore
|
||||
|
||||
def get_model_card_by_id(self, model_card_id: str) -> Optional[speckle_model_card]:
|
||||
"""Find a model card by its ID."""
|
||||
for model_card in self.model_cards:
|
||||
if model_card.get_model_card_id() == model_card_id:
|
||||
return model_card
|
||||
return None
|
||||
|
||||
|
||||
def register() -> None:
|
||||
bpy.utils.register_class(SpeckleState)
|
||||
bpy.types.Scene.speckle_state = bpy.props.PointerProperty(type=SpeckleState) # type: ignore
|
||||
|
||||
|
||||
def unregister() -> None:
|
||||
del bpy.types.Scene.speckle_state
|
||||
bpy.utils.unregister_class(SpeckleState)
|
||||
@@ -0,0 +1 @@
|
||||
from .main_panel import SPECKLE_PT_main_panel # noqa: F401
|
||||
@@ -0,0 +1,147 @@
|
||||
import bpy
|
||||
from bpy.types import Context, Event
|
||||
from typing import List, Tuple
|
||||
from ..utils.account_manager import (
|
||||
get_account_enum_items,
|
||||
speckle_account,
|
||||
speckle_workspace,
|
||||
get_workspaces,
|
||||
get_active_workspace,
|
||||
get_account_from_id,
|
||||
)
|
||||
from ..utils.project_manager import get_projects_for_account
|
||||
from ..ui.project_selection_dialog import speckle_project
|
||||
|
||||
|
||||
class SPECKLE_UL_accounts_list(bpy.types.UIList):
|
||||
"""
|
||||
UIList for displaying accounts
|
||||
"""
|
||||
|
||||
def draw_item(
|
||||
self,
|
||||
context: Context,
|
||||
layout: bpy.types.UILayout,
|
||||
data: bpy.types.PropertyGroup,
|
||||
item: bpy.types.PropertyGroup,
|
||||
icon: str,
|
||||
active_data: bpy.types.PropertyGroup,
|
||||
active_propname: str,
|
||||
) -> None:
|
||||
if self.layout_type in {"DEFAULT", "COMPACT"}:
|
||||
row = layout.row()
|
||||
row.label(text=item.user_name)
|
||||
row.label(text=item.server_url)
|
||||
row.label(text=item.user_email)
|
||||
elif self.layout_type == "GRID":
|
||||
layout.alignment = "CENTER"
|
||||
layout.label(text=item.user_name)
|
||||
|
||||
|
||||
class SPECKLE_OT_account_selection_dialog(bpy.types.Operator):
|
||||
"""
|
||||
operator for displaying and handling the account selection dialog
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.account_selection_dialog"
|
||||
bl_label = "Select Account"
|
||||
bl_description = "Select account"
|
||||
|
||||
account_index: bpy.props.IntProperty(default=0) # type: ignore
|
||||
|
||||
def invoke(self, context: Context, event: Event) -> set[str]:
|
||||
wm = context.window_manager
|
||||
# Clear existing accounts
|
||||
wm.speckle_accounts.clear()
|
||||
|
||||
# Save selected account
|
||||
current_account_index = 0
|
||||
|
||||
# Fetch accounts
|
||||
for i, (id, user_name, server_url, user_email) in enumerate(
|
||||
get_account_enum_items()
|
||||
):
|
||||
account: speckle_account = wm.speckle_accounts.add()
|
||||
account.id = id
|
||||
account.user_name = user_name
|
||||
account.server_url = server_url
|
||||
account.user_email = user_email
|
||||
if id == wm.selected_account_id:
|
||||
current_account_index = i
|
||||
|
||||
self.account_index = current_account_index
|
||||
return context.window_manager.invoke_props_dialog(self)
|
||||
|
||||
def draw(self, context: Context) -> None:
|
||||
layout = self.layout
|
||||
wm = context.window_manager
|
||||
row = layout.row()
|
||||
# add account button
|
||||
if wm.selected_account_id == "NO_ACCOUNTS":
|
||||
add_account_button_text = "Sign In"
|
||||
add_account_button_icon = "WORLD"
|
||||
else:
|
||||
add_account_button_text = "Add Account"
|
||||
add_account_button_icon = "ADD"
|
||||
row.operator(
|
||||
"speckle.add_account",
|
||||
icon=add_account_button_icon,
|
||||
text=add_account_button_text,
|
||||
)
|
||||
|
||||
if wm.selected_account_id != "NO_ACCOUNTS":
|
||||
layout.template_list(
|
||||
"SPECKLE_UL_accounts_list",
|
||||
"",
|
||||
context.window_manager,
|
||||
"speckle_accounts",
|
||||
self,
|
||||
"account_index",
|
||||
)
|
||||
|
||||
def execute(self, context: Context) -> set[str]:
|
||||
wm = context.window_manager
|
||||
# update the selected account id
|
||||
account = get_account_from_id(wm.speckle_accounts[self.account_index].id)
|
||||
wm.selected_account_id = account.id
|
||||
self.report(
|
||||
{"INFO"},
|
||||
f"Selected account: {account.userInfo.name} - {account.userInfo.email} - {account.serverInfo.url}",
|
||||
)
|
||||
update_workspaces_list(context)
|
||||
update_projects_list(context)
|
||||
# redraw the area
|
||||
context.area.tag_redraw()
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
def update_workspaces_list(context: Context) -> None:
|
||||
wm = context.window_manager
|
||||
wm.speckle_workspaces.clear()
|
||||
workspaces = get_workspaces(wm.selected_account_id)
|
||||
for id, name in workspaces:
|
||||
workspace: speckle_workspace = wm.speckle_workspaces.add()
|
||||
workspace.id = id
|
||||
workspace.name = name
|
||||
active_workspace = get_active_workspace(wm.selected_account_id)
|
||||
if active_workspace:
|
||||
wm.selected_workspace.id = active_workspace["id"]
|
||||
elif wm.speckle_workspaces:
|
||||
wm.selected_workspace.id = wm.speckle_workspaces[0].id
|
||||
print("Updated Workspaces List!")
|
||||
|
||||
|
||||
def update_projects_list(context: Context) -> None:
|
||||
wm = context.window_manager
|
||||
wm.speckle_projects.clear()
|
||||
projects: List[Tuple[str, str, str, str, bool]] = get_projects_for_account(
|
||||
wm.selected_account_id, workspace_id=wm.selected_workspace.id
|
||||
)
|
||||
for name, role, updated, id, can_receive in projects:
|
||||
project: speckle_project = wm.speckle_projects.add()
|
||||
project.name = name
|
||||
project.role = role
|
||||
project.updated = updated
|
||||
project.id = id
|
||||
project.can_receive = can_receive
|
||||
print("Updated Projects List!")
|
||||
@@ -0,0 +1,34 @@
|
||||
from typing import Optional, Dict
|
||||
import os
|
||||
import bpy.utils.previews
|
||||
|
||||
speckle_icons: Optional[Dict[str, bpy.types.ImagePreview]] = None
|
||||
|
||||
|
||||
def load_icons() -> None:
|
||||
global speckle_icons
|
||||
speckle_icons = bpy.utils.previews.new()
|
||||
icons_dir = os.path.dirname(__file__)
|
||||
speckle_logo_icon_path = os.path.join(icons_dir, "speckle-logo.png")
|
||||
if os.path.exists(speckle_logo_icon_path):
|
||||
speckle_icons.load("speckle_logo", speckle_logo_icon_path, "IMAGE")
|
||||
else:
|
||||
print(f"[Speckle] WARNING ‑ icon file not found: {speckle_logo_icon_path}")
|
||||
object_highlight_icon_path = os.path.join(icons_dir, "object-highlight.png")
|
||||
if os.path.exists(object_highlight_icon_path):
|
||||
speckle_icons.load("object_highlight", object_highlight_icon_path, "IMAGE")
|
||||
else:
|
||||
print(f"[Speckle] WARNING ‑ icon file not found: {object_highlight_icon_path}")
|
||||
|
||||
|
||||
def unload_icons() -> None:
|
||||
global speckle_icons
|
||||
if speckle_icons is not None:
|
||||
bpy.utils.previews.remove(speckle_icons)
|
||||
|
||||
|
||||
def get_icon(icon_name: str) -> int:
|
||||
global speckle_icons
|
||||
if speckle_icons is None:
|
||||
raise ValueError("Icons not loaded")
|
||||
return speckle_icons[icon_name].icon_id
|
||||
@@ -0,0 +1,106 @@
|
||||
import bpy
|
||||
from bpy.types import UILayout, Context
|
||||
from .icons import get_icon
|
||||
|
||||
|
||||
class SPECKLE_PT_main_panel(bpy.types.Panel):
|
||||
"""
|
||||
main panel for the Speckle addon.
|
||||
"""
|
||||
|
||||
bl_label = "Speckle"
|
||||
|
||||
bl_idname = "SPECKLE_PT_main_panel"
|
||||
bl_space_type = "VIEW_3D"
|
||||
bl_region_type = "UI"
|
||||
bl_category = "Speckle"
|
||||
|
||||
def draw(self, context: Context) -> None:
|
||||
layout: UILayout = self.layout
|
||||
layout.label(text="Speckle Connector", icon_value=get_icon("speckle_logo"))
|
||||
|
||||
# check to see if there are any speckle models in the file
|
||||
if not context.scene.speckle_state.model_cards:
|
||||
layout.label(text="Hello!")
|
||||
layout.label(text="There are no Speckle models in this file yet.")
|
||||
|
||||
layout.separator()
|
||||
|
||||
wm = context.window_manager
|
||||
project_selected = bool(getattr(wm, "selected_project_name", None))
|
||||
model_selected = bool(getattr(wm, "selected_model_name", None))
|
||||
version_selected = bool(getattr(wm, "selected_version_id", None))
|
||||
selection_made = bool(getattr(wm, "speckle_objects", None))
|
||||
|
||||
# UI Mode Switch
|
||||
row = layout.row()
|
||||
row.prop(wm, "ui_mode", expand=True)
|
||||
|
||||
# select Project button
|
||||
row = layout.row()
|
||||
project_name = getattr(wm, "selected_project_name", "")
|
||||
project_button_text = project_name if project_selected else "Select Project"
|
||||
project_button_icon = "CHECKMARK" if project_selected else "PLUS"
|
||||
row.operator(
|
||||
"speckle.project_selection_dialog",
|
||||
text=project_button_text,
|
||||
icon=project_button_icon,
|
||||
)
|
||||
|
||||
# select Model button
|
||||
row = layout.row()
|
||||
model_name = getattr(wm, "selected_model_name", "")
|
||||
model_button_text = model_name if model_selected else "Select Model"
|
||||
model_button_icon = "CHECKMARK" if model_selected else "PLUS"
|
||||
row.enabled = project_selected
|
||||
row.operator(
|
||||
"speckle.model_selection_dialog",
|
||||
text=model_button_text,
|
||||
icon=model_button_icon,
|
||||
)
|
||||
if wm.ui_mode == "PUBLISH":
|
||||
# TODO: implement Publish flow
|
||||
# Selection filter
|
||||
row = layout.row()
|
||||
row.enabled = project_selected and model_selected
|
||||
selection_button_text = (
|
||||
f"{len(wm.speckle_objects)} Objects"
|
||||
if wm.speckle_objects
|
||||
else "Select Objects"
|
||||
)
|
||||
row.operator(
|
||||
"speckle.selection_filter_dialog",
|
||||
text=selection_button_text,
|
||||
icon="PLUS",
|
||||
).model_card_id = ""
|
||||
|
||||
# Publish button
|
||||
row = layout.row()
|
||||
row.enabled = project_selected and model_selected and selection_made
|
||||
row.operator("speckle.publish", text="Publish Model", icon="EXPORT")
|
||||
pass
|
||||
|
||||
if wm.ui_mode == "LOAD":
|
||||
# select Version button
|
||||
row = layout.row()
|
||||
version_id = getattr(wm, "selected_version_id", "")
|
||||
load_option = getattr(wm, "selected_version_load_option", "")
|
||||
if load_option == "LATEST":
|
||||
version_button_text = "Latest"
|
||||
elif load_option == "SPECIFIC":
|
||||
version_button_text = version_id
|
||||
else:
|
||||
version_button_text = "Select Version"
|
||||
|
||||
version_button_icon = "CHECKMARK" if version_selected else "PLUS"
|
||||
row.enabled = project_selected and model_selected
|
||||
row.operator(
|
||||
"speckle.version_selection_dialog",
|
||||
text=version_button_text,
|
||||
icon=version_button_icon,
|
||||
).model_card_id = ""
|
||||
|
||||
# load button
|
||||
row = layout.row()
|
||||
row.enabled = project_selected and model_selected and version_selected
|
||||
row.operator("speckle.load", text="Load Model", icon="IMPORT")
|
||||
@@ -0,0 +1,89 @@
|
||||
import bpy
|
||||
from bpy.types import UILayout, Context
|
||||
from .icons import get_icon
|
||||
|
||||
|
||||
class SPECKLE_PT_model_cards_panel(bpy.types.Panel):
|
||||
"""
|
||||
Panel for displaying Speckle model cards.
|
||||
"""
|
||||
|
||||
bl_label = "Model Cards"
|
||||
bl_idname = "SPECKLE_PT_model_cards_panel"
|
||||
bl_space_type = "VIEW_3D"
|
||||
bl_region_type = "UI"
|
||||
bl_category = "Speckle"
|
||||
bl_order = 1
|
||||
|
||||
@classmethod
|
||||
def poll(cls, context: Context) -> bool:
|
||||
"""Only show panel when model cards exist"""
|
||||
return bool(context.scene.speckle_state.model_cards)
|
||||
|
||||
def draw(self, context: Context) -> None:
|
||||
layout: UILayout = self.layout
|
||||
|
||||
# group model cards by project name
|
||||
project_groups = {}
|
||||
for model_card in context.scene.speckle_state.model_cards:
|
||||
project_name = (
|
||||
model_card.project_name if model_card.project_name else "No Project"
|
||||
)
|
||||
if project_name not in project_groups:
|
||||
project_groups[project_name] = []
|
||||
project_groups[project_name].append(model_card)
|
||||
|
||||
for project_name, model_cards in project_groups.items():
|
||||
project_box = layout.box()
|
||||
project_row = project_box.row()
|
||||
project_row.label(text=f"Project: {project_name}", icon="TRIA_RIGHT")
|
||||
|
||||
for model_card in model_cards:
|
||||
box: UILayout = project_box.box()
|
||||
row_1: UILayout = box.row()
|
||||
row_2: UILayout = box.row()
|
||||
|
||||
if model_card.is_publish:
|
||||
# Publish button in the model card
|
||||
row_1.operator(
|
||||
"speckle.model_card_publish", text="", icon="EXPORT"
|
||||
).model_card_id = model_card.get_model_card_id()
|
||||
# Selection filter button in the model card
|
||||
row_2.operator(
|
||||
"speckle.selection_filter_dialog",
|
||||
text=f"Selection: {len(model_card.objects)} objects",
|
||||
).model_card_id = model_card.get_model_card_id()
|
||||
elif not model_card.is_publish:
|
||||
# Load button in the model card
|
||||
row_1.operator(
|
||||
"speckle.model_card_load", text="", icon="IMPORT"
|
||||
).model_card_id = model_card.get_model_card_id()
|
||||
version_button_text = (
|
||||
f"Latest: {model_card.version_id}"
|
||||
if model_card.load_option == "LATEST"
|
||||
else f"{model_card.version_id}"
|
||||
)
|
||||
row_2.operator(
|
||||
"speckle.version_selection_dialog",
|
||||
text=version_button_text,
|
||||
).model_card_id = model_card.get_model_card_id()
|
||||
# TODO: Get last updated time
|
||||
|
||||
else:
|
||||
print({"ERROR"}, "Model card state unknown")
|
||||
return
|
||||
|
||||
row_1.label(text=f"{model_card.model_name}")
|
||||
|
||||
# Select button in the model card
|
||||
select_op = row_1.operator(
|
||||
"speckle.select_objects",
|
||||
text="",
|
||||
icon_value=get_icon("object_highlight"),
|
||||
)
|
||||
select_op.model_card_id = model_card.get_model_card_id()
|
||||
|
||||
# Settings button in the model card
|
||||
row_1.operator(
|
||||
"speckle.model_card_settings", text="", icon="COLLAPSEMENU"
|
||||
).model_card_id = model_card.get_model_card_id()
|
||||
@@ -0,0 +1,127 @@
|
||||
import bpy
|
||||
from bpy.types import UILayout, Context, PropertyGroup, Event
|
||||
from ..utils.model_manager import get_models_for_project
|
||||
from ..utils.version_manager import get_latest_version
|
||||
from ..utils.account_manager import can_create_model
|
||||
from ..blender_operators.create_model import SPECKLE_OT_create_model
|
||||
|
||||
|
||||
class SPECKLE_UL_models_list(bpy.types.UIList):
|
||||
"""
|
||||
UIList for displaying a list of Speckle models
|
||||
"""
|
||||
|
||||
def draw_item(
|
||||
self,
|
||||
context: Context,
|
||||
layout: UILayout,
|
||||
data: PropertyGroup,
|
||||
item: PropertyGroup,
|
||||
icon: str,
|
||||
active_data: PropertyGroup,
|
||||
active_propname: str,
|
||||
) -> None:
|
||||
if self.layout_type in {"DEFAULT", "COMPACT"}:
|
||||
row = layout.row(align=True)
|
||||
split = row.split(factor=0.5)
|
||||
split.label(text=item.name)
|
||||
|
||||
right_split = split.split(factor=0.25)
|
||||
right_split.label(text=item.id)
|
||||
right_split.label(text=item.updated)
|
||||
|
||||
elif self.layout_type == "GRID":
|
||||
layout.alignment = "CENTER"
|
||||
layout.label(text=item.name)
|
||||
|
||||
|
||||
class SPECKLE_OT_model_selection_dialog(bpy.types.Operator):
|
||||
"""
|
||||
operator for displaying and handling the model selection dialog
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.model_selection_dialog"
|
||||
bl_label = "Select Model"
|
||||
bl_description = "Select a model to load"
|
||||
|
||||
def update_models_list(self, context: Context) -> None:
|
||||
wm = context.window_manager
|
||||
|
||||
wm.speckle_models.clear()
|
||||
|
||||
search = self.search_query if self.search_query.strip() else None
|
||||
models = get_models_for_project(
|
||||
wm.selected_account_id, wm.selected_project_id, search=search
|
||||
)
|
||||
|
||||
for name, id, updated in models:
|
||||
model = wm.speckle_models.add()
|
||||
model.name = name
|
||||
model.updated = updated
|
||||
model.id = id
|
||||
|
||||
return None
|
||||
|
||||
search_query: bpy.props.StringProperty( # type: ignore
|
||||
name="Search",
|
||||
description="Search a model",
|
||||
default="",
|
||||
update=update_models_list,
|
||||
)
|
||||
|
||||
model_index: bpy.props.IntProperty(name="Model Index", default=0) # type: ignore
|
||||
|
||||
def execute(self, context: Context) -> set[str]:
|
||||
wm = context.window_manager
|
||||
if 0 <= self.model_index < len(wm.speckle_models):
|
||||
selected_model = wm.speckle_models[self.model_index]
|
||||
|
||||
wm.selected_model_id = selected_model.id
|
||||
wm.selected_model_name = selected_model.name
|
||||
|
||||
latest_version = get_latest_version(
|
||||
account_id=wm.selected_account_id,
|
||||
project_id=wm.selected_project_id,
|
||||
model_id=wm.selected_model_id,
|
||||
)
|
||||
if latest_version:
|
||||
wm.selected_version_load_option = "LATEST"
|
||||
wm.selected_version_id = latest_version[0]
|
||||
|
||||
print(f"Selected model: {selected_model.name} ({selected_model.id})")
|
||||
|
||||
context.area.tag_redraw()
|
||||
return {"FINISHED"}
|
||||
|
||||
def invoke(self, context: Context, event: Event) -> set[str]:
|
||||
self.update_models_list(context)
|
||||
|
||||
wm = context.window_manager
|
||||
authorized, _ = can_create_model(wm.selected_account_id, wm.selected_project_id)
|
||||
self._can_create_model = authorized
|
||||
SPECKLE_OT_create_model._can_create = authorized
|
||||
|
||||
return context.window_manager.invoke_props_dialog(self)
|
||||
|
||||
def draw(self, context: Context) -> None:
|
||||
layout: UILayout = self.layout
|
||||
wm = context.window_manager
|
||||
layout.label(text=f"Project: {wm.selected_project_name}")
|
||||
|
||||
row = layout.row(align=True)
|
||||
row.prop(self, "search_query", icon="VIEWZOOM", text="") # search bar
|
||||
if wm.ui_mode != "LOAD":
|
||||
sub = row.row(align=True)
|
||||
sub.enabled = getattr(self, "_can_create_model", True)
|
||||
sub.operator("speckle.create_model", icon="ADD", text="")
|
||||
|
||||
layout.template_list(
|
||||
"SPECKLE_UL_models_list",
|
||||
"",
|
||||
context.window_manager,
|
||||
"speckle_models",
|
||||
self,
|
||||
"model_index",
|
||||
)
|
||||
|
||||
layout.separator()
|
||||
Binary file not shown.
|
After Width: | Height: | Size: 1.2 KiB |
@@ -0,0 +1,199 @@
|
||||
import bpy
|
||||
from bpy.types import UILayout, Context, PropertyGroup, Event
|
||||
from typing import List, Tuple
|
||||
from ..utils.account_manager import (
|
||||
can_create_project_in_workspace,
|
||||
get_active_workspace,
|
||||
get_default_account_id,
|
||||
get_account_from_id,
|
||||
)
|
||||
from ..utils.project_manager import get_projects_for_account
|
||||
from ..utils.property_groups import speckle_project
|
||||
|
||||
|
||||
class SPECKLE_UL_projects_list(bpy.types.UIList):
|
||||
"""
|
||||
UIList for displaying a list of Speckle projects
|
||||
"""
|
||||
|
||||
def draw_item(
|
||||
self,
|
||||
context: Context,
|
||||
layout: UILayout,
|
||||
data: PropertyGroup,
|
||||
item: PropertyGroup,
|
||||
icon: str,
|
||||
active_data: PropertyGroup,
|
||||
active_propname: str,
|
||||
) -> None:
|
||||
if self.layout_type in {"DEFAULT", "COMPACT"}:
|
||||
row = layout.row(align=True)
|
||||
# enable/disable the row based on permission
|
||||
row.enabled = item.can_receive
|
||||
|
||||
split = row.split(factor=0.5)
|
||||
split.label(text=item.name)
|
||||
|
||||
right_split = split.split(factor=0.5)
|
||||
right_split.label(text=item.role)
|
||||
right_split.label(text=item.updated)
|
||||
|
||||
# handles when the list is in a grid layout
|
||||
elif self.layout_type == "GRID":
|
||||
layout.alignment = "CENTER"
|
||||
layout.enabled = item.can_receive
|
||||
layout.label(text=item.name)
|
||||
|
||||
|
||||
class SPECKLE_OT_project_selection_dialog(bpy.types.Operator):
|
||||
"""
|
||||
operator for displaying and handling the project selection dialog
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.project_selection_dialog"
|
||||
bl_label = "Select Project"
|
||||
bl_description = "Select a project to load models from"
|
||||
|
||||
def update_projects_list(self, context: Context) -> None:
|
||||
"""
|
||||
updates the list of projects based on the selected account and search query
|
||||
"""
|
||||
wm = context.window_manager
|
||||
|
||||
wm.can_create_project_in_workspace = can_create_project_in_workspace(
|
||||
wm.selected_account_id, wm.selected_workspace.id
|
||||
)
|
||||
wm.speckle_projects.clear()
|
||||
|
||||
# get projects for the selected account, using search if provided
|
||||
search = self.search_query if self.search_query.strip() else None
|
||||
projects: List[Tuple[str, str, str, str, bool]] = get_projects_for_account(
|
||||
wm.selected_account_id, search=search, workspace_id=wm.selected_workspace.id
|
||||
)
|
||||
|
||||
for name, role, updated, id, can_receive in projects:
|
||||
project: speckle_project = wm.speckle_projects.add()
|
||||
project.name = name
|
||||
project.role = role
|
||||
project.updated = updated
|
||||
project.id = id
|
||||
project.can_receive = can_receive
|
||||
print("Updated Projects List!")
|
||||
return None
|
||||
|
||||
search_query: bpy.props.StringProperty( # type: ignore
|
||||
name="Search or Paste a URL",
|
||||
description="Search a project or paste a URL to add a project",
|
||||
default="",
|
||||
update=update_projects_list,
|
||||
)
|
||||
|
||||
project_index: bpy.props.IntProperty(name="Project Index", default=0) # type: ignore
|
||||
|
||||
def execute(self, context: Context) -> set[str]:
|
||||
wm = context.window_manager
|
||||
if 0 <= self.project_index < len(wm.speckle_projects):
|
||||
selected_project = wm.speckle_projects[self.project_index]
|
||||
|
||||
# verify the user has permission to receive from this project
|
||||
if not selected_project.can_receive:
|
||||
self.report(
|
||||
{"ERROR"},
|
||||
"Your role on this project doesn't give you permission to load.",
|
||||
)
|
||||
return {"CANCELLED"}
|
||||
|
||||
wm.selected_project_id = selected_project.id
|
||||
wm.selected_project_name = selected_project.name
|
||||
|
||||
print(f"Selected project: {selected_project.name} ({selected_project.id})")
|
||||
|
||||
context.area.tag_redraw()
|
||||
return {"FINISHED"}
|
||||
|
||||
def invoke(self, context: Context, event: Event) -> set[str]:
|
||||
wm = context.window_manager
|
||||
|
||||
# Clear existing projects
|
||||
wm.speckle_projects.clear()
|
||||
|
||||
if wm.selected_account_id == "":
|
||||
wm.selected_account_id = get_default_account_id()
|
||||
|
||||
active_workspace = get_active_workspace(wm.selected_account_id)
|
||||
if active_workspace:
|
||||
wm.selected_workspace.id = active_workspace["id"]
|
||||
wm.selected_workspace.name = active_workspace["name"]
|
||||
else:
|
||||
from .account_selection_dialog import update_workspaces_list
|
||||
|
||||
update_workspaces_list(context)
|
||||
workspaces = list(wm.speckle_workspaces)
|
||||
if workspaces:
|
||||
wm.selected_workspace.id = workspaces[0].id
|
||||
wm.selected_workspace.name = workspaces[0].name
|
||||
|
||||
# Fetch projects from server
|
||||
projects: List[Tuple[str, str, str, str, bool]] = get_projects_for_account(
|
||||
wm.selected_account_id, wm.selected_workspace.id
|
||||
)
|
||||
|
||||
for name, role, updated, id, can_receive in projects:
|
||||
project: speckle_project = wm.speckle_projects.add()
|
||||
project.name = name
|
||||
project.role = role
|
||||
project.updated = updated
|
||||
project.id = id
|
||||
project.can_receive = can_receive
|
||||
|
||||
return context.window_manager.invoke_props_dialog(self)
|
||||
|
||||
def draw(self, context: Context) -> None:
|
||||
layout: UILayout = self.layout
|
||||
wm = context.window_manager
|
||||
|
||||
# Account selection
|
||||
row = layout.row()
|
||||
|
||||
if wm.selected_account_id == "NO_ACCOUNTS":
|
||||
row.operator("speckle.add_account", icon="WORLD", text="Sign In")
|
||||
|
||||
# if no accounts then don't show workspaces or projects list
|
||||
if wm.selected_account_id != "NO_ACCOUNTS":
|
||||
account = get_account_from_id(wm.selected_account_id)
|
||||
|
||||
row.operator(
|
||||
"speckle.account_selection_dialog",
|
||||
icon="USER",
|
||||
text=f"{account.userInfo.name} - {account.userInfo.email} - {account.serverInfo.url}",
|
||||
)
|
||||
# Workspace selection
|
||||
row = layout.row()
|
||||
row.operator(
|
||||
"speckle.workspace_selection_dialog",
|
||||
icon="WORKSPACE",
|
||||
text=wm.selected_workspace.name,
|
||||
)
|
||||
|
||||
# Search field
|
||||
row = layout.row(align=True)
|
||||
row.prop(self, "search_query", icon="VIEWZOOM", text="")
|
||||
# add project by url button
|
||||
split = row.split()
|
||||
split.operator("speckle.add_project_by_url", icon="LINKED", text="")
|
||||
# create project button
|
||||
# hide if in load mode
|
||||
if wm.ui_mode != "LOAD":
|
||||
split = row.split()
|
||||
split.operator("speckle.create_project", icon="ADD", text="")
|
||||
split.enabled = wm.can_create_project_in_workspace
|
||||
|
||||
layout.template_list(
|
||||
"SPECKLE_UL_projects_list",
|
||||
"",
|
||||
context.window_manager,
|
||||
"speckle_projects",
|
||||
self,
|
||||
"project_index",
|
||||
)
|
||||
layout.separator()
|
||||
@@ -0,0 +1,143 @@
|
||||
import bpy
|
||||
from typing import List
|
||||
from bpy.types import Operator, Context, Object
|
||||
from bpy.props import EnumProperty
|
||||
from ..utils.model_card_utils import update_model_card_objects
|
||||
from ..utils.account_manager import can_create_version
|
||||
|
||||
|
||||
class SPECKLE_OT_selection_filter_dialog(Operator):
|
||||
"""
|
||||
operator for handling object selection and filtering
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.selection_filter_dialog"
|
||||
bl_label = "Select Objects"
|
||||
bl_description = "Select objects to publish"
|
||||
|
||||
selection_type: EnumProperty(
|
||||
name="Selection",
|
||||
items=[
|
||||
("SELECTION", "Selection", "Select objects manually"),
|
||||
],
|
||||
default="SELECTION",
|
||||
) # type: ignore
|
||||
|
||||
model_card_id: bpy.props.StringProperty(
|
||||
name="Model Card ID",
|
||||
description="This is used to indicate the function is called from a model card",
|
||||
default="",
|
||||
) # type: ignore
|
||||
|
||||
version_message: bpy.props.StringProperty(
|
||||
name="Version Message",
|
||||
description="Message to be used for the version",
|
||||
default="",
|
||||
) # type: ignore
|
||||
|
||||
def execute(self, context: Context) -> set:
|
||||
wm = context.window_manager
|
||||
wm.speckle_objects.clear()
|
||||
user_selection = context.selected_objects
|
||||
if self.model_card_id != "":
|
||||
model_card = context.scene.speckle_state.get_model_card_by_id(
|
||||
self.model_card_id
|
||||
)
|
||||
update_model_card_objects(model_card, user_selection)
|
||||
self.report({"INFO"}, "Selection updated")
|
||||
|
||||
# On-demand permission check before publishing
|
||||
authorized, auth_message = can_create_version(
|
||||
model_card.account_id, model_card.project_id, model_card.model_id
|
||||
)
|
||||
if not authorized:
|
||||
self.report({"ERROR"}, auth_message)
|
||||
return {"CANCELLED"}
|
||||
|
||||
# Call the publish operator
|
||||
bpy.ops.speckle.model_card_publish(
|
||||
model_card_id=self.model_card_id, version_message=self.version_message
|
||||
)
|
||||
|
||||
context.area.tag_redraw()
|
||||
return {"FINISHED"}
|
||||
|
||||
for sel in user_selection:
|
||||
obj = wm.speckle_objects.add()
|
||||
obj.name = sel.name
|
||||
context.area.tag_redraw()
|
||||
return {"FINISHED"}
|
||||
|
||||
def invoke(self, context: Context, event: bpy.types.Event) -> set:
|
||||
return context.window_manager.invoke_props_dialog(self)
|
||||
|
||||
def draw(self, context: Context):
|
||||
layout = self.layout
|
||||
wm = context.window_manager
|
||||
|
||||
project_name = wm.selected_project_name
|
||||
model_name = wm.selected_model_name
|
||||
if self.model_card_id != "":
|
||||
model_card = context.scene.speckle_state.get_model_card_by_id(
|
||||
self.model_card_id
|
||||
)
|
||||
project_name = model_card.project_name
|
||||
model_name = model_card.model_name
|
||||
|
||||
layout.label(text=f"Project: {project_name}")
|
||||
layout.label(text=f"Model: {model_name}")
|
||||
|
||||
# layout.prop(self, "selection_type")
|
||||
layout.separator()
|
||||
|
||||
selected_objects: List[Object] = context.selected_objects
|
||||
total_selected: int = len(selected_objects)
|
||||
|
||||
box = layout.box()
|
||||
row = box.row()
|
||||
row.label(text="Selection Summary", icon="OUTLINER_OB_GROUP_INSTANCE")
|
||||
row.label(text=f"Total: {total_selected}", icon="OBJECT_DATA")
|
||||
|
||||
object_types: dict[str, int] = {}
|
||||
for obj in selected_objects:
|
||||
if obj.type not in object_types:
|
||||
object_types[obj.type] = 1
|
||||
else:
|
||||
object_types[obj.type] += 1
|
||||
|
||||
col = box.column(align=True)
|
||||
for obj_type, count in object_types.items():
|
||||
row = col.row()
|
||||
row.label(text=f"{obj_type}:", icon=self.get_icon_for_type(obj_type))
|
||||
row.label(text=str(count))
|
||||
|
||||
layout.separator()
|
||||
|
||||
if self.model_card_id != "":
|
||||
layout.label(text="Version Message")
|
||||
layout.prop(self, "version_message", text="")
|
||||
layout.label(
|
||||
text="New version will be published after updating selection",
|
||||
icon="INFO_LARGE",
|
||||
)
|
||||
|
||||
def get_icon_for_type(self, obj_type: str) -> str:
|
||||
icon_map: dict[str, str] = {
|
||||
"MESH": "OUTLINER_OB_MESH",
|
||||
"CURVE": "OUTLINER_OB_CURVE",
|
||||
"SURFACE": "OUTLINER_OB_SURFACE",
|
||||
"META": "OUTLINER_OB_META",
|
||||
"FONT": "OUTLINER_OB_FONT",
|
||||
"ARMATURE": "OUTLINER_OB_ARMATURE",
|
||||
"LATTICE": "OUTLINER_OB_LATTICE",
|
||||
"EMPTY": "OUTLINER_OB_EMPTY",
|
||||
"GPENCIL": "OUTLINER_OB_GREASEPENCIL",
|
||||
"CAMERA": "OUTLINER_OB_CAMERA",
|
||||
"LIGHT": "OUTLINER_OB_LIGHT",
|
||||
"SPEAKER": "OUTLINER_OB_SPEAKER",
|
||||
"LIGHT_PROBE": "OUTLINER_OB_LIGHTPROBE",
|
||||
}
|
||||
return icon_map.get(obj_type, "OBJECT_DATA")
|
||||
|
||||
def check(self, context: Context) -> bool:
|
||||
return True # this forces the dialog to redraw
|
||||
Binary file not shown.
|
After Width: | Height: | Size: 446 B |
@@ -0,0 +1,48 @@
|
||||
import bpy
|
||||
from bpy.types import UILayout, Context
|
||||
|
||||
|
||||
class SPECKLE_PT_update_panel(bpy.types.Panel):
|
||||
"""Panel for displaying connector update notifications"""
|
||||
|
||||
bl_label = "Update Speckle"
|
||||
bl_idname = "SPECKLE_PT_update_panel"
|
||||
bl_space_type = "VIEW_3D"
|
||||
bl_region_type = "UI"
|
||||
bl_category = "Speckle"
|
||||
bl_order = 0 # This ensures it appears above the main panel
|
||||
|
||||
@classmethod
|
||||
def poll(cls, context: Context) -> bool:
|
||||
"""Only show this panel when an update is available"""
|
||||
wm = context.window_manager
|
||||
return getattr(wm, "update_available", False)
|
||||
|
||||
def draw(self, context: Context) -> None:
|
||||
layout: UILayout = self.layout
|
||||
wm = context.window_manager
|
||||
|
||||
# Get current version from bl_info
|
||||
from ... import bl_info
|
||||
|
||||
current_version = bl_info["version"]
|
||||
current_version_str = (
|
||||
f"{current_version[0]}.{current_version[1]}.{current_version[2]}"
|
||||
)
|
||||
|
||||
# Update notification
|
||||
box = layout.box()
|
||||
box.alert = True # Makes the box stand out with alert styling
|
||||
|
||||
col = box.column()
|
||||
col.label(text="New version available!", icon="INFO")
|
||||
|
||||
row = col.row()
|
||||
row.label(text=f"Current: v{current_version_str}")
|
||||
|
||||
row = col.row()
|
||||
row.label(text=f"Latest: v{wm.latest_version}")
|
||||
|
||||
# Update button
|
||||
row = col.row()
|
||||
row.operator("speckle.update_button", text="Download Update", icon="LINKED")
|
||||
@@ -0,0 +1,185 @@
|
||||
import bpy
|
||||
from bpy.types import UILayout, Context, PropertyGroup, Event
|
||||
from ..utils.version_manager import get_versions_for_model, get_latest_version
|
||||
|
||||
|
||||
class SPECKLE_UL_versions_list(bpy.types.UIList):
|
||||
"""
|
||||
UIList for displaying a list of Speckle versions
|
||||
"""
|
||||
|
||||
# TODO: Adjust column widths so message has the most space.
|
||||
def draw_item(
|
||||
self,
|
||||
context: Context,
|
||||
layout: UILayout,
|
||||
data: PropertyGroup,
|
||||
item: PropertyGroup,
|
||||
icon: str,
|
||||
active_data: PropertyGroup,
|
||||
active_propname: str,
|
||||
) -> None:
|
||||
if self.layout_type in {"DEFAULT", "COMPACT"}:
|
||||
row = layout.row(align=True)
|
||||
split = row.split(factor=0.166)
|
||||
split.label(text=item.id)
|
||||
right_split = split.split(factor=0.7)
|
||||
right_split.label(text=item.message)
|
||||
right_split.label(text=item.updated)
|
||||
|
||||
elif self.layout_type == "GRID":
|
||||
layout.alignment = "CENTER"
|
||||
layout.label(text=item.id)
|
||||
|
||||
|
||||
class SPECKLE_OT_version_selection_dialog(bpy.types.Operator):
|
||||
bl_idname = "speckle.version_selection_dialog"
|
||||
bl_label = "Select Version"
|
||||
bl_description = "Select a model version to load. Default is the latest version. You can also select a specific version."
|
||||
|
||||
version_index: bpy.props.IntProperty(name="Model Index", default=0) # type: ignore
|
||||
|
||||
load_option: bpy.props.EnumProperty( # type: ignore
|
||||
name="Load Option",
|
||||
description="Choose how to load the version",
|
||||
items=[
|
||||
("LATEST", "Load latest version", "Load the latest version available"),
|
||||
(
|
||||
"SPECIFIC",
|
||||
"Load a specific version",
|
||||
"Load a specific version from the list",
|
||||
),
|
||||
],
|
||||
default="LATEST",
|
||||
)
|
||||
|
||||
model_card_id: bpy.props.StringProperty(
|
||||
name="Model Card ID",
|
||||
description="This is used to indicate the function is called from a model card",
|
||||
default="",
|
||||
) # type: ignore
|
||||
|
||||
def update_versions_list(self, context: Context) -> None:
|
||||
wm = context.window_manager
|
||||
wm.speckle_versions.clear()
|
||||
|
||||
versions = get_versions_for_model(
|
||||
account_id=wm.selected_account_id,
|
||||
project_id=wm.selected_project_id,
|
||||
model_id=wm.selected_model_id,
|
||||
)
|
||||
|
||||
for id, message, updated in versions:
|
||||
version = wm.speckle_versions.add()
|
||||
version.id = id
|
||||
version.message = message
|
||||
version.updated = updated
|
||||
|
||||
return None
|
||||
|
||||
def execute(self, context: Context) -> set[str]:
|
||||
wm = context.window_manager
|
||||
|
||||
version_id_to_store = ""
|
||||
|
||||
if self.load_option == "LATEST":
|
||||
latest_version = get_latest_version(
|
||||
account_id=wm.selected_account_id,
|
||||
project_id=wm.selected_project_id,
|
||||
model_id=wm.selected_model_id,
|
||||
)
|
||||
if latest_version:
|
||||
version_id_to_store = latest_version[0]
|
||||
else:
|
||||
print(
|
||||
f"Could not fetch latest version for model {wm.selected_model_id}"
|
||||
)
|
||||
return {"CANCELLED"}
|
||||
|
||||
elif self.load_option == "SPECIFIC":
|
||||
if 0 <= self.version_index < len(wm.speckle_versions):
|
||||
selected_version = wm.speckle_versions[self.version_index]
|
||||
version_id_to_store = selected_version.id
|
||||
else:
|
||||
print(f"Invalid version index {self.version_index}")
|
||||
return {"CANCELLED"}
|
||||
wm.selected_version_id = version_id_to_store
|
||||
|
||||
if self.model_card_id != "":
|
||||
model_card = context.scene.speckle_state.get_model_card_by_id(
|
||||
self.model_card_id
|
||||
)
|
||||
if model_card is None:
|
||||
self.report({"ERROR"}, f"Model card '{self.model_card_id}' not found")
|
||||
return {"CANCELLED"}
|
||||
|
||||
model_card.load_option = self.load_option
|
||||
model_card.version_id = version_id_to_store
|
||||
self.report(
|
||||
{"INFO"},
|
||||
f"Model card updated: Selected version: {model_card.version_id}, Option: {self.load_option}",
|
||||
)
|
||||
bpy.ops.speckle.model_card_load(model_card_id=self.model_card_id)
|
||||
context.area.tag_redraw()
|
||||
|
||||
return {"FINISHED"}
|
||||
|
||||
wm.selected_version_load_option = self.load_option
|
||||
self.report(
|
||||
{"INFO"},
|
||||
f"Selected version: {version_id_to_store} (Option: {self.load_option})",
|
||||
)
|
||||
|
||||
context.area.tag_redraw()
|
||||
|
||||
return {"FINISHED"}
|
||||
|
||||
def invoke(self, context: Context, event: Event) -> set[str]:
|
||||
if self.model_card_id != "":
|
||||
wm = context.window_manager
|
||||
model_card = context.scene.speckle_state.get_model_card_by_id(
|
||||
self.model_card_id
|
||||
)
|
||||
self.load_option = model_card.load_option
|
||||
wm.selected_account_id = model_card.account_id
|
||||
wm.selected_project_id = model_card.project_id
|
||||
wm.selected_model_id = model_card.model_id
|
||||
wm.selected_version_id = model_card.version_id
|
||||
|
||||
self.update_versions_list(context)
|
||||
|
||||
return context.window_manager.invoke_props_dialog(self)
|
||||
|
||||
def draw(self, context: Context) -> None:
|
||||
layout: UILayout = self.layout
|
||||
wm = context.window_manager
|
||||
project_name = wm.selected_project_name
|
||||
model_name = wm.selected_model_name
|
||||
if self.model_card_id != "":
|
||||
model_card = context.scene.speckle_state.get_model_card_by_id(
|
||||
self.model_card_id
|
||||
)
|
||||
project_name = model_card.project_name
|
||||
model_name = model_card.model_name
|
||||
|
||||
layout.label(text=f"Project: {project_name}")
|
||||
layout.label(text=f"Model: {model_name}")
|
||||
|
||||
layout.prop(
|
||||
self,
|
||||
"load_option",
|
||||
expand=True,
|
||||
)
|
||||
|
||||
if self.load_option == "SPECIFIC":
|
||||
# Versions UIList
|
||||
layout.template_list(
|
||||
"SPECKLE_UL_versions_list",
|
||||
"",
|
||||
context.window_manager,
|
||||
"speckle_versions",
|
||||
self,
|
||||
"version_index",
|
||||
)
|
||||
|
||||
layout.separator()
|
||||
@@ -0,0 +1,108 @@
|
||||
import bpy
|
||||
from bpy.types import Context, UILayout, Event, PropertyGroup
|
||||
from typing import List, Tuple
|
||||
from ..utils.account_manager import get_workspaces, speckle_workspace
|
||||
from ..utils.project_manager import get_projects_for_account
|
||||
from ..utils.account_manager import can_create_project_in_workspace
|
||||
|
||||
|
||||
class SPECKLE_UL_workspaces_list(bpy.types.UIList):
|
||||
"""
|
||||
UIList for workspaces
|
||||
"""
|
||||
|
||||
def draw_item(
|
||||
self,
|
||||
context: Context,
|
||||
layout: UILayout,
|
||||
data: PropertyGroup,
|
||||
item: PropertyGroup,
|
||||
icon: str,
|
||||
active_data: PropertyGroup,
|
||||
active_propname: str,
|
||||
) -> None:
|
||||
if self.layout_type in {"DEFAULT", "COMPACT"}:
|
||||
row = layout.row(align=True)
|
||||
row.label(text=item.name)
|
||||
|
||||
elif self.layout_type == "GRID":
|
||||
layout.alignment = "CENTER"
|
||||
layout.label(text=item.name)
|
||||
|
||||
|
||||
class SPECKLE_OT_workspace_selection_dialog(bpy.types.Operator):
|
||||
"""
|
||||
Operator for selecting a workspace
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.workspace_selection_dialog"
|
||||
bl_label = "Select Workspace"
|
||||
bl_description = "Select a workspace to load projects from"
|
||||
|
||||
workspace_index: bpy.props.IntProperty(name="Workspace Index", default=0) # type: ignore
|
||||
|
||||
def invoke(self, context: Context, event: Event) -> set[str]:
|
||||
wm = context.window_manager
|
||||
wm.speckle_workspaces.clear()
|
||||
workspaces: List[Tuple[str, str]] = get_workspaces(wm.selected_account_id)
|
||||
current_workspace_index = 0
|
||||
for i, (id, name) in enumerate(workspaces):
|
||||
workspace: speckle_workspace = wm.speckle_workspaces.add()
|
||||
workspace.id = id
|
||||
workspace.name = name
|
||||
if id == wm.selected_workspace.id:
|
||||
current_workspace_index = i
|
||||
self.workspace_index = current_workspace_index
|
||||
return context.window_manager.invoke_props_dialog(self)
|
||||
|
||||
def draw(self, context: Context) -> None:
|
||||
layout: UILayout = self.layout
|
||||
wm = context.window_manager
|
||||
layout.label(text=f"Selected Workspace: {wm.selected_workspace.name}")
|
||||
layout.template_list(
|
||||
"SPECKLE_UL_workspaces_list",
|
||||
"",
|
||||
context.window_manager,
|
||||
"speckle_workspaces",
|
||||
self,
|
||||
"workspace_index",
|
||||
)
|
||||
|
||||
def execute(self, context: Context) -> set[str]:
|
||||
wm = context.window_manager
|
||||
if 0 <= self.workspace_index < len(wm.speckle_workspaces):
|
||||
selected_workspace = wm.speckle_workspaces[self.workspace_index]
|
||||
wm.selected_workspace.id = selected_workspace.id
|
||||
wm.selected_workspace.name = selected_workspace.name
|
||||
update_projects_list(context)
|
||||
context.area.tag_redraw()
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
def update_projects_list(context):
|
||||
"""Update projects list when workspace changes"""
|
||||
|
||||
wm = context.window_manager
|
||||
wm.speckle_projects.clear()
|
||||
|
||||
# get projects for the selected account and workspace
|
||||
projects = get_projects_for_account(
|
||||
wm.selected_account_id, wm.selected_workspace.id
|
||||
)
|
||||
|
||||
for name, role, updated, id, can_receive in projects:
|
||||
project = wm.speckle_projects.add()
|
||||
project.name = name
|
||||
project.role = role
|
||||
project.updated = updated
|
||||
project.id = id
|
||||
project.can_receive = can_receive
|
||||
|
||||
# Update can_create_project_in_workspace flag
|
||||
wm.can_create_project_in_workspace = can_create_project_in_workspace(
|
||||
wm.selected_account_id, wm.selected_workspace.id
|
||||
)
|
||||
print(f"Workspace changed to: {wm.selected_workspace.id}")
|
||||
print("Projects list updated")
|
||||
|
||||
context.area.tag_redraw()
|
||||
@@ -0,0 +1,325 @@
|
||||
import bpy
|
||||
from specklepy.core.api.credentials import get_local_accounts
|
||||
from typing import List, Tuple, Optional, Dict
|
||||
from urllib.parse import urlparse
|
||||
from specklepy.core.api.credentials import Account
|
||||
from specklepy.core.api.client import SpeckleClient
|
||||
from specklepy.core.api.wrapper import StreamWrapper
|
||||
from .misc import strip_non_ascii
|
||||
|
||||
|
||||
class SpeckleClientCache:
|
||||
def __init__(self):
|
||||
self._clients: Dict[str, SpeckleClient] = {}
|
||||
|
||||
def get_client(self, account_id: str) -> SpeckleClient:
|
||||
# Check cache first
|
||||
if account_id in self._clients:
|
||||
print(f"[Cache HIT] Using cached client for account {account_id}")
|
||||
return self._clients[account_id]
|
||||
|
||||
# Create new client if needed
|
||||
print(f"[Cache MISS] Creating new client for account {account_id}")
|
||||
account = get_account_from_id(account_id)
|
||||
if not account:
|
||||
raise ValueError(f"No account found for ID: {account_id}")
|
||||
|
||||
url = account.serverInfo.url
|
||||
use_ssl = urlparse(url).scheme.lower() != "http"
|
||||
client = SpeckleClient(host=url, use_ssl=use_ssl)
|
||||
client.authenticate_with_account(account)
|
||||
self._clients[account_id] = client
|
||||
return client
|
||||
|
||||
def clear(self) -> None:
|
||||
"""Clear all cached clients."""
|
||||
print("[Cache] Clearing all cached clients")
|
||||
self._clients.clear()
|
||||
|
||||
|
||||
# Global cache instance
|
||||
_client_cache = SpeckleClientCache()
|
||||
|
||||
|
||||
class speckle_account(bpy.types.PropertyGroup):
|
||||
id: bpy.props.StringProperty() # type: ignore
|
||||
user_name: bpy.props.StringProperty() # type: ignore
|
||||
server_url: bpy.props.StringProperty() # type: ignore
|
||||
user_email: bpy.props.StringProperty() # type: ignore
|
||||
|
||||
|
||||
class speckle_workspace(bpy.types.PropertyGroup):
|
||||
"""
|
||||
PropertyGroup for storing workspace information
|
||||
"""
|
||||
|
||||
id: bpy.props.StringProperty(name="ID") # type: ignore
|
||||
name: bpy.props.StringProperty() # type: ignore
|
||||
|
||||
|
||||
def get_account_enum_items() -> List[Tuple[str, str, str, str]]:
|
||||
accounts: List[Account] = get_local_accounts()
|
||||
if not accounts:
|
||||
print("No accounts found!")
|
||||
return [("NO_ACCOUNTS", "No accounts found!", "", "")]
|
||||
print("Accounts added")
|
||||
speckle_accounts = []
|
||||
for acc in accounts:
|
||||
speckle_accounts.append(
|
||||
(
|
||||
acc.id,
|
||||
strip_non_ascii(acc.userInfo.name),
|
||||
acc.serverInfo.url,
|
||||
acc.userInfo.email,
|
||||
)
|
||||
)
|
||||
return speckle_accounts
|
||||
|
||||
|
||||
def get_workspaces(account_id: str) -> List[Tuple[str, str]]:
|
||||
"""
|
||||
retrieves the workspaces for a given account ID
|
||||
"""
|
||||
|
||||
try:
|
||||
# Get client from cache
|
||||
client = _client_cache.get_client(account_id)
|
||||
|
||||
workspaces_enabled = client.server.get().workspaces.workspaces_enabled
|
||||
|
||||
if workspaces_enabled:
|
||||
workspaces = client.active_user.get_workspaces().items
|
||||
|
||||
workspace_list = [
|
||||
(ws.id, strip_non_ascii(ws.name))
|
||||
for ws in workspaces
|
||||
if ws.creation_state is None or ws.creation_state.completed
|
||||
]
|
||||
|
||||
active_workspace = client.active_user.get_active_workspace()
|
||||
default_workspace_id = (
|
||||
active_workspace.id
|
||||
if active_workspace
|
||||
else (workspaces[0].id if workspaces else None)
|
||||
)
|
||||
|
||||
if default_workspace_id:
|
||||
result = reorder_tuple(workspace_list, default_workspace_id)
|
||||
else:
|
||||
result = workspace_list
|
||||
else:
|
||||
result = []
|
||||
|
||||
return result
|
||||
except Exception as e:
|
||||
print(f"Error in get_workspaces: {str(e)}")
|
||||
_client_cache.clear() # Clear cache on error
|
||||
return [("", "")]
|
||||
|
||||
|
||||
def get_default_account_id() -> Optional[str]:
|
||||
"""
|
||||
retrieves the ID of the default Speckle account
|
||||
"""
|
||||
return next(
|
||||
(acc.id for acc in get_local_accounts() if acc.isDefault), "NO_ACCOUNTS"
|
||||
)
|
||||
|
||||
|
||||
def get_server_url_by_account_id(account_id: str) -> Optional[str]:
|
||||
"""
|
||||
retrieves the server URL for a given account ID
|
||||
"""
|
||||
accounts: List[Account] = get_local_accounts()
|
||||
for acc in accounts:
|
||||
if acc.id == account_id:
|
||||
return acc.serverInfo.url
|
||||
return None
|
||||
|
||||
|
||||
def get_active_workspace(account_id: str) -> Optional[Dict[str, str]]:
|
||||
"""
|
||||
retrieves the ID of the default workspace for a given account ID
|
||||
"""
|
||||
try:
|
||||
client = _client_cache.get_client(account_id)
|
||||
active_workspace = client.active_user.get_active_workspace()
|
||||
if active_workspace:
|
||||
return {"id": active_workspace.id, "name": active_workspace.name}
|
||||
return None
|
||||
except Exception as e:
|
||||
print(f"Error in get_active_workspace: {str(e)}")
|
||||
_client_cache.clear()
|
||||
return None
|
||||
|
||||
|
||||
def get_account_from_id(account_id: str) -> Optional[Account]:
|
||||
return next((acc for acc in get_local_accounts() if acc.id == account_id), None)
|
||||
|
||||
|
||||
def reorder_tuple(tuple_list, target_id):
|
||||
for i, (id, value) in enumerate(tuple_list):
|
||||
if id == target_id:
|
||||
# Remove the tuple from its current position
|
||||
target_tuple = tuple_list.pop(i)
|
||||
# Insert it at the beginning of the list
|
||||
tuple_list.insert(0, target_tuple)
|
||||
return tuple_list
|
||||
|
||||
# If the target_id wasn't found
|
||||
print(f"Tuple with ID {target_id} not found in the list")
|
||||
return tuple_list
|
||||
|
||||
|
||||
def get_project_from_url(
|
||||
url: str,
|
||||
) -> Tuple[Optional[StreamWrapper], Optional[object], Optional[object], str]:
|
||||
"""
|
||||
get a project from a URL, handling all the client setup.
|
||||
"""
|
||||
try:
|
||||
wrapper = StreamWrapper(url)
|
||||
account = wrapper.get_account()
|
||||
client = _client_cache.get_client(account.id)
|
||||
|
||||
# get the stream_id (project_id) from the wrapper
|
||||
if not wrapper.stream_id:
|
||||
return wrapper, client, None, "Could not extract project ID from URL"
|
||||
|
||||
project = client.project.get(wrapper.stream_id)
|
||||
|
||||
if not project:
|
||||
return wrapper, client, None, "Could not access project"
|
||||
|
||||
return wrapper, client, project, ""
|
||||
|
||||
except Exception as e:
|
||||
return None, None, None, f"Failed to process URL: {str(e)}"
|
||||
|
||||
|
||||
def get_model_details_by_wrapper(
|
||||
wrapper: StreamWrapper,
|
||||
) -> Tuple[str, str, str, str, str, str, str]:
|
||||
"""
|
||||
extract model details from a StreamWrapper object.
|
||||
"""
|
||||
client = wrapper.get_client()
|
||||
client.authenticate_with_account(wrapper.get_account())
|
||||
(
|
||||
account_id,
|
||||
project_id,
|
||||
project_name,
|
||||
model_id,
|
||||
model_name,
|
||||
version_id,
|
||||
load_option,
|
||||
) = "", "", "", "", "", "", ""
|
||||
account_id = wrapper.get_account().id
|
||||
if wrapper.stream_id:
|
||||
project_id = wrapper.stream_id
|
||||
project_name = client.project.get(project_id).name
|
||||
if wrapper.model_id:
|
||||
model_id = wrapper.model_id
|
||||
model = client.model.get(model_id, project_id)
|
||||
model_name = model.name
|
||||
load_option = "LATEST" if not wrapper.commit_id else "SPECIFIC"
|
||||
if wrapper.commit_id:
|
||||
version_id = wrapper.commit_id
|
||||
else:
|
||||
versions = client.version.get_versions(
|
||||
wrapper.model_id, wrapper.stream_id, limit=1
|
||||
)
|
||||
if versions.items and len(versions.items) > 0:
|
||||
version_id = versions.items[0].id
|
||||
else:
|
||||
version_id = ""
|
||||
return (
|
||||
account_id,
|
||||
project_id,
|
||||
project_name,
|
||||
model_id,
|
||||
model_name,
|
||||
version_id,
|
||||
load_option,
|
||||
)
|
||||
|
||||
|
||||
def can_load(client, project) -> Tuple[bool, str]:
|
||||
try:
|
||||
permissions = client.project.get_permissions(project.id)
|
||||
|
||||
if permissions.can_load.authorized:
|
||||
return True, ""
|
||||
else:
|
||||
return (
|
||||
False,
|
||||
"Your role on this project doesn't give you permission to load.",
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
error_msg = f"Failed to check permissions: {str(e)}"
|
||||
print(error_msg)
|
||||
return False, error_msg
|
||||
|
||||
|
||||
def can_create_version(
|
||||
account_id: str, project_id: str, model_id: str
|
||||
) -> Tuple[bool, str]:
|
||||
try:
|
||||
client = _client_cache.get_client(account_id)
|
||||
permissions = client.model.get_permissions(project_id, model_id)
|
||||
|
||||
if permissions.can_create_version.authorized:
|
||||
return True, ""
|
||||
else:
|
||||
message = getattr(permissions.can_create_version, "message", None)
|
||||
return (
|
||||
False,
|
||||
message
|
||||
or "Your role on this project doesn't give you permission to publish.",
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
error_msg = f"Failed to check permissions: {str(e)}"
|
||||
print(error_msg)
|
||||
return False, error_msg
|
||||
|
||||
|
||||
def can_create_model(account_id: str, project_id: str) -> Tuple[bool, str]:
|
||||
try:
|
||||
client = _client_cache.get_client(account_id)
|
||||
permissions = client.project.get_permissions(project_id)
|
||||
|
||||
if permissions.can_create_model.authorized:
|
||||
return True, ""
|
||||
else:
|
||||
message = getattr(permissions.can_create_model, "message", None)
|
||||
return (
|
||||
False,
|
||||
message
|
||||
or "You don't have permission to create models in this project.",
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
error_msg = f"Failed to check permissions: {str(e)}"
|
||||
print(error_msg)
|
||||
return False, error_msg
|
||||
|
||||
|
||||
def can_create_project_in_workspace(account_id: str, workspace_id: str) -> bool:
|
||||
"""
|
||||
Check if the user can create a project in the specified workspace.
|
||||
"""
|
||||
try:
|
||||
client = _client_cache.get_client(account_id)
|
||||
|
||||
try:
|
||||
workspace = client.workspace.get(workspace_id)
|
||||
return workspace.permissions.can_create_project.authorized
|
||||
except Exception as e:
|
||||
print(f"Failed to get workspace: {str(e)}")
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f"Error in can_create_project_in_workspace: {str(e)}")
|
||||
_client_cache.clear() # Clear cache on error
|
||||
return False
|
||||
@@ -0,0 +1,575 @@
|
||||
"""
|
||||
Speckle authentication module for Blender connector.
|
||||
|
||||
Implements OAuth-style authentication flow with a local HTTP server,
|
||||
eliminating the dependency on the desktop service.
|
||||
"""
|
||||
|
||||
import errno
|
||||
import json
|
||||
import secrets
|
||||
import string
|
||||
import sys
|
||||
import threading
|
||||
import webbrowser
|
||||
from http.server import BaseHTTPRequestHandler, HTTPServer
|
||||
from typing import Optional, Dict, Any, Tuple
|
||||
from urllib.parse import urlparse, parse_qs
|
||||
from urllib.request import Request, urlopen
|
||||
from urllib.error import URLError, HTTPError
|
||||
|
||||
|
||||
# Speckle Blender dedicated app constants (registered server-side)
|
||||
SPECKLE_APP_ID = "sblndrdui" # Dedicated app ID for Blender connector
|
||||
SPECKLE_AUTH_PORT = 29365 # Port for local auth callback server
|
||||
|
||||
|
||||
def get_user_agent() -> str:
|
||||
"""Get User-Agent string identifying the Blender connector to prevent Cloudflare blocking."""
|
||||
try:
|
||||
from pathlib import Path
|
||||
|
||||
# Get the extension directory
|
||||
addon_dir = Path(__file__).parent.parent.parent
|
||||
|
||||
# Try to read version from blender_manifest.toml
|
||||
manifest_path = addon_dir / "blender_manifest.toml"
|
||||
if manifest_path.exists():
|
||||
with open(manifest_path, "r") as f:
|
||||
for line in f:
|
||||
if line.startswith("version = "):
|
||||
version = line.split("=")[1].strip().strip('"')
|
||||
break
|
||||
else:
|
||||
version = "3.0.0"
|
||||
else:
|
||||
version = "3.0.0"
|
||||
except Exception:
|
||||
# Fallback if we can't determine version
|
||||
version = "3.0.0"
|
||||
|
||||
python_version = f"{sys.version_info.major}.{sys.version_info.minor}"
|
||||
return f"Speckle-Blender-Connector/{version} (Python/{python_version})"
|
||||
|
||||
|
||||
class AuthenticationError(Exception):
|
||||
"""Raised when authentication fails."""
|
||||
|
||||
pass
|
||||
|
||||
|
||||
def generate_challenge() -> str:
|
||||
"""Generate a random 12-character alphanumeric challenge string."""
|
||||
chars = string.ascii_letters + string.digits
|
||||
return "".join(secrets.choice(chars) for _ in range(12))
|
||||
|
||||
|
||||
class ThreadSafeAuthServer(HTTPServer):
|
||||
"""Thread-safe HTTP server for Speckle authentication with locked state management."""
|
||||
|
||||
def __init__(self, *args, **kwargs):
|
||||
super().__init__(*args, **kwargs)
|
||||
self._lock = threading.Lock()
|
||||
self._server_url: Optional[str] = None
|
||||
self._challenge: Optional[str] = None
|
||||
self._auth_complete: bool = False
|
||||
self._auth_success: bool = False
|
||||
self._error_message: Optional[str] = None
|
||||
self._request_count: int = 0
|
||||
|
||||
@property
|
||||
def server_url(self) -> Optional[str]:
|
||||
with self._lock:
|
||||
return self._server_url
|
||||
|
||||
@server_url.setter
|
||||
def server_url(self, value: str) -> None:
|
||||
with self._lock:
|
||||
self._server_url = value
|
||||
|
||||
@property
|
||||
def challenge(self) -> Optional[str]:
|
||||
with self._lock:
|
||||
return self._challenge
|
||||
|
||||
@challenge.setter
|
||||
def challenge(self, value: str) -> None:
|
||||
with self._lock:
|
||||
self._challenge = value
|
||||
|
||||
@property
|
||||
def is_complete(self) -> bool:
|
||||
with self._lock:
|
||||
return self._auth_complete
|
||||
|
||||
@property
|
||||
def is_successful(self) -> bool:
|
||||
with self._lock:
|
||||
return self._auth_success
|
||||
|
||||
@property
|
||||
def error_message(self) -> Optional[str]:
|
||||
with self._lock:
|
||||
return self._error_message
|
||||
|
||||
def set_auth_success(self) -> None:
|
||||
"""Mark authentication as successful (sets auth_complete LAST for atomicity)."""
|
||||
with self._lock:
|
||||
self._auth_success = True
|
||||
self._error_message = None
|
||||
self._auth_complete = True # Set LAST to prevent partial reads
|
||||
|
||||
def set_auth_failure(self, error_message: str) -> None:
|
||||
"""Mark authentication as failed (sets auth_complete LAST for atomicity)."""
|
||||
with self._lock:
|
||||
self._auth_success = False
|
||||
self._error_message = error_message
|
||||
self._auth_complete = True # Set LAST to prevent partial reads
|
||||
|
||||
def increment_request_count(self) -> int:
|
||||
"""Increment and return request count."""
|
||||
with self._lock:
|
||||
self._request_count += 1
|
||||
return self._request_count
|
||||
|
||||
@property
|
||||
def request_count(self) -> int:
|
||||
with self._lock:
|
||||
return self._request_count
|
||||
|
||||
|
||||
class SpeckleAuthHandler(BaseHTTPRequestHandler):
|
||||
"""HTTP request handler for Speckle authentication flow with /auth/add-account and callback routes."""
|
||||
|
||||
def log_message(self, format, *args):
|
||||
print(f"[Auth Server] {format % args}")
|
||||
|
||||
def do_GET(self):
|
||||
self.server.increment_request_count()
|
||||
|
||||
parsed_path = urlparse(self.path)
|
||||
query_params = parse_qs(parsed_path.query)
|
||||
|
||||
if parsed_path.path == "/auth/add-account":
|
||||
self._handle_add_account(query_params)
|
||||
elif parsed_path.path == "/":
|
||||
self._handle_callback(query_params)
|
||||
else:
|
||||
self._send_error_response(404, "Not Found")
|
||||
|
||||
def _handle_add_account(self, query_params: Dict[str, list]):
|
||||
"""Handle initial add-account request, generate challenge and redirect to Speckle server."""
|
||||
# Get server URL from query params
|
||||
server_url = query_params.get("serverUrl", ["https://app.speckle.systems"])[0]
|
||||
self.server.server_url = server_url.rstrip("/")
|
||||
|
||||
# Generate challenge
|
||||
self.server.challenge = generate_challenge()
|
||||
|
||||
# Construct redirect URL
|
||||
auth_url = f"{self.server.server_url}/authn/verify/{SPECKLE_APP_ID}/{self.server.challenge}"
|
||||
|
||||
print(f"[Auth Server] Redirecting to: {auth_url}")
|
||||
|
||||
# Send redirect response
|
||||
self.send_response(302)
|
||||
self.send_header("Location", auth_url)
|
||||
self.end_headers()
|
||||
|
||||
def _handle_callback(self, query_params: Dict[str, list]):
|
||||
"""Handle callback from Speckle server, exchange access code for tokens and save account."""
|
||||
# Get access code from query params
|
||||
access_code_list = query_params.get("access_code", [])
|
||||
|
||||
if not access_code_list:
|
||||
self._redirect_to_failure("fail-no-access-code")
|
||||
return
|
||||
|
||||
access_code = access_code_list[0]
|
||||
|
||||
try:
|
||||
# Exchange access code for tokens
|
||||
tokens = exchange_access_code_for_tokens(
|
||||
access_code, self.server.challenge, self.server.server_url
|
||||
)
|
||||
|
||||
# Get user and server info
|
||||
user_info, server_info = get_user_and_server_info(
|
||||
tokens["token"], self.server.server_url
|
||||
)
|
||||
|
||||
# Save account
|
||||
save_account_to_storage(
|
||||
tokens["token"], tokens["refreshToken"], user_info, server_info
|
||||
)
|
||||
|
||||
# Mark as successful (sets auth_complete LAST atomically)
|
||||
self.server.set_auth_success()
|
||||
|
||||
# Redirect to success page
|
||||
self._redirect_to_success()
|
||||
|
||||
except Exception as e:
|
||||
print(f"[Auth Server] Error during authentication: {e}")
|
||||
# Mark as failed (sets auth_complete LAST atomically)
|
||||
self.server.set_auth_failure(str(e))
|
||||
self._redirect_to_failure("fail")
|
||||
|
||||
def _redirect_to_success(self):
|
||||
self.send_response(302)
|
||||
self.send_header(
|
||||
"Location", "https://www.speckle.systems/connector-auth/success"
|
||||
)
|
||||
self.end_headers()
|
||||
|
||||
def _redirect_to_failure(self, reason: str):
|
||||
self.send_response(302)
|
||||
self.send_header(
|
||||
"Location", f"https://www.speckle.systems/connector-auth/{reason}"
|
||||
)
|
||||
self.end_headers()
|
||||
|
||||
def _send_error_response(self, code: int, message: str):
|
||||
self.send_response(code)
|
||||
self.send_header("Content-type", "text/html")
|
||||
self.end_headers()
|
||||
self.wfile.write(
|
||||
f"<html><body><h1>{code} {message}</h1></body></html>".encode()
|
||||
)
|
||||
|
||||
|
||||
def _handle_auth_error(e: AuthenticationError) -> None:
|
||||
"""Re-raise AuthenticationError with user-friendly message for network errors."""
|
||||
error_str = str(e)
|
||||
if "Network error" in error_str or "URLError" in error_str:
|
||||
raise AuthenticationError(
|
||||
"Network error while authenticating. Please check your internet connection."
|
||||
) from e
|
||||
raise
|
||||
|
||||
|
||||
def _post_json(
|
||||
url: str,
|
||||
body: Dict[str, Any],
|
||||
auth_token: Optional[str] = None,
|
||||
error_context: str = "Request",
|
||||
) -> Dict[str, Any]:
|
||||
"""Make POST request with JSON body and optional Bearer token."""
|
||||
# Encode body as JSON
|
||||
data = json.dumps(body).encode("utf-8")
|
||||
|
||||
# Build headers
|
||||
headers = {"Content-Type": "application/json", "User-Agent": get_user_agent()}
|
||||
|
||||
# Add Authorization header if token provided
|
||||
if auth_token:
|
||||
headers["Authorization"] = f"Bearer {auth_token}"
|
||||
|
||||
try:
|
||||
request = Request(url, data=data, headers=headers)
|
||||
with urlopen(request, timeout=30) as response:
|
||||
response_data = json.loads(response.read().decode("utf-8"))
|
||||
return response_data
|
||||
|
||||
except HTTPError as e:
|
||||
error_body = e.read().decode("utf-8") if e.fp else "No error details"
|
||||
raise AuthenticationError(f"{error_context} failed: {e.code} {error_body}")
|
||||
except URLError as e:
|
||||
raise AuthenticationError(f"Network error during {error_context}: {e.reason}")
|
||||
except json.JSONDecodeError as e:
|
||||
raise AuthenticationError(f"Invalid JSON response from {error_context}: {e}")
|
||||
|
||||
|
||||
def exchange_access_code_for_tokens(
|
||||
access_code: str, challenge: str, server_url: str
|
||||
) -> Dict[str, str]:
|
||||
"""Exchange access code and challenge for authentication tokens."""
|
||||
if not challenge:
|
||||
raise AuthenticationError("No challenge available")
|
||||
|
||||
# Prepare request body
|
||||
body = {
|
||||
"appId": SPECKLE_APP_ID,
|
||||
"appSecret": SPECKLE_APP_ID,
|
||||
"accessCode": access_code,
|
||||
"challenge": challenge,
|
||||
}
|
||||
|
||||
# Make POST request
|
||||
url = f"{server_url}/auth/token"
|
||||
try:
|
||||
response_data = _post_json(url, body, error_context="token exchange")
|
||||
except AuthenticationError as e:
|
||||
_handle_auth_error(e)
|
||||
|
||||
# Validate response
|
||||
if "token" not in response_data or "refreshToken" not in response_data:
|
||||
raise AuthenticationError("Invalid response from token endpoint")
|
||||
|
||||
return {
|
||||
"token": response_data["token"],
|
||||
"refreshToken": response_data["refreshToken"],
|
||||
}
|
||||
|
||||
|
||||
def get_user_and_server_info(
|
||||
token: str, server_url: str
|
||||
) -> Tuple[Dict[str, Any], Dict[str, Any]]:
|
||||
"""Get user and server information using GraphQL query with auth token."""
|
||||
# Prepare GraphQL query
|
||||
query = """
|
||||
query {
|
||||
activeUser {
|
||||
id
|
||||
name
|
||||
email
|
||||
company
|
||||
avatar
|
||||
}
|
||||
serverInfo {
|
||||
name
|
||||
company
|
||||
adminContact
|
||||
description
|
||||
version
|
||||
}
|
||||
}
|
||||
"""
|
||||
|
||||
body = {"query": query}
|
||||
|
||||
# Make POST request
|
||||
url = f"{server_url}/graphql"
|
||||
try:
|
||||
response_data = _post_json(
|
||||
url, body, auth_token=token, error_context="user info request"
|
||||
)
|
||||
except AuthenticationError as e:
|
||||
_handle_auth_error(e)
|
||||
|
||||
# Validate response
|
||||
if "data" not in response_data:
|
||||
raise AuthenticationError("Invalid GraphQL response")
|
||||
|
||||
data = response_data["data"]
|
||||
|
||||
if "activeUser" not in data or "serverInfo" not in data:
|
||||
raise AuthenticationError("Missing user or server info in response")
|
||||
|
||||
user_info = data["activeUser"]
|
||||
server_info = data["serverInfo"]
|
||||
|
||||
# Ensure server URL is set correctly
|
||||
server_info["url"] = server_url.rstrip("/")
|
||||
|
||||
return user_info, server_info
|
||||
|
||||
|
||||
def save_account_to_storage(
|
||||
token: str,
|
||||
refresh_token: str,
|
||||
user_info: Dict[str, Any],
|
||||
server_info: Dict[str, Any],
|
||||
) -> None:
|
||||
"""Save account to Accounts.db SQLite database for compatibility with specklepy."""
|
||||
try:
|
||||
import sqlite3
|
||||
import hashlib
|
||||
import os
|
||||
from specklepy.core.api.credentials import speckle_path_provider
|
||||
|
||||
# Generate account ID (hash of email + server URL)
|
||||
account_id_string = f"{user_info['email']}-{server_info['url']}"
|
||||
account_id = hashlib.md5(account_id_string.encode()).hexdigest().upper()
|
||||
|
||||
# Construct account object matching the expected format
|
||||
account_data = {
|
||||
"id": account_id,
|
||||
"token": token,
|
||||
"refreshToken": refresh_token,
|
||||
"isDefault": True,
|
||||
"isOnline": True,
|
||||
"serverInfo": {
|
||||
"name": server_info["name"],
|
||||
"company": server_info.get("company"),
|
||||
"version": server_info.get("version"),
|
||||
"description": server_info.get("description"),
|
||||
"url": server_info["url"],
|
||||
},
|
||||
"userInfo": {
|
||||
"id": user_info["id"],
|
||||
"name": user_info["name"],
|
||||
"email": user_info["email"],
|
||||
"company": user_info.get("company"),
|
||||
"avatar": user_info.get("avatar"),
|
||||
},
|
||||
}
|
||||
|
||||
# Get database path
|
||||
speckle_folder = speckle_path_provider.user_speckle_folder_path()
|
||||
db_path = os.path.join(speckle_folder, "Accounts.db")
|
||||
|
||||
# Ensure the Speckle folder exists
|
||||
os.makedirs(speckle_folder, exist_ok=True)
|
||||
|
||||
# Connect to database and save account
|
||||
# Use IMMEDIATE isolation level to acquire write lock immediately,
|
||||
# preventing race conditions in concurrent account additions
|
||||
conn = sqlite3.connect(db_path, isolation_level="IMMEDIATE")
|
||||
try:
|
||||
with conn:
|
||||
cursor = conn.cursor()
|
||||
|
||||
# Create table if it doesn't exist
|
||||
cursor.execute("""
|
||||
CREATE TABLE IF NOT EXISTS objects (
|
||||
hash TEXT PRIMARY KEY,
|
||||
content TEXT
|
||||
)
|
||||
""")
|
||||
|
||||
# If setting as default, remove default flag from other accounts
|
||||
# Use batch update to make the operation more atomic
|
||||
if account_data["isDefault"]:
|
||||
cursor.execute("SELECT hash, content FROM objects")
|
||||
rows = cursor.fetchall()
|
||||
|
||||
# Build list of updates to execute in batch
|
||||
updates = []
|
||||
for existing_id, existing_content in rows:
|
||||
try:
|
||||
existing_account = json.loads(existing_content)
|
||||
if existing_account.get("isDefault", False):
|
||||
existing_account["isDefault"] = False
|
||||
updates.append(
|
||||
(json.dumps(existing_account), existing_id)
|
||||
)
|
||||
except json.JSONDecodeError:
|
||||
# Skip malformed accounts
|
||||
continue
|
||||
|
||||
# Execute all updates in batch for better atomicity
|
||||
if updates:
|
||||
cursor.executemany(
|
||||
"UPDATE objects SET content = ? WHERE hash = ?", updates
|
||||
)
|
||||
|
||||
# Insert or replace the account
|
||||
cursor.execute(
|
||||
"INSERT OR REPLACE INTO objects (hash, content) VALUES (?, ?)",
|
||||
(account_id, json.dumps(account_data)),
|
||||
)
|
||||
|
||||
conn.commit()
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
print(
|
||||
f"[Auth] Successfully saved account: {user_info['email']} @ {server_info['url']} (ID: {account_id})"
|
||||
)
|
||||
|
||||
# Track account creation event
|
||||
try:
|
||||
from specklepy.logging import metrics
|
||||
|
||||
metrics.track(metrics.HOST_APP, "connector", "account", {"action": "add"})
|
||||
except Exception as e:
|
||||
# Don't fail if metrics tracking fails
|
||||
print(f"[Auth] Failed to track metrics: {e}")
|
||||
|
||||
except Exception as e:
|
||||
raise AuthenticationError(f"Failed to save account: {e}")
|
||||
|
||||
|
||||
class AuthenticationServer:
|
||||
"""Manages local HTTP server for Speckle authentication in a background thread."""
|
||||
|
||||
def __init__(self, port: int = SPECKLE_AUTH_PORT):
|
||||
self.port = port
|
||||
self.server: Optional[ThreadSafeAuthServer] = None
|
||||
self.thread: Optional[threading.Thread] = None
|
||||
self.shutdown_event = threading.Event()
|
||||
|
||||
def start(self) -> bool:
|
||||
"""Start HTTP server in background thread."""
|
||||
try:
|
||||
# Create thread-safe server (state initialized in constructor)
|
||||
self.server = ThreadSafeAuthServer(
|
||||
("127.0.0.1", self.port), SpeckleAuthHandler
|
||||
)
|
||||
|
||||
# Start server in background thread
|
||||
self.thread = threading.Thread(target=self._run_server, daemon=True)
|
||||
self.thread.start()
|
||||
|
||||
print(f"[Auth Server] Started on http://127.0.0.1:{self.port}")
|
||||
return True
|
||||
|
||||
except OSError as e:
|
||||
if e.errno in (errno.EADDRINUSE, 10048): # Address already in use
|
||||
print(f"[Auth Server] Port {self.port} is already in use.")
|
||||
else:
|
||||
print(f"[Auth Server] Failed to start server: {e}")
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f"[Auth Server] Unexpected error starting server: {e}")
|
||||
return False
|
||||
|
||||
def _run_server(self):
|
||||
try:
|
||||
# Set a timeout so handle_request doesn't block forever
|
||||
self.server.timeout = 0.5
|
||||
|
||||
# Server should handle a maximum of 3 requests:
|
||||
# 1. /auth/add-account (redirect to Speckle)
|
||||
# 2. / callback (from Speckle with access_code)
|
||||
# 3. Maybe a favicon or other browser request
|
||||
# After that or when shutdown is signaled, stop
|
||||
max_requests = 5 # Allow a few extra for browser quirks
|
||||
|
||||
while (
|
||||
not self.shutdown_event.is_set()
|
||||
and self.server.request_count < max_requests
|
||||
):
|
||||
self.server.handle_request()
|
||||
|
||||
# If auth is complete, we can stop serving
|
||||
if self.server.is_complete:
|
||||
print("[Auth Server] Authentication complete, stopping server")
|
||||
break
|
||||
|
||||
except Exception as e:
|
||||
print(f"[Auth Server] Error in server thread: {e}")
|
||||
self.server.set_auth_failure(f"Server thread crashed: {e}")
|
||||
|
||||
def shutdown(self):
|
||||
if self.server:
|
||||
self.shutdown_event.set()
|
||||
try:
|
||||
# Give the server thread a moment to see the shutdown event
|
||||
if self.thread and self.thread.is_alive():
|
||||
self.thread.join(timeout=2.0)
|
||||
|
||||
self.server.server_close()
|
||||
except Exception as e:
|
||||
print(f"[Auth Server] Error during shutdown: {e}")
|
||||
|
||||
self.server = None
|
||||
self.thread = None
|
||||
print("[Auth Server] Shutdown complete")
|
||||
|
||||
def is_complete(self) -> bool:
|
||||
return self.server.is_complete if self.server else False
|
||||
|
||||
def is_successful(self) -> bool:
|
||||
return self.server.is_successful if self.server else False
|
||||
|
||||
def get_error_message(self) -> Optional[str]:
|
||||
return self.server.error_message if self.server else None
|
||||
|
||||
def open_auth_url(self, server_url: str = "https://app.speckle.systems"):
|
||||
"""Open authentication URL in browser to initiate auth flow."""
|
||||
# Trigger the add-account endpoint
|
||||
url = f"http://127.0.0.1:{self.port}/auth/add-account?serverUrl={server_url}"
|
||||
webbrowser.open(url)
|
||||
print("[Auth Server] Opening browser to initiate authentication...")
|
||||
@@ -0,0 +1,26 @@
|
||||
from typing import Iterator, TypeVar, Type
|
||||
from specklepy.objects.base import Base
|
||||
from specklepy.objects.graph_traversal.traversal import TraversalContext
|
||||
|
||||
|
||||
def get_ascendants(context: TraversalContext) -> Iterator[Base]:
|
||||
"""
|
||||
walks up the tree, returning all ascendants, including context
|
||||
"""
|
||||
head = context
|
||||
while head is not None:
|
||||
yield head.current
|
||||
head = head.parent
|
||||
|
||||
|
||||
T = TypeVar("T", bound=Base)
|
||||
|
||||
|
||||
def get_ascendant_of_type(context: TraversalContext, type_cls: Type[T]) -> Iterator[T]:
|
||||
"""
|
||||
walks up the tree, returning all ascendants of the given type,
|
||||
starting with the context, walking up parent nodes
|
||||
"""
|
||||
for ascendant in get_ascendants(context):
|
||||
if isinstance(ascendant, type_cls):
|
||||
yield ascendant
|
||||
@@ -0,0 +1,53 @@
|
||||
from datetime import datetime, timezone
|
||||
import re
|
||||
|
||||
|
||||
def format_relative_time(timestamp) -> str:
|
||||
"""
|
||||
convert UTC timestamp to local timezone and return relative time string
|
||||
"""
|
||||
if not timestamp:
|
||||
return "Unknown"
|
||||
|
||||
# convert to local timezone
|
||||
try:
|
||||
try:
|
||||
dt = datetime.fromisoformat(str(timestamp).replace("Z", "+00:00"))
|
||||
except ValueError:
|
||||
try:
|
||||
ts = float(timestamp)
|
||||
dt = datetime.fromtimestamp(ts / 1000, tz=timezone.utc)
|
||||
except (ValueError, TypeError):
|
||||
return "Invalid timestamp"
|
||||
|
||||
local_dt = dt.astimezone()
|
||||
|
||||
# calculate relative time
|
||||
now = datetime.now(timezone.utc).astimezone()
|
||||
delta = now - local_dt
|
||||
|
||||
if delta.days == 0:
|
||||
if delta.seconds < 3600:
|
||||
minutes = delta.seconds // 60
|
||||
return f"{minutes} minutes ago"
|
||||
else:
|
||||
hours = delta.seconds // 3600
|
||||
return f"{hours} hours ago"
|
||||
else:
|
||||
return f"{delta.days} days ago"
|
||||
except ValueError:
|
||||
return "Invalid timestamp"
|
||||
|
||||
|
||||
def format_role(role: str) -> str:
|
||||
"""
|
||||
This function takes a Speckle role string in the format "prefix:role" and
|
||||
returns just the role part
|
||||
"""
|
||||
split_role = role.split(":")
|
||||
return f"{split_role[1]}"
|
||||
|
||||
|
||||
def strip_non_ascii(text):
|
||||
# Keep English letters, digits, spaces and basic punctuation
|
||||
return re.sub(r"[^a-zA-Z0-9\s.,!?]", "", text)
|
||||
@@ -0,0 +1,428 @@
|
||||
from typing import Any, Dict, Optional
|
||||
|
||||
import bpy
|
||||
from bpy.types import Context
|
||||
|
||||
from ..utils.property_groups import speckle_model_card
|
||||
|
||||
|
||||
def find_layer_collection(layer_collection, collection_name):
|
||||
"""
|
||||
Recursively find a layer collection by collection name
|
||||
"""
|
||||
if layer_collection.collection.name == collection_name:
|
||||
return layer_collection
|
||||
for child in layer_collection.children:
|
||||
result = find_layer_collection(child, collection_name)
|
||||
if result:
|
||||
return result
|
||||
return None
|
||||
|
||||
|
||||
def get_object_by_application_id(app_id: str):
|
||||
"""
|
||||
Find a Blender object by its applicationId stored in custom property
|
||||
"""
|
||||
if not app_id:
|
||||
return None
|
||||
|
||||
for obj in bpy.data.objects:
|
||||
if "applicationId" in obj and obj["applicationId"] == app_id:
|
||||
return obj
|
||||
return None
|
||||
|
||||
|
||||
def get_objects_by_application_ids(app_ids: list):
|
||||
"""
|
||||
Find multiple Blender objects by their applicationIds
|
||||
"""
|
||||
if not app_ids:
|
||||
return {}
|
||||
|
||||
result = {}
|
||||
for obj in bpy.data.objects:
|
||||
if "applicationId" in obj and obj["applicationId"] in app_ids:
|
||||
result[obj["applicationId"]] = obj
|
||||
return result
|
||||
|
||||
|
||||
def get_collection_by_application_id(app_id: str):
|
||||
"""
|
||||
Find a Blender collection by its applicationId stored in custom property
|
||||
"""
|
||||
if not app_id:
|
||||
return None
|
||||
|
||||
for collection in bpy.data.collections:
|
||||
if "applicationId" in collection and collection["applicationId"] == app_id:
|
||||
return collection
|
||||
return None
|
||||
|
||||
|
||||
def get_collection_identifier(blender_col: bpy.types.Collection) -> str:
|
||||
"""
|
||||
Get collection identifier: applicationId if exists, fallback to name
|
||||
"""
|
||||
if "applicationId" in blender_col and blender_col["applicationId"]:
|
||||
return blender_col["applicationId"]
|
||||
return blender_col.name
|
||||
|
||||
|
||||
def find_collection_by_identifier(identifier: str):
|
||||
"""
|
||||
Find collection by identifier: try applicationId first, then name
|
||||
"""
|
||||
# first try to find by applicationId
|
||||
collection = get_collection_by_application_id(identifier)
|
||||
if collection:
|
||||
return collection
|
||||
|
||||
# fallback to name-based lookup
|
||||
return bpy.data.collections.get(identifier)
|
||||
|
||||
|
||||
def capture_modifier_data(blender_obj: bpy.types.Object) -> list:
|
||||
"""
|
||||
Capture modifier data from a Blender object as dictionaries
|
||||
"""
|
||||
modifiers_data = []
|
||||
for modifier in blender_obj.modifiers:
|
||||
modifier_data = {
|
||||
"name": modifier.name,
|
||||
"type": modifier.type,
|
||||
"show_viewport": modifier.show_viewport,
|
||||
"show_render": modifier.show_render,
|
||||
"show_in_editmode": modifier.show_in_editmode,
|
||||
"show_on_cage": modifier.show_on_cage,
|
||||
"properties": {},
|
||||
}
|
||||
|
||||
# Capture modifier-specific properties
|
||||
for prop_name in modifier.bl_rna.properties.keys():
|
||||
if prop_name in [
|
||||
"rna_type",
|
||||
"name",
|
||||
"type",
|
||||
"show_viewport",
|
||||
"show_render",
|
||||
"show_in_editmode",
|
||||
"show_on_cage",
|
||||
]:
|
||||
continue
|
||||
try:
|
||||
if hasattr(modifier, prop_name):
|
||||
prop_value = getattr(modifier, prop_name)
|
||||
# Handle different property types
|
||||
if isinstance(prop_value, (int, float, bool, str)):
|
||||
modifier_data["properties"][prop_name] = prop_value
|
||||
elif hasattr(prop_value, "name"): # Object references
|
||||
modifier_data["properties"][prop_name] = prop_value.name
|
||||
elif (
|
||||
hasattr(prop_value, "__len__") and len(prop_value) <= 4
|
||||
): # Vectors/colors
|
||||
modifier_data["properties"][prop_name] = list(prop_value)
|
||||
except (AttributeError, TypeError):
|
||||
continue
|
||||
|
||||
modifiers_data.append(modifier_data)
|
||||
|
||||
return modifiers_data
|
||||
|
||||
|
||||
def has_visibility_modifications(obj: bpy.types.Object) -> bool:
|
||||
"""Check if object has non-default visibility settings"""
|
||||
return obj.hide_viewport or obj.hide_select or obj.hide_render or obj.hide_get()
|
||||
|
||||
|
||||
def has_modifier_modifications(obj: bpy.types.Object) -> bool:
|
||||
"""Check if object has any modifiers applied"""
|
||||
return hasattr(obj, "modifiers") and len(obj.modifiers) > 0
|
||||
|
||||
|
||||
def has_collection_visibility_modifications(layer_col, collection) -> bool:
|
||||
"""Check if collection has non-default visibility settings"""
|
||||
return (
|
||||
layer_col.hide_viewport
|
||||
or collection.hide_select
|
||||
or collection.hide_render
|
||||
or layer_col.exclude
|
||||
)
|
||||
|
||||
|
||||
def collect_objects_with_properties(
|
||||
model_card: speckle_model_card,
|
||||
) -> Dict[str, Dict[str, Any]]:
|
||||
"""
|
||||
Collect objects and collections with their current properties before deletion
|
||||
Only stores data for objects that have been modified from defaults
|
||||
"""
|
||||
collected_data = {"objects": {}, "collections": {}}
|
||||
|
||||
# Collect object properties (only for modified objects)
|
||||
for s_obj in model_card.objects:
|
||||
blender_obj = get_object_by_application_id(s_obj.applicationId)
|
||||
if blender_obj:
|
||||
obj_data = {}
|
||||
|
||||
# Only collect visibility if modified from defaults
|
||||
if has_visibility_modifications(blender_obj):
|
||||
obj_data["visibility"] = {
|
||||
"hide_get": blender_obj.hide_get(),
|
||||
"hide_viewport": blender_obj.hide_viewport,
|
||||
"hide_select": blender_obj.hide_select,
|
||||
"hide_render": blender_obj.hide_render,
|
||||
}
|
||||
|
||||
# Only collect modifiers if object has any
|
||||
if has_modifier_modifications(blender_obj):
|
||||
obj_data["modifiers"] = capture_modifier_data(blender_obj)
|
||||
|
||||
# Only store object data if it has modifications
|
||||
if obj_data:
|
||||
collected_data["objects"][s_obj.applicationId] = obj_data
|
||||
|
||||
# Collect collection properties (only for modified collections)
|
||||
for s_col in model_card.collections:
|
||||
# try to find collection by applicationId first, then fallback to name
|
||||
blender_col = None
|
||||
if s_col.applicationId:
|
||||
blender_col = get_collection_by_application_id(s_col.applicationId)
|
||||
if not blender_col:
|
||||
blender_col = bpy.data.collections.get(s_col.name)
|
||||
|
||||
if blender_col:
|
||||
view_layer = bpy.context.view_layer
|
||||
if view_layer:
|
||||
layer_col = find_layer_collection(
|
||||
view_layer.layer_collection, blender_col.name
|
||||
)
|
||||
if layer_col and has_collection_visibility_modifications(
|
||||
layer_col, blender_col
|
||||
):
|
||||
# use collection identifier as key
|
||||
collection_id = get_collection_identifier(blender_col)
|
||||
collected_data["collections"][collection_id] = {
|
||||
"hide_viewport": layer_col.hide_viewport,
|
||||
"hide_select": layer_col.collection.hide_select,
|
||||
"hide_render": layer_col.collection.hide_render,
|
||||
"exclude_from_view_layer": layer_col.exclude,
|
||||
}
|
||||
|
||||
return collected_data
|
||||
|
||||
|
||||
def transfer_object_properties(
|
||||
new_obj: bpy.types.Object, old_obj_data: Dict[str, Any]
|
||||
) -> None:
|
||||
"""
|
||||
Transfer visibility and modifiers from old object data to new object
|
||||
Handles sparse data gracefully - applies defaults when data is missing
|
||||
"""
|
||||
# Transfer visibility settings (if any were modified)
|
||||
visibility = old_obj_data.get("visibility")
|
||||
if visibility:
|
||||
new_obj.hide_set(visibility.get("hide_get", False))
|
||||
new_obj.hide_viewport = visibility.get("hide_viewport", False)
|
||||
new_obj.hide_select = visibility.get("hide_select", False)
|
||||
new_obj.hide_render = visibility.get("hide_render", False)
|
||||
# If no visibility data, object keeps defaults (all False)
|
||||
|
||||
# Transfer modifiers (if any were present)
|
||||
old_modifiers = old_obj_data.get("modifiers")
|
||||
if old_modifiers and hasattr(new_obj, "modifiers"):
|
||||
# Clear existing modifiers
|
||||
new_obj.modifiers.clear()
|
||||
|
||||
# Transfer each modifier
|
||||
for modifier_data in old_modifiers:
|
||||
recreate_modifier_from_data(new_obj, modifier_data)
|
||||
# If no modifier data, object keeps default (no modifiers)
|
||||
|
||||
|
||||
def transfer_collection_properties(
|
||||
new_col: bpy.types.Collection, old_col_data: Dict[str, Any]
|
||||
) -> None:
|
||||
"""
|
||||
Transfer visibility properties from old collection data to new collection
|
||||
Handles sparse data gracefully - applies defaults when data is missing
|
||||
"""
|
||||
view_layer = bpy.context.view_layer
|
||||
if view_layer:
|
||||
layer_col = find_layer_collection(view_layer.layer_collection, new_col.name)
|
||||
if layer_col:
|
||||
# Only apply properties if collection had modifications
|
||||
# (otherwise it keeps defaults: all False)
|
||||
layer_col.hide_viewport = old_col_data.get("hide_viewport", False)
|
||||
layer_col.collection.hide_select = old_col_data.get("hide_select", False)
|
||||
layer_col.collection.hide_render = old_col_data.get("hide_render", False)
|
||||
layer_col.exclude = old_col_data.get("exclude_from_view_layer", False)
|
||||
|
||||
|
||||
def recreate_modifier_from_data(
|
||||
new_obj: bpy.types.Object, modifier_data: Dict[str, Any]
|
||||
) -> Optional[bpy.types.Modifier]:
|
||||
"""
|
||||
Recreate a modifier from captured data
|
||||
"""
|
||||
try:
|
||||
# Validate modifier data
|
||||
if not modifier_data.get("type") or not modifier_data.get("name"):
|
||||
print(f"Invalid modifier data: {modifier_data}")
|
||||
return None
|
||||
|
||||
# Create new modifier
|
||||
new_modifier = new_obj.modifiers.new(
|
||||
modifier_data["name"], modifier_data["type"]
|
||||
)
|
||||
|
||||
# Set visibility properties
|
||||
new_modifier.show_viewport = modifier_data.get("show_viewport", True)
|
||||
new_modifier.show_render = modifier_data.get("show_render", True)
|
||||
new_modifier.show_in_editmode = modifier_data.get("show_in_editmode", True)
|
||||
new_modifier.show_on_cage = modifier_data.get("show_on_cage", False)
|
||||
|
||||
# Set modifier-specific properties
|
||||
for prop_name, prop_value in modifier_data.get("properties", {}).items():
|
||||
try:
|
||||
if hasattr(new_modifier, prop_name):
|
||||
current_value = getattr(new_modifier, prop_name)
|
||||
# Handle object references
|
||||
if hasattr(current_value, "name") and isinstance(prop_value, str):
|
||||
referenced_obj = bpy.data.objects.get(prop_value)
|
||||
if referenced_obj:
|
||||
setattr(new_modifier, prop_name, referenced_obj)
|
||||
else:
|
||||
setattr(new_modifier, prop_name, prop_value)
|
||||
except (AttributeError, TypeError):
|
||||
continue
|
||||
|
||||
return new_modifier
|
||||
except Exception as e:
|
||||
print(f"Error recreating modifier {modifier_data.get('name', 'unknown')}: {e}")
|
||||
return None
|
||||
|
||||
|
||||
def update_model_card_objects(
|
||||
model_card: speckle_model_card,
|
||||
converted_objects: Dict[str, bpy.types.Object | bpy.types.Collection],
|
||||
old_properties: Optional[Dict[str, Dict[str, Any]]] = None,
|
||||
):
|
||||
"""
|
||||
Update model card with new objects and apply properties from old objects if provided
|
||||
"""
|
||||
# Clear model card objects
|
||||
model_card.objects.clear()
|
||||
model_card.collections.clear()
|
||||
|
||||
# Convert list to dictionary if needed
|
||||
if isinstance(converted_objects, list):
|
||||
converted_objects = {obj.name: obj for obj in converted_objects}
|
||||
|
||||
# Using a set keeps lookup O(1)
|
||||
object_names = set()
|
||||
collection_names = set()
|
||||
|
||||
for obj in converted_objects.values():
|
||||
# Handle collections
|
||||
if isinstance(obj, bpy.types.Collection):
|
||||
if obj.name in collection_names:
|
||||
continue
|
||||
collection_names.add(obj.name)
|
||||
|
||||
s_col = model_card.collections.add()
|
||||
s_col.name = obj.name
|
||||
s_col.applicationId = obj.get("applicationId", "")
|
||||
|
||||
# apply old collection properties if available (use identifier-based lookup)
|
||||
if old_properties:
|
||||
collection_id = get_collection_identifier(obj)
|
||||
if collection_id in old_properties.get("collections", {}):
|
||||
old_col_data = old_properties["collections"][collection_id]
|
||||
transfer_collection_properties(obj, old_col_data)
|
||||
|
||||
# Handle objects
|
||||
elif isinstance(obj, bpy.types.Object):
|
||||
if obj.name in object_names:
|
||||
continue
|
||||
object_names.add(obj.name)
|
||||
|
||||
s_obj = model_card.objects.add()
|
||||
s_obj.name = obj.name
|
||||
s_obj.applicationId = obj.get("applicationId", "")
|
||||
|
||||
# Apply old object properties if available
|
||||
if (
|
||||
old_properties
|
||||
and s_obj.applicationId
|
||||
and s_obj.applicationId in old_properties.get("objects", {})
|
||||
):
|
||||
old_obj_data = old_properties["objects"][s_obj.applicationId]
|
||||
transfer_object_properties(obj, old_obj_data)
|
||||
|
||||
|
||||
def delete_model_card_objects(model_card: speckle_model_card, context: Context) -> None:
|
||||
"""
|
||||
deletes the model card objects
|
||||
"""
|
||||
# Delete objects directly without requiring selection
|
||||
for obj in model_card.objects:
|
||||
blender_obj = get_object_by_application_id(obj.applicationId)
|
||||
if not blender_obj:
|
||||
continue
|
||||
|
||||
# Remove object from all collections first
|
||||
for collection in blender_obj.users_collection:
|
||||
collection.objects.unlink(blender_obj)
|
||||
|
||||
# Delete the object directly
|
||||
bpy.data.objects.remove(blender_obj)
|
||||
|
||||
# delete model card/currently loaded collections
|
||||
for col in model_card.collections:
|
||||
coll = bpy.data.collections.get(col.name)
|
||||
if not coll:
|
||||
continue
|
||||
# unlink from scenes
|
||||
for scene in bpy.data.scenes:
|
||||
if scene.collection.children.get(coll.name):
|
||||
scene.collection.children.unlink(coll)
|
||||
bpy.data.collections.remove(coll)
|
||||
|
||||
|
||||
def select_model_card_objects(model_card, context: Context):
|
||||
# deselect all objects first
|
||||
bpy.ops.object.select_all(action="DESELECT")
|
||||
# select objects in model card
|
||||
for obj in model_card.objects:
|
||||
blender_obj = get_object_by_application_id(obj.applicationId)
|
||||
if not blender_obj:
|
||||
continue
|
||||
if blender_obj.name in context.view_layer.objects:
|
||||
blender_obj.select_set(True)
|
||||
|
||||
selected = context.selected_objects
|
||||
if selected:
|
||||
context.view_layer.objects.active = selected[0]
|
||||
|
||||
|
||||
def zoom_to_selected_objects(context: Context):
|
||||
"""
|
||||
zooms to the selected objects
|
||||
"""
|
||||
bpy.ops.view3d.view_selected()
|
||||
|
||||
|
||||
def model_card_exists(
|
||||
project_id: str, model_id: str, is_publish: bool, context: Context
|
||||
) -> bool:
|
||||
"""
|
||||
checks if a model card exists
|
||||
"""
|
||||
for model_card in context.scene.speckle_state.model_cards:
|
||||
if (
|
||||
model_card.project_id == project_id
|
||||
and model_card.model_id == model_id
|
||||
and model_card.is_publish == is_publish
|
||||
):
|
||||
return True
|
||||
return False
|
||||
@@ -0,0 +1,52 @@
|
||||
from specklepy.core.api.inputs.project_inputs import ProjectModelsFilter
|
||||
from specklepy.core.api.models.current import Model
|
||||
from typing import List, Tuple, Optional
|
||||
from .misc import format_relative_time, strip_non_ascii
|
||||
from .account_manager import _client_cache
|
||||
|
||||
|
||||
def get_models_for_project(
|
||||
account_id: str, project_id: str, search: Optional[str] = None
|
||||
) -> List[Tuple[str, str, str]]:
|
||||
"""
|
||||
fetches models for a given project from the Speckle server
|
||||
"""
|
||||
try:
|
||||
if not account_id or not project_id:
|
||||
print(
|
||||
f"Error: Invalid inputs - account_id: {account_id}, project_id: {project_id}"
|
||||
)
|
||||
return []
|
||||
|
||||
# Get cached client
|
||||
client = _client_cache.get_client(account_id)
|
||||
if not client:
|
||||
print(f"Error: Could not get client for account: {account_id}")
|
||||
return []
|
||||
|
||||
try:
|
||||
client.project.get(project_id)
|
||||
except Exception as e:
|
||||
print(f"Error: Project with ID {project_id} not found: {str(e)}")
|
||||
return []
|
||||
|
||||
filter = ProjectModelsFilter(search=search) if search else None
|
||||
|
||||
models: List[Model] = client.model.get_models(
|
||||
project_id=project_id, models_limit=10, models_filter=filter
|
||||
).items
|
||||
|
||||
return [
|
||||
(
|
||||
strip_non_ascii(model.name),
|
||||
model.id,
|
||||
format_relative_time(model.updated_at),
|
||||
)
|
||||
for model in models
|
||||
]
|
||||
|
||||
except Exception as e:
|
||||
print(f"Error fetching models: {str(e)}")
|
||||
# Clear cache on error to prevent stale clients
|
||||
_client_cache.clear()
|
||||
return []
|
||||
@@ -0,0 +1,131 @@
|
||||
from specklepy.core.api.client import SpeckleClient
|
||||
from specklepy.core.api.resources.current.workspace_resource import WorkspaceResource
|
||||
from specklepy.core.api.inputs.project_inputs import WorksaceProjectsFilter
|
||||
from typing import List, Tuple, Optional
|
||||
from specklepy.core.api.credentials import Account
|
||||
from .misc import format_relative_time, format_role, strip_non_ascii
|
||||
from .account_manager import _client_cache
|
||||
|
||||
|
||||
def get_projects_for_account(
|
||||
account_id: str, workspace_id: str = None, search: Optional[str] = None
|
||||
) -> List[Tuple[str, str, str, str, bool]]:
|
||||
"""
|
||||
fetches projects for a given account from the Speckle server
|
||||
"""
|
||||
try:
|
||||
# Get cached client
|
||||
client = _client_cache.get_client(account_id)
|
||||
if not client:
|
||||
print(f"Error: Could not get client for account: {account_id}")
|
||||
return []
|
||||
|
||||
# Get account for workspace operations that still need it
|
||||
from specklepy.core.api.credentials import get_local_accounts
|
||||
|
||||
account: Optional[Account] = next(
|
||||
(acc for acc in get_local_accounts() if acc.id == account_id), None
|
||||
)
|
||||
if not account:
|
||||
print(f"Error: Could not find account with ID: {account_id}")
|
||||
return []
|
||||
|
||||
try:
|
||||
workspace_resource = WorkspaceResource(
|
||||
account, client.url, client.httpclient, client.server.version()
|
||||
)
|
||||
|
||||
# create filter with search parameter
|
||||
filter = (
|
||||
WorksaceProjectsFilter(search=search, with_project_role_only=False)
|
||||
if search
|
||||
else None
|
||||
)
|
||||
|
||||
projects_with_permissions = (
|
||||
workspace_resource.get_projects_with_permissions(
|
||||
workspace_id=workspace_id, limit=10, filter=filter
|
||||
)
|
||||
)
|
||||
|
||||
result = []
|
||||
for project in projects_with_permissions.items:
|
||||
can_load_permission = False
|
||||
|
||||
if hasattr(project, "permissions") and project.permissions:
|
||||
can_load_permission = (
|
||||
hasattr(project.permissions, "can_load")
|
||||
and project.permissions.can_load
|
||||
and project.permissions.can_load.authorized
|
||||
)
|
||||
|
||||
result.append(
|
||||
(
|
||||
strip_non_ascii(project.name),
|
||||
format_role(getattr(project, "role", ""))
|
||||
if hasattr(project, "role") and project.role
|
||||
else "",
|
||||
format_relative_time(project.updated_at),
|
||||
project.id,
|
||||
can_load_permission,
|
||||
)
|
||||
)
|
||||
|
||||
return result
|
||||
|
||||
except Exception as workspace_error:
|
||||
print(
|
||||
f"WorkspaceResource failed, falling back to old method: {workspace_error}"
|
||||
)
|
||||
return _get_projects_with_individual_permissions(
|
||||
client, workspace_id, search
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
import traceback
|
||||
|
||||
error_msg = f"Error: {str(e)}\n"
|
||||
error_msg += f"Traceback:\n{''.join(traceback.format_tb(e.__traceback__))}"
|
||||
print(error_msg)
|
||||
# Clear cache on error to prevent stale clients
|
||||
_client_cache.clear()
|
||||
return []
|
||||
|
||||
|
||||
def _get_projects_with_individual_permissions(
|
||||
client: SpeckleClient,
|
||||
workspace_id: str,
|
||||
search: Optional[str] = None,
|
||||
) -> List[Tuple[str, str, str, str, bool]]:
|
||||
"""
|
||||
Fallback helper function to get projects with permissions using individual API calls
|
||||
"""
|
||||
from specklepy.core.api.inputs.user_inputs import UserProjectsFilter
|
||||
from .account_manager import can_load
|
||||
|
||||
filter = UserProjectsFilter(
|
||||
search=search,
|
||||
workspaceId=workspace_id,
|
||||
personalOnly=False,
|
||||
include_implicit_access=True,
|
||||
)
|
||||
|
||||
projects = client.active_user.get_projects(limit=10, filter=filter).items
|
||||
|
||||
result = []
|
||||
for project in projects:
|
||||
can_load_permission, _ = can_load(client, project)
|
||||
|
||||
result.append(
|
||||
(
|
||||
strip_non_ascii(project.name),
|
||||
format_role(getattr(project, "role", ""))
|
||||
if hasattr(project, "role") and project.role
|
||||
else "",
|
||||
format_relative_time(project.updated_at),
|
||||
project.id,
|
||||
can_load_permission,
|
||||
)
|
||||
)
|
||||
|
||||
return result
|
||||
@@ -0,0 +1,115 @@
|
||||
import bpy
|
||||
|
||||
|
||||
class speckle_project(bpy.types.PropertyGroup):
|
||||
"""
|
||||
PropertyGroup for storing project information
|
||||
"""
|
||||
|
||||
name: bpy.props.StringProperty() # type: ignore
|
||||
role: bpy.props.StringProperty(name="Role") # type: ignore
|
||||
updated: bpy.props.StringProperty(name="Updated") # type: ignore
|
||||
id: bpy.props.StringProperty(name="ID") # type: ignore
|
||||
can_receive: bpy.props.BoolProperty(name="Can Receive", default=False) # type: ignore
|
||||
|
||||
|
||||
class speckle_model(bpy.types.PropertyGroup):
|
||||
"""
|
||||
PropertyGroup for storing model information
|
||||
"""
|
||||
|
||||
name: bpy.props.StringProperty() # type: ignore
|
||||
id: bpy.props.StringProperty(name="ID") # type: ignore
|
||||
updated: bpy.props.StringProperty(name="Updated") # type: ignore
|
||||
|
||||
|
||||
class speckle_version(bpy.types.PropertyGroup):
|
||||
"""
|
||||
PropertyGroup for storing version information
|
||||
"""
|
||||
|
||||
id: bpy.props.StringProperty(name="ID") # type: ignore
|
||||
message: bpy.props.StringProperty(name="Message") # type: ignore
|
||||
updated: bpy.props.StringProperty(name="Updated") # type: ignore
|
||||
source_app: bpy.props.StringProperty(name="Source") # type: ignore
|
||||
|
||||
|
||||
class speckle_object(bpy.types.PropertyGroup):
|
||||
"""
|
||||
PropertyGroup for storing object names and applicationIds
|
||||
"""
|
||||
|
||||
name: bpy.props.StringProperty() # type: ignore
|
||||
applicationId: bpy.props.StringProperty(name="Application ID", default="") # type: ignore
|
||||
|
||||
|
||||
class speckle_collection(bpy.types.PropertyGroup):
|
||||
"""
|
||||
PropertyGroup for storing collections
|
||||
"""
|
||||
|
||||
name: bpy.props.StringProperty() # type: ignore
|
||||
applicationId: bpy.props.StringProperty(name="Application ID", default="") # type: ignore
|
||||
|
||||
|
||||
class speckle_model_card(bpy.types.PropertyGroup):
|
||||
"""
|
||||
represents a Speckle model card in the Blender UI
|
||||
"""
|
||||
|
||||
account_id: bpy.props.StringProperty(
|
||||
name="Account ID", description="ID of the account", default=""
|
||||
) # type: ignore
|
||||
server_url: bpy.props.StringProperty(
|
||||
name="Server URL",
|
||||
description="URL of the Server",
|
||||
default="app.speckle.systems",
|
||||
) # type: ignore
|
||||
project_name: bpy.props.StringProperty(
|
||||
name="Project Name", description="Name of the project", default=""
|
||||
) # type: ignore
|
||||
project_id: bpy.props.StringProperty(
|
||||
name="Project ID", description="ID of the selected project", default=""
|
||||
) # type: ignore
|
||||
model_id: bpy.props.StringProperty(
|
||||
name="Model ID", description="ID of the model", default=""
|
||||
) # type: ignore
|
||||
model_name: bpy.props.StringProperty(
|
||||
name="Model Name", description="Name of the model", default=""
|
||||
) # type: ignore
|
||||
is_publish: bpy.props.BoolProperty(
|
||||
name="Publish/Load",
|
||||
description="If the model is published or loaded",
|
||||
default=False,
|
||||
) # type: ignore
|
||||
selection_summary: bpy.props.StringProperty(
|
||||
name="Selection Summary", description="Summary of the selection", default=""
|
||||
) # type: ignore
|
||||
version_id: bpy.props.StringProperty(
|
||||
name="Version ID", description="ID of the selected version", default=""
|
||||
) # type: ignore
|
||||
load_option: bpy.props.StringProperty(
|
||||
name="Load Option", description="Option of loading the model", default=""
|
||||
) # type: ignore
|
||||
objects: bpy.props.CollectionProperty(type=speckle_object) # type: ignore
|
||||
collections: bpy.props.CollectionProperty(type=speckle_collection) # type: ignore
|
||||
instance_loading_mode: bpy.props.StringProperty(
|
||||
name="Instance Loading Mode",
|
||||
description="Mode of loading instances",
|
||||
default="INSTANCE_PROXIES",
|
||||
) # type: ignore
|
||||
apply_modifiers: bpy.props.BoolProperty(
|
||||
name="Apply Modifiers",
|
||||
description="Apply modifiers to the objects",
|
||||
default=True,
|
||||
) # type: ignore
|
||||
|
||||
def get_model_card_id(self) -> str:
|
||||
if not self.project_id or not self.model_id:
|
||||
raise ValueError(
|
||||
"Project ID and Model ID are required to generate a model card ID."
|
||||
)
|
||||
if self.is_publish:
|
||||
return f"PUBLISH-{self.project_id}-{self.model_id}"
|
||||
else:
|
||||
return f"LOAD-{self.project_id}-{self.model_id}"
|
||||
@@ -0,0 +1,93 @@
|
||||
from specklepy.core.api.client import SpeckleClient
|
||||
from .account_manager import _client_cache
|
||||
from typing import List, Tuple
|
||||
from .misc import format_relative_time
|
||||
from specklepy.core.api.inputs.model_inputs import ModelVersionsFilter
|
||||
from specklepy.core.api.models.current import Version
|
||||
|
||||
|
||||
def get_versions_for_model(
|
||||
account_id: str, project_id: str, model_id: str
|
||||
) -> List[Tuple[str, str, str]]:
|
||||
"""
|
||||
fetches versions for a given model from the Speckle server
|
||||
"""
|
||||
try:
|
||||
# Validate inputs
|
||||
if not account_id or not project_id or not model_id:
|
||||
print(
|
||||
f"Error: Invalid inputs - account_id: {account_id}, project_id: {project_id}, model_id: {model_id}"
|
||||
)
|
||||
return []
|
||||
|
||||
# Get cached client
|
||||
client: SpeckleClient = _client_cache.get_client(account_id)
|
||||
if not client:
|
||||
print(f"Error: Could not get client for account: {account_id}")
|
||||
return []
|
||||
|
||||
filter: ModelVersionsFilter = ModelVersionsFilter(priorityIds=[])
|
||||
|
||||
# Get versions
|
||||
versions: List[Version] = client.version.get_versions(
|
||||
project_id=project_id, model_id=model_id, limit=10, filter=filter
|
||||
)
|
||||
versions_list: List[Tuple[str, str, str]] = []
|
||||
for version in versions.items:
|
||||
if version.referenced_object != "":
|
||||
versions_list.append(
|
||||
(
|
||||
version.id,
|
||||
version.message
|
||||
if version.message is not None
|
||||
else "No message",
|
||||
format_relative_time(version.created_at),
|
||||
)
|
||||
)
|
||||
return versions_list
|
||||
|
||||
except Exception as e:
|
||||
print(f"Error fetching versions: {str(e)}")
|
||||
# Clear cache on error to prevent stale clients
|
||||
_client_cache.clear()
|
||||
return []
|
||||
|
||||
|
||||
def get_latest_version(
|
||||
account_id: str, project_id: str, model_id: str
|
||||
) -> Tuple[str, str, str]:
|
||||
try:
|
||||
# Validate inputs
|
||||
if not account_id or not project_id or not model_id:
|
||||
print(
|
||||
f"Error: Invalid inputs - account_id: {account_id}, project_id: {project_id}, model_id: {model_id}"
|
||||
)
|
||||
return ("", "", "")
|
||||
|
||||
# Get cached client
|
||||
client: SpeckleClient = _client_cache.get_client(account_id)
|
||||
if not client:
|
||||
print(f"Error: Could not get client for account: {account_id}")
|
||||
return ("", "", "")
|
||||
|
||||
# Get versions (limit to 1 since we only need the latest)
|
||||
versions: List[Version] = client.version.get_versions(
|
||||
project_id=project_id, model_id=model_id, limit=1
|
||||
).items
|
||||
|
||||
if not versions:
|
||||
print(f"Error: No versions found for model_id: {model_id}")
|
||||
return ("", "", "")
|
||||
|
||||
latest = versions[0]
|
||||
return (
|
||||
latest.id,
|
||||
latest.message if latest.message is not None else "No message",
|
||||
format_relative_time(latest.created_at),
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
print(f"Error fetching latest version: {str(e)}")
|
||||
# Clear cache on error to prevent stale clients
|
||||
_client_cache.clear()
|
||||
return ("", "", "")
|
||||
@@ -1,29 +0,0 @@
|
||||
from typing import Union
|
||||
from bpy_speckle.convert.to_native import convert_to_native
|
||||
from specklepy.objects.base import Base
|
||||
|
||||
def get_speckle_subobjects(attr: Union[dict, Base], scale: float, name: str) -> list:
|
||||
subobjects = []
|
||||
keys = attr.keys() if isinstance(attr, dict) else attr.get_dynamic_member_names()
|
||||
for key in keys:
|
||||
if isinstance(attr[key], dict):
|
||||
subtype = attr[key].get("type", None)
|
||||
if subtype:
|
||||
name = f"{name}.{key}"
|
||||
subobject = convert_to_native(attr[key], name)
|
||||
|
||||
subobjects.append(subobject)
|
||||
props = attr[key].get("properties", None)
|
||||
if props:
|
||||
subobjects.extend(get_speckle_subobjects(props, scale, name))
|
||||
elif hasattr(attr[key], "type"):
|
||||
subtype = attr[key].type
|
||||
if subtype:
|
||||
name = "{}.{}".format(name, key)
|
||||
subobject = convert_to_native(attr[key], name)
|
||||
|
||||
subobjects.append(subobject)
|
||||
props = attr[key].get("properties", None)
|
||||
if props:
|
||||
subobjects.extend(get_speckle_subobjects(props, scale, name))
|
||||
return subobjects
|
||||
@@ -1,665 +0,0 @@
|
||||
import math
|
||||
from typing import Tuple, Union, Collection
|
||||
from bpy_speckle.functions import get_scale_length, _report
|
||||
from mathutils import (
|
||||
Matrix as MMatrix,
|
||||
Vector as MVector,
|
||||
Quaternion as MQuaternion,
|
||||
)
|
||||
import bpy, bmesh
|
||||
from specklepy.objects.other import (
|
||||
Instance,
|
||||
Transform,
|
||||
BlockDefinition,
|
||||
)
|
||||
from specklepy.objects.geometry import *
|
||||
from bpy.types import Object
|
||||
from .util import (
|
||||
get_render_material,
|
||||
link_object_to_collection_nested,
|
||||
render_material_to_native,
|
||||
add_custom_properties,
|
||||
add_vertices,
|
||||
add_faces,
|
||||
add_colors,
|
||||
add_uv_coords,
|
||||
)
|
||||
|
||||
SUPPORTED_CURVES = (Line, Polyline, Curve, Arc, Polycurve, Ellipse, Circle)
|
||||
|
||||
CAN_CONVERT_TO_NATIVE = (
|
||||
Mesh,
|
||||
*SUPPORTED_CURVES,
|
||||
Instance,
|
||||
)
|
||||
|
||||
|
||||
def _has_native_convesion(speckle_object: Base) -> bool:
|
||||
return any(isinstance(speckle_object, t) for t in CAN_CONVERT_TO_NATIVE)
|
||||
|
||||
def _has_fallback_conversion(speckle_object: Base) -> bool:
|
||||
return any(getattr(speckle_object, alias, None) for alias in DISPLAY_VALUE_PROPERTY_ALIASES)
|
||||
|
||||
def can_convert_to_native(speckle_object: Base) -> bool:
|
||||
|
||||
if(_has_native_convesion(speckle_object) or _has_fallback_conversion(speckle_object)):
|
||||
return True
|
||||
|
||||
_report(f"Could not convert unsupported Speckle object: {speckle_object}")
|
||||
return False
|
||||
|
||||
def create_new_object(obj_data: Optional[bpy.types.ID], desired_name: str, counter: int = 0) -> bpy.types.Object:
|
||||
"""
|
||||
Creates a new blender object with a unique name,
|
||||
if the desired_name is already taken
|
||||
we'll append a number, with the format .xxx to the desired_name to ensure the name is unique.
|
||||
"""
|
||||
name = desired_name if counter == 0 else f"{desired_name[:OBJECT_NAME_MAX_LENGTH - 4]}.{counter:03d}" # format counter as name.xxx, truncate to ensure we don't exceed the object name max length
|
||||
|
||||
#TODO: This is very slow, and gets slower the more objects you receive with the same name...
|
||||
# We could use a binary/galloping search, and/or cache the name -> index within a receive.
|
||||
if name in bpy.data.objects.keys():
|
||||
#Object already exists, increment counter and try again!
|
||||
return create_new_object(obj_data, desired_name, counter + 1)
|
||||
|
||||
blender_object = bpy.data.objects.new(name, obj_data)
|
||||
return blender_object
|
||||
|
||||
convert_instances_as: str #HACK: This is hacky, we need a better way to pass settings down to the converter
|
||||
def set_convert_instances_as(value: str):
|
||||
global convert_instances_as
|
||||
convert_instances_as = value
|
||||
|
||||
def convert_to_native(speckle_object: Base) -> list[Object]:
|
||||
|
||||
speckle_type = type(speckle_object)
|
||||
try:
|
||||
object_name = _generate_object_name(speckle_object)
|
||||
scale = get_scale_factor(speckle_object)
|
||||
|
||||
obj_data: Optional[Union[bpy.types.ID, bpy.types.Object]] = None
|
||||
converted: list[Object] = []
|
||||
|
||||
# convert elements/breps
|
||||
if not _has_native_convesion(speckle_object):
|
||||
(obj_data, converted) = element_to_native(speckle_object, object_name, scale)
|
||||
if not obj_data and not converted:
|
||||
_report(f"Unsupported type {speckle_object.speckle_type}")
|
||||
|
||||
# convert supported geometry
|
||||
elif isinstance(speckle_object, Mesh):
|
||||
obj_data = mesh_to_native(speckle_object, object_name, scale)
|
||||
elif speckle_type in SUPPORTED_CURVES:
|
||||
obj_data = icurve_to_native(speckle_object, object_name, scale)
|
||||
elif isinstance(speckle_object, Instance):
|
||||
if convert_instances_as == "linked_duplicates":
|
||||
(obj_data, converted) = instance_to_native_object(speckle_object, scale)
|
||||
elif convert_instances_as != "collection_instance":
|
||||
obj_data = instance_to_native_collection_instance(speckle_object, scale)
|
||||
else:
|
||||
_report(f"convert_instances_as = '{convert_instances_as}' is not implemented, Instances will be converted as collection instances!")
|
||||
obj_data = instance_to_native_collection_instance(speckle_object, scale)
|
||||
|
||||
else:
|
||||
_report(f"Unsupported type {speckle_type}")
|
||||
return []
|
||||
except Exception as ex: # conversion error
|
||||
_report(f"Error converting {speckle_object} \n{ex}")
|
||||
return []
|
||||
|
||||
|
||||
blender_object = obj_data if isinstance(obj_data, Object) else create_new_object(obj_data, object_name)
|
||||
|
||||
blender_object.speckle.object_id = str(speckle_object.id)
|
||||
blender_object.speckle.enabled = True
|
||||
add_custom_properties(speckle_object, blender_object)
|
||||
|
||||
for child in converted:
|
||||
child.parent = blender_object
|
||||
|
||||
converted.append(blender_object)
|
||||
_report(f"Successfully converted {object_name} as {blender_object.type}")
|
||||
return converted
|
||||
|
||||
|
||||
DISPLAY_VALUE_PROPERTY_ALIASES = ["displayValue", "@displayValue", "displayMesh", "@displayMesh", "elements", "@elements"]
|
||||
|
||||
def element_to_native(speckle_object: Base, name: str, scale: float, combineMeshes: bool = True) -> tuple[Optional[bpy.types.Mesh], list[bpy.types.Object]]:
|
||||
"""
|
||||
Converts a given speckle_object by converting displayValue properties (elements treated the same as displayValues)
|
||||
|
||||
if combineMeshes == True
|
||||
Converts mesh displayValues as one mesh
|
||||
Converts non-mesh displayValues as child Objects
|
||||
if combineMeshes == False
|
||||
Converts all displayValues as child objects (first item of the returned tuple will be None)
|
||||
"""
|
||||
meshes: list[Mesh] = []
|
||||
elements: list[Base] = []
|
||||
|
||||
#NOTE: raw Mesh elements will be treated like displayValues, which is not ideal, but no connector sends raw Mesh elements so it's fine
|
||||
for alias in DISPLAY_VALUE_PROPERTY_ALIASES:
|
||||
display = getattr(speckle_object, alias, None)
|
||||
|
||||
count = 0
|
||||
MAX_DEPTH = 255 # some large value, to prevent infinite reccursion
|
||||
def seperate(value: Any) -> bool:
|
||||
nonlocal meshes, elements, count, MAX_DEPTH
|
||||
|
||||
if combineMeshes and isinstance(value, Mesh):
|
||||
meshes.append(value)
|
||||
elif isinstance(value, Base):
|
||||
elements.append(value)
|
||||
elif isinstance(value, list):
|
||||
count += 1
|
||||
if(count > MAX_DEPTH):
|
||||
return True
|
||||
for x in value:
|
||||
seperate(x)
|
||||
|
||||
return False
|
||||
|
||||
did_halt = seperate(display)
|
||||
|
||||
if did_halt:
|
||||
_report(f"Traversal of {speckle_object.speckle_type} {speckle_object.id} halted after traversal depth exceeds MAX_DEPTH={MAX_DEPTH}. Are there circular references object structure?")
|
||||
|
||||
|
||||
converted: list[Object] = []
|
||||
mesh = None
|
||||
|
||||
if meshes:
|
||||
mesh = meshes_to_native(speckle_object, meshes, name, scale)
|
||||
|
||||
for item in elements:
|
||||
# add parent type here so we can use it as a blender custom prop
|
||||
# not making it hidden, so it will get added on send as i think it might be helpful? can reconsider
|
||||
item.parent_speckle_type = speckle_object.speckle_type #TODO: consider if this is still useful, as we now properly structure object parenting
|
||||
blender_object = convert_to_native(item)
|
||||
if isinstance(blender_object, list):
|
||||
converted.extend(blender_object)
|
||||
else:
|
||||
add_custom_properties(speckle_object, blender_object)
|
||||
converted.append(blender_object)
|
||||
|
||||
return (mesh, converted)
|
||||
|
||||
|
||||
def mesh_to_native(speckle_mesh: Mesh, name: str, scale: float) -> bpy.types.Mesh:
|
||||
return meshes_to_native(speckle_mesh, [speckle_mesh], name, scale)
|
||||
|
||||
def meshes_to_native(element: Base, meshes: Collection[Mesh], name: str, scale: float) -> bpy.types.Mesh:
|
||||
if name in bpy.data.meshes.keys():
|
||||
return bpy.data.meshes[name]
|
||||
|
||||
blender_mesh = bpy.data.meshes.new(name=name)
|
||||
|
||||
fallback_material = get_render_material(element)
|
||||
|
||||
bm = bmesh.new()
|
||||
|
||||
# First pass, add vertex data
|
||||
for mesh in meshes:
|
||||
scale = get_scale_factor(mesh, scale)
|
||||
add_vertices(mesh, bm, scale)
|
||||
|
||||
bm.verts.ensure_lookup_table()
|
||||
|
||||
# Second pass, add face data
|
||||
offset = 0
|
||||
for i, mesh in enumerate(meshes):
|
||||
add_faces(mesh, bm, offset, i)
|
||||
|
||||
render_material = get_render_material(mesh) or fallback_material
|
||||
if render_material is not None:
|
||||
native_material = render_material_to_native(render_material)
|
||||
blender_mesh.materials.append(native_material)
|
||||
|
||||
offset += len(mesh.vertices) // 3
|
||||
|
||||
bm.faces.ensure_lookup_table()
|
||||
bm.verts.index_update()
|
||||
|
||||
# Third pass, add vertex instance data
|
||||
for mesh in meshes:
|
||||
add_colors(mesh, bm)
|
||||
add_uv_coords(mesh, bm)
|
||||
|
||||
bmesh.ops.recalc_face_normals(bm, faces=bm.faces)
|
||||
|
||||
bm.to_mesh(blender_mesh)
|
||||
bm.free()
|
||||
|
||||
return blender_mesh
|
||||
|
||||
|
||||
"""
|
||||
Curves
|
||||
"""
|
||||
|
||||
def line_to_native(speckle_curve: Line, blender_curve: bpy.types.Curve, scale: float) -> list[bpy.types.Spline]:
|
||||
if not speckle_curve.end: return []
|
||||
|
||||
line = blender_curve.splines.new("POLY")
|
||||
line.points.add(1)
|
||||
|
||||
line.points[0].co = (
|
||||
float(speckle_curve.start.x) * scale,
|
||||
float(speckle_curve.start.y) * scale,
|
||||
float(speckle_curve.start.z) * scale,
|
||||
1,
|
||||
)
|
||||
|
||||
line.points[1].co = (
|
||||
float(speckle_curve.end.x) * scale,
|
||||
float(speckle_curve.end.y) * scale,
|
||||
float(speckle_curve.end.z) * scale,
|
||||
1,
|
||||
)
|
||||
|
||||
return [line]
|
||||
|
||||
|
||||
def polyline_to_native(scurve: Polyline, bcurve: bpy.types.Curve, scale: float) -> list[bpy.types.Spline]:
|
||||
if not (value := scurve.value): return []
|
||||
N = len(value) // 3
|
||||
|
||||
polyline = bcurve.splines.new("POLY")
|
||||
|
||||
if hasattr(scurve, "closed"):
|
||||
polyline.use_cyclic_u = scurve.closed
|
||||
|
||||
polyline.points.add(N - 1)
|
||||
for i in range(N):
|
||||
polyline.points[i].co = (
|
||||
float(value[i * 3]) * scale,
|
||||
float(value[i * 3 + 1]) * scale,
|
||||
float(value[i * 3 + 2]) * scale,
|
||||
1,
|
||||
)
|
||||
|
||||
return [polyline]
|
||||
|
||||
|
||||
|
||||
def nurbs_to_native(scurve: Curve, bcurve: bpy.types.Curve, scale: float) -> list[bpy.types.Spline]:
|
||||
if not (points := scurve.points): return []
|
||||
|
||||
# Closed curves from rhino will have n + degree points. We ignore the extras
|
||||
num_points = len(points) // 3 - scurve.degree if (scurve.closed) else (
|
||||
len(points) // 3)
|
||||
|
||||
nurbs = bcurve.splines.new("NURBS")
|
||||
nurbs.use_cyclic_u = scurve.closed
|
||||
nurbs.use_endpoint_u = not scurve.periodic
|
||||
|
||||
nurbs.points.add(num_points - 1)
|
||||
use_weights = len(scurve.weights) >= num_points
|
||||
for i in range(num_points):
|
||||
nurbs.points[i].co = (
|
||||
float(points[i * 3]) * scale,
|
||||
float(points[i * 3 + 1]) * scale,
|
||||
float(points[i * 3 + 2]) * scale,
|
||||
1,
|
||||
)
|
||||
|
||||
nurbs.points[i].weight = scurve.weights[i] if use_weights else 1
|
||||
|
||||
nurbs.order_u = scurve.degree + 1
|
||||
|
||||
return [nurbs]
|
||||
|
||||
|
||||
def arc_to_native(rcurve: Arc, bcurve: bpy.types.Curve, scale: float) -> Optional[bpy.types.Spline]:
|
||||
# TODO: improve Blender representation of arc - check autocad test stream
|
||||
|
||||
plane = rcurve.plane
|
||||
if not plane:
|
||||
return None
|
||||
|
||||
normal = MVector([plane.normal.x, plane.normal.y, plane.normal.z])
|
||||
|
||||
radius = rcurve.radius * scale
|
||||
startAngle = rcurve.startAngle
|
||||
endAngle = rcurve.endAngle
|
||||
|
||||
startQuat = MQuaternion(normal, startAngle)
|
||||
endQuat = MQuaternion(normal, endAngle)
|
||||
|
||||
# Get start and end vectors, centre point, angles, etc.
|
||||
r1 = MVector([plane.xdir.x, plane.xdir.y, plane.xdir.z])
|
||||
r1.rotate(startQuat)
|
||||
|
||||
r2 = MVector([plane.xdir.x, plane.xdir.y, plane.xdir.z])
|
||||
r2.rotate(endQuat)
|
||||
|
||||
c = MVector([plane.origin.x, plane.origin.y, plane.origin.z]) * scale
|
||||
|
||||
spt = c + r1 * radius
|
||||
ept = c + r2 * radius
|
||||
|
||||
angle = endAngle - startAngle
|
||||
|
||||
t1 = normal.cross(r1)
|
||||
|
||||
# Initialize arc data and calculate subdivisions
|
||||
arc = bcurve.splines.new("NURBS")
|
||||
|
||||
arc.use_cyclic_u = False
|
||||
|
||||
Ndiv = max(int(math.floor(angle / 0.3)), 2)
|
||||
step = angle / float(Ndiv)
|
||||
stepQuat = MQuaternion(normal, step)
|
||||
tan = math.tan(step / 2) * radius
|
||||
|
||||
arc.points.add(Ndiv + 1)
|
||||
|
||||
# Set start and end points
|
||||
arc.points[0].co = (spt.x, spt.y, spt.z, 1)
|
||||
arc.points[Ndiv + 1].co = (ept.x, ept.y, ept.z, 1)
|
||||
|
||||
# Set intermediate points
|
||||
for i in range(Ndiv):
|
||||
t1 = normal.cross(r1)
|
||||
pt = c + r1 * radius + t1 * tan
|
||||
arc.points[i + 1].co = (pt.x, pt.y, pt.z, 1)
|
||||
r1.rotate(stepQuat)
|
||||
|
||||
# Set curve settings
|
||||
arc.use_endpoint_u = True
|
||||
arc.order_u = 3
|
||||
|
||||
return arc
|
||||
|
||||
|
||||
def polycurve_to_native(scurve: Polycurve, bcurve: bpy.types.Curve, scale: float) -> list[bpy.types.Spline]:
|
||||
"""
|
||||
Convert Polycurve object
|
||||
"""
|
||||
segments = scurve.segments
|
||||
|
||||
curves = []
|
||||
|
||||
for seg in segments:
|
||||
speckle_type = type(seg)
|
||||
|
||||
if speckle_type in SUPPORTED_CURVES:
|
||||
curves.append(icurve_to_native_spline(seg, bcurve, scale))
|
||||
else:
|
||||
_report(f"Unsupported curve type: {speckle_type}")
|
||||
|
||||
return curves
|
||||
|
||||
def circle_to_native(circle: Circle, bcurve: bpy.types.Curve, units_scale: float) -> list[bpy.types.Spline]:
|
||||
#HACK: violates typing, but it works...
|
||||
circle["firstRadius"] = circle.radius
|
||||
circle["secondRadius"] = circle.radius
|
||||
return ellipse_to_native(circle, bcurve, units_scale)
|
||||
|
||||
def ellipse_to_native(ellipse: Ellipse, bcurve: bpy.types.Curve, units_scale: float) -> list[bpy.types.Spline]:
|
||||
plane = ellipse.plane
|
||||
|
||||
radX = ellipse.firstRadius * units_scale
|
||||
radY = ellipse.secondRadius * units_scale
|
||||
|
||||
D = 0.5522847498307936 # (4/3)*tan(pi/8)
|
||||
|
||||
right_handles = [
|
||||
(+radX, +radY * D, 0.0),
|
||||
(-radX * D, +radY, 0.0),
|
||||
(-radX, -radY * D, 0.0),
|
||||
(+radX * D, -radY, 0.0),
|
||||
]
|
||||
|
||||
left_handles = [
|
||||
(+radX, -radY * D, 0.0),
|
||||
(+radX * D, +radY, 0.0),
|
||||
(-radX, +radY * D, 0.0),
|
||||
(-radX * D, -radY, 0.0),
|
||||
]
|
||||
|
||||
points = [
|
||||
(+radX, 0.0, 0.0),
|
||||
(0.0, +radY, 0.0),
|
||||
(-radX, 0.0, 0.0),
|
||||
(0.0, -radY, 0.0),
|
||||
]
|
||||
transform = plane_to_native_transform(plane, units_scale)
|
||||
|
||||
spline = bcurve.splines.new("BEZIER")
|
||||
spline.bezier_points.add(len(points) - 1)
|
||||
|
||||
for i in range(len(points)):
|
||||
spline.bezier_points[i].co = transform @ MVector(points[i])
|
||||
spline.bezier_points[i].handle_left = transform @ MVector(left_handles[i])
|
||||
spline.bezier_points[i].handle_right = transform @ MVector(right_handles[i])
|
||||
|
||||
spline.use_cyclic_u = True
|
||||
|
||||
#TODO support trims?
|
||||
return [spline]
|
||||
|
||||
|
||||
def icurve_to_native_spline(speckle_curve: Base, blender_curve: bpy.types.Curve, scale: float) -> list[bpy.types.Spline]:
|
||||
# polycurves
|
||||
if isinstance(speckle_curve, Polycurve):
|
||||
return polycurve_to_native(speckle_curve, blender_curve, scale)
|
||||
|
||||
# single curves
|
||||
if isinstance(speckle_curve, Line):
|
||||
spline = line_to_native(speckle_curve, blender_curve, scale)
|
||||
elif isinstance(speckle_curve, Curve):
|
||||
spline = nurbs_to_native(speckle_curve, blender_curve, scale)
|
||||
elif isinstance(speckle_curve, Polyline):
|
||||
spline = polyline_to_native(speckle_curve, blender_curve, scale)
|
||||
elif isinstance(speckle_curve, Arc):
|
||||
spline = arc_to_native(speckle_curve, blender_curve, scale)
|
||||
elif isinstance(speckle_curve, Ellipse):
|
||||
spline = ellipse_to_native(speckle_curve, blender_curve, scale)
|
||||
elif isinstance(speckle_curve, Circle):
|
||||
spline = circle_to_native(speckle_curve, blender_curve, scale)
|
||||
else:
|
||||
raise TypeError(f"{speckle_curve} is not a supported curve type. Supported types: {SUPPORTED_CURVES}")
|
||||
|
||||
return [spline] if spline is not None else []
|
||||
|
||||
|
||||
def icurve_to_native(speckle_curve: Base, name: str, scale: float) -> Optional[bpy.types.Curve]:
|
||||
curve_type = type(speckle_curve)
|
||||
if curve_type not in SUPPORTED_CURVES:
|
||||
_report(f"Unsupported curve type: {curve_type}")
|
||||
return None
|
||||
blender_curve = (
|
||||
bpy.data.curves[name]
|
||||
if name in bpy.data.curves.keys()
|
||||
else bpy.data.curves.new(name, type="CURVE")
|
||||
)
|
||||
blender_curve.dimensions = "3D"
|
||||
blender_curve.resolution_u = 12 #TODO: We could maybe decern the resolution from the ployline displayValue
|
||||
|
||||
icurve_to_native_spline(speckle_curve, blender_curve, scale)
|
||||
|
||||
return blender_curve
|
||||
|
||||
|
||||
"""
|
||||
Transforms and Intances
|
||||
"""
|
||||
|
||||
def transform_to_native(transform: Transform, scale: float) -> MMatrix:
|
||||
mat = MMatrix(
|
||||
[
|
||||
transform.value[:4],
|
||||
transform.value[4:8],
|
||||
transform.value[8:12],
|
||||
transform.value[12:16],
|
||||
]
|
||||
)
|
||||
# scale the translation
|
||||
for i in range(3):
|
||||
mat[i][3] *= scale
|
||||
return mat
|
||||
|
||||
def plane_to_native_transform(plane: Plane, fallback_scale:float = 1) -> MMatrix:
|
||||
scale_factor = get_scale_factor(plane, fallback_scale)
|
||||
tx = (plane.origin.x * scale_factor)
|
||||
ty = (plane.origin.y * scale_factor)
|
||||
tz = (plane.origin.z * scale_factor)
|
||||
|
||||
|
||||
return MMatrix((
|
||||
(plane.xdir.x, plane.ydir.x, plane.normal.x, tx),
|
||||
(plane.xdir.y, plane.ydir.y, plane.normal.y, ty),
|
||||
(plane.xdir.z, plane.ydir.z, plane.normal.z, tz),
|
||||
(0, 0, 0, 1 )
|
||||
))
|
||||
|
||||
|
||||
"""
|
||||
Instances / Blocks
|
||||
"""
|
||||
|
||||
def _get_instance_name(instance: Instance) -> str:
|
||||
name_prefix = _get_friendly_object_name(instance) or _get_friendly_object_name(instance.definition) or _simplified_speckle_type(instance.speckle_type)
|
||||
return f"{name_prefix}{OBJECT_NAME_SEPERATOR}{instance.id}"
|
||||
|
||||
|
||||
def instance_to_native_object(instance: Instance, scale: float) -> Tuple[bpy.types.Object, List[bpy.types.Object]]:
|
||||
"""
|
||||
Converts Instance to a unique object with (potentially) shared data (linked duplicate)
|
||||
"""
|
||||
if not instance.definition: raise Exception(f"Instance is missing a definition")
|
||||
if not instance.transform: raise Exception(f"Instance is missing a transform")
|
||||
|
||||
name = _get_instance_name(instance)
|
||||
definition = instance.definition
|
||||
|
||||
native_instance: Object
|
||||
native_elements: List[Object] = []
|
||||
elements_on_instance: List[Object] = []
|
||||
|
||||
if isinstance(definition, BlockDefinition): #NOTE: We have to handle BlockDefinitions specially here, since they don't follow normal traversal rules
|
||||
native_instance = create_new_object(None, name) #Instance will be empty
|
||||
native_instance.empty_display_size = 0
|
||||
for geo in definition.geometry:
|
||||
native_elements.append(convert_to_native(geo)[-1])
|
||||
else:
|
||||
native_instance = convert_to_native(instance.definition)[-1] # Convert assuming that definition is convertable
|
||||
|
||||
instance_transform = transform_to_native(instance.transform, scale)
|
||||
instance_transform_inverted = instance_transform.inverted()
|
||||
native_instance.matrix_world = instance_transform
|
||||
|
||||
(_, elements_on_instance) = element_to_native(instance, name, scale)
|
||||
for c in elements_on_instance:
|
||||
c.matrix_world = instance_transform_inverted @ c.matrix_world #Undo the instance transform on elements
|
||||
|
||||
native_elements.extend(elements_on_instance)
|
||||
|
||||
return (native_instance, native_elements) #TODO: need to double check that all child objects have custom props attached correctly
|
||||
|
||||
def instance_to_native_collection_instance(instance: Instance, scale: float) -> bpy.types.Object:
|
||||
"""
|
||||
Convert an Instance as a transformed Object with the `instance_collection` property
|
||||
set to be the `instance.Definition` converted as a collection
|
||||
|
||||
The definition collection won't be linked to the current scene
|
||||
Any Elements on the instance object will also be converted (and spacially transformed)
|
||||
"""
|
||||
if not instance.definition: raise Exception(f"Instance is missing a definition")
|
||||
if not instance.transform: raise Exception(f"Instance is missing a transform")
|
||||
|
||||
name = _get_instance_name(instance)
|
||||
|
||||
# Get/Convert definition collection
|
||||
collection_def = _instance_definition_to_native(instance.definition)
|
||||
|
||||
# Convert elements as children of collection instance object
|
||||
(_, elements) = element_to_native(instance, name, scale, False)
|
||||
|
||||
instance_transform = transform_to_native(instance.transform, scale)
|
||||
instance_transform_inverted = instance_transform.inverted()
|
||||
|
||||
native_instance = bpy.data.objects.new(name, None)
|
||||
|
||||
#add_custom_properties(instance, native_instance)
|
||||
# hide the instance axes so they don't clutter the viewport
|
||||
native_instance.empty_display_size = 0
|
||||
native_instance.instance_collection = collection_def
|
||||
native_instance.instance_type = "COLLECTION"
|
||||
native_instance.matrix_world =instance_transform
|
||||
|
||||
for c in elements:
|
||||
c.matrix_world = instance_transform_inverted @ c.matrix_world #Undo the instance transform on elements
|
||||
c.parent = native_instance #TODO: need to double check that all child objects have custom props attached correctly
|
||||
|
||||
return native_instance
|
||||
|
||||
def _instance_definition_to_native(definition: Union[Base, BlockDefinition]) -> bpy.types.Collection:
|
||||
"""
|
||||
Converts a geometry carrying Base as a collection (does not link it to the scene)
|
||||
"""
|
||||
name = _generate_object_name(definition)
|
||||
native_def = bpy.data.collections.get(name)
|
||||
if native_def:
|
||||
return native_def
|
||||
|
||||
native_def = bpy.data.collections.new(name)
|
||||
native_def["applicationId"] = definition.applicationId
|
||||
|
||||
#TODO could maybe replace BlockDefinition awareness with a single traverse member call
|
||||
geometry = definition.geometry if isinstance(definition, BlockDefinition) else [definition]
|
||||
|
||||
for geo in geometry:
|
||||
if not geo: continue
|
||||
converted = convert_to_native(geo)[-1] #NOTE: we assume the last item is the root converted item
|
||||
link_object_to_collection_nested(converted, native_def)
|
||||
|
||||
|
||||
return native_def
|
||||
|
||||
|
||||
"""
|
||||
Object Naming
|
||||
"""
|
||||
|
||||
def _get_friendly_object_name(speckle_object: Base) -> Optional[str]:
|
||||
return (getattr(speckle_object, "name", None)
|
||||
or getattr(speckle_object, "Name", None)
|
||||
or getattr(speckle_object, "family", None)
|
||||
)
|
||||
|
||||
|
||||
# Blender object names must not exceed 62 characters
|
||||
# We need to ensure the complete ID is included in the name (to prevent identity collisions)
|
||||
# So we if the name is too long, we need to truncate
|
||||
OBJECT_NAME_MAX_LENGTH = 62
|
||||
SPECKLE_ID_LENGTH = 32
|
||||
OBJECT_NAME_SEPERATOR = " -- "
|
||||
|
||||
def _truncate_object_name(name: str) -> str:
|
||||
|
||||
MAX_NAME_LENGTH = OBJECT_NAME_MAX_LENGTH - SPECKLE_ID_LENGTH - len(OBJECT_NAME_SEPERATOR)
|
||||
|
||||
return name[:MAX_NAME_LENGTH]
|
||||
|
||||
|
||||
def _simplified_speckle_type(speckle_type: str) -> str:
|
||||
return(speckle_type.rsplit('.')[-1]) #Take only the most specific object type name (without namespace)
|
||||
|
||||
def _generate_object_name(speckle_object: Base) -> str:
|
||||
prefix: str
|
||||
name = _get_friendly_object_name(speckle_object)
|
||||
if name:
|
||||
prefix = _truncate_object_name(name)
|
||||
else:
|
||||
prefix = _simplified_speckle_type(speckle_object.speckle_type)
|
||||
|
||||
return f"{prefix}{OBJECT_NAME_SEPERATOR}{speckle_object.id}"
|
||||
|
||||
|
||||
def get_scale_factor(speckle_object: Base, fallback: float = 1.0) -> float:
|
||||
scale = fallback
|
||||
if units := getattr(speckle_object, "units", None):
|
||||
scale = get_scale_length(units) / bpy.context.scene.unit_settings.scale_length
|
||||
return scale
|
||||
@@ -1,448 +0,0 @@
|
||||
from typing import Dict, Iterable, Optional, Tuple
|
||||
import bpy
|
||||
from bpy.types import Depsgraph, MeshPolygon, Object
|
||||
from deprecated import deprecated
|
||||
from mathutils.geometry import interpolate_bezier
|
||||
from mathutils import (
|
||||
Matrix as MMatrix,
|
||||
Vector as MVector,
|
||||
)
|
||||
from specklepy.objects.geometry import (
|
||||
Mesh, Curve, Interval, Box, Point, Polyline
|
||||
)
|
||||
from specklepy.objects.other import *
|
||||
from bpy_speckle.functions import _report
|
||||
from bpy_speckle.convert.util import (
|
||||
get_blender_custom_properties,
|
||||
make_knots,
|
||||
nurb_make_curve,
|
||||
to_argb_int,
|
||||
)
|
||||
|
||||
UNITS = "m"
|
||||
|
||||
CAN_CONVERT_TO_SPECKLE = ("MESH", "CURVE", "EMPTY")
|
||||
|
||||
|
||||
def convert_to_speckle(raw_blender_object: Object, scale: float, units: str, depsgraph: Optional[Depsgraph]) -> Optional[list]:
|
||||
global UNITS
|
||||
UNITS = units
|
||||
|
||||
blender_type = raw_blender_object.type
|
||||
if blender_type not in CAN_CONVERT_TO_SPECKLE:
|
||||
return None
|
||||
|
||||
blender_object: Object = (
|
||||
raw_blender_object.evaluated_get(depsgraph)
|
||||
if depsgraph
|
||||
else raw_blender_object
|
||||
)
|
||||
|
||||
converted = None
|
||||
if blender_type == "MESH":
|
||||
converted = mesh_to_speckle(blender_object, blender_object.data, scale)
|
||||
elif blender_type == "CURVE":
|
||||
converted = icurve_to_speckle(blender_object, blender_object.data, scale)
|
||||
elif blender_type == "EMPTY":
|
||||
converted = empty_to_speckle(blender_object, scale)
|
||||
if not converted:
|
||||
return None
|
||||
|
||||
speckle_objects = []
|
||||
if isinstance(converted, list):
|
||||
speckle_objects.extend([c for c in converted if c != None])
|
||||
else:
|
||||
speckle_objects.append(converted)
|
||||
|
||||
for so in speckle_objects:
|
||||
so["properties"] = get_blender_custom_properties(raw_blender_object) #NOTE: Depsgraph copies don't have custom properties so we use the raw version
|
||||
so["applicationId"] = so.properties.pop("applicationId", None)
|
||||
|
||||
|
||||
# Set object transform
|
||||
if blender_type != "EMPTY": #TODO: this could be deprecated once we add proper instancing support
|
||||
so["properties"]["transform"] = transform_to_speckle(
|
||||
blender_object.matrix_world
|
||||
)
|
||||
|
||||
return speckle_objects
|
||||
|
||||
def mesh_to_speckle(blender_object: Object, data: bpy.types.Mesh, scale: float = 1.0) -> List[Mesh]:
|
||||
#if data.loop_triangles is None or len(data.loop_triangles) < 1:
|
||||
# data.calc_loop_triangles()
|
||||
|
||||
# Categorise polygons by material index
|
||||
submesh_data: Dict[int, List[MeshPolygon]] = {}
|
||||
|
||||
for p in data.polygons:
|
||||
if p.material_index not in submesh_data:
|
||||
submesh_data[p.material_index] = []
|
||||
submesh_data[p.material_index].append(p)
|
||||
|
||||
transform = blender_object.matrix_world
|
||||
scaled_vertices = [tuple(transform @ x.co * scale) for x in data.vertices]
|
||||
|
||||
# Create Speckle meshes for each material
|
||||
submeshes = []
|
||||
index_counter = 0
|
||||
for i in submesh_data:
|
||||
index_mapping: Dict[int, int] = {}
|
||||
|
||||
#Loop through each polygon, and map indicies to their new index in m_verts
|
||||
|
||||
mesh_area = 0
|
||||
m_verts: List[float] = []
|
||||
m_faces: List[int] = []
|
||||
m_texcoords: List[float] = []
|
||||
for face in submesh_data[i]:
|
||||
u_indices = face.vertices
|
||||
m_faces.append(len(u_indices))
|
||||
|
||||
mesh_area += face.area
|
||||
for u_index in u_indices:
|
||||
if u_index not in index_mapping:
|
||||
# Create mapping between index in blender mesh, and new index in speckle submesh
|
||||
index_mapping[u_index] = len(m_verts) // 3
|
||||
vert = scaled_vertices[u_index]
|
||||
m_verts.append(vert[0])
|
||||
m_verts.append(vert[1])
|
||||
m_verts.append(vert[2])
|
||||
|
||||
if data.uv_layers.active:
|
||||
vt = data.uv_layers.active.data[index_counter]
|
||||
m_texcoords.extend([vt.uv.x, vt.uv.y])
|
||||
|
||||
m_faces.append(index_mapping[u_index])
|
||||
index_counter += 1
|
||||
|
||||
speckle_mesh = Mesh(
|
||||
vertices=m_verts,
|
||||
faces=m_faces,
|
||||
colors=[],
|
||||
textureCoordinates=m_texcoords,
|
||||
units=UNITS,
|
||||
area = mesh_area,
|
||||
bbox=Box(area=0.0, volume=0.0),
|
||||
)
|
||||
|
||||
if i < len(data.materials):
|
||||
material = data.materials[i]
|
||||
if material is not None:
|
||||
speckle_mesh["renderMaterial"] = material_to_speckle(material)
|
||||
submeshes.append(speckle_mesh)
|
||||
|
||||
return submeshes
|
||||
|
||||
|
||||
def bezier_to_speckle(matrix: MMatrix, spline: bpy.types.Spline, scale: float, name: Optional[str] = None) -> Curve:
|
||||
degree = 3
|
||||
closed = spline.use_cyclic_u
|
||||
|
||||
points = []
|
||||
for i, bp in enumerate(spline.bezier_points):
|
||||
if i > 0:
|
||||
points.append(tuple(matrix @ bp.handle_left * scale))
|
||||
points.append(tuple(matrix @ bp.co * scale))
|
||||
if i < len(spline.bezier_points) - 1:
|
||||
points.append(tuple(matrix @ bp.handle_right * scale))
|
||||
|
||||
if closed:
|
||||
points.extend(
|
||||
(
|
||||
tuple(matrix @ spline.bezier_points[-1].handle_right * scale),
|
||||
tuple(matrix @ spline.bezier_points[0].handle_left * scale),
|
||||
tuple(matrix @ spline.bezier_points[0].co * scale),
|
||||
)
|
||||
)
|
||||
|
||||
num_points = len(points)
|
||||
|
||||
flattend_points = []
|
||||
for row in points: flattend_points.extend(row)
|
||||
|
||||
knot_count = num_points + degree - 1
|
||||
knots = [0] * knot_count
|
||||
|
||||
for i in range(1, len(knots)):
|
||||
knots[i] = i // 3
|
||||
|
||||
length = spline.calc_length()
|
||||
domain = Interval(start=0, end=length, totalChildrenCount=0)
|
||||
return Curve(
|
||||
name=name,
|
||||
degree=degree,
|
||||
closed=spline.use_cyclic_u,
|
||||
periodic= not spline.use_endpoint_u,
|
||||
points=flattend_points,
|
||||
weights=[1] * num_points,
|
||||
knots=knots,
|
||||
rational=True,
|
||||
area=0,
|
||||
volume=0,
|
||||
length=length,
|
||||
domain=domain,
|
||||
units=UNITS,
|
||||
bbox=Box(area=0.0, volume=0.0),
|
||||
displayValue = bezier_to_speckle_polyline(matrix, spline, scale, length),
|
||||
)
|
||||
|
||||
|
||||
def nurbs_to_speckle(matrix: MMatrix, spline: bpy.types.Spline, scale: float, name: Optional[str] = None) -> Curve:
|
||||
|
||||
degree = spline.order_u - 1
|
||||
knots = make_knots(spline)
|
||||
|
||||
length = spline.calc_length()
|
||||
domain = Interval(start=0, end=length, totalChildrenCount=0)
|
||||
|
||||
weights = [pt.weight for pt in spline.points]
|
||||
is_rational = all(w == weights[0] for w in weights)
|
||||
|
||||
points = [tuple(matrix @ pt.co.xyz * scale) for pt in spline.points]
|
||||
|
||||
flattend_points = []
|
||||
for row in points: flattend_points.extend(row)
|
||||
|
||||
if spline.use_cyclic_u:
|
||||
for i in range(0, degree * 3, 3):
|
||||
# Rhino expects n + degree number of points (for closed curves). So we need to add an extra point for each degree
|
||||
flattend_points.append(flattend_points[i + 0])
|
||||
flattend_points.append(flattend_points[i + 1])
|
||||
flattend_points.append(flattend_points[i + 2])
|
||||
|
||||
for i in range(0, degree):
|
||||
weights.append(weights[i])
|
||||
|
||||
return Curve(
|
||||
name=name,
|
||||
degree=degree,
|
||||
closed=spline.use_cyclic_u,
|
||||
periodic= not spline.use_endpoint_u,
|
||||
points=flattend_points,
|
||||
weights=weights,
|
||||
knots=knots,
|
||||
rational=is_rational,
|
||||
area=0,
|
||||
volume=0,
|
||||
length=length,
|
||||
domain=domain,
|
||||
units=UNITS,
|
||||
bbox=Box(area=0.0, volume=0.0),
|
||||
displayValue=nurbs_to_speckle_polyline(matrix, spline, scale, length),
|
||||
)
|
||||
|
||||
def nurbs_to_speckle_polyline(matrix: MMatrix, spline: bpy.types.Spline, scale: float, length: Optional[float] = None) -> Polyline:
|
||||
"""
|
||||
Samples a nurbs curve with resolution_u creating a polyline
|
||||
"""
|
||||
points = []
|
||||
sampled_points = nurb_make_curve(spline, spline.resolution_u, 3)
|
||||
for i in range(0, len(sampled_points), 3):
|
||||
scaled_point = matrix @ MVector((
|
||||
sampled_points[i + 0],
|
||||
sampled_points[i + 1],
|
||||
sampled_points[i + 2])) * scale
|
||||
|
||||
points.append(scaled_point.x)
|
||||
points.append(scaled_point.y)
|
||||
points.append(scaled_point.z)
|
||||
|
||||
length = length or spline.calc_length()
|
||||
domain = Interval(start=0, end=length, totalChildrenCount=0)
|
||||
return Polyline(value=points, closed = spline.use_cyclic_u, domain=domain, area=0, len=length)
|
||||
|
||||
|
||||
#Inspired by https://blender.stackexchange.com/a/689 (CC BY-SA 3.0)
|
||||
def bezier_to_speckle_polyline(matrix: MMatrix, spline: bpy.types.Spline, scale: float, length: Optional[float] = None) -> Optional[Polyline]:
|
||||
"""
|
||||
Samples a Bézier curve with resolution_u creating a polyline
|
||||
"""
|
||||
segments = len(spline.bezier_points)
|
||||
if segments < 2: return None
|
||||
|
||||
R = spline.resolution_u + 1
|
||||
|
||||
points = []
|
||||
if not spline.use_cyclic_u:
|
||||
segments -= 1
|
||||
|
||||
points: List[float] = []
|
||||
for i in range(segments):
|
||||
inext = (i + 1) % len(spline.bezier_points)
|
||||
|
||||
knot1 = spline.bezier_points[i].co
|
||||
handle1 = spline.bezier_points[i].handle_right
|
||||
handle2 = spline.bezier_points[inext].handle_left
|
||||
knot2 = spline.bezier_points[inext].co
|
||||
|
||||
_points = interpolate_bezier(knot1, handle1, handle2, knot2, R)
|
||||
for p in _points:
|
||||
scaled_point = matrix @ p * scale
|
||||
points.append(scaled_point.x)
|
||||
points.append(scaled_point.y)
|
||||
points.append(scaled_point.z)
|
||||
|
||||
length = length or spline.calc_length()
|
||||
domain = Interval(start=0, end=length, totalChildrenCount=0)
|
||||
return Polyline(value=points, closed = spline.use_cyclic_u, domain=domain, area=0, len=length)
|
||||
|
||||
def poly_to_speckle(matrix: MMatrix, spline: bpy.types.Spline, scale: float, name: Optional[str] = None) -> Polyline:
|
||||
points = [tuple(matrix @ pt.co.xyz * scale) for pt in spline.points]
|
||||
|
||||
flattend_points = []
|
||||
for row in points: flattend_points.extend(row)
|
||||
|
||||
length = spline.calc_length()
|
||||
domain = Interval(start=0, end=length, totalChildrenCount=0)
|
||||
return Polyline(
|
||||
name=name,
|
||||
closed=bool(spline.use_cyclic_u),
|
||||
value=list(flattend_points),
|
||||
length=length,
|
||||
domain=domain,
|
||||
bbox=Box(area=0.0, volume=0.0),
|
||||
area=0,
|
||||
units=UNITS,
|
||||
)
|
||||
|
||||
|
||||
def icurve_to_speckle(blender_object: Object, data: bpy.types.Curve, scale=1.0) -> Optional[List[Base]]:
|
||||
UNITS = "m" if bpy.context.scene.unit_settings.system == "METRIC" else "ft"
|
||||
|
||||
if blender_object.type != "CURVE":
|
||||
return None
|
||||
|
||||
blender_object = blender_object.evaluated_get(bpy.context.view_layer.depsgraph)
|
||||
|
||||
mat = blender_object.matrix_world
|
||||
|
||||
curves = []
|
||||
|
||||
if data.bevel_mode == "OBJECT" and data.bevel_object != None:
|
||||
mesh = mesh_to_speckle(blender_object, blender_object.to_mesh(), scale)
|
||||
curves.extend(mesh)
|
||||
|
||||
for spline in data.splines:
|
||||
if spline.type == "BEZIER":
|
||||
curves.append(bezier_to_speckle(mat, spline, scale, blender_object.name))
|
||||
|
||||
elif spline.type == "NURBS":
|
||||
curves.append(nurbs_to_speckle(mat, spline, scale, blender_object.name))
|
||||
|
||||
elif spline.type == "POLY":
|
||||
curves.append(poly_to_speckle(mat, spline, scale, blender_object.name))
|
||||
|
||||
return curves
|
||||
|
||||
@deprecated
|
||||
def ngons_to_speckle_polylines(blender_object: Object, data: bpy.types.Mesh, scale=1.0) -> Optional[List[Polyline]]:
|
||||
UNITS = "m" if bpy.context.scene.unit_settings.system == "METRIC" else "ft"
|
||||
|
||||
if blender_object.type != "MESH":
|
||||
return None
|
||||
|
||||
mat = blender_object.matrix_world
|
||||
|
||||
verts = data.vertices
|
||||
polylines = []
|
||||
for i, poly in enumerate(data.polygons):
|
||||
value = []
|
||||
for v in poly.vertices:
|
||||
value.extend(mat @ verts[v].co * scale)
|
||||
|
||||
domain = Interval(start=0, end=1)
|
||||
poly = Polyline(
|
||||
name="{}_{}".format(blender_object.name, i),
|
||||
closed=True,
|
||||
value=value,
|
||||
length=0,
|
||||
domain=domain,
|
||||
bbox=Box(area=0.0, volume=0.0),
|
||||
area=0,
|
||||
units=UNITS,
|
||||
)
|
||||
|
||||
polylines.append(poly)
|
||||
|
||||
return polylines
|
||||
|
||||
|
||||
def material_to_speckle(blender_mat: bpy.types.Material) -> RenderMaterial:
|
||||
speckle_mat = RenderMaterial()
|
||||
speckle_mat.name = blender_mat.name
|
||||
|
||||
if blender_mat.use_nodes is True and blender_mat.node_tree.nodes.get(
|
||||
"Principled BSDF"
|
||||
):
|
||||
inputs = blender_mat.node_tree.nodes["Principled BSDF"].inputs
|
||||
speckle_mat.diffuse = to_argb_int(inputs["Base Color"].default_value)
|
||||
speckle_mat.emissive = to_argb_int(inputs["Emission"].default_value)
|
||||
speckle_mat.roughness = inputs["Roughness"].default_value
|
||||
speckle_mat.metalness = inputs["Metallic"].default_value
|
||||
speckle_mat.opacity = inputs["Alpha"].default_value
|
||||
|
||||
else:
|
||||
speckle_mat.diffuse = to_argb_int(blender_mat.diffuse_color)
|
||||
speckle_mat.metalness = blender_mat.metallic
|
||||
speckle_mat.roughness = blender_mat.roughness
|
||||
|
||||
return speckle_mat
|
||||
|
||||
|
||||
def material_to_speckle_old(blender_object: Object) -> Optional[RenderMaterial]:
|
||||
"""Create and return a render material from a blender object"""
|
||||
if not getattr(blender_object.data, "materials", None):
|
||||
return None
|
||||
|
||||
blender_mat: bpy.types.Material = blender_object.data.materials[0]
|
||||
if not blender_mat:
|
||||
return None
|
||||
|
||||
return material_to_speckle(blender_mat)
|
||||
|
||||
|
||||
def transform_to_speckle(blender_transform: Iterable[Iterable[float]], scale=1.0) -> Transform:
|
||||
value = [y for x in blender_transform for y in x]
|
||||
# scale the translation
|
||||
for i in (3, 7, 11):
|
||||
value[i] *= scale
|
||||
|
||||
return Transform(value=value, units=UNITS)
|
||||
|
||||
|
||||
def block_def_to_speckle(blender_definition: bpy.types.Collection, scale=1.0) -> BlockDefinition:
|
||||
geometry = []
|
||||
for geo in blender_definition.objects:
|
||||
geometry.extend(convert_to_speckle(geo, scale, UNITS, None))
|
||||
block_def = BlockDefinition(
|
||||
units=UNITS,
|
||||
name=blender_definition.name,
|
||||
geometry=geometry,
|
||||
basePoint=Point(units=UNITS),
|
||||
)
|
||||
blender_props = get_blender_custom_properties(blender_definition)
|
||||
block_def.applicationId = blender_props.pop("applicationId", None)
|
||||
return block_def
|
||||
|
||||
|
||||
def block_instance_to_speckle(blender_instance: Object, scale=1.0) -> BlockInstance:
|
||||
return BlockInstance(
|
||||
blockDefinition=block_def_to_speckle(
|
||||
blender_instance.instance_collection, scale
|
||||
),
|
||||
transform=transform_to_speckle(blender_instance.matrix_world),
|
||||
name=blender_instance.name,
|
||||
units=UNITS,
|
||||
)
|
||||
|
||||
|
||||
def empty_to_speckle(blender_object: Object, scale=1.0) -> Optional[BlockInstance]:
|
||||
# probably an instance collection (block) so let's try it
|
||||
try:
|
||||
geo = blender_object.instance_collection.objects.items()
|
||||
return block_instance_to_speckle(blender_object, scale)
|
||||
except AttributeError as err:
|
||||
_report(
|
||||
f"No instance collection found in empty. Skipping object {blender_object.name}"
|
||||
)
|
||||
return None
|
||||
@@ -1,431 +0,0 @@
|
||||
import math
|
||||
from typing import Any, Optional, Tuple
|
||||
from bmesh.types import BMesh
|
||||
import bpy, struct, idprop
|
||||
|
||||
from specklepy.objects.base import Base
|
||||
from specklepy.objects.geometry import Circle, Mesh, Ellipse
|
||||
from specklepy.objects.other import RenderMaterial
|
||||
from bpy_speckle.functions import _report
|
||||
from bpy.types import Material, Object
|
||||
|
||||
IGNORED_PROPERTY_KEYS = {
|
||||
"id",
|
||||
"elements",
|
||||
"displayMesh",
|
||||
"displayValue",
|
||||
"speckle_type",
|
||||
"parameters",
|
||||
"faces",
|
||||
"colors",
|
||||
"vertices",
|
||||
"renderMaterial",
|
||||
"textureCoordinates",
|
||||
"totalChildrenCount"
|
||||
}
|
||||
|
||||
|
||||
def to_rgba(argb_int: int) -> Tuple[float, float, float, float]:
|
||||
"""Converts the int representation of a colour into a percent RGBA tuple"""
|
||||
alpha = ((argb_int >> 24) & 255) / 255
|
||||
red = ((argb_int >> 16) & 255) / 255
|
||||
green = ((argb_int >> 8) & 255) / 255
|
||||
blue = (argb_int & 255) / 255
|
||||
|
||||
return (red, green, blue, alpha)
|
||||
|
||||
|
||||
def to_argb_int(rgba_color: list[float]) -> int:
|
||||
"""Converts an RGBA array to an ARGB integer"""
|
||||
argb_color = rgba_color[-1:] + rgba_color[:3]
|
||||
int_color = [int(val * 255) for val in argb_color]
|
||||
|
||||
return int.from_bytes(int_color, byteorder="big", signed=True)
|
||||
|
||||
def set_custom_property(key: str, value: Any, blender_object: Object) -> None:
|
||||
try:
|
||||
#Expected c types: float, int, string, float[], int[]
|
||||
blender_object[key] = value
|
||||
except (OverflowError, TypeError) as ex:
|
||||
print(f"Skipping setting property ({key}={value}) on {blender_object.name_full}, Reason: {ex}")
|
||||
except Exception as ex:
|
||||
#TODO: Log this as it's unexpected!!!
|
||||
print(f"Skipping setting property ({key}={value}) on {blender_object.name_full}, Reason: {ex}")
|
||||
|
||||
def add_custom_properties(speckle_object: Base, blender_object: Object):
|
||||
if blender_object is None:
|
||||
return
|
||||
|
||||
blender_object["_speckle_type"] = type(speckle_object).__name__
|
||||
|
||||
app_id = getattr(speckle_object, "applicationId", None)
|
||||
if app_id:
|
||||
blender_object["applicationId"] = speckle_object.applicationId
|
||||
keys = speckle_object.get_dynamic_member_names() if "Geometry" in speckle_object.speckle_type else (set(speckle_object.get_member_names()) - IGNORED_PROPERTY_KEYS)
|
||||
for key in keys:
|
||||
val = getattr(speckle_object, key, None)
|
||||
if val is None:
|
||||
continue
|
||||
|
||||
if isinstance(val, (int, str, float)):
|
||||
set_custom_property(key, val, blender_object)
|
||||
elif key == "properties" and isinstance(val, Base):
|
||||
val["applicationId"] = None
|
||||
add_custom_properties(val, blender_object)
|
||||
elif isinstance(val, list):
|
||||
items = [item for item in val if not isinstance(item, Base)]
|
||||
if items:
|
||||
set_custom_property(key, items, blender_object)
|
||||
elif isinstance(val,dict):
|
||||
for (k,v) in val.items():
|
||||
if not isinstance(v, Base):
|
||||
set_custom_property(k, v, blender_object)
|
||||
|
||||
|
||||
def render_material_to_native(speckle_mat: RenderMaterial) -> Material:
|
||||
|
||||
mat_name = speckle_mat.name
|
||||
if not mat_name:
|
||||
mat_name = speckle_mat.applicationId or speckle_mat.id or speckle_mat.get_id()
|
||||
|
||||
blender_mat = bpy.data.materials.get(mat_name)
|
||||
if blender_mat is None:
|
||||
blender_mat = bpy.data.materials.new(mat_name)
|
||||
|
||||
# for now, we're not updating these materials. as per tom's suggestion, we should have a toggle
|
||||
# that enables this as the blender mats will prob be much more complex than whatever is coming in
|
||||
blender_mat.use_nodes = True
|
||||
inputs = blender_mat.node_tree.nodes["Principled BSDF"].inputs
|
||||
|
||||
inputs["Base Color"].default_value = to_rgba(speckle_mat.diffuse)
|
||||
inputs["Emission"].default_value = to_rgba(speckle_mat.emissive)
|
||||
inputs["Roughness"].default_value = speckle_mat.roughness
|
||||
inputs["Metallic"].default_value = speckle_mat.metalness
|
||||
inputs["Alpha"].default_value = speckle_mat.opacity
|
||||
|
||||
if speckle_mat.opacity < 1.0:
|
||||
blender_mat.blend_method = "BLEND"
|
||||
|
||||
return blender_mat
|
||||
|
||||
def get_render_material(speckle_object: Base) -> Optional[RenderMaterial]:
|
||||
"""Trys to get a RenderMaterial on given speckle_object and convert it to a blender material"""
|
||||
|
||||
speckle_mat = getattr(
|
||||
speckle_object,
|
||||
"renderMaterial",
|
||||
getattr(speckle_object, "@renderMaterial", None),
|
||||
)
|
||||
if not isinstance(speckle_mat, RenderMaterial):
|
||||
return None
|
||||
|
||||
return speckle_mat
|
||||
|
||||
|
||||
|
||||
def add_vertices(speckle_mesh: Mesh, blender_mesh: BMesh, scale=1.0):
|
||||
sverts = speckle_mesh.vertices
|
||||
|
||||
if sverts and len(sverts) > 0:
|
||||
for i in range(0, len(sverts), 3):
|
||||
blender_mesh.verts.new(
|
||||
(
|
||||
float(sverts[i]) * scale,
|
||||
float(sverts[i + 1]) * scale,
|
||||
float(sverts[i + 2]) * scale,
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
|
||||
def add_faces(speckle_mesh: Mesh, blender_mesh: BMesh, indexOffset: int, materialIndex: int = 0, smooth:bool = True):
|
||||
sfaces = speckle_mesh.faces
|
||||
|
||||
if sfaces and len(sfaces) > 0:
|
||||
i = 0
|
||||
while i < len(sfaces):
|
||||
n = sfaces[i]
|
||||
if n < 3:
|
||||
n += 3 # 0 -> 3, 1 -> 4
|
||||
|
||||
i += 1
|
||||
try:
|
||||
f = blender_mesh.faces.new(
|
||||
[blender_mesh.verts[x + indexOffset] for x in sfaces[i : i + n]]
|
||||
)
|
||||
f.material_index = materialIndex
|
||||
f.smooth = smooth
|
||||
except Exception as e:
|
||||
_report(f"Failed to create face for mesh {speckle_mesh.id} \n{e}")
|
||||
i += n
|
||||
|
||||
|
||||
def add_colors(speckle_mesh: Mesh, blender_mesh: BMesh):
|
||||
|
||||
scolors = speckle_mesh.colors
|
||||
|
||||
if scolors:
|
||||
colors = []
|
||||
if len(scolors) > 0:
|
||||
|
||||
for i in range(len(scolors)):
|
||||
col = int(scolors[i])
|
||||
(a, r, g, b) = [
|
||||
int(x) for x in struct.unpack("!BBBB", struct.pack("!i", col))
|
||||
]
|
||||
colors.append(
|
||||
(
|
||||
float(r) / 255.0,
|
||||
float(g) / 255.0,
|
||||
float(b) / 255.0,
|
||||
float(a) / 255.0,
|
||||
)
|
||||
)
|
||||
|
||||
# Make vertex colors
|
||||
if len(scolors) == len(blender_mesh.verts):
|
||||
color_layer = blender_mesh.loops.layers.color.new("Col")
|
||||
|
||||
for face in blender_mesh.faces:
|
||||
for loop in face.loops:
|
||||
loop[color_layer] = colors[loop.vert.index]
|
||||
|
||||
|
||||
def add_uv_coords(speckle_mesh: Mesh, blender_mesh: BMesh):
|
||||
s_uvs = speckle_mesh.textureCoordinates
|
||||
if not s_uvs:
|
||||
return
|
||||
try:
|
||||
uv = []
|
||||
|
||||
if len(s_uvs) // 2 == len(blender_mesh.verts):
|
||||
uv.extend(
|
||||
(float(s_uvs[i]), float(s_uvs[i + 1]))
|
||||
for i in range(0, len(s_uvs), 2)
|
||||
)
|
||||
else:
|
||||
_report(
|
||||
f"Failed to match UV coordinates to vert data. Blender mesh verts: {len(blender_mesh.verts)}, Speckle UVs: {len(s_uvs) // 2}"
|
||||
)
|
||||
return
|
||||
|
||||
# Make UVs
|
||||
uv_layer = blender_mesh.loops.layers.uv.verify()
|
||||
|
||||
for f in blender_mesh.faces:
|
||||
for l in f.loops:
|
||||
luv = l[uv_layer]
|
||||
luv.uv = uv[l.vert.index]
|
||||
except:
|
||||
_report("Failed to decode texture coordinates.")
|
||||
raise
|
||||
|
||||
|
||||
ignored_keys = {
|
||||
"id",
|
||||
"speckle",
|
||||
"speckle_type"
|
||||
"_speckle_type",
|
||||
"_speckle_name",
|
||||
"_speckle_transform",
|
||||
"_RNA_UI",
|
||||
"elements",
|
||||
"transform",
|
||||
"_units",
|
||||
"_chunkable",
|
||||
}
|
||||
|
||||
def get_blender_custom_properties(obj, max_depth: int = 1000):
|
||||
if max_depth < 0:
|
||||
return obj
|
||||
|
||||
if hasattr(obj, "keys"):
|
||||
keys = set(obj.keys()) - ignored_keys
|
||||
return {
|
||||
key: get_blender_custom_properties(obj[key], max_depth - 1)
|
||||
for key in keys
|
||||
if not key.startswith("_")
|
||||
}
|
||||
|
||||
if isinstance(obj, (list, tuple, idprop.types.IDPropertyArray)):
|
||||
return [get_blender_custom_properties(o, max_depth - 1) for o in obj]
|
||||
|
||||
return obj
|
||||
|
||||
"""
|
||||
Python implementation of Blender's NURBS curve generation for to Speckle conversion
|
||||
from: https://blender.stackexchange.com/a/34276
|
||||
based on https://projects.blender.org/blender/blender/src/branch/main/source/blender/blenkernel/intern/curve.cc (check old version)
|
||||
"""
|
||||
|
||||
def macro_knotsu(nu: bpy.types.Spline) -> int:
|
||||
return nu.order_u + nu.point_count_u + (nu.order_u - 1 if nu.use_cyclic_u else 0)
|
||||
|
||||
def macro_segmentsu(nu: bpy.types.Spline) -> int:
|
||||
return nu.point_count_u if nu.use_cyclic_u else nu.point_count_u - 1
|
||||
|
||||
def make_knots(nu: bpy.types.Spline) -> list[float]:
|
||||
knots = [0.0] * macro_knotsu(nu)
|
||||
flag = nu.use_endpoint_u + (nu.use_bezier_u << 1)
|
||||
if nu.use_cyclic_u:
|
||||
calc_knots(knots, nu.point_count_u, nu.order_u, 0)
|
||||
else:
|
||||
calc_knots(knots, nu.point_count_u, nu.order_u, flag)
|
||||
return knots
|
||||
|
||||
|
||||
def calc_knots(knots: list[float], point_count: int, order: int, flag: int) -> None:
|
||||
pts_order = point_count + order
|
||||
if flag == 1: # CU_NURB_ENDPOINT
|
||||
k = 0.0
|
||||
for a in range(1, pts_order + 1):
|
||||
knots[a - 1] = k
|
||||
if a >= order and a <= point_count:
|
||||
k += 1.0
|
||||
elif flag == 2: # CU_NURB_BEZIER
|
||||
if order == 4:
|
||||
k = 0.34
|
||||
for a in range(pts_order):
|
||||
knots[a] = math.floor(k)
|
||||
k += 1.0 / 3.0
|
||||
elif order == 3:
|
||||
k = 0.6
|
||||
for a in range(pts_order):
|
||||
if a >= order and a <= point_count:
|
||||
k += 0.5
|
||||
knots[a] = math.floor(k)
|
||||
else:
|
||||
for a in range(1, len(knots) - 1):
|
||||
knots[a] = a - 1
|
||||
|
||||
knots[-1] = knots[-2]
|
||||
|
||||
def basis_nurb(t: float, order: int, point_count: int, knots: list[float], basis: list[float], start: int, end: int) -> Tuple[int, int]:
|
||||
i1 = i2 = 0
|
||||
orderpluspnts = order + point_count
|
||||
opp2 = orderpluspnts - 1
|
||||
|
||||
# this is for float inaccuracy
|
||||
if t < knots[0]:
|
||||
t = knots[0]
|
||||
elif t > knots[opp2]:
|
||||
t = knots[opp2]
|
||||
|
||||
# this part is order '1'
|
||||
o2 = order + 1
|
||||
for i in range(opp2):
|
||||
if knots[i] != knots[i + 1] and t >= knots[i] and t <= knots[i + 1]:
|
||||
basis[i] = 1.0
|
||||
i1 = i - o2
|
||||
if i1 < 0:
|
||||
i1 = 0
|
||||
i2 = i
|
||||
i += 1
|
||||
while i < opp2:
|
||||
basis[i] = 0.0
|
||||
i += 1
|
||||
break
|
||||
|
||||
else:
|
||||
basis[i] = 0.0
|
||||
|
||||
basis[i] = 0.0
|
||||
|
||||
# this is order 2, 3, ...
|
||||
for j in range(2, order + 1):
|
||||
|
||||
if i2 + j >= orderpluspnts:
|
||||
i2 = opp2 - j
|
||||
|
||||
for i in range(i1, i2 + 1):
|
||||
if basis[i] != 0.0:
|
||||
d = ((t - knots[i]) * basis[i]) / (knots[i + j - 1] - knots[i])
|
||||
else:
|
||||
d = 0.0
|
||||
|
||||
if basis[i + 1] != 0.0:
|
||||
e = ((knots[i + j] - t) * basis[i + 1]) / (knots[i + j] - knots[i + 1])
|
||||
else:
|
||||
e = 0.0
|
||||
|
||||
basis[i] = d + e
|
||||
|
||||
start = 1000
|
||||
end = 0
|
||||
|
||||
for i in range(i1, i2 + 1):
|
||||
if basis[i] > 0.0:
|
||||
end = i
|
||||
if start == 1000:
|
||||
start = i
|
||||
|
||||
return start, end
|
||||
|
||||
def nurb_make_curve(nu: bpy.types.Spline, resolu: int, stride: int = 3) -> list[float]:
|
||||
""""BKE_nurb_makeCurve"""
|
||||
EPS = 1e-6
|
||||
coord_index = istart = iend = 0
|
||||
|
||||
coord_array = [0.0] * (3 * nu.resolution_u * macro_segmentsu(nu))
|
||||
sum_array = [0] * nu.point_count_u
|
||||
basisu = [0.0] * macro_knotsu(nu)
|
||||
knots = make_knots(nu)
|
||||
|
||||
resolu = resolu * macro_segmentsu(nu)
|
||||
ustart = knots[nu.order_u - 1]
|
||||
uend = knots[nu.point_count_u + nu.order_u - 1] if nu.use_cyclic_u else \
|
||||
knots[nu.point_count_u]
|
||||
ustep = (uend - ustart) / (resolu - (0 if nu.use_cyclic_u else 1))
|
||||
cycl = nu.order_u - 1 if nu.use_cyclic_u else 0
|
||||
|
||||
u = ustart
|
||||
while resolu:
|
||||
resolu -= 1
|
||||
istart, iend = basis_nurb(u, nu.order_u, nu.point_count_u + cycl, knots, basisu, istart, iend)
|
||||
|
||||
#/* calc sum */
|
||||
sumdiv = 0.0
|
||||
sum_index = 0
|
||||
pt_index = istart - 1
|
||||
for i in range(istart, iend + 1):
|
||||
if i >= nu.point_count_u:
|
||||
pt_index = i - nu.point_count_u
|
||||
else:
|
||||
pt_index += 1
|
||||
|
||||
sum_array[sum_index] = basisu[i] * nu.points[pt_index].co[3]
|
||||
sumdiv += sum_array[sum_index]
|
||||
sum_index += 1
|
||||
|
||||
if (sumdiv != 0.0) and (sumdiv < 1.0 - EPS or sumdiv > 1.0 + EPS):
|
||||
sum_index = 0
|
||||
for i in range(istart, iend + 1):
|
||||
sum_array[sum_index] /= sumdiv
|
||||
sum_index += 1
|
||||
|
||||
coord_array[coord_index: coord_index + 3] = (0.0, 0.0, 0.0)
|
||||
|
||||
sum_index = 0
|
||||
pt_index = istart - 1
|
||||
for i in range(istart, iend + 1):
|
||||
if i >= nu.point_count_u:
|
||||
pt_index = i - nu.point_count_u
|
||||
else:
|
||||
pt_index += 1
|
||||
|
||||
if sum_array[sum_index] != 0.0:
|
||||
for j in range(3):
|
||||
coord_array[coord_index + j] += sum_array[sum_index] * nu.points[pt_index].co[j]
|
||||
sum_index += 1
|
||||
|
||||
coord_index += stride
|
||||
u += ustep
|
||||
|
||||
return coord_array
|
||||
|
||||
def link_object_to_collection_nested(obj: bpy.types.Object, col: bpy.types.Collection):
|
||||
if obj.name not in col.objects:
|
||||
col.objects.link(obj)
|
||||
|
||||
for child in obj.children:
|
||||
link_object_to_collection_nested(child, col)
|
||||
@@ -0,0 +1,3 @@
|
||||
from ..converter.to_native import * # noqa: F403
|
||||
from ..converter.to_speckle import * # noqa: F403
|
||||
from ..converter.utils import * # noqa: F403
|
||||
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,6 @@
|
||||
from .to_speckle import convert_to_speckle # noqa: F401
|
||||
from .material_to_speckle import ( # noqa: F401
|
||||
blender_material_to_speckle,
|
||||
create_render_material_proxies,
|
||||
add_render_material_proxies_to_base,
|
||||
)
|
||||
@@ -0,0 +1,272 @@
|
||||
from bpy.types import Object
|
||||
from typing import Union, Optional, Tuple, List
|
||||
from specklepy.objects.geometry import Polyline, Curve
|
||||
from specklepy.objects.primitive import Interval
|
||||
from specklepy.objects.base import Base
|
||||
from mathutils import Matrix
|
||||
from mathutils.geometry import interpolate_bezier
|
||||
from .utils import nurb_make_curve, make_knots
|
||||
|
||||
|
||||
def curve_to_speckle(
|
||||
blender_obj: Object, scale_factor: float = 1.0
|
||||
) -> Union[Base, None]:
|
||||
assert blender_obj.type == "CURVE", "Object must be a curve"
|
||||
assert blender_obj.data is not None, "Curve data cannot be None"
|
||||
|
||||
curve_data = blender_obj.data
|
||||
matrix = blender_obj.matrix_world
|
||||
units = "m" # TODO: Use the unit system from the scene
|
||||
|
||||
base = Base()
|
||||
curves = []
|
||||
|
||||
for spline in curve_data.splines:
|
||||
if spline.type == "BEZIER":
|
||||
curves.append(
|
||||
bezier_to_speckle(matrix, spline, blender_obj.name, scale_factor, units)
|
||||
)
|
||||
elif spline.type == "NURBS":
|
||||
curves.append(
|
||||
nurbs_to_speckle(matrix, spline, blender_obj.name, scale_factor, units)
|
||||
)
|
||||
|
||||
if curves:
|
||||
base["@elements"] = curves
|
||||
base["name"] = blender_obj.name
|
||||
return base
|
||||
|
||||
return None
|
||||
|
||||
|
||||
def bezier_to_speckle(
|
||||
matrix: Matrix,
|
||||
spline,
|
||||
name: Optional[str] = None,
|
||||
scale_factor: float = 1.0,
|
||||
units: str = "m",
|
||||
) -> Curve:
|
||||
degree = 3
|
||||
closed = spline.use_cyclic_u
|
||||
points: List[Tuple[float, float, float]] = []
|
||||
|
||||
for i, bp in enumerate(spline.bezier_points):
|
||||
if i > 0:
|
||||
transformed_point = matrix @ bp.handle_left * scale_factor
|
||||
points.append(
|
||||
(transformed_point.x, transformed_point.y, transformed_point.z)
|
||||
)
|
||||
|
||||
transformed_point = matrix @ bp.co * scale_factor
|
||||
points.append((transformed_point.x, transformed_point.y, transformed_point.z))
|
||||
|
||||
if i < len(spline.bezier_points) - 1:
|
||||
transformed_point = matrix @ bp.handle_right * scale_factor
|
||||
points.append(
|
||||
(transformed_point.x, transformed_point.y, transformed_point.z)
|
||||
)
|
||||
|
||||
if closed:
|
||||
transformed_point = (
|
||||
matrix @ spline.bezier_points[-1].handle_right * scale_factor
|
||||
)
|
||||
points.append((transformed_point.x, transformed_point.y, transformed_point.z))
|
||||
|
||||
transformed_point = matrix @ spline.bezier_points[0].handle_left * scale_factor
|
||||
points.append((transformed_point.x, transformed_point.y, transformed_point.z))
|
||||
|
||||
transformed_point = matrix @ spline.bezier_points[0].co * scale_factor
|
||||
points.append((transformed_point.x, transformed_point.y, transformed_point.z))
|
||||
|
||||
num_points = len(points)
|
||||
|
||||
flattened_points = []
|
||||
for point in points:
|
||||
flattened_points.extend(point)
|
||||
|
||||
knot_count = num_points + degree - 1
|
||||
knots = [0] * knot_count
|
||||
|
||||
for i in range(1, len(knots)):
|
||||
knots[i] = i // 3
|
||||
|
||||
length = spline.calc_length()
|
||||
|
||||
domain = Interval(start=0, end=length)
|
||||
display_value = bezier_to_speckle_polyline(
|
||||
matrix, spline, length, scale_factor, units
|
||||
)
|
||||
|
||||
curve = Curve(
|
||||
degree=degree,
|
||||
periodic=not spline.use_endpoint_u,
|
||||
rational=True,
|
||||
points=flattened_points,
|
||||
weights=[1] * num_points,
|
||||
knots=knots,
|
||||
closed=spline.use_cyclic_u,
|
||||
displayValue=display_value,
|
||||
units=units,
|
||||
bbox=None,
|
||||
)
|
||||
|
||||
curve.__dict__["_length"] = length
|
||||
curve.__dict__["_area"] = 0.0
|
||||
|
||||
curve["domain"] = domain
|
||||
|
||||
if name:
|
||||
curve["name"] = name
|
||||
|
||||
return curve
|
||||
|
||||
|
||||
def bezier_to_speckle_polyline(
|
||||
matrix: Matrix,
|
||||
spline,
|
||||
length: Optional[float] = None,
|
||||
scale_factor: float = 1.0,
|
||||
units: str = "m",
|
||||
) -> Optional[Polyline]:
|
||||
segments = len(spline.bezier_points)
|
||||
if segments < 2:
|
||||
return None
|
||||
|
||||
resolution = spline.resolution_u + 1
|
||||
points: List[float] = []
|
||||
|
||||
if not spline.use_cyclic_u:
|
||||
segments -= 1
|
||||
|
||||
for i in range(segments):
|
||||
inext = (i + 1) % len(spline.bezier_points)
|
||||
|
||||
knot1 = spline.bezier_points[i].co
|
||||
handle1 = spline.bezier_points[i].handle_right
|
||||
handle2 = spline.bezier_points[inext].handle_left
|
||||
knot2 = spline.bezier_points[inext].co
|
||||
|
||||
sampled_points = interpolate_bezier(knot1, handle1, handle2, knot2, resolution)
|
||||
for p in sampled_points:
|
||||
scaled_point = matrix @ p * scale_factor
|
||||
points.append(scaled_point.x)
|
||||
points.append(scaled_point.y)
|
||||
points.append(scaled_point.z)
|
||||
|
||||
length = length or spline.calc_length()
|
||||
|
||||
polyline = Polyline(value=points, units=units)
|
||||
|
||||
polyline["domain"] = {"start": 0, "end": length}
|
||||
polyline["closed"] = spline.use_cyclic_u
|
||||
|
||||
return polyline
|
||||
|
||||
|
||||
def nurbs_to_speckle(
|
||||
matrix: Matrix,
|
||||
spline,
|
||||
name: Optional[str] = None,
|
||||
scale_factor: float = 1.0,
|
||||
units: str = "m",
|
||||
) -> Curve:
|
||||
degree = spline.order_u - 1
|
||||
knots = make_knots(spline)
|
||||
|
||||
length = spline.calc_length()
|
||||
domain = Interval(start=0, end=length)
|
||||
|
||||
weights = [pt.weight for pt in spline.points]
|
||||
first_weight = weights[0] if weights else 1.0
|
||||
is_rational = any(abs(w - first_weight) > 1e-9 for w in weights)
|
||||
|
||||
points = []
|
||||
for pt in spline.points:
|
||||
transformed_point = matrix @ pt.co.xyz * scale_factor
|
||||
points.append((transformed_point.x, transformed_point.y, transformed_point.z))
|
||||
|
||||
flattened_points = []
|
||||
for point in points:
|
||||
flattened_points.extend(point)
|
||||
|
||||
if spline.use_cyclic_u:
|
||||
for i in range(0, degree * 3, 3):
|
||||
flattened_points.append(flattened_points[i + 0])
|
||||
flattened_points.append(flattened_points[i + 1])
|
||||
flattened_points.append(flattened_points[i + 2])
|
||||
|
||||
for i in range(0, degree):
|
||||
weights.append(weights[i])
|
||||
|
||||
resolution_multiplier = (
|
||||
4 if (spline.use_cyclic_u and spline.point_count_u <= 16) else 1
|
||||
)
|
||||
display_value = nurbs_to_speckle_polyline(
|
||||
matrix, spline, length, scale_factor, units, resolution_multiplier
|
||||
)
|
||||
|
||||
curve = Curve(
|
||||
degree=degree,
|
||||
periodic=not spline.use_endpoint_u,
|
||||
rational=is_rational,
|
||||
points=flattened_points,
|
||||
weights=weights,
|
||||
knots=knots,
|
||||
closed=spline.use_cyclic_u,
|
||||
displayValue=display_value,
|
||||
units=units,
|
||||
bbox=None,
|
||||
)
|
||||
|
||||
curve.__dict__["_length"] = length
|
||||
|
||||
curve["domain"] = domain
|
||||
|
||||
if name:
|
||||
curve["name"] = name
|
||||
|
||||
return curve
|
||||
|
||||
|
||||
def nurbs_to_speckle_polyline(
|
||||
matrix: Matrix,
|
||||
spline,
|
||||
length: Optional[float] = None,
|
||||
scale_factor: float = 1.0,
|
||||
units: str = "m",
|
||||
resolution_multiplier: int = 1,
|
||||
) -> Polyline:
|
||||
from mathutils import Vector
|
||||
|
||||
points: List[float] = []
|
||||
|
||||
resolution = spline.resolution_u * resolution_multiplier
|
||||
|
||||
sampled_points = nurb_make_curve(spline, resolution)
|
||||
|
||||
for i in range(0, len(sampled_points), 3):
|
||||
point_vector = Vector(
|
||||
(sampled_points[i], sampled_points[i + 1], sampled_points[i + 2])
|
||||
)
|
||||
transformed_point = matrix @ point_vector * scale_factor
|
||||
|
||||
points.append(transformed_point.x)
|
||||
points.append(transformed_point.y)
|
||||
points.append(transformed_point.z)
|
||||
|
||||
length = length or spline.calc_length()
|
||||
|
||||
polyline = Polyline(value=points, units=units)
|
||||
|
||||
polyline["domain"] = {"start": 0, "end": length}
|
||||
polyline["closed"] = spline.use_cyclic_u
|
||||
|
||||
# Set length property if needed
|
||||
if hasattr(polyline, "length") or hasattr(polyline, "_length"):
|
||||
polyline.__dict__["_length"] = length
|
||||
|
||||
# Set area property if needed
|
||||
if hasattr(polyline, "area") or hasattr(polyline, "_area"):
|
||||
polyline.__dict__["_area"] = 0
|
||||
|
||||
return polyline
|
||||
@@ -0,0 +1,257 @@
|
||||
from typing import Dict, List, Set
|
||||
import bpy
|
||||
from bpy.types import Material, Object
|
||||
from specklepy.objects.base import Base
|
||||
from specklepy.objects.other import RenderMaterial
|
||||
from specklepy.objects.proxies import RenderMaterialProxy
|
||||
from ..utils import to_argb_int
|
||||
|
||||
from .utils import get_submesh_id, get_unique_id
|
||||
|
||||
|
||||
def blender_material_to_speckle(material: Material) -> RenderMaterial:
|
||||
"""
|
||||
convert a Blender material to a Speckle RenderMaterial
|
||||
"""
|
||||
diffuse = -1 # default white
|
||||
opacity = 1.0
|
||||
emissive = -16777216 # default black
|
||||
metalness = 0.0
|
||||
roughness = 1.0
|
||||
|
||||
# extract material properties if using nodes
|
||||
if material.use_nodes and material.node_tree:
|
||||
output_node = None
|
||||
for node in material.node_tree.nodes:
|
||||
if node.type == "OUTPUT_MATERIAL":
|
||||
output_node = node
|
||||
break
|
||||
|
||||
# find the main shader node connected to output
|
||||
main_shader = None
|
||||
if output_node and output_node.inputs["Surface"].is_linked:
|
||||
main_shader = output_node.inputs["Surface"].links[0].from_node
|
||||
|
||||
# handle different shader types
|
||||
# we're supporting: principled, diffuse, emmision and glass - for now
|
||||
if main_shader:
|
||||
if main_shader.type == "BSDF_PRINCIPLED":
|
||||
diffuse, opacity, metalness, roughness, emissive = (
|
||||
_extract_principled_properties(main_shader)
|
||||
)
|
||||
|
||||
elif main_shader.type == "BSDF_DIFFUSE":
|
||||
color_input = main_shader.inputs.get("Color")
|
||||
if color_input:
|
||||
if color_input.is_linked:
|
||||
rgba = _get_color_from_connected_node(
|
||||
color_input.links[0].from_node
|
||||
)
|
||||
else:
|
||||
rgba = list(color_input.default_value)
|
||||
diffuse = to_argb_int(rgba)
|
||||
roughness = 1.0
|
||||
|
||||
elif main_shader.type == "EMISSION":
|
||||
color_input = main_shader.inputs.get("Color")
|
||||
strength_input = main_shader.inputs.get("Strength")
|
||||
if color_input and strength_input:
|
||||
if color_input.is_linked:
|
||||
rgba = _get_color_from_connected_node(
|
||||
color_input.links[0].from_node
|
||||
)
|
||||
else:
|
||||
rgba = list(color_input.default_value)
|
||||
|
||||
strength = (
|
||||
float(strength_input.default_value)
|
||||
if not strength_input.is_linked
|
||||
else 1.0
|
||||
)
|
||||
|
||||
if strength > 0:
|
||||
emission_rgba = [c * strength for c in rgba[:3]] + [rgba[3]]
|
||||
emission_rgba = [min(1.0, max(0.0, c)) for c in emission_rgba]
|
||||
emissive = to_argb_int(emission_rgba)
|
||||
diffuse = to_argb_int(rgba)
|
||||
|
||||
elif main_shader.type == "BSDF_GLASS":
|
||||
color_input = main_shader.inputs.get("Color")
|
||||
if color_input:
|
||||
if color_input.is_linked:
|
||||
rgba = _get_color_from_connected_node(
|
||||
color_input.links[0].from_node
|
||||
)
|
||||
else:
|
||||
rgba = list(color_input.default_value)
|
||||
diffuse = to_argb_int(rgba)
|
||||
roughness_input = main_shader.inputs.get("Roughness")
|
||||
if roughness_input:
|
||||
roughness = (
|
||||
float(roughness_input.default_value)
|
||||
if not roughness_input.is_linked
|
||||
else 0.0
|
||||
)
|
||||
opacity = 0.5
|
||||
|
||||
else:
|
||||
# fallback to legacy material properties
|
||||
if hasattr(material, "diffuse_color"):
|
||||
rgba = list(material.diffuse_color) + [1.0]
|
||||
diffuse = to_argb_int(rgba)
|
||||
|
||||
if hasattr(material, "metallic"):
|
||||
metalness = float(material.metallic)
|
||||
|
||||
if hasattr(material, "roughness"):
|
||||
roughness = float(material.roughness)
|
||||
|
||||
render_material = RenderMaterial(
|
||||
name=material.name,
|
||||
diffuse=diffuse,
|
||||
opacity=opacity,
|
||||
emissive=emissive,
|
||||
metalness=metalness,
|
||||
roughness=roughness,
|
||||
)
|
||||
|
||||
return render_material
|
||||
|
||||
|
||||
def _extract_principled_properties(principled_node):
|
||||
diffuse = -1
|
||||
opacity = 1.0
|
||||
metalness = 0.0
|
||||
roughness = 1.0
|
||||
emissive = -16777216
|
||||
|
||||
base_color_input = principled_node.inputs.get("Base Color")
|
||||
if base_color_input:
|
||||
if base_color_input.is_linked:
|
||||
rgba = _get_color_from_connected_node(base_color_input.links[0].from_node)
|
||||
else:
|
||||
rgba = list(base_color_input.default_value)
|
||||
diffuse = to_argb_int(rgba)
|
||||
|
||||
# Alpha/Opacity
|
||||
alpha_input = principled_node.inputs.get("Alpha")
|
||||
if alpha_input and not alpha_input.is_linked:
|
||||
opacity = float(alpha_input.default_value)
|
||||
|
||||
# Metallic
|
||||
metallic_input = principled_node.inputs.get("Metallic")
|
||||
if metallic_input and not metallic_input.is_linked:
|
||||
metalness = float(metallic_input.default_value)
|
||||
|
||||
# Roughness
|
||||
roughness_input = principled_node.inputs.get("Roughness")
|
||||
if roughness_input and not roughness_input.is_linked:
|
||||
roughness = float(roughness_input.default_value)
|
||||
|
||||
# Emission - try different possible input names for different versions
|
||||
emission_color_input = principled_node.inputs.get(
|
||||
"Emission Color"
|
||||
) or principled_node.inputs.get("Emission")
|
||||
|
||||
emission_strength_input = principled_node.inputs.get("Emission Strength")
|
||||
|
||||
if emission_color_input:
|
||||
if emission_color_input.is_linked:
|
||||
emission_rgba = _get_color_from_connected_node(
|
||||
emission_color_input.links[0].from_node
|
||||
)
|
||||
else:
|
||||
emission_rgba = list(emission_color_input.default_value)
|
||||
|
||||
emission_strength = 1.0
|
||||
if emission_strength_input and not emission_strength_input.is_linked:
|
||||
emission_strength = float(emission_strength_input.default_value)
|
||||
|
||||
if emission_strength > 0 and any(
|
||||
c > 0.01 for c in emission_rgba[:3]
|
||||
): # Check if color is not black
|
||||
final_emission_rgba = [c * emission_strength for c in emission_rgba[:3]] + [
|
||||
emission_rgba[3]
|
||||
]
|
||||
final_emission_rgba = [min(1.0, max(0.0, c)) for c in final_emission_rgba]
|
||||
emissive = to_argb_int(final_emission_rgba)
|
||||
|
||||
return diffuse, opacity, metalness, roughness, emissive
|
||||
|
||||
|
||||
def _get_color_from_connected_node(node):
|
||||
if node.type == "RGB":
|
||||
rgba = list(node.outputs["Color"].default_value)
|
||||
return rgba
|
||||
elif node.type == "VALTORGB":
|
||||
if node.color_ramp.elements:
|
||||
rgba = list(node.color_ramp.elements[0].color)
|
||||
return rgba
|
||||
elif hasattr(node, "color"):
|
||||
rgba = list(node.color) + [1.0]
|
||||
return rgba
|
||||
|
||||
# fallback to white
|
||||
return [1.0, 1.0, 1.0, 1.0]
|
||||
|
||||
|
||||
def collect_material_assignments(objects: List[Object]) -> Dict[str, Set[str]]:
|
||||
"""
|
||||
collect material assignments for objects, creating unique applicationIds
|
||||
for each material slot use a unique id
|
||||
"""
|
||||
material_assignments: Dict[str, Set[str]] = {}
|
||||
|
||||
for obj in objects:
|
||||
if not obj or not hasattr(obj, "data") or not obj.data:
|
||||
continue
|
||||
|
||||
# check if object has materials
|
||||
if hasattr(obj.data, "materials") and obj.data.materials:
|
||||
for material_index, material_slot in enumerate(obj.data.materials):
|
||||
if material_slot:
|
||||
material_name = material_slot.name
|
||||
|
||||
# set unique ID for submeshes
|
||||
application_id = get_submesh_id(obj, material_index)
|
||||
|
||||
if material_name not in material_assignments:
|
||||
material_assignments[material_name] = set()
|
||||
|
||||
material_assignments[material_name].add(application_id)
|
||||
|
||||
return material_assignments
|
||||
|
||||
|
||||
def create_render_material_proxies(objects: List[Object]) -> List[RenderMaterialProxy]:
|
||||
material_assignments = collect_material_assignments(objects)
|
||||
|
||||
if not material_assignments:
|
||||
return []
|
||||
|
||||
proxies = []
|
||||
|
||||
for material_name, object_ids in material_assignments.items():
|
||||
blender_material = bpy.data.materials.get(material_name)
|
||||
if not blender_material:
|
||||
continue
|
||||
|
||||
speckle_material = blender_material_to_speckle(blender_material)
|
||||
|
||||
proxy = RenderMaterialProxy(objects=list(object_ids), value=speckle_material)
|
||||
|
||||
proxy.applicationId = get_unique_id(blender_material)
|
||||
|
||||
proxies.append(proxy)
|
||||
|
||||
return proxies
|
||||
|
||||
|
||||
def add_render_material_proxies_to_base(base: Base, objects: List[Object]) -> None:
|
||||
"""
|
||||
add render material proxies to the base object.
|
||||
"""
|
||||
proxies = create_render_material_proxies(objects)
|
||||
|
||||
if proxies:
|
||||
base.renderMaterialProxies = proxies
|
||||
@@ -0,0 +1,132 @@
|
||||
from typing import Dict, List, cast
|
||||
|
||||
import bpy
|
||||
from bpy.types import Mesh as BMesh
|
||||
from bpy.types import MeshPolygon, Object
|
||||
from mathutils import Matrix as MMatrix
|
||||
from mathutils import Vector as MVector
|
||||
|
||||
from specklepy.objects.base import Base
|
||||
from specklepy.objects.geometry.mesh import Mesh
|
||||
from .utils import get_submesh_id
|
||||
|
||||
|
||||
def mesh_to_speckle(
|
||||
blender_object: Object, data: bpy.types.Mesh, units_scale: float, units: str
|
||||
) -> Base:
|
||||
"""
|
||||
convert a Blender mesh object
|
||||
"""
|
||||
meshes = mesh_to_speckle_meshes(blender_object, data, units_scale, units)
|
||||
return meshes
|
||||
|
||||
|
||||
def mesh_to_speckle_meshes(
|
||||
blender_object: Object, data: bpy.types.Mesh, units_scale: float, units: str
|
||||
) -> List[Mesh]:
|
||||
"""
|
||||
convert a Blender mesh to a list of Speckle meshes
|
||||
each face corner (loop) gets its own vertex
|
||||
"""
|
||||
assert isinstance(data, BMesh), "Data must be a Blender mesh"
|
||||
assert units_scale > 0, "Units scale must be positive"
|
||||
|
||||
submesh_data: Dict[int, List[MeshPolygon]] = {}
|
||||
for p in data.polygons:
|
||||
if p.material_index not in submesh_data:
|
||||
submesh_data[p.material_index] = []
|
||||
submesh_data[p.material_index].append(p)
|
||||
|
||||
transform = cast(MMatrix, blender_object.matrix_world)
|
||||
normal_transform = transform.to_3x3().inverted().transposed()
|
||||
|
||||
submeshes = []
|
||||
|
||||
# sort material indices to ensure consistent ordering
|
||||
for material_index in sorted(submesh_data.keys()):
|
||||
mesh_area = 0
|
||||
m_verts: List[float] = []
|
||||
m_faces: List[int] = []
|
||||
m_texcoords: List[float] = []
|
||||
m_normals: List[float] = []
|
||||
|
||||
vertex_counter = 0
|
||||
|
||||
for face in submesh_data[material_index]:
|
||||
mesh_area += face.area
|
||||
|
||||
loop_indices = face.loop_indices
|
||||
m_faces.append(len(loop_indices))
|
||||
|
||||
for loop_index in loop_indices:
|
||||
loop = data.loops[loop_index]
|
||||
|
||||
vertex = data.vertices[loop.vertex_index]
|
||||
transformed_vertex = transform @ vertex.co * units_scale
|
||||
|
||||
m_verts.extend(
|
||||
[transformed_vertex.x, transformed_vertex.y, transformed_vertex.z]
|
||||
)
|
||||
|
||||
# get and transform the loop normal
|
||||
# try to get split normal, fallback to face normal if not available
|
||||
try:
|
||||
if hasattr(loop, "normal") and len(loop.normal) > 0:
|
||||
# Use split normal from loop
|
||||
loop_normal = normal_transform @ loop.normal
|
||||
else:
|
||||
# Fallback to face normal
|
||||
loop_normal = normal_transform @ face.normal
|
||||
except: # noqa: E722
|
||||
# Final fallback: use face normal
|
||||
loop_normal = normal_transform @ face.normal
|
||||
|
||||
loop_normal.normalize()
|
||||
m_normals.extend([loop_normal.x, loop_normal.y, loop_normal.z])
|
||||
|
||||
# add UV coordinates if available
|
||||
if data.uv_layers.active:
|
||||
uv_data = data.uv_layers.active.data[loop_index]
|
||||
uv = cast(MVector, uv_data.uv)
|
||||
m_texcoords.extend([uv.x, uv.y])
|
||||
|
||||
m_faces.append(vertex_counter)
|
||||
vertex_counter += 1
|
||||
|
||||
speckle_mesh = Mesh(
|
||||
vertices=m_verts,
|
||||
faces=m_faces,
|
||||
colors=[],
|
||||
textureCoordinates=m_texcoords,
|
||||
vertexNormals=m_normals,
|
||||
units=units,
|
||||
)
|
||||
|
||||
if len(m_verts) > 0:
|
||||
speckle_mesh.area = mesh_area
|
||||
|
||||
speckle_mesh.applicationId = get_submesh_id(blender_object, material_index)
|
||||
|
||||
submeshes.append(speckle_mesh)
|
||||
|
||||
return submeshes
|
||||
|
||||
|
||||
def is_closed_mesh(faces: List[int]) -> bool:
|
||||
"""
|
||||
check if a mesh is closed by verifying that each edge is shared by exactly 2 faces.
|
||||
"""
|
||||
edge_counts = {}
|
||||
|
||||
i = 0
|
||||
while i < len(faces):
|
||||
vertex_count = faces[i]
|
||||
for j in range(vertex_count):
|
||||
v1 = faces[i + 1 + j]
|
||||
v2 = faces[i + 1 + ((j + 1) % vertex_count)]
|
||||
edge = tuple(sorted([v1, v2]))
|
||||
edge_counts[edge] = edge_counts.get(edge, 0) + 1
|
||||
|
||||
i += vertex_count + 1
|
||||
|
||||
return all(count == 2 for count in edge_counts.values())
|
||||
@@ -0,0 +1,17 @@
|
||||
from bpy.types import Object
|
||||
from specklepy.objects.geometry import Point
|
||||
|
||||
|
||||
def point_to_speckle(blender_object: Object, scale_factor: float = 1.0) -> Point:
|
||||
assert blender_object.type == "EMPTY", "Object must be an empty."
|
||||
|
||||
location = blender_object.location
|
||||
|
||||
speckle_point = Point(
|
||||
x=location.x * scale_factor,
|
||||
y=location.y * scale_factor,
|
||||
z=location.z * scale_factor,
|
||||
units="", # TODO: implement units in object level
|
||||
)
|
||||
|
||||
return speckle_point
|
||||
@@ -0,0 +1,85 @@
|
||||
from bpy.types import Object
|
||||
from typing import Optional
|
||||
from specklepy.objects.data_objects import BlenderObject
|
||||
from .curve_to_speckle import curve_to_speckle
|
||||
from .mesh_to_speckle import mesh_to_speckle_meshes
|
||||
from .utils import get_object_id, get_curve_element_id
|
||||
|
||||
|
||||
def convert_to_speckle(
|
||||
blender_object: Object,
|
||||
scale_factor: float = 1.0,
|
||||
units: str = "m",
|
||||
apply_modifiers: bool = True,
|
||||
) -> Optional[BlenderObject]:
|
||||
display_value = []
|
||||
properties = {}
|
||||
|
||||
if blender_object.type == "CURVE":
|
||||
# handle curve modifiers apply_modifiers is True
|
||||
if apply_modifiers and blender_object.modifiers:
|
||||
import bpy
|
||||
|
||||
# Convert curve with modifiers to mesh
|
||||
depsgraph = bpy.context.evaluated_depsgraph_get()
|
||||
evaluated_obj = blender_object.evaluated_get(depsgraph)
|
||||
evaluated_mesh = evaluated_obj.to_mesh()
|
||||
|
||||
if evaluated_mesh:
|
||||
meshes = mesh_to_speckle_meshes(
|
||||
blender_object, evaluated_mesh, scale_factor, units
|
||||
)
|
||||
blender_object.to_mesh_clear()
|
||||
if meshes:
|
||||
display_value = meshes
|
||||
else:
|
||||
# curve conversion without modifiers
|
||||
curve_result = curve_to_speckle(blender_object, scale_factor)
|
||||
if curve_result and hasattr(curve_result, "@elements"):
|
||||
display_value = curve_result["@elements"]
|
||||
for i, element in enumerate(display_value):
|
||||
if hasattr(element, "applicationId"):
|
||||
element.applicationId = get_curve_element_id(blender_object, i)
|
||||
elif curve_result:
|
||||
if hasattr(curve_result, "applicationId"):
|
||||
curve_result.applicationId = get_curve_element_id(blender_object, 0)
|
||||
display_value = [curve_result]
|
||||
|
||||
elif blender_object.type == "MESH":
|
||||
# get mesh data - apply modifiers if requested
|
||||
mesh_data = blender_object.data
|
||||
if apply_modifiers and blender_object.modifiers:
|
||||
import bpy
|
||||
|
||||
# use evaluated object to get mesh with modifiers applied
|
||||
depsgraph = bpy.context.evaluated_depsgraph_get()
|
||||
evaluated_obj = blender_object.evaluated_get(depsgraph)
|
||||
evaluated_mesh = evaluated_obj.to_mesh()
|
||||
mesh_data = evaluated_mesh
|
||||
|
||||
meshes = mesh_to_speckle_meshes(blender_object, mesh_data, scale_factor, units)
|
||||
|
||||
if (
|
||||
apply_modifiers
|
||||
and blender_object.modifiers
|
||||
and mesh_data != blender_object.data
|
||||
):
|
||||
blender_object.to_mesh_clear()
|
||||
|
||||
if meshes:
|
||||
display_value = meshes
|
||||
|
||||
if not display_value:
|
||||
return None
|
||||
|
||||
if not isinstance(display_value, list):
|
||||
display_value = [display_value]
|
||||
|
||||
return BlenderObject(
|
||||
name=blender_object.name,
|
||||
type=blender_object.type,
|
||||
displayValue=display_value,
|
||||
applicationId=get_object_id(blender_object),
|
||||
properties=properties,
|
||||
units=units,
|
||||
)
|
||||
@@ -0,0 +1,242 @@
|
||||
import bpy
|
||||
from bpy.types import ID, Object
|
||||
import math
|
||||
from typing import Tuple, Optional
|
||||
|
||||
OBJECT_NAME_SPECKLE_SEPARATOR = " -- "
|
||||
SPECKLE_ID_LENGTH = 32
|
||||
_QUICK_TEST_NAME_LENGTH = SPECKLE_ID_LENGTH + len(OBJECT_NAME_SPECKLE_SEPARATOR)
|
||||
|
||||
|
||||
def to_speckle_name(blender_object: bpy.types.ID) -> str:
|
||||
does_name_contain_id = (
|
||||
len(blender_object.name) > _QUICK_TEST_NAME_LENGTH
|
||||
and OBJECT_NAME_SPECKLE_SEPARATOR in blender_object.name
|
||||
)
|
||||
if does_name_contain_id:
|
||||
return blender_object.name.rsplit(OBJECT_NAME_SPECKLE_SEPARATOR, 1)[0]
|
||||
else:
|
||||
return blender_object.name
|
||||
|
||||
|
||||
"""
|
||||
Python implementation of Blender's NURBS curve generation for to Speckle conversion
|
||||
from: https://blender.stackexchange.com/a/34276
|
||||
based on https://projects.blender.org/blender/blender/src/branch/main/source/blender/blenkernel/intern/curve.cc (check old version)
|
||||
"""
|
||||
|
||||
|
||||
def macro_knotsu(nu: bpy.types.Spline) -> int:
|
||||
return nu.order_u + nu.point_count_u + (nu.order_u - 1 if nu.use_cyclic_u else 0)
|
||||
|
||||
|
||||
def macro_segmentsu(nu: bpy.types.Spline) -> int:
|
||||
return nu.point_count_u if nu.use_cyclic_u else nu.point_count_u - 1
|
||||
|
||||
|
||||
def make_knots(nu: bpy.types.Spline) -> list[float]:
|
||||
knots = [0.0] * macro_knotsu(nu)
|
||||
flag = nu.use_endpoint_u + (nu.use_bezier_u << 1)
|
||||
if nu.use_cyclic_u:
|
||||
calc_knots(knots, nu.point_count_u, nu.order_u, 0)
|
||||
else:
|
||||
calc_knots(knots, nu.point_count_u, nu.order_u, flag)
|
||||
return knots
|
||||
|
||||
|
||||
def calc_knots(knots: list[float], point_count: int, order: int, flag: int) -> None:
|
||||
pts_order = point_count + order
|
||||
if flag == 1: # CU_NURB_ENDPOINT
|
||||
k = 0.0
|
||||
for a in range(1, pts_order + 1):
|
||||
knots[a - 1] = k
|
||||
if a >= order and a <= point_count:
|
||||
k += 1.0
|
||||
elif flag == 2: # CU_NURB_BEZIER
|
||||
if order == 4:
|
||||
k = 0.34
|
||||
for a in range(pts_order):
|
||||
knots[a] = math.floor(k)
|
||||
k += 1.0 / 3.0
|
||||
elif order == 3:
|
||||
k = 0.6
|
||||
for a in range(pts_order):
|
||||
if a >= order and a <= point_count:
|
||||
k += 0.5
|
||||
knots[a] = math.floor(k)
|
||||
else:
|
||||
for a in range(1, len(knots) - 1):
|
||||
knots[a] = a - 1
|
||||
|
||||
knots[-1] = knots[-2]
|
||||
|
||||
|
||||
def basis_nurb(
|
||||
t: float,
|
||||
order: int,
|
||||
point_count: int,
|
||||
knots: list[float],
|
||||
basis: list[float],
|
||||
start: int,
|
||||
end: int,
|
||||
) -> Tuple[int, int]:
|
||||
i1 = i2 = 0
|
||||
orderpluspnts = order + point_count
|
||||
opp2 = orderpluspnts - 1
|
||||
|
||||
# this is for float inaccuracy
|
||||
if t < knots[0]:
|
||||
t = knots[0]
|
||||
elif t > knots[opp2]:
|
||||
t = knots[opp2]
|
||||
|
||||
# this part is order '1'
|
||||
o2 = order + 1
|
||||
for i in range(opp2):
|
||||
if knots[i] != knots[i + 1] and t >= knots[i] and t <= knots[i + 1]:
|
||||
basis[i] = 1.0
|
||||
i1 = i - o2
|
||||
if i1 < 0:
|
||||
i1 = 0
|
||||
i2 = i
|
||||
i += 1
|
||||
while i < opp2:
|
||||
basis[i] = 0.0
|
||||
i += 1
|
||||
break
|
||||
|
||||
else:
|
||||
basis[i] = 0.0
|
||||
|
||||
basis[i] = 0.0 # type: ignore
|
||||
|
||||
# this is order 2, 3, ...
|
||||
for j in range(2, order + 1):
|
||||
if i2 + j >= orderpluspnts:
|
||||
i2 = opp2 - j
|
||||
|
||||
for i in range(i1, i2 + 1):
|
||||
if basis[i] != 0.0:
|
||||
d = ((t - knots[i]) * basis[i]) / (knots[i + j - 1] - knots[i])
|
||||
else:
|
||||
d = 0.0
|
||||
|
||||
if basis[i + 1] != 0.0:
|
||||
e = ((knots[i + j] - t) * basis[i + 1]) / (knots[i + j] - knots[i + 1])
|
||||
else:
|
||||
e = 0.0
|
||||
|
||||
basis[i] = d + e
|
||||
|
||||
start = 1000
|
||||
end = 0
|
||||
|
||||
for i in range(i1, i2 + 1):
|
||||
if basis[i] > 0.0:
|
||||
end = i
|
||||
if start == 1000:
|
||||
start = i
|
||||
|
||||
return start, end
|
||||
|
||||
|
||||
def nurb_make_curve(nu: bpy.types.Spline, resolu: int, stride: int = 3) -> list[float]:
|
||||
""" "BKE_nurb_makeCurve"""
|
||||
EPS = 1e-6
|
||||
coord_index = istart = iend = 0
|
||||
|
||||
coord_array = [0.0] * (3 * nu.resolution_u * macro_segmentsu(nu))
|
||||
sum_array = [0] * nu.point_count_u
|
||||
basisu = [0.0] * macro_knotsu(nu)
|
||||
knots = make_knots(nu)
|
||||
|
||||
resolu = resolu * macro_segmentsu(nu)
|
||||
ustart = knots[nu.order_u - 1]
|
||||
uend = (
|
||||
knots[nu.point_count_u + nu.order_u - 1]
|
||||
if nu.use_cyclic_u
|
||||
else knots[nu.point_count_u]
|
||||
)
|
||||
ustep = (uend - ustart) / (resolu - (0 if nu.use_cyclic_u else 1))
|
||||
cycl = nu.order_u - 1 if nu.use_cyclic_u else 0
|
||||
|
||||
u = ustart
|
||||
while resolu:
|
||||
resolu -= 1
|
||||
istart, iend = basis_nurb(
|
||||
u, nu.order_u, nu.point_count_u + cycl, knots, basisu, istart, iend
|
||||
)
|
||||
|
||||
# /* calc sum */
|
||||
sumdiv = 0.0
|
||||
sum_index = 0
|
||||
pt_index = istart - 1
|
||||
for i in range(istart, iend + 1):
|
||||
if i >= nu.point_count_u:
|
||||
pt_index = i - nu.point_count_u
|
||||
else:
|
||||
pt_index += 1
|
||||
|
||||
sum_array[sum_index] = basisu[i] * nu.points[pt_index].co[3] # type: ignore
|
||||
sumdiv += sum_array[sum_index]
|
||||
sum_index += 1
|
||||
|
||||
if (sumdiv != 0.0) and (sumdiv < 1.0 - EPS or sumdiv > 1.0 + EPS):
|
||||
sum_index = 0
|
||||
for i in range(istart, iend + 1):
|
||||
sum_array[sum_index] /= sumdiv # type: ignore
|
||||
sum_index += 1
|
||||
|
||||
coord_array[coord_index : coord_index + 3] = (0.0, 0.0, 0.0)
|
||||
|
||||
sum_index = 0
|
||||
pt_index = istart - 1
|
||||
for i in range(istart, iend + 1):
|
||||
if i >= nu.point_count_u:
|
||||
pt_index = i - nu.point_count_u
|
||||
else:
|
||||
pt_index += 1
|
||||
|
||||
if sum_array[sum_index] != 0.0:
|
||||
for j in range(3):
|
||||
coord_array[coord_index + j] += (
|
||||
sum_array[sum_index] * nu.points[pt_index].co[j]
|
||||
)
|
||||
sum_index += 1
|
||||
|
||||
coord_index += stride
|
||||
u += ustep
|
||||
|
||||
return coord_array
|
||||
|
||||
|
||||
def get_unique_id(native_object: ID, suffix: Optional[str] = None) -> str:
|
||||
base_id = f"{type(native_object).__name__}:{native_object.name_full}"
|
||||
|
||||
if suffix:
|
||||
return f"{base_id}:{suffix}"
|
||||
|
||||
return base_id
|
||||
|
||||
|
||||
def get_submesh_id(blender_object: Object, material_index: int) -> str:
|
||||
mesh_data = blender_object.data
|
||||
if not mesh_data:
|
||||
return f"Mesh:{blender_object.name_full}_mat{material_index}"
|
||||
|
||||
return f"Mesh:{mesh_data.name_full}_mat{material_index}"
|
||||
|
||||
|
||||
def get_curve_element_id(blender_object: Object, curve_index: int = 0) -> str:
|
||||
curve_data = blender_object.data
|
||||
if not curve_data:
|
||||
return f"Curve:{blender_object.name_full}_curve{curve_index}"
|
||||
|
||||
if curve_index == 0:
|
||||
return f"Curve:{curve_data.name_full}"
|
||||
|
||||
return f"Curve:{curve_data.name_full}_curve{curve_index}"
|
||||
|
||||
|
||||
def get_object_id(blender_object: Object) -> str:
|
||||
return get_unique_id(blender_object)
|
||||
@@ -0,0 +1,226 @@
|
||||
from typing import Tuple, List, Optional, Dict
|
||||
import bpy
|
||||
import mathutils
|
||||
from specklepy.objects import Base
|
||||
from specklepy.objects.graph_traversal.default_traversal import (
|
||||
create_default_traversal_function,
|
||||
)
|
||||
from specklepy.core.api.client import SpeckleClient
|
||||
|
||||
|
||||
def to_rgba(argb_int: int) -> Tuple[float, float, float, float]:
|
||||
"""
|
||||
converts the int representation of a colour into a RGBA tuple
|
||||
"""
|
||||
alpha = ((argb_int >> 24) & 255) / 255
|
||||
red = ((argb_int >> 16) & 255) / 255
|
||||
green = ((argb_int >> 8) & 255) / 255
|
||||
blue = (argb_int & 255) / 255
|
||||
return (red, green, blue, alpha)
|
||||
|
||||
|
||||
def to_argb_int(rgba_color: List[float]) -> int:
|
||||
"""
|
||||
converts an RGBA array to an ARGB integer
|
||||
"""
|
||||
argb_color = rgba_color[-1:] + rgba_color[:3]
|
||||
int_color = [int(val * 255) for val in argb_color]
|
||||
return int.from_bytes(int_color, byteorder="big", signed=True)
|
||||
|
||||
|
||||
def create_material_from_proxy(
|
||||
render_material, material_name: str
|
||||
) -> bpy.types.Material:
|
||||
"""
|
||||
creates a Blender material from a Speckle RenderMaterial
|
||||
"""
|
||||
if material_name in bpy.data.materials:
|
||||
return bpy.data.materials[material_name]
|
||||
|
||||
# create new material
|
||||
material = bpy.data.materials.new(name=material_name)
|
||||
material.use_nodes = True
|
||||
node_tree = material.node_tree
|
||||
nodes = node_tree.nodes
|
||||
|
||||
for node in nodes:
|
||||
nodes.remove(node)
|
||||
|
||||
bsdf = nodes.new(type="ShaderNodeBsdfPrincipled")
|
||||
output = nodes.new(type="ShaderNodeOutputMaterial")
|
||||
|
||||
node_tree.links.new(bsdf.outputs["BSDF"], output.inputs["Surface"])
|
||||
|
||||
if hasattr(render_material, "diffuse"):
|
||||
diffuse_rgba = to_rgba(render_material.diffuse)
|
||||
bsdf.inputs["Base Color"].default_value = (
|
||||
diffuse_rgba[0],
|
||||
diffuse_rgba[1],
|
||||
diffuse_rgba[2],
|
||||
1.0,
|
||||
)
|
||||
|
||||
if hasattr(render_material, "opacity"):
|
||||
opacity = float(render_material.opacity)
|
||||
if opacity < 1.0:
|
||||
material.blend_method = "BLEND"
|
||||
bsdf.inputs["Alpha"].default_value = opacity
|
||||
|
||||
if hasattr(render_material, "metalness"):
|
||||
metalness = float(render_material.metalness)
|
||||
bsdf.inputs["Metallic"].default_value = metalness
|
||||
|
||||
if hasattr(render_material, "roughness"):
|
||||
roughness = float(render_material.roughness)
|
||||
bsdf.inputs["Roughness"].default_value = roughness
|
||||
|
||||
if (
|
||||
hasattr(render_material, "emissive") and render_material.emissive != -16777216
|
||||
): # default black
|
||||
emissive_rgba = to_rgba(render_material.emissive)
|
||||
# only add emission if it's not black (default)
|
||||
if any(val > 0.01 for val in emissive_rgba[:3]):
|
||||
bsdf.inputs["Emission Color"].default_value = (
|
||||
emissive_rgba[0],
|
||||
emissive_rgba[1],
|
||||
emissive_rgba[2],
|
||||
1.0,
|
||||
)
|
||||
bsdf.inputs["Emission Strength"].default_value = 1.0
|
||||
|
||||
# set viewport display color
|
||||
if hasattr(render_material, "diffuse") and hasattr(render_material, "opacity"):
|
||||
material.diffuse_color = (
|
||||
diffuse_rgba[0],
|
||||
diffuse_rgba[1],
|
||||
diffuse_rgba[2],
|
||||
opacity,
|
||||
)
|
||||
|
||||
return material
|
||||
|
||||
|
||||
def transform_matrix(transform: List[float]) -> mathutils.Matrix:
|
||||
"""
|
||||
converts a speckle transform array to a 4x4 matrix (blender needs it)
|
||||
"""
|
||||
|
||||
if len(transform) != 16:
|
||||
raise ValueError(f"Expected transform with 16 values, got {len(transform)}")
|
||||
|
||||
return mathutils.Matrix(
|
||||
(
|
||||
(transform[0], transform[4], transform[8], transform[12]),
|
||||
(transform[1], transform[5], transform[9], transform[13]),
|
||||
(transform[2], transform[6], transform[10], transform[14]),
|
||||
(transform[3], transform[7], transform[11], transform[15]),
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
def build_object_id_map(root_object: Base) -> Dict[str, Base]:
|
||||
"""
|
||||
Builds a dictionary mapping object IDs (both id and applicationId) to objects.
|
||||
"""
|
||||
id_map = {}
|
||||
traversal_function = create_default_traversal_function()
|
||||
|
||||
for traversal_item in traversal_function.traverse(root_object):
|
||||
obj = traversal_item.current
|
||||
|
||||
if hasattr(obj, "id") and obj.id:
|
||||
id_map[obj.id] = obj
|
||||
|
||||
if hasattr(obj, "applicationId") and obj.applicationId:
|
||||
id_map[obj.applicationId] = obj
|
||||
|
||||
return id_map
|
||||
|
||||
|
||||
def find_object_by_id(root_object: Base, target_id: str) -> Optional[Base]:
|
||||
"""
|
||||
finds an object using traversal, checking both id and applicationId
|
||||
"""
|
||||
if hasattr(root_object, "__closure") and root_object.__closure:
|
||||
if target_id in root_object.__closure:
|
||||
if hasattr(root_object, "elements"):
|
||||
for element in root_object.elements:
|
||||
if hasattr(element, "id") and element.id == target_id:
|
||||
return element
|
||||
if (
|
||||
hasattr(element, "referencedId")
|
||||
and element.referencedId == target_id
|
||||
):
|
||||
return find_object_by_id(root_object, element.referencedId)
|
||||
|
||||
if hasattr(root_object, "@elements"):
|
||||
for element in root_object["@elements"]:
|
||||
if hasattr(element, "id") and element.id == target_id:
|
||||
return element
|
||||
if (
|
||||
hasattr(element, "referencedId")
|
||||
and element.referencedId == target_id
|
||||
):
|
||||
return find_object_by_id(root_object, element.referencedId)
|
||||
|
||||
traversal_function = create_default_traversal_function()
|
||||
|
||||
for traversal_item in traversal_function.traverse(root_object):
|
||||
obj = traversal_item.current
|
||||
|
||||
if not hasattr(obj, "id"):
|
||||
continue
|
||||
|
||||
if obj.id == target_id:
|
||||
return obj
|
||||
|
||||
if hasattr(obj, "applicationId"):
|
||||
app_id = obj.applicationId
|
||||
if app_id == target_id:
|
||||
return obj
|
||||
|
||||
def deep_search(search_obj):
|
||||
if hasattr(search_obj, "id") and search_obj.id == target_id:
|
||||
return search_obj
|
||||
|
||||
elements_attrs = ["elements", "@elements"]
|
||||
for attr in elements_attrs:
|
||||
if hasattr(search_obj, attr):
|
||||
elements = getattr(search_obj, attr)
|
||||
if elements and isinstance(elements, list):
|
||||
for element in elements:
|
||||
if hasattr(element, "id") and element.id == target_id:
|
||||
return element
|
||||
if (
|
||||
hasattr(element, "referencedId")
|
||||
and element.referencedId == target_id
|
||||
):
|
||||
ref_obj = find_object_by_id(
|
||||
root_object, element.referencedId
|
||||
)
|
||||
if ref_obj:
|
||||
return ref_obj
|
||||
result = deep_search(element)
|
||||
if result:
|
||||
return result
|
||||
return None
|
||||
|
||||
return deep_search(root_object)
|
||||
|
||||
|
||||
def get_project_workspace_id(client: SpeckleClient, project_id: str) -> Optional[str]:
|
||||
workspace_id = None
|
||||
server_version = client.project.server_version or client.server.version()
|
||||
|
||||
# Local yarn builds of server will report a server version if "dev"
|
||||
# We'll assume that local builds are up-to-date with the latest features
|
||||
if server_version[0] == "dev":
|
||||
maj = 999
|
||||
min = 999
|
||||
else:
|
||||
maj = server_version[0]
|
||||
min = server_version[1]
|
||||
|
||||
if maj > 2 or (maj == 2 and min > 20):
|
||||
workspace_id = client.project.get(project_id).workspace_id
|
||||
return workspace_id
|
||||
@@ -1,72 +0,0 @@
|
||||
from bpy_speckle.clients import speckle_clients
|
||||
|
||||
"""
|
||||
Speckle functions
|
||||
"""
|
||||
|
||||
unit_scale = {
|
||||
"meters": 1.0,
|
||||
"centimeters": 0.01,
|
||||
"millimeters": 0.001,
|
||||
"inches": 0.0254,
|
||||
"feet": 0.3048,
|
||||
"kilometers": 1000.0,
|
||||
"mm": 0.001,
|
||||
"cm": 0.01,
|
||||
"m": 1.0,
|
||||
"km": 1000.0,
|
||||
"in": 0.0254,
|
||||
"ft": 0.3048,
|
||||
"yd": 0.9144,
|
||||
"mi": 1609.340,
|
||||
}
|
||||
|
||||
"""
|
||||
Utility functions
|
||||
"""
|
||||
|
||||
|
||||
def _report(msg):
|
||||
"""
|
||||
Function for printing messages to the console
|
||||
"""
|
||||
print("SpeckleBlender: {}".format(msg))
|
||||
|
||||
|
||||
def get_scale_length(units: str) -> float:
|
||||
if units.lower() in unit_scale.keys():
|
||||
return unit_scale[units]
|
||||
_report("Units <{}> are not supported.".format(units))
|
||||
return 1.0
|
||||
|
||||
|
||||
"""
|
||||
Client, user, and stream functions
|
||||
"""
|
||||
|
||||
|
||||
def _check_speckle_client_user_stream(scene):
|
||||
"""
|
||||
Verify that there is a valid user and stream
|
||||
"""
|
||||
speckle = scene.speckle
|
||||
|
||||
user = (
|
||||
speckle.users[int(speckle.active_user)]
|
||||
if len(speckle.users) > int(speckle.active_user)
|
||||
else None
|
||||
)
|
||||
|
||||
if user is None:
|
||||
print("No users loaded.")
|
||||
|
||||
stream = (
|
||||
user.streams[user.active_stream]
|
||||
if len(user.streams) > user.active_stream
|
||||
else None
|
||||
)
|
||||
|
||||
if stream is None:
|
||||
print("Account contains no streams.")
|
||||
|
||||
return (user, stream)
|
||||
+42
-22
@@ -1,6 +1,7 @@
|
||||
"""
|
||||
Provides uniform and consistent path helpers for `specklepy`
|
||||
"""
|
||||
|
||||
import os
|
||||
import sys
|
||||
from pathlib import Path
|
||||
@@ -55,9 +56,7 @@ def user_application_data_path() -> Path:
|
||||
if sys.platform.startswith("win"):
|
||||
app_data_path = os.getenv("APPDATA")
|
||||
if not app_data_path:
|
||||
raise Exception(
|
||||
"Cannot get appdata path from environment."
|
||||
)
|
||||
raise Exception("Cannot get appdata path from environment.")
|
||||
return Path(app_data_path)
|
||||
else:
|
||||
# try getting the standard XDG_DATA_HOME value
|
||||
@@ -68,9 +67,7 @@ def user_application_data_path() -> Path:
|
||||
else:
|
||||
return _ensure_folder_exists(Path.home(), ".config")
|
||||
except Exception as ex:
|
||||
raise Exception(
|
||||
"Failed to initialize user application data path.", ex
|
||||
)
|
||||
raise Exception("Failed to initialize user application data path.", ex)
|
||||
|
||||
|
||||
def user_speckle_folder_path() -> Path:
|
||||
@@ -90,19 +87,16 @@ def user_speckle_connector_installation_path(host_application: str) -> Path:
|
||||
)
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
print("Starting module dependency installation")
|
||||
print(sys.executable)
|
||||
|
||||
PYTHON_PATH = sys.executable
|
||||
|
||||
|
||||
|
||||
def connector_installation_path(host_application: str) -> Path:
|
||||
connector_installation_path = user_speckle_connector_installation_path(host_application)
|
||||
connector_installation_path = user_speckle_connector_installation_path(
|
||||
host_application
|
||||
)
|
||||
connector_installation_path.mkdir(exist_ok=True, parents=True)
|
||||
|
||||
# set user modules path at beginning of paths for earlier hit
|
||||
@@ -113,7 +107,6 @@ def connector_installation_path(host_application: str) -> Path:
|
||||
return connector_installation_path
|
||||
|
||||
|
||||
|
||||
def is_pip_available() -> bool:
|
||||
try:
|
||||
import_module("pip") # noqa F401
|
||||
@@ -132,13 +125,14 @@ def ensure_pip() -> None:
|
||||
if completed_process.returncode == 0:
|
||||
print("Successfully installed pip")
|
||||
else:
|
||||
raise Exception(f"Failed to install pip, got {completed_process.returncode} return code")
|
||||
raise Exception(
|
||||
f"Failed to install pip, got {completed_process.returncode} return code"
|
||||
)
|
||||
|
||||
|
||||
def get_requirements_path() -> Path:
|
||||
# we assume that a requirements.txt exists next to the __init__.py file
|
||||
path = Path(Path(__file__).parent, "requirements.txt")
|
||||
assert path.exists()
|
||||
return path
|
||||
|
||||
|
||||
@@ -147,30 +141,55 @@ def install_requirements(host_application: str) -> None:
|
||||
# script path. Here we'll install the
|
||||
# dependencies
|
||||
path = connector_installation_path(host_application)
|
||||
print(f"Installing Speckle dependencies to {path}")
|
||||
|
||||
from subprocess import run
|
||||
|
||||
def debugger_is_active() -> bool:
|
||||
"""Return if the debugger is currently active"""
|
||||
return hasattr(sys, "gettrace") and sys.gettrace() is not None
|
||||
|
||||
requirements_path = get_requirements_path()
|
||||
|
||||
is_debug = debugger_is_active()
|
||||
|
||||
if not is_debug and not requirements_path.exists():
|
||||
print("Skipped installing dependencies")
|
||||
return
|
||||
|
||||
print(f"Installing Speckle dependencies to {path}")
|
||||
completed_process = run(
|
||||
[
|
||||
PYTHON_PATH,
|
||||
"-m",
|
||||
"pip",
|
||||
"-q",
|
||||
"--disable-pip-version-check",
|
||||
"install",
|
||||
"--prefer-binary",
|
||||
"--ignore-installed",
|
||||
"--no-compile",
|
||||
"--no-deps",
|
||||
"-t",
|
||||
str(path),
|
||||
"-r",
|
||||
str(get_requirements_path()),
|
||||
str(requirements_path),
|
||||
],
|
||||
capture_output=True,
|
||||
text=True,
|
||||
)
|
||||
|
||||
if completed_process.returncode != 0:
|
||||
m = f"Failed to install dependenices through pip, got {completed_process.returncode} return code"
|
||||
print(completed_process.stdout)
|
||||
print(completed_process.stderr)
|
||||
m = f"Failed to install dependencies through pip, got {completed_process.returncode} return code"
|
||||
print(m)
|
||||
raise Exception(m)
|
||||
|
||||
print("Successfully installed dependencies")
|
||||
|
||||
if not is_debug:
|
||||
requirements_path.unlink()
|
||||
|
||||
|
||||
def install_dependencies(host_application: str) -> None:
|
||||
if not is_pip_available():
|
||||
@@ -184,7 +203,7 @@ def _import_dependencies() -> None:
|
||||
# the code above doesn't work for now, it fails on importing graphql-core
|
||||
# despite that, the connector seams to be working as expected
|
||||
# But it would be nice to make this solution work
|
||||
# it would ensure that all dependencies are fully loaded
|
||||
# it would ensure that all dependencies are fully loaded
|
||||
# requirements = get_requirements_path().read_text()
|
||||
# reqs = [
|
||||
# req.split(" ; ")[0].split("==")[0].split("[")[0].replace("-", "_")
|
||||
@@ -195,6 +214,7 @@ def _import_dependencies() -> None:
|
||||
# print(req)
|
||||
# import_module("specklepy")
|
||||
|
||||
|
||||
def ensure_dependencies(host_application: str) -> None:
|
||||
try:
|
||||
install_dependencies(host_application)
|
||||
@@ -202,6 +222,6 @@ def ensure_dependencies(host_application: str) -> None:
|
||||
_import_dependencies()
|
||||
print("Successfully found dependencies")
|
||||
except ImportError:
|
||||
raise Exception(f"Cannot automatically ensure Speckle dependencies. Please try restarting the host application {host_application}!")
|
||||
|
||||
|
||||
raise Exception(
|
||||
f"Cannot automatically ensure Speckle dependencies. Please try restarting the host application {host_application}!"
|
||||
)
|
||||
|
||||
@@ -1,64 +0,0 @@
|
||||
from .users import LoadUsers, LoadUserStreams, ResetUsers
|
||||
from .object import (
|
||||
UpdateObject,
|
||||
ResetObject,
|
||||
DeleteObject,
|
||||
UploadNgonsAsPolylines,
|
||||
SelectIfSameCustomProperty,
|
||||
SelectIfHasCustomProperty,
|
||||
)
|
||||
from .streams import (
|
||||
ReceiveStreamObjects,
|
||||
SendStreamObjects,
|
||||
ViewStreamDataApi,
|
||||
DeleteStream,
|
||||
SelectOrphanObjects,
|
||||
)
|
||||
from .streams import (
|
||||
UpdateGlobal,
|
||||
AddStreamFromURL,
|
||||
CreateStream,
|
||||
CopyStreamId,
|
||||
CopyCommitId,
|
||||
CopyBranchName,
|
||||
)
|
||||
from .commit import DeleteCommit
|
||||
from .misc import OpenSpeckleGuide, OpenSpeckleTutorials, OpenSpeckleForum
|
||||
|
||||
operator_classes = [
|
||||
LoadUsers,
|
||||
ResetUsers,
|
||||
ReceiveStreamObjects,
|
||||
SendStreamObjects,
|
||||
LoadUserStreams,
|
||||
CopyStreamId,
|
||||
CopyCommitId,
|
||||
CopyBranchName,
|
||||
]
|
||||
|
||||
operator_classes.extend([DeleteCommit])
|
||||
|
||||
operator_classes.extend(
|
||||
[
|
||||
UpdateObject,
|
||||
ResetObject,
|
||||
DeleteObject,
|
||||
UploadNgonsAsPolylines,
|
||||
SelectIfSameCustomProperty,
|
||||
SelectIfHasCustomProperty,
|
||||
]
|
||||
)
|
||||
|
||||
operator_classes.extend(
|
||||
[
|
||||
ViewStreamDataApi,
|
||||
DeleteStream,
|
||||
SelectOrphanObjects,
|
||||
UpdateGlobal,
|
||||
AddStreamFromURL,
|
||||
CreateStream,
|
||||
OpenSpeckleGuide,
|
||||
OpenSpeckleTutorials,
|
||||
OpenSpeckleForum,
|
||||
]
|
||||
)
|
||||
@@ -1,77 +0,0 @@
|
||||
"""
|
||||
Commit operators
|
||||
"""
|
||||
import bpy
|
||||
from bpy.props import BoolProperty
|
||||
from bpy_speckle.functions import _check_speckle_client_user_stream, _report
|
||||
from bpy_speckle.clients import speckle_clients
|
||||
from bpy_speckle.properties.scene import SpeckleSceneSettings
|
||||
|
||||
|
||||
class DeleteCommit(bpy.types.Operator):
|
||||
"""
|
||||
Deletes the selected commit from the selected stream.
|
||||
To execute from code, call: `bpy.ops.speckle.delete_commit(are_you_sure=True)`
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.delete_commit"
|
||||
bl_label = "Delete commit"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
bl_description = "Delete active commit permanently"
|
||||
|
||||
are_you_sure: BoolProperty(
|
||||
name="Confirm",
|
||||
default=False,
|
||||
)
|
||||
|
||||
def draw(self, context):
|
||||
layout = self.layout
|
||||
col = layout.column()
|
||||
col.prop(self, "are_you_sure")
|
||||
|
||||
def invoke(self, context, event):
|
||||
wm = context.window_manager
|
||||
if len(context.scene.speckle.users) > 0:
|
||||
return wm.invoke_props_dialog(self)
|
||||
|
||||
return {"CANCELLED"}
|
||||
|
||||
def execute(self, context):
|
||||
|
||||
if not self.are_you_sure:
|
||||
_report(f"{self.bl_idname}: cancelled by user")
|
||||
return {"CANCELLED"}
|
||||
|
||||
self.are_you_sure = False
|
||||
|
||||
speckle: SpeckleSceneSettings = context.scene.speckle
|
||||
|
||||
user = speckle.get_active_user()
|
||||
if user is None:
|
||||
print(f"{self.bl_idname}: failed - No user selected/found")
|
||||
return {"CANCELLED"}
|
||||
|
||||
stream = user.get_active_stream()
|
||||
if stream is None:
|
||||
print(f"{self.bl_idname}: failed - No stream selected/found")
|
||||
return {"CANCELLED"}
|
||||
|
||||
branch = stream.get_active_branch()
|
||||
if branch is None:
|
||||
print(f"{self.bl_idname}: failed - No branch selected/found")
|
||||
return {"CANCELLED"}
|
||||
|
||||
commit = branch.get_active_commit()
|
||||
if commit is None:
|
||||
print(f"{self.bl_idname}: failed - No commit selected/found")
|
||||
return {"CANCELLED"}
|
||||
|
||||
client = speckle_clients[int(speckle.active_user)]
|
||||
|
||||
deleted = client.commit.delete(stream_id=stream.id, commit_id=commit.id)
|
||||
if not deleted:
|
||||
print(f"{self.bl_idname}: failed - Delete operation failed")
|
||||
return {"CANCELLED"}
|
||||
|
||||
print(f"{self.bl_idname}: succeeded - commit {commit.id} ({commit.message}) has been deleted from stream {stream.id}")
|
||||
return {"FINISHED"}
|
||||
@@ -1,35 +0,0 @@
|
||||
import bpy
|
||||
import webbrowser
|
||||
|
||||
|
||||
class OpenSpeckleGuide(bpy.types.Operator):
|
||||
bl_idname = "speckle.open_speckle_guide"
|
||||
bl_label = "Speckle Guide"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
bl_description = "Browse the documentation on the Speckle Guide"
|
||||
|
||||
def execute(self, context):
|
||||
webbrowser.open("https://speckle.guide/user/blender.html")
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
class OpenSpeckleTutorials(bpy.types.Operator):
|
||||
bl_idname = "speckle.open_speckle_tutorials"
|
||||
bl_label = "Tutorials Portal"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
bl_description = "Visit our tutorials portal for learning resources"
|
||||
|
||||
def execute(self, context):
|
||||
webbrowser.open("https://speckle.systems/tutorials/")
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
class OpenSpeckleForum(bpy.types.Operator):
|
||||
bl_idname = "speckle.open_speckle_forum"
|
||||
bl_label = "Community Forum"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
bl_description = "Ask questions and join the discussion on our community forum"
|
||||
|
||||
def execute(self, context):
|
||||
webbrowser.open("https://speckle.community/")
|
||||
return {"FINISHED"}
|
||||
@@ -1,318 +0,0 @@
|
||||
"""
|
||||
Object operators
|
||||
"""
|
||||
|
||||
import bpy
|
||||
from bpy.props import BoolProperty, EnumProperty
|
||||
from deprecated import deprecated
|
||||
from bpy_speckle.convert.to_speckle import (
|
||||
convert_to_speckle,
|
||||
ngons_to_speckle_polylines,
|
||||
)
|
||||
from bpy_speckle.functions import get_scale_length, _report
|
||||
from bpy_speckle.clients import speckle_clients
|
||||
|
||||
|
||||
class UpdateObject(bpy.types.Operator):
|
||||
"""
|
||||
Update local (receive) or remote (send) object depending on
|
||||
the update direction. If sending, updates the object on the
|
||||
server in-place.
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.update_object"
|
||||
bl_label = "Update Object"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
|
||||
client = None
|
||||
|
||||
def execute(self, context):
|
||||
user = context.scene.speckle.users[int(context.scene.speckle.active_user)]
|
||||
client = speckle_clients[int(context.scene.speckle.active_user)]
|
||||
|
||||
active = context.active_object
|
||||
_report(active)
|
||||
|
||||
if active is not None and active.speckle.enabled:
|
||||
if active.speckle.send_or_receive == "send" and active.speckle.stream_id:
|
||||
sstream = client.streams.get(active.speckle.stream_id)
|
||||
# res = client.StreamGetAsync(active.speckle.stream_id)['resource']
|
||||
# res = client.streams.get(active.speckle.stream_id)
|
||||
|
||||
if sstream is None:
|
||||
_report("Getting stream failed.")
|
||||
return {"CANCELLED"}
|
||||
|
||||
stream_units = "Meters"
|
||||
if sstream.baseProperties:
|
||||
stream_units = sstream.baseProperties.units
|
||||
|
||||
scale = context.scene.unit_settings.scale_length / get_scale_length(
|
||||
stream_units
|
||||
)
|
||||
|
||||
sm = convert_to_speckle(active, scale)
|
||||
|
||||
_report("Updating object {}".format(sm["_id"]))
|
||||
client.objects.update(active.speckle.object_id, sm)
|
||||
|
||||
return {"FINISHED"}
|
||||
|
||||
return {"CANCELLED"}
|
||||
return {"CANCELLED"}
|
||||
|
||||
|
||||
class ResetObject(bpy.types.Operator):
|
||||
"""
|
||||
Reset Speckle object settings
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.reset_object"
|
||||
bl_label = "Reset Object"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
|
||||
def execute(self, context):
|
||||
|
||||
context.object.speckle.send_or_receive = "send"
|
||||
context.object.speckle.stream_id = ""
|
||||
context.object.speckle.object_id = ""
|
||||
context.object.speckle.enabled = False
|
||||
context.view_layer.update()
|
||||
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
class DeleteObject(bpy.types.Operator):
|
||||
"""
|
||||
Delete object from the server and update relevant stream
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.delete_object"
|
||||
bl_label = "Delete Object"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
|
||||
def execute(self, context):
|
||||
client = speckle_clients[int(context.scene.speckle.active_user)]
|
||||
active = context.object
|
||||
if active.speckle.enabled:
|
||||
res = client.StreamGetAsync(active.speckle.stream_id)
|
||||
existing = [
|
||||
x
|
||||
for x in res["resource"]["objects"]
|
||||
if x["_id"] == active.speckle.object_id
|
||||
]
|
||||
if existing is None:
|
||||
return {"CANCELLED"}
|
||||
new_objects = [
|
||||
x
|
||||
for x in res["resource"]["objects"]
|
||||
if x["_id"] != active.speckle.object_id
|
||||
]
|
||||
|
||||
res = client.GetLayers(active.speckle.stream_id)
|
||||
new_layers = res["resource"]["layers"]
|
||||
new_layers[-1]["objectCount"] = new_layers[-1]["objectCount"] - 1
|
||||
new_layers[-1]["topology"] = "0-%s" % new_layers[-1]["objectCount"]
|
||||
|
||||
res = client.StreamUpdateAsync(
|
||||
{"objects": new_objects, "layers": new_layers}, active.speckle.stream_id
|
||||
)
|
||||
res = client.ObjectDeleteAsync(active.speckle.object_id)
|
||||
|
||||
active.speckle.send_or_receive = "send"
|
||||
active.speckle.stream_id = ""
|
||||
active.speckle.object_id = ""
|
||||
active.speckle.enabled = False
|
||||
context.view_layer.update()
|
||||
|
||||
return {"FINISHED"}
|
||||
|
||||
@deprecated
|
||||
class UploadNgonsAsPolylines(bpy.types.Operator):
|
||||
"""
|
||||
Upload mesh ngon faces as polyline outlines
|
||||
TODO: move to another category of specialized operators and fix to work with API 2.0
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.upload_ngons_as_polylines"
|
||||
bl_label = "Upload Ngons As Polylines"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
|
||||
clear_stream: BoolProperty(
|
||||
name="Clear stream",
|
||||
default=False,
|
||||
)
|
||||
|
||||
def execute(self, context):
|
||||
active = context.active_object
|
||||
if active is not None and active.type == "MESH":
|
||||
|
||||
user = context.scene.speckle.users[int(context.scene.speckle.active_user)]
|
||||
client = speckle_clients[int(context.scene.speckle.active_user)]
|
||||
stream = user.streams[user.active_stream]
|
||||
|
||||
# scale = context.scene.unit_settings.scale_length / get_scale_length(
|
||||
# stream.units
|
||||
# )
|
||||
scale = 1.0
|
||||
|
||||
sp = ngons_to_speckle_polylines(active, scale)
|
||||
|
||||
if sp is None:
|
||||
return {"CANCELLED"}
|
||||
|
||||
placeholders = []
|
||||
for polyline in sp:
|
||||
|
||||
res = client.objects.create([polyline])
|
||||
|
||||
if res is None:
|
||||
_report(client.me)
|
||||
continue
|
||||
placeholders.extend(res)
|
||||
|
||||
if not placeholders:
|
||||
return {"CANCELLED"}
|
||||
|
||||
# Get list of existing objects in stream and append new object to list
|
||||
_report("Fetching stream...")
|
||||
sstream = client.streams.get(stream.id)
|
||||
|
||||
if self.clear_stream:
|
||||
_report("Clearing stream...")
|
||||
sstream.objects = placeholders
|
||||
N = 0
|
||||
else:
|
||||
sstream.objects.extend(placeholders)
|
||||
|
||||
N = sstream.layers[-1].objectCount
|
||||
if self.clear_stream:
|
||||
N = 0
|
||||
sstream.layers[-1].objectCount = N + len(placeholders)
|
||||
sstream.layers[-1].topology = "0-%s" % (N + len(placeholders))
|
||||
|
||||
res = client.streams.update(sstream.id, sstream)
|
||||
|
||||
# Update view layer
|
||||
context.view_layer.update()
|
||||
_report("Done.")
|
||||
|
||||
return {"FINISHED"}
|
||||
|
||||
def invoke(self, context, event):
|
||||
wm = context.window_manager
|
||||
return wm.invoke_props_dialog(self)
|
||||
|
||||
def draw(self, context):
|
||||
layout = self.layout
|
||||
layout.prop(self, "clear_stream")
|
||||
|
||||
|
||||
def get_custom_speckle_props(self, context):
|
||||
ignore = ["speckle", "cycles", "cycles_visibility"]
|
||||
|
||||
active = context.active_object
|
||||
if not active:
|
||||
return []
|
||||
|
||||
return [(x, "{}".format(x), "") for x in active.keys()]
|
||||
|
||||
|
||||
class SelectIfSameCustomProperty(bpy.types.Operator):
|
||||
"""
|
||||
Select scene objects if they have the same custom property
|
||||
value as the active object
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.select_if_same_custom_props"
|
||||
bl_label = "Select Identical Custom Props"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
|
||||
custom_prop: EnumProperty(
|
||||
name="Custom properties",
|
||||
description="Available streams associated with user.",
|
||||
items=get_custom_speckle_props,
|
||||
)
|
||||
|
||||
def draw(self, context):
|
||||
layout = self.layout
|
||||
col = layout.column()
|
||||
col.prop(self, "custom_prop")
|
||||
|
||||
def invoke(self, context, event):
|
||||
wm = context.window_manager
|
||||
return wm.invoke_props_dialog(self)
|
||||
|
||||
def execute(self, context):
|
||||
|
||||
active = context.active_object
|
||||
if not active:
|
||||
return {"CANCELLED"}
|
||||
|
||||
if self.custom_prop not in active.keys():
|
||||
return {"CANCELLED"}
|
||||
|
||||
value = active[self.custom_prop]
|
||||
|
||||
_report(
|
||||
"Looking for '{}' property with a value of '{}'.".format(
|
||||
self.custom_prop, value
|
||||
)
|
||||
)
|
||||
|
||||
for obj in bpy.data.objects:
|
||||
|
||||
if self.custom_prop in obj.keys() and obj[self.custom_prop] == value:
|
||||
obj.select_set(True)
|
||||
else:
|
||||
obj.select_set(False)
|
||||
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
class SelectIfHasCustomProperty(bpy.types.Operator):
|
||||
"""
|
||||
Select scene objects if they have the same custom property
|
||||
as the active object, regardless of the value
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.select_if_has_custom_props"
|
||||
bl_label = "Select Same Custom Prop"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
|
||||
custom_prop: EnumProperty(
|
||||
name="Custom properties",
|
||||
description="Custom properties yo",
|
||||
items=get_custom_speckle_props,
|
||||
)
|
||||
|
||||
def draw(self, context):
|
||||
layout = self.layout
|
||||
col = layout.column()
|
||||
col.prop(self, "custom_prop")
|
||||
|
||||
def invoke(self, context, event):
|
||||
wm = context.window_manager
|
||||
return wm.invoke_props_dialog(self)
|
||||
|
||||
def execute(self, context):
|
||||
|
||||
active = context.active_object
|
||||
if not active:
|
||||
return {"CANCELLED"}
|
||||
|
||||
if self.custom_prop not in active.keys():
|
||||
return {"CANCELLED"}
|
||||
|
||||
value = active[self.custom_prop]
|
||||
|
||||
_report("Looking for '{}' property.".format(self.custom_prop))
|
||||
|
||||
for obj in bpy.data.objects:
|
||||
|
||||
if self.custom_prop in obj.keys():
|
||||
obj.select_set(True)
|
||||
else:
|
||||
obj.select_set(False)
|
||||
|
||||
return {"FINISHED"}
|
||||
@@ -1,959 +0,0 @@
|
||||
"""
|
||||
Stream operators
|
||||
"""
|
||||
from itertools import chain
|
||||
from math import radians
|
||||
from typing import Callable, Dict, Iterable, Optional
|
||||
import bpy
|
||||
from bpy_speckle.convert.util import link_object_to_collection_nested
|
||||
from bpy_speckle.properties.scene import SpeckleSceneSettings
|
||||
from specklepy.api.models import Commit
|
||||
import webbrowser
|
||||
from bpy.props import (
|
||||
StringProperty,
|
||||
BoolProperty,
|
||||
EnumProperty,
|
||||
)
|
||||
from bpy.types import Context, Object
|
||||
from bpy_speckle.convert.to_native import (
|
||||
can_convert_to_native,
|
||||
convert_to_native,
|
||||
set_convert_instances_as,
|
||||
)
|
||||
from bpy_speckle.convert.to_speckle import (
|
||||
convert_to_speckle,
|
||||
)
|
||||
from bpy_speckle.functions import (
|
||||
_check_speckle_client_user_stream,
|
||||
get_scale_length,
|
||||
_report,
|
||||
)
|
||||
from bpy_speckle.clients import speckle_clients
|
||||
from bpy_speckle.operators.users import add_user_stream
|
||||
|
||||
from specklepy.api import operations, host_applications
|
||||
from specklepy.api.wrapper import StreamWrapper
|
||||
from specklepy.api.resources.stream import Stream
|
||||
from specklepy.transports.server import ServerTransport
|
||||
from specklepy.objects.geometry import *
|
||||
from specklepy.logging.exceptions import SpeckleException
|
||||
from specklepy.logging import metrics
|
||||
|
||||
|
||||
def get_objects_collections(base: Base) -> Dict[str, list]:
|
||||
"""Create collections based on the dynamic members on a root commit object"""
|
||||
collections = {}
|
||||
for name in base.get_dynamic_member_names():
|
||||
value = base[name]
|
||||
if isinstance(value, list):
|
||||
col = create_collection(name)
|
||||
collections[name] = get_objects_nested_lists(value, col)
|
||||
if isinstance(value, Base):
|
||||
col = create_collection(name)
|
||||
collections[name] = get_objects_collections_recursive(value, col)
|
||||
|
||||
return collections
|
||||
|
||||
|
||||
def get_objects_nested_lists(items: list, parent_col: Optional[bpy.types.Collection] = None) -> List:
|
||||
"""For handling the weird nested lists that come from Grasshopper"""
|
||||
objects = []
|
||||
if not items:
|
||||
return objects
|
||||
|
||||
if isinstance(items[0], list):
|
||||
items = list(chain.from_iterable(items))
|
||||
objects.extend(get_objects_nested_lists(items, parent_col))
|
||||
else:
|
||||
objects = [
|
||||
get_objects_collections_recursive(item, parent_col)
|
||||
for item in items
|
||||
if isinstance(item, Base)
|
||||
]
|
||||
|
||||
return objects
|
||||
|
||||
|
||||
def get_objects_collections_recursive(base: Base, parent_col: Optional[bpy.types.Collection] = None) -> List:
|
||||
"""Recursively create collections based on the dynamic members on nested `Base` objects within the root commit object"""
|
||||
# if it's a convertable (registered) class and not just a plain `Base`, return the object itself
|
||||
if can_convert_to_native(base):
|
||||
return [base]
|
||||
|
||||
# if it's an unknown type, try to drill further down to find convertable objects
|
||||
objects = []
|
||||
|
||||
for name in base.get_dynamic_member_names():
|
||||
value = base[name]
|
||||
if isinstance(value, list):
|
||||
objects.extend(item for item in value if isinstance(item, Base))
|
||||
if isinstance(value, Base):
|
||||
col = parent_col.children.get(name)
|
||||
if not col:
|
||||
col = create_collection(name)
|
||||
try:
|
||||
parent_col.children.link(col)
|
||||
except:
|
||||
_report(
|
||||
f"Problem linking collection {col.name} to parent {parent_col.name}; skipping"
|
||||
)
|
||||
objects.append({name: get_objects_collections_recursive(value, col)})
|
||||
|
||||
return objects
|
||||
|
||||
|
||||
ObjectCallback = Optional[Callable[[bpy.types.Context, Object, Base], Object]]
|
||||
ReceiveCompleteCallback = Optional[Callable[[bpy.types.Context, Dict[str, Object]], None]]
|
||||
|
||||
def get_receive_funcs(context: Context, created_objects: Dict[str, Object]) -> tuple[ObjectCallback, ReceiveCompleteCallback]:
|
||||
"""
|
||||
Fetches the injected callback functions from user specified "Receive Script"
|
||||
"""
|
||||
|
||||
objectCallback: ObjectCallback = None
|
||||
receiveCompleteCallback: ReceiveCompleteCallback = None
|
||||
|
||||
if context.scene.speckle.receive_script in bpy.data.texts:
|
||||
mod = bpy.data.texts[context.scene.speckle.receive_script].as_module()
|
||||
if hasattr(mod, "execute_for_each"):
|
||||
objectCallback = mod.execute_for_each
|
||||
elif hasattr(mod, "execute"):
|
||||
objectCallback = lambda c, o, _ : mod.execute(c.scene, o)
|
||||
|
||||
if hasattr(mod, "execute_for_all"):
|
||||
receiveCompleteCallback = mod.execute_for_all
|
||||
|
||||
|
||||
progress = 0
|
||||
|
||||
def for_each_object(context: bpy.types.Context, obj: Object, base: Base) -> Object:
|
||||
nonlocal progress
|
||||
nonlocal created_objects
|
||||
nonlocal objectCallback
|
||||
|
||||
progress += 1 #NOTE:XXX Progress bar never reaches 100 because func is only called for convertible objects
|
||||
context.window_manager.progress_update(progress)
|
||||
created_objects[obj.name] = obj
|
||||
|
||||
if objectCallback:
|
||||
return objectCallback(context, obj, base)
|
||||
else:
|
||||
return obj
|
||||
|
||||
return (for_each_object, receiveCompleteCallback)
|
||||
|
||||
def bases_to_native(context: bpy.types.Context, collections: Dict[str, list], scale: float, stream_id: str, func: ObjectCallback = None):
|
||||
for col_name, objects in collections.items():
|
||||
col = bpy.data.collections[col_name]
|
||||
existing = get_existing_collection_objs(col)
|
||||
if isinstance(objects, dict):
|
||||
bases_to_native(context, objects, scale, stream_id)
|
||||
elif isinstance(objects, list):
|
||||
for obj in objects:
|
||||
if isinstance(obj, dict):
|
||||
bases_to_native(context, obj, scale, stream_id, func)
|
||||
elif isinstance(obj, list): #FIXME: wtf are these nested if statement, can this not be a recursive call?
|
||||
for item in obj:
|
||||
if isinstance(item, dict):
|
||||
bases_to_native(context, item, scale, stream_id, func)
|
||||
elif isinstance(item, Base):
|
||||
base_to_native(
|
||||
context, item, scale, stream_id, col, existing, func
|
||||
)
|
||||
elif isinstance(obj, Base):
|
||||
base_to_native(context, obj, scale, stream_id, col, existing, func)
|
||||
|
||||
else:
|
||||
_report(
|
||||
f"Something went wrong when receiving collection: {col_name}" #FIXME: undescript report message
|
||||
)
|
||||
|
||||
bpy.context.view_layer.update()
|
||||
|
||||
if context.area:
|
||||
context.area.tag_redraw()
|
||||
|
||||
|
||||
|
||||
def base_to_native(context: bpy.types.Context,
|
||||
base: Base,
|
||||
scale: float,
|
||||
stream_id: str,
|
||||
col: bpy.types.Collection,
|
||||
existing: Dict[str, Object],
|
||||
func: ObjectCallback = None
|
||||
):
|
||||
|
||||
new_objects = convert_to_native(base)
|
||||
|
||||
#NOTE: this code is ancient, and in testing does nothing, so we are removing it.
|
||||
# if hasattr(base, "properties") and base.properties is not None:
|
||||
# new_objects.extend(get_speckle_subobjects(base.properties, scale, base.id))
|
||||
# elif isinstance(base, dict) and "properties" in base.keys():
|
||||
# new_objects.extend(
|
||||
# get_speckle_subobjects(base["properties"], scale, base["id"])
|
||||
# )
|
||||
|
||||
"""
|
||||
Set object Speckle settings
|
||||
"""
|
||||
for new_object in new_objects:
|
||||
if new_object is None:
|
||||
continue
|
||||
|
||||
"""
|
||||
Run injected function
|
||||
"""
|
||||
if func:
|
||||
new_object = func(context, new_object, base) #this base object isn't always the right one for hosted elements! #TODO: may be it now, need to double check!
|
||||
|
||||
if (
|
||||
new_object is None
|
||||
): # If the injected function returned None, then we should ignore this object.
|
||||
_report(f"Script '{func.__module__}' returned None.")
|
||||
continue
|
||||
|
||||
new_object.speckle.stream_id = stream_id
|
||||
new_object.speckle.send_or_receive = "receive"
|
||||
|
||||
if new_object.speckle.object_id in existing.keys():
|
||||
name = existing[new_object.speckle.object_id].name
|
||||
existing[new_object.speckle.object_id].name = f"{name}__deleted"
|
||||
new_object.name = name
|
||||
col.objects.unlink(existing[new_object.speckle.object_id])
|
||||
|
||||
link_object_to_collection_nested(new_object, col)
|
||||
#if new_object.name not in col.objects:
|
||||
#col.objects.link(new_object)
|
||||
|
||||
|
||||
def create_collection(name: str, clear_collection=True) -> bpy.types.Collection:
|
||||
if name in bpy.data.collections:
|
||||
col = bpy.data.collections[name]
|
||||
if clear_collection:
|
||||
for obj in col.objects:
|
||||
col.objects.unlink(obj)
|
||||
else:
|
||||
col = bpy.data.collections.new(name)
|
||||
|
||||
return col
|
||||
|
||||
|
||||
def create_child_collections(parent_col: bpy.types.Collection, children_names: Iterable[str]):
|
||||
for name in children_names:
|
||||
col = create_collection(name)
|
||||
parent_col.children.link(col)
|
||||
|
||||
|
||||
def get_existing_collection_objs(col: bpy.types.Collection) -> Dict[str, bpy.types.Object]:
|
||||
return {
|
||||
obj.speckle.object_id: obj for obj in col.objects if obj.speckle.object_id != ""
|
||||
}
|
||||
|
||||
|
||||
def get_collection_parents(collection: bpy.types.Collection, names: list[str]) -> None:
|
||||
for parent in bpy.data.collections:
|
||||
if collection.name in parent.children.keys():
|
||||
# TODO: this should be rethought to make it clear when this is an IFC delim so we know to replace it
|
||||
# with `/` again on receive
|
||||
names.append(parent.name.replace("/", "::").replace(".", "::"))
|
||||
get_collection_parents(parent, names)
|
||||
|
||||
|
||||
def get_collection_hierarchy(collection: Optional[bpy.types.Collection]) -> list[str]:
|
||||
if not collection:
|
||||
return []
|
||||
names = [collection.name.replace("/", "::").replace(".", "::")]
|
||||
get_collection_parents(collection, names)
|
||||
|
||||
return names
|
||||
|
||||
|
||||
def create_nested_hierarchy(base: Base, hierarchy: List[str], objects: Any):
|
||||
child = base
|
||||
|
||||
while hierarchy:
|
||||
name = hierarchy.pop()
|
||||
if not hasattr(child, name):
|
||||
child[name] = Base()
|
||||
child.add_detachable_attrs({name})
|
||||
child = child[name]
|
||||
|
||||
if not hasattr(child, "@elements"):
|
||||
child["@elements"] = []
|
||||
child["@elements"].extend(objects)
|
||||
|
||||
return base
|
||||
|
||||
#RECEIVE_MODES = [#TODO: modes
|
||||
# ("create", "Create", "Add new geometry, without removing any existing objects"),
|
||||
# ("replace", "Replace", "Replace objects from previous receive operations from the same stream"),
|
||||
# #("update","Update", "") #TODO: update mode!
|
||||
#]
|
||||
|
||||
INSTANCES_SETTINGS = [
|
||||
("collection_instance", "Collection Instace", "Receive Instances as Collection Instances"),
|
||||
("linked_duplicates", "Linked Duplicates", "Receive Instances as Linked Duplicates"),
|
||||
]
|
||||
|
||||
class ReceiveStreamObjects(bpy.types.Operator):
|
||||
"""
|
||||
Receive stream objects
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.receive_stream_objects"
|
||||
bl_label = "Download Stream Objects"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
bl_description = "Receive objects from active stream"
|
||||
|
||||
|
||||
clean_meshes: BoolProperty(name="Clean Meshes", default=False)
|
||||
|
||||
#receive_mode: EnumProperty(items=RECEIVE_MODES, name="Receive Type", default="replace", description="The behaviour of the recieve operation")
|
||||
receive_instances_as: EnumProperty(items=INSTANCES_SETTINGS, name="Receive Instances As", default="collection_instance", description="How to receive speckle Instances")
|
||||
|
||||
|
||||
def draw(self, context):
|
||||
layout = self.layout
|
||||
col = layout.column()
|
||||
col.prop(self, "clean_meshes")
|
||||
#col.prop(self, "receive_mode")
|
||||
col.prop(self, "receive_instances_as")
|
||||
|
||||
def invoke(self, context, event):
|
||||
return context.window_manager.invoke_props_dialog(self)
|
||||
|
||||
@staticmethod
|
||||
def clean_converted_meshes(context: bpy.types.Context, convertedObjects: dict[str, Object]):
|
||||
|
||||
bpy.ops.object.select_all(action='DESELECT')
|
||||
|
||||
active = None
|
||||
for obj in convertedObjects.values():
|
||||
if obj.type != 'MESH':
|
||||
continue
|
||||
|
||||
obj.select_set(True, view_layer=context.scene.view_layers[0])
|
||||
active = obj
|
||||
|
||||
|
||||
if active == None:
|
||||
return
|
||||
context.view_layer.objects.active = active
|
||||
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
bpy.ops.mesh.select_all(action='SELECT')
|
||||
bpy.ops.mesh.dissolve_limited(angle_limit=radians(0.1))
|
||||
|
||||
# Reset state to previous (not quite sure if this is 100% necessary)
|
||||
bpy.ops.object.mode_set(mode='OBJECT')
|
||||
bpy.ops.object.select_all(action='DESELECT')
|
||||
bpy.context.view_layer.objects.active = None
|
||||
|
||||
def execute(self, context):
|
||||
bpy.context.view_layer.objects.active = None
|
||||
|
||||
speckle: SpeckleSceneSettings = context.scene.speckle
|
||||
|
||||
#Get UI Selection
|
||||
user = speckle.get_active_user()
|
||||
if not user:
|
||||
print("No user selected/found")
|
||||
return {"CANCELLED"}
|
||||
|
||||
stream = user.get_active_stream()
|
||||
if not stream:
|
||||
print("No stream selected/found")
|
||||
return {"CANCELLED"}
|
||||
|
||||
branch = stream.get_active_branch()
|
||||
if not branch:
|
||||
print("No branch selected/found")
|
||||
return {"CANCELLED"}
|
||||
|
||||
commit = branch.get_active_commit()
|
||||
if commit is None:
|
||||
print("No commit selected/found")
|
||||
return {"CANCELLED"}
|
||||
|
||||
#Get actual stream data
|
||||
client = speckle_clients[int(speckle.active_user)]
|
||||
|
||||
transport = ServerTransport(stream.id, client)
|
||||
|
||||
metrics.track(
|
||||
metrics.RECEIVE,
|
||||
getattr(transport, "account", None),
|
||||
custom_props={
|
||||
"sourceHostApp": host_applications.get_host_app_from_string(commit.source_application).slug,
|
||||
"sourceHostAppVersion": commit.source_application,
|
||||
"isMultiplayer": commit.author_id != user.id,
|
||||
},
|
||||
)
|
||||
commit_object = operations._untracked_receive(commit.referenced_object, transport)
|
||||
client.commit.received(
|
||||
stream.id,
|
||||
commit.id,
|
||||
source_application="blender",
|
||||
message="received commit from Speckle Blender",
|
||||
)
|
||||
|
||||
context.window_manager.progress_begin(0, commit_object.totalChildrenCount or 1)
|
||||
|
||||
set_convert_instances_as(self.receive_instances_as) #HACK: we need a better way to pass settings down to the converter
|
||||
|
||||
"""
|
||||
Create or get Collection for stream objects
|
||||
"""
|
||||
collections = get_objects_collections(commit_object)
|
||||
|
||||
if not collections:
|
||||
print("Unusual commit structure - did not correctly create collections")
|
||||
return {"CANCELLED"}
|
||||
|
||||
# name = ""
|
||||
# if self.receive_mode == "create":
|
||||
name = "{} [ {} @ {} ]".format(stream.name, branch.name, commit.id) # Matches Rhino "Create" naming
|
||||
# else:
|
||||
# name = stream.name # Doesn't quite match rhino's Update layer naming, but is close enough no?
|
||||
|
||||
col = create_collection(name)
|
||||
col.speckle.stream_id = stream.id
|
||||
col.speckle.units = commit_object.units or "m"
|
||||
|
||||
if col.name not in bpy.context.scene.collection.children:
|
||||
bpy.context.scene.collection.children.link(col)
|
||||
|
||||
for child_col in collections.keys():
|
||||
try:
|
||||
col.children.link(bpy.data.collections[child_col])
|
||||
except:
|
||||
pass
|
||||
"""
|
||||
Set conversion scale from stream units
|
||||
"""
|
||||
scale = (
|
||||
get_scale_length(col.speckle.units)
|
||||
/ context.scene.unit_settings.scale_length
|
||||
)
|
||||
|
||||
"""
|
||||
Get script from text editor for injection
|
||||
"""
|
||||
created_objects = {}
|
||||
(func, on_complete) = get_receive_funcs(context, created_objects)
|
||||
|
||||
|
||||
"""
|
||||
Iterate through retrieved resources
|
||||
"""
|
||||
|
||||
bases_to_native(context, collections, scale, stream.id, func)
|
||||
context.window_manager.progress_end()
|
||||
|
||||
|
||||
if self.clean_meshes:
|
||||
self.clean_converted_meshes(context, created_objects)
|
||||
|
||||
if on_complete:
|
||||
on_complete(context, created_objects)
|
||||
|
||||
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
|
||||
|
||||
class SendStreamObjects(bpy.types.Operator):
|
||||
"""
|
||||
Send stream objects
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.send_stream_objects"
|
||||
bl_label = "Send stream objects"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
bl_description = "Send selected objects to active stream"
|
||||
|
||||
apply_modifiers: BoolProperty(name="Apply modifiers", default=True)
|
||||
commit_message: StringProperty(
|
||||
name="Message",
|
||||
default="Pushed elements from Blender.",
|
||||
)
|
||||
|
||||
def draw(self, context):
|
||||
layout = self.layout
|
||||
col = layout.column()
|
||||
col.prop(self, "commit_message")
|
||||
col.prop(self, "apply_modifiers")
|
||||
|
||||
def invoke(self, context, event):
|
||||
wm = context.window_manager
|
||||
if len(context.scene.speckle.users) > 0:
|
||||
N = len(context.selected_objects)
|
||||
if N == 1:
|
||||
self.commit_message = "Pushed {} element from Blender.".format(N)
|
||||
else:
|
||||
self.commit_message = "Pushed {} elements from Blender.".format(N)
|
||||
return wm.invoke_props_dialog(self)
|
||||
|
||||
return {"CANCELLED"}
|
||||
|
||||
def execute(self, context):
|
||||
|
||||
selected = context.selected_objects
|
||||
|
||||
if len(selected) < 1:
|
||||
return {"CANCELLED"}
|
||||
|
||||
check = _check_speckle_client_user_stream(context.scene)
|
||||
user, bstream = check
|
||||
|
||||
if user is None:
|
||||
return {"CANCELLED"}
|
||||
|
||||
stream = user.streams[user.active_stream]
|
||||
branch = stream.branches[int(stream.branch)]
|
||||
|
||||
client = speckle_clients[int(context.scene.speckle.active_user)]
|
||||
|
||||
# scale = context.scene.unit_settings.scale_length / get_scale_length(
|
||||
# stream.units.lower()
|
||||
# )
|
||||
|
||||
|
||||
scale = 1.0
|
||||
|
||||
units = "m" if bpy.context.scene.unit_settings.system == "METRIC" else "ft"
|
||||
|
||||
"""
|
||||
Get script from text editor for injection
|
||||
"""
|
||||
func = None
|
||||
if context.scene.speckle.send_script in bpy.data.texts:
|
||||
mod = bpy.data.texts[context.scene.speckle.send_script].as_module()
|
||||
if hasattr(mod, "execute"):
|
||||
func = mod.execute
|
||||
|
||||
export = {}
|
||||
|
||||
for obj in selected:
|
||||
|
||||
# if obj.type != 'MESH':
|
||||
# continue
|
||||
|
||||
new_object = obj
|
||||
|
||||
"""
|
||||
Run injected function
|
||||
"""
|
||||
if func:
|
||||
new_object = func(context.scene, obj)
|
||||
|
||||
if (
|
||||
new_object is None
|
||||
): # Make sure that the injected function returned an object
|
||||
new_obj = obj
|
||||
_report("Script '{}' returned None.".format(func.__module__))
|
||||
continue
|
||||
|
||||
_report("Converting {}".format(obj.name))
|
||||
|
||||
converted = convert_to_speckle(
|
||||
obj,
|
||||
scale,
|
||||
units,
|
||||
bpy.context.evaluated_depsgraph_get()
|
||||
if self.apply_modifiers
|
||||
else None,
|
||||
)
|
||||
|
||||
if not converted:
|
||||
continue
|
||||
|
||||
collection_name = obj.users_collection[0].name
|
||||
if not export.get(collection_name):
|
||||
export[collection_name] = []
|
||||
|
||||
export[collection_name].extend(converted)
|
||||
|
||||
base = Base()
|
||||
for name, objects in export.items():
|
||||
collection = bpy.data.collections.get(name)
|
||||
hierarchy = get_collection_hierarchy(collection)
|
||||
create_nested_hierarchy(base, hierarchy, objects)
|
||||
|
||||
transport = ServerTransport(stream.id, client)
|
||||
|
||||
_report(f"Sending to {stream}")
|
||||
obj_id = operations.send(
|
||||
base,
|
||||
[transport],
|
||||
)
|
||||
|
||||
commitId = client.commit.create(
|
||||
stream.id,
|
||||
obj_id,
|
||||
branch.name,
|
||||
message=self.commit_message,
|
||||
source_application="blender",
|
||||
)
|
||||
_report(f"Commit Created {user.server_url}/streams/{stream.id}/commits/{commitId}")
|
||||
|
||||
bpy.ops.speckle.load_user_streams()
|
||||
|
||||
context.view_layer.update()
|
||||
|
||||
if context.area:
|
||||
context.area.tag_redraw()
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
class ViewStreamDataApi(bpy.types.Operator):
|
||||
bl_idname = "speckle.view_stream_data_api"
|
||||
bl_label = "Open Stream in Web"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
bl_description = "View the stream in the web browser"
|
||||
|
||||
def execute(self, context):
|
||||
|
||||
if len(context.scene.speckle.users) > 0:
|
||||
user = context.scene.speckle.users[int(context.scene.speckle.active_user)]
|
||||
if len(user.streams) > 0:
|
||||
stream = user.streams[user.active_stream]
|
||||
|
||||
webbrowser.open("%s/streams/%s" % (user.server_url, stream.id), new=2)
|
||||
return {"FINISHED"}
|
||||
return {"CANCELLED"}
|
||||
|
||||
|
||||
class AddStreamFromURL(bpy.types.Operator):
|
||||
"""
|
||||
Add / select a stream using its url
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.add_stream_from_url"
|
||||
bl_label = "Add stream from URL"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
bl_description = "Add an existing stream by providing its URL"
|
||||
stream_url: StringProperty(
|
||||
name="Stream URL", default="https://speckle.xyz/streams/3073b96e86"
|
||||
)
|
||||
|
||||
def draw(self, context):
|
||||
layout = self.layout
|
||||
col = layout.column()
|
||||
col.prop(self, "stream_url")
|
||||
|
||||
def invoke(self, context, event):
|
||||
wm = context.window_manager
|
||||
if len(context.scene.speckle.users) > 0:
|
||||
return wm.invoke_props_dialog(self)
|
||||
|
||||
return {"CANCELLED"}
|
||||
|
||||
def execute(self, context):
|
||||
speckle = context.scene.speckle
|
||||
|
||||
wrapper = StreamWrapper(self.stream_url)
|
||||
user_index = next(
|
||||
(i for i, u in enumerate(speckle.users) if wrapper.host in u.server_url),
|
||||
None,
|
||||
)
|
||||
if user_index is None:
|
||||
return {"CANCELLED"}
|
||||
speckle.active_user = str(user_index)
|
||||
user = speckle.users[user_index]
|
||||
|
||||
client = speckle_clients[user_index]
|
||||
stream = client.stream.get(wrapper.stream_id, branch_limit=20)
|
||||
if not isinstance(stream, Stream):
|
||||
raise SpeckleException("Could not get the requested stream")
|
||||
|
||||
index, b_stream = next(
|
||||
((i, s) for i, s in enumerate(user.streams) if s.id == stream.id),
|
||||
(None, None),
|
||||
)
|
||||
|
||||
if index is None:
|
||||
add_user_stream(user, stream)
|
||||
user.active_stream, b_stream = next(
|
||||
(i, s) for i, s in enumerate(user.streams) if s.id == stream.id
|
||||
)
|
||||
else:
|
||||
user.active_stream = index
|
||||
|
||||
if wrapper.branch_name:
|
||||
b_index = b_stream.branches.find(wrapper.branch_name)
|
||||
b_stream.branch = str(b_index if b_index != -1 else 0)
|
||||
elif wrapper.commit_id:
|
||||
commit = client.commit.get(wrapper.stream_id, wrapper.commit_id)
|
||||
if isinstance(commit, Commit):
|
||||
b_index = b_stream.branches.find(commit.branchName)
|
||||
if b_index == -1:
|
||||
b_index = 0
|
||||
b_stream.branch = str(b_index)
|
||||
c_index = b_stream.branches[b_index].commits.find(commit.id)
|
||||
b_stream.branches[b_index].commit = str(c_index if c_index != -1 else 0)
|
||||
|
||||
# Update view layer
|
||||
context.view_layer.update()
|
||||
|
||||
if context.area:
|
||||
context.area.tag_redraw()
|
||||
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
class CreateStream(bpy.types.Operator):
|
||||
"""
|
||||
Create new stream
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.create_stream"
|
||||
bl_label = "Create stream"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
bl_description = "Create new stream"
|
||||
|
||||
stream_name: StringProperty(name="Stream name")
|
||||
stream_description: StringProperty(
|
||||
name="Stream description", default="This is a Blender stream."
|
||||
)
|
||||
|
||||
def draw(self, context):
|
||||
layout = self.layout
|
||||
col = layout.column()
|
||||
col.prop(self, "stream_name")
|
||||
col.prop(self, "stream_description")
|
||||
|
||||
def invoke(self, context, event):
|
||||
wm = context.window_manager
|
||||
if len(context.scene.speckle.users) > 0:
|
||||
return wm.invoke_props_dialog(self)
|
||||
|
||||
return {"CANCELLED"}
|
||||
|
||||
def execute(self, context):
|
||||
|
||||
check = _check_speckle_client_user_stream(context.scene)
|
||||
if check is None:
|
||||
return {"CANCELLED"}
|
||||
|
||||
user, bstream = check
|
||||
|
||||
client = speckle_clients[int(context.scene.speckle.active_user)]
|
||||
|
||||
client.stream.create(
|
||||
name=self.stream_name, description=self.stream_description, is_public=True
|
||||
)
|
||||
|
||||
bpy.ops.speckle.load_user_streams()
|
||||
user.active_stream = user.streams.find(self.stream_name)
|
||||
|
||||
# Update view layer
|
||||
context.view_layer.update()
|
||||
|
||||
if context.area:
|
||||
context.area.tag_redraw()
|
||||
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
class DeleteStream(bpy.types.Operator):
|
||||
"""
|
||||
Delete stream
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.delete_stream"
|
||||
bl_label = "Delete stream"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
bl_description = "Delete selected stream permanently"
|
||||
|
||||
are_you_sure: BoolProperty(
|
||||
name="Confirm",
|
||||
default=False,
|
||||
)
|
||||
|
||||
delete_collection: BoolProperty(name="Delete collection", default=False)
|
||||
|
||||
def draw(self, context):
|
||||
layout = self.layout
|
||||
col = layout.column()
|
||||
col.prop(self, "are_you_sure")
|
||||
col.prop(self, "delete_collection")
|
||||
|
||||
def invoke(self, context, event):
|
||||
wm = context.window_manager
|
||||
if len(context.scene.speckle.users) > 0:
|
||||
return wm.invoke_props_dialog(self)
|
||||
|
||||
return {"CANCELLED"}
|
||||
|
||||
def execute(self, context):
|
||||
|
||||
if not self.are_you_sure:
|
||||
return {"CANCELLED"}
|
||||
|
||||
self.are_you_sure = False
|
||||
|
||||
check = _check_speckle_client_user_stream(context.scene)
|
||||
if check is None:
|
||||
return {"CANCELLED"}
|
||||
|
||||
user, stream = check
|
||||
client = speckle_clients[int(context.scene.speckle.active_user)]
|
||||
|
||||
client.stream.delete(id=stream.id)
|
||||
|
||||
if self.delete_collection:
|
||||
col_name = "SpeckleStream_{}_{}".format(stream.name, stream.id)
|
||||
if col_name in bpy.data.collections:
|
||||
collection = bpy.data.collections[col_name]
|
||||
bpy.data.collections.remove(collection)
|
||||
|
||||
bpy.ops.speckle.load_user_streams()
|
||||
context.view_layer.update()
|
||||
|
||||
if context.area:
|
||||
context.area.tag_redraw()
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
class SelectOrphanObjects(bpy.types.Operator):
|
||||
"""
|
||||
Select Speckle objects that don't belong to any stream
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.select_orphans"
|
||||
bl_label = "Select orphaned objects"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
bl_description = "Select Speckle objects that don't belong to any stream"
|
||||
|
||||
def draw(self, context):
|
||||
layout = self.layout
|
||||
|
||||
def execute(self, context):
|
||||
|
||||
for o in context.scene.objects:
|
||||
if (
|
||||
o.speckle.stream_id
|
||||
and o.speckle.stream_id not in context.scene["speckle_streams"]
|
||||
):
|
||||
o.select = True
|
||||
else:
|
||||
o.select = False
|
||||
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
class UpdateGlobal(bpy.types.Operator):
|
||||
"""
|
||||
DEPRECATED
|
||||
Update all Speckle objects
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.update_global"
|
||||
bl_label = "Update Global"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
bl_description = "Update all Speckle objects"
|
||||
|
||||
client = None
|
||||
|
||||
def draw(self, context):
|
||||
layout = self.layout
|
||||
row = layout.row()
|
||||
label = row.label(text="Update everything.")
|
||||
|
||||
def execute(self, context):
|
||||
|
||||
client = context.scene.speckle.client
|
||||
|
||||
profiles = client.load_local_profiles()
|
||||
if len(profiles) < 1:
|
||||
raise ValueError("No profiles found.")
|
||||
client.use_existing_profile(sorted(profiles.keys())[0])
|
||||
context.scene.speckle.user = sorted(profiles.keys())[0]
|
||||
|
||||
for obj in context.scene.objects:
|
||||
if obj.speckle.enabled:
|
||||
UpdateObject(context.scene.speckle_client, obj)
|
||||
|
||||
context.scene.update()
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
class CopyStreamId(bpy.types.Operator):
|
||||
"""
|
||||
Copy stream ID to clipboard
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.stream_copy_id"
|
||||
bl_label = "Copy stream ID"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
bl_description = "Copy stream ID to clipboard"
|
||||
|
||||
def execute(self, context):
|
||||
speckle = context.scene.speckle
|
||||
|
||||
if len(speckle.users) < 1:
|
||||
return {"CANCELLED"}
|
||||
user = speckle.users[int(speckle.active_user)]
|
||||
if len(user.streams) < 1:
|
||||
return {"CANCELLED"}
|
||||
stream = user.streams[user.active_stream]
|
||||
bpy.context.window_manager.clipboard = stream.id
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
class CopyCommitId(bpy.types.Operator):
|
||||
"""
|
||||
Copy commit ID to clipboard
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.commit_copy_id"
|
||||
bl_label = "Copy commit ID"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
bl_description = "Copy commit ID to clipboard"
|
||||
|
||||
def execute(self, context):
|
||||
speckle = context.scene.speckle
|
||||
|
||||
if len(speckle.users) < 1:
|
||||
return {"CANCELLED"}
|
||||
user = speckle.users[int(speckle.active_user)]
|
||||
if len(user.streams) < 1:
|
||||
return {"CANCELLED"}
|
||||
stream = user.streams[user.active_stream]
|
||||
if len(stream.branches) < 1:
|
||||
return {"CANCELLED"}
|
||||
branch = stream.branches[int(stream.branch)]
|
||||
if len(branch.commits) < 1:
|
||||
return {"CANCELLED"}
|
||||
commit = branch.commits[int(branch.commit)]
|
||||
bpy.context.window_manager.clipboard = commit.id
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
class CopyBranchName(bpy.types.Operator):
|
||||
"""
|
||||
Copy branch name to clipboard
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.branch_copy_name"
|
||||
bl_label = "Copy branch name"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
bl_description = "Copy branch name to clipboard"
|
||||
|
||||
def execute(self, context):
|
||||
speckle = context.scene.speckle
|
||||
|
||||
if len(speckle.users) < 1:
|
||||
return {"CANCELLED"}
|
||||
user = speckle.users[int(speckle.active_user)]
|
||||
if len(user.streams) < 1:
|
||||
return {"CANCELLED"}
|
||||
stream = user.streams[user.active_stream]
|
||||
if len(stream.branches) < 1:
|
||||
return {"CANCELLED"}
|
||||
branch = stream.branches[int(stream.branch)]
|
||||
bpy.context.window_manager.clipboard = branch.name
|
||||
return {"FINISHED"}
|
||||
@@ -1,161 +0,0 @@
|
||||
"""
|
||||
User account operators
|
||||
"""
|
||||
import bpy
|
||||
from bpy_speckle.functions import _report
|
||||
from bpy_speckle.clients import speckle_clients
|
||||
from bpy_speckle.properties.scene import SpeckleCommitObject, SpeckleSceneSettings
|
||||
from specklepy.api.client import SpeckleClient
|
||||
from specklepy.api.models import Stream, User
|
||||
from specklepy.api.credentials import get_local_accounts
|
||||
from datetime import datetime
|
||||
|
||||
class ResetUsers(bpy.types.Operator):
|
||||
"""
|
||||
Reset loaded users
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.users_reset"
|
||||
bl_label = "Reset users"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
|
||||
def execute(self, context):
|
||||
self.reset_ui(context)
|
||||
|
||||
bpy.context.view_layer.update()
|
||||
if context.area:
|
||||
context.area.tag_redraw()
|
||||
return {"FINISHED"}
|
||||
|
||||
@staticmethod
|
||||
def reset_ui(context: bpy.types.Context):
|
||||
speckle: SpeckleSceneSettings = context.scene.speckle
|
||||
|
||||
speckle.users.clear()
|
||||
speckle_clients.clear()
|
||||
|
||||
class LoadUsers(bpy.types.Operator):
|
||||
"""
|
||||
Load all users from local user database
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.users_load"
|
||||
bl_label = "Load users"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
|
||||
def execute(self, context):
|
||||
|
||||
_report("Loading users...")
|
||||
|
||||
speckle: SpeckleSceneSettings = context.scene.speckle
|
||||
users = speckle.users
|
||||
|
||||
ResetUsers.reset_ui(context)
|
||||
|
||||
profiles = get_local_accounts()
|
||||
active_user_index = 0
|
||||
|
||||
for profile in profiles:
|
||||
user = users.add()
|
||||
user.server_name = profile.serverInfo.name or "Speckle Server"
|
||||
user.server_url = profile.serverInfo.url
|
||||
user.id = profile.userInfo.id
|
||||
user.name = profile.userInfo.name
|
||||
user.email = profile.userInfo.email
|
||||
user.company = profile.userInfo.company or ""
|
||||
try:
|
||||
client = SpeckleClient(
|
||||
host=profile.serverInfo.url,
|
||||
use_ssl="https" in profile.serverInfo.url,
|
||||
)
|
||||
client.authenticate_with_account(profile)
|
||||
speckle_clients.append(client)
|
||||
except Exception as ex:
|
||||
_report(ex)
|
||||
users.remove(len(users) - 1)
|
||||
if profile.isDefault:
|
||||
active_user_index = len(users) - 1
|
||||
|
||||
#speckle.active_user_index = int(speckle.active_user) #TODO Wtf is this?
|
||||
speckle.active_user = str(active_user_index)
|
||||
bpy.context.view_layer.update()
|
||||
|
||||
if context.area:
|
||||
context.area.tag_redraw()
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
def add_user_stream(user: User, stream: Stream):
|
||||
s = user.streams.add()
|
||||
s.name = stream.name
|
||||
s.id = stream.id
|
||||
s.description = stream.description
|
||||
|
||||
if not stream.branches:
|
||||
return
|
||||
|
||||
# branches = [branch for branch in stream.branches.items if branch.name != "globals"]
|
||||
for b in stream.branches.items:
|
||||
branch = s.branches.add()
|
||||
branch.name = b.name
|
||||
|
||||
if not b.commits:
|
||||
continue
|
||||
|
||||
for c in b.commits.items:
|
||||
commit: SpeckleCommitObject = branch.commits.add()
|
||||
commit.id = commit.name = c.id
|
||||
commit.message = c.message or ""
|
||||
commit.author_name = c.authorName
|
||||
commit.author_id = c.authorId
|
||||
commit.created_at = datetime.strftime(c.createdAt, "%Y-%m-%d %H:%M:%S.%f%Z")
|
||||
commit.source_application = str(c.sourceApplication)
|
||||
commit.referenced_object = c.referencedObject
|
||||
|
||||
if hasattr(s, "baseProperties"):
|
||||
s.units = stream.baseProperties.units
|
||||
else:
|
||||
s.units = "Meters"
|
||||
|
||||
|
||||
class LoadUserStreams(bpy.types.Operator):
|
||||
"""
|
||||
Load all available streams for active user user
|
||||
"""
|
||||
|
||||
bl_idname = "speckle.load_user_streams"
|
||||
bl_label = "Load user streams"
|
||||
bl_options = {"REGISTER", "UNDO"}
|
||||
bl_description = "(Re)load all available user streams"
|
||||
|
||||
def execute(self, context):
|
||||
speckle = context.scene.speckle
|
||||
|
||||
if len(speckle.users) > 0:
|
||||
user = speckle.users[int(context.scene.speckle.active_user)]
|
||||
client = speckle_clients[int(context.scene.speckle.active_user)]
|
||||
|
||||
try:
|
||||
streams = client.stream.list(stream_limit=20)
|
||||
except Exception as e:
|
||||
_report(f"Failed to retrieve streams: {e}")
|
||||
return
|
||||
if not streams:
|
||||
_report("Failed to retrieve streams.")
|
||||
return
|
||||
|
||||
user.streams.clear()
|
||||
|
||||
default_units = "Meters"
|
||||
|
||||
for s in streams:
|
||||
sstream = client.stream.get(id=s.id, branch_limit=20)
|
||||
add_user_stream(user, sstream)
|
||||
|
||||
bpy.context.view_layer.update()
|
||||
|
||||
return {"FINISHED"}
|
||||
|
||||
if context.area:
|
||||
context.area.tag_redraw()
|
||||
return {"CANCELLED"}
|
||||
@@ -1,23 +0,0 @@
|
||||
from .scene import (
|
||||
SpeckleSceneSettings,
|
||||
SpeckleSceneObject,
|
||||
SpeckleUserObject,
|
||||
SpeckleStreamObject,
|
||||
SpeckleBranchObject,
|
||||
SpeckleCommitObject,
|
||||
)
|
||||
from .object import SpeckleObjectSettings
|
||||
from .collection import SpeckleCollectionSettings
|
||||
from .addon import SpeckleAddonPreferences
|
||||
|
||||
property_classes = [
|
||||
SpeckleSceneObject,
|
||||
SpeckleCommitObject,
|
||||
SpeckleBranchObject,
|
||||
SpeckleStreamObject,
|
||||
SpeckleUserObject,
|
||||
SpeckleSceneSettings,
|
||||
SpeckleObjectSettings,
|
||||
SpeckleCollectionSettings,
|
||||
SpeckleAddonPreferences,
|
||||
]
|
||||
@@ -1,17 +0,0 @@
|
||||
"""
|
||||
Addon properties
|
||||
"""
|
||||
import bpy
|
||||
|
||||
|
||||
class SpeckleAddonPreferences(bpy.types.AddonPreferences):
|
||||
"""
|
||||
Add-on preferences
|
||||
TODO: add any preferences that might be relevant here
|
||||
"""
|
||||
|
||||
bl_idname = __package__
|
||||
|
||||
def draw(self, context):
|
||||
layout = self.layout
|
||||
layout.label(text="SpeckleBlender preferences")
|
||||
@@ -1,19 +0,0 @@
|
||||
"""
|
||||
Collection properties
|
||||
"""
|
||||
import bpy
|
||||
|
||||
|
||||
class SpeckleCollectionSettings(bpy.types.PropertyGroup):
|
||||
enabled: bpy.props.BoolProperty(default=False, name="Enabled")
|
||||
|
||||
send_or_receive: bpy.props.EnumProperty(
|
||||
name="Mode",
|
||||
items=(
|
||||
("send", "Send", "Send data to Speckle server."),
|
||||
("receive", "Receive", "Receive data from Speckle server."),
|
||||
),
|
||||
)
|
||||
stream_id: bpy.props.StringProperty(default="")
|
||||
name: bpy.props.StringProperty(default="")
|
||||
units: bpy.props.StringProperty(default="")
|
||||
@@ -1,18 +0,0 @@
|
||||
"""
|
||||
Object properties
|
||||
"""
|
||||
import bpy
|
||||
|
||||
|
||||
class SpeckleObjectSettings(bpy.types.PropertyGroup):
|
||||
enabled: bpy.props.BoolProperty(default=False, name="Enabled")
|
||||
|
||||
send_or_receive: bpy.props.EnumProperty(
|
||||
name="Mode",
|
||||
items=(
|
||||
("send", "Send", "Send data to Speckle server."),
|
||||
("receive", "Receive", "Receive data from Speckle server."),
|
||||
),
|
||||
)
|
||||
stream_id: bpy.props.StringProperty(default="")
|
||||
object_id: bpy.props.StringProperty(default="")
|
||||
@@ -1,159 +0,0 @@
|
||||
"""
|
||||
Scene properties
|
||||
"""
|
||||
from typing import Optional
|
||||
import bpy
|
||||
from bpy.props import (
|
||||
StringProperty,
|
||||
BoolProperty,
|
||||
FloatProperty,
|
||||
CollectionProperty,
|
||||
EnumProperty,
|
||||
IntProperty,
|
||||
PointerProperty,
|
||||
)
|
||||
|
||||
|
||||
class SpeckleSceneObject(bpy.types.PropertyGroup):
|
||||
name: bpy.props.StringProperty(default="")
|
||||
|
||||
|
||||
class SpeckleCommitObject(bpy.types.PropertyGroup):
|
||||
id: StringProperty(default="")
|
||||
message: StringProperty(default="")
|
||||
author_name: StringProperty(default="")
|
||||
author_id: StringProperty(default="")
|
||||
created_at: StringProperty(default="")
|
||||
source_application: StringProperty(default="")
|
||||
referenced_object: StringProperty(default="")
|
||||
|
||||
|
||||
class SpeckleBranchObject(bpy.types.PropertyGroup):
|
||||
def get_commits(self, context):
|
||||
if self.commits != None and len(self.commits) > 0:
|
||||
return [
|
||||
(str(i), commit.id, commit.message, i)
|
||||
for i, commit in enumerate(self.commits)
|
||||
]
|
||||
return [("0", "<none>", "<none>", 0)]
|
||||
|
||||
name: StringProperty(default="main")
|
||||
commits: CollectionProperty(type=SpeckleCommitObject)
|
||||
commit: EnumProperty(
|
||||
name="Commit",
|
||||
description="Active commit",
|
||||
items=get_commits,
|
||||
)
|
||||
|
||||
def get_active_commit(self) -> Optional[SpeckleCommitObject]:
|
||||
selected_index = int(self.commit)
|
||||
if 0 <= selected_index < len(self.commits):
|
||||
return self.commits[selected_index]
|
||||
return None
|
||||
|
||||
|
||||
class SpeckleStreamObject(bpy.types.PropertyGroup):
|
||||
def get_branches(self, context):
|
||||
if self.branches:
|
||||
return [
|
||||
(str(i), branch.name, branch.name, i)
|
||||
for i, branch in enumerate(self.branches)
|
||||
if branch.name != "globals"
|
||||
]
|
||||
return [("0", "<none>", "<none>", 0)]
|
||||
|
||||
name: StringProperty(default="SpeckleStream")
|
||||
description: StringProperty(default="No description provided.")
|
||||
id: StringProperty(default="")
|
||||
units: StringProperty(default="Meters")
|
||||
query: StringProperty(default="")
|
||||
branches: CollectionProperty(type=SpeckleBranchObject)
|
||||
branch: EnumProperty(
|
||||
name="Branch",
|
||||
description="Active branch",
|
||||
items=get_branches,
|
||||
)
|
||||
|
||||
def get_active_branch(self) -> Optional[SpeckleBranchObject]:
|
||||
selected_index = int(self.branch)
|
||||
if 0 <= selected_index < len(self.branches):
|
||||
return self.branches[selected_index]
|
||||
return None
|
||||
|
||||
|
||||
class SpeckleUserObject(bpy.types.PropertyGroup):
|
||||
server_name: StringProperty(default="SpeckleXYZ")
|
||||
server_url: StringProperty(default="https://speckle.xyz")
|
||||
id: StringProperty(default="")
|
||||
name: StringProperty(default="Speckle User")
|
||||
email: StringProperty(default="user@speckle.xyz")
|
||||
company: StringProperty(default="SpeckleSystems")
|
||||
streams: CollectionProperty(type=SpeckleStreamObject)
|
||||
active_stream: IntProperty(default=0)
|
||||
|
||||
def get_active_stream(self) -> Optional[SpeckleStreamObject]:
|
||||
selected_index = int(self.active_stream)
|
||||
if 0 <= selected_index < len(self.streams):
|
||||
return self.streams[selected_index]
|
||||
return None
|
||||
|
||||
class SpeckleSceneSettings(bpy.types.PropertyGroup):
|
||||
def get_scripts(self, context):
|
||||
return [
|
||||
("<none>", "<none>", "<none>"),
|
||||
*[(t.name, t.name, t.name) for t in bpy.data.texts],
|
||||
]
|
||||
|
||||
streams: EnumProperty(
|
||||
name="Available streams",
|
||||
description="Available streams associated with user.",
|
||||
items=[],
|
||||
)
|
||||
|
||||
users: CollectionProperty(type=SpeckleUserObject)
|
||||
|
||||
def get_users(self, context):
|
||||
return [
|
||||
(str(i), "{} ({})".format(user.email, user.server_name), user.server_url, i)
|
||||
for i, user in enumerate(self.users)
|
||||
]
|
||||
|
||||
def set_user(self, context):
|
||||
bpy.ops.speckle.load_user_streams()
|
||||
|
||||
active_user: EnumProperty(
|
||||
items=get_users,
|
||||
name="Account",
|
||||
description="Select account",
|
||||
update=set_user,
|
||||
get=None,
|
||||
set=None,
|
||||
)
|
||||
|
||||
objects: CollectionProperty(type=SpeckleSceneObject)
|
||||
|
||||
scale: FloatProperty(default=0.001)
|
||||
|
||||
user: StringProperty(
|
||||
name="User",
|
||||
description="Current user.",
|
||||
default="Speckle User",
|
||||
)
|
||||
|
||||
receive_script: EnumProperty(
|
||||
name="Receive script",
|
||||
description="Script to run when receiving stream objects.",
|
||||
items=get_scripts,
|
||||
)
|
||||
|
||||
send_script: EnumProperty(
|
||||
name="Send script",
|
||||
description="Script to run when sending stream objects.",
|
||||
items=get_scripts,
|
||||
)
|
||||
|
||||
def get_active_user(self) -> Optional[SpeckleUserObject]:
|
||||
selected_index = int(self.active_user)
|
||||
if 0 <= selected_index < len(self.users):
|
||||
return self.users[selected_index]
|
||||
return None
|
||||
@@ -1,18 +0,0 @@
|
||||
from .object import OBJECT_PT_speckle
|
||||
from .view3d import (
|
||||
VIEW3D_UL_SpeckleUsers,
|
||||
VIEW3D_UL_SpeckleStreams,
|
||||
VIEW3D_PT_SpeckleUser,
|
||||
VIEW3D_PT_SpeckleStreams,
|
||||
VIEW3D_PT_SpeckleActiveStream,
|
||||
VIEW3D_PT_SpeckleHelp,
|
||||
)
|
||||
|
||||
ui_classes = [
|
||||
VIEW3D_PT_SpeckleUser,
|
||||
VIEW3D_PT_SpeckleStreams,
|
||||
VIEW3D_PT_SpeckleActiveStream,
|
||||
VIEW3D_UL_SpeckleUsers,
|
||||
VIEW3D_UL_SpeckleStreams,
|
||||
VIEW3D_PT_SpeckleHelp,
|
||||
]
|
||||
@@ -1,35 +0,0 @@
|
||||
"""
|
||||
Object UI elements
|
||||
"""
|
||||
|
||||
import bpy
|
||||
from bpy.props import (
|
||||
StringProperty,
|
||||
BoolProperty,
|
||||
FloatProperty,
|
||||
CollectionProperty,
|
||||
EnumProperty,
|
||||
)
|
||||
|
||||
|
||||
class OBJECT_PT_speckle(bpy.types.Panel):
|
||||
bl_space_type = "PROPERTIES"
|
||||
# bl_idname = 'OBJECT_PT_speckle'
|
||||
bl_region_type = "WINDOW"
|
||||
bl_context = "object"
|
||||
bl_label = "Speckle"
|
||||
|
||||
def draw_header(self, context):
|
||||
self.layout.prop(context.object.speckle, "enabled", text="")
|
||||
|
||||
def draw(self, context):
|
||||
ob = context.object
|
||||
layout = self.layout
|
||||
layout.active = ob.speckle.enabled
|
||||
col = layout.column()
|
||||
col.prop(ob.speckle, "send_or_receive", expand=True)
|
||||
col.prop(ob.speckle, "stream_id", text="Stream ID")
|
||||
col.prop(ob.speckle, "object_id", text="Object ID")
|
||||
col.operator("speckle.update_object", text="Update")
|
||||
col.operator("speckle.reset_object", text="Reset")
|
||||
col.operator("speckle.delete_object", text="Delete")
|
||||
@@ -1,271 +0,0 @@
|
||||
"""
|
||||
Speckle UI elements for the 3d viewport
|
||||
"""
|
||||
|
||||
|
||||
import bpy
|
||||
from bpy.props import (
|
||||
StringProperty,
|
||||
BoolProperty,
|
||||
FloatProperty,
|
||||
CollectionProperty,
|
||||
EnumProperty,
|
||||
)
|
||||
|
||||
from datetime import datetime
|
||||
|
||||
"""
|
||||
Compatibility
|
||||
TODO: evaluate if we should still support Blender <2.80
|
||||
"""
|
||||
|
||||
Region = "TOOLS" if bpy.app.version < (2, 80, 0) else "UI"
|
||||
|
||||
|
||||
def wrap(width, text):
|
||||
"""
|
||||
Split strings into width for
|
||||
wrapping
|
||||
"""
|
||||
lines = []
|
||||
|
||||
arr = text.split()
|
||||
lengthSum = 0
|
||||
|
||||
line = []
|
||||
for var in arr:
|
||||
lengthSum += len(var) + 1
|
||||
if lengthSum <= width:
|
||||
line.append(var)
|
||||
else:
|
||||
lines.append(" ".join(line))
|
||||
line = [var]
|
||||
lengthSum = len(var)
|
||||
|
||||
lines.append(" ".join(line))
|
||||
|
||||
return lines
|
||||
|
||||
|
||||
def get_available_users(self, context):
|
||||
"""
|
||||
Function to populate users list
|
||||
"""
|
||||
return [(a, a, a.name) for a in context.scene.speckle.users]
|
||||
|
||||
|
||||
class VIEW3D_UL_SpeckleUsers(bpy.types.UIList):
|
||||
"""
|
||||
Speckle user list
|
||||
"""
|
||||
|
||||
def draw_item(self, context, layout, data, user, active_data, active_propname):
|
||||
if self.layout_type in {"DEFAULT", "COMPACT"}:
|
||||
if user:
|
||||
# layout.prop(user, "name", text=user.name, emboss=False, icon_value=0)
|
||||
layout.label(
|
||||
text=user.name + " (" + user.email + ")",
|
||||
translate=False,
|
||||
icon_value=0,
|
||||
)
|
||||
else:
|
||||
layout.label(text="", translate=False, icon_value=0)
|
||||
|
||||
elif self.layout_type in {"GRID"}:
|
||||
layout.alignment = "CENTER"
|
||||
layout.label(text="Users", icon_value=0)
|
||||
|
||||
|
||||
class VIEW3D_UL_SpeckleStreams(bpy.types.UIList):
|
||||
"""
|
||||
Speckle stream list
|
||||
"""
|
||||
|
||||
def draw_item(self, context, layout, data, stream, active_data, active_propname):
|
||||
if self.layout_type in {"DEFAULT", "COMPACT"}:
|
||||
if stream:
|
||||
layout.label(
|
||||
text=f"{stream.name} ({stream.id})",
|
||||
translate=False,
|
||||
icon_value=0,
|
||||
)
|
||||
else:
|
||||
layout.label(text=" ", translate=False, icon_value=0)
|
||||
|
||||
elif self.layout_type in {"GRID"}:
|
||||
layout.alignment = "CENTER"
|
||||
layout.label(text="Streams", icon_value=0)
|
||||
|
||||
|
||||
class VIEW3D_PT_SpeckleUser(bpy.types.Panel):
|
||||
"""
|
||||
Speckle Users UI panel in the 3d viewport
|
||||
"""
|
||||
|
||||
bl_space_type = "VIEW_3D"
|
||||
bl_region_type = Region
|
||||
bl_category = "Speckle"
|
||||
bl_context = "objectmode"
|
||||
bl_label = "User"
|
||||
|
||||
def draw(self, context):
|
||||
speckle = context.scene.speckle
|
||||
|
||||
layout = self.layout
|
||||
col = layout.column()
|
||||
|
||||
if len(speckle.users) < 1:
|
||||
col.label(text="Refresh to initialise")
|
||||
else:
|
||||
col.prop(speckle, "active_user", text="")
|
||||
user = speckle.users[int(speckle.active_user)]
|
||||
col.label(text="{} ({})".format(user.server_name, user.server_url))
|
||||
col.label(text="{} ({})".format(user.name, user.email))
|
||||
|
||||
col.operator("speckle.users_load", text="", icon="FILE_REFRESH")
|
||||
|
||||
class VIEW3D_PT_SpeckleStreams(bpy.types.Panel):
|
||||
"""
|
||||
Speckle Streams UI panel in the 3d viewport
|
||||
"""
|
||||
|
||||
bl_space_type = "VIEW_3D"
|
||||
bl_region_type = Region
|
||||
bl_category = "Speckle"
|
||||
bl_context = "objectmode"
|
||||
bl_label = "Streams"
|
||||
|
||||
def draw(self, context):
|
||||
speckle = context.scene.speckle
|
||||
col = self.layout.column()
|
||||
|
||||
if len(speckle.users) < 1:
|
||||
col.label(text="No stream data.")
|
||||
else:
|
||||
user = speckle.users[int(speckle.active_user)]
|
||||
col.template_list(
|
||||
"VIEW3D_UL_SpeckleStreams", "", user, "streams", user, "active_stream"
|
||||
)
|
||||
row = col.row(align=True)
|
||||
row.operator("speckle.add_stream_from_url", text="", icon="URL")
|
||||
row.operator("speckle.create_stream", text="", icon="ADD")
|
||||
row.operator("speckle.delete_stream", text="", icon="REMOVE")
|
||||
row.operator("speckle.load_user_streams", text="", icon="FILE_REFRESH")
|
||||
|
||||
|
||||
class VIEW3D_PT_SpeckleActiveStream(bpy.types.Panel):
|
||||
"""
|
||||
Speckle Active Streams UI panel in the 3d viewport
|
||||
"""
|
||||
|
||||
bl_space_type = "VIEW_3D"
|
||||
bl_region_type = Region
|
||||
bl_category = "Speckle"
|
||||
bl_context = "objectmode"
|
||||
bl_label = "Active stream"
|
||||
|
||||
def draw(self, context):
|
||||
speckle = context.scene.speckle
|
||||
col = self.layout.column()
|
||||
|
||||
if len(speckle.users) < 1:
|
||||
col.label(text="No stream data.")
|
||||
else:
|
||||
user = speckle.users[int(speckle.active_user)]
|
||||
if len(user.streams) < 1:
|
||||
col.label(text="No active stream.")
|
||||
else:
|
||||
stream = user.streams[user.active_stream]
|
||||
# user.active_stream = min(user.active_stream, len(user.streams) - 1)
|
||||
row = col.row()
|
||||
row.label(text=f"{stream.name} ({stream.id})")
|
||||
row.operator("speckle.stream_copy_id", text="", icon="COPY_ID")
|
||||
col.separator()
|
||||
|
||||
row = col.row()
|
||||
row.prop(stream, "branch", text="")
|
||||
row.operator("speckle.branch_copy_name", text="", icon="COPY_ID")
|
||||
|
||||
if len(stream.branches) > 0:
|
||||
branch = stream.branches[int(stream.branch)]
|
||||
|
||||
row = col.row()
|
||||
row.prop(branch, "commit", text="")
|
||||
row.operator("speckle.commit_copy_id", text="", icon="COPY_ID")
|
||||
|
||||
if len(branch.commits) > 0:
|
||||
commit = branch.commits[int(branch.commit)]
|
||||
area = col.box()
|
||||
area.separator()
|
||||
|
||||
lines = wrap(32, commit.message)
|
||||
for line in lines:
|
||||
row = area.row(align=True)
|
||||
row.alignment = "EXPAND"
|
||||
row.scale_y = 0.4
|
||||
row.label(text=line)
|
||||
area.separator()
|
||||
|
||||
dt = datetime.strptime(
|
||||
commit.created_at, "%Y-%m-%d %H:%M:%S.%f%Z"
|
||||
)
|
||||
col.label(text=f"{dt.ctime()}")
|
||||
col.label(text=f"{commit.author_name} ({commit.author_id})")
|
||||
col.label(text=commit.source_application)
|
||||
else:
|
||||
col.label(text="No branches found!")
|
||||
|
||||
col.separator()
|
||||
|
||||
area = col.box()
|
||||
row = area.row()
|
||||
subcol = row.column()
|
||||
subcol.operator("speckle.receive_stream_objects", text="Receive")
|
||||
subcol.prop(speckle, "receive_script", text="")
|
||||
subcol = row.column()
|
||||
subcol.operator("speckle.send_stream_objects", text="Send")
|
||||
subcol.prop(speckle, "send_script", text="")
|
||||
area.prop(stream, "query", text="Filter")
|
||||
|
||||
col.separator()
|
||||
|
||||
row = col.row(align=True)
|
||||
subcol = row.column()
|
||||
|
||||
col.label(text="Description:")
|
||||
area = col.box()
|
||||
area.separator()
|
||||
|
||||
lines = wrap(32, stream.description)
|
||||
|
||||
for line in lines:
|
||||
row = area.row(align=True)
|
||||
row.alignment = "EXPAND"
|
||||
row.scale_y = 0.4
|
||||
row.label(text=line)
|
||||
|
||||
area.separator()
|
||||
col.separator()
|
||||
col.operator("speckle.view_stream_data_api", text="Open Stream in Web")
|
||||
|
||||
|
||||
class VIEW3D_PT_SpeckleHelp(bpy.types.Panel):
|
||||
"""
|
||||
Speckle Help UI panel in the 3d viewport
|
||||
"""
|
||||
|
||||
bl_space_type = "VIEW_3D"
|
||||
bl_region_type = Region
|
||||
bl_category = "Speckle"
|
||||
bl_context = "objectmode"
|
||||
bl_label = "Help"
|
||||
|
||||
def draw(self, context):
|
||||
layout = self.layout
|
||||
col = layout.column()
|
||||
|
||||
col.operator("speckle.open_speckle_guide")
|
||||
col.separator()
|
||||
col.operator("speckle.open_speckle_tutorials")
|
||||
col.separator()
|
||||
col.operator("speckle.open_speckle_forum")
|
||||
@@ -1,54 +0,0 @@
|
||||
def find_key_case_insensitive(data, key, default=None):
|
||||
value = data.get(key)
|
||||
if value:
|
||||
return value
|
||||
|
||||
"""
|
||||
Necessary to find keys where the first character
|
||||
is capitalized
|
||||
"""
|
||||
value = data.get(key[0].upper() + key[1:])
|
||||
if value:
|
||||
return value
|
||||
|
||||
value = data.get(key.upper())
|
||||
if value:
|
||||
return value
|
||||
|
||||
return default
|
||||
|
||||
|
||||
def get_iddata(base, uuid, name, obdata):
|
||||
"""
|
||||
This is taken from the import_3dm add-on:
|
||||
https://github.com/jesterKing/import_3dm
|
||||
# Copyright (c) 2018-2019 Nathan Letwory, Joel Putnam,
|
||||
Tom Svilans
|
||||
|
||||
Get an iddata. If an object with given uuid is found in
|
||||
this .blend use that. Otherwise new up one with base.new,
|
||||
potentially with obdata if that is set
|
||||
"""
|
||||
founditem = None
|
||||
if uuid is not None:
|
||||
for item in base:
|
||||
if item.get("speckle_id", None) == str(uuid):
|
||||
founditem = item
|
||||
break
|
||||
elif name:
|
||||
for item in base:
|
||||
if item.get("name", None) == name:
|
||||
founditem = item
|
||||
break
|
||||
if founditem:
|
||||
theitem = founditem
|
||||
theitem["name"] = name
|
||||
if obdata:
|
||||
theitem.data = obdata
|
||||
else:
|
||||
if obdata:
|
||||
theitem = base.new(name=name, object_data=obdata)
|
||||
else:
|
||||
theitem = base.new(name=name)
|
||||
tag_data(theitem, uuid, name)
|
||||
return theitem
|
||||
@@ -0,0 +1,4 @@
|
||||
#!/usr/bin/env pwsh
|
||||
$ErrorActionPreference = "Stop"
|
||||
|
||||
uv pip compile pyproject.toml --output-file bpy_speckle/requirements.txt --generate-hashes
|
||||
@@ -1,4 +1,4 @@
|
||||
#!/usr/bin/env bash
|
||||
set -e -o pipefail
|
||||
|
||||
poetry export --only main -o bpy_speckle/requirements.txt
|
||||
uv pip compile pyproject.toml --output-file bpy_speckle/requirements.txt --generate-hashes
|
||||
|
||||
@@ -1,20 +0,0 @@
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
def patch_installer(tag: str):
|
||||
"""Patches the installer with the correct connector version and specklepy version"""
|
||||
tag = tag.replace("\n", "")
|
||||
iss_file = "speckle-sharp-ci-tools/blender.iss"
|
||||
iss_path = Path(iss_file)
|
||||
lines = iss_path.read_text().split("\n")
|
||||
lines.insert(12, f'#define AppVersion "{tag.split("-")[0]}"')
|
||||
lines.insert(13, f'#define AppInfoVersion "{tag}"')
|
||||
|
||||
iss_path.write_text("\n".join(lines))
|
||||
print(f"Patched installer with connector v{tag}")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
tag = sys.argv[1]
|
||||
patch_installer(tag)
|
||||
+32
-10
@@ -1,30 +1,52 @@
|
||||
import re
|
||||
import sys
|
||||
|
||||
def patch_connector(tag):
|
||||
"""Patches the connector version within the connector init file"""
|
||||
bpy_file = "bpy_speckle/__init__.py"
|
||||
tag = tag.split(".")
|
||||
|
||||
with open(bpy_file, "r") as file:
|
||||
def patch_addon(simple_version: str):
|
||||
"""Patches the __init__.py bl_info version within the connector init file"""
|
||||
FILE_PATH = "bpy_speckle/__init__.py"
|
||||
version = simple_version.split(".")
|
||||
|
||||
with open(FILE_PATH, "r") as file:
|
||||
lines = file.readlines()
|
||||
|
||||
for (index, line) in enumerate(lines):
|
||||
for index, line in enumerate(lines):
|
||||
if '"version":' in line:
|
||||
lines[index] = f' "version": ({tag[0]}, {tag[1]}, {tag[2]}),\n'
|
||||
print(f"Patched connector version number in {bpy_file}")
|
||||
lines[index] = (
|
||||
f' "version": ({version[0]}, {version[1]}, {version[2]}),\n'
|
||||
)
|
||||
|
||||
with open(FILE_PATH, "w") as file:
|
||||
file.writelines(lines)
|
||||
|
||||
|
||||
def patch_manifest(simple_version: str):
|
||||
"""Patches the connector version within the connector init file"""
|
||||
FILE_PATH = "bpy_speckle/blender_manifest.toml"
|
||||
version = simple_version.split(".")
|
||||
|
||||
with open(FILE_PATH, "r") as file:
|
||||
lines = file.readlines()
|
||||
|
||||
for index, line in enumerate(lines):
|
||||
if line.startswith("version ="):
|
||||
lines[index] = f'version = "{version[0]}.{version[1]}.{version[2]}"\n'
|
||||
print(f"Patched connector version number in {FILE_PATH}")
|
||||
break
|
||||
|
||||
with open(bpy_file, "w") as file:
|
||||
with open(FILE_PATH, "w") as file:
|
||||
file.writelines(lines)
|
||||
|
||||
|
||||
def main():
|
||||
tag = sys.argv[1]
|
||||
if not re.match(r"([0-9]+)\.([0-9]+)\.([0-9]+)", tag):
|
||||
raise ValueError(f"Invalid tag provided: {tag}")
|
||||
|
||||
print(f"Patching version: {tag}")
|
||||
patch_connector(tag.split("-")[0])
|
||||
simple_version = tag.split("-")[0]
|
||||
patch_addon(simple_version)
|
||||
patch_manifest(simple_version)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
Generated
-1105
File diff suppressed because it is too large
Load Diff
+13
-21
@@ -1,24 +1,16 @@
|
||||
[tool.poetry]
|
||||
[project]
|
||||
name = "speckle-blender"
|
||||
version = "2.0.0"
|
||||
description = "the Speckle 2.0 connector for Blender!"
|
||||
authors = ["izzy lyseggen <izzy.lyseggen@gmail.com>", "Gergő Jedlicska <gergo@jedlicska.com>"]
|
||||
version = "3.0.0"
|
||||
description = "Next-Gen Speckle connector for Blender!"
|
||||
requires-python = ">=3.11.9, <4.0.0"
|
||||
license = "Apache-2.0"
|
||||
dependencies = [
|
||||
"specklepy>=3.2.4",
|
||||
]
|
||||
|
||||
[tool.poetry.dependencies]
|
||||
python = ">=3.8, <4.0.0"
|
||||
specklepy = "^2.13.0"
|
||||
|
||||
# [tool.poetry.group.local_specklepy.dependencies]
|
||||
# specklepy = {path = "../specklepy", develop = true}
|
||||
|
||||
[tool.poetry.group.dev.dependencies]
|
||||
numpy = "^1.23.5"
|
||||
fake-bpy-module-latest = "^20230117"
|
||||
black = "^22.10.0"
|
||||
pylint = "^2.15.7"
|
||||
ruff = "^0.0.166"
|
||||
|
||||
[build-system]
|
||||
requires = ["poetry-core>=1.0.0"]
|
||||
build-backend = "poetry.core.masonry.api"
|
||||
[dependency-groups]
|
||||
dev = [
|
||||
"fake-bpy-module-latest>=20260126",
|
||||
"ruff==0.14.14",
|
||||
"pre-commit>=4.0.1",
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user