cannam@0: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
cannam@0: 
cannam@0: /*
cannam@0:     Vamp feature extraction plugin using the MATCH audio alignment
cannam@0:     algorithm.
cannam@0: 
cannam@0:     Centre for Digital Music, Queen Mary, University of London.
cannam@0:     This file copyright 2007 Simon Dixon, Chris Cannam and QMUL.
cannam@0:     
cannam@0:     This program is free software; you can redistribute it and/or
cannam@0:     modify it under the terms of the GNU General Public License as
cannam@0:     published by the Free Software Foundation; either version 2 of the
cannam@0:     License, or (at your option) any later version.  See the file
cannam@0:     COPYING included with this distribution for more information.
cannam@0: */
cannam@0: 
cannam@0: #include "Path.h"
cannam@0: 
cannam@0: int
cannam@0: Path::smooth(std::vector<int> &x, std::vector<int> &y, int length)
cannam@0: {
cannam@0:     if (length == 0)
cannam@0:         return 0;
Chris@14:     while ((int)val.size() < length) {
cannam@0:         val.push_back(0);
cannam@0:         len.push_back(0);
cannam@0:     }
cannam@0:     int p = 0;
cannam@0:     val[0] = len[0] = 0;
cannam@0:     for (int i = 1; i < length; i++) {	// H = 1; V = 2; D = 3
cannam@0:         int current = x[i] - x[i-1] + 2 * (y[i] - y[i-1]);
cannam@0:         if (current == val[p]) {
cannam@0:             len[p]++;
cannam@0:         } else if ((current == 3) || (val[p] == 0)) {
cannam@0:             val[++p] = current;
cannam@0:             len[p] = 1;
cannam@0:         } else if (val[p] + current == 3) {	// 1 + 2
cannam@0:             if (--len[p] == 0)
cannam@0:                 p--;
cannam@0:             if (val[p] == 3)
cannam@0:                 len[p]++;
cannam@0:             else {
cannam@0:                 val[++p] = 3;
cannam@0:                 len[p] = 1;
cannam@0:             }
cannam@0:         } else {	// val[p] == 3 && current != 3
cannam@0:             if ((val[p-1] == current) ||
cannam@0:                 (val[p-1] == 0) ||
cannam@0:                 (len[p] > MAX_RUN_LENGTH)) {
cannam@0:                 val[++p] = current;
cannam@0:                 len[p] = 1;
cannam@0:             } else {
cannam@0:                 if (--len[p-1] == 0) {
cannam@0:                     val[p-1] = val[p];
cannam@0:                     len[p-1] = len[p];
cannam@0:                     p--;
cannam@0:                     if (val[p-1] == 3) {
cannam@0:                         len[p-1] += len[p];
cannam@0:                         p--;
cannam@0:                     }
cannam@0:                 }
cannam@0:                 len[p]++;
cannam@0:             }
cannam@0:         }
cannam@0:     }
cannam@0:     int i = 1;
cannam@0:     for (int pp = 1; pp <= p; pp++) {
cannam@0:         int dx = val[pp] & 1;
cannam@0:         int dy = val[pp] >> 1;
cannam@0:         for (int j = len[pp]; j > 0; j--, i++) {
cannam@0:             x[i] = x[i-1] + dx;
cannam@0:             y[i] = y[i-1] + dy;
cannam@0:         }
cannam@0:     }
cannam@0:     return i;
cannam@0: }