/* $Id: flow_objects.c 3969 2008-07-04 19:22:23Z matju $ GridFlow Copyright (c) 2001-2008 by Mathieu Bouchard This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version. See file ../COPYING for further informations on licensing terms. This program 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 General Public License for more details. You should have received a copy of the GNU General Public License along with this program; if not, write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. */ #include #include #include #include #include #include #include "../gridflow.h.fcs" #ifndef DESIREDATA extern "C" { #include "bundled/g_canvas.h" }; #endif extern "C" t_canvas *canvas_getrootfor(t_canvas *x); /* both oprintf are copied from desiredata */ //using namespace std; // can't static void voprintf(std::ostream &buf, const char *s, va_list args) { char *b; vasprintf(&b,s,args); buf << b; free(b); } static void oprintf(std::ostream &buf, const char *s, ...) { va_list args; va_start(args,s); voprintf(buf,s,args); va_end(args); } /* ---------------------------------------------------------------- */ // BAD HACK: GCC complains: unimplemented (--debug mode only) (i don't remember which GCC this was) #ifdef HAVE_DEBUG #define SCOPY(a,b,n) COPY(a,b,n) #else #define SCOPY(a,b,n) SCopy::f(a,b) #endif template class SCopy { public: template static inline void __attribute__((always_inline)) f(T *a, T *b) { *a=*b; SCopy::f(a+1,b+1);}}; template <> class SCopy<0> { public: template static inline void __attribute__((always_inline)) f(T *a, T *b) {}}; /*template <> class SCopy<4> { public: template static inline void __attribute__((always_inline)) f(T *a, T *b) { *a=*b; SCopy<3>::f(a+1,b+1);} static inline void __attribute__((always_inline)) f(uint8 *a, uint8 *b) { *(int32 *)a=*(int32 *)b; } };*/ Numop *op_add, *op_sub, *op_mul, *op_div, *op_mod, *op_shl, *op_and, *op_put; static void expect_dim_dim_list (P d) { if (d->n!=1) RAISE("dimension list should be Dim[n], not %s",d->to_s());} //static void expect_min_one_dim (P d) { // if (d->n<1) RAISE("minimum 1 dimension");} static void expect_max_one_dim (P d) { if (d->n>1) RAISE("expecting Dim[] or Dim[n], got %s",d->to_s());} //static void expect_exactly_one_dim (P d) { // if (d->n!=1) RAISE("expecting Dim[n], got %s",d->to_s());} //**************************************************************** \class GridCast : FObject { \attr NumberTypeE nt; \constructor (NumberTypeE nt) {this->nt = nt;} \grin 0 }; GRID_INLET(GridCast,0) { out = new GridOutlet(this,0,in->dim,nt); } GRID_FLOW { out->send(n,data); } GRID_END \end class {install("#cast",1,1); add_creator("@cast");} //**************************************************************** GridHandler *stromgol; // remove this asap //{ ?,Dim[B] -> Dim[*Cs] } // out0 nt to be specified explicitly \class GridImport : FObject { \attr NumberTypeE cast; \attr P dim; // size of grids to send PtrGrid dim_grid; \constructor (...) { dim_grid.constrain(expect_dim_dim_list); this->cast = argc>=2 ? NumberTypeE_find(argv[1]) : int32_e; if (argc>2) RAISE("too many arguments"); if (argc>0 && argv[0]!=gensym("per_message")) { dim_grid=new Grid(argv[0]); dim = dim_grid->to_dim(); if (!dim->prod()) RAISE("target grid size must not be zero"); } } ~GridImport() {} \decl 0 reset(); \decl 0 symbol(t_symbol *x); \decl 0 to_ascii(...); //\decl 0 list(...); \decl 1 per_message(); \grin 0 \grin 1 int32 template void process (long n, T *data) { if (in.size()<=0) in.resize(1); if (!in[0]) in[0]=new GridInlet((FObject *)this,stromgol); while (n) { if (!out || !out->dim) out = new GridOutlet(this,0,dim?dim:in[0]->dim,cast); long n2 = min((long)n,out->dim->prod()-out->dex); out->send(n2,data); n-=n2; data+=n2; } } }; GRID_INLET(GridImport,0) {} GRID_FLOW { process(n,data); } GRID_END GRID_INPUT(GridImport,1,dim_grid) { P d = dim_grid->to_dim(); if (!d->prod()) RAISE("target grid size must not be zero"); dim = d; } GRID_END \def 0 symbol(t_symbol *x) { const char *name = x->s_name; long n = strlen(name); if (!dim) out=new GridOutlet(this,0,new Dim(n)); process(n,(uint8 *)name); } \def 0 to_ascii(...) { std::ostringstream os; pd_oprint(os,argc,argv); string s = os.str(); long n = s.length(); if (!dim) out=new GridOutlet(this,0,new Dim(n),cast); process(n,(uint8 *)s.data()); } \def 0 list(...) {//first two lines are there until grins become strictly initialized. if (in.size()<=0) in.resize(1); if (!in[0]) in[0]=new GridInlet((FObject *)this,stromgol); in[0]->from_list(argc,argv,cast);} \def 1 per_message() {dim=0; dim_grid=0;} \def 0 reset() {int32 foo[1]={0}; while (out->dim) out->send(1,foo);} \end class {install("#import",2,1); add_creator("@import"); stromgol = &GridImport_grid_0_hand;} //**************************************************************** /*{ Dim[*As] -> ? }*/ /* in0: integer nt */ \class GridToFloat : FObject { \constructor () {} \grin 0 }; GRID_INLET(GridToFloat,0) { } GRID_FLOW { for (int i=0; ioutlets[0],data[i]); } GRID_END \end class {install("#to_float",1,1); add_creator("#export"); add_creator("@export");} \class GridToSymbol : FObject { \constructor () {} \grin 0 }; GRID_INLET(GridToSymbol,0) { in->set_chunk(0); } GRID_FLOW { char c[n+1]; for (int i=0; ioutlets[0],gensym(c)); } GRID_END \end class {install("#to_symbol",1,1); add_creator("#export_symbol"); add_creator("@export_symbol");} /*{ Dim[*As] -> ? }*/ /* in0: integer nt */ \class GridExportList : FObject { \constructor () {} int n; \grin 0 }; GRID_INLET(GridExportList,0) { long n = in->dim->prod(); if (n>1000000) RAISE("list too big (%ld elements, max 1000000)", n); this->n = n; in->set_chunk(0); } GRID_FLOW { send_out(0,n,data); } GRID_FINISH { if (in->dim->prod()==0) send_out(0,0,data); } GRID_END \end class {install("#to_list",1,1); add_creator("#export_list"); add_creator("@export_list");} /* **************************************************************** */ \class GridPrint : FObject { \constructor (t_symbol *name=0) { this->dest = 0; this->name = name; base=10; trunc=70; maxrows=50; } \attr t_symbol *name; \grin 0 int base; uint32 trunc; int maxrows; int columns; t_pd *dest; \decl 0 dest (void *p); \decl void end_hook (); \decl 0 base (int x); \decl 0 trunc (int x); \decl 0 maxrows (int y); void puts (const char *s) { if (!dest) post("%s",s); else { int n = strlen(s); t_atom a[n]; for (int i=0; i void make_columns (int n, T *data); template void dump(std::ostream &s, int n, T *data, char sep=' ', int trunc=-1) { if (trunc<0) trunc=this->trunc; std::string f = format(NumberTypeE_type_of(data)); for (int i=0; i=0; j--) s<<' '; if (data[i]!=x) s<<'-'; for (int j=ndigits-1; j>=0; j--) { s<>j)&1)); } } if (is_name << ": "; s << "Dim["; for (int i=0; idim->n; i++) { s << in->dim->v[i]; if (idim->n-1) s << ','; } s << "]"; if (in->nt!=int32_e) s << "(" << number_type_table[in->nt].name << ")"; s << ": "; } std::string format (NumberTypeE nt) { if (nt==float32_e) return "%6.6f"; if (nt==float64_e) return "%14.14f"; std::ostringstream r; r << "%"; r << columns; //if (nt==int64_e) r << "l"; if (base==2) r << "b"; else if (base==8) r << "o"; else if (base==10) r << "d"; else if (base==16) r << "x"; return r.str(); } }; \def 0 dest (void *p) {dest = (t_pd *)p;} \def void end_hook () {} \def 0 base (int x) { if (x==2 || x==8 || x==10 || x==16) base=x; else RAISE("base %d not supported",x); } \def 0 trunc (int x) { if (x<0 || x>240) RAISE("out of range (not in 0..240 range)"); trunc = x; } \def 0 maxrows (int y) {maxrows = y;} template void GridPrint::make_columns (int n, T *data) { long maxv=0; long minv=0; for (int i=0; idata[i]) minv=long(data[i]); } int maxd = 1 + (maxv<0) + int(log(max(1.,fabs(maxv)))/log(base)); int mind = 1 + (minv<0) + int(log(max(1.,fabs(minv)))/log(base)); //fprintf(stderr,"v=(%d,%d) d=(%d,%d)\n",minv,maxv,mind,maxd); columns = max(maxd,mind); } GRID_INLET(GridPrint,0) { in->set_chunk(0); } GRID_FLOW { std::ostringstream head; dump_dims(head,in); int ndim = in->dim->n; if (ndim > 3) { head << " (not printed)"; puts(head); } else if (ndim < 2) { make_columns(n,data); dump(head,n,data,' ',trunc); puts(head); } else if (ndim == 2) { puts(head); make_columns(n,data); long sy = in->dim->v[0]; long sx = n/sy; for (int row=0; rowdim->v[0]; row++) { std::ostringstream body; dump(body,sx,&data[sx*row],' ',trunc); puts(body); if (row>maxrows) {puts("..."); break;} } } else if (ndim == 3) { puts(head); make_columns(n,data); int sy = in->dim->v[0]; int sx = in->dim->v[1]; int sz = n/sy; int sz2 = sz/in->dim->v[1]; for (int row=0; rowtrunc) break; } puts(str); if (row>maxrows) {puts("..."); break;} } } end_hook(0,0); } GRID_FINISH { std::ostringstream head; dump_dims(head,in); if (in->dim->prod()==0) puts(head); } GRID_END \end class {install("#print",1,1); add_creator("@print");} /* **************************************************************** */ // [#store] is the class for storing a grid and restituting it on demand. // The right inlet receives the grid. The left inlet receives either a bang // (which forwards the whole image) or a grid describing what to send. //{ Dim[*As,B],Dim[*Cs,*Ds] -> Dim[*As,*Ds] } // in0: integer nt // in1: whatever nt // out0: same nt as in1 \class GridStore : FObject { PtrGrid r; // can't be \attr PtrGrid put_at; // can't be //\attr \attr Numop *op; int32 *wdex ; // temporary buffer, copy of put_at int32 *fromb; int32 *to2 ; int lsd; // lsd = Last Same Dimension (for put_at) int d; // goes with wdex \constructor (Grid *r=0) { put_at.constrain(expect_max_one_dim); this->r = r?r:new Grid(new Dim(),int32_e,true); op = op_put; wdex = NEWBUF(int32,Dim::MAX_DIM); // temporary buffer, copy of put_at fromb = NEWBUF(int32,Dim::MAX_DIM); to2 = NEWBUF(int32,Dim::MAX_DIM); } ~GridStore () { DELBUF(wdex); DELBUF(fromb); DELBUF(to2); } \decl 0 bang (); \decl 1 reassign (); \decl 1 put_at (Grid *index); \grin 0 int \grin 1 template void compute_indices(T *v, long nc, long nd); }; // takes the backstore of a grid and puts it back into place. a backstore // is a grid that is filled while the grid it would replace has not // finished being used. static void snap_backstore (PtrGrid &r) { if (r.next) {r=r.next.p; r.next=0;} } template void GridStore::compute_indices(T *v, long nc, long nd) { for (int i=0; idim->v[i]; bool fast = lowest_bit(wrap)==highest_bit(wrap); // is power of two? if (i) { if (fast) op_shl->map(nd,v,(T)highest_bit(wrap)); else op_mul->map(nd,v,(T)wrap); } if (fast) op_and->map(nd,v+nd*i,(T)(wrap-1)); else op_mod->map(nd,v+nd*i,(T)(wrap)); if (i) op_add->zip(nd,v,v+nd*i); } } // !@#$ i should ensure that n is not exceedingly large // !@#$ worse: the size of the foo buffer may still be too large GRID_INLET(GridStore,0) { // snap_backstore must be done before *anything* else snap_backstore(r); int na = in->dim->n; int nb = r->dim->n; int32 v[Dim::MAX_DIM]; if (na<1) RAISE("must have at least 1 dimension.",na,1,1+nb); long nc = in->dim->get(na-1); if (nc>nb) RAISE("got %d elements in last dimension, expecting <= %d", nc, nb); long nnc = r->dim->prod(nc); int lastindexable = nnc ? r->dim->prod()/nnc-1 : 0; // SIGFPE happened when r was especially empty (nnc==0) int ngreatest = nt_greatest((T *)0); if (lastindexable > ngreatest) RAISE("lastindexable=%d > ngreatest=%d (ask matju)",lastindexable,ngreatest); int nd = nb-nc+na-1; COPY(v,in->dim->v,na-1); COPY(v+na-1,r->dim->v+nc,nb-nc); out=new GridOutlet(this,0,new Dim(nd,v),r->nt); if (nc>0) in->set_chunk(na-1); } GRID_FLOW { int na = in->dim->n; int nc = in->dim->get(na-1); long size = r->dim->prod(nc); long nd = n/nc; T w[n]; T *v=w; if (sizeof(T)==1 && nc==1 && r->dim->v[0]<=256) { // bug? shouldn't modulo be done here? v=data; } else { COPY(v,data,n); for (long k=0,i=0; igive(size*nd,foo-size*nd); \ } else { \ for (int i=0; isend(size,p+size*v[i]); \ } \ } TYPESWITCH(r->nt,FOO,) #undef FOO } GRID_FINISH { if (in->dim->prod()==0) { long n = in->dim->prod(0,-2); long size = r->dim->prod(); #define FOO(T) while (n--) out->send(size,(T *)*r); TYPESWITCH(r->nt,FOO,) #undef FOO } } GRID_END GRID_INLET(GridStore,1) { NumberTypeE nt = NumberTypeE_type_of(data); if (!put_at) { // reassign if (in[0].dim) r.next = new Grid(in->dim,nt); else r = new Grid(in->dim,nt); return; } SAME_TYPE(in,r); // put_at ( ... ) //!@#$ should check types. if (r->nt!=in->nt) RAISE("shoo"); long nn=r->dim->n, na=put_at->dim->v[0], nb=in->dim->n; int32 sizeb[nn]; for (int i=0; idim->prod()); COPY(fromb+nn-na,(int32 *)*put_at ,na); COPY(sizeb+nn-nb,(int32 *)in->dim->v,nb); for (int i=0; i=nn-in->dim->n) { lsd--; //int cs = in->dim->prod(lsd-nn+in->dim->n); if (/*cs*(number_type_table[in->nt].size/8)>GridOutlet::MAX_PACKET_SIZE ||*/ fromb[lsd]!=0 || sizeb[lsd]!=r->dim->v[lsd]) break; } lsd++; in->set_chunk(lsd-nn+in->dim->n); } GRID_FLOW { //fprintf(stderr,"d=%d\n",d); if (!put_at) { // reassign COPY(((T *)*(r.next ? r.next.p : &*r.p))+in->dex, data, n); return; } // put_at (...) long cs = in->factor(); // chunksize int32 v[lsd]; int32 *x = wdex; while (n) { // here d is the dim# to reset; d=n for none for(;dzip(cs,(T *)*r+v[0]*cs,data); data+=cs; n-=cs; // find next set of indices; here d is the dim# to increment for(;;) {d--; if (d<0) return; x[d]++; if (x[d] -> Dim[*As] } or //{ Dim[*As],Dim[*Bs] -> Dim[*As] } \class GridOp : FObject { \attr Numop *op; PtrGrid r; \constructor (Numop *op, Grid *r=0) { this->op=op; this->r=r?r:new Grid(new Dim(),int32_e,true); } \grin 0 \grin 1 }; GRID_INLET(GridOp,0) { snap_backstore(r); SAME_TYPE(in,r); out=new GridOutlet(this,0,in->dim,in->nt); in->set_mode(6); if (op->size>1 && (in->dim->get(in->dim->n-1)!=op->size || r->dim->get(r->dim->n-1)!=op->size)) RAISE("using %s requires Dim(...,%d) in both inlets but got: left=%s right=%s", op->name,op->size,in->dim->to_s(),r->dim->to_s()); } GRID_ALLOC { //out->ask(in->allocn,(T * &)in->alloc,in->allocfactor,in->allocmin,in->allocmax); } GRID_FLOW { T *rdata = (T *)*r; long loop = r->dim->prod(); if (sizeof(T)==8) { fprintf(stderr,"1: data=%p rdata=%p\n",data,rdata); WATCH(n,data); } //fprintf(stderr,"[#] op=%s loop=%ld\n",op->name,loop); if (loop>1) { if (in->dex+n <= loop) { op->zip(n/op->size,data,rdata+in->dex); } else { // !@#$ should prebuild and reuse this array when "loop" is small T data2[n]; long ii = mod(in->dex,loop); long m = min(loop-ii,n); COPY(data2,rdata+ii,m); long nn = m+((n-m)/loop)*loop; for (long i=m; inn) COPY(data2+nn,rdata,n-nn); if (sizeof(T)==8) { fprintf(stderr,"2: data=%p data2=%p\n",data,data2); WATCH(n,data); WATCH(n,data2); } op->zip(n/op->size,data,data2); if (sizeof(T)==8) {WATCH(n,data); WATCH(n,data2);} } } else { op->map(n,data,*rdata); } out->give(n,data); } GRID_END GRID_INPUT2(GridOp,1,r) {} GRID_END \end class {install("#",2,1); add_creator("@");} //**************************************************************** \class GridFold : FObject { \attr Numop *op; \attr PtrGrid seed; \constructor (Numop *op) {this->op=op;} \grin 0 }; GRID_INLET(GridFold,0) { //{ Dim[*As,B,*Cs],Dim[*Cs] -> Dim[*As,*Cs] } if (seed) SAME_TYPE(in,seed); int an = in->dim->n; int bn = seed?seed->dim->n:0; if (an<=bn) RAISE("minimum 1 more dimension than the seed (%d vs %d)",an,bn); int32 v[an-1]; int yi = an-bn-1; COPY(v,in->dim->v,yi); COPY(v+yi,in->dim->v+an-bn,bn); if (seed) SAME_DIM(an-(yi+1),in->dim,(yi+1),seed->dim,0); out=new GridOutlet(this,0,new Dim(an-1,v),in->nt); in->set_chunk(yi); if (in->dim->prod(yi)==0) { long n = out->dim->prod(); T x=0; op->on(x)->neutral(&x,at_left); for(long i=0; isend(1,&x); } } GRID_FLOW { int an = in->dim->n; int bn = seed?seed->dim->n:0; long yn = in->dim->v[an-bn-1]; long zn = in->dim->prod(an-bn); T buf[n/yn]; long nn=n; long yzn=yn*zn; for (long i=0; n; i+=zn, data+=yzn, n-=yzn) { if (seed) COPY(buf+i,((T *)*seed),zn); else {T neu; op->on(*buf)->neutral(&neu,at_left); op_put->map(zn,buf+i,neu);} op->fold(zn,yn,buf+i,data); } out->send(nn/yn,buf); } GRID_FINISH { } GRID_END \end class {install("#fold",1,1);} \class GridScan : FObject { \attr Numop *op; \attr PtrGrid seed; \constructor (Numop *op) {this->op = op;} \grin 0 }; GRID_INLET(GridScan,0) { //{ Dim[*As,B,*Cs],Dim[*Cs] -> Dim[*As,B,*Cs] } if (seed) SAME_TYPE(in,seed); int an = in->dim->n; int bn = seed?seed->dim->n:0; if (an<=bn) RAISE("minimum 1 more dimension than the right hand"); if (seed) SAME_DIM(bn,in->dim,an-bn,seed->dim,0); out=new GridOutlet(this,0,in->dim,in->nt); in->set_chunk(an-bn-1); } GRID_FLOW { int an = in->dim->n; int bn = seed?seed->dim->n:0; long yn = in->dim->v[an-bn-1]; long zn = in->dim->prod(an-bn); long factor = in->factor(); T buf[n]; COPY(buf,data,n); if (seed) { for (long i=0; iscan(zn,yn,(T *)*seed,buf+i); } else { T neu; op->on(*buf)->neutral(&neu,at_left); T seed[zn]; op_put->map(zn,seed,neu); for (long i=0; iscan(zn,yn, seed,buf+i); } out->send(n,buf); } GRID_END \end class {install("#scan",1,1);} //**************************************************************** // L is a Dim[*si,sj, *ss] // R is a Dim[ sj,*sk,*ss] // Seed is a Dim[ *ss] // result is a Dim[*si, *sk,*ss] // Currently *ss can only be = Dim[] \class GridInner : FObject { \attr Numop *op; \attr Numop *fold; \attr PtrGrid seed; PtrGrid r; PtrGrid r2; // temporary bool use_dot; \constructor (Grid *r=0) { this->op = op_mul; this->fold = op_add; this->seed = new Grid(new Dim(),int32_e,true); this->r = r ? r : new Grid(new Dim(),int32_e,true); } \grin 0 \grin 1 }; // let's see this as a matrix product like L[i,j]*R[j,k] in Einstein notation // L: matrix of size si by sj // R: matrix of size sj by sk // LR: matrix of size si by sk template void inner_child_a (T *as, T *bs, int sj, int sk, int chunk) { for (int j=0; jmap(sk,as,*bs);} template void inner_child_b (T *as, T *bs, int sj, int chunk) { for (int j=0; jmap(sk,as,*bs);} // Inner product in a Module on the (+,*) Ring // | BBBBB // j BBBBB // | BBBBB // --j--*---k--- // AAAAA CCCCC template void dot_add_mul (long sk, long sj, T *cs, T *as, T *bs) { for (long k=0; k void dot_add_mul (long sk, T *cs, T *as, T *bs) { for (long k=0; k void dot_add_mul (T *cs, T *as, T *bs) { for (long k=0; k a=in->dim, b=r->dim; if (a->n<1) RAISE("a: minimum 1 dimension"); if (b->n<1) RAISE("b: minimum 1 dimension"); if (seed->dim->n != 0) RAISE("seed must be a scalar"); int n = a->n+b->n-2; SAME_DIM(1,a,a->n-1,b,0); int32 v[n]; COPY(v,a->v,a->n-1); COPY(v+a->n-1,b->v+1,b->n-1); out=new GridOutlet(this,0,new Dim(n,v),in->nt); in->set_chunk(a->n-1); long sjk=r->dim->prod(), sj=in->factor(), sk=sjk/sj; long chunk = GridOutlet::MAX_PACKET_SIZE/sjk; T *rdata = (T *)*r; r2=new Grid(new Dim(chunk*sjk),r->nt); T *buf3 = (T *)*r2; for (long i=0; idim->n==0 && *(T *)*seed==0; } GRID_FLOW { long sjk=r->dim->prod(), sj=in->factor(), sk=sjk/sj; long chunk = GridOutlet::MAX_PACKET_SIZE/sjk; T buf [chunk*sk]; T buf2[chunk*sk]; long off = chunk; if (use_dot) { while (n) { if (chunk*sj>n) chunk=n/sj; if (sj<=4 && sk<=4) switch ((sj-1)*4+(sk-1)) { #define DOT_ADD_MUL_3(sj,sk) for (int i=0; i( buf2+sk*i,data+sj*i,(T *)*r); case 0: DOT_ADD_MUL_3(1,1); break; case 1: DOT_ADD_MUL_3(1,2); break; case 2: DOT_ADD_MUL_3(1,3); break; case 3: DOT_ADD_MUL_3(1,4); break; case 4: DOT_ADD_MUL_3(2,1); break; case 5: DOT_ADD_MUL_3(2,2); break; case 6: DOT_ADD_MUL_3(2,3); break; case 7: DOT_ADD_MUL_3(2,4); break; case 8: DOT_ADD_MUL_3(3,1); break; case 9: DOT_ADD_MUL_3(3,2); break; case 10: DOT_ADD_MUL_3(3,3); break; case 11: DOT_ADD_MUL_3(3,4); break; case 12: DOT_ADD_MUL_3(4,1); break; case 13: DOT_ADD_MUL_3(4,2); break; case 14: DOT_ADD_MUL_3(4,3); break; case 15: DOT_ADD_MUL_3(4,4); break; } else switch (sj) { #define DOT_ADD_MUL_2(sj) for (int i=0; i(sk,buf2+sk*i,data+sj*i,(T *)*r); case 1: DOT_ADD_MUL_2(1); break; case 2: DOT_ADD_MUL_2(2); break; case 3: DOT_ADD_MUL_2(3); break; case 4: DOT_ADD_MUL_2(4); break; default:for (int i=0; isend(chunk*sk,buf2); n-=chunk*sj; data+=chunk*sj; } } else { while (n) { if (chunk*sj>n) chunk=n/sj; op_put->map(chunk*sk,buf2,*(T *)*seed); for (long i=0; i(buf,data+i,sj,chunk); break; case 2: inner_child_b(buf,data+i,sj,chunk); break; case 3: inner_child_b(buf,data+i,sj,chunk); break; case 4: inner_child_b(buf,data+i,sj,chunk); break; default: inner_child_a(buf,data+i,sj,sk,chunk); } op->zip(chunk*sk,buf,(T *)*r2+i*off*sk); fold->zip(chunk*sk,buf2,buf); } out->send(chunk*sk,buf2); n-=chunk*sj; data+=chunk*sj; } } } GRID_FINISH { r2=0; } GRID_END GRID_INPUT(GridInner,1,r) {} GRID_END \end class {install("#inner",2,1);} /* **************************************************************** */ /*{ Dim[*As],Dim[*Bs] -> Dim[*As,*Bs] }*/ \class GridOuter : FObject { \attr Numop *op; PtrGrid r; \constructor (Numop *op, Grid *r=0) { this->op = op; this->r = r ? r : new Grid(new Dim(),int32_e,true); } \grin 0 \grin 1 }; GRID_INLET(GridOuter,0) { SAME_TYPE(in,r); P a = in->dim; P b = r->dim; int n = a->n+b->n; int32 v[n]; COPY(v,a->v,a->n); COPY(v+a->n,b->v,b->n); out=new GridOutlet(this,0,new Dim(n,v),in->nt); } GRID_FLOW { long b_prod = r->dim->prod(); if (b_prod > 4) { T buf[b_prod]; while (n) { for (long j=0; jzip(b_prod,buf,(T *)*r); out->send(b_prod,buf); data++; n--; } return; } n*=b_prod; T *buf = NEWBUF(T,n); T buf2[b_prod*64]; for (int i=0; i<64; i++) COPY(buf2+i*b_prod,(T *)*r,b_prod); switch (b_prod) { #define Z buf[k++]=data[i] case 1: for (long i=0,k=0; kzip(ch,buf+j,buf2); op->zip(n-nn,buf+nn,buf2); out->give(n,buf); } GRID_END GRID_INPUT(GridOuter,1,r) {} GRID_END \end class {install("#outer",2,1); add_creator("@outer");} //**************************************************************** //{ Dim[],Dim[],Dim[] -> Dim[A] } or //{ Dim[B],Dim[B],Dim[B] -> Dim[*As,B] } \class GridFor : FObject { \attr PtrGrid from; \attr PtrGrid to; \attr PtrGrid step; \constructor (Grid *from, Grid *to, Grid *step) { this->from.constrain(expect_max_one_dim); this->to .constrain(expect_max_one_dim); this->step.constrain(expect_max_one_dim); this->from=from; this->to =to; this->step=step; } \decl 0 set (Grid *r=0); \decl 0 bang (); \grin 0 int \grin 1 int \grin 2 int template void trigger (T bogus); }; template void GridFor::trigger (T bogus) { int n = from->dim->prod(); int32 nn[n+1]; T x[64*n]; T *fromb = (T *)*from; T * tob = (T *)*to ; T *stepb = (T *)*step; T to2[n]; for (int i=step->dim->prod()-1; i>=0; i--) if (!stepb[i]) RAISE("step must not contain zeroes"); for (int i=0; i d; if (from->dim->n==0) { d = new Dim(*nn); } else { nn[n]=n; d = new Dim(n+1,nn); } int total = d->prod(); out=new GridOutlet(this,0,d,from->nt); if (total==0) return; int k=0; for(int d=0;;d++) { // here d is the dim# to reset; d=n for none for(;dsend(k,x); k=0; COPY(x,x+63*n,n);} else { COPY(x+k,x+k-n,n);} d--; // here d is the dim# to increment for(;;d--) { if (d<0) goto end; x[k+d]+=stepb[d]; if (x[k+d]!=to2[d]) break; } } end: if (k) out->send(k,x); } \def 0 bang () { SAME_TYPE(from,to); SAME_TYPE(from,step); if (!from->dim->equal(to->dim) || !to->dim->equal(step->dim)) RAISE("dimension mismatch: from:%s to:%s step:%s", from->dim->to_s(),to->dim->to_s(),step->dim->to_s()); #define FOO(T) trigger((T)0); TYPESWITCH_JUSTINT(from->nt,FOO,); #undef FOO } \def 0 set (Grid *r) { from=new Grid(argv[0]); } GRID_INPUT(GridFor,2,step) {} GRID_END GRID_INPUT(GridFor,1,to) {} GRID_END GRID_INPUT(GridFor,0,from) {_0_bang(0,0);} GRID_END \end class {install("#for",3,1); add_creator("@for");} //**************************************************************** \class GridFinished : FObject { \constructor () {} \grin 0 }; GRID_INLET(GridFinished,0) { in->set_mode(0); } GRID_FINISH { outlet_bang(bself->outlets[0]); } GRID_END \end class {install("#finished",1,1); add_creator("@finished");} \class GridDim : FObject { \constructor () {} \grin 0 }; GRID_INLET(GridDim,0) { GridOutlet out(this,0,new Dim(in->dim->n)); out.send(in->dim->n,in->dim->v); in->set_mode(0); } GRID_END \end class {install("#dim",1,1); add_creator("@dim");} \class GridType : FObject { \constructor () {} \grin 0 }; GRID_INLET(GridType,0) { outlet_symbol(bself->outlets[0],gensym(const_cast(number_type_table[in->nt].name))); in->set_mode(0); } GRID_END \end class {install("#type",1,1); add_creator("@type");} //**************************************************************** //{ Dim[*As],Dim[B] -> Dim[*Cs] } \class GridRedim : FObject { \attr P dim; PtrGrid dim_grid; PtrGrid temp; // temp->dim is not of the same shape as dim ~GridRedim() {} \constructor (Grid *d) { dim_grid.constrain(expect_dim_dim_list); dim_grid=d; dim = dim_grid->to_dim(); // if (!dim->prod()) RAISE("target grid size must not be zero"); } \grin 0 \grin 1 int32 }; GRID_INLET(GridRedim,0) { long a=in->dim->prod(), b=dim->prod(); if (ant); out=new GridOutlet(this,0,dim,in->nt); } GRID_FLOW { long i = in->dex; if (!temp) { long n2 = min(n,dim->prod()-i); if (n2>0) out->send(n2,data); // discard other values if any } else { long n2 = min(n,in->dim->prod()-i); COPY((T *)*temp+i,data,n2); if (n2>0) out->send(n2,data); } } GRID_FINISH { if (!!temp) { long a = in->dim->prod(), b = dim->prod(); if (a) { for (long i=a; isend(min(a,b-i),(T *)*temp); } else { T foo[1]={0}; for (long i=0; isend(1,foo); } } temp=0; } GRID_END GRID_INPUT(GridRedim,1,dim_grid) { P d = dim_grid->to_dim(); // if (!d->prod()) RAISE("target grid size must not be zero"); else post("d->prod=%d",d->prod()); dim = d; } GRID_END \end class {install("#redim",2,1); add_creator("@redim");} //**************************************************************** \class GridJoin : FObject { \attr int which_dim; PtrGrid r; \grin 0 \grin 1 \constructor (int which_dim=-1, Grid *r=0) { this->which_dim = which_dim; this->r=r; } }; GRID_INLET(GridJoin,0) { NOTEMPTY(r); SAME_TYPE(in,r); P d = in->dim; if (d->n != r->dim->n) RAISE("wrong number of dimensions"); int w = which_dim; if (w<0) w+=d->n; if (w<0 || w>=d->n) RAISE("can't join on dim number %d on %d-dimensional grids", which_dim,d->n); int32 v[d->n]; for (int i=0; in; i++) { v[i] = d->get(i); if (i==w) { v[i]+=r->dim->v[i]; } else { if (v[i]!=r->dim->v[i]) RAISE("dimensions mismatch: dim #%i, left is %d, right is %d",i,v[i],r->dim->v[i]); } } out=new GridOutlet(this,0,new Dim(d->n,v),in->nt); in->set_chunk(w); } GRID_FLOW { int w = which_dim; if (w<0) w+=in->dim->n; long a = in->factor(); long b = r->dim->prod(w); T *data2 = (T *)*r + in->dex*b/a; if (a==3 && b==1) { int m = n+n*b/a; T data3[m]; T *data4 = data3; while (n) { SCOPY(data4,data,3); SCOPY(data4+3,data2,1); n-=3; data+=3; data2+=1; data4+=4; } out->send(m,data3); } else if (a+b<=16) { int m = n+n*b/a; T data3[m]; int i=0; while (n) { COPY(data3+i,data,a); data+=a; i+=a; n-=a; COPY(data3+i,data2,b); data2+=b; i+=b; } out->send(m,data3); } else { while (n) { out->send(a,data); out->send(b,data2); data+=a; data2+=b; n-=a; } } } GRID_FINISH { if (in->dim->prod()==0) out->send(r->dim->prod(),(T *)*r); } GRID_END GRID_INPUT(GridJoin,1,r) {} GRID_END \end class {install("@join",2,1);} //**************************************************************** \class GridGrade : FObject { \constructor () {} \grin 0 }; typedef int (*comparator_t)(const void *, const void *); template struct GradeFunction { static int comparator (T **a, T **b) {return **a-**b;}}; #define FOO(T) \ template <> struct GradeFunction { \ static int comparator (T **a, T **b) {T x = **a-**b; return x<0 ? -1 : x>0;}}; FOO(int64) FOO(float32) FOO(float64) #undef FOO GRID_INLET(GridGrade,0) { out=new GridOutlet(this,0,in->dim); in->set_chunk(in->dim->n-1); } GRID_FLOW { long m = in->factor(); T *foo[m]; T bar[m]; for (; n; n-=m,data+=m) { for (int i=0; i::comparator); for (int i=0; isend(m,bar); } } GRID_END \end class {install("#grade",1,1); add_creator("@grade");} //**************************************************************** //\class GridMedian : FObject //**************************************************************** \class GridTranspose : FObject { \attr int dim1; \attr int dim2; int d1,d2,na,nb,nc,nd; // temporaries \constructor (int dim1=0, int dim2=1) { this->dim1 = dim1; this->dim2 = dim2; } \decl 1 float (int dim1); \decl 2 float (int dim2); \grin 0 }; \def 1 float (int dim1) { this->dim1=dim1; } \def 2 float (int dim2) { this->dim2=dim2; } GRID_INLET(GridTranspose,0) { int32 v[in->dim->n]; COPY(v,in->dim->v,in->dim->n); d1=dim1; d2=dim2; if (d1<0) d1+=in->dim->n; if (d2<0) d2+=in->dim->n; if (d1>=in->dim->n || d2>=in->dim->n || d1<0 || d2<0) RAISE("would swap dimensions %d and %d but this grid has only %d dimensions", dim1,dim2,in->dim->n); memswap(v+d1,v+d2,1); if (d1==d2) { out=new GridOutlet(this,0,new Dim(in->dim->n,v), in->nt); } else { nd = in->dim->prod(1+max(d1,d2)); nc = in->dim->v[max(d1,d2)]; nb = in->dim->prod(1+min(d1,d2))/nc/nd; na = in->dim->v[min(d1,d2)]; out=new GridOutlet(this,0,new Dim(in->dim->n,v), in->nt); in->set_chunk(min(d1,d2)); } // Turns a Grid[*,na,*nb,nc,*nd] into a Grid[*,nc,*nb,na,*nd]. } GRID_FLOW { //T res[na*nb*nc*nd]; T *res = NEWBUF(T,na*nb*nc*nd); if (dim1==dim2) { out->send(n,data); return; } int prod = na*nb*nc*nd; for (; n; n-=prod, data+=prod) { for (long a=0; asend(na*nb*nc*nd,res); } DELBUF(res); //!@#$ if an exception was thrown by out->send, this never gets done } GRID_END \end class {install("#transpose",3,1); add_creator("@transpose");} //**************************************************************** \class GridReverse : FObject { \attr int dim1; // dimension to act upon int d; // temporaries \constructor (int dim1=0) {this->dim1 = dim1;} \decl 1 float (int dim1); \grin 0 }; \def 1 float (int dim1) { this->dim1=dim1; } GRID_INLET(GridReverse,0) { d=dim1; if (d<0) d+=in->dim->n; if (d>=in->dim->n || d<0) RAISE("would reverse dimension %d but this grid has only %d dimensions", dim1,in->dim->n); out=new GridOutlet(this,0,new Dim(in->dim->n,in->dim->v), in->nt); in->set_chunk(d); } GRID_FLOW { long f1=in->factor(), f2=in->dim->prod(d+1); while (n) { long hf1=f1/2; T *data2 = data+f1-f2; for (long i=0; isend(f1,data); data+=f1; n-=f1; } } GRID_END \end class {install("#reverse",2,1);} //**************************************************************** \class GridCentroid : FObject { \constructor () {} \grin 0 int int sumx,sumy,sum,y; // temporaries }; GRID_INLET(GridCentroid,0) { if (in->dim->n != 3) RAISE("expecting 3 dims"); if (in->dim->v[2] != 1) RAISE("expecting 1 channel"); in->set_chunk(1); out=new GridOutlet(this,0,new Dim(2), in->nt); sumx=0; sumy=0; sum=0; y=0; } GRID_FLOW { int sx = in->dim->v[1]; while (n) { for (int x=0; xsend(2,blah); outlet_float(bself->outlets[1],blah[0]); outlet_float(bself->outlets[2],blah[1]); } GRID_END \end class {install("#centroid",1,3);} //**************************************************************** static void expect_pair (P dim) {if (dim->prod()!=2) RAISE("expecting only two numbers. Dim(2)");} \class GridMoment : FObject { \constructor (int order=1) { offset.constrain(expect_pair); //t_atom2 a[2] = {t_atom2(0),t_atom2(0)}; t_atom a[2]; SETFLOAT(a,0); SETFLOAT(a+1,0); offset=new Grid(2,a,int32_e); if (order!=1 && order!=2) RAISE("supports only orders 1 and 2 for now"); this->order=order; } \grin 0 int \grin 1 int \attr int order; // order \attr PtrGrid offset; int64 sumy,sumxy,sumx,sum,y; // temporaries }; GRID_INLET(GridMoment,0) { if (in->dim->n != 3) RAISE("expecting 3 dims"); if (in->dim->v[2] != 1) RAISE("expecting 1 channel"); in->set_chunk(1); switch (order) { case 1: out=new GridOutlet(this,0,new Dim(2 ), in->nt); break; case 2: out=new GridOutlet(this,0,new Dim(2,2), in->nt); break; default: RAISE("supports only orders 1 and 2 for now"); } sumx=0; sumy=0; sumxy=0; sum=0; y=0; } GRID_FLOW { int sx = in->dim->v[1]; int oy = ((int*)*offset)[0]; int ox = ((int*)*offset)[1]; while (n) { switch (order) { case 1: for (int x=0; xsend(2,blah); break; case 2: /* covariance matrix */ blah[0] = sum ? sumy/sum : 0; blah[1] = sum ? sumxy/sum : 0; blah[2] = sum ? sumxy/sum : 0; blah[3] = sum ? sumx/sum : 0; out->send(4,blah); break; } } GRID_END GRID_INPUT(GridMoment,1,offset) {} GRID_END \end class {install("#moment",2,1);} //**************************************************************** \class GridLabeling : FObject { \grin 0 \attr int form(); \attr int form_val; \constructor (int form=0) {form_val=form; initialize3();} void initialize3(); }; struct Stats { int64 yy,yx,xx,y,x,area; int64 x1,x2; Stats() {yy=yx=xx=y=x=area=0;} }; #define AT(y,x) dat[(y)*sx+(x)] template void flood_fill(T *dat, int sy, int sx, int y, int x, Stats *stat, int label, int form) { /* find x1,x2 such that all the x of that horizontal segment are x1<=x=0; x1--) if (AT(y,x1)!=1) break; x1++; if (form==0) { for (x=x1; xyy += y*y; stat->y += y; stat->yx += y*x; stat->area++; stat->xx += x*x; stat->x += x; } for (x=x1; x0 && AT(y-1,x)==1) flood_fill(dat,sy,sx,y-1,x,stat,label,form); if (yy=y; stat->x1=x1; stat->x2=x2; } } GRID_INLET(GridLabeling,0) { if (in->dim->n<2 || in->dim->prod(2)!=1) RAISE("requires dim (y,x) or (y,x,1)"); in->set_chunk(0); } GRID_FLOW { int sy=in->dim->v[0], sx=in->dim->v[1]; T *dat = NEWBUF(T,n); for (int i=0; int); out->send(n,dat); DELBUF(dat); } GRID_END \def int form() {return form_val;} \def 0 form(int form) { if (form<0 || form>1) RAISE("form must be 0 or 1, not %d",form); form_val=form; initialize3(); } void GridLabeling::initialize3() { bself->noutlets_set(form_val ? 2 : 4); } \end class {install("#labeling",1,0);} //**************************************************************** \class GridPerspective : FObject { \attr int32 z; \grin 0 \constructor (int32 z=256) {this->z=z;} }; GRID_INLET(GridPerspective,0) { int n = in->dim->n; int32 v[n]; COPY(v,in->dim->v,n); v[n-1]--; in->set_chunk(in->dim->n-1); out=new GridOutlet(this,0,new Dim(n,v),in->nt); } GRID_FLOW { int m = in->factor(); for (; n; n-=m,data+=m) { op_mul->map(m-1,data,(T)z); op_div->map(m-1,data,data[m-1]); out->send(m-1,data); } } GRID_END \end class {install("#perspective",1,1); add_creator("@perspective");} //**************************************************************** \class GridBorder : FObject { \attr P diml; \attr P dimr; PtrGrid diml_grid; PtrGrid dimr_grid; \grin 0 \grin 1 int \grin 2 int \constructor (Grid *dl=0, Grid *dr=0) { t_atom a[2]; SETFLOAT(a+0,1); SETFLOAT(a+1,1); SETFLOAT(a+2,0); diml_grid=dl?dl:new Grid(3,a,int32_e); dimr_grid=dr?dr:new Grid(3,a,int32_e); diml = diml_grid->to_dim(); dimr = dimr_grid->to_dim(); } }; GRID_INLET(GridBorder,0) { int n = in->dim->n; if (n!=3) RAISE("only 3 dims supported for now"); if (diml->n != n) RAISE("diml mismatch"); if (dimr->n != n) RAISE("dimr mismatch"); if (diml->v[2] || dimr->v[2]) RAISE("can't augment channels (todo)"); int32 v[n]; for (int i=0; idim->v[i]+diml->v[i]+dimr->v[i]; in->set_chunk(0); out=new GridOutlet(this,0,new Dim(n,v),in->nt); } GRID_FLOW { int sy = in->dim->v[0]; int sx = in->dim->v[1]; int zx = sx+diml->v[1]+dimr->v[1]; int sc = in->dim->v[2]; int zc = sc+diml->v[2]+dimr->v[2]; int sxc = sx*sc; int zxc = zx*zc; int32 duh[zxc]; for (int x=0; xv[0]; y++) out->send(zxc,duh); for (int y=0; ysend(diml->v[1]*sc,duh); out->send(sxc,data+y*sxc); out->send(dimr->v[1]*sc,duh); } for (int i=0; iv[0]; i++) out->send(zxc,duh); } GRID_END GRID_INPUT(GridBorder,1,diml_grid) {diml = diml_grid->to_dim();} GRID_END GRID_INPUT(GridBorder,2,dimr_grid) {dimr = dimr_grid->to_dim();} GRID_END \end class {install("#border",3,1);} static void expect_picture (P d) { if (d->n!=3) RAISE("(height,width,chans) dimensions please");} static void expect_rgb_picture (P d) { expect_picture(d); if (d->get(2)!=3) RAISE("(red,green,blue) channels please");} static void expect_rgba_picture (P d) { expect_picture(d); if (d->get(2)!=4) RAISE("(red,green,blue,alpha) channels please");} //**************************************************************** //{ Dim[A,B,*Cs],Dim[D,E] -> Dim[A,B,*Cs] } static void expect_convolution_matrix (P d) { if (d->n != 2) RAISE("only exactly two dimensions allowed for now (got %d)", d->n); } // entry in a compiled convolution kernel struct PlanEntry { long y,x; bool neutral; }; \class GridConvolve : FObject { \attr Numop *op; \attr Numop *fold; \attr PtrGrid seed; \attr PtrGrid b; \attr bool wrap; \attr bool anti; PtrGrid a; int plann; PlanEntry *plan; int margx,margy; // margins \constructor (Grid *r=0) { plan=0; b.constrain(expect_convolution_matrix); plan=0; this->op = op_mul; this->fold = op_add; this->seed = new Grid(new Dim(),int32_e,true); this->b= r ? r : new Grid(new Dim(1,1),int32_e,true); this->wrap = true; this->anti = true; } \grin 0 \grin 1 template void copy_row (T *buf, long sx, long y, long x); template void make_plan (T bogus); ~GridConvolve () {if (plan) delete[] plan;} }; template void GridConvolve::copy_row (T *buf, long sx, long y, long x) { long day = a->dim->get(0), dax = a->dim->get(1), dac = a->dim->prod(2); y=mod(y,day); x=mod(x,dax); T *ap = (T *)*a + y*dax*dac; while (sx) { long sx1 = min(sx,dax-x); COPY(buf,ap+x*dac,sx1*dac); x=0; buf += sx1*dac; sx -= sx1; } } template void GridConvolve::make_plan (T bogus) { P da = a->dim, db = b->dim; long dby = db->get(0); long dbx = db->get(1); if (plan) delete[] plan; plan = new PlanEntry[dbx*dby]; long i=0; for (long y=0; yon(rh)->is_neutral( rh,at_right); bool absorbent = op->on(rh)->is_absorbent(rh,at_right); T foo[1]={0}; if (absorbent) { op->map(1,foo,rh); absorbent = fold->on(rh)->is_neutral(foo[0],at_right); } if (absorbent) continue; plan[i].y = y; plan[i].x = x; plan[i].neutral = neutral; i++; } } plann = i; } GRID_INLET(GridConvolve,0) { SAME_TYPE(in,b); SAME_TYPE(in,seed); P da=in->dim, db=b->dim; if (!db) RAISE("right inlet has no grid"); if (!seed) RAISE("seed missing"); if (db->n != 2) RAISE("right grid must have two dimensions"); if (da->n < 2) RAISE("left grid has less than two dimensions"); if (seed->dim->n != 0) RAISE("seed must be scalar"); if (da->get(0) < db->get(0)) RAISE("grid too small (y): %d < %d", da->get(0), db->get(0)); if (da->get(1) < db->get(1)) RAISE("grid too small (x): %d < %d", da->get(1), db->get(1)); margy = (db->get(0)-1)/2; margx = (db->get(1)-1)/2; //if (a) post("for %p, a->dim=%s da=%s",this,a->dim->to_s(),da->to_s()); if (!a || !a->dim->equal(da)) a=new Grid(da,in->nt); // with this condition it's 2% faster on Linux but takes more RAM. //a=new Grid(da,in->nt); // with this condition it's 2% faster but takes more RAM. int v[da->n]; COPY(v,da->v,da->n); if (!wrap) {v[0]-=db->v[0]-1; v[1]-=db->v[1]-1;} out=new GridOutlet(this,0,new Dim(da->n,v),in->nt); } GRID_FLOW { COPY((T *)*a+in->dex, data, n); } GRID_FINISH { make_plan((T)0); long dbx = b->dim->get(1); long dby = b->dim->get(0); long day = out->dim->get(0); long n = out->dim->prod(1); long sx = out->dim->get(1)+dbx-1; long sxc = out->dim->prod(2)*sx; T buf[n]; T buf2[sxc]; T orh=0; for (long ay=0; aymap(n,buf,*(T *)*seed); for (long i=0; imap(sxc,buf2,rh); } fold->zip(n,buf,buf2+bx*out->dim->prod(2)); orh=rh; } out->send(n,buf); } //a=0; // comment this out when trying to recycle a (use the dim->equal above) } GRID_END GRID_INPUT(GridConvolve,1,b) {} GRID_END \end class {install("#convolve",2,1);} /* ---------------------------------------------------------------- */ /* "#scale_by" does quick scaling of pictures by integer factors */ /*{ Dim[A,B,3] -> Dim[C,D,3] }*/ static void expect_scale_factor (P dim) { if (dim->prod()!=1 && dim->prod()!=2) RAISE("expecting only one or two numbers"); } \class GridScaleBy : FObject { \attr PtrGrid scale; // integer scale factor int scaley; int scalex; \constructor (Grid *factor=0) { scale.constrain(expect_scale_factor); t_atom a[1]; SETFLOAT(a,2); scale = factor?factor:new Grid(1,a,int32_e); prepare_scale_factor(); } \grin 0 \grin 1 void prepare_scale_factor () { scaley = ((int32 *)*scale)[0]; scalex = ((int32 *)*scale)[scale->dim->prod()==1 ? 0 : 1]; if (scaley<1) scaley=2; if (scalex<1) scalex=2; } }; GRID_INLET(GridScaleBy,0) { P a = in->dim; expect_picture(a); out=new GridOutlet(this,0,new Dim(a->get(0)*scaley,a->get(1)*scalex,a->get(2)),in->nt); in->set_chunk(1); } GRID_FLOW { int rowsize = in->dim->prod(1); T buf[rowsize*scalex]; int chans = in->dim->get(2); #define Z(z) buf[p+z]=data[i+z] for (; n>0; data+=rowsize, n-=rowsize) { int p=0; #define LOOP(z) \ for (int i=0; isend(rowsize*scalex,buf); } #undef Z } GRID_END GRID_INPUT(GridScaleBy,1,scale) { prepare_scale_factor(); } GRID_END \end class {install("#scale_by",2,1); add_creator("@scale_by");} // ---------------------------------------------------------------- //{ Dim[A,B,3] -> Dim[C,D,3] } \class GridDownscaleBy : FObject { \attr PtrGrid scale; \attr bool smoothly; int scaley; int scalex; PtrGrid temp; \constructor (Grid *factor=0, t_symbol *option=0) { scale.constrain(expect_scale_factor); t_atom a[1]; SETFLOAT(a,2); scale = factor?factor:new Grid(1,a,int32_e); prepare_scale_factor(); smoothly = option==gensym("smoothly"); } \grin 0 \grin 1 void prepare_scale_factor () { scaley = ((int32 *)*scale)[0]; scalex = ((int32 *)*scale)[scale->dim->prod()==1 ? 0 : 1]; if (scaley<1) scaley=2; if (scalex<1) scalex=2; } }; GRID_INLET(GridDownscaleBy,0) { P a = in->dim; if (a->n!=3) RAISE("(height,width,chans) please"); out=new GridOutlet(this,0,new Dim(a->get(0)/scaley,a->get(1)/scalex,a->get(2)),in->nt); in->set_chunk(1); // i don't remember why two rows instead of just one. temp=new Grid(new Dim(2,in->dim->get(1)/scalex,in->dim->get(2)),in->nt); } GRID_FLOW { int rowsize = in->dim->prod(1); int rowsize2 = temp->dim->prod(1); T *buf = (T *)*temp; //!@#$ maybe should be something else than T ? int xinc = in->dim->get(2)*scalex; int y = in->dex / rowsize; int chans=in->dim->get(2); #define Z(z) buf[p+z]+=data[i+z] if (smoothly) { while (n>0) { if (y%scaley==0) CLEAR(buf,rowsize2); #define LOOP(z) \ for (int i=0,p=0; pdim) { op_div->map(rowsize2,buf,(T)(scalex*scaley)); out->send(rowsize2,buf); CLEAR(buf,rowsize2); } data+=rowsize; n-=rowsize; } #undef Z } else { #define Z(z) buf[p+z]=data[i+z] for (; n>0 && out->dim; data+=rowsize, n-=rowsize,y++) { if (y%scaley!=0) continue; #define LOOP(z) for (int i=0,p=0; pdim->get(2)) { case 1: LOOP(1) {Z(0);} break; case 2: LOOP(2) {Z(0);Z(1);} break; case 3: LOOP(3) {Z(0);Z(1);Z(2);} break; case 4: LOOP(4) {Z(0);Z(1);Z(2);Z(3);} break; default:LOOP(chans) {for (int k=0; ksend(rowsize2,buf); } } #undef Z } GRID_END GRID_INPUT(GridDownscaleBy,1,scale) { prepare_scale_factor(); } GRID_END \end class {install("#downscale_by",2,1); add_creator("@downscale_by");} //**************************************************************** \class GridLayer : FObject { PtrGrid r; \constructor () {r.constrain(expect_rgb_picture);} \grin 0 int \grin 1 int }; GRID_INLET(GridLayer,0) { NOTEMPTY(r); SAME_TYPE(in,r); P a = in->dim; expect_rgba_picture(a); if (a->get(1)!=r->dim->get(1)) RAISE("same width please"); if (a->get(0)!=r->dim->get(0)) RAISE("same height please"); in->set_chunk(2); out=new GridOutlet(this,0,r->dim); } GRID_FLOW { T *rr = ((T *)*r) + in->dex*3/4; T foo[n*3/4]; #define COMPUTE_ALPHA(c,a) \ foo[j+c] = (data[i+c]*data[i+a] + rr[j+c]*(256-data[i+a])) >> 8 for (int i=0,j=0; isend(n*3/4,foo); } GRID_END GRID_INPUT(GridLayer,1,r) {} GRID_END \end class {install("#layer",2,1); add_creator("@layer");} // **************************************************************** // pad1,pad2 only are there for 32-byte alignment struct Line {int32 y1,x1,y2,x2,x,m,ox,pad2;}; static void expect_polygon (P d) { if (d->n!=2 || d->get(1)!=2) RAISE("expecting Dim[n,2] polygon"); } enum DrawMode {DRAW_FILL,DRAW_LINE,DRAW_POINT}; enum OmitMode {OMIT_NONE,OMIT_LAST,OMIT_ODD}; DrawMode convert(const t_atom &x, DrawMode *foo) { t_symbol *s = convert(x,(t_symbol **)0); if (s==gensym("fill")) return DRAW_FILL; if (s==gensym("line")) return DRAW_LINE; if (s==gensym("point")) return DRAW_POINT; RAISE("unknown DrawMode '%s' (want fill or line)",s->s_name); } OmitMode convert(const t_atom &x, OmitMode *foo) { t_symbol *s = convert(x,(t_symbol **)0); if (s==gensym("none")) return OMIT_NONE; if (s==gensym("last")) return OMIT_LAST; if (s==gensym("odd")) return OMIT_ODD; RAISE("unknown OmitMode '%s' (want none or last or odd)",s->s_name); } \class DrawPolygon : FObject { \attr Numop *op; \attr PtrGrid color; \attr PtrGrid polygon; \attr DrawMode draw; \attr OmitMode omit; PtrGrid color2; PtrGrid lines; int lines_start; int lines_stop; \constructor (Numop *op=op_put, Grid *color=0, Grid *polygon=0) { draw=DRAW_FILL; omit=OMIT_NONE; this->color.constrain(expect_max_one_dim); this->polygon.constrain(expect_polygon); this->op = op; if (color) this->color=color; if (polygon) {this->polygon=polygon; init_lines();} } \grin 0 \grin 1 \grin 2 int32 void init_lines(); void changed(t_symbol *s) {init_lines();} }; void DrawPolygon::init_lines () { if (!polygon) return; int tnl = polygon->dim->get(0); int nl = omit==OMIT_LAST ? tnl-1 : omit==OMIT_ODD ? (tnl+1)/2 : tnl; lines=new Grid(new Dim(nl,8), int32_e); Line *ld = (Line *)(int32 *)*lines; int32 *pd = *polygon; for (int i=0,j=0; ild[i].y2) memswap((int32 *)(ld+i)+0,(int32 *)(ld+i)+2,2); long dy = ld[i].y2-ld[i].y1; long dx = ld[i].x2-ld[i].x1; ld[i].m = dy ? (dx<<16)/dy : 0; } } static int order_by_starting_scanline (const void *a, const void *b) { return ((Line *)a)->y1 - ((Line *)b)->y1; } static int order_by_column (const void *a, const void *b) { return ((Line *)a)->x - ((Line *)b)->x; } GRID_INLET(DrawPolygon,0) { NOTEMPTY(color); NOTEMPTY(polygon); NOTEMPTY(lines); SAME_TYPE(in,color); if (in->dim->n!=3) RAISE("expecting 3 dimensions"); if (in->dim->get(2)!=color->dim->get(0)) RAISE("image does not have same number of channels as stored color"); out=new GridOutlet(this,0,in->dim,in->nt); lines_start = lines_stop = 0; in->set_chunk(1); int nl = lines->dim->get(0); qsort((int32 *)*lines,nl,sizeof(Line),order_by_starting_scanline); int cn = color->dim->prod(); color2=new Grid(new Dim(cn*16), color->nt); for (int i=0; i<16; i++) COPY((T *)*color2+cn*i,(T *)*color,cn); } GRID_FLOW { int nl = lines->dim->get(0); Line *ld = (Line *)(int32 *)*lines; int f = in->factor(); int y = in->dex/f; int cn = color->dim->prod(); T *cd = (T *)*color2; while (n) { while (lines_stop != nl && ld[lines_stop].y1<=y) { Line &l = ld[lines_stop]; l.x = l.x1 + (((y-l.y1)*l.m)>>16); lines_stop++; } if (draw!=DRAW_POINT) { int fudge = draw==DRAW_FILL?0:1; for (int i=lines_start; isend(f,data); } else { int32 xl = in->dim->get(1); T *data2 = NEWBUF(T,f); COPY(data2,data,f); for (int i=lines_start; i>16); } if (draw!=DRAW_POINT) qsort(ld+lines_start,lines_stop-lines_start,sizeof(Line),order_by_column); if (draw==DRAW_FILL) { for (int i=lines_start; i=xe) continue; /* !@#$ WHAT? */ while (xe-xs>=16) {op->zip(16*cn,data2+cn*xs,cd); xs+=16;} op->zip((xe-xs)*cn,data2+cn*xs,cd); } } else if (draw==DRAW_LINE) { for (int i=lines_start; i=xl && xe>=xl)) continue; xs = max(0,xs); xe = min(xl,xe); while (xe-xs>=16) {op->zip(16*cn,data2+cn*xs,cd); xs+=16;} op->zip((xe-xs)*cn,data2+cn*xs,cd); } } else { for (int i=lines_start; i=xl) continue; op->zip((xe-xs)*cn,data2+cn*xs,cd); } lines_start=lines_stop; } out->give(f,data2); } n-=f; data+=f; y++; } } GRID_END GRID_INPUT(DrawPolygon,1,color) {} GRID_END GRID_INPUT(DrawPolygon,2,polygon) {init_lines();} GRID_END \end class {install("#draw_polygon",3,1); add_creator("@draw_polygon");} //**************************************************************** static void expect_position(P d) { if (d->n!=1) RAISE("position should have 1 dimension, not %d", d->n); if (d->v[0]!=2) RAISE("position dim 0 should have 2 elements, not %d", d->v[0]); } \class DrawImage : FObject { \attr Numop *op; \attr PtrGrid image; \attr PtrGrid position; \attr bool alpha; \attr bool tile; \constructor (Numop *op=op_put, Grid *image=0, Grid *position=0) { alpha=false; tile=false; this->op = op; this->position.constrain(expect_position); this->image.constrain(expect_picture); if (image) this->image=image; if (position) this->position=position; else this->position=new Grid(new Dim(2),int32_e,true); } \grin 0 \grin 1 \grin 2 int32 // draw row # ry of right image in row buffer buf, starting at xs // overflow on both sides has to be handled automatically by this method template void draw_segment(T *obuf, T *ibuf, int ry, int x0); }; #define COMPUTE_ALPHA(c,a) obuf[j+(c)] = ibuf[j+(c)] + (rbuf[a])*(obuf[j+(c)]-ibuf[j+(c)])/256; #define COMPUTE_ALPHA4(b) \ COMPUTE_ALPHA(b+0,b+3); \ COMPUTE_ALPHA(b+1,b+3); \ COMPUTE_ALPHA(b+2,b+3); \ obuf[b+3] = rbuf[b+3] + (255-rbuf[b+3])*(ibuf[j+b+3])/256; template void DrawImage::draw_segment(T *obuf, T *ibuf, int ry, int x0) { if (ry<0 || ry>=image->dim->get(0)) return; // outside of image int sx = in[0]->dim->get(1), rsx = image->dim->get(1); int sc = in[0]->dim->get(2), rsc = image->dim->get(2); T *rbuf = (T *)*image + ry*rsx*rsc; if (x0>sx || x0<=-rsx) return; // outside of buffer int n=rsx; if (x0+n>sx) n=sx-x0; if (x0<0) { rbuf-=rsc*x0; n+=x0; x0=0; } if (alpha && rsc==4 && sc==3) { // RGB by RGBA //!@#$ optimise int j=sc*x0; for (; n; n--, rbuf+=4, j+=3) { op->zip(sc,obuf+j,rbuf); COMPUTE_ALPHA(0,3); COMPUTE_ALPHA(1,3); COMPUTE_ALPHA(2,3); } } else if (alpha && rsc==4 && sc==4) { // RGBA by RGBA op->zip(n*rsc,obuf+x0*rsc,rbuf); int j=sc*x0; for (; n>=4; n-=4, rbuf+=16, j+=16) { COMPUTE_ALPHA4(0);COMPUTE_ALPHA4(4); COMPUTE_ALPHA4(8);COMPUTE_ALPHA4(12); } for (; n; n--, rbuf+=4, j+=4) { COMPUTE_ALPHA4(0); } } else { // RGB by RGB, etc op->zip(n*rsc,obuf+sc*x0,rbuf); } } GRID_INLET(DrawImage,0) { NOTEMPTY(image); NOTEMPTY(position); SAME_TYPE(in,image); if (in->dim->n!=3) RAISE("expecting 3 dimensions"); if (image->dim->n!=3) RAISE("expecting 3 dimensions in right_hand"); int lchan = in->dim->get(2); int rchan = image->dim->get(2); if (alpha && rchan!=4) { RAISE("alpha mode works only with 4 channels in right_hand"); } if (lchan != rchan-(alpha?1:0) && lchan != rchan) { RAISE("right_hand has %d channels, alpha=%d, left_hand has %d, expecting %d or %d", rchan, alpha?1:0, lchan, rchan-(alpha?1:0), rchan); } out=new GridOutlet(this,0,in->dim,in->nt); in->set_chunk(1); } GRID_FLOW { int f = in->factor(); int y = in->dex/f; if (position->nt != int32_e) RAISE("position has to be int32"); int py = ((int32*)*position)[0], rsy = image->dim->v[0]; int px = ((int32*)*position)[1], rsx = image->dim->v[1], sx=in->dim->get(1); for (; n; y++, n-=f, data+=f) { int ty = div2(y-py,rsy); if (tile || ty==0) { T *data2 = NEWBUF(T,f); COPY(data2,data,f); if (tile) { for (int x=px-div2(px+rsx-1,rsx)*rsx; xgive(f,data2); } else { out->send(f,data); } } } GRID_END GRID_INPUT(DrawImage,1,image) {} GRID_END GRID_INPUT(DrawImage,2,position) {} GRID_END \end class {install("#draw_image",3,1); add_creator("@draw_image");} //**************************************************************** // Dim[*A],Dim[*B],Dim[C,size(A)-size(B)] -> Dim[*A] /* NOT FINISHED */ \class GridDrawPoints : FObject { \attr Numop *op; \attr PtrGrid color; \attr PtrGrid points; \grin 0 \grin 1 int32 \grin 2 int32 \constructor (Numop *op=op_put, Grid *color=0, Grid *points=0) { this->op = op; if (color) this->color=color; if (points) this->points=points; } }; GRID_INPUT(GridDrawPoints,1,color) {} GRID_END GRID_INPUT(GridDrawPoints,2,points) {} GRID_END GRID_INLET(GridDrawPoints,0) { NOTEMPTY(color); NOTEMPTY(points); SAME_TYPE(in,color); out=new GridOutlet(this,0,in->dim,in->nt); if (points->dim->n!=2) RAISE("points should be a 2-D grid"); if (points->dim->v[1] != in->dim->n - color->dim->n) RAISE("wrong number of dimensions"); in->set_chunk(0); } GRID_FLOW { long m = points->dim->v[1]; long cn = in->dim->prod(-color->dim->n); /* size of color (RGB=3, greyscale=1, ...) */ int32 *pdata = (int32 *)points->data; T *cdata = (T *)color->data; for (long i=0; im; k++) off = off*in->dim->v[k] + pdata[i*points->dim->v[1]+k]; off *= cn; for (long j=0; jsend(data); } GRID_END \end class {install("#draw_points",3,1);} //**************************************************************** \class GridPolygonize : FObject { \constructor () {} \grin 0 }; GRID_INLET(GridPolygonize,0) { if (in->dim->n<2 || in->dim->prod(2)!=1) RAISE("requires dim (y,x) or (y,x,1)"); in->set_chunk(0); } GRID_FLOW { /* WRITE ME */ } GRID_END \end class {install("#polygonize",1,1);} //**************************************************************** \class GridNoiseGateYuvs : FObject { \grin 0 int thresh; \decl 1 float(int v); \constructor (int v=0) {thresh=v;} }; GRID_INLET(GridNoiseGateYuvs,0) { if (in->dim->n!=3) RAISE("requires 3 dimensions: dim(y,x,3)"); if (in->dim->v[2]!=3) RAISE("requires 3 channels"); out=new GridOutlet(this,0,in->dim,in->nt); in->set_chunk(2); } GRID_FLOW { T tada[n]; for (long i=0; isend(n,tada); } GRID_END \def 1 float(int v) {thresh=v;} \end class {install("#noise_gate_yuvs",2,1);} //**************************************************************** \class GridPack : FObject { int n; PtrGrid a; \constructor (int n=2, NumberTypeE nt=int32_e) { if (n<1) RAISE("n=%d must be at least 1",n); if (n>32) RAISE("n=%d is too many?",n); a = new Grid(new Dim(n),nt,true); this->n=n; bself->ninlets_set(this->n); } \decl void _n_float (int inlet, float f); \decl void _n_list (int inlet, float f); \decl 0 bang (); //\grin 0 }; \def void _n_float (int inlet, float f) { #define FOO(T) ((T *)*a)[inlet] = (T)f; TYPESWITCH(a->nt,FOO,); #undef FOO _0_bang(argc,argv); } \def void _n_list (int inlet, float f) {_n_float(argc,argv,inlet,f);} \def 0 bang () { out=new GridOutlet(this,0,a->dim,a->nt); #define FOO(T) out->send(n,(T *)*a); TYPESWITCH(a->nt,FOO,); #undef FOO } \end class {install("#pack",1,1); add_creator("@pack");} \class GridUnpack : FObject { int n; \constructor (int n=2) { if (n<1) RAISE("n=%d must be at least 1",n); if (n>32) RAISE("n=%d is too many?",n); this->n=n; bself->noutlets_set(this->n); } \grin 0 }; GRID_INLET(GridUnpack,0) { in->set_chunk(0); } GRID_FLOW { for (int i=n-1; i>=0; i--) outlet_float(bself->outlets[i],(t_float)data[i]); } GRID_END \end class {install("#unpack",1,0);} //**************************************************************** \class ForEach : FObject { \constructor () {} \decl 0 list (...); }; \def 0 list (...) { t_outlet *o = bself->outlets[0]; for (int i=0; imom); string s = o.str(); pd_error(canvas,"%s",s.data()); } \end class {install("gf.error",1,0);} //**************************************************************** \class GridRotatificator : FObject { int angle; int from, to, n; \decl 0 float (int scale); \decl 0 axis (int from, int to, int n); \constructor (int from=0, int to=1, int n=2) { angle=0; _0_axis(0,0,from,to,n); } \decl 1 float(int angle); }; \def 0 float (int scale) { int32 rotator[n*n]; for (int i=0; i=n) RAISE("from-axis number incorrect"); if (to <0 || to >=n) RAISE( "to-axis number incorrect"); this->from = from; this-> to = to; this-> n = n; } \def 1 float(int angle) {this->angle = angle;} \end class {install("#rotatificator",2,1);} //**************************************************************** template void swap (T &a, T &b) {T c; c=a; a=b; b=c;} \class ListReverse : FObject { \constructor () {} \decl 0 list(...); }; \def 0 list (...) { for (int i=(argc-1)/2; i>=0; i--) swap(argv[i],argv[argc-i-1]); outlet_list(bself->te_outlet,&s_list,argc,argv); } \end class {install("listreverse",1,1);} \class ListFlatten : FObject { std::vector contents; \constructor () {} \decl 0 list(...); void traverse (int argc, t_atom2 *argv) { for (int i=0; ite_outlet,&s_list,contents.size(),&contents[0]); contents.clear(); } \end class {install("listflatten",1,1);} // does not do recursive comparison of lists. static bool atom_eq (t_atom &a, t_atom &b) { if (a.a_type!=b.a_type) return false; if (a.a_type==A_FLOAT) return a.a_float ==b.a_float; if (a.a_type==A_SYMBOL) return a.a_symbol ==b.a_symbol; if (a.a_type==A_POINTER) return a.a_gpointer==b.a_gpointer; if (a.a_type==A_LIST) return a.a_gpointer==b.a_gpointer; RAISE("don't know how to compare elements of type %d",a.a_type); } \class ListFind : FObject { int ac; t_atom *at; ~ListFind() {if (at) delete[] at;} \constructor (...) {ac=0; at=0; _1_list(argc,argv);} \decl 0 list(...); \decl 1 list(...); \decl 0 float(float f); \decl 0 symbol(t_symbol *s); }; \def 1 list (...) { if (at) delete[] at; ac = argc; at = new t_atom[argc]; for (int i=0; ioutlets[0],i==ac?-1:i); } \def 0 float (float f) { int i=0; for (; ioutlets[0],i==ac?-1:i); } \def 0 symbol (t_symbol *s) { int i=0; for (; ioutlets[0],i==ac?-1:i); } //doc:_1_list,"list to search into" //doc:_0_float,"float to find in that list" //doc_out:_0_float,"position of the incoming float in the stored list" \end class {install("listfind",2,1);} void outlet_atom (t_outlet *self, t_atom *av) { if (av->a_type==A_FLOAT) outlet_float( self,av->a_float); else if (av->a_type==A_SYMBOL) outlet_symbol( self,av->a_symbol); else if (av->a_type==A_POINTER) outlet_pointer(self,av->a_gpointer); else outlet_list(self,gensym("list"),1,av); } \class ListRead : FObject { /* sounds like tabread */ int ac; t_atom *at; ~ListRead() {if (at) delete[] at;} \constructor (...) {ac=0; at=0; _1_list(argc,argv);} \decl 0 float(float f); \decl 1 list(...); }; \def 0 float(float f) { int i = int(f); if (i<0) i+=ac; if (i<0 || i>=ac) {outlet_bang(bself->outlets[0]); return;} /* out-of-range */ outlet_atom(bself->outlets[0],&at[i]); } \def 1 list (...) { if (at) delete[] at; ac = argc; at = new t_atom[argc]; for (int i=0; i ninlets_set(1+nmosusses); bself->noutlets_set(1+nmosusses); } ~Range () {delete[] mosusses;} \decl 0 float(float f); \decl 0 list(float f); \decl void _n_float(int i, float f); }; \def 0 list(float f) {_0_float(argc,argv,f);} \def 0 float(float f) { int i; for (i=0; ioutlets[i],f); } \def void _n_float(int i, float f) { if (!i) _0_float(argc,argv,f); // precedence problem in rubyext... else mosusses[i-1] = f; } \end class {install("range",1,1);} //**************************************************************** static void display_update(void *x); string ssprintf(const char *fmt, ...) { std::ostringstream os; va_list va; va_start(va,fmt); voprintf(os,fmt,va); va_end(va); return os.str(); } #ifndef HAVE_DESIREDATA \class Display : FObject { bool selected; t_glist *canvas; t_symbol *rsym; int y,x,sy,sx; bool vis; std::ostringstream text; t_clock *clock; t_pd *gp; \constructor () { selected=false; canvas=0; y=0; x=0; sy=16; sx=80; vis=false; clock=0; std::ostringstream os; rsym = gensym(const_cast(ssprintf("display:%08x",this).data())); pd_typedmess(&pd_objectmaker,gensym("#print"),0,0); gp = pd_newest(); t_atom a[1]; SETFLOAT(a,20); pd_typedmess(gp,gensym("maxrows"),1,a); text << "..."; pd_bind((t_pd *)bself,rsym); SETPOINTER(a,(t_gpointer *)bself); pd_typedmess(gp,gensym("dest"),1,a); clock = clock_new((void *)this,(void(*)())display_update); } ~Display () { pd_unbind((t_pd *)bself,rsym); pd_free(gp); if (clock) clock_free(clock); } \decl void anything (...); \decl 0 set_size(int sy, int sx); \decl 0 grid(...); \decl 0 very_long_name_that_nobody_uses(...); void show() { std::ostringstream quoted; // def quote(text) "\"" + text.gsub(/["\[\]\n\$]/m) {|x| if x=="\n" then "\\n" else "\\"+x end } + "\"" end std::string ss = text.str(); const char *s = ss.data(); int n = ss.length(); for (int i=0;is_name,bself->te_xpix,bself->te_ypix,selected?"#0000ff":"#000000",canvas,ss.data()); } }; static void display_getrectfn(t_gobj *x, t_glist *glist, int *x1, int *y1, int *x2, int *y2) { BFObject *bself = (BFObject*)x; Display *self = (Display *)bself->self; self->canvas = glist; *x1 = bself->te_xpix-1; *y1 = bself->te_ypix-1; *x2 = bself->te_xpix+1+self->sx; *y2 = bself->te_ypix+1+self->sy; } static void display_displacefn(t_gobj *x, t_glist *glist, int dx, int dy) { BFObject *bself = (BFObject*)x; Display *self = (Display *)bself->self; self->canvas = glist; bself->te_xpix+=dx; bself->te_ypix+=dy; self->canvas = glist_getcanvas(glist); self->show(); canvas_fixlinesfor(glist, (t_text *)x); } static void display_selectfn(t_gobj *x, t_glist *glist, int state) { BFObject *bself = (BFObject*)x; Display *self = (Display *)bself->self; self->canvas = glist; self->selected=!!state; sys_vgui(".x%x.c itemconfigure %s -outline %s\n",glist_getcanvas(glist),self->rsym->s_name,self->selected?"#0000ff":"#000000"); } static void display_deletefn(t_gobj *x, t_glist *glist) { BFObject *bself = (BFObject*)x; Display *self = (Display *)bself->self; self->canvas = glist; if (self->vis) sys_vgui(".x%x.c delete %s %sTEXT\n",glist_getcanvas(glist),self->rsym->s_name,self->rsym->s_name); canvas_deletelinesfor(glist, (t_text *)x); } static void display_visfn(t_gobj *x, t_glist *glist, int flag) { BFObject *bself = (BFObject*)x; Display *self = (Display *)bself->self; self->canvas = glist; self->vis = !!flag; display_update(self); } static void display_update(void *x) { Display *self = (Display *)x; if (self->vis) self->show(); } \def 0 set_size(int sy, int sx) {this->sy=sy; this->sx=sx;} \def void anything (...) { string sel = string(argv[0]).data()+3; text.str(""); if (sel != "float") {text << sel; if (argc>1) text << " ";} long col = text.str().length(); char buf[MAXPDSTRING]; for (int i=1; i56) {text << "\\\\\n"; col=0;} } } clock_delay(clock,0); } \def 0 grid(...) { text.str(""); pd_typedmess(gp,gensym("grid"),argc,argv); clock_delay(clock,0); } \def 0 very_long_name_that_nobody_uses(...) { if (text.str().length()) text << "\n"; for (int i=0; ibfclass; t_widgetbehavior *wb = new t_widgetbehavior; wb->w_getrectfn = display_getrectfn; wb->w_displacefn = display_displacefn; wb->w_selectfn = display_selectfn; wb->w_activatefn = 0; wb->w_deletefn = display_deletefn; wb->w_visfn = display_visfn; wb->w_clickfn = 0; class_setwidget(qlass,wb); sys_gui("proc display_update {self x y fg bg outline font canvas text} { \n\ $canvas delete ${self}TEXT \n\ $canvas create text [expr $x+2] [expr $y+2] -fill $fg -font $font -text $text -anchor nw -tag ${self}TEXT \n\ foreach {x1 y1 x2 y2} [$canvas bbox ${self}TEXT] {} \n\ incr x -1 \n\ incr y -1 \n\ set sx [expr $x2-$x1+2] \n\ set sy [expr $y2-$y1+4] \n\ $canvas delete ${self} \n\ $canvas create rectangle $x $y [expr $x+$sx] [expr $y+$sy] -fill $bg -tags $self -outline $outline \n\ $canvas create rectangle $x $y [expr $x+7] $y -fill red -tags $self -outline $outline \n\ $canvas lower $self ${self}TEXT \n\ pd \"$self set_size $sy $sx;\" \n\ }\n"); } #endif // ndef HAVE_DESIREDATA //**************************************************************** // from pd/src/g_canvas.c struct _canvasenvironment { t_symbol *ce_dir; /* directory patch lives in */ int ce_argc; /* number of "$" arguments */ t_atom *ce_argv; /* array of "$" arguments */ int ce_dollarzero; /* value of "$0" */ }; struct ArgSpec { t_symbol *name; t_symbol *type; t_atom defaultv; }; \class Args : FObject { ArgSpec *sargv; int sargc; \constructor (...) { sargc = argc; sargv = new ArgSpec[argc]; for (int i=0; inoutlets_set(sargc+1); } ~Args () {delete[] sargv;} \decl 0 bang (); void process_args (int argc, t_atom *argv); }; void outlet_anything2 (t_outlet *o, int argc, t_atom *argv) { if (!argc) outlet_bang(o); else if (argv[0].a_type==A_SYMBOL) outlet_anything(o,argv[0].a_symbol,argc-1,argv+1); else if (argv[0].a_type==A_FLOAT && argc==1) outlet_float(o,argv[0].a_float); else outlet_anything(o,&s_list,argc,argv); } \def 0 bang () { _canvasenvironment *env = canvas_getenv(bself->mom); int ac = env->ce_argc; t_atom av[ac]; for (int i=0; ice_argv[i]; //ac = handle_braces(ac,av); int j; t_symbol *comma = gensym(","); for (j=0; joutlets[sargc],j-k,av+k); } } void Args::process_args (int argc, t_atom *argv) { t_canvas *canvas = canvas_getrootfor(bself->mom); t_symbol *wildcard = gensym("*"); for (int i=sargc-1; i>=0; i--) { t_atom *v; if (i>=argc) { if (sargv[i].defaultv.a_type != A_NULL) { v = &sargv[i].defaultv; } else if (sargv[i].name!=wildcard) { pd_error(canvas,"missing argument $%d named \"%s\"", i+1,sargv[i].name->s_name); continue; } } else v = &argv[i]; if (sargv[i].name==wildcard) { if (argc-i>0) outlet_list(bself->outlets[i],&s_list,argc-i,argv+i); else outlet_bang(bself->outlets[i]); } else { if (v->a_type==A_LIST) { t_binbuf *b = (t_binbuf *)v->a_gpointer; outlet_list(bself->outlets[i],&s_list,binbuf_getnatom(b),binbuf_getvec(b)); } else if (v->a_type==A_SYMBOL) outlet_symbol(bself->outlets[i],v->a_symbol); else outlet_anything2(bself->outlets[i],1,v); } } if (argc>sargc && sargv[sargc-1].name!=wildcard) pd_error(canvas,"warning: too many args (got %d, want %d)", argc, sargc); } \end class {install("args",1,1);} //**************************************************************** \class UnixTime : FObject { \constructor () {} \decl 0 bang (); }; \def 0 bang () { timeval tv; gettimeofday(&tv,0); time_t t = time(0); struct tm *tmp = localtime(&t); if (!tmp) RAISE("localtime: %s",strerror(errno)); char tt[MAXPDSTRING]; strftime(tt,MAXPDSTRING,"%a %b %d %H:%M:%S %Z %Y",tmp); t_atom a[6]; SETFLOAT(a+0,tmp->tm_year+1900); SETFLOAT(a+1,tmp->tm_mon-1); SETFLOAT(a+2,tmp->tm_mday); SETFLOAT(a+3,tmp->tm_hour); SETFLOAT(a+4,tmp->tm_min); SETFLOAT(a+5,tmp->tm_sec); t_atom b[3]; SETFLOAT(b+0,tv.tv_sec/86400); SETFLOAT(b+1,mod(tv.tv_sec,86400)); SETFLOAT(b+2,tv.tv_usec); outlet_anything(bself->outlets[2],&s_list,6,a); outlet_anything(bself->outlets[1],&s_list,3,b); send_out(0,strlen(tt),tt); } \end class UnixTime {install("unix_time",1,3);} //**************************************************************** /* if using a DB-25 female connector as found on a PC, then the pin numbering is like: 13 _____ 1 25 \___/ 14 1 = STROBE = the clock line is a square wave, often at 9600 Hz, which determines the data rate in usual circumstances. 2..9 = D0..D7 = the eight ordinary data bits 10 = -ACK (status bit 6 ?) 11 = BUSY (status bit 7) 12 = PAPER_END (status bit 5) 13 = SELECT (status bit 4 ?) 14 = -AUTOFD 15 = -ERROR (status bit 3 ?) 16 = -INIT 17 = -SELECT_IN 18..25 = GROUND */ //#include #define LPCHAR 0x0601 #define LPCAREFUL 0x0609 /* obsoleted??? wtf? */ #define LPGETSTATUS 0x060b /* return LP_S(minor) */ #define LPGETFLAGS 0x060e /* get status flags */ #include struct ParallelPort; void ParallelPort_call(ParallelPort *self); \class ParallelPort : FObject { FILE *f; int fd; int status; int flags; bool manually; t_clock *clock; ~ParallelPort () {if (clock) clock_free(clock); if (f) fclose(f);} \constructor (string port, bool manually=0) { f = fopen(port.data(),"r+"); if (!f) RAISE("open %s: %s",port.data(),strerror(errno)); fd = fileno(f); status = 0xdeadbeef; flags = 0xdeadbeef; this->manually = manually; clock = manually ? 0 : clock_new(this,(void(*)())ParallelPort_call); clock_delay(clock,0); } void call (); \decl 0 float (float x); \decl 0 bang (); }; \def 0 float (float x) { uint8 foo = (uint8) x; fwrite(&foo,1,1,f); fflush(f); } void ParallelPort_call(ParallelPort *self) {self->call();} void ParallelPort::call() { int flags; if (ioctl(fd,LPGETFLAGS,&flags)<0) post("ioctl: %s",strerror(errno)); if (this->flags!=flags) outlet_float(bself->outlets[2],flags); this->flags = flags; int status; if (ioctl(fd,LPGETSTATUS,&status)<0) post("ioctl: %s",strerror(errno)); if (this->status!=status) outlet_float(bself->outlets[1],status); this->status = status; if (clock) clock_delay(clock,2000); } \def 0 bang () {status = flags = 0xdeadbeef; call();} //outlet 0 reserved (future use) \end class {install("parallel_port",1,3);} //**************************************************************** \class Route2 : FObject { int nsels; t_symbol **sels; ~Route2() {if (sels) delete[] sels;} \constructor (...) {nsels=0; sels=0; _1_list(argc,argv); bself->noutlets_set(1+nsels);} \decl void anything(...); \decl 1 list(...); }; \def void anything(...) { t_symbol *sel = gensym(argv[0].a_symbol->s_name+3); int i=0; for (i=0; ioutlets[i],sel,argc-1,argv+1); } \def 1 list(...) { for (int i=0; i int sgn(T a, T b=0) {return ab;} \class Shunt : FObject { int n; \attr int index; \attr int mode; \attr int hi; \attr int lo; \constructor (int n=2, int i=0) { this->n=n; this->hi=n-1; this->lo=0; this->mode=0; this->index=i; bself->noutlets_set(n); } \decl void anything(...); \decl 1 float(int i); }; \def void anything(...) { t_symbol *sel = gensym(argv[0].a_symbol->s_name+3); outlet_anything(bself->outlets[index],sel,argc-1,argv+1); if (mode) { index += sgn(mode); if (indexhi) { int k = max(hi-lo+1,0); int m = gf_abs(mode); if (m==1) index = mod(index-lo,k)+lo; else {mode=-mode; index+=mode;} } } } \def 1 float(int i) {index = mod(i,n);} \end class {install("shunt",2,0); add_creator("demux");} \class Send39 : FObject { \attr t_symbol *dest; \constructor (t_symbol *dest) { char buf[MAXPDSTRING]; sprintf(buf,"pd-%s",dest->s_name); this->dest = gensym(buf); } \decl void anything (...); }; \def void anything(...) { t_symbol *sel = gensym(argv[0].a_symbol->s_name+3); pd_typedmess(this->dest->s_thing,sel,argc-1,argv+1); } \end class {install("send39",1,0);} struct Receives; struct ReceivesProxy { t_pd x_pd; Receives *parent; t_symbol *suffix; }; t_class *ReceivesProxy_class; \class Receives : FObject { int ac; ReceivesProxy **av; t_symbol *prefix; t_symbol *local (t_symbol *suffix) {return gensym((string(prefix->s_name) + string(suffix->s_name)).data());} \constructor (t_symbol *prefix, ...) { this->prefix = prefix; ac = argc-1; av = new ReceivesProxy *[argc-1]; for (int i=0; iparent = this; av[i]->suffix = argv[i+1]; pd_bind( (t_pd *)av[i],local(av[i]->suffix)); } } ~Receives () { for (int i=0; isuffix)); pd_free((t_pd *)av[i]); } delete[] av; } }; void ReceivesProxy_anything (ReceivesProxy *self, t_symbol *s, int argc, t_atom *argv) { outlet_symbol( self->parent->bself->outlets[1],self->suffix); outlet_anything(self->parent->bself->outlets[0],s,argc,argv); } \end class { install("receives",1,2); ReceivesProxy_class = class_new(gensym("receives.proxy"),0,0,sizeof(ReceivesProxy),CLASS_PD|CLASS_NOINLET, A_NULL); class_addanything(ReceivesProxy_class,(t_method)ReceivesProxy_anything); } /* this can't report on bang,float,symbol,pointer,list because zgetfn can't either */ \class ClassExists : FObject { \constructor () {} \decl void _0_symbol(t_symbol *s); }; \def void _0_symbol(t_symbol *s) { outlet_float(bself->outlets[0],!!zgetfn(&pd_objectmaker,s)); } \end class {install("class_exists",1,1);} typedef t_binbuf t_list; static t_list *list_new (int argc, t_atom *argv) { t_list *b = binbuf_new(); binbuf_add(b,argc,argv); return b; } static void list_free (t_list *self) {binbuf_free(self);} \class ListEqual : FObject { t_list *list; \constructor (...) {list=0; _1_list(argc,argv);} \decl 0 list (...); \decl 1 list (...); }; \def 1 list (...) { if (list) list_free(list); list = list_new(argc,argv); } \def 0 list (...) { if (binbuf_getnatom(list) != argc) {outlet_float(bself->outlets[0],0); return;} t_atom2 *at = (t_atom2 *)binbuf_getvec(list); for (int i=0; ioutlets[0],0); return;} outlet_float(bself->outlets[0],1); } \end class {install("list.==",2,1);} //**************************************************************** //#ifdef UNISTD #include #include #include #include #include //#endif #if defined (__APPLE__) || defined (__FreeBSD__) #define HZ CLK_TCK #endif uint64 cpu_hertz; int uint64_compare(uint64 &a, uint64 &b) {return ab;} \class UserTime : FObject { clock_t time; \constructor () {_0_bang(argc,argv);} \decl 0 bang (); \decl 1 bang (); }; \def 0 bang () {struct tms t; times(&t); time = t.tms_utime;} \def 1 bang () {struct tms t; times(&t); outlet_float(bself->outlets[0],(t.tms_utime-time)*1000/HZ);} \end class {install("usertime",2,1);} \class SystemTime : FObject { clock_t time; \constructor () {_0_bang(argc,argv);} \decl 0 bang (); \decl 1 bang (); }; \def 0 bang () {struct tms t; times(&t); time = t.tms_stime;} \def 1 bang () {struct tms t; times(&t); outlet_float(bself->outlets[0],(t.tms_stime-time)*1000/HZ);} \end class {install("systemtime",2,1);} \class TSCTime : FObject { uint64 time; \constructor () {_0_bang(argc,argv);} \decl 0 bang (); \decl 1 bang (); }; \def 0 bang () {time=rdtsc();} \def 1 bang () {outlet_float(bself->outlets[0],(rdtsc()-time)*1000.0/cpu_hertz);} \end class {install("tsctime",2,1); struct timeval t0,t1; uint64 u0,u1; uint64 estimates[3]; for (int i=0; i<3; i++) { u0=rdtsc(); gettimeofday(&t0,0); usleep(10000); u1=rdtsc(); gettimeofday(&t1,0); uint64 t = (t1.tv_sec-t0.tv_sec)*1000000+(t1.tv_usec-t0.tv_usec); estimates[i] = (u1-u0)*1000000/t; } qsort(estimates,3,sizeof(uint64),(comparator_t)uint64_compare); cpu_hertz = estimates[1]; } //**************************************************************** #define OP(x) op_dict[string(#x)] void startup_flow_objects () { op_add = OP(+); op_sub = OP(-); op_mul = OP(*); op_shl = OP(<<); op_mod = OP(%); op_and = OP(&); op_div = OP(/); op_put = OP(put); \startall }