mvn_build.py

#!/usr/bin/python
# -*- coding: iso-8859-15 -*-
#
# Lima compilation script
#
# Version 1.0
#
#
# Author: FL / AN (Hacked from S. Poirier)

from multiprocessing import Pool
from argparse import ArgumentParser

import os
import sys
import shutil
import string
import time


copied_files = []

#------------------------------------------------------------------------------
# build exception
#------------------------------------------------------------------------------
class BuildError(Exception):
  pass

#------------------------------------------------------------------------------
# Project directory
#------------------------------------------------------------------------------
def set_project_dir(sub_dir):
  project_dir = os.path.join(current_dir, sub_dir)
  os.chdir(project_dir)

#------------------------------------------------------------------------------
# Copy library
#------------------------------------------------------------------------------

def copy_file_ext(from_path, to_path, file_ext):

  if not os.path.isdir(to_path):
    os.makedirs(to_path)

  files = os.listdir(from_path)
  for file in files:
    full_name = os.path.join(from_path, file)
    if os.path.isfile(full_name):
      root, ext = os.path.splitext(full_name)
      if ext == file_ext:
        shutil.copy(full_name, to_path)
        file_to_copy = file + ' copied in ' + to_path
        copied_files.append(file_to_copy)

#------------------------------------------------------------------------------
# Compilation
#------------------------------------------------------------------------------
def build():
  rc = os.system(maven_clean_install + maven_options)
  if rc != 0:
    raise BuildError

#------------------------------------------------------------------------------
# Clean
#------------------------------------------------------------------------------
def clean():
  rc = os.system(maven_clean)
  if rc != 0:
    raise BuildError

#------------------------------------------------------------------------------
# Clean all modules
#------------------------------------------------------------------------------
def clean_all():
    print 'Cleaning all\n'
    set_project_dir('.');clean()
    set_project_dir('third-party/Processlib');clean()
    for cam in camera_list:
        set_project_dir('camera/'+ cam);clean()
    set_project_dir('applications/tango/LimaDetector');clean()

#------------------------------------------------------------------------------
# build the LimaDetector device
#------------------------------------------------------------------------------
def build_device(target_path):
  print 'Build Device LimaDetector\n'
  set_project_dir('applications/tango/LimaDetector')

  print "Maven command = ", maven_clean_install + maven_platform_options + maven_options
  rc = os.system(maven_clean_install + maven_platform_options + maven_options)
  if rc != 0:
    raise BuildError

  if "linux" in sys.platform:
    if target_path is not None:
      dest_path = os.path.join(target_path, '')
      copy_file_ext(device_src_path, dest_path, '')
  if "win32" in sys.platform:
    if target_path is not None:
      dest_path = os.path.join(target_path, '')
      copy_file_ext(device_src_path, dest_path, '.exe')
      copy_file_ext(device_src_path, dest_path, '.pdb')

  print '\n'

#------------------------------------------------------------------------------
# build the Lima Core
#------------------------------------------------------------------------------
def build_lima_core(target_path):
  print 'Build Lima\n'
  set_project_dir('.')
  build()

  if "linux" in sys.platform:
    if target_path is not None:
      dest_path = os.path.join(target_path, '')
      copy_file_ext(src_path, dest_path, '.so')
  if "win32" in sys.platform:
    if target_path is not None:
      dest_path = os.path.join(target_path, '')
      copy_file_ext(src_path, dest_path, '.dll')
      copy_file_ext(src_path, dest_path, '.pdb')
  print '\n'

#------------------------------------------------------------------------------
# build the Plugin 'plugin'
#------------------------------------------------------------------------------
def build_plugin(plugin,target_path):

  """Build the selected plugin"""

  print "Building:    " , plugin, "\n"
  set_project_dir(plugin)
  build()

  if "linux" in sys.platform:
    if target_path is not None:
      dest_path = os.path.join(target_path, '')
      copy_file_ext(src_path, dest_path, '.so')
  if "win32" in sys.platform:
    if target_path is not None:
      dest_path = os.path.join(target_path, '')
      copy_file_ext(src_path, dest_path, '.dll')
      copy_file_ext(src_path, dest_path, '.pdb')

  print '\n'


