def _minimal_ext_cmd(cmd):

# construct minimal environment

env = {}

for k in ['SYSTEMROOT', 'PATH', 'HOME']:

v = os.environ.get(k)

if v is not None:

env[k] = v

# LANGUAGE is used on win32

env['LANGUAGE'] = 'C'

env['LANG'] = 'C'

env['LC_ALL'] = 'C'

out = subprocess.Popen(

cmd, stdout=subprocess.PIPE, env=env).communicate()[0]

return out

try:

out = _minimal_ext_cmd(['git', 'rev-parse', 'HEAD'])

sha = out.strip().decode('ascii')

except OSError:

sha = 'unknown'

if os.path.exists('.git'):

sha = get_git_hash()[:7]

elif os.path.exists(version_file):

try:

from mmagic.version import __version__

sha = __version__.split('+')[-1]

except ImportError:

raise ImportError('Unable to get git version')

else:

sha = 'unknown'

"""Parse the package dependencies listed in a requirements file but strips

import re

import sys

from os.path import exists

require_fpath = fname

def parse_line(line):

"""Parse information from a line in a requirements text file."""

if line.startswith('-r '):

# Allow specifying requirements in other files

target = line.split(' ')[1]

for info in parse_require_file(target):

yield info

else:

info = {'line': line}

if line.startswith('-e '):

info['package'] = line.split('#egg=')[1]

elif '@git+' in line:

info['package'] = line

else:

# Remove versioning from the package

pat = '(' + '|'.join(['>=', '<=', '==', '>', '<']) + ')'

parts = re.split(pat, line, maxsplit=1)

parts = [p.strip() for p in parts]

info['package'] = parts[0]

if len(parts) > 1:

op, rest = parts[1:]

if ';' in rest:

# Handle platform specific dependencies

# http://setuptools.readthedocs.io/en/latest/setuptools.html#declaring-platform-specific-dependencies

version, platform_deps = map(str.strip,

rest.split(';'))

info['platform_deps'] = platform_deps

else:

version = rest # NOQA

info['version'] = (op, version)

yield info

def parse_require_file(fpath):

with open(fpath, 'r') as f:

for line in f.readlines():

line = line.strip()

if line and not line.startswith('#'):

for info in parse_line(line):

yield info

def gen_packages_items():

if exists(require_fpath):

for info in parse_require_file(require_fpath):

parts = [info['package']]

if with_version and 'version' in info:

parts.extend(info['version'])

if not sys.version.startswith('3.4'):

# apparently package_deps are broken in 3.4

platform_deps = info.get('platform_deps')

if platform_deps is not None:

parts.append(';' + platform_deps)

item = ''.join(parts)

yield item

packages = list(gen_packages_items())

define_macros = []

extra_compile_args = {'cxx': []}

if torch.cuda.is_available() or os.getenv('FORCE_CUDA', '0') == '1':

define_macros += [('WITH_CUDA', None)]

extension = CUDAExtension

extra_compile_args['nvcc'] = [

'-D__CUDA_NO_HALF_OPERATORS__',

'-D__CUDA_NO_HALF_CONVERSIONS__',

'-D__CUDA_NO_HALF2_OPERATORS__',

]

sources += sources_cuda

else:

print(f'Compiling {name} without CUDA')

extension = CppExtension

return extension(

name=f'{module}.{name}',

sources=[os.path.join(*module.split('.'), p) for p in sources],

define_macros=define_macros,

"""Add extra files that are required to support MIM into the package.

# parse installment mode

if 'develop' in sys.argv:

# installed by `pip install -e .`

mode = 'symlink'

elif 'sdist' in sys.argv or 'bdist_wheel' in sys.argv:

# installed by `pip install .`

# or create source distribution by `python setup.py sdist`

mode = 'copy'

else:

return

filenames = ['tools', 'configs', 'demo', 'model-index.yml']

repo_path = osp.dirname(__file__)

mim_path = osp.join(repo_path, 'mmagic', '.mim')

os.makedirs(mim_path, exist_ok=True)

for filename in filenames:

if osp.exists(filename):

src_path = osp.join(repo_path, filename)

tar_path = osp.join(mim_path, filename)

if osp.isfile(tar_path) or osp.islink(tar_path):

os.remove(tar_path)

elif osp.isdir(tar_path):

shutil.rmtree(tar_path)

if mode == 'symlink':

src_relpath = osp.relpath(src_path, osp.dirname(tar_path))

try:

os.symlink(src_relpath, tar_path)

except OSError:

# Creating a symbolic link on windows may raise an

# `OSError: [WinError 1314]` due to privilege. If

# the error happens, the src file will be copied

mode = 'copy'

warnings.warn(

f'Failed to create a symbolic link for {src_relpath}, '

f'and it will be copied to {tar_path}')

else:

continue

if mode == 'copy':

if osp.isfile(src_path):

shutil.copyfile(src_path, tar_path)

elif osp.isdir(src_path):

shutil.copytree(src_path, tar_path)

else:

warnings.warn(f'Cannot copy file {src_path}.')

else: