optimization: creation of images

[pmachines.git] / setup.py

'''Setuptools' configuration file
e.g. python setup.py models --cores=1
e.g. python setup.py bdist_apps --nowin=1'''


from os import system, getcwd, chdir
from sys import argv, executable
from collections import namedtuple
from subprocess import Popen
from distutils.cmd import Command
from setuptools import setup
from setuptools.command.develop import develop
from multiprocessing import cpu_count
from direct.dist.commands import bdist_apps
from lib.build.build import branch, files, ver, files, bld_dpath
from lib.build.docs import bld_docs
from lib.build.models import ModelsBuilder
from lib.build.images import bld_images
from lib.build.screenshots import bld_screenshots
from lib.build.lang import LanguageBuilder
from p3d_appimage import AppImageBuilder
from p3d_flatpak import FlatpakBuilder
import lib.engine.log  # so logging's info/debug are logged


appname = longname = 'pmachines'



msg = '''NOTE: please be sure that you've already created the assets with:
    * python setup.py images models lang'''


class DevelopPyCmd(develop):
    '''Command for setting up the development.'''

    def run(self):
        '''Prepare the development environment.'''
        develop.run(self)
        Popen([executable, __file__, 'lang']).communicate()
        Popen([executable, __file__, 'models']).communicate()


class AbsCmd(Command):
    '''Common functionalities for commands.'''

    user_options = [('cores=', None, '#cores')]
    cores = cpu_count()

    def initialize_options(self):
        for arg in argv[:]:
            if arg.startswith('--cores='):
                AbsCmd.cores = int(arg.split('=')[1])

    def finalize_options(self):  # must override
        pass


class DocsCmd(AbsCmd):
    '''Command for building the docs.'''

    def run(self):
        '''Builds the docs.'''
        bld_docs()


class ModelsCmd(AbsCmd):
    '''Command for building the models.'''

    def run(self):
        '''Builds the models.'''
        ModelsBuilder().build('assets/models', int(AbsCmd.cores))


class ImagesCmd(AbsCmd):
    '''Command for building the models.'''

    def run(self):
        '''Builds the images.'''
        bld_screenshots()
        bld_images(
            files(['jpg', 'png'], ['models', 'gltf', 'bam'], ['_png.png']), int(AbsCmd.cores))


class LangCmd(AbsCmd):
    '''Command for building po, pot and mo files.'''

    lang_path = 'assets/locale/'

    def _process_lang(self, lang_code):
        '''Processes a single language.'''
        poname = 'assets/po/%s.po' % lang_code
        LanguageBuilder.merge(lang_code, 'assets/po/', self.lang_path, appname)
        mo_tmpl = '%s%s/LC_MESSAGES/%s.mo'
        moname = mo_tmpl % (self.lang_path, lang_code, appname)
        LanguageBuilder.mo(moname, self.lang_path, appname)

    def run(self):
        '''Builds the language files.'''
        LanguageBuilder.pot(appname, 'assets/po/')
        list(map(self._process_lang, ['it_IT']))


class BDistAppsCmd(bdist_apps):
    '''Customization of BDistApps.'''

    user_options = bdist_apps.user_options + [
        ('cores', None, '#cores'),
        ('nowin=', None, "don't build for windows"),
        ('nolinux=', None, "don't build for linux")]

    def initialize_options(self):
        '''Default values.'''
        bdist_apps.initialize_options(self)
        self.nowin, self.nolinux = None, None

    #def finalize_options(self):  # must override
    #    bdist_apps.finalize_options(self)

    def run(self):
        '''Our bdist_apps' customization.'''
        print(msg)
        cmd = 'patch --forward ' + \
              '../venv/lib/python3.7/site-packages/direct/dist/commands.py' + \
              ' lib/build/commands.patch'
        system(cmd)
        bdist_apps.run(self)
        if not self.nolinux:
            hbranch = {'master': 'alpha', 'rc': 'rc', 'stable': ''}[branch]
            AppImageBuilder(self).build(longname, hbranch,
                                        'http://www.ya2tech.it/downloads/')
            fbranch = {'master': 'alpha', 'rc': 'rc', 'stable': 'stable'}[branch]
            bld = FlatpakBuilder(
                self,
                'it.ya2.Pmachines',
                '/home/flavio/pmachines_builder/flatpak',
                'D43B6D401912B520B6805FCC8E019E6340E3BAB5',
                '/home/flavio/pmachines_builder/gpg',
                'http://www.ya2tech.it/flatpak',
                ['options*.ini'],
                fbranch,
                ['assets'])
            bld.build()


if __name__ == '__main__':
    platform_lst, installers_dct = [], {}
    if all('--nowin' not in arg for arg in argv):
        platform_lst += ['win_amd64']
        installers_dct['win_amd64'] = ['nsis']
    if all('--nolinux' not in arg for arg in argv):
        platform_lst += ['manylinux1_x86_64']
        installers_dct['manylinux1_x86_64'] = []
    log_path = '$USER_APPDATA/pmachines/logs/%Y/%B/%d/%H_%M_%S.log'
    package_data_dirs = {'simplepbr': [('simplepbr/shaders*', '', {})]}
    setup(
        name=appname,
        version=ver,
        cmdclass={
            'develop': DevelopPyCmd,
            'docs': DocsCmd,
            'models': ModelsCmd,
            'images': ImagesCmd,
            'lang': LangCmd,
            'bdist_apps': BDistAppsCmd},
        install_requires=['panda3d'],
        options={
            'build_apps': {
                'exclude_patterns': [
                    'build/*', 'built/*', 'setup.py', 'requirements.txt',
                    'venv/*', '.git*', '*.pyc', 'options.ini', '__pycache__',
                    'assets/gltf/*', 'assets/models/**/*.blend',
                    'assets/models/**/models/*.png',
                    'assets/models/**/models/*.jpg'],
                'log_filename_strftime': True,
                'log_filename': log_path,
                'plugins': ['pandagl', 'p3openal_audio', 'pandadx9'],
                'gui_apps': {appname: 'main.py'},
                'package_data_dirs': package_data_dirs,
                'icons': {
                    appname: [
                        'assets/images/icon/icon256_png.png',
                        'assets/images/icon/icon128_png.png',
                        'assets/images/icon/icon48_png.png',
                        'assets/images/icon/icon32_png.png',
                        'assets/images/icon/icon16_png.png']},
                'icons': {
                    appname: [
                        'assets/icon/icon256.png',
                        'assets/icon/icon128.png',
                        'assets/icon/icon48.png',
                        'assets/icon/icon32.png',
                        'assets/icon/icon16.png']},
                'include_patterns': [
                    '**/lib/licenses/*',
                    '**/licenses/*',
                    '**/*.bam',
                    '**/*.dds',
                    '**/*.bin',
                    #'**/*.png',
                    #'**/*.jpg',
                    '**/*.ico',
                    '**/*.json',
                    '**/track_tr.py',
                    '**/*.txt',
                    '**/*.ttf',
                    '**/*.vert',
                    '**/*.frag',
                    '**/*.ogg',
                    '**/*.wav',
                    '**/*.mo'],
                'platforms': platform_lst,
                'include_modules': {'*': ['encodings.hex_codec']}},
            'bdist_apps': {'installers': installers_dct}})