#------------------------------------------------------------------------------
# build all linux cameras
#------------------------------------------------------------------------------
def build_all_camera(target_path):
    start=time.time()
    # this take 2 min 45 sec on a quad core
    if multi_proc == False:
        for cam in camera_list:
            build_plugin('camera/' + cam, target_path)
    else:
        # this take 1 min 26 sec on a quad core
        pool = Pool()
        for cam in camera_list:
            print "multi proc: Building:    " , cam, "\n"
            pool.apply_async(build_plugin,('camera/' + cam, target_path))
        pool.close()
        pool.join()
        time.strftime("%M min %S sec", time.localtime(170))
    print "duration for compiling all cameras (MultiProc = ",multi_proc, ") = ", time.strftime("%M min %S sec", time.localtime(time.time() - start))

#------------------------------------------------------------------------------
# Main Entry point
#------------------------------------------------------------------------------
if __name__ == "__main__":

  if "linux" in sys.platform:
    platform = "linux"
    camera_list = ["adsc", "aviex", "basler", "eiger", "marccd","pilatus","prosilica","simulator","xpad"]
    maven_platform_options = " --file pom-linux.xml"
    src_path = './target/nar/lib/i386-Linux-g++/shared/'
    device_src_path = './target/nar/bin/i386-Linux-g++/'
  if "win32" in sys.platform:
    platform = "win32"
    camera_list = ["andor", "hamamatsu", "pco","perkinelmer","roperscientific","simulator","uview"]
    maven_platform_options = " --file pom-win.xml"
    src_path = './target/nar/lib/x86-Windows-msvc/shared/'
    device_src_path = './target/nar/bin/x86-Windows-msvc/'

  print "platform : ", platform

  # command line parsing
  parser = ArgumentParser(description="Lima compilation script")
  cams_string = ""
  for cam in camera_list:
    cams_string += cam + "|"
  help_string = "module to compile (possible values are: all|processlib|lima|cameras|"+ cams_string+ "|device||cleanall)"
  parser.add_argument("module", help=help_string)
  parser.add_argument("-o","--offline", help="mvn will be offline",action="store_true")
  parser.add_argument("-f","--pomfile", help="name of the pom file(for the Tango device only)")
  parser.add_argument("-q","--quiet", help="mvn will be quiet", action="store_true")
  parser.add_argument("-m","--multiproc", help="cameras will be compiled in multiprocessing way",action="store_true")
  parser.add_argument("-d","--directory", help="automatically install Lima binaries into the specified installation directory")

  args = parser.parse_args()

  maven_options = ""
  # manage command line option  
  if args.offline:
    maven_options += " -o"

  if args.pomfile:
    maven_platform_options = " --file " + args.pomfile

  if args.quiet:
    maven_options += " -q"

  if args.multiproc:
    multi_proc = True
  else:
    multi_proc = False

  if args.directory:
    target_path = args.directory
  else:
    target_path = None

  # variables
  maven_clean_install   = "mvn clean install -DenableCheckRelease=false"
  maven_clean           = "mvn clean"
  current_dir           = os.getcwd()

  try:
    # Build all
    if args.module == 'all':
        print 'BUILD ALL\n'
        build_plugin('third-party/Processlib', target_path)
        build_lima_core(target_path)
        build_all_camera(target_path)
        build_device(target_path)
    # Build processlib
    elif args.module == 'processlib':
        print 'BUILD ProcessLib\n'
        build_plugin('third-party/Processlib', target_path)
    # Build device
    elif args.module == 'device':
        print 'BUILD Device\n'
        build_device(target_path)
    # Build lima
    elif args.module == 'lima':
        print 'BUILD Lima Core\n'
        build_lima_core(target_path)
    # Build cameras
    elif args.module == 'cameras':
        print 'BUILD All ',platform,' Cameras\n'
        build_all_camera(target_path)
    # Clean all
    elif args.module =='cleanall':
        clean_all()
    # Build cam
    else:
        for cam in camera_list:
            if args.module == cam:
                build_plugin('camera/'+cam, target_path)
                break

    # display list of copied files, if -d option is used
    if args.directory:
        print '\n'
        print '============================================='
        print 'Modules are compiled & copied as shown below:'
        print '============================================='
        for file in copied_files:
            print '- ',file
        print '\n'

  except BuildError, e:
    sys.stderr.write("!!!   BUILD FAILED    !!!\n")