recSchema.cpp

Go to the documentation of this file.

/************************************************************************
 ************************************************************************
    FAUST compiler
    Copyright (C) 2003-2004 GRAME, Centre National de Creation Musicale
    ---------------------------------------------------------------------
    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.

    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., 675 Mass Ave, Cambridge, MA 02139, USA.
 ************************************************************************
 ************************************************************************/


#include "recSchema.h"
#include <iostream>
#include <assert.h>

using namespace std;

schema* makeRecSchema (schema* s1, schema* s2)
{
    schema* a = makeEnlargedSchema(s1, s2->width());
    schema* b = makeEnlargedSchema(s2, s1->width());
    double  m = dWire * max(b->inputs(), b->outputs());
    double  w = a->width() + 2*m;

    return new recSchema(a,b,w);
}

recSchema::recSchema (schema* s1, schema* s2, double width)
    :   schema( s1->inputs() - s2->outputs(),
                s1->outputs(),
                width,
                s1->height() + s2->height() ),
        fSchema1(s1),
        fSchema2(s2)
{
    // this version only accepts legal expressions of same width
    assert(s1->inputs() >= s2->outputs());
    assert(s1->outputs() >= s2->inputs());
    assert(s1->width() >= s2->width());

    // create the input and output points
    for (unsigned int i=0; i<inputs(); i++)     fInputPoint.push_back(point(0,0));
    for (unsigned int i=0; i<outputs(); i++)    fOutputPoint.push_back(point(0,0));

}

void recSchema::place(double ox, double oy, int orientation)
{
    beginPlace(ox, oy, orientation);

    double dx1 = (width() - fSchema1->width())/2;
    double dx2 = (width() - fSchema2->width())/2;

    // place the two sub diagrams
    if (orientation == kLeftRight) {
        fSchema2->place(ox+dx2, oy, kRightLeft);
        fSchema1->place(ox+dx1, oy+fSchema2->height(), kLeftRight);
    } else {
        fSchema1->place(ox+dx1, oy, kRightLeft);
        fSchema2->place(ox+dx2, oy+fSchema1->height(), kLeftRight);
    }


    // adjust delta space to orientation
    if (orientation == kRightLeft) { dx1 = -dx1; }

    // place input points
    for (unsigned int i=0; i<inputs(); i++) {
        point p = fSchema1->inputPoint(i+fSchema2->outputs());
        fInputPoint[i] = point(p.x-dx1, p.y);
    }

    // place output points
    for (unsigned int i=0; i<outputs(); i++) {
        point p = fSchema1->outputPoint(i);
        fOutputPoint[i] = point(p.x+dx1, p.y);
    }

    endPlace();
}


point recSchema::inputPoint(unsigned int i) const
{
    return fInputPoint[i];
}


point recSchema::outputPoint(unsigned int i) const
{
    return fOutputPoint[i];
}


void recSchema::draw(device& dev)
{
    assert(placed());

    // draw the two subdiagrams
    fSchema1->draw(dev);
    fSchema2->draw(dev);

    // draw the output lines
    for (unsigned int i=0; i<outputs(); i++) {
        point p = fSchema1->outputPoint(i);
        point q = outputPoint(i);
        //dev.trait(p.x, p.y, q.x, q.y);
    }

    // draw the input lines
    unsigned int skip = fSchema2->outputs();
    for (unsigned int i=0; i<inputs(); i++) {
        point p = fSchema1->inputPoint(i+skip);
        point q = inputPoint(i);
        //dev.trait(p.x, p.y, q.x, q.y);
    }

    // draw the feedback connections to each fSchema2 input
    for (unsigned int i=0; i<fSchema2->inputs(); i++) {
        drawFeedback(dev, fSchema1->outputPoint(i), fSchema2->inputPoint(i), i*dWire);
    }

    // draw the feedfront connections from each fSchema2 output
    for (unsigned int i=0; i<fSchema2->outputs(); i++) {
        drawFeedfront(dev, fSchema2->outputPoint(i), fSchema1->inputPoint(i), i*dWire);
    }
}


void recSchema::drawDelaySign(device& dev, double x, double y, double size)
{
    dev.trait(x-size/2, y, x-size/2, y-size);
    dev.trait(x-size/2, y-size, x+size/2, y-size);
    dev.trait(x+size/2, y-size, x+size/2, y);
}


void recSchema::collectTraits(collector& c)
{
    assert(placed());

    // draw the two subdiagrams
    fSchema1->collectTraits(c);
    fSchema2->collectTraits(c);

    // draw the feedback connections to each fSchema2 input
    for (unsigned int i=0; i<fSchema2->inputs(); i++) {
        collectFeedback(c, fSchema1->outputPoint(i), fSchema2->inputPoint(i), i*dWire, outputPoint(i));
    }

    // draw the non recursive output lines
    for (unsigned int i=fSchema2->inputs(); i<outputs(); i++) {
        point p = fSchema1->outputPoint(i);
        point q = outputPoint(i);
        c.addTrait(trait(p,q));     // in->out order
    }

    // draw the input lines
    unsigned int skip = fSchema2->outputs();
    for (unsigned int i=0; i<inputs(); i++) {
        point p = inputPoint(i);
        point q = fSchema1->inputPoint(i+skip);
        c.addTrait(trait(p,q));     // in->out order
    }

    // draw the feedfront connections from each fSchema2 output
    for (unsigned int i=0; i<fSchema2->outputs(); i++) {
        collectFeedfront(c, fSchema2->outputPoint(i), fSchema1->inputPoint(i), i*dWire);
    }
}



void recSchema::drawFeedback(device& dev, const point& src, const point& dst, double dx)
{
    double  ox = src.x + ((orientation()==kLeftRight) ? dx : -dx);
    double  ct = (orientation()==kLeftRight) ? dWire/2 : -dWire/2;

    drawDelaySign(dev, ox, src.y, ct);
    //dev.trait(ox, src.y-ct, ox, dst.y);
    //dev.trait(ox, dst.y, dst.x, dst.y);
}



void recSchema::collectFeedback(collector& c, const point& src, const point& dst, double dx, const point& out)
{
    double  ox = src.x + ((orientation()==kLeftRight) ? dx : -dx);
    double  ct = (orientation()==kLeftRight) ? dWire/2 : -dWire/2;

    point   up(ox, src.y-ct);
    point   br(ox+ct/2.0, src.y);

    c.addOutput(up);
    c.addOutput(br);
    c.addInput(br);

    c.addTrait(trait(up, point(ox, dst.y)));
    c.addTrait(trait(point(ox, dst.y), point(dst.x, dst.y)));
    c.addTrait(trait(src,br));
    c.addTrait(trait(br,out));

}


void recSchema::drawFeedfront(device& dev, const point& src, const point& dst, double dx)
{
//    double    ox = src.x + ((orientation()==kLeftRight) ? -dx : dx);

//  dev.trait(ox, src.y, src.x, src.y);
//  dev.trait(ox, src.y, ox, dst.y);
//  dev.trait(ox, dst.y, dst.x, dst.y);
}


void recSchema::collectFeedfront(collector& c, const point& src, const point& dst, double dx)
{
    double  ox = src.x + ((orientation()==kLeftRight) ? -dx : dx);

    c.addTrait(trait(point(src.x, src.y), point(ox, src.y)));
    c.addTrait(trait(point(ox, src.y), point(ox, dst.y)));
    c.addTrait(trait(point(ox, dst.y), point(dst.x, dst.y)));
}