Initial commit
This commit is contained in:
10
src/build
Executable file
10
src/build
Executable file
@@ -0,0 +1,10 @@
|
|||||||
|
#!/bin/sh
|
||||||
|
|
||||||
|
SUBDIRS="common lsc lsi"
|
||||||
|
|
||||||
|
for dir in ${SUBDIRS}
|
||||||
|
do
|
||||||
|
cd ${dir}
|
||||||
|
make $*
|
||||||
|
cd ..
|
||||||
|
done
|
||||||
19
src/common/Makefile
Normal file
19
src/common/Makefile
Normal file
@@ -0,0 +1,19 @@
|
|||||||
|
# Makefile
|
||||||
|
|
||||||
|
OBJS= hash.o mem.o
|
||||||
|
SRCS= hash.c mem.c
|
||||||
|
|
||||||
|
INCDIR=../include
|
||||||
|
|
||||||
|
CFLAGS+= -Wall -Werror
|
||||||
|
CPPFLAGS+= -I${INCDIR}
|
||||||
|
|
||||||
|
all: ${OBJS}
|
||||||
|
|
||||||
|
install:
|
||||||
|
|
||||||
|
depend:
|
||||||
|
mkdep -- ${CFLAGS} ${CPPFLAGS} ${SRCS}
|
||||||
|
|
||||||
|
clean:
|
||||||
|
rm -f ${OBJS}
|
||||||
105
src/common/hash.c
Normal file
105
src/common/hash.c
Normal file
@@ -0,0 +1,105 @@
|
|||||||
|
/* hash.c */
|
||||||
|
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include "hash.h"
|
||||||
|
#include "mem.h"
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Declare hash as:
|
||||||
|
* struct hashentry hash[HASHSIZE];
|
||||||
|
*
|
||||||
|
* Initialise with hash_init(hash);
|
||||||
|
* free all hash items with hash_clear(hash);
|
||||||
|
*/
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Based on Jenkins one-at-a-time hash, from
|
||||||
|
* http://en.wikipedia.org/wiki/Hash_table
|
||||||
|
*/
|
||||||
|
|
||||||
|
int hashfn(char *key)
|
||||||
|
{
|
||||||
|
uint32_t hash = 0;
|
||||||
|
size_t i;
|
||||||
|
size_t len = strlen(key);
|
||||||
|
|
||||||
|
for (i = 0; i < len; i++)
|
||||||
|
{
|
||||||
|
hash += key[i];
|
||||||
|
hash += (hash << 10);
|
||||||
|
hash ^= (hash >> 6);
|
||||||
|
}
|
||||||
|
|
||||||
|
hash += (hash << 3);
|
||||||
|
hash ^= (hash >> 11);
|
||||||
|
hash += (hash << 15);
|
||||||
|
|
||||||
|
return hash & HASHMASK;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
void hash_iterate(struct hashentry **hash, void (*fn)(struct hashentry *))
|
||||||
|
{
|
||||||
|
int i;
|
||||||
|
struct hashentry *hashptr;
|
||||||
|
|
||||||
|
for (i = 0; i < HASHSIZE; i++)
|
||||||
|
if (hash[i]) {
|
||||||
|
hashptr = hash[i];
|
||||||
|
while (hashptr) {
|
||||||
|
fn(hashptr);
|
||||||
|
hashptr = hashptr->next;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void hash_clear(struct hashentry **hash)
|
||||||
|
{
|
||||||
|
int i;
|
||||||
|
struct hashentry *hashptr;
|
||||||
|
struct hashentry *last;
|
||||||
|
|
||||||
|
for (i = 0; i < HASHSIZE; i++)
|
||||||
|
if (hash[i]) {
|
||||||
|
hashptr = hash[i];
|
||||||
|
while (hashptr) {
|
||||||
|
last = hashptr;
|
||||||
|
hashptr = hashptr->next;
|
||||||
|
free(last);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void hash_init(struct hashentry **hash)
|
||||||
|
{
|
||||||
|
bzero(hash, sizeof(struct hashentry *) * HASHSIZE);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
struct hashentry *hash_lookup(struct hashentry **base, char *key, int create)
|
||||||
|
{
|
||||||
|
struct hashentry *hashptr;
|
||||||
|
int hash;
|
||||||
|
int found;
|
||||||
|
|
||||||
|
hash = hashfn(key);
|
||||||
|
hashptr = base[hash];
|
||||||
|
found = 0;
|
||||||
|
|
||||||
|
while (hashptr) {
|
||||||
|
if (strcmp(key, hashptr->name) == 0)
|
||||||
|
return hashptr;
|
||||||
|
hashptr = hashptr->next;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!create)
|
||||||
|
return NULL;
|
||||||
|
|
||||||
|
hashptr = safe_malloc(sizeof(struct hashentry));
|
||||||
|
hashptr->next = base[hash];
|
||||||
|
hashptr->name = NULL; /* Signal that this is new */
|
||||||
|
base[hash] = hashptr;
|
||||||
|
|
||||||
|
return hashptr;
|
||||||
|
}
|
||||||
26
src/common/mem.c
Normal file
26
src/common/mem.c
Normal file
@@ -0,0 +1,26 @@
|
|||||||
|
/* mem.c */
|
||||||
|
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <err.h>
|
||||||
|
|
||||||
|
void *safe_malloc(size_t size)
|
||||||
|
{
|
||||||
|
void *mem;
|
||||||
|
|
||||||
|
mem = malloc(size);
|
||||||
|
if (mem == NULL)
|
||||||
|
err(EXIT_FAILURE, "Failed to malloc %d bytes", size);
|
||||||
|
// printf("Allocated %d bytes at %p\n", size, mem);
|
||||||
|
return mem;
|
||||||
|
}
|
||||||
|
|
||||||
|
void *safe_realloc(void *ptr, size_t size)
|
||||||
|
{
|
||||||
|
void *mem;
|
||||||
|
|
||||||
|
mem = realloc(ptr, size);
|
||||||
|
if (mem == NULL)
|
||||||
|
err(EXIT_FAILURE, "Failed to relloc %d bytes", size);
|
||||||
|
return mem;
|
||||||
|
}
|
||||||
154
src/include/code.h
Normal file
154
src/include/code.h
Normal file
@@ -0,0 +1,154 @@
|
|||||||
|
/* code.h */
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Opcode is one byte
|
||||||
|
*/
|
||||||
|
|
||||||
|
/* Stack opcodes */
|
||||||
|
#define OP_NOP 0 /* Do nothing */
|
||||||
|
#define OP_PUSH 1 /* Push immediate */
|
||||||
|
#define OP_PUSHSTR 2 /* Push immediate string */
|
||||||
|
#define OP_POP 3 /* sp += n */
|
||||||
|
#define OP_ALLOC 4 /* sp -= n */
|
||||||
|
#define OP_LOAD 5 /* push(*(sp+n)) */
|
||||||
|
#define OP_STORE 6 /* *(sp+n) = pop() ; use sp after pop */
|
||||||
|
|
||||||
|
/* Branch opcodes are odd if link, even otherwise */
|
||||||
|
#define OP_B 8 /* relative branch */
|
||||||
|
#define OP_BL 9
|
||||||
|
#define OP_BI 10 /* indirect branch */
|
||||||
|
#define OP_BIL 11
|
||||||
|
#define OP_BZ 12 /* pop and relative branch if zero */
|
||||||
|
#define OP_BZL 13
|
||||||
|
#define OP_BNZ 14 /* pop and relative branch if not zero */
|
||||||
|
#define OP_BNZL 15
|
||||||
|
#define OP_RET 16 /* pop and branch to return address */
|
||||||
|
|
||||||
|
/* Calls */
|
||||||
|
#define OP_CALLNUM 17 /* call numbered function */
|
||||||
|
#define OP_CALLSTR 18 /* call string function */
|
||||||
|
|
||||||
|
/* Arithmetic */
|
||||||
|
#define OP_ADD 19
|
||||||
|
#define OP_SUB 20
|
||||||
|
#define OP_MUL 21
|
||||||
|
#define OP_DIV 22
|
||||||
|
#define OP_EQ 23
|
||||||
|
#define OP_NE 24
|
||||||
|
#define OP_LT 25
|
||||||
|
#define OP_GT 26
|
||||||
|
#define OP_LE 27
|
||||||
|
#define OP_GE 28
|
||||||
|
|
||||||
|
#define FLOP_ADD 29
|
||||||
|
#define FLOP_SUB 30
|
||||||
|
#define FLOP_MUL 31
|
||||||
|
#define FLOP_DIV 32
|
||||||
|
#define FLOP_EQ 33
|
||||||
|
#define FLOP_NE 34
|
||||||
|
#define FLOP_LT 35
|
||||||
|
#define FLOP_GT 36
|
||||||
|
#define FLOP_LE 37
|
||||||
|
#define FLOP_GE 38
|
||||||
|
|
||||||
|
char *instr_names[] = {
|
||||||
|
"NOP",
|
||||||
|
"PUSH",
|
||||||
|
"PUSHSTR",
|
||||||
|
"POP",
|
||||||
|
"ALLOC",
|
||||||
|
"LOAD",
|
||||||
|
"STORE",
|
||||||
|
"ERR",
|
||||||
|
"B",
|
||||||
|
"BL",
|
||||||
|
"BI",
|
||||||
|
"BIL",
|
||||||
|
"BZ",
|
||||||
|
"BZL",
|
||||||
|
"BNZ",
|
||||||
|
"BNZL",
|
||||||
|
"RET",
|
||||||
|
"CALLNUM",
|
||||||
|
"CALLSTR",
|
||||||
|
"ADD",
|
||||||
|
"SUB",
|
||||||
|
"MUL",
|
||||||
|
"DIV",
|
||||||
|
"EQ",
|
||||||
|
"NE",
|
||||||
|
"LT",
|
||||||
|
"GT",
|
||||||
|
"LE",
|
||||||
|
"GE",
|
||||||
|
"FLADD",
|
||||||
|
"FLSUB",
|
||||||
|
"FLMUL",
|
||||||
|
"FLDIV",
|
||||||
|
"FLEQ",
|
||||||
|
"FLNE",
|
||||||
|
"FLLT",
|
||||||
|
"FLGT",
|
||||||
|
"FLLE",
|
||||||
|
"FLGE",
|
||||||
|
};
|
||||||
|
|
||||||
|
#define NINSTR (sizeof(instr_names)/sizeof(char *))
|
||||||
|
#define INSTR_NAME(x) (instr_names[x])
|
||||||
|
|
||||||
|
/*
|
||||||
|
* For opcodes which take an operand, the byte count is stored
|
||||||
|
* in the top 2 bits
|
||||||
|
*/
|
||||||
|
#define BYTECOUNT(x) ((((x) >> 6) & 3) + 1)
|
||||||
|
#define SET_BYTECOUNT(x) (((x)-1)<<6)
|
||||||
|
#define OPCODE_MASK(x) ((x) & 0x3f)
|
||||||
|
|
||||||
|
/*
|
||||||
|
* For OP_PUSHSTR and OP_CALLSTR, the byte count is the number of bytes
|
||||||
|
* which follow the opcode to describe the length of the string. In
|
||||||
|
* most cases, the string will be <256 bytes long, so the byte count will
|
||||||
|
* be 1.
|
||||||
|
*/
|
||||||
|
|
||||||
|
|
||||||
|
/* Internal functions */
|
||||||
|
|
||||||
|
#define INTFN_NOP 0
|
||||||
|
#define INTFN_GLOBAL_STORE 1
|
||||||
|
#define INTFN_GLOBAL_LOAD 2
|
||||||
|
#define INTFN_GLOBAL_ARRAY_STORE 3
|
||||||
|
#define INTFN_GLOBAL_ARRAY_LOAD 4
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
* This is important because sizeof(stkentry) must be the same both
|
||||||
|
* in the compiler and the vm, in order to get string handling right.
|
||||||
|
*/
|
||||||
|
typedef int stkentry;
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Structure of the file header:
|
||||||
|
*
|
||||||
|
* 2 "LC" (magic)
|
||||||
|
* 2 0 (version)
|
||||||
|
* 4 ptr (pointer to function table)
|
||||||
|
*
|
||||||
|
* ptr-4 code
|
||||||
|
*
|
||||||
|
* 1 or more of:
|
||||||
|
* 4 length of block, or 0 to terminate
|
||||||
|
* 4 type of block (0 for function)
|
||||||
|
* 4 pointer to function (offset within file)
|
||||||
|
* 4 number of arguments (unused for now)
|
||||||
|
* n NULL-terminated function name.
|
||||||
|
* - align to 4-byte boundary
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
|
||||||
|
#define MAGIC1 'L'
|
||||||
|
#define MAGIC2 'C'
|
||||||
|
#define VERSION1 0
|
||||||
|
#define VERSION2 0
|
||||||
|
|
||||||
39
src/include/hash.h
Normal file
39
src/include/hash.h
Normal file
@@ -0,0 +1,39 @@
|
|||||||
|
/* hash.h */
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Declare hash as:
|
||||||
|
* struct hashentry *hash[HASHSIZE];
|
||||||
|
*
|
||||||
|
* Initialise with hash_init(hash);
|
||||||
|
* free all hash items with hash_clear(hash);
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
|
||||||
|
struct hashentry {
|
||||||
|
char *name;
|
||||||
|
int value;
|
||||||
|
int flags;
|
||||||
|
struct hashentry *next;
|
||||||
|
};
|
||||||
|
|
||||||
|
#define HASHSIZE 512
|
||||||
|
#define HASHMASK (HASHSIZE - 1)
|
||||||
|
|
||||||
|
typedef void (*hash_action)(struct hashentry *);
|
||||||
|
|
||||||
|
int hashfn(char *);
|
||||||
|
void hash_iterate(struct hashentry **, hash_action);
|
||||||
|
void hash_clear(struct hashentry **);
|
||||||
|
void hash_init(struct hashentry **);
|
||||||
|
|
||||||
|
/*
|
||||||
|
* hash_lookup - lookup item in the hash
|
||||||
|
*
|
||||||
|
* The action on failure is determined by arg 3. If non-zero,
|
||||||
|
* then a hash entry will be created and returned. Otherwise,
|
||||||
|
* NULL is returned. If a new item is created, the name field
|
||||||
|
* will be NULL, and it is the responsibility of the caller
|
||||||
|
* to fill this in correctly.
|
||||||
|
*/
|
||||||
|
struct hashentry *hash_lookup(struct hashentry **, char *, int);
|
||||||
|
|
||||||
4
src/include/mem.h
Normal file
4
src/include/mem.h
Normal file
@@ -0,0 +1,4 @@
|
|||||||
|
/* mem.h */
|
||||||
|
|
||||||
|
void *safe_malloc(size_t);
|
||||||
|
void *safe_realloc(void *, size_t);
|
||||||
40
src/lsc/Makefile
Normal file
40
src/lsc/Makefile
Normal file
@@ -0,0 +1,40 @@
|
|||||||
|
# Makefile
|
||||||
|
|
||||||
|
PREFIX?= /usr/local
|
||||||
|
|
||||||
|
OBJS= lexer.o parser.o ast.o codegen.o
|
||||||
|
SRCS= ast.c codegen.c
|
||||||
|
|
||||||
|
COMMONOBJS= mem.o hash.o
|
||||||
|
COMMONDIR= ../common
|
||||||
|
|
||||||
|
INCDIR=../include
|
||||||
|
|
||||||
|
PROGOBJS= ${OBJS} ${COMMONOBJS:S/^/${COMMONDIR}\//}
|
||||||
|
|
||||||
|
YACC= yacc -d
|
||||||
|
|
||||||
|
CFLAGS+= -Wall -Werror
|
||||||
|
CPPFLAGS+= -I${INCDIR}
|
||||||
|
|
||||||
|
|
||||||
|
lsc: ${OBJS}
|
||||||
|
${LINK.c} -o ${.TARGET} ${PROGOBJS}
|
||||||
|
|
||||||
|
y.tab.h: parser.o
|
||||||
|
|
||||||
|
lexer.o: lexer.l y.tab.h types.h
|
||||||
|
|
||||||
|
parser.o: parser.y ast.h types.h codegen.h
|
||||||
|
|
||||||
|
|
||||||
|
install: lsc
|
||||||
|
${INSTALL} -d ${PREFIX}/bin
|
||||||
|
${INSTALL} -c lsc ${PREFIX}/bin/lsc
|
||||||
|
|
||||||
|
depend:
|
||||||
|
mkdep -- ${CFLAGS} ${CPPFLAGS} ${SRCS}
|
||||||
|
|
||||||
|
clean:
|
||||||
|
rm -f ${OBJS} lsc y.tab.h
|
||||||
|
|
||||||
413
src/lsc/ast.c
Normal file
413
src/lsc/ast.c
Normal file
@@ -0,0 +1,413 @@
|
|||||||
|
/* ast.c */
|
||||||
|
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <err.h>
|
||||||
|
#include "types.h"
|
||||||
|
#include "ast.h"
|
||||||
|
#include "mem.h"
|
||||||
|
|
||||||
|
ast_list *make_list(ast *elem, ast_list *tail)
|
||||||
|
{
|
||||||
|
ast_list *list = safe_malloc(sizeof(ast_list));
|
||||||
|
|
||||||
|
list->elem = elem;
|
||||||
|
list->next = tail;
|
||||||
|
|
||||||
|
return list;
|
||||||
|
}
|
||||||
|
|
||||||
|
ast_list *reverse_list(ast_list *list)
|
||||||
|
{
|
||||||
|
ast_list *last = NULL;
|
||||||
|
ast_list *next;
|
||||||
|
|
||||||
|
while (list) {
|
||||||
|
next = list->next;
|
||||||
|
list->next = last;
|
||||||
|
last = list;
|
||||||
|
list = next;
|
||||||
|
}
|
||||||
|
|
||||||
|
return last;
|
||||||
|
}
|
||||||
|
|
||||||
|
ast *make_list_node(ast_list *list)
|
||||||
|
{
|
||||||
|
ast *elem = safe_malloc(sizeof(ast));
|
||||||
|
|
||||||
|
elem->tag = node_ast;
|
||||||
|
elem->info.node.tag = kind_list;
|
||||||
|
elem->info.node.head = list;
|
||||||
|
|
||||||
|
return elem;
|
||||||
|
}
|
||||||
|
|
||||||
|
ast *make_string(char *string)
|
||||||
|
{
|
||||||
|
ast *elem = safe_malloc(sizeof(ast));
|
||||||
|
|
||||||
|
elem->tag = str_ast;
|
||||||
|
elem->info.string = string;
|
||||||
|
|
||||||
|
return elem;
|
||||||
|
}
|
||||||
|
|
||||||
|
ast *make_integer(long integer)
|
||||||
|
{
|
||||||
|
ast *elem = safe_malloc(sizeof(ast));
|
||||||
|
|
||||||
|
elem->tag = int_ast;
|
||||||
|
elem->info.integer = integer;
|
||||||
|
|
||||||
|
return elem;
|
||||||
|
}
|
||||||
|
|
||||||
|
ast *make_real(float real)
|
||||||
|
{
|
||||||
|
ast *elem = safe_malloc(sizeof(ast));
|
||||||
|
|
||||||
|
elem->tag = real_ast;
|
||||||
|
elem->info.real = real;
|
||||||
|
|
||||||
|
return elem;
|
||||||
|
}
|
||||||
|
|
||||||
|
ast *make_function(char *identifier, ast_list *args, ast_list *stmts)
|
||||||
|
{
|
||||||
|
ast *elem = safe_malloc(sizeof(ast));
|
||||||
|
ast_list *list = NULL;
|
||||||
|
|
||||||
|
elem->tag = node_ast;
|
||||||
|
elem->info.node.tag = kind_fndef;
|
||||||
|
list = make_list(make_list_node(stmts), list);
|
||||||
|
list = make_list(make_list_node(args), list);
|
||||||
|
list = make_list(make_string(identifier), list);
|
||||||
|
elem->info.node.head = list;
|
||||||
|
|
||||||
|
return elem;
|
||||||
|
}
|
||||||
|
|
||||||
|
ast *make_fndefint(char *identifier, long num)
|
||||||
|
{
|
||||||
|
ast *elem = safe_malloc(sizeof(ast));
|
||||||
|
ast_list *list = NULL;
|
||||||
|
|
||||||
|
elem->tag = node_ast;
|
||||||
|
elem->info.node.tag = kind_fndefint;
|
||||||
|
list = make_list(make_integer(num), list);
|
||||||
|
list = make_list(make_string(identifier), list);
|
||||||
|
elem->info.node.head = list;
|
||||||
|
|
||||||
|
return elem;
|
||||||
|
}
|
||||||
|
|
||||||
|
ast *make_constant(char *identifier, long num)
|
||||||
|
{
|
||||||
|
ast *elem = safe_malloc(sizeof(ast));
|
||||||
|
ast_list *list = NULL;
|
||||||
|
|
||||||
|
elem->tag = node_ast;
|
||||||
|
elem->info.node.tag = kind_constant;
|
||||||
|
list = make_list(make_integer(num), list);
|
||||||
|
list = make_list(make_string(identifier), list);
|
||||||
|
elem->info.node.head = list;
|
||||||
|
|
||||||
|
return elem;
|
||||||
|
}
|
||||||
|
|
||||||
|
ast *make_realconstant(char *identifier, float num)
|
||||||
|
{
|
||||||
|
ast *elem = safe_malloc(sizeof(ast));
|
||||||
|
ast_list *list = NULL;
|
||||||
|
|
||||||
|
elem->tag = node_ast;
|
||||||
|
elem->info.node.tag = kind_constant;
|
||||||
|
list = make_list(make_real(num), list);
|
||||||
|
list = make_list(make_string(identifier), list);
|
||||||
|
elem->info.node.head = list;
|
||||||
|
|
||||||
|
return elem;
|
||||||
|
}
|
||||||
|
|
||||||
|
ast *make_fndefext(char *identifier)
|
||||||
|
{
|
||||||
|
ast *elem = safe_malloc(sizeof(ast));
|
||||||
|
ast_list *list = NULL;
|
||||||
|
|
||||||
|
elem->tag = node_ast;
|
||||||
|
elem->info.node.tag = kind_fndefext;
|
||||||
|
list = make_list(make_string(identifier), list);
|
||||||
|
elem->info.node.head = list;
|
||||||
|
|
||||||
|
return elem;
|
||||||
|
}
|
||||||
|
|
||||||
|
ast *make_variable(char *identifier)
|
||||||
|
{
|
||||||
|
ast *elem = safe_malloc(sizeof(ast));
|
||||||
|
|
||||||
|
elem->tag = var_ast;
|
||||||
|
elem->info.variable = identifier;
|
||||||
|
|
||||||
|
return elem;
|
||||||
|
}
|
||||||
|
|
||||||
|
ast *make_array(char *identifier, ast *expression)
|
||||||
|
{
|
||||||
|
ast *elem = safe_malloc(sizeof(ast));
|
||||||
|
ast_list *list;
|
||||||
|
|
||||||
|
elem->tag = array_ast;
|
||||||
|
elem->info.node.tag = kind_array;
|
||||||
|
list = make_list(expression, NULL);
|
||||||
|
list = make_list(make_variable(identifier), list);
|
||||||
|
elem->info.node.head = list;
|
||||||
|
|
||||||
|
return elem;
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
ast *make_assignment(char *identifier, ast *expression)
|
||||||
|
{
|
||||||
|
ast *elem = safe_malloc(sizeof(ast));
|
||||||
|
ast_list *list;
|
||||||
|
|
||||||
|
elem->tag = node_ast;
|
||||||
|
elem->info.node.tag = kind_assign;
|
||||||
|
list = make_list(expression, NULL);
|
||||||
|
list = make_list(make_variable(identifier), list);
|
||||||
|
elem->info.node.head = list;
|
||||||
|
|
||||||
|
return elem;
|
||||||
|
}
|
||||||
|
*/
|
||||||
|
|
||||||
|
ast *make_assignment(ast_list *identifier_list, ast *expression)
|
||||||
|
{
|
||||||
|
ast *elem = safe_malloc(sizeof(ast));
|
||||||
|
ast_list *list;
|
||||||
|
|
||||||
|
elem->tag = node_ast;
|
||||||
|
elem->info.node.tag = kind_assign;
|
||||||
|
list = make_list(expression, NULL);
|
||||||
|
list = make_list(make_list_node(identifier_list), list);
|
||||||
|
elem->info.node.head = list;
|
||||||
|
|
||||||
|
return elem;
|
||||||
|
}
|
||||||
|
|
||||||
|
ast *make_binary_op(ast_kind op, ast *exp1, ast *exp2)
|
||||||
|
{
|
||||||
|
ast *elem = safe_malloc(sizeof(ast));
|
||||||
|
ast_list *list;
|
||||||
|
|
||||||
|
elem->tag = node_ast;
|
||||||
|
elem->info.node.tag = op;
|
||||||
|
list = make_list(exp2, NULL);
|
||||||
|
list = make_list(exp1, list);
|
||||||
|
elem->info.node.head = list;
|
||||||
|
|
||||||
|
return elem;
|
||||||
|
}
|
||||||
|
|
||||||
|
ast *make_unary_op(ast_kind op, ast *exp1)
|
||||||
|
{
|
||||||
|
ast *elem = safe_malloc(sizeof(ast));
|
||||||
|
ast_list *list;
|
||||||
|
|
||||||
|
elem->tag = node_ast;
|
||||||
|
elem->info.node.tag = op;
|
||||||
|
list = make_list(exp1, NULL);
|
||||||
|
elem->info.node.head = list;
|
||||||
|
|
||||||
|
return elem;
|
||||||
|
}
|
||||||
|
|
||||||
|
ast *make_call(char *identifier, ast_list *args)
|
||||||
|
{
|
||||||
|
ast *elem = safe_malloc(sizeof(ast));
|
||||||
|
ast_list *list;
|
||||||
|
|
||||||
|
elem->tag = node_ast;
|
||||||
|
elem->info.node.tag = kind_call;
|
||||||
|
list = make_list(make_list_node(args), NULL);
|
||||||
|
list = make_list(make_string(identifier), list);
|
||||||
|
elem->info.node.head = list;
|
||||||
|
|
||||||
|
return elem;
|
||||||
|
}
|
||||||
|
|
||||||
|
ast *make_statement(ast_kind type, ast *expression, ast_list *statements,
|
||||||
|
ast_list *more_statements)
|
||||||
|
{
|
||||||
|
ast *elem = safe_malloc(sizeof(ast));
|
||||||
|
ast_list *list;
|
||||||
|
|
||||||
|
elem->tag = node_ast;
|
||||||
|
elem->info.node.tag = type;
|
||||||
|
list = NULL;
|
||||||
|
if (more_statements)
|
||||||
|
list = make_list(make_list_node(more_statements), list);
|
||||||
|
list = make_list(make_list_node(statements), list);
|
||||||
|
list = make_list(expression, list);
|
||||||
|
elem->info.node.head = list;
|
||||||
|
|
||||||
|
return elem;
|
||||||
|
}
|
||||||
|
|
||||||
|
ast *make_case_statement_variable(char *identifier)
|
||||||
|
{
|
||||||
|
ast *elem = safe_malloc(sizeof(ast));
|
||||||
|
|
||||||
|
elem->tag = casevar_ast;
|
||||||
|
elem->info.variable = identifier;
|
||||||
|
|
||||||
|
return elem;
|
||||||
|
}
|
||||||
|
|
||||||
|
ast *make_case_statement_number(long num)
|
||||||
|
{
|
||||||
|
ast *elem = safe_malloc(sizeof(ast));
|
||||||
|
|
||||||
|
elem->tag = casenum_ast;
|
||||||
|
elem->info.integer = num;
|
||||||
|
|
||||||
|
return elem;
|
||||||
|
}
|
||||||
|
|
||||||
|
ast *make_return_statement(void)
|
||||||
|
{
|
||||||
|
ast *elem = safe_malloc(sizeof(ast));
|
||||||
|
|
||||||
|
elem->tag = node_ast;
|
||||||
|
elem->info.node.tag = stmt_return;
|
||||||
|
elem->info.node.head = NULL;
|
||||||
|
|
||||||
|
return elem;
|
||||||
|
}
|
||||||
|
|
||||||
|
ast *make_break_statement(void)
|
||||||
|
{
|
||||||
|
ast *elem = safe_malloc(sizeof(ast));
|
||||||
|
|
||||||
|
elem->tag = node_ast;
|
||||||
|
elem->info.node.tag = stmt_break;
|
||||||
|
elem->info.node.head = NULL;
|
||||||
|
|
||||||
|
return elem;
|
||||||
|
}
|
||||||
|
|
||||||
|
ast *make_default_statement(void)
|
||||||
|
{
|
||||||
|
ast *elem = safe_malloc(sizeof(ast));
|
||||||
|
|
||||||
|
elem->tag = node_ast;
|
||||||
|
elem->info.node.tag = stmt_default;
|
||||||
|
elem->info.node.head = NULL;
|
||||||
|
|
||||||
|
return elem;
|
||||||
|
}
|
||||||
|
|
||||||
|
void indent(int spaces)
|
||||||
|
{
|
||||||
|
spaces = spaces * 2;
|
||||||
|
while (spaces--) {
|
||||||
|
printf(" ");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void ast_dump_main(ast *, int);
|
||||||
|
|
||||||
|
void ast_dump_list(ast_list *list, int spaces)
|
||||||
|
{
|
||||||
|
while (list) {
|
||||||
|
ast_dump_main(list->elem, spaces);
|
||||||
|
list = list->next;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void ast_dump_main(ast *node, int spaces)
|
||||||
|
{
|
||||||
|
indent(spaces);
|
||||||
|
switch (node->tag) {
|
||||||
|
case int_ast:
|
||||||
|
printf("integer: %ld\n", node->info.integer);
|
||||||
|
break;
|
||||||
|
case real_ast:
|
||||||
|
printf("real: %g\n", node->info.real);
|
||||||
|
break;
|
||||||
|
case var_ast:
|
||||||
|
printf("variable: %s\n", node->info.variable);
|
||||||
|
break;
|
||||||
|
case str_ast:
|
||||||
|
printf("string: %s\n", node->info.string);
|
||||||
|
break;
|
||||||
|
case node_ast:
|
||||||
|
switch (node->info.node.tag) {
|
||||||
|
case kind_fndef:
|
||||||
|
printf("function definition:\n");
|
||||||
|
break;
|
||||||
|
case kind_assign:
|
||||||
|
printf("assignment:\n");
|
||||||
|
break;
|
||||||
|
case kind_list:
|
||||||
|
printf("generic list:\n");
|
||||||
|
break;
|
||||||
|
case kind_call:
|
||||||
|
printf("function call:\n");
|
||||||
|
break;
|
||||||
|
case op_plus:
|
||||||
|
printf("plus operation:\n");
|
||||||
|
break;
|
||||||
|
case op_minus:
|
||||||
|
printf("minus operation:\n");
|
||||||
|
break;
|
||||||
|
case op_times:
|
||||||
|
printf("times operation:\n");
|
||||||
|
break;
|
||||||
|
case op_divide:
|
||||||
|
printf("divide operation:\n");
|
||||||
|
break;
|
||||||
|
case op_gt:
|
||||||
|
printf("greater than operation:\n");
|
||||||
|
break;
|
||||||
|
case op_lt:
|
||||||
|
printf("less than operation:\n");
|
||||||
|
break;
|
||||||
|
case op_ge:
|
||||||
|
printf("greater than or equal operation:\n");
|
||||||
|
break;
|
||||||
|
case op_le:
|
||||||
|
printf("less than or equal operation:\n");
|
||||||
|
break;
|
||||||
|
case op_eq:
|
||||||
|
printf("equality operation:\n");
|
||||||
|
break;
|
||||||
|
case op_ne:
|
||||||
|
printf("not equal operation:\n");
|
||||||
|
break;
|
||||||
|
case stmt_if:
|
||||||
|
printf("if statement:\n");
|
||||||
|
break;
|
||||||
|
case stmt_while:
|
||||||
|
printf("while statement:\n");
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
printf("unknown list type:\n");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
ast_dump_list(node->info.node.head, spaces+1);
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
printf("unknown node type\n");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void ast_dump(ast *node)
|
||||||
|
{
|
||||||
|
ast_dump_main(node, 0);
|
||||||
|
}
|
||||||
|
|
||||||
27
src/lsc/ast.h
Normal file
27
src/lsc/ast.h
Normal file
@@ -0,0 +1,27 @@
|
|||||||
|
/* ast.h */
|
||||||
|
|
||||||
|
ast_list *make_list(ast *, ast_list *);
|
||||||
|
ast_list *reverse_list(ast_list *);
|
||||||
|
ast *make_list_node(ast_list *);
|
||||||
|
ast *make_function(char *, ast_list *, ast_list *);
|
||||||
|
ast *make_fndefint(char *, long);
|
||||||
|
ast *make_constant(char *, long);
|
||||||
|
ast *make_realconstant(char *, float);
|
||||||
|
ast *make_fndefext(char *);
|
||||||
|
ast *make_variable(char *);
|
||||||
|
ast *make_array(char *, ast *);
|
||||||
|
ast *make_string(char *);
|
||||||
|
ast *make_assignment(ast_list *, ast *);
|
||||||
|
ast *make_binary_op(ast_kind, ast *, ast *);
|
||||||
|
ast *make_unary_op(ast_kind, ast *);
|
||||||
|
ast *make_call(char *, ast_list *);
|
||||||
|
ast *make_integer(long);
|
||||||
|
ast *make_real(float);
|
||||||
|
ast *make_statement(ast_kind, ast *, ast_list *, ast_list *);
|
||||||
|
ast *make_case_statement_number(long);
|
||||||
|
ast *make_case_statement_variable(char *);
|
||||||
|
ast *make_break_statement();
|
||||||
|
ast *make_default_statement();
|
||||||
|
ast *make_return_statement();
|
||||||
|
|
||||||
|
void ast_dump(ast *);
|
||||||
1314
src/lsc/codegen.c
Normal file
1314
src/lsc/codegen.c
Normal file
File diff suppressed because it is too large
Load Diff
4
src/lsc/codegen.h
Normal file
4
src/lsc/codegen.h
Normal file
@@ -0,0 +1,4 @@
|
|||||||
|
/* codegen.h */
|
||||||
|
|
||||||
|
void codegen(ast *);
|
||||||
|
void output_code(void);
|
||||||
37
src/lsc/header.lh
Normal file
37
src/lsc/header.lh
Normal file
@@ -0,0 +1,37 @@
|
|||||||
|
/* header.lh */
|
||||||
|
|
||||||
|
/* Standard header */
|
||||||
|
|
||||||
|
fndefint __nop 0;
|
||||||
|
fndefint __global_store 1;
|
||||||
|
fndefint __global_load 2;
|
||||||
|
fndefint __global_array_store 3;
|
||||||
|
fndefint __global_array_load 4;
|
||||||
|
fndefint __printint 5;
|
||||||
|
fndefint __printreal 6;
|
||||||
|
fndefint __printstr 7;
|
||||||
|
fndefint dmx_setchannel 8;
|
||||||
|
fndefint dmx_output 9;
|
||||||
|
fndefint gettime 10;
|
||||||
|
fndefint waittime 11;
|
||||||
|
fndefint wait 12;
|
||||||
|
fndefint wakeup 13;
|
||||||
|
fndefint spawn 14;
|
||||||
|
fndefint midi_read 15;
|
||||||
|
fndefint beatdetect_read 16;
|
||||||
|
fndefint beatdetect_phase 17;
|
||||||
|
fndefint beatdetect_confidence 18;
|
||||||
|
fndefint realtoint 19;
|
||||||
|
fndefint inttoreal 20;
|
||||||
|
fndefint map3d_setcal 21;
|
||||||
|
fndefint map3d_calibrate 22;
|
||||||
|
fndefint map3d_transform 23;
|
||||||
|
fndefint map3d_setparams 24;
|
||||||
|
fndefint map3d_load 25;
|
||||||
|
fndefint map3d_save 26;
|
||||||
|
fndefint sin 27;
|
||||||
|
fndefint cos 28;
|
||||||
|
fndefint random 29;
|
||||||
|
|
||||||
|
constant _dmxoutput 10;
|
||||||
|
constant _beat 11;
|
||||||
146
src/lsc/lexer.l
Normal file
146
src/lsc/lexer.l
Normal file
@@ -0,0 +1,146 @@
|
|||||||
|
/* lexer.l */
|
||||||
|
|
||||||
|
/*
|
||||||
|
* The incl state is used for picking up the name of an include file
|
||||||
|
*/
|
||||||
|
%x incl
|
||||||
|
|
||||||
|
%x comment
|
||||||
|
%x str
|
||||||
|
|
||||||
|
%{
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <err.h>
|
||||||
|
#include "types.h"
|
||||||
|
#include "y.tab.h"
|
||||||
|
|
||||||
|
#define MAX_INCLUDE_DEPTH 10
|
||||||
|
YY_BUFFER_STATE include_stack[MAX_INCLUDE_DEPTH];
|
||||||
|
int lineno_stack[MAX_INCLUDE_DEPTH];
|
||||||
|
int include_stack_ptr = 0;
|
||||||
|
|
||||||
|
#define MAX_STR_CONST 1024
|
||||||
|
char string_buf[MAX_STR_CONST];
|
||||||
|
char *string_buf_ptr;
|
||||||
|
|
||||||
|
%}
|
||||||
|
|
||||||
|
%option yylineno
|
||||||
|
|
||||||
|
%%
|
||||||
|
|
||||||
|
%include BEGIN(incl);
|
||||||
|
"/*" BEGIN(comment);
|
||||||
|
|
||||||
|
<comment>{
|
||||||
|
[^*]* /* eat anything that's not a '*' */
|
||||||
|
"*"+[^*/]* /* eat up '*'s not followed by '/'s */
|
||||||
|
"*"+"/" BEGIN(INITIAL);
|
||||||
|
}
|
||||||
|
|
||||||
|
\" string_buf_ptr = string_buf; BEGIN(str);
|
||||||
|
|
||||||
|
<str>{
|
||||||
|
\" {
|
||||||
|
BEGIN(INITIAL);
|
||||||
|
*string_buf_ptr = '\0';
|
||||||
|
yylval.Tstring = strdup(string_buf);
|
||||||
|
return STRING;
|
||||||
|
}
|
||||||
|
\n {
|
||||||
|
fprintf(stderr, "Unterminated string at line %d\n", yylineno);
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
\\[0-7]{1,3} {
|
||||||
|
/* octal escape sequence */
|
||||||
|
int result;
|
||||||
|
(void) sscanf(yytext + 1, "%o", &result);
|
||||||
|
if (result > 0xff) {
|
||||||
|
fprintf(stderr,
|
||||||
|
"octal escape out of bounds at line %d\n",
|
||||||
|
yylineno);
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
\\[0-9]+ {
|
||||||
|
/* bad escape sequence */
|
||||||
|
fprintf(stderr, "bad escape sequence at line %d\n", yylineno);
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
\\n *string_buf_ptr++ = '\n';
|
||||||
|
\\t *string_buf_ptr++ = '\t';
|
||||||
|
\\r *string_buf_ptr++ = '\r';
|
||||||
|
\\b *string_buf_ptr++ = '\b';
|
||||||
|
\\f *string_buf_ptr++ = '\f';
|
||||||
|
|
||||||
|
\\(.|\n) *string_buf_ptr++ = yytext[1];
|
||||||
|
|
||||||
|
[^\\\n\"]+ {
|
||||||
|
char *yptr = yytext;
|
||||||
|
|
||||||
|
while (*yptr)
|
||||||
|
*string_buf_ptr++ = *yptr++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function return TOKFUNCTION;
|
||||||
|
fndefint return TOKFNDEFINT;
|
||||||
|
fndefext return TOKFNDEFEXT;
|
||||||
|
constant return TOKCONSTANT;
|
||||||
|
while return TOKWHILE;
|
||||||
|
if return TOKIF;
|
||||||
|
else return TOKELSE;
|
||||||
|
switch return TOKSWITCH;
|
||||||
|
case return TOKCASE;
|
||||||
|
default return TOKDEFAULT;
|
||||||
|
break return TOKBREAK;
|
||||||
|
return return TOKRETURN;
|
||||||
|
== return TOKEQ;
|
||||||
|
!= return TOKNE;
|
||||||
|
\<= return TOKLE;
|
||||||
|
\>= return TOKGE;
|
||||||
|
&& return TOKAND;
|
||||||
|
\|\| return TOKOR;
|
||||||
|
|
||||||
|
[0-9]+ yylval.Tinteger = atoi(yytext); return NUMBER;
|
||||||
|
-[0-9]+ yylval.Tinteger = atoi(yytext); return NUMBER;
|
||||||
|
[0-9]+\.[0-9]+ yylval.Treal = atof(yytext); return REAL;
|
||||||
|
-[0-9]+\.[0-9]+ yylval.Treal = atof(yytext); return REAL;
|
||||||
|
[_a-zA-Z0-9]+ yylval.Tstring = strdup(yytext); return IDENTIFIER;
|
||||||
|
%[_a-zA-Z0-9]+ yylval.Tstring = strdup(yytext); return IDENTIFIER;
|
||||||
|
$[_a-zA-Z0-9]+ yylval.Tstring = strdup(yytext); return IDENTIFIER;
|
||||||
|
|
||||||
|
[ \t\n]+ /* eat whitespace */
|
||||||
|
|
||||||
|
. return (int) yytext[0];
|
||||||
|
|
||||||
|
<incl>[ \t]* /* eat the whitespace */
|
||||||
|
<incl>[^ \t\n]+ { /* got the include filename */
|
||||||
|
if (include_stack_ptr >= MAX_INCLUDE_DEPTH) {
|
||||||
|
fprintf(stderr, "Includes nested too deeply");
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
include_stack[include_stack_ptr] = YY_CURRENT_BUFFER;
|
||||||
|
lineno_stack[include_stack_ptr++] = yylineno;
|
||||||
|
yyin = fopen(yytext, "r");
|
||||||
|
yylineno = 1;
|
||||||
|
if (!yyin)
|
||||||
|
err(1, "%s", yytext);
|
||||||
|
yy_switch_to_buffer(yy_create_buffer(yyin, YY_BUF_SIZE));
|
||||||
|
BEGIN(INITIAL);
|
||||||
|
}
|
||||||
|
|
||||||
|
<<EOF>> {
|
||||||
|
if (--include_stack_ptr < 0) {
|
||||||
|
yyterminate();
|
||||||
|
} else {
|
||||||
|
yy_delete_buffer(YY_CURRENT_BUFFER);
|
||||||
|
yy_switch_to_buffer(
|
||||||
|
include_stack[include_stack_ptr]);
|
||||||
|
yylineno = lineno_stack[include_stack_ptr];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
%%
|
||||||
|
|
||||||
335
src/lsc/parser.y
Normal file
335
src/lsc/parser.y
Normal file
@@ -0,0 +1,335 @@
|
|||||||
|
%{
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <err.h>
|
||||||
|
#include "types.h"
|
||||||
|
#include "ast.h"
|
||||||
|
#include "codegen.h"
|
||||||
|
|
||||||
|
int yyparse(void);
|
||||||
|
int yylex(void);
|
||||||
|
|
||||||
|
extern int yylineno;
|
||||||
|
char *yytext;
|
||||||
|
|
||||||
|
/* XXX */
|
||||||
|
extern FILE *binout;
|
||||||
|
extern FILE *yyin;
|
||||||
|
|
||||||
|
void yyerror(const char * str)
|
||||||
|
{
|
||||||
|
fprintf(stderr, "\n%s on line %d before '%s'\n", str, yylineno, yytext);
|
||||||
|
}
|
||||||
|
|
||||||
|
int yywrap()
|
||||||
|
{
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
int main(int argc, char *argv[])
|
||||||
|
{
|
||||||
|
argv++; /* skip over program name */
|
||||||
|
argc--;
|
||||||
|
if (argc > 0) {
|
||||||
|
char *outfile;
|
||||||
|
int len;
|
||||||
|
|
||||||
|
outfile = strdup(argv[0]);
|
||||||
|
if (outfile == NULL)
|
||||||
|
err(1, "Failed to allocate memory");
|
||||||
|
len = strlen(outfile);
|
||||||
|
if ((len < 3) ||
|
||||||
|
(outfile[len-1] != 's') ||
|
||||||
|
(outfile[len-2] != 'l') ||
|
||||||
|
(outfile[len-3] != '.'))
|
||||||
|
errx(1, "Can only compile .ls files");
|
||||||
|
outfile[len-1] = 'c';
|
||||||
|
|
||||||
|
yyin = fopen(argv[0], "r");
|
||||||
|
if (yyin == NULL)
|
||||||
|
err(1, "%s: Failed to open input file", argv[0]);
|
||||||
|
binout = fopen(outfile, "w");
|
||||||
|
if (binout == NULL)
|
||||||
|
err(1, "%s: Failed to open output file", outfile);
|
||||||
|
|
||||||
|
free(outfile);
|
||||||
|
} else {
|
||||||
|
yyin = stdin;
|
||||||
|
binout = fopen("out.lc", "w");
|
||||||
|
if (binout == NULL)
|
||||||
|
err(1, "%s: Failed to open output file", "out.lc");
|
||||||
|
}
|
||||||
|
|
||||||
|
yyparse();
|
||||||
|
fclose(binout);
|
||||||
|
fclose(yyin);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
%}
|
||||||
|
|
||||||
|
%token TOKFUNCTION TOKFNDEFINT TOKFNDEFEXT TOKCONSTANT
|
||||||
|
%token TOKWHILE TOKIF TOKELSE TOKSWITCH TOKCASE TOKDEFAULT
|
||||||
|
%token TOKBREAK TOKRETURN TOKEQ TOKNE TOKAND TOKOR STRING
|
||||||
|
|
||||||
|
%union {
|
||||||
|
ast* Tast;
|
||||||
|
ast_list* Tast_list;
|
||||||
|
char * Tstring;
|
||||||
|
long Tinteger;
|
||||||
|
float Treal;
|
||||||
|
}
|
||||||
|
|
||||||
|
%type <Tast> argument
|
||||||
|
%type <Tast> constant
|
||||||
|
%type <Tast> function
|
||||||
|
%type <Tast> fndefint
|
||||||
|
%type <Tast> fndefext
|
||||||
|
%type <Tast> statement
|
||||||
|
%type <Tast> assignment
|
||||||
|
%type <Tast> expression
|
||||||
|
%type <Tast> realconstant
|
||||||
|
%type <Tast> statement_if
|
||||||
|
%type <Tast> statement_case
|
||||||
|
%type <Tast> statement_while
|
||||||
|
%type <Tast> statement_break
|
||||||
|
%type <Tast> statement_switch
|
||||||
|
%type <Tast> statement_return
|
||||||
|
%type <Tast> statement_default
|
||||||
|
%type <Tast_list> stmt_list_inner
|
||||||
|
%type <Tast_list> exp_list_inner
|
||||||
|
%type <Tast_list> arg_list_inner
|
||||||
|
%type <Tast_list> fn_list_inner
|
||||||
|
%type <Tast_list> assign_list
|
||||||
|
%type <Tast_list> stmt_list
|
||||||
|
%type <Tast_list> exp_list
|
||||||
|
%type <Tast_list> arg_list
|
||||||
|
%type <Tast_list> fn_list
|
||||||
|
%token <Tstring> IDENTIFIER
|
||||||
|
%token <Tstring> STRING
|
||||||
|
%token <Tinteger> NUMBER
|
||||||
|
%token <Treal> REAL
|
||||||
|
|
||||||
|
%left '+'
|
||||||
|
%left '-'
|
||||||
|
%left '*'
|
||||||
|
%left '/'
|
||||||
|
%left TOKAND
|
||||||
|
%left TOKOR
|
||||||
|
%nonassoc '<'
|
||||||
|
%nonassoc '>'
|
||||||
|
%nonassoc TOKLE
|
||||||
|
%nonassoc TOKGE
|
||||||
|
%nonassoc TOKEQ
|
||||||
|
%nonassoc TOKNE
|
||||||
|
|
||||||
|
%start start
|
||||||
|
|
||||||
|
%%
|
||||||
|
|
||||||
|
start:
|
||||||
|
fn_list
|
||||||
|
{
|
||||||
|
codegen(make_list_node($1));
|
||||||
|
output_code();
|
||||||
|
}
|
||||||
|
;
|
||||||
|
|
||||||
|
fn_list:
|
||||||
|
fn_list_inner { $$ = reverse_list($1); }
|
||||||
|
;
|
||||||
|
|
||||||
|
fn_list_inner:
|
||||||
|
fn_list_inner function { $$ = make_list($2, $1); }
|
||||||
|
| fn_list_inner fndefint { $$ = make_list($2, $1); }
|
||||||
|
| fn_list_inner fndefext { $$ = make_list($2, $1); }
|
||||||
|
| fn_list_inner constant { $$ = make_list($2, $1); }
|
||||||
|
| fn_list_inner realconstant { $$ = make_list($2, $1); }
|
||||||
|
| { $$ = NULL; }
|
||||||
|
;
|
||||||
|
|
||||||
|
function:
|
||||||
|
TOKFUNCTION IDENTIFIER '(' arg_list ')' '{' stmt_list '}'
|
||||||
|
{ $$ = make_function($2, $4, $7); }
|
||||||
|
;
|
||||||
|
|
||||||
|
fndefint:
|
||||||
|
TOKFNDEFINT IDENTIFIER NUMBER ';'
|
||||||
|
{ $$ = make_fndefint($2, $3); }
|
||||||
|
;
|
||||||
|
|
||||||
|
constant:
|
||||||
|
TOKCONSTANT IDENTIFIER NUMBER ';'
|
||||||
|
{ $$ = make_constant($2, $3); }
|
||||||
|
;
|
||||||
|
|
||||||
|
realconstant:
|
||||||
|
TOKCONSTANT IDENTIFIER REAL ';'
|
||||||
|
{ $$ = make_realconstant($2, $3); }
|
||||||
|
;
|
||||||
|
|
||||||
|
fndefext:
|
||||||
|
TOKFNDEFEXT IDENTIFIER ';'
|
||||||
|
{ $$ = make_fndefext($2); }
|
||||||
|
;
|
||||||
|
|
||||||
|
argument:
|
||||||
|
IDENTIFIER
|
||||||
|
{ $$ = make_variable($1); }
|
||||||
|
;
|
||||||
|
|
||||||
|
arg_list:
|
||||||
|
arg_list_inner { $$ = reverse_list($1); }
|
||||||
|
;
|
||||||
|
|
||||||
|
arg_list_inner:
|
||||||
|
arg_list_inner ',' argument { $$ = make_list($3, $1); }
|
||||||
|
| argument { $$ = make_list($1, NULL); }
|
||||||
|
| { $$ = NULL; }
|
||||||
|
;
|
||||||
|
|
||||||
|
stmt_list:
|
||||||
|
stmt_list_inner { $$ = reverse_list($1); }
|
||||||
|
;
|
||||||
|
|
||||||
|
stmt_list_inner:
|
||||||
|
stmt_list_inner statement { $$ = make_list($2, $1); }
|
||||||
|
| { $$ = NULL; }
|
||||||
|
;
|
||||||
|
|
||||||
|
statement:
|
||||||
|
statement_if
|
||||||
|
|
|
||||||
|
statement_while
|
||||||
|
|
|
||||||
|
statement_switch
|
||||||
|
|
|
||||||
|
statement_case
|
||||||
|
|
|
||||||
|
statement_default
|
||||||
|
|
|
||||||
|
statement_break ';'
|
||||||
|
|
|
||||||
|
assignment ';'
|
||||||
|
|
|
||||||
|
expression ';'
|
||||||
|
{ $$ = make_assignment(NULL, $1); }
|
||||||
|
|
|
||||||
|
statement_return ';'
|
||||||
|
;
|
||||||
|
|
||||||
|
assignment:
|
||||||
|
assign_list '=' expression
|
||||||
|
{ $$ = make_assignment($1, $3); }
|
||||||
|
;
|
||||||
|
|
||||||
|
assign_list:
|
||||||
|
assign_list ',' IDENTIFIER
|
||||||
|
{ $$ = make_list(make_variable($3), $1); }
|
||||||
|
| IDENTIFIER
|
||||||
|
{ $$ = make_list(make_variable($1), NULL); }
|
||||||
|
| assign_list ',' IDENTIFIER '[' expression ']'
|
||||||
|
{ $$ = make_list(make_array($3, $5), $1); }
|
||||||
|
| IDENTIFIER '[' expression ']'
|
||||||
|
{ $$ = make_list(make_array($1, $3), NULL); }
|
||||||
|
;
|
||||||
|
|
||||||
|
expression:
|
||||||
|
expression '+' expression { $$ = make_binary_op(op_plus, $1, $3); }
|
||||||
|
|
|
||||||
|
expression '-' expression { $$ = make_binary_op(op_minus, $1, $3); }
|
||||||
|
|
|
||||||
|
expression '*' expression { $$ = make_binary_op(op_times, $1, $3); }
|
||||||
|
|
|
||||||
|
expression '/' expression { $$ = make_binary_op(op_divide, $1, $3); }
|
||||||
|
|
|
||||||
|
expression '<' expression { $$ = make_binary_op(op_lt, $1, $3); }
|
||||||
|
|
|
||||||
|
expression '>' expression { $$ = make_binary_op(op_gt, $1, $3); }
|
||||||
|
|
|
||||||
|
expression TOKLE expression { $$ = make_binary_op(op_le, $1, $3); }
|
||||||
|
|
|
||||||
|
expression TOKGE expression { $$ = make_binary_op(op_ge, $1, $3); }
|
||||||
|
|
|
||||||
|
expression TOKEQ expression { $$ = make_binary_op(op_eq, $1, $3); }
|
||||||
|
|
|
||||||
|
expression TOKNE expression { $$ = make_binary_op(op_ne, $1, $3); }
|
||||||
|
|
|
||||||
|
expression TOKAND expression { $$ = make_binary_op(op_and, $1, $3); }
|
||||||
|
|
|
||||||
|
expression TOKOR expression { $$ = make_binary_op(op_or, $1, $3); }
|
||||||
|
|
|
||||||
|
'-' expression { $$ = make_unary_op(op_minus, $2); }
|
||||||
|
|
|
||||||
|
'(' expression ')' { $$ = $2; }
|
||||||
|
|
|
||||||
|
IDENTIFIER '(' exp_list ')'
|
||||||
|
{ $$ = make_call($1, $3); }
|
||||||
|
|
|
||||||
|
IDENTIFIER '[' expression ']'
|
||||||
|
{ $$ = make_array($1, $3); }
|
||||||
|
|
|
||||||
|
IDENTIFIER
|
||||||
|
{ $$ = make_variable($1); }
|
||||||
|
|
|
||||||
|
NUMBER
|
||||||
|
{ $$ = make_integer($1); }
|
||||||
|
|
|
||||||
|
REAL
|
||||||
|
{ $$ = make_real($1); }
|
||||||
|
|
|
||||||
|
STRING
|
||||||
|
{ $$ = make_string($1); }
|
||||||
|
;
|
||||||
|
|
||||||
|
exp_list:
|
||||||
|
exp_list_inner { $$ = reverse_list($1); }
|
||||||
|
;
|
||||||
|
|
||||||
|
exp_list_inner:
|
||||||
|
exp_list_inner ',' expression { $$ = make_list($3, $1); }
|
||||||
|
| expression { $$ = make_list($1, NULL); }
|
||||||
|
| { $$ = NULL; }
|
||||||
|
;
|
||||||
|
|
||||||
|
statement_if:
|
||||||
|
TOKIF '(' expression ')' '{' stmt_list '}'
|
||||||
|
{ $$ = make_statement(stmt_if, $3, $6, NULL); }
|
||||||
|
|
|
||||||
|
TOKIF '(' expression ')' '{' stmt_list '}' TOKELSE '{' stmt_list '}'
|
||||||
|
{ $$ = make_statement(stmt_if, $3, $6, $10); }
|
||||||
|
;
|
||||||
|
|
||||||
|
statement_while:
|
||||||
|
TOKWHILE '(' expression ')' '{' stmt_list '}'
|
||||||
|
{ $$ = make_statement(stmt_while, $3, $6, NULL); }
|
||||||
|
;
|
||||||
|
|
||||||
|
statement_switch:
|
||||||
|
TOKSWITCH '(' expression ')' '{' stmt_list '}'
|
||||||
|
{ $$ = make_statement(stmt_switch, $3, $6, NULL); }
|
||||||
|
;
|
||||||
|
|
||||||
|
statement_case:
|
||||||
|
TOKCASE NUMBER ':'
|
||||||
|
{ $$ = make_case_statement_number($2); }
|
||||||
|
|
|
||||||
|
TOKCASE IDENTIFIER ':'
|
||||||
|
{ $$ = make_case_statement_variable($2); }
|
||||||
|
;
|
||||||
|
|
||||||
|
statement_return:
|
||||||
|
TOKRETURN
|
||||||
|
{ $$ = make_return_statement(); }
|
||||||
|
;
|
||||||
|
|
||||||
|
statement_break:
|
||||||
|
TOKBREAK
|
||||||
|
{ $$ = make_break_statement(); }
|
||||||
|
;
|
||||||
|
|
||||||
|
statement_default:
|
||||||
|
TOKDEFAULT ':'
|
||||||
|
{ $$ = make_default_statement(); }
|
||||||
|
;
|
||||||
|
|
||||||
37
src/lsc/types.h
Normal file
37
src/lsc/types.h
Normal file
@@ -0,0 +1,37 @@
|
|||||||
|
/* types.h */
|
||||||
|
|
||||||
|
typedef enum {
|
||||||
|
|
||||||
|
kind_fndef, kind_fndefint, kind_fndefext, kind_constant,
|
||||||
|
kind_assign, kind_list, kind_call, kind_array,
|
||||||
|
|
||||||
|
op_plus, op_minus, op_times, op_divide,
|
||||||
|
op_gt, op_lt, op_ge, op_le, op_eq, op_ne,
|
||||||
|
op_and, op_or,
|
||||||
|
|
||||||
|
stmt_if, stmt_while, stmt_switch, stmt_casenum, stmt_casevar,
|
||||||
|
stmt_break, stmt_default, stmt_return
|
||||||
|
|
||||||
|
} ast_kind;
|
||||||
|
|
||||||
|
typedef struct ast {
|
||||||
|
enum {
|
||||||
|
int_ast, real_ast, var_ast, casenum_ast, casevar_ast,
|
||||||
|
str_ast, node_ast, array_ast
|
||||||
|
} tag;
|
||||||
|
union {
|
||||||
|
long integer;
|
||||||
|
float real;
|
||||||
|
char* variable;
|
||||||
|
char* string;
|
||||||
|
struct {
|
||||||
|
ast_kind tag;
|
||||||
|
struct ast_list* head;
|
||||||
|
} node;
|
||||||
|
} info;
|
||||||
|
} ast;
|
||||||
|
|
||||||
|
typedef struct ast_list {
|
||||||
|
ast* elem;
|
||||||
|
struct ast_list* next;
|
||||||
|
} ast_list;
|
||||||
31
src/lsi/Makefile
Normal file
31
src/lsi/Makefile
Normal file
@@ -0,0 +1,31 @@
|
|||||||
|
# Makefile
|
||||||
|
|
||||||
|
PREFIX?= /usr/local
|
||||||
|
|
||||||
|
OBJS= main.o vm.o dmx.o midi.o beatdetect.o fft.o map3d.o mouse.o
|
||||||
|
SRCS= main.c vm.c dmx.c midi.c beatdetect.c fft.c map3d.c mouse.c
|
||||||
|
|
||||||
|
COMMONOBJS= mem.o hash.o
|
||||||
|
COMMONDIR= ../common
|
||||||
|
|
||||||
|
INCDIR= ../include
|
||||||
|
|
||||||
|
CFLAGS+= -Wall -Werror
|
||||||
|
CPPFLAGS+= -I${INCDIR}
|
||||||
|
LDLIBS+= -lm
|
||||||
|
|
||||||
|
PROGOBJS= ${OBJS} ${COMMONOBJS:S/^/${COMMONDIR}\//}
|
||||||
|
|
||||||
|
|
||||||
|
lsi: ${OBJS}
|
||||||
|
${LINK.c} -o ${.TARGET} ${PROGOBJS} ${LDLIBS}
|
||||||
|
|
||||||
|
install: lsi
|
||||||
|
${INSTALL} -d ${PREFIX}/bin
|
||||||
|
${INSTALL} -c lsi ${PREFIX}/bin/lsi
|
||||||
|
|
||||||
|
depend:
|
||||||
|
mkdep -- ${CFLAGS} ${CPPFLAGS} ${SRCS}
|
||||||
|
|
||||||
|
clean:
|
||||||
|
rm -f ${OBJS} lsi
|
||||||
377
src/lsi/beatdetect.c
Normal file
377
src/lsi/beatdetect.c
Normal file
@@ -0,0 +1,377 @@
|
|||||||
|
/* beatdetect.c */
|
||||||
|
|
||||||
|
#include <sys/types.h>
|
||||||
|
#include <sys/ioctl.h>
|
||||||
|
#include <sys/audioio.h>
|
||||||
|
#include <fcntl.h>
|
||||||
|
#include <signal.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <math.h>
|
||||||
|
#include <stdint.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
#include <err.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <errno.h>
|
||||||
|
|
||||||
|
#include "fft.h"
|
||||||
|
#include "vm.h"
|
||||||
|
|
||||||
|
#define AUDIO_DEVICE "/dev/sound"
|
||||||
|
|
||||||
|
#define BUFSIZE 2048
|
||||||
|
|
||||||
|
#define CLIP 1
|
||||||
|
|
||||||
|
#define HISTSIZE 43
|
||||||
|
#define NOUTPUTS 10
|
||||||
|
#define BAR 1.5
|
||||||
|
|
||||||
|
#define MAXP 44
|
||||||
|
#define MINP 22
|
||||||
|
|
||||||
|
#define CHISTSIZE 512
|
||||||
|
|
||||||
|
double outputslist[NOUTPUTS] = {
|
||||||
|
1, 3, 7, 15, 31, 63, 127, 255, 511, 1023
|
||||||
|
};
|
||||||
|
|
||||||
|
#if 0
|
||||||
|
/*
|
||||||
|
double profile[NOUTPUTS] = {
|
||||||
|
1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0
|
||||||
|
};
|
||||||
|
*/
|
||||||
|
double profile[NOUTPUTS] = {
|
||||||
|
1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 1.0
|
||||||
|
};
|
||||||
|
|
||||||
|
#else
|
||||||
|
|
||||||
|
double profile[NOUTPUTS] = {
|
||||||
|
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif
|
||||||
|
|
||||||
|
int audiofd;
|
||||||
|
#if 0
|
||||||
|
double lavg[NOUTPUTS];
|
||||||
|
double var[NOUTPUTS];
|
||||||
|
#endif
|
||||||
|
int lastmi = 0;
|
||||||
|
int lastmj = 0;
|
||||||
|
int lastmp = 0;
|
||||||
|
double lastmax = 0;
|
||||||
|
int minp, maxp;
|
||||||
|
double lastsum[NOUTPUTS];
|
||||||
|
double history[NOUTPUTS][HISTSIZE];
|
||||||
|
double chistory[NOUTPUTS][CHISTSIZE];
|
||||||
|
int histptr;
|
||||||
|
int chistptr;
|
||||||
|
char buffer[BUFSIZE];
|
||||||
|
char *bufptr;
|
||||||
|
int bufleft;
|
||||||
|
double phase;
|
||||||
|
double confidence;
|
||||||
|
int audio_initialised = 0;
|
||||||
|
|
||||||
|
#define HISTORY(i, x) (history[i][((histptr+(x) >= HISTSIZE) ? histptr+(x)-HISTSIZE : histptr+(x))])
|
||||||
|
|
||||||
|
#define HISTWRITE(i, x) history[i][histptr] = (x)
|
||||||
|
|
||||||
|
#define HISTNEXT do { \
|
||||||
|
histptr++; \
|
||||||
|
if (histptr >= HISTSIZE) \
|
||||||
|
histptr = 0; \
|
||||||
|
} while (0)
|
||||||
|
|
||||||
|
#define CHISTORY(i, x) (chistory[i][((chistptr+(x) >= CHISTSIZE) ? chistptr+(x)-CHISTSIZE : chistptr+(x))])
|
||||||
|
|
||||||
|
#define CHISTWRITE(i, x) chistory[i][chistptr] = (x)
|
||||||
|
|
||||||
|
#define CHISTNEXT do { \
|
||||||
|
chistptr++; \
|
||||||
|
if (chistptr >= CHISTSIZE) \
|
||||||
|
chistptr = 0; \
|
||||||
|
} while (0)
|
||||||
|
|
||||||
|
#define SUMSQ(a, b) ((a) * (a) + (b) * (b))
|
||||||
|
#define MAGSQ(i) SUMSQ(freqs[2*(i)], freqs[2*(i)+1])
|
||||||
|
|
||||||
|
void beatdetect_close(void)
|
||||||
|
{
|
||||||
|
close(audiofd);
|
||||||
|
}
|
||||||
|
|
||||||
|
double beatdetect_getphase(void)
|
||||||
|
{
|
||||||
|
return phase;
|
||||||
|
}
|
||||||
|
|
||||||
|
double beatdetect_getconfidence(void)
|
||||||
|
{
|
||||||
|
return confidence;
|
||||||
|
}
|
||||||
|
|
||||||
|
void beatdetect_init(void)
|
||||||
|
{
|
||||||
|
audio_info_t info, oinfo;
|
||||||
|
|
||||||
|
audiofd = open(AUDIO_DEVICE, O_RDONLY | O_NONBLOCK, 0);
|
||||||
|
if (audiofd < 0)
|
||||||
|
err(1, "failed to open audio device");
|
||||||
|
|
||||||
|
if (ioctl(audiofd, AUDIO_GETINFO, &oinfo) < 0)
|
||||||
|
err(1, "failed to get audio info");
|
||||||
|
|
||||||
|
AUDIO_INITINFO(&info);
|
||||||
|
|
||||||
|
info.record.sample_rate = 44100;
|
||||||
|
info.record.channels = 1;
|
||||||
|
info.record.precision = 16;
|
||||||
|
info.record.encoding = AUDIO_ENCODING_SLINEAR;
|
||||||
|
info.record.gain = oinfo.record.gain;
|
||||||
|
info.record.port = oinfo.record.port;
|
||||||
|
info.record.balance = oinfo.record.balance;
|
||||||
|
|
||||||
|
info.monitor_gain = oinfo.monitor_gain;
|
||||||
|
|
||||||
|
info.mode = AUMODE_RECORD;
|
||||||
|
|
||||||
|
if (ioctl(audiofd, AUDIO_SETINFO, &info) < 0)
|
||||||
|
err(1, "failed to set audio info");
|
||||||
|
|
||||||
|
fft_init(BUFSIZE/2);
|
||||||
|
|
||||||
|
bzero(history, sizeof(history));
|
||||||
|
bzero(chistory, sizeof(chistory));
|
||||||
|
histptr = 0;
|
||||||
|
chistptr = 0;
|
||||||
|
|
||||||
|
bufptr = buffer;
|
||||||
|
bufleft = BUFSIZE;
|
||||||
|
|
||||||
|
audio_initialised = 1;
|
||||||
|
|
||||||
|
/* vm_register_blah */
|
||||||
|
vm_register_signal_fd(audiofd, VM_BEATQ);
|
||||||
|
}
|
||||||
|
|
||||||
|
int beatdetect_read(void)
|
||||||
|
{
|
||||||
|
#if 0
|
||||||
|
double elem;
|
||||||
|
#endif
|
||||||
|
int i, j, n, p;
|
||||||
|
int rv;
|
||||||
|
int clip;
|
||||||
|
double localsum[NOUTPUTS];
|
||||||
|
fft_type *freqs;
|
||||||
|
int count;
|
||||||
|
int nitems;
|
||||||
|
double mean, variance, sd;
|
||||||
|
int beat;
|
||||||
|
double r[NOUTPUTS][MAXP][MAXP-MINP];
|
||||||
|
double max;
|
||||||
|
double cmax;
|
||||||
|
int mi, mj, mp;
|
||||||
|
int cmi, cmj, cmp;
|
||||||
|
|
||||||
|
if (!audio_initialised)
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
while (1) {
|
||||||
|
rv = read(audiofd, bufptr, bufleft);
|
||||||
|
if (rv == -1) {
|
||||||
|
if (errno == EAGAIN) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
printf("audio read failed\n");
|
||||||
|
audio_initialised = 0;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (rv == 0)
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
bufptr += rv;
|
||||||
|
bufleft -= rv;
|
||||||
|
|
||||||
|
if (bufleft != 0) {
|
||||||
|
// printf("audio: short read (read %d)\n", rv);
|
||||||
|
if (bufleft < 0) {
|
||||||
|
printf("audio: oversize read\n");
|
||||||
|
bufptr = buffer;
|
||||||
|
bufleft = BUFSIZE;
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
bufptr = buffer;
|
||||||
|
bufleft = BUFSIZE;
|
||||||
|
|
||||||
|
fft_data_signed16((int16_t *)buffer);
|
||||||
|
|
||||||
|
clip = 0;
|
||||||
|
|
||||||
|
/* Check for clip and compute rms */
|
||||||
|
for (i = 0; i < BUFSIZE/2; i++) {
|
||||||
|
int sample = ((int16_t *)buffer)[i];
|
||||||
|
if ((sample == INT16_MAX) || (sample == (-1 - INT16_MAX)))
|
||||||
|
clip = 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
fft_window();
|
||||||
|
fft_compute();
|
||||||
|
freqs = fft_getresult();
|
||||||
|
|
||||||
|
n = 0;
|
||||||
|
for (i = 0; i < NOUTPUTS; i++) {
|
||||||
|
double output = 0;
|
||||||
|
for (j = 0; i+j < outputslist[n]; j++)
|
||||||
|
output += MAGSQ(i+j);
|
||||||
|
output = output / j;
|
||||||
|
localsum[i] = output;
|
||||||
|
HISTWRITE(i, output);
|
||||||
|
n++;
|
||||||
|
}
|
||||||
|
|
||||||
|
HISTNEXT;
|
||||||
|
|
||||||
|
#if 0
|
||||||
|
for (i = 0; i < NOUTPUTS; i++) {
|
||||||
|
lavg[i] = 0;
|
||||||
|
for (j = 0; j < HISTSIZE; j++) {
|
||||||
|
lavg[i] += HISTORY(i, j);
|
||||||
|
}
|
||||||
|
lavg[i] = lavg[i] / (double)HISTSIZE;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (i = 0; i < NOUTPUTS; i++) {
|
||||||
|
var[i] = 0;
|
||||||
|
for (j = 0; j < HISTSIZE; j++) {
|
||||||
|
elem = HISTORY(i, j) - lavg[i];
|
||||||
|
var[i] += elem * elem;
|
||||||
|
}
|
||||||
|
var[i] = var[i] / (double)HISTSIZE;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#if CLIP
|
||||||
|
if (clip)
|
||||||
|
printf("C");
|
||||||
|
else
|
||||||
|
printf(" ");
|
||||||
|
#endif
|
||||||
|
|
||||||
|
for (i = 0; i < NOUTPUTS; i++) {
|
||||||
|
CHISTWRITE(i, localsum[i] - lastsum[i]);
|
||||||
|
lastsum[i] = localsum[i];
|
||||||
|
}
|
||||||
|
CHISTNEXT;
|
||||||
|
|
||||||
|
max = 0;
|
||||||
|
mi = 0;
|
||||||
|
mp = 0;
|
||||||
|
mj = 0;
|
||||||
|
mean = 0;
|
||||||
|
nitems = 0;
|
||||||
|
|
||||||
|
for (i = 0; i < NOUTPUTS; i++) {
|
||||||
|
for (p = MINP; p < MAXP; p++) {
|
||||||
|
for (j = 0; j < p; j++) {
|
||||||
|
r[i][j][p-MINP] =
|
||||||
|
(CHISTORY(i, CHISTSIZE-j-1) +
|
||||||
|
CHISTORY(i, CHISTSIZE-(p+j)-1) +
|
||||||
|
CHISTORY(i, CHISTSIZE-(2*p+j)-1) +
|
||||||
|
CHISTORY(i, CHISTSIZE-(3*p+j)-1) +
|
||||||
|
CHISTORY(i, CHISTSIZE-(4*p+j)-1) +
|
||||||
|
CHISTORY(i, CHISTSIZE-(5*p+j)-1) +
|
||||||
|
CHISTORY(i, CHISTSIZE-(6*p+j)-1) +
|
||||||
|
CHISTORY(i, CHISTSIZE-(7*p+j)-1))
|
||||||
|
* profile[i];
|
||||||
|
// printf("i j p = %d %d %d\n", i, j, p);
|
||||||
|
if (r[i][j][p-MINP] < 0)
|
||||||
|
r[i][j][p-MINP] = 0;
|
||||||
|
if (r[i][j][p-MINP] > max) {
|
||||||
|
max = r[i][j][p-MINP];
|
||||||
|
mi = i;
|
||||||
|
mp = p;
|
||||||
|
mj = j;
|
||||||
|
}
|
||||||
|
mean += r[i][j][p-MINP];
|
||||||
|
nitems++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
mean /= nitems;
|
||||||
|
|
||||||
|
cmax = 0;
|
||||||
|
cmi = 0;
|
||||||
|
cmp = 0;
|
||||||
|
cmj = 0;
|
||||||
|
|
||||||
|
minp = lastmp - 1;
|
||||||
|
if (minp < MINP)
|
||||||
|
minp = MINP;
|
||||||
|
maxp = lastmp + 1;
|
||||||
|
if (maxp >= MAXP)
|
||||||
|
maxp = MAXP-1;
|
||||||
|
|
||||||
|
for (i = 0; i < NOUTPUTS; i++) {
|
||||||
|
for (p = minp; p <= maxp; p++) {
|
||||||
|
for (j = lastmj; j <= lastmj+2; j++) {
|
||||||
|
int nj = j;
|
||||||
|
if (nj >= p)
|
||||||
|
nj -= p;
|
||||||
|
if (r[i][nj][p-MINP] > cmax) {
|
||||||
|
cmax = r[i][nj][p-MINP];
|
||||||
|
cmi = i;
|
||||||
|
cmj = nj;
|
||||||
|
cmp = p;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
count = 0;
|
||||||
|
variance = 0;
|
||||||
|
for (i = 0; i < NOUTPUTS; i++)
|
||||||
|
for (p = MINP; p < MAXP; p++)
|
||||||
|
for (j = 0; j < p; j++) {
|
||||||
|
double val;
|
||||||
|
if (r[i][j][p-MINP] > max / 1.2)
|
||||||
|
count++;
|
||||||
|
val = r[i][j][p-MINP] - mean;
|
||||||
|
variance += val * val;
|
||||||
|
}
|
||||||
|
|
||||||
|
variance /= nitems;
|
||||||
|
sd = sqrt(variance);
|
||||||
|
|
||||||
|
if (cmax > (max / 1.2)) {
|
||||||
|
max = cmax;
|
||||||
|
mi = cmi;
|
||||||
|
mj = cmj;
|
||||||
|
mp = cmp;
|
||||||
|
}
|
||||||
|
|
||||||
|
phase = (((double)mj) / ((double)mp));
|
||||||
|
confidence = sd;
|
||||||
|
beat = 0;
|
||||||
|
if (sd > /*200*/ 10)
|
||||||
|
if (mj < lastmj) {
|
||||||
|
beat = 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
lastmax = max;
|
||||||
|
lastmj = mj;
|
||||||
|
lastmi = mi;
|
||||||
|
lastmp = mp;
|
||||||
|
|
||||||
|
if (beat)
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
7
src/lsi/beatdetect.h
Normal file
7
src/lsi/beatdetect.h
Normal file
@@ -0,0 +1,7 @@
|
|||||||
|
/* beatdetect.h */
|
||||||
|
|
||||||
|
void beatdetect_init(void);
|
||||||
|
void beatdetect_close(void);
|
||||||
|
double beatdetect_getphase(void);
|
||||||
|
double beatdetect_getconfidence(void);
|
||||||
|
int beatdetect_read(void);
|
||||||
129
src/lsi/dmx.c
Normal file
129
src/lsi/dmx.c
Normal file
@@ -0,0 +1,129 @@
|
|||||||
|
/* dmx.c */
|
||||||
|
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
#include <fcntl.h>
|
||||||
|
#include <err.h>
|
||||||
|
#include <termios.h>
|
||||||
|
#include <strings.h>
|
||||||
|
#include <assert.h>
|
||||||
|
#include "dmx.h"
|
||||||
|
|
||||||
|
#define PORT "/dev/ttyU0"
|
||||||
|
#define DMX_PACKETSIZE (DMX_UNIVERSESIZE + 6)
|
||||||
|
|
||||||
|
int dmxfd;
|
||||||
|
|
||||||
|
char dmxpacket[DMX_UNIVERSESIZE + DMX_PACKETSIZE];
|
||||||
|
char *dmxuniverse;
|
||||||
|
|
||||||
|
void dmx_open(void)
|
||||||
|
{
|
||||||
|
struct termios t;
|
||||||
|
int flags;
|
||||||
|
|
||||||
|
dmxfd = open(PORT, O_NONBLOCK | O_RDWR, 0);
|
||||||
|
if (dmxfd == -1) {
|
||||||
|
err(1, "failed to open DMX port");
|
||||||
|
}
|
||||||
|
|
||||||
|
flags = fcntl(dmxfd, F_GETFL);
|
||||||
|
fcntl(dmxfd, F_SETFL, flags & ~O_NONBLOCK);
|
||||||
|
|
||||||
|
tcgetattr(dmxfd, &t);
|
||||||
|
cfmakeraw(&t);
|
||||||
|
t.c_cflag = CLOCAL | CREAD | CS8;
|
||||||
|
tcsetattr(dmxfd, TCSANOW, &t);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
void dmx_close(void)
|
||||||
|
{
|
||||||
|
close(dmxfd);
|
||||||
|
}
|
||||||
|
|
||||||
|
void dmx_dumpparams(void)
|
||||||
|
{
|
||||||
|
char buf[1024];
|
||||||
|
int more;
|
||||||
|
char *ptr;
|
||||||
|
int bytes;
|
||||||
|
int len;
|
||||||
|
|
||||||
|
buf[0] = 0x7e;
|
||||||
|
buf[1] = 3;
|
||||||
|
buf[2] = 2;
|
||||||
|
buf[3] = 0;
|
||||||
|
buf[4] = 0;
|
||||||
|
buf[5] = 0;
|
||||||
|
buf[6] = 0xe7;
|
||||||
|
if (write(dmxfd, buf, 7) != 7)
|
||||||
|
printf("didn't write 7 bytes\n");
|
||||||
|
|
||||||
|
more = 1;
|
||||||
|
ptr = buf;
|
||||||
|
bytes = 0;
|
||||||
|
len = 0;
|
||||||
|
while (more) {
|
||||||
|
if (read(dmxfd, ptr, 1) != 1)
|
||||||
|
goto out;
|
||||||
|
bytes++;
|
||||||
|
ptr++;
|
||||||
|
if (buf[0] != 0x7e) {
|
||||||
|
printf("Invalid packet received\n");
|
||||||
|
goto out;
|
||||||
|
}
|
||||||
|
if (bytes == 4)
|
||||||
|
len = buf[2] | (buf[3] << 8);
|
||||||
|
if (bytes == len + 5) {
|
||||||
|
if (buf[bytes-1] != (char)0xe7) {
|
||||||
|
printf("Invalid packet end\n");
|
||||||
|
goto out;
|
||||||
|
}
|
||||||
|
more = 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
printf("Received packet:\n");
|
||||||
|
printf("%x %x %x %x %x %x %x %x %x %x %x %x\n", buf[0], buf[1], buf[2], buf[3], buf[4], buf[5], buf[6], buf[7], buf[8], buf[9], buf[10], buf[11]);
|
||||||
|
|
||||||
|
out:
|
||||||
|
printf("buffer contents:\n");
|
||||||
|
printf("%x %x %x %x %x %x %x %x %x %x %x %x\n", buf[0], buf[1], buf[2], buf[3], buf[4], buf[5], buf[6], buf[7], buf[8], buf[9], buf[10], buf[11]);
|
||||||
|
}
|
||||||
|
|
||||||
|
void dmx_init(void)
|
||||||
|
{
|
||||||
|
dmx_open();
|
||||||
|
dmxpacket[0] = 0x7e;
|
||||||
|
dmxpacket[1] = 6;
|
||||||
|
dmxpacket[2] = (DMX_UNIVERSESIZE+1) & 0xff;
|
||||||
|
dmxpacket[3] = ((DMX_UNIVERSESIZE+1) >> 8) & 0xff;
|
||||||
|
dmxpacket[4] = 0; /* start code */
|
||||||
|
bzero(dmxpacket+5, DMX_UNIVERSESIZE);
|
||||||
|
dmxpacket[DMX_UNIVERSESIZE+5] = 0xe7;
|
||||||
|
dmxuniverse = dmxpacket+5;
|
||||||
|
// dmx_dumpparams();
|
||||||
|
}
|
||||||
|
|
||||||
|
void dmx_setchannel(int channel, int value)
|
||||||
|
{
|
||||||
|
assert(channel < DMX_UNIVERSESIZE);
|
||||||
|
dmxuniverse[channel] = (char)value;
|
||||||
|
}
|
||||||
|
|
||||||
|
void dmx_output(void)
|
||||||
|
{
|
||||||
|
int off = 0;
|
||||||
|
// char *buf = dmxpacket;
|
||||||
|
// printf("%x %x %x %x %x %x %x %x %x %x %x %x\n", buf[0], buf[1], buf[2], buf[3], buf[4], buf[5], buf[6], buf[7], buf[8], buf[9], buf[10], buf[11]);
|
||||||
|
while (off < DMX_PACKETSIZE) {
|
||||||
|
int r;
|
||||||
|
r = write(dmxfd, dmxpacket + off, DMX_PACKETSIZE - off);
|
||||||
|
// printf("write returned %d\n", r);
|
||||||
|
if (r == -1)
|
||||||
|
err(1, "error writing packet");
|
||||||
|
off += r;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
9
src/lsi/dmx.h
Normal file
9
src/lsi/dmx.h
Normal file
@@ -0,0 +1,9 @@
|
|||||||
|
/* dmx.h */
|
||||||
|
|
||||||
|
#define DMX_UNIVERSESIZE 512
|
||||||
|
|
||||||
|
void dmx_close(void);
|
||||||
|
void dmx_init(void);
|
||||||
|
void dmx_setchannel(int channel, int value);
|
||||||
|
void dmx_output(void);
|
||||||
|
|
||||||
138
src/lsi/fft.c
Normal file
138
src/lsi/fft.c
Normal file
@@ -0,0 +1,138 @@
|
|||||||
|
/* fft.c */
|
||||||
|
|
||||||
|
#include <math.h>
|
||||||
|
#include <stdint.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <assert.h>
|
||||||
|
#include "mem.h"
|
||||||
|
#include "fft.h"
|
||||||
|
|
||||||
|
#define FFT_SIGN (-1)
|
||||||
|
|
||||||
|
int fft_size = 0;
|
||||||
|
fft_type *fft_buffer = NULL;
|
||||||
|
long fft_logN;
|
||||||
|
|
||||||
|
void fft_init(int size)
|
||||||
|
{
|
||||||
|
assert(fft_size == 0);
|
||||||
|
assert(fft_buffer == NULL);
|
||||||
|
|
||||||
|
/* We use complex pairs, so we need twice the number of elements */
|
||||||
|
fft_buffer = safe_malloc(sizeof(fft_type) * size * 2);
|
||||||
|
fft_size = size;
|
||||||
|
fft_logN = (long)(log(fft_size)/log(2.0)+0.5);
|
||||||
|
}
|
||||||
|
|
||||||
|
void fft_data_float(float *buf)
|
||||||
|
{
|
||||||
|
int i;
|
||||||
|
|
||||||
|
for (i = 0; i < fft_size; i++) {
|
||||||
|
fft_buffer[i*2] = buf[i];
|
||||||
|
fft_buffer[i*2+1] = 0.0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void fft_data_signed16(int16_t *buf)
|
||||||
|
{
|
||||||
|
int i;
|
||||||
|
|
||||||
|
for (i = 0; i < fft_size; i++) {
|
||||||
|
fft_buffer[i*2] = (fft_type)buf[i] / (INT16_MAX+1);
|
||||||
|
fft_buffer[i*2+1] = 0.0;
|
||||||
|
// printf("input: %f, %f\n", fft_buffer[i*2], fft_buffer[i*2+1]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void fft_data_unsigned16(uint16_t *buf)
|
||||||
|
{
|
||||||
|
int i;
|
||||||
|
|
||||||
|
for (i = 0; i < fft_size; i++) {
|
||||||
|
fft_buffer[i*2] = (fft_type)buf[i] / (INT16_MAX+1) -
|
||||||
|
(INT16_MAX+1);
|
||||||
|
fft_buffer[i*2+1] = 0.0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fft_type *fft_getresult(void)
|
||||||
|
{
|
||||||
|
return fft_buffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
void fft_window(void)
|
||||||
|
{
|
||||||
|
int i;
|
||||||
|
|
||||||
|
for (i = 0; i < fft_size; i++) {
|
||||||
|
fft_buffer[2*i] = fft_buffer[2*i] * (0.5 + 0.5 *
|
||||||
|
cos(M_PI * (i - fft_size/2)/(fft_size/2 + 1)));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* FFT routine
|
||||||
|
* Based on routine (C) 1996 S.M.Bernsee.
|
||||||
|
*/
|
||||||
|
void fft_compute(void)
|
||||||
|
{
|
||||||
|
fft_type wr, wi, arg, *p1, *p2, temp;
|
||||||
|
fft_type tr, ti, ur, ui, *p1r, *p1i, *p2r, *p2i;
|
||||||
|
long i, bitm, j, le, le2, k;
|
||||||
|
|
||||||
|
for (i = 2; i < 2*fft_size-2; i += 2) {
|
||||||
|
|
||||||
|
for (bitm = 2, j = 0; bitm < 2*fft_size; bitm <<= 1) {
|
||||||
|
|
||||||
|
if (i & bitm) j++;
|
||||||
|
j <<= 1;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
if (i < j) {
|
||||||
|
|
||||||
|
p1 = fft_buffer+i; p2 = fft_buffer+j;
|
||||||
|
temp = *p1; *(p1++) = *p2;
|
||||||
|
*(p2++) = temp; temp = *p1;
|
||||||
|
*p1 = *p2; *p2 = temp;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
for (k = 0, le = 2; k < fft_logN; k++) {
|
||||||
|
|
||||||
|
le <<= 1;
|
||||||
|
le2 = le>>1;
|
||||||
|
ur = 1.0;
|
||||||
|
ui = 0.0;
|
||||||
|
arg = M_PI / (le2>>1);
|
||||||
|
wr = cos(arg);
|
||||||
|
wi = FFT_SIGN*sin(arg);
|
||||||
|
|
||||||
|
for (j = 0; j < le2; j += 2) {
|
||||||
|
|
||||||
|
p1r = fft_buffer+j; p1i = p1r+1;
|
||||||
|
p2r = p1r+le2; p2i = p2r+1;
|
||||||
|
|
||||||
|
for (i = j; i < 2*fft_size; i += le) {
|
||||||
|
|
||||||
|
tr = *p2r * ur - *p2i * ui;
|
||||||
|
ti = *p2r * ui + *p2i * ur;
|
||||||
|
*p2r = *p1r - tr; *p2i = *p1i - ti;
|
||||||
|
*p1r += tr; *p1i += ti;
|
||||||
|
p1r += le; p1i += le;
|
||||||
|
p2r += le; p2i += le;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
tr = ur*wr - ui*wi;
|
||||||
|
ui = ur*wi + ui*wr;
|
||||||
|
ur = tr;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
12
src/lsi/fft.h
Normal file
12
src/lsi/fft.h
Normal file
@@ -0,0 +1,12 @@
|
|||||||
|
/* fft.h */
|
||||||
|
|
||||||
|
typedef float fft_type;
|
||||||
|
|
||||||
|
/* Initialise fft with the size of the data */
|
||||||
|
void fft_init(int);
|
||||||
|
void fft_data_float(float *);
|
||||||
|
void fft_data_signed16(int16_t *);
|
||||||
|
void fft_data_unsigned16(uint16_t *);
|
||||||
|
fft_type *fft_getresult(void);
|
||||||
|
void fft_window(void);
|
||||||
|
void fft_compute(void);
|
||||||
42
src/lsi/main.c
Normal file
42
src/lsi/main.c
Normal file
@@ -0,0 +1,42 @@
|
|||||||
|
/* main.c */
|
||||||
|
|
||||||
|
#include <signal.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <err.h>
|
||||||
|
#include "vm.h"
|
||||||
|
#include "dmx.h"
|
||||||
|
#include "midi.h"
|
||||||
|
#include "beatdetect.h"
|
||||||
|
#include "mouse.h"
|
||||||
|
|
||||||
|
void finish(void)
|
||||||
|
{
|
||||||
|
dmx_close();
|
||||||
|
midi_close();
|
||||||
|
beatdetect_close();
|
||||||
|
exit(0);
|
||||||
|
}
|
||||||
|
|
||||||
|
void sigint_handler(int signal)
|
||||||
|
{
|
||||||
|
finish();
|
||||||
|
}
|
||||||
|
|
||||||
|
int main(int argc, char *argv[])
|
||||||
|
{
|
||||||
|
argv++;
|
||||||
|
argc--;
|
||||||
|
if (argc != 1)
|
||||||
|
errx(1, "Usage: lsi <filename>");
|
||||||
|
vm_init();
|
||||||
|
vm_load(argv[0]);
|
||||||
|
signal(SIGINT, sigint_handler);
|
||||||
|
midi_init();
|
||||||
|
dmx_init();
|
||||||
|
beatdetect_init();
|
||||||
|
mouse_init();
|
||||||
|
vm_spawn("main");
|
||||||
|
vm_run();
|
||||||
|
finish();
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
645
src/lsi/map3d.c
Normal file
645
src/lsi/map3d.c
Normal file
@@ -0,0 +1,645 @@
|
|||||||
|
/* map3d.c */
|
||||||
|
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <math.h>
|
||||||
|
#include <strings.h>
|
||||||
|
#include <fcntl.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
|
||||||
|
#include "vm.h"
|
||||||
|
|
||||||
|
#define NLIGHTS 16
|
||||||
|
|
||||||
|
#define PANMAX 127
|
||||||
|
#define PANRANGE M_PI_2
|
||||||
|
#define PANOFFSET 128
|
||||||
|
#define TILTMAX 127
|
||||||
|
#define TILTRANGE M_PI_4
|
||||||
|
#define TILTOFFSET 128
|
||||||
|
|
||||||
|
#define MAP3D_FILENAME ".map3d.caldata"
|
||||||
|
|
||||||
|
struct light {
|
||||||
|
double M[4][3];
|
||||||
|
double cp[3][3];
|
||||||
|
double pan[3];
|
||||||
|
double tilt[3];
|
||||||
|
};
|
||||||
|
|
||||||
|
struct light map3d_cal[NLIGHTS];
|
||||||
|
|
||||||
|
#define MAG(x) (sqrt(SQUARE(x[0]) + SQUARE(x[1]) + SQUARE(x[2])))
|
||||||
|
#define SQUARE(x) ((x) * (x))
|
||||||
|
#define PYTHAG3(a, b) sqrt(SQUARE(a[0]-b[0]) + SQUARE(a[1]-b[1]) + SQUARE(a[2]-b[2]))
|
||||||
|
|
||||||
|
#define DEG(x) (180 * (x) / M_PI)
|
||||||
|
|
||||||
|
void normalise(double *v)
|
||||||
|
{
|
||||||
|
double mag;
|
||||||
|
int i;
|
||||||
|
|
||||||
|
mag = MAG(v);
|
||||||
|
|
||||||
|
for (i = 0; i < 3; i++)
|
||||||
|
v[i] /= mag;
|
||||||
|
}
|
||||||
|
|
||||||
|
#define MM map3d_cal[light].M
|
||||||
|
|
||||||
|
void multiply(int light, double *i, double *o) {
|
||||||
|
o[0] = i[0] * MM[0][0] + i[1] * MM[1][0] + i[2] * MM[2][0] + MM[3][0];
|
||||||
|
o[1] = i[0] * MM[0][1] + i[1] * MM[1][1] + i[2] * MM[2][1] + MM[3][1];
|
||||||
|
o[2] = i[0] * MM[0][2] + i[1] * MM[1][2] + i[2] * MM[2][2] + MM[3][2];
|
||||||
|
}
|
||||||
|
|
||||||
|
#undef MM
|
||||||
|
|
||||||
|
void map3d_init(void)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
void map3d_close(void)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
void map3d_save(void)
|
||||||
|
{
|
||||||
|
int fd, rv;
|
||||||
|
|
||||||
|
fd = open(MAP3D_FILENAME, O_WRONLY | O_CREAT, 0666);
|
||||||
|
rv = write(fd, map3d_cal, sizeof(map3d_cal));
|
||||||
|
if (rv != sizeof(map3d_cal))
|
||||||
|
printf("Warning: Calibration data not saved correctly\n");
|
||||||
|
close(fd);
|
||||||
|
}
|
||||||
|
|
||||||
|
int map3d_load(void)
|
||||||
|
{
|
||||||
|
int fd, rv;
|
||||||
|
|
||||||
|
fd = open(MAP3D_FILENAME, O_RDONLY, 0666);
|
||||||
|
if (!fd)
|
||||||
|
return 0;
|
||||||
|
rv = read(fd, map3d_cal, sizeof(map3d_cal));
|
||||||
|
if (rv != sizeof(map3d_cal))
|
||||||
|
printf("Warning: Calibration data not read correctly\n");
|
||||||
|
close(fd);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
void map3d_transform(int light, double x, double y, double z,
|
||||||
|
int *pan, int *tilt)
|
||||||
|
{
|
||||||
|
double pv[3];
|
||||||
|
double rv[3];
|
||||||
|
double p, t;
|
||||||
|
|
||||||
|
// printf("Transforming for light %d: (%f, %f, %f)\n", light, x, y, z);
|
||||||
|
fflush(stdout);
|
||||||
|
pv[0] = x;
|
||||||
|
pv[1] = y;
|
||||||
|
pv[2] = z;
|
||||||
|
multiply(light, pv, rv);
|
||||||
|
normalise(rv);
|
||||||
|
t = asin(rv[1]);
|
||||||
|
p = asin(rv[0]/cos(t));
|
||||||
|
|
||||||
|
*pan = (int)round((p * PANMAX)/PANRANGE) + PANOFFSET;
|
||||||
|
*tilt = (int)round((t * TILTMAX)/TILTRANGE) + TILTOFFSET;
|
||||||
|
if (*pan < 0)
|
||||||
|
*pan = 0;
|
||||||
|
if (*pan > 255)
|
||||||
|
*pan = 255;
|
||||||
|
if (*tilt < 0)
|
||||||
|
*tilt = 0;
|
||||||
|
if (*tilt > 255)
|
||||||
|
*tilt = 255;
|
||||||
|
// printf("pan = %d, tilt = %d\n", *pan, *tilt);
|
||||||
|
}
|
||||||
|
|
||||||
|
void map3d_setcal(int light, int n, double x, double y, double z,
|
||||||
|
int pan, int tilt)
|
||||||
|
{
|
||||||
|
printf("setcal(%d, %d, %f, %f, %f, %d, %d)\n", light, n, x, y, z, pan, tilt);
|
||||||
|
map3d_cal[light].cp[n][0] = x;
|
||||||
|
map3d_cal[light].cp[n][1] = y;
|
||||||
|
map3d_cal[light].cp[n][2] = z;
|
||||||
|
map3d_cal[light].pan[n] = PANRANGE * ((double)pan - PANOFFSET) / PANMAX;
|
||||||
|
map3d_cal[light].tilt[n] = TILTRANGE * ((double)tilt - TILTOFFSET)
|
||||||
|
/ TILTMAX;
|
||||||
|
printf("Setcal: pan = %f, tilt = %f\n", map3d_cal[light].pan[n], map3d_cal[light].tilt[n]);
|
||||||
|
}
|
||||||
|
|
||||||
|
void unitvector(double *vec, double pan, double tilt)
|
||||||
|
{
|
||||||
|
double tmp[3];
|
||||||
|
double cosp, sinp, cost, sint;
|
||||||
|
|
||||||
|
/* Precalculate some values */
|
||||||
|
cosp = cos(pan);
|
||||||
|
sinp = sin(pan);
|
||||||
|
cost = cos(tilt);
|
||||||
|
sint = sin(tilt);
|
||||||
|
|
||||||
|
/* Start with a unit vector */
|
||||||
|
vec[0] = 0;
|
||||||
|
vec[1] = 0;
|
||||||
|
vec[2] = 1;
|
||||||
|
|
||||||
|
/* Rotate around X axis (tilt) */
|
||||||
|
tmp[0] = vec[0];
|
||||||
|
tmp[1] = vec[1]*cost + vec[2]*sint;
|
||||||
|
tmp[2] = vec[2]*cost - vec[1]*sint;
|
||||||
|
|
||||||
|
/* Rotate around Y axis (pan) */
|
||||||
|
vec[0] = tmp[0]*cosp + tmp[2]*sinp;
|
||||||
|
vec[1] = tmp[1];
|
||||||
|
vec[2] = tmp[2]*cosp - tmp[0]*sinp;
|
||||||
|
}
|
||||||
|
|
||||||
|
double dotproduct(double *a, double *b)
|
||||||
|
{
|
||||||
|
return a[0]*b[0] + a[1]*b[1] + a[2]*b[2];
|
||||||
|
}
|
||||||
|
|
||||||
|
void crossproduct(double *a, double *b, double *r)
|
||||||
|
{
|
||||||
|
r[0] = a[1]*b[2] - a[2]*b[1];
|
||||||
|
r[1] = a[2]*b[0] - a[0]*b[2];
|
||||||
|
r[2] = a[0]*b[1] - a[1]*b[0];
|
||||||
|
}
|
||||||
|
|
||||||
|
void dumpmatrix(double sim[12][13]) {
|
||||||
|
int i, j;
|
||||||
|
|
||||||
|
for (i = 0; i < 12; i++) {
|
||||||
|
for (j = 0; j < 13; j++)
|
||||||
|
printf("\t%f", sim[i][j]);
|
||||||
|
printf("\n");
|
||||||
|
}
|
||||||
|
printf("\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
void eliminate(double sim[12][13], double *var)
|
||||||
|
{
|
||||||
|
int i, j, k, maxpos;
|
||||||
|
double max, val, tmp, x;
|
||||||
|
|
||||||
|
dumpmatrix(sim);
|
||||||
|
|
||||||
|
for (i = 0; i < 12; i++) {
|
||||||
|
/* Find the maximum element in the ith column */
|
||||||
|
max = 0.0;
|
||||||
|
maxpos = i;
|
||||||
|
for (j = i; j < 12; j++) {
|
||||||
|
val = fabs(sim[j][i]);
|
||||||
|
if (val > max) {
|
||||||
|
max = val;
|
||||||
|
maxpos = j;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (maxpos != i)
|
||||||
|
for (j = 0; j < 13; j++) {
|
||||||
|
tmp = sim[maxpos][j];
|
||||||
|
sim[maxpos][j] = sim[i][j];
|
||||||
|
sim[i][j] = tmp;
|
||||||
|
}
|
||||||
|
if (fabs(sim[i][i]) < 0.0001)
|
||||||
|
printf("Warning: Lost accuracy at row %d\n", i);
|
||||||
|
for (j = i+1; j < 12; j++) {
|
||||||
|
if (fabs(sim[j][i]) > 0.0) {
|
||||||
|
/*
|
||||||
|
* Subtract x times row i from row j,
|
||||||
|
* where x is chosen such that
|
||||||
|
* sim[i][i] * x = sim[j][i]
|
||||||
|
*/
|
||||||
|
x = sim[j][i] / sim[i][i];
|
||||||
|
for (k = 0; k < 13; k++)
|
||||||
|
sim[j][k] -= x*sim[i][k];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
dumpmatrix(sim);
|
||||||
|
|
||||||
|
/* Next, substitute in the unknowns. */
|
||||||
|
for (i = 12-1; i >= 0; i--) {
|
||||||
|
var[i] = sim[i][12];
|
||||||
|
for (j = i+1; j < 12; j++)
|
||||||
|
var[i] -= var[j] * sim[i][j];
|
||||||
|
var[i] /= sim[i][i];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/* In: Guess, length, angle, space for solutions */
|
||||||
|
void solvetriangle(double c, double a, double A, double *solp, double *soln)
|
||||||
|
{
|
||||||
|
double root;
|
||||||
|
double cosA;
|
||||||
|
double med;
|
||||||
|
|
||||||
|
cosA = cos(A);
|
||||||
|
med = 4*c*c*cosA*cosA - 4*(c*c-a*a);
|
||||||
|
root = sqrt(med);
|
||||||
|
|
||||||
|
if (isnan(root))
|
||||||
|
/* sqrt() of a negative number */
|
||||||
|
if (med > -0.000000001)
|
||||||
|
root = 0;
|
||||||
|
|
||||||
|
*solp = (2*c*cosA + root)/2;
|
||||||
|
*soln = (2*c*cosA - root)/2;
|
||||||
|
}
|
||||||
|
|
||||||
|
double try(int n, double dguess, double a, double b, double c, double A, double B, double C)
|
||||||
|
{
|
||||||
|
double s1, s2;
|
||||||
|
double e, f, d;
|
||||||
|
|
||||||
|
solvetriangle(dguess, a, A, &s1, &s2);
|
||||||
|
e = (n>=4?s2:s1);
|
||||||
|
n = n % 4;
|
||||||
|
solvetriangle(e, c, C, &s1, &s2);
|
||||||
|
f = (n>=2?s2:s1);
|
||||||
|
n = n % 2;
|
||||||
|
solvetriangle(f, b, B, &s1, &s2);
|
||||||
|
d = (n>=1?s2:s1);
|
||||||
|
|
||||||
|
return d;
|
||||||
|
}
|
||||||
|
|
||||||
|
void getsolution(int n, double dguess, double a, double b, double c, double A, double B, double C, double *e, double *f)
|
||||||
|
{
|
||||||
|
double s1, s2;
|
||||||
|
|
||||||
|
solvetriangle(dguess, a, A, &s1, &s2);
|
||||||
|
*e = (n>=4?s2:s1);
|
||||||
|
n = n % 4;
|
||||||
|
solvetriangle(*e, c, C, &s1, &s2);
|
||||||
|
*f = (n>=2?s2:s1);
|
||||||
|
}
|
||||||
|
|
||||||
|
#define DINCR 0.005
|
||||||
|
|
||||||
|
/*
|
||||||
|
* A is angle opposite a, between d and e
|
||||||
|
* B is angle opposite b, between d and f
|
||||||
|
* C is angle opposite c, between e and f
|
||||||
|
*/
|
||||||
|
int trysolvetetra(int n, double a, double b, double c,
|
||||||
|
double A, double B, double C,
|
||||||
|
double *dr, double *er, double *fr)
|
||||||
|
{
|
||||||
|
double dguess;
|
||||||
|
double thresh = 0.000000000001;
|
||||||
|
double d, lastd;
|
||||||
|
double mind, maxd;
|
||||||
|
double d1, d2, d3;
|
||||||
|
double e, f;
|
||||||
|
int found;
|
||||||
|
|
||||||
|
lastd = 0.0/0.0; /* NaN */
|
||||||
|
mind = maxd = 0; /* Shut up compiler */
|
||||||
|
|
||||||
|
/* Find a bracket */
|
||||||
|
found = 0;
|
||||||
|
for (dguess = 0.0; dguess < 100.0; dguess+=DINCR) {
|
||||||
|
d = try(n, dguess, a, b, c, A, B, C) - dguess;
|
||||||
|
if (((d >= 0) && (lastd < 0)) || ((d <= 0) && (lastd > 0))) {
|
||||||
|
mind = dguess - DINCR;
|
||||||
|
maxd = dguess;
|
||||||
|
printf("Found bracket: (%f, %f)\n", mind, maxd);
|
||||||
|
found = 1;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
lastd = d;
|
||||||
|
}
|
||||||
|
if (!found)
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
while ((maxd-mind) > thresh) {
|
||||||
|
d1 = try(n, mind, a, b, c, A, B, C) - mind;
|
||||||
|
d2 = try(n, (mind+maxd)/2, a, b, c, A, B, C) - (mind+maxd)/2;
|
||||||
|
d3 = try(n, maxd, a, b, c, A, B, C) - maxd;
|
||||||
|
if (((d2 >= 0) && (d1 < 0)) || ((d2 < 0) && (d1 >= 0)))
|
||||||
|
maxd = (mind+maxd)/2;
|
||||||
|
else
|
||||||
|
mind = (mind+maxd)/2;
|
||||||
|
}
|
||||||
|
|
||||||
|
d = (mind+maxd)/2;
|
||||||
|
|
||||||
|
getsolution(n, d, a, b, c, A, B, C, &e, &f);
|
||||||
|
|
||||||
|
printf("Found solution: (%f, %f, %f)\n", d, e, f);
|
||||||
|
|
||||||
|
*dr = d;
|
||||||
|
*er = e;
|
||||||
|
*fr = f;
|
||||||
|
|
||||||
|
if (!isfinite(d))
|
||||||
|
return 0;
|
||||||
|
if (!isfinite(e))
|
||||||
|
return 0;
|
||||||
|
if (!isfinite(f))
|
||||||
|
return 0;
|
||||||
|
if (d < 0)
|
||||||
|
return 0;
|
||||||
|
if (e < 0)
|
||||||
|
return 0;
|
||||||
|
if (f < 0)
|
||||||
|
return 0;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
int solvetetra(double a, double b, double c,
|
||||||
|
double A, double B, double C,
|
||||||
|
double *dr, double *er, double *fr)
|
||||||
|
{
|
||||||
|
int i;
|
||||||
|
|
||||||
|
for (i = 0; i < 8; i++)
|
||||||
|
if (trysolvetetra(i, a, b, c, A, B, C, dr, er, fr))
|
||||||
|
return 1;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int map3d_calibrate(int light)
|
||||||
|
{
|
||||||
|
double v1[3], v2[3], v3[3], p4[3];
|
||||||
|
double angle1, angle2, angle3;
|
||||||
|
double a, b, c, d, e, f;
|
||||||
|
double pAB[3], pBC[3];
|
||||||
|
double axis1[3], axis2[3], axis3[3];
|
||||||
|
double axm1, axm2, axm3;
|
||||||
|
double hypot, alpha, beta, gamma, delta;
|
||||||
|
double sim[12][13];
|
||||||
|
double var[12];
|
||||||
|
double pv[3], tv[3];
|
||||||
|
int success, i;
|
||||||
|
|
||||||
|
/*
|
||||||
|
* First, create unit vectors in the directions of the
|
||||||
|
* pan & tilt values given
|
||||||
|
*/
|
||||||
|
unitvector(v1, map3d_cal[light].pan[0], map3d_cal[light].tilt[0]);
|
||||||
|
unitvector(v2, map3d_cal[light].pan[1], map3d_cal[light].tilt[1]);
|
||||||
|
unitvector(v3, map3d_cal[light].pan[2], map3d_cal[light].tilt[2]);
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Now, we need the angles between them
|
||||||
|
*/
|
||||||
|
angle1 = acos(dotproduct(v1, v2));
|
||||||
|
angle2 = acos(dotproduct(v2, v3));
|
||||||
|
angle3 = acos(dotproduct(v3, v1));
|
||||||
|
|
||||||
|
printf("angles (%f, %f, %f)\n", DEG(angle1), DEG(angle2), DEG(angle3));
|
||||||
|
|
||||||
|
/*
|
||||||
|
* And the lengths of the edges which we know
|
||||||
|
*/
|
||||||
|
a = PYTHAG3(map3d_cal[light].cp[0], map3d_cal[light].cp[1]);
|
||||||
|
b = PYTHAG3(map3d_cal[light].cp[1], map3d_cal[light].cp[2]);
|
||||||
|
c = PYTHAG3(map3d_cal[light].cp[2], map3d_cal[light].cp[0]);
|
||||||
|
|
||||||
|
/* Solve the tetrahedron */
|
||||||
|
success = solvetetra(a, b, c, angle1, angle2, angle3, &d, &e, &f);
|
||||||
|
if (!success) {
|
||||||
|
printf("failed to solve tetrahedron\n");
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Multiply the vectors by their magnitudes */
|
||||||
|
for (i = 0; i < 3; i++) {
|
||||||
|
v1[i] *= e;
|
||||||
|
v2[i] *= d;
|
||||||
|
v3[i] *= f;
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Find two vectors to define the triangle between
|
||||||
|
* the calibration points
|
||||||
|
*/
|
||||||
|
for (i = 0; i < 3; i++) {
|
||||||
|
pAB[i] = map3d_cal[light].cp[1][i]-map3d_cal[light].cp[0][i];
|
||||||
|
pBC[i] = map3d_cal[light].cp[2][i]-map3d_cal[light].cp[1][i];
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Create some perpendicular vectors in terms of which we can
|
||||||
|
* calculate a vector to the fourth point
|
||||||
|
*/
|
||||||
|
axis1[0] = pAB[0];
|
||||||
|
axis1[1] = pAB[1];
|
||||||
|
axis1[2] = pAB[2];
|
||||||
|
normalise(axis1);
|
||||||
|
crossproduct(axis1, pBC, axis2);
|
||||||
|
normalise(axis2);
|
||||||
|
crossproduct(axis1, axis2, axis3);
|
||||||
|
normalise(axis3);
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Now we do some trigonometry to find out the distance to
|
||||||
|
* the fourth point in terms of the three axes
|
||||||
|
*/
|
||||||
|
beta = asin(d*sin(angle1)/a);
|
||||||
|
gamma = asin(f*sin(angle3)/c);
|
||||||
|
delta = acos(((a*a)+(c*c)-(b*b))/(2*a*c));
|
||||||
|
|
||||||
|
alpha = acos((cos(gamma)-cos(beta)*cos(delta))/(sin(beta)*sin(delta)));
|
||||||
|
|
||||||
|
hypot = e*sin(beta);
|
||||||
|
axm1 = e*cos(beta);
|
||||||
|
axm2 = hypot*sin(alpha);
|
||||||
|
axm3 = hypot*cos(alpha);
|
||||||
|
|
||||||
|
/* Now we have the magnitudes, let's get the vectors */
|
||||||
|
for (i = 0; i < 3; i++) {
|
||||||
|
axis1[i] *= axm1;
|
||||||
|
axis2[i] *= axm2;
|
||||||
|
axis3[i] *= axm3;
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Now we can simply add these vectors to point A
|
||||||
|
* to get the fourth point
|
||||||
|
*/
|
||||||
|
for (i = 0; i < 3; i++)
|
||||||
|
p4[i] = map3d_cal[light].cp[0][i] + axis1[i] +
|
||||||
|
axis2[i] + axis3[i];
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Now we can construct a matrix to represent 12 simultaneous
|
||||||
|
* equations, which we then solve to get our transformation matrix.
|
||||||
|
*/
|
||||||
|
bzero(sim, sizeof(sim));
|
||||||
|
|
||||||
|
/* First point */
|
||||||
|
sim[0][0] = map3d_cal[light].cp[0][0];
|
||||||
|
sim[0][1] = map3d_cal[light].cp[0][1];
|
||||||
|
sim[0][2] = map3d_cal[light].cp[0][2];
|
||||||
|
sim[0][9] = 1;
|
||||||
|
sim[0][12] = v1[0];
|
||||||
|
|
||||||
|
sim[1][3] = map3d_cal[light].cp[0][0];
|
||||||
|
sim[1][4] = map3d_cal[light].cp[0][1];
|
||||||
|
sim[1][5] = map3d_cal[light].cp[0][2];
|
||||||
|
sim[1][10] = 1;
|
||||||
|
sim[1][12] = v1[1];
|
||||||
|
|
||||||
|
sim[2][6] = map3d_cal[light].cp[0][0];
|
||||||
|
sim[2][7] = map3d_cal[light].cp[0][1];
|
||||||
|
sim[2][8] = map3d_cal[light].cp[0][2];
|
||||||
|
sim[2][11] = 1;
|
||||||
|
sim[2][12] = v1[2];
|
||||||
|
|
||||||
|
/* Second point */
|
||||||
|
sim[3][0] = map3d_cal[light].cp[1][0];
|
||||||
|
sim[3][1] = map3d_cal[light].cp[1][1];
|
||||||
|
sim[3][2] = map3d_cal[light].cp[1][2];
|
||||||
|
sim[3][9] = 1;
|
||||||
|
sim[3][12] = v2[0];
|
||||||
|
|
||||||
|
sim[4][3] = map3d_cal[light].cp[1][0];
|
||||||
|
sim[4][4] = map3d_cal[light].cp[1][1];
|
||||||
|
sim[4][5] = map3d_cal[light].cp[1][2];
|
||||||
|
sim[4][10] = 1;
|
||||||
|
sim[4][12] = v2[1];
|
||||||
|
|
||||||
|
sim[5][6] = map3d_cal[light].cp[1][0];
|
||||||
|
sim[5][7] = map3d_cal[light].cp[1][1];
|
||||||
|
sim[5][8] = map3d_cal[light].cp[1][2];
|
||||||
|
sim[5][11] = 1;
|
||||||
|
sim[5][12] = v2[2];
|
||||||
|
|
||||||
|
/* Third point */
|
||||||
|
sim[6][0] = map3d_cal[light].cp[2][0];
|
||||||
|
sim[6][1] = map3d_cal[light].cp[2][1];
|
||||||
|
sim[6][2] = map3d_cal[light].cp[2][2];
|
||||||
|
sim[6][9] = 1;
|
||||||
|
sim[6][12] = v3[0];
|
||||||
|
|
||||||
|
sim[7][3] = map3d_cal[light].cp[2][0];
|
||||||
|
sim[7][4] = map3d_cal[light].cp[2][1];
|
||||||
|
sim[7][5] = map3d_cal[light].cp[2][2];
|
||||||
|
sim[7][10] = 1;
|
||||||
|
sim[7][12] = v3[1];
|
||||||
|
|
||||||
|
sim[8][6] = map3d_cal[light].cp[2][0];
|
||||||
|
sim[8][7] = map3d_cal[light].cp[2][1];
|
||||||
|
sim[8][8] = map3d_cal[light].cp[2][2];
|
||||||
|
sim[8][11] = 1;
|
||||||
|
sim[8][12] = v3[2];
|
||||||
|
|
||||||
|
/* Fourth point */
|
||||||
|
sim[9][0] = p4[0];
|
||||||
|
sim[9][1] = p4[1];
|
||||||
|
sim[9][2] = p4[2];
|
||||||
|
sim[9][9] = 1;
|
||||||
|
sim[9][12] = 0;
|
||||||
|
|
||||||
|
sim[10][3] = p4[0];
|
||||||
|
sim[10][4] = p4[1];
|
||||||
|
sim[10][5] = p4[2];
|
||||||
|
sim[10][10] = 1;
|
||||||
|
sim[10][12] = 0;
|
||||||
|
|
||||||
|
sim[11][6] = p4[0];
|
||||||
|
sim[11][7] = p4[1];
|
||||||
|
sim[11][8] = p4[2];
|
||||||
|
sim[11][11] = 1;
|
||||||
|
sim[11][12] = 0;
|
||||||
|
|
||||||
|
eliminate(sim, var);
|
||||||
|
|
||||||
|
map3d_cal[light].M[0][0] = var[0];
|
||||||
|
map3d_cal[light].M[1][0] = var[1];
|
||||||
|
map3d_cal[light].M[2][0] = var[2];
|
||||||
|
map3d_cal[light].M[0][1] = var[3];
|
||||||
|
map3d_cal[light].M[1][1] = var[4];
|
||||||
|
map3d_cal[light].M[2][1] = var[5];
|
||||||
|
map3d_cal[light].M[0][2] = var[6];
|
||||||
|
map3d_cal[light].M[1][2] = var[7];
|
||||||
|
map3d_cal[light].M[2][2] = var[8];
|
||||||
|
map3d_cal[light].M[3][0] = var[9];
|
||||||
|
map3d_cal[light].M[3][1] = var[10];
|
||||||
|
map3d_cal[light].M[3][2] = var[11];
|
||||||
|
|
||||||
|
printf("%f\t%f\t%f\t\t%f\n", var[0], var[3], var[6], var[9]);
|
||||||
|
printf("%f\t%f\t%f\t\t%f\n", var[1], var[4], var[7], var[10]);
|
||||||
|
printf("%f\t%f\t%f\t\t%f\n", var[2], var[5], var[8], var[11]);
|
||||||
|
|
||||||
|
printf("Calibration points are:\n");
|
||||||
|
multiply(light, map3d_cal[light].cp[0], tv);
|
||||||
|
printf("(%f, %f, %f) => (%f, %f, %f)\n", map3d_cal[light].cp[0][0], map3d_cal[light].cp[0][1], map3d_cal[light].cp[0][2], tv[0], tv[1], tv[2]);
|
||||||
|
multiply(light, map3d_cal[light].cp[1], tv);
|
||||||
|
printf("(%f, %f, %f) => (%f, %f, %f)\n", map3d_cal[light].cp[1][0], map3d_cal[light].cp[1][1], map3d_cal[light].cp[1][2], tv[0], tv[1], tv[2]);
|
||||||
|
multiply(light, map3d_cal[light].cp[2], tv);
|
||||||
|
printf("(%f, %f, %f) => (%f, %f, %f)\n", map3d_cal[light].cp[2][0], map3d_cal[light].cp[2][1], map3d_cal[light].cp[2][2], tv[0], tv[1], tv[2]);
|
||||||
|
multiply(light, p4, tv);
|
||||||
|
printf("(%f, %f, %f) => (%f, %f, %f)\n", p4[0], p4[1], p4[2], tv[0], tv[1], tv[2]);
|
||||||
|
for (i = 0; i < 10; i++) {
|
||||||
|
pv[0] = 0;
|
||||||
|
pv[1] = (double)i/10.0;
|
||||||
|
pv[2] = 0;
|
||||||
|
multiply(light, pv, tv);
|
||||||
|
printf("(%f, %f, %f) => (%f, %f, %f)\n", pv[0], pv[1], pv[2], tv[0], tv[1], tv[2]);
|
||||||
|
}
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Set parameters based on position (x, y, z) and direction (tx, ty, tz) */
|
||||||
|
/* Temp: x, y = pan, tilt measured to (0, 0, 0)
|
||||||
|
tx, ty = pan, tilt of light fixture
|
||||||
|
tz = distance from light fixture to (0, 0, 0)
|
||||||
|
*/
|
||||||
|
void map3d_setparams(int light, int opan, int otilt, double lpan, double ltilt, double dist)
|
||||||
|
{
|
||||||
|
double n[3];
|
||||||
|
double up[3];
|
||||||
|
double right[3];
|
||||||
|
double v[3];
|
||||||
|
double op, ot;
|
||||||
|
|
||||||
|
op = PANRANGE * ((double)opan - PANOFFSET) / PANMAX;
|
||||||
|
ot = TILTRANGE * ((double)otilt - TILTOFFSET) / TILTMAX;
|
||||||
|
|
||||||
|
unitvector(v, op, ot);
|
||||||
|
unitvector(n, lpan, ltilt);
|
||||||
|
|
||||||
|
up[0] = 0;
|
||||||
|
up[1] = (n[1] + sqrt(n[1]*n[1]+4*n[2]*n[2])) /
|
||||||
|
(4*n[2]);
|
||||||
|
up[2] = -up[1]*n[1]/n[2];
|
||||||
|
normalise(up);
|
||||||
|
|
||||||
|
crossproduct(n, up, right);
|
||||||
|
normalise(right);
|
||||||
|
|
||||||
|
printf("n:\t%f\t%f\t%f\n", n[0], n[1], n[2]);
|
||||||
|
printf("up: \t%f\t%f\t%f\n", up[0], up[1], up[2]);
|
||||||
|
printf("right: \t%f\t%f\t%f\n", right[0], right[1], right[2]);
|
||||||
|
printf("\n");
|
||||||
|
|
||||||
|
/* Construct matrix */
|
||||||
|
|
||||||
|
map3d_cal[light].M[0][0] = right[0];
|
||||||
|
map3d_cal[light].M[0][1] = right[1];
|
||||||
|
map3d_cal[light].M[0][2] = right[2];
|
||||||
|
map3d_cal[light].M[1][0] = up[0];
|
||||||
|
map3d_cal[light].M[1][1] = up[1];
|
||||||
|
map3d_cal[light].M[1][2] = up[2];
|
||||||
|
map3d_cal[light].M[2][0] = n[0];
|
||||||
|
map3d_cal[light].M[2][1] = n[1];
|
||||||
|
map3d_cal[light].M[2][2] = n[2];
|
||||||
|
map3d_cal[light].M[3][0] = dist*v[0];
|
||||||
|
map3d_cal[light].M[3][1] = dist*v[1];
|
||||||
|
map3d_cal[light].M[3][2] = dist*v[2];
|
||||||
|
|
||||||
|
printf("\t%f\t%f\t%f\t\t%f\n", map3d_cal[light].M[0][0], map3d_cal[light].M[0][1], map3d_cal[light].M[0][2], map3d_cal[light].M[3][0]);
|
||||||
|
printf("\t%f\t%f\t%f\t\t%f\n", map3d_cal[light].M[1][0], map3d_cal[light].M[1][1], map3d_cal[light].M[1][2], map3d_cal[light].M[3][1]);
|
||||||
|
printf("\t%f\t%f\t%f\t\t%f\n", map3d_cal[light].M[2][0], map3d_cal[light].M[2][1], map3d_cal[light].M[2][2], map3d_cal[light].M[3][2]);
|
||||||
|
printf("\n");
|
||||||
|
|
||||||
|
}
|
||||||
11
src/lsi/map3d.h
Normal file
11
src/lsi/map3d.h
Normal file
@@ -0,0 +1,11 @@
|
|||||||
|
/* map3d.h */
|
||||||
|
|
||||||
|
void map3d_init(void);
|
||||||
|
|
||||||
|
void map3d_close(void);
|
||||||
|
int map3d_load(void);
|
||||||
|
void map3d_save(void);
|
||||||
|
void map3d_transform(int, double, double, double, int *, int *);
|
||||||
|
void map3d_setcal(int, int, double, double, double, int, int);
|
||||||
|
int map3d_calibrate(int);
|
||||||
|
void map3d_setparams(int, int, int, double, double, double);
|
||||||
101
src/lsi/midi.c
Normal file
101
src/lsi/midi.c
Normal file
@@ -0,0 +1,101 @@
|
|||||||
|
/* midi.c */
|
||||||
|
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
#include <fcntl.h>
|
||||||
|
#include <err.h>
|
||||||
|
#include <sys/midiio.h>
|
||||||
|
#include <errno.h>
|
||||||
|
#include <sys/ioctl.h>
|
||||||
|
#include "vm.h"
|
||||||
|
|
||||||
|
#define PORT "/dev/music"
|
||||||
|
|
||||||
|
int midi_fd;
|
||||||
|
int midi_bytes;
|
||||||
|
seq_event_rec midi_buf;
|
||||||
|
int midi_initialised = 0;
|
||||||
|
|
||||||
|
void midi_init(void)
|
||||||
|
{
|
||||||
|
midi_fd = open(PORT, O_NONBLOCK | O_RDONLY, 0);
|
||||||
|
if (midi_fd == -1) {
|
||||||
|
err(1, "failed to open MIDI port");
|
||||||
|
}
|
||||||
|
midi_bytes = 0;
|
||||||
|
midi_initialised = 1;
|
||||||
|
vm_register_signal_fd(midi_fd, VM_MIDIQ);
|
||||||
|
}
|
||||||
|
|
||||||
|
void midi_close(void)
|
||||||
|
{
|
||||||
|
printf("Closing MIDI\n");
|
||||||
|
ioctl(midi_fd, SEQUENCER_SYNC, NULL);
|
||||||
|
ioctl(midi_fd, SEQUENCER_RESET, NULL);
|
||||||
|
printf("...\n");
|
||||||
|
close(midi_fd);
|
||||||
|
printf("X.\n");
|
||||||
|
midi_fd = 0;
|
||||||
|
midi_initialised = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Returns non-zero if we're interested in the packet
|
||||||
|
* received. Zero otherwise.
|
||||||
|
*/
|
||||||
|
int midi_filter(void)
|
||||||
|
{
|
||||||
|
switch (midi_buf.arr[0]) {
|
||||||
|
case SEQ_CHN_COMMON:
|
||||||
|
/* We don't care about the unit number - yet */
|
||||||
|
/* We don't care about the channel - yet */
|
||||||
|
if (midi_buf.arr[2] == 0xb0) /* cmd */
|
||||||
|
return 1;
|
||||||
|
return 0;
|
||||||
|
case SEQ_TIMING:
|
||||||
|
return 0;
|
||||||
|
default:
|
||||||
|
printf("Unknown MIDI message received\n");
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
int midi_read(void)
|
||||||
|
{
|
||||||
|
if (!midi_initialised)
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
while (1) {
|
||||||
|
int rv;
|
||||||
|
int left = sizeof(midi_buf) - midi_bytes;
|
||||||
|
rv = read(midi_fd, midi_buf.arr + midi_bytes, left);
|
||||||
|
if (rv == -1) {
|
||||||
|
if (errno == EAGAIN)
|
||||||
|
return 0;
|
||||||
|
printf("Error reading from MIDI\n");
|
||||||
|
close(midi_fd);
|
||||||
|
midi_initialised = 0;
|
||||||
|
}
|
||||||
|
if (rv == 0)
|
||||||
|
return 0;
|
||||||
|
midi_bytes += rv;
|
||||||
|
if (midi_bytes == sizeof(midi_buf)) {
|
||||||
|
midi_bytes = 0;
|
||||||
|
if (midi_filter())
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Return the interesting bits of the MIDI command.
|
||||||
|
* The contents of this function is likely to change
|
||||||
|
* as we become interested in more types of MIDI device.
|
||||||
|
* Pre-condition: midi_filter() has returned 1
|
||||||
|
*/
|
||||||
|
void midi_getcmd(int *button, int *value)
|
||||||
|
{
|
||||||
|
*button = midi_buf.arr[4];
|
||||||
|
*value = midi_buf.arr[6];
|
||||||
|
}
|
||||||
7
src/lsi/midi.h
Normal file
7
src/lsi/midi.h
Normal file
@@ -0,0 +1,7 @@
|
|||||||
|
/* midi.h */
|
||||||
|
|
||||||
|
void midi_init(void);
|
||||||
|
void midi_close(void);
|
||||||
|
int midi_filter(void);
|
||||||
|
int midi_read(void);
|
||||||
|
int midi_getcmd(int *, int *);
|
||||||
132
src/lsi/mouse.c
Normal file
132
src/lsi/mouse.c
Normal file
@@ -0,0 +1,132 @@
|
|||||||
|
/* mouse.c */
|
||||||
|
|
||||||
|
#include <sys/time.h>
|
||||||
|
#include <dev/wscons/wsconsio.h>
|
||||||
|
|
||||||
|
#include <fcntl.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
#include <err.h>
|
||||||
|
#include <errno.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include "vm.h"
|
||||||
|
|
||||||
|
#define MOUSEDEVICE "/dev/wsmouse"
|
||||||
|
|
||||||
|
char *events[] = {
|
||||||
|
"undefined",
|
||||||
|
"WSCONS_EVENT_KEY_UP",
|
||||||
|
"WSCONS_EVENT_KEY_DOWN",
|
||||||
|
"WSCONS_EVENT_ALL_KEYS_UP",
|
||||||
|
"WSCONS_EVENT_MOUSE_UP",
|
||||||
|
"WSCONS_EVENT_MOUSE_DOWN",
|
||||||
|
"WSCONS_EVENT_MOUSE_DELTA_X",
|
||||||
|
"WSCONS_EVENT_MOUSE_DELTA_Y",
|
||||||
|
"WSCONS_EVENT_MOUSE_ABSOLUTE_X",
|
||||||
|
"WSCONS_EVENT_MOUSE_ABSOLUTE_Y",
|
||||||
|
"WSCONS_EVENT_MOUSE_DELTA_Z",
|
||||||
|
"WSCONS_EVENT_MOUSE_ABSOLUTE_Z",
|
||||||
|
"WSCONS_EVENT_SCREEN_SWITCH",
|
||||||
|
"WSCONS_EVENT_ASCII",
|
||||||
|
"WSCONS_EVENT_MOUSE_DELTA_W",
|
||||||
|
"WSCONS_EVENT_MOUSE_ABSOLUTE_W"
|
||||||
|
};
|
||||||
|
|
||||||
|
#define WSCONS_EVENT_DELTA_X 6
|
||||||
|
#define WSCONS_EVENT_DELTA_Y 7
|
||||||
|
#define WSCONS_EVENT_DELTA_Z 10
|
||||||
|
|
||||||
|
#define MOUSE_SCALE 4
|
||||||
|
|
||||||
|
#define MOUSE_X_MIN (-127 * MOUSE_SCALE)
|
||||||
|
#define MOUSE_X_MAX (127 * MOUSE_SCALE)
|
||||||
|
#define MOUSE_Y_MIN (-127 * MOUSE_SCALE)
|
||||||
|
#define MOUSE_Y_MAX (127 * MOUSE_SCALE)
|
||||||
|
#define MOUSE_Z_MIN -127
|
||||||
|
#define MOUSE_Z_MAX 127
|
||||||
|
|
||||||
|
int mouse_fd;
|
||||||
|
int mouse_bytes;
|
||||||
|
struct wscons_event mouse_buf;
|
||||||
|
int mouse_initialised = 0;
|
||||||
|
int mouse_x, mouse_y, mouse_z;
|
||||||
|
|
||||||
|
void mouse_init(void)
|
||||||
|
{
|
||||||
|
mouse_fd = open(MOUSEDEVICE, O_RDONLY);
|
||||||
|
if (mouse_fd < 0)
|
||||||
|
err(1, "can't open mouse device");
|
||||||
|
|
||||||
|
vm_register_signal_fd(mouse_fd, VM_MOUSEQ);
|
||||||
|
mouse_initialised = 1;
|
||||||
|
mouse_x = 0;
|
||||||
|
mouse_y = 0;
|
||||||
|
mouse_z = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
void mouse_close(void)
|
||||||
|
{
|
||||||
|
close(mouse_fd);
|
||||||
|
mouse_initialised = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int mouse_filter(void)
|
||||||
|
{
|
||||||
|
switch (mouse_buf.type) {
|
||||||
|
case WSCONS_EVENT_MOUSE_DELTA_X:
|
||||||
|
mouse_x += mouse_buf.value;
|
||||||
|
if (mouse_x > MOUSE_X_MAX)
|
||||||
|
mouse_x = MOUSE_X_MAX;
|
||||||
|
if (mouse_x < MOUSE_X_MIN)
|
||||||
|
mouse_x = MOUSE_X_MIN;
|
||||||
|
return 1;
|
||||||
|
case WSCONS_EVENT_MOUSE_DELTA_Y:
|
||||||
|
mouse_y += mouse_buf.value;
|
||||||
|
if (mouse_y > MOUSE_Y_MAX)
|
||||||
|
mouse_y = MOUSE_Y_MAX;
|
||||||
|
if (mouse_y < MOUSE_Y_MIN)
|
||||||
|
mouse_y = MOUSE_Y_MIN;
|
||||||
|
return 1;
|
||||||
|
case WSCONS_EVENT_MOUSE_DELTA_Z:
|
||||||
|
mouse_z += mouse_buf.value;
|
||||||
|
if (mouse_z > MOUSE_Z_MAX)
|
||||||
|
mouse_z = MOUSE_Z_MAX;
|
||||||
|
if (mouse_z < MOUSE_Z_MIN)
|
||||||
|
mouse_z = MOUSE_Z_MIN;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
void mouse_getpos(int *x, int *y, int *z)
|
||||||
|
{
|
||||||
|
*x = mouse_x / MOUSE_SCALE;
|
||||||
|
*y = mouse_y / MOUSE_SCALE;
|
||||||
|
*z = mouse_z;
|
||||||
|
}
|
||||||
|
|
||||||
|
int mouse_read(void)
|
||||||
|
{
|
||||||
|
if (!mouse_initialised)
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
while (1) {
|
||||||
|
int rv;
|
||||||
|
int left = sizeof(mouse_buf) - mouse_bytes;
|
||||||
|
rv = read(mouse_fd, ((char *)&mouse_buf) + mouse_bytes, left);
|
||||||
|
if (rv == -1) {
|
||||||
|
if (errno == EAGAIN)
|
||||||
|
return 0;
|
||||||
|
printf("Error reading from mouse\n");
|
||||||
|
close(mouse_fd);
|
||||||
|
mouse_initialised = 0;
|
||||||
|
}
|
||||||
|
if (rv == 0)
|
||||||
|
return 0;
|
||||||
|
mouse_bytes += rv;
|
||||||
|
if (mouse_bytes == sizeof(mouse_buf)) {
|
||||||
|
mouse_bytes = 0;
|
||||||
|
if (mouse_filter())
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
7
src/lsi/mouse.h
Normal file
7
src/lsi/mouse.h
Normal file
@@ -0,0 +1,7 @@
|
|||||||
|
/* mouse.h */
|
||||||
|
|
||||||
|
void mouse_init(void);
|
||||||
|
void mouse_close(void);
|
||||||
|
int mouse_filter(void);
|
||||||
|
void mouse_getpos(int *, int *, int *);
|
||||||
|
int mouse_read(void);
|
||||||
1237
src/lsi/vm.c
Normal file
1237
src/lsi/vm.c
Normal file
File diff suppressed because it is too large
Load Diff
20
src/lsi/vm.h
Normal file
20
src/lsi/vm.h
Normal file
@@ -0,0 +1,20 @@
|
|||||||
|
/* vm.h */
|
||||||
|
|
||||||
|
void vm_init(void);
|
||||||
|
void vm_load(char *);
|
||||||
|
int vm_spawn(char *);
|
||||||
|
void vm_run(void);
|
||||||
|
void vm_register_signal_fd(int /* fd */, int /* queue */);
|
||||||
|
|
||||||
|
#define VM_MAXQUEUES 512
|
||||||
|
#define VM_RUNQ 0
|
||||||
|
#define VM_TIMEQ 1
|
||||||
|
|
||||||
|
#define VM_MOUSEQ 6
|
||||||
|
#define VM_BEATQ 7
|
||||||
|
#define VM_MIDIQ 8
|
||||||
|
#define VM_SOCKQ 9
|
||||||
|
|
||||||
|
#define VM_USERQMIN 10
|
||||||
|
#define VM_NOQUEUE (-1)
|
||||||
|
|
||||||
12
src/maketar
Executable file
12
src/maketar
Executable file
@@ -0,0 +1,12 @@
|
|||||||
|
#!/bin/sh
|
||||||
|
|
||||||
|
./build clean
|
||||||
|
|
||||||
|
RELEASENAME=lightscript-0.0
|
||||||
|
CONTENTS="build common include lsc lsi maketar"
|
||||||
|
|
||||||
|
|
||||||
|
mkdir ${RELEASENAME}
|
||||||
|
cp -pR ${CONTENTS} ${RELEASENAME}
|
||||||
|
tar czf ${RELEASENAME}.tar.gz ${RELEASENAME}
|
||||||
|
rm -rf ${RELEASENAME}
|
||||||
Reference in New Issue
Block a user