setup.py - lecrunch - LeCrunch is a set of libraries and scripts for communicating and extracting waveform traces from LeCroy X-stream oscilloscopes.

blob: 0c153ca4939fa5e7a1ba1499e8fcc5dbd46cd7a4 (plain)

1	`scope_ip = ''`

/* Name.Variable.Global */ .highlight .vi { color: #3333bb } /* Name.Variable.Instance */ .highlight .vm { color: #336699 } /* Name.Variable.Magic */ .highlight .il { color: #0000DD; font-weight: bold } /* Literal.Number.Integer.Long */

import numpy as np
from os.path import realpath, dirname
from chroma.tools import read_csv

color_map = read_csv(dirname(realpath(__file__)) + '/ciexyz64_1.csv')

def map_wavelength(wavelength):
    r = np.interp(wavelength, color_map[:,0], color_map[:,1])
    g = np.interp(wavelength, color_map[:,0], color_map[:,2])
    b = np.interp(wavelength, color_map[:,0], color_map[:,3])

    if np.iterable(wavelength):
        rgb = np.empty((len(wavelength),3))

        rgb[:,0] = r
        rgb[:,1] = g
        rgb[:,2] = b

        return rgb
    else:
        return np.array([r,g,b])