mirror of
https://github.com/alicevision/Meshroom.git
synced 2025-04-28 17:57:16 +02:00
123 lines
4 KiB
Python
123 lines
4 KiB
Python
__version__ = "2021.1.0"
|
|
__version_name__ = __version__
|
|
|
|
from distutils import util
|
|
from enum import Enum
|
|
import logging
|
|
import os
|
|
import sys
|
|
|
|
# sys.frozen is initialized by cx_Freeze and identifies a release package
|
|
isFrozen = getattr(sys, "frozen", False)
|
|
if not isFrozen:
|
|
# development mode: add git branch name (if any) to __version_name__
|
|
scriptPath = os.path.dirname(os.path.abspath(__file__))
|
|
headFilepath = os.path.join(scriptPath, "../.git/HEAD")
|
|
if os.path.exists(headFilepath):
|
|
with open(headFilepath, "r") as headFile:
|
|
data = headFile.readlines()
|
|
branchName = data[0].split('/')[-1].strip()
|
|
__version_name__ += "-" + branchName
|
|
|
|
# Allow override from env variable
|
|
__version_name__ = os.environ.get("REZ_MESHROOM_VERSION", __version_name__)
|
|
|
|
useMultiChunks = util.strtobool(os.environ.get("MESHROOM_USE_MULTI_CHUNKS", "True"))
|
|
|
|
|
|
class Backend(Enum):
|
|
STANDALONE = 1
|
|
PYSIDE = 2
|
|
|
|
|
|
backend = Backend.STANDALONE
|
|
|
|
|
|
def useUI():
|
|
global backend
|
|
backend = Backend.PYSIDE
|
|
|
|
|
|
def setupEnvironment():
|
|
"""
|
|
Setup environment for Meshroom to work in a prebuilt, standalone configuration.
|
|
|
|
Use 'MESHROOM_INSTALL_DIR' to simulate standalone configuration with a path to a Meshroom installation folder.
|
|
|
|
# Meshroom standalone structure
|
|
|
|
- Meshroom/
|
|
- aliceVision/
|
|
- bin/ # runtime bundled binaries (windows: exe + libs, unix: executables)
|
|
- lib/ # runtime bundled libraries (unix: libs)
|
|
- share/ # resource files
|
|
- aliceVision/
|
|
- COPYING.md # AliceVision COPYING file
|
|
- cameraSensors.db # sensor database
|
|
- vlfeat_K80L3.tree # voctree file
|
|
- lib/ # Python lib folder
|
|
- qtPlugins/
|
|
Meshroom # main executable
|
|
COPYING.md # Meshroom COPYING file
|
|
"""
|
|
|
|
def addToEnvPath(var, val, index=-1):
|
|
"""
|
|
Add paths to the given environment variable.
|
|
|
|
Args:
|
|
var (str): the name of the variable to add paths to
|
|
val (str or list of str): the path(s) to add
|
|
index (int): insertion index
|
|
"""
|
|
if not val:
|
|
return
|
|
|
|
paths = os.environ.get(var, "").split(os.pathsep)
|
|
|
|
if not isinstance(val, (list, tuple)):
|
|
val = [val]
|
|
|
|
if index == -1:
|
|
paths.extend(val)
|
|
elif index == 0:
|
|
paths = val + paths
|
|
else:
|
|
raise ValueError("addToEnvPath: index must be -1 or 0.")
|
|
os.environ[var] = os.pathsep.join(paths)
|
|
|
|
# setup root directory (override possible by setting "MESHROOM_INSTALL_DIR" environment variable)
|
|
rootDir = os.path.dirname(sys.executable) if isFrozen else os.environ.get("MESHROOM_INSTALL_DIR", None)
|
|
|
|
if rootDir:
|
|
os.environ["MESHROOM_INSTALL_DIR"] = rootDir
|
|
|
|
aliceVisionDir = os.path.join(rootDir, "aliceVision")
|
|
# default directories
|
|
aliceVisionBinDir = os.path.join(aliceVisionDir, "bin")
|
|
aliceVisionShareDir = os.path.join(aliceVisionDir, "share", "aliceVision")
|
|
qtPluginsDir = os.path.join(rootDir, "qtPlugins")
|
|
sensorDBPath = os.path.join(aliceVisionShareDir, "cameraSensors.db")
|
|
voctreePath = os.path.join(aliceVisionShareDir, "vlfeat_K80L3.SIFT.tree")
|
|
|
|
env = {
|
|
'PATH': aliceVisionBinDir,
|
|
'QT_PLUGIN_PATH': [qtPluginsDir],
|
|
'QML2_IMPORT_PATH': [os.path.join(qtPluginsDir, "qml")]
|
|
}
|
|
|
|
for key, value in env.items():
|
|
logging.info("Add to {}: {}".format(key, value))
|
|
addToEnvPath(key, value, 0)
|
|
|
|
variables = {
|
|
"ALICEVISION_SENSOR_DB": sensorDBPath,
|
|
"ALICEVISION_VOCTREE": voctreePath
|
|
}
|
|
|
|
for key, value in variables.items():
|
|
if key not in os.environ and os.path.exists(value):
|
|
logging.info("Set {}: {}".format(key, value))
|
|
os.environ[key] = value
|
|
else:
|
|
addToEnvPath("PATH", os.environ.get("ALICEVISION_BIN_PATH", ""))
|