NLeSC / PattyAnalytics

scripts/transform.py

#!/usr/bin/env python2.7
"""Apply transformation to pointcloud.

The transformations (if given) are applied in the following order:
1. rotation
2. scaling
3. offset

Usage:
  transform.py [-o <origin>] [-r <rot>] [-t <translate>] [-s <scaling>] <source> <target>

This line is too long as per the coding-style (89/79).

Positional arguments:
  source            Source pointcloud file
  target            Target pointcloud file

Options:
  -r <rot>          CSV file of a 4x4 transformation matrix, assumed to be
                    normalized.
  -o <rot_origin>   CSV file with a 3d point of the origin for the
                    rotation and scaling
  -s <scaling>      CSV file with a single scaling multiplication factor.
  -t <translate>    CSV file with a 3d vector to translate the pointcloud with.
"""

from __future__ import print_function
from docopt import docopt

import numpy as np
from patty.utils import load, save


def csv_read(path):
    return np.genfromtxt(path, dtype=float, delimiter=',')


if __name__ == '__main__':
    args = docopt(__doc__)

The name args does not conform to the constant naming conventions ((([A-Z_][A-Z0-9_]*)|(__.*__))$).

    pc = load(args['<source>'])

The name pc does not conform to the constant naming conventions ((([A-Z_][A-Z0-9_]*)|(__.*__))$).

    try:
        offset = csv_read(args['-o'])

The name offset does not conform to the constant naming conventions ((([A-Z_][A-Z0-9_]*)|(__.*__))$).

    except:
        offset = None

The name offset does not conform to the constant naming conventions ((([A-Z_][A-Z0-9_]*)|(__.*__))$).

    try:
        matrix = csv_read(args['-r'])

The name matrix does not conform to the constant naming conventions ((([A-Z_][A-Z0-9_]*)|(__.*__))$).

        pc.rotate(matrix, origin=offset)
    except Exception as e:

The name e does not conform to the variable naming conventions ([a-z_][a-z0-9_]{1,30}$).

        print('Problem with rotate: ', e)

    try:
        factor = csv_read(args['-s'])

The name factor does not conform to the constant naming conventions ((([A-Z_][A-Z0-9_]*)|(__.*__))$).

        pc.scale(factor, origin=offset)
    except Exception as e:

The name e does not conform to the variable naming conventions ([a-z_][a-z0-9_]{1,30}$).

        print('Problem with scale: ', e)

    try:
        vector = csv_read(args['-t'])

The name vector does not conform to the constant naming conventions ((([A-Z_][A-Z0-9_]*)|(__.*__))$).

        pc.translate(vector)
    except Exception as e:

The name e does not conform to the variable naming conventions ([a-z_][a-z0-9_]{1,30}$).

        print('Problem with translate: ', e)

    save(pc, args['<target>'])