OSC.jl/src/OSC.jl

325 lines
9.7 KiB
Julia

#OSC.jl
#Copyright (c) 2014, Mark McCurry, All rights reserved.
#
#This library is free software; you can redistribute it and/or
#modify it under the terms of the GNU Lesser General Public
#License as published by the Free Software Foundation; either
#version 3.0 of the License, or (at your option) any later version.
#
#This library is distributed in the hope that it will be useful,
#but WITHOUT ANY WARRANTY; without even the implied warranty of
#MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
#Lesser General Public License for more details.
#
#You should have received a copy of the GNU Lesser General Public
#License along with this library.
module OSC
import Base.show
macro incfp(x) quote begin
local gensym_ = $(esc(x))
$(esc(x)) = $(esc(x))+1
gensym_
end end end
type OscMsg
data::Array{Uint8}
end
function stringify(data::Array{Uint8})
zeroInd = find(data.== 0)
if(length(zeroInd) == 0)
return string(map(char, data)...)
elseif(zeroInd[1] == 0)
return nothing
else
return string(map(char, data[1:zeroInd[1]-1])...)
end
end
function names(msg::OscMsg)#::ASCIIString
pos = 1
while(msg.data[pos += 1] != 0) end #skip pattern
while(msg.data[pos += 1] == 0) end #skip null
return stringify(msg.data[pos+1:end]); #skip comma
end
strip_args(args::ASCIIString) = replace(replace(args,"]",""),"[","")
function narguments(msg::OscMsg)
length(strip_args(names(msg)))
end
has_reserved(typeChar::Char) = typeChar in "isbfhtdSrmc"
nreserved(args::ASCIIString) = sum(map(has_reserved, collect(args)))
function argType(msg::OscMsg, nargument::Int)#::Char
@assert(nargument < narguments(msg));
return strip_args(names(msg))[nargument+1]
end
align(pos) = pos+(4-(pos-1)%4)
function arg_off(msg::OscMsg, idx::Int)#::Int
if(!has_reserved(argType(msg,idx)))
return 0;
end
#Iterate to the right position
args::ASCIIString = names(msg);
argc::Int = 1
pos::Int = 1
#Get past the Argument String
while(msg.data[pos] != ',') pos += 1 end
while(msg.data[pos] != 0) pos += 1 end
#Alignment
pos = align(pos)
#ignore any leading '[' or ']'
while(args[argc] in "[]") argc += 1 end
while(idx != 0)
bundle_length::Uint32 = 0;
arg = args[argc]
argc += 1
if(arg in "htd")
pos +=8;
elseif(arg in "mrfci")
pos += 4;
elseif(arg in "Ss")
while(msg.data[pos += 1] != 0) end
pos = align(pos)
elseif(arg == 'b')
bundle_length |= (msg.data[@incfp(pos)] << 24);
bundle_length |= (msg.data[@incfp(pos)] << 16);
bundle_length |= (msg.data[@incfp(pos)] << 8);
bundle_length |= (msg.data[@incfp(pos)]);
bundle_length += 4-bundle_length%4;
pos += bundle_length;
elseif(arg in "[]")#completely ignore array chars
idx += 1;
else #TFI
end
idx -= 1
end
pos;
end
#Calculate the size of the message without writing to a buffer
function vsosc_null(address::ASCIIString,
arguments::ASCIIString,
args...)
pos::Int = length(address)+1
pos = align(pos)
pos += 1+length(arguments)
pos = align(pos)
arg_pos = 1;
#Take care of varargs
for(arg = arguments)
if(arg in "htd")
arg_pos += 1
pos += 8;
elseif(arg in "mrcfi")
arg_pos += 1
pos += 4;
elseif(arg in "sS")
s::ASCIIString = args[@incfp(arg_pos)];
pos += length(s);
pos = align(pos)
elseif(arg in "b")
i::Int32 = sizeof(args[@incfp(arg_pos)])
pos += 4 + i;
pos = align(pos)
end #other args classes are ignored
end
return pos-1;
end
function rtosc_amessage(buffer::Array{Uint8},
len::Int,
address::ASCIIString,
arguments::ASCIIString,
args...)
total_len::Int = vsosc_null(address, arguments, args...);
for(i=1:total_len)
buffer[i] = 0
end
#Abort if the message cannot fit
if(total_len>len)
return 0;
end
pos::Int = 1;
#Address
for(C = address) buffer[@incfp(pos)] = C end
pos = align(pos)
#Arguments
buffer[@incfp(pos)] = ',';
for(A=arguments) buffer[@incfp(pos)] = A; end
pos = align(pos)
arg_pos::Int = 1;
for(arg = arguments)
@assert(arg != 0);
if(arg in "htd")
d::Uint64 = reinterpret(Uint64, args[@incfp(arg_pos)]);
buffer[@incfp(pos)] = ((d>>56) & 0xff);
buffer[@incfp(pos)] = ((d>>48) & 0xff);
buffer[@incfp(pos)] = ((d>>40) & 0xff);
buffer[@incfp(pos)] = ((d>>32) & 0xff);
buffer[@incfp(pos)] = ((d>>24) & 0xff);
buffer[@incfp(pos)] = ((d>>16) & 0xff);
buffer[@incfp(pos)] = ((d>>8) & 0xff);
buffer[@incfp(pos)] = (d & 0xff);
elseif(arg in "rfci")
i::Int32 = reinterpret(Int32, args[@incfp(arg_pos)]);
buffer[@incfp(pos)] = ((i>>24) & 0xff);
buffer[@incfp(pos)] = ((i>>16) & 0xff);
buffer[@incfp(pos)] = ((i>>8) & 0xff);
buffer[@incfp(pos)] = (i & 0xff);
elseif(arg in "m")
m = args[@incfp(arg_pos)];
buffer[@incfp(pos)] = m[1];
buffer[@incfp(pos)] = m[2];
buffer[@incfp(pos)] = m[3];
buffer[@incfp(pos)] = m[4];
elseif(arg in "Ss")
s = args[@incfp(arg_pos)];
for(C = s)
buffer[@incfp(pos)] = C
end
pos = align(pos)
elseif(arg == 'b')
b = args[@incfp(arg_pos)];
i = sizeof(b);
buffer[@incfp(pos)] = ((i>>24) & 0xff);
buffer[@incfp(pos)] = ((i>>16) & 0xff);
buffer[@incfp(pos)] = ((i>>8) & 0xff);
buffer[@incfp(pos)] = (i & 0xff);
for(U = b)
buffer[@incfp(pos)] = uint8(U);
end
pos = align(pos)
end
end
return pos-1;
end
OscMsg(address, arguments, args...) = message(address, arguments, args...)
function message(address::ASCIIString,
arguments::ASCIIString,
args...)
len::Int = vsosc_null(address, arguments, args...);
data::Vector{Uint8} = Array(Uint8, len);
rtosc_amessage(data,len,address,arguments,args...);
return OscMsg(data)
end
function rtosc_argument(msg::OscMsg, idx::Int)
typeChar::Char = argType(msg, idx);
#trivial case
if(!has_reserved(typeChar))
if(typeChar == 'T')
return true
elseif(typeChar == 'F')
return false;
end
else
arg_pos::Int = arg_off(msg, idx)
if(typeChar in "htd")
t::Uint64 = 0
t |= (uint64(msg.data[@incfp(arg_pos)]) << 56);
t |= (uint64(msg.data[@incfp(arg_pos)]) << 48);
t |= (uint64(msg.data[@incfp(arg_pos)]) << 40);
t |= (uint64(msg.data[@incfp(arg_pos)]) << 32);
t |= (uint64(msg.data[@incfp(arg_pos)]) << 24);
t |= (uint64(msg.data[@incfp(arg_pos)]) << 16);
t |= (uint64(msg.data[@incfp(arg_pos)]) << 8);
t |= (uint64(msg.data[@incfp(arg_pos)]));
if(typeChar == 'h')
return int64(t)
elseif(typeChar == 'd')
return reinterpret(Float64, t);
else
return t;
end
elseif(typeChar in "f")
return reinterpret(Float32,msg.data[arg_pos+(3:-1:0)])[1]
elseif(typeChar in "rci")
i::Int32 = 0
i |= (uint32(msg.data[@incfp(arg_pos)]) << 24);
i |= (uint32(msg.data[@incfp(arg_pos)]) << 16);
i |= (uint32(msg.data[@incfp(arg_pos)]) << 8);
i |= (uint32(msg.data[@incfp(arg_pos)]));
if(typeChar == 'r')
return uint32(i)
elseif(typeChar == 'c')
return char(i)
else
return i
end
elseif(typeChar in "m")
m = Array(Uint8, 4)
m[1] = msg.data[@incfp(arg_pos)]
m[2] = msg.data[@incfp(arg_pos)]
m[3] = msg.data[@incfp(arg_pos)]
m[4] = msg.data[@incfp(arg_pos)]
return m
elseif(typeChar in "b")
len::Int32 = 0
len |= (msg.data[@incfp(arg_pos)] << 24);
len |= (msg.data[@incfp(arg_pos)] << 16);
len |= (msg.data[@incfp(arg_pos)] << 8);
len |= (msg.data[@incfp(arg_pos)]);
return msg.data[arg_pos+(0:len-1)];
elseif(typeChar in "Ss")
return stringify(msg.data[arg_pos:end]);
end
end
return nothing;
end
getindex(msg::OscMsg, idx::Int) = rtosc_argument(msg, idx)
function show(io::IO, msg::OscMsg)
println(io, "OSC Message to ", stringify(msg.data))
println(io, " Arguments:");
for i=1:narguments(msg)
showField(io, msg,i)
end
end
function showField(io::IO, msg::OscMsg, arg_id)
map = ['i' Int32; 'f' Float32; 's' String; 'b' :Blob; 'h' Int32; 't' Uint64;
'd' Float64; 'S' Symbol; 'c' Char; 'r' :RBG; 'm' :Midi; 'T' true;
'F' false; 'N' Nothing]
dict = Dict{Char, Any}(map[:,1][:],map[:,2][:])
dict['I'] = Inf
typeChar::Char = argType(msg, arg_id-1)
value = msg[arg_id-1]
if(issubtype(typeof(value), Array))
value = value'
end
@printf(io, " #%2d %c:", arg_id, typeChar);
print(dict[typeChar]," - ", value)
if(!issubtype(typeof(value), Array))
println()
end
end
export OscMsg
end