@bluemath/linalg/src/lapack/common.ts

Bluemath Linear Algebra library

/*

Copyright (C) 2017 Jayesh Salvi, Blue Math Software Inc.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.

*/

import * as lapacklite from '../../ext/lapacklite'
let em = lapacklite.Module;

import {TypedArray} from '@bluemath/common'

/**
 * @hidden
 */
export const SIZE_CHAR = 1;
/**
 * @hidden
 */
export const SIZE_INT = 4;
/**
 * @hidden
 */
export const SIZE_DOUBLE = 8;
/**
 * @hidden
 */
export const SIZE_SINGLE = 4;

/**
 * @hidden
 */
export const spotrf_wrap = em.cwrap('spotrf_',null,
  ['number','number','number','number','number']);

/**
 * @hidden
 */
export const dpotrf_wrap = em.cwrap('dpotrf_',null,
  ['number','number','number','number','number']);

/**
 * @hidden
 */
export const sgesv_wrap = em.cwrap('sgesv_',
  null,
  [
    'number', 'number', 'number', 'number', 'number',
    'number', 'number', 'number'
  ]);

/**
 * @hidden
 */
export const dgesv_wrap = em.cwrap('dgesv_',
  null,
  [
    'number', 'number', 'number', 'number', 'number',
    'number', 'number', 'number'
  ]);

/**
 * @hidden
 */
export const sgemm_wrap = em.cwrap('sgemm_',
  null,
  [
    'number', 'number', 'number', 'number', 'number',
    'number', 'number', 'number', 'number', 'number',
    'number', 'number', 'number'
  ]);

/**
 * @hidden
 */
export const dgemm_wrap = em.cwrap('dgemm_',
  null,
  [
    'number', 'number', 'number', 'number', 'number',
    'number', 'number', 'number', 'number', 'number',
    'number', 'number', 'number'
  ]);

/**
 * @hidden
 */
export const dgemv_wrap = em.cwrap('dgemv_',
  null,
  [
    'number', 'number', 'number', 'number', 'number',
    'number', 'number', 'number', 'number', 'number'
  ]);

/**
 * @hidden
 */
export const sgemv_wrap = em.cwrap('sgemv_',
  null,
  [
    'number', 'number', 'number', 'number', 'number',
    'number', 'number', 'number', 'number', 'number'
  ]);

/**
 * @hidden
 */
export const sdot_wrap = em.cwrap('sdot_',
  null,
  ['number', 'number', 'number', 'number', 'number']);

/**
 * @hidden
 */
export const ddot_wrap = em.cwrap('ddot_',
  null,
  ['number', 'number', 'number', 'number', 'number']);

/**
 * @hidden
 */
export const dgesdd_wrap = em.cwrap('dgesdd_', null,
  [
    'number','number','number','number','number',
    'number','number','number','number','number',
    'number','number','number','number'
  ]);

/**
 * @hidden
 */
export const sgesdd_wrap = em.cwrap('sgesdd_', null,
  [
    'number','number','number','number','number',
    'number','number','number','number','number',
    'number','number','number','number'
  ]);

/**
 * @hidden
 */
export const sgeqrf_wrap = em.cwrap('sgeqrf_', null,
  [
    'number','number','number','number','number',
    'number','number','number'
  ]);

/**
 * @hidden
 */
export const dgeqrf_wrap = em.cwrap('dgeqrf_', null,
  [
    'number','number','number','number','number',
    'number','number','number'
  ]);

/**
 * @hidden
 */
export const sorgqr_wrap = em.cwrap('sorgqr_', null,
  [
    'number','number','number','number','number',
    'number','number','number','number'
  ]);

/**
 * @hidden
 */
export const dorgqr_wrap = em.cwrap('dorgqr_', null,
  [
    'number','number','number','number','number',
    'number','number','number','number'
  ]);

/**
 * @hidden
 */
export const dgelsd_wrap = em.cwrap('dgelsd_', null,
  [
    'number','number','number','number','number',
    'number','number','number','number','number',
    'number','number','number','number'
  ]);

/**
 * @hidden
 */
export const sgetrf_wrap = em.cwrap('sgetrf_', null,
  [
    'number','number','number','number','number','number'
  ]);

/**
 * @hidden
 */
export const dgetrf_wrap = em.cwrap('dgetrf_', null,
  [
    'number','number','number','number','number','number'
  ]);

/**
 * @hidden
 */
export const sgeev_wrap = em.cwrap('sgeev_', null,
  [
    'number','number','number','number','number',
    'number','number','number','number','number',
    'number','number','number','number'
  ]);

/**
 * @hidden
 */
export const dgeev_wrap = em.cwrap('dgeev_', null,
  [
    'number','number','number','number','number',
    'number','number','number','number','number',
    'number','number','number','number'
  ]);

/**
 * @hidden
 */
export function defineEmVariable(
  type:'i8'|'i32'|'f32'|'f64',
  init?:number)
{
  let p;
  switch(type) {
    case 'i32':
      p = em._malloc(SIZE_INT);
      if(init) {
        em.setValue(p, init, type);
      }
      break;
    case 'i8':
      p = em._malloc(SIZE_CHAR);
      if(init) {
        em.setValue(p, init, type);
      }
      break;
    case 'f32':
      p = em._malloc(SIZE_SINGLE);
      if(init) {
        em.setValue(p, init, 'float');
      }
      break;
    case 'f64':
      p = em._malloc(SIZE_DOUBLE);
      if(init) {
        em.setValue(p, init, 'double');
      }
      break;
    default:
      throw new Error('Unhandled variable type');
  }
  return p;
}

/**
 * @hidden
 */
export function defineEmArrayVariable(
  type:'i8'|'i32'|'f32'|'f64',
  len:number,
  init?:TypedArray
) : [number,TypedArray]
{
  let p:number, arr:TypedArray;
  switch(type) {
    case 'i8':
      p = em._malloc(len * SIZE_CHAR);
      arr = new Int8Array(em.HEAP8.buffer, p, len);
      if(init) {
        arr.set(init);
      }
      return [p,arr];
    case 'i32':
      p = em._malloc(len * SIZE_INT);
      arr = new Int32Array(em.HEAP32.buffer, p, len);
      if(init) {
        arr.set(init);
      }
      return [p,arr];
    case 'f32':
      p = em._malloc(len * SIZE_SINGLE);
      arr = new Float32Array(em.HEAPF32.buffer, p, len);
      if(init) {
        arr.set(init);
      }
      return [p,arr];
    case 'f64':
      p = em._malloc(len * SIZE_DOUBLE);
      arr = new Float64Array(em.HEAPF64.buffer, p, len);
      if(init) {
        arr.set(init);
      }
      return [p,arr];
    default:
      throw new Error('Unhandled type');
  }
}