annotate framework/Document.cpp @ 53:de2b3c6479c8

* Introduce new Transform class which contains data necessary to describe the context for a plugin -- the plugin's name and output, the step/block size etc (formerly spread across ExecutionContext and TransformFactory). Other code hasn't been updated to use this yet. * Rename existing Transform stuff to Transformers (because they run Transforms) I'm still not 100% sure about this change, don't rely on it.
author Chris Cannam
date Mon, 05 Nov 2007 15:31:06 +0000
parents 43ad8d909e28
children a798f5e6fc5e
rev   line source
Chris@45 1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
Chris@45 2
Chris@45 3 /*
Chris@45 4 Sonic Visualiser
Chris@45 5 An audio file viewer and annotation editor.
Chris@45 6 Centre for Digital Music, Queen Mary, University of London.
Chris@45 7 This file copyright 2006 Chris Cannam and QMUL.
Chris@45 8
Chris@45 9 This program is free software; you can redistribute it and/or
Chris@45 10 modify it under the terms of the GNU General Public License as
Chris@45 11 published by the Free Software Foundation; either version 2 of the
Chris@45 12 License, or (at your option) any later version. See the file
Chris@45 13 COPYING included with this distribution for more information.
Chris@45 14 */
Chris@45 15
Chris@45 16 #include "Document.h"
Chris@45 17
Chris@45 18 #include "data/model/WaveFileModel.h"
Chris@45 19 #include "data/model/WritableWaveFileModel.h"
Chris@45 20 #include "data/model/DenseThreeDimensionalModel.h"
Chris@45 21 #include "data/model/DenseTimeValueModel.h"
Chris@45 22 #include "layer/Layer.h"
Chris@45 23 #include "base/CommandHistory.h"
Chris@45 24 #include "base/Command.h"
Chris@45 25 #include "view/View.h"
Chris@45 26 #include "base/PlayParameterRepository.h"
Chris@45 27 #include "base/PlayParameters.h"
Chris@53 28 #include "plugin/transform/TransformerFactory.h"
Chris@45 29 #include <QApplication>
Chris@45 30 #include <QTextStream>
Chris@45 31 #include <iostream>
Chris@45 32
Chris@45 33 // For alignment:
Chris@45 34 #include "data/model/AggregateWaveModel.h"
Chris@45 35 #include "data/model/SparseTimeValueModel.h"
Chris@45 36 #include "data/model/AlignmentModel.h"
Chris@45 37
Chris@45 38 //!!! still need to handle command history, documentRestored/documentModified
Chris@45 39
Chris@45 40 Document::Document() :
Chris@47 41 m_mainModel(0),
Chris@47 42 m_autoAlignment(false)
Chris@45 43 {
Chris@45 44 connect(this, SIGNAL(modelAboutToBeDeleted(Model *)),
Chris@53 45 TransformerFactory::getInstance(),
Chris@45 46 SLOT(modelAboutToBeDeleted(Model *)));
Chris@45 47 }
Chris@45 48
Chris@45 49 Document::~Document()
Chris@45 50 {
Chris@45 51 //!!! Document should really own the command history. atm we
Chris@45 52 //still refer to it in various places that don't have access to
Chris@45 53 //the document, be nice to fix that
Chris@45 54
Chris@45 55 // std::cerr << "\n\nDocument::~Document: about to clear command history" << std::endl;
Chris@45 56 CommandHistory::getInstance()->clear();
Chris@45 57
Chris@52 58 std::cerr << "Document::~Document: about to delete layers" << std::endl;
Chris@45 59 while (!m_layers.empty()) {
Chris@45 60 deleteLayer(*m_layers.begin(), true);
Chris@45 61 }
Chris@45 62
Chris@45 63 if (!m_models.empty()) {
Chris@45 64 std::cerr << "Document::~Document: WARNING: "
Chris@45 65 << m_models.size() << " model(s) still remain -- "
Chris@45 66 << "should have been garbage collected when deleting layers"
Chris@45 67 << std::endl;
Chris@45 68 while (!m_models.empty()) {
Chris@45 69 Model *model = m_models.begin()->first;
Chris@45 70 if (model == m_mainModel) {
Chris@45 71 // just in case!
Chris@45 72 std::cerr << "Document::~Document: WARNING: Main model is also"
Chris@45 73 << " in models list!" << std::endl;
Chris@45 74 } else if (model) {
Chris@45 75 emit modelAboutToBeDeleted(model);
Chris@45 76 model->aboutToDelete();
Chris@45 77 delete model;
Chris@45 78 }
Chris@45 79 m_models.erase(m_models.begin());
Chris@45 80 }
Chris@45 81 }
Chris@45 82
Chris@45 83 // std::cerr << "Document::~Document: About to get rid of main model"
Chris@45 84 // << std::endl;
Chris@45 85 if (m_mainModel) {
Chris@45 86 emit modelAboutToBeDeleted(m_mainModel);
Chris@45 87 m_mainModel->aboutToDelete();
Chris@45 88 }
Chris@45 89
Chris@45 90 emit mainModelChanged(0);
Chris@45 91 delete m_mainModel;
Chris@45 92
Chris@45 93 }
Chris@45 94
Chris@45 95 Layer *
Chris@45 96 Document::createLayer(LayerFactory::LayerType type)
Chris@45 97 {
Chris@45 98 Layer *newLayer = LayerFactory::getInstance()->createLayer(type);
Chris@45 99 if (!newLayer) return 0;
Chris@45 100
Chris@45 101 newLayer->setObjectName(getUniqueLayerName(newLayer->objectName()));
Chris@45 102
Chris@45 103 m_layers.insert(newLayer);
Chris@52 104
Chris@52 105 std::cerr << "Document::createLayer: Added layer of type " << type
Chris@52 106 << ", now have " << m_layers.size() << " layers" << std::endl;
Chris@52 107
Chris@45 108 emit layerAdded(newLayer);
Chris@45 109
Chris@45 110 return newLayer;
Chris@45 111 }
Chris@45 112
Chris@45 113 Layer *
Chris@45 114 Document::createMainModelLayer(LayerFactory::LayerType type)
Chris@45 115 {
Chris@45 116 Layer *newLayer = createLayer(type);
Chris@45 117 if (!newLayer) return 0;
Chris@45 118 setModel(newLayer, m_mainModel);
Chris@45 119 return newLayer;
Chris@45 120 }
Chris@45 121
Chris@45 122 Layer *
Chris@45 123 Document::createImportedLayer(Model *model)
Chris@45 124 {
Chris@45 125 LayerFactory::LayerTypeSet types =
Chris@45 126 LayerFactory::getInstance()->getValidLayerTypes(model);
Chris@45 127
Chris@45 128 if (types.empty()) {
Chris@45 129 std::cerr << "WARNING: Document::importLayer: no valid display layer for model" << std::endl;
Chris@45 130 return 0;
Chris@45 131 }
Chris@45 132
Chris@45 133 //!!! for now, just use the first suitable layer type
Chris@45 134 LayerFactory::LayerType type = *types.begin();
Chris@45 135
Chris@45 136 Layer *newLayer = LayerFactory::getInstance()->createLayer(type);
Chris@45 137 if (!newLayer) return 0;
Chris@45 138
Chris@45 139 newLayer->setObjectName(getUniqueLayerName(newLayer->objectName()));
Chris@45 140
Chris@45 141 addImportedModel(model);
Chris@45 142 setModel(newLayer, model);
Chris@45 143
Chris@45 144 //!!! and all channels
Chris@45 145 setChannel(newLayer, -1);
Chris@45 146
Chris@45 147 m_layers.insert(newLayer);
Chris@52 148
Chris@52 149 std::cerr << "Document::createImportedLayer: Added layer of type " << type
Chris@52 150 << ", now have " << m_layers.size() << " layers" << std::endl;
Chris@52 151
Chris@45 152 emit layerAdded(newLayer);
Chris@45 153 return newLayer;
Chris@45 154 }
Chris@45 155
Chris@45 156 Layer *
Chris@45 157 Document::createEmptyLayer(LayerFactory::LayerType type)
Chris@45 158 {
Chris@45 159 Model *newModel =
Chris@45 160 LayerFactory::getInstance()->createEmptyModel(type, m_mainModel);
Chris@45 161 if (!newModel) return 0;
Chris@45 162
Chris@45 163 Layer *newLayer = createLayer(type);
Chris@45 164 if (!newLayer) {
Chris@45 165 delete newModel;
Chris@45 166 return 0;
Chris@45 167 }
Chris@45 168
Chris@45 169 addImportedModel(newModel);
Chris@45 170 setModel(newLayer, newModel);
Chris@45 171
Chris@45 172 return newLayer;
Chris@45 173 }
Chris@45 174
Chris@45 175 Layer *
Chris@45 176 Document::createDerivedLayer(LayerFactory::LayerType type,
Chris@53 177 TransformerId transform)
Chris@45 178 {
Chris@45 179 Layer *newLayer = createLayer(type);
Chris@45 180 if (!newLayer) return 0;
Chris@45 181
Chris@45 182 newLayer->setObjectName(getUniqueLayerName
Chris@53 183 (TransformerFactory::getInstance()->
Chris@53 184 getTransformerFriendlyName(transform)));
Chris@45 185
Chris@45 186 return newLayer;
Chris@45 187 }
Chris@45 188
Chris@45 189 Layer *
Chris@53 190 Document::createDerivedLayer(TransformerId transform,
Chris@45 191 Model *inputModel,
Chris@53 192 const PluginTransformer::ExecutionContext &context,
Chris@45 193 QString configurationXml)
Chris@45 194 {
Chris@45 195 Model *newModel = addDerivedModel(transform, inputModel,
Chris@45 196 context, configurationXml);
Chris@45 197 if (!newModel) {
Chris@45 198 // error already printed to stderr by addDerivedModel
Chris@45 199 emit modelGenerationFailed(transform);
Chris@45 200 return 0;
Chris@45 201 }
Chris@45 202
Chris@45 203 LayerFactory::LayerTypeSet types =
Chris@45 204 LayerFactory::getInstance()->getValidLayerTypes(newModel);
Chris@45 205
Chris@45 206 if (types.empty()) {
Chris@53 207 std::cerr << "WARNING: Document::createLayerForTransformer: no valid display layer for output of transform " << transform.toStdString() << std::endl;
Chris@45 208 delete newModel;
Chris@45 209 return 0;
Chris@45 210 }
Chris@45 211
Chris@45 212 //!!! for now, just use the first suitable layer type
Chris@45 213
Chris@45 214 Layer *newLayer = createLayer(*types.begin());
Chris@45 215 setModel(newLayer, newModel);
Chris@45 216
Chris@45 217 //!!! We need to clone the model when adding the layer, so that it
Chris@45 218 //can be edited without affecting other layers that are based on
Chris@45 219 //the same model. Unfortunately we can't just clone it now,
Chris@45 220 //because it probably hasn't been completed yet -- the transform
Chris@45 221 //runs in the background. Maybe the transform has to handle
Chris@45 222 //cloning and cacheing models itself.
Chris@45 223 //
Chris@45 224 // Once we do clone models here, of course, we'll have to avoid
Chris@45 225 // leaking them too.
Chris@45 226 //
Chris@45 227 // We want the user to be able to add a model to a second layer
Chris@45 228 // _while it's still being calculated in the first_ and have it
Chris@45 229 // work quickly. That means we need to put the same physical
Chris@45 230 // model pointer in both layers, so they can't actually be cloned.
Chris@45 231
Chris@45 232 if (newLayer) {
Chris@45 233 newLayer->setObjectName(getUniqueLayerName
Chris@53 234 (TransformerFactory::getInstance()->
Chris@53 235 getTransformerFriendlyName(transform)));
Chris@45 236 }
Chris@45 237
Chris@45 238 emit layerAdded(newLayer);
Chris@45 239 return newLayer;
Chris@45 240 }
Chris@45 241
Chris@45 242 void
Chris@45 243 Document::setMainModel(WaveFileModel *model)
Chris@45 244 {
Chris@45 245 Model *oldMainModel = m_mainModel;
Chris@45 246 m_mainModel = model;
Chris@45 247
Chris@45 248 emit modelAdded(m_mainModel);
Chris@45 249
Chris@45 250 std::vector<Layer *> obsoleteLayers;
Chris@53 251 std::set<QString> failedTransformers;
Chris@45 252
Chris@45 253 // We need to ensure that no layer is left using oldMainModel or
Chris@45 254 // any of the old derived models as its model. Either replace the
Chris@45 255 // model, or delete the layer for each layer that is currently
Chris@45 256 // using one of these. Carry out this replacement before we
Chris@45 257 // delete any of the models.
Chris@45 258
Chris@52 259 std::cerr << "Document::setMainModel: Have "
Chris@52 260 << m_layers.size() << " layers" << std::endl;
Chris@52 261
Chris@45 262 for (LayerSet::iterator i = m_layers.begin(); i != m_layers.end(); ++i) {
Chris@45 263
Chris@45 264 Layer *layer = *i;
Chris@45 265 Model *model = layer->getModel();
Chris@45 266
Chris@45 267 // std::cerr << "Document::setMainModel: inspecting model "
Chris@45 268 // << (model ? model->objectName().toStdString() : "(null)") << " in layer "
Chris@45 269 // << layer->objectName().toStdString() << std::endl;
Chris@45 270
Chris@45 271 if (model == oldMainModel) {
Chris@45 272 // std::cerr << "... it uses the old main model, replacing" << std::endl;
Chris@45 273 LayerFactory::getInstance()->setModel(layer, m_mainModel);
Chris@45 274 continue;
Chris@45 275 }
Chris@45 276
Chris@45 277 if (m_models.find(model) == m_models.end()) {
Chris@45 278 std::cerr << "WARNING: Document::setMainModel: Unknown model "
Chris@45 279 << model << " in layer " << layer << std::endl;
Chris@45 280 // get rid of this hideous degenerate
Chris@45 281 obsoleteLayers.push_back(layer);
Chris@45 282 continue;
Chris@45 283 }
Chris@45 284
Chris@45 285 if (m_models[model].source == oldMainModel) {
Chris@45 286
Chris@45 287 // std::cerr << "... it uses a model derived from the old main model, regenerating" << std::endl;
Chris@45 288
Chris@45 289 // This model was derived from the previous main
Chris@45 290 // model: regenerate it.
Chris@45 291
Chris@53 292 TransformerId transform = m_models[model].transform;
Chris@53 293 PluginTransformer::ExecutionContext context = m_models[model].context;
Chris@45 294
Chris@45 295 Model *replacementModel =
Chris@45 296 addDerivedModel(transform,
Chris@45 297 m_mainModel,
Chris@45 298 context,
Chris@45 299 m_models[model].configurationXml);
Chris@45 300
Chris@45 301 if (!replacementModel) {
Chris@45 302 std::cerr << "WARNING: Document::setMainModel: Failed to regenerate model for transform \""
Chris@45 303 << transform.toStdString() << "\"" << " in layer " << layer << std::endl;
Chris@53 304 if (failedTransformers.find(transform) == failedTransformers.end()) {
Chris@45 305 emit modelRegenerationFailed(layer->objectName(),
Chris@45 306 transform);
Chris@53 307 failedTransformers.insert(transform);
Chris@45 308 }
Chris@45 309 obsoleteLayers.push_back(layer);
Chris@45 310 } else {
Chris@45 311 std::cerr << "Replacing model " << model << " (type "
Chris@45 312 << typeid(*model).name() << ") with model "
Chris@45 313 << replacementModel << " (type "
Chris@45 314 << typeid(*replacementModel).name() << ") in layer "
Chris@45 315 << layer << " (name " << layer->objectName().toStdString() << ")"
Chris@45 316 << std::endl;
Chris@45 317 RangeSummarisableTimeValueModel *rm =
Chris@45 318 dynamic_cast<RangeSummarisableTimeValueModel *>(replacementModel);
Chris@45 319 if (rm) {
Chris@45 320 std::cerr << "new model has " << rm->getChannelCount() << " channels " << std::endl;
Chris@45 321 } else {
Chris@45 322 std::cerr << "new model is not a RangeSummarisableTimeValueModel!" << std::endl;
Chris@45 323 }
Chris@45 324 setModel(layer, replacementModel);
Chris@45 325 }
Chris@45 326 }
Chris@45 327 }
Chris@45 328
Chris@45 329 for (size_t k = 0; k < obsoleteLayers.size(); ++k) {
Chris@45 330 deleteLayer(obsoleteLayers[k], true);
Chris@45 331 }
Chris@45 332
Chris@48 333 for (ModelMap::iterator i = m_models.begin(); i != m_models.end(); ++i) {
Chris@48 334 if (i->first->getAlignmentReference() == oldMainModel) {
Chris@48 335 alignModel(i->first);
Chris@48 336 }
Chris@48 337 }
Chris@48 338
Chris@45 339 emit mainModelChanged(m_mainModel);
Chris@45 340
Chris@45 341 // we already emitted modelAboutToBeDeleted for this
Chris@45 342 delete oldMainModel;
Chris@45 343 }
Chris@45 344
Chris@45 345 void
Chris@53 346 Document::addDerivedModel(TransformerId transform,
Chris@45 347 Model *inputModel,
Chris@53 348 const PluginTransformer::ExecutionContext &context,
Chris@45 349 Model *outputModelToAdd,
Chris@45 350 QString configurationXml)
Chris@45 351 {
Chris@45 352 if (m_models.find(outputModelToAdd) != m_models.end()) {
Chris@45 353 std::cerr << "WARNING: Document::addDerivedModel: Model already added"
Chris@45 354 << std::endl;
Chris@45 355 return;
Chris@45 356 }
Chris@45 357
Chris@45 358 // std::cerr << "Document::addDerivedModel: source is " << inputModel << " \"" << inputModel->objectName().toStdString() << "\"" << std::endl;
Chris@45 359
Chris@45 360 ModelRecord rec;
Chris@45 361 rec.source = inputModel;
Chris@45 362 rec.transform = transform;
Chris@45 363 rec.context = context;
Chris@45 364 rec.configurationXml = configurationXml;
Chris@45 365 rec.refcount = 0;
Chris@45 366
Chris@45 367 outputModelToAdd->setSourceModel(inputModel);
Chris@45 368
Chris@45 369 m_models[outputModelToAdd] = rec;
Chris@45 370
Chris@45 371 emit modelAdded(outputModelToAdd);
Chris@45 372 }
Chris@45 373
Chris@45 374
Chris@45 375 void
Chris@45 376 Document::addImportedModel(Model *model)
Chris@45 377 {
Chris@45 378 if (m_models.find(model) != m_models.end()) {
Chris@45 379 std::cerr << "WARNING: Document::addImportedModel: Model already added"
Chris@45 380 << std::endl;
Chris@45 381 return;
Chris@45 382 }
Chris@45 383
Chris@45 384 ModelRecord rec;
Chris@45 385 rec.source = 0;
Chris@45 386 rec.transform = "";
Chris@45 387 rec.refcount = 0;
Chris@45 388
Chris@45 389 m_models[model] = rec;
Chris@45 390
Chris@47 391 if (m_autoAlignment) alignModel(model);
Chris@47 392
Chris@45 393 emit modelAdded(model);
Chris@45 394 }
Chris@45 395
Chris@45 396 Model *
Chris@53 397 Document::addDerivedModel(TransformerId transform,
Chris@45 398 Model *inputModel,
Chris@53 399 const PluginTransformer::ExecutionContext &context,
Chris@45 400 QString configurationXml)
Chris@45 401 {
Chris@45 402 Model *model = 0;
Chris@45 403
Chris@45 404 for (ModelMap::iterator i = m_models.begin(); i != m_models.end(); ++i) {
Chris@45 405 if (i->second.transform == transform &&
Chris@45 406 i->second.source == inputModel &&
Chris@45 407 i->second.context == context &&
Chris@45 408 i->second.configurationXml == configurationXml) {
Chris@45 409 return i->first;
Chris@45 410 }
Chris@45 411 }
Chris@45 412
Chris@53 413 model = TransformerFactory::getInstance()->transform
Chris@45 414 (transform, inputModel, context, configurationXml);
Chris@45 415
Chris@45 416 if (!model) {
Chris@45 417 std::cerr << "WARNING: Document::addDerivedModel: no output model for transform " << transform.toStdString() << std::endl;
Chris@45 418 } else {
Chris@45 419 addDerivedModel(transform, inputModel, context, model, configurationXml);
Chris@45 420 }
Chris@45 421
Chris@45 422 return model;
Chris@45 423 }
Chris@45 424
Chris@45 425 void
Chris@45 426 Document::releaseModel(Model *model) // Will _not_ release main model!
Chris@45 427 {
Chris@45 428 if (model == 0) {
Chris@45 429 return;
Chris@45 430 }
Chris@45 431
Chris@45 432 if (model == m_mainModel) {
Chris@45 433 return;
Chris@45 434 }
Chris@45 435
Chris@45 436 bool toDelete = false;
Chris@45 437
Chris@45 438 if (m_models.find(model) != m_models.end()) {
Chris@45 439
Chris@45 440 if (m_models[model].refcount == 0) {
Chris@45 441 std::cerr << "WARNING: Document::releaseModel: model " << model
Chris@45 442 << " reference count is zero already!" << std::endl;
Chris@45 443 } else {
Chris@45 444 if (--m_models[model].refcount == 0) {
Chris@45 445 toDelete = true;
Chris@45 446 }
Chris@45 447 }
Chris@45 448 } else {
Chris@45 449 std::cerr << "WARNING: Document::releaseModel: Unfound model "
Chris@45 450 << model << std::endl;
Chris@45 451 toDelete = true;
Chris@45 452 }
Chris@45 453
Chris@45 454 if (toDelete) {
Chris@45 455
Chris@45 456 int sourceCount = 0;
Chris@45 457
Chris@45 458 for (ModelMap::iterator i = m_models.begin(); i != m_models.end(); ++i) {
Chris@45 459 if (i->second.source == model) {
Chris@45 460 ++sourceCount;
Chris@45 461 i->second.source = 0;
Chris@45 462 }
Chris@45 463 }
Chris@45 464
Chris@45 465 if (sourceCount > 0) {
Chris@45 466 std::cerr << "Document::releaseModel: Deleting model "
Chris@45 467 << model << " even though it is source for "
Chris@45 468 << sourceCount << " other derived model(s) -- resetting "
Chris@45 469 << "their source fields appropriately" << std::endl;
Chris@45 470 }
Chris@45 471
Chris@45 472 emit modelAboutToBeDeleted(model);
Chris@45 473 model->aboutToDelete();
Chris@45 474 m_models.erase(model);
Chris@45 475 delete model;
Chris@45 476 }
Chris@45 477 }
Chris@45 478
Chris@45 479 void
Chris@45 480 Document::deleteLayer(Layer *layer, bool force)
Chris@45 481 {
Chris@45 482 if (m_layerViewMap.find(layer) != m_layerViewMap.end() &&
Chris@45 483 m_layerViewMap[layer].size() > 0) {
Chris@45 484
Chris@45 485 std::cerr << "WARNING: Document::deleteLayer: Layer "
Chris@45 486 << layer << " [" << layer->objectName().toStdString() << "]"
Chris@45 487 << " is still used in " << m_layerViewMap[layer].size()
Chris@45 488 << " views!" << std::endl;
Chris@45 489
Chris@45 490 if (force) {
Chris@45 491
Chris@45 492 std::cerr << "(force flag set -- deleting from all views)" << std::endl;
Chris@45 493
Chris@45 494 for (std::set<View *>::iterator j = m_layerViewMap[layer].begin();
Chris@45 495 j != m_layerViewMap[layer].end(); ++j) {
Chris@45 496 // don't use removeLayerFromView, as it issues a command
Chris@45 497 layer->setLayerDormant(*j, true);
Chris@45 498 (*j)->removeLayer(layer);
Chris@45 499 }
Chris@45 500
Chris@45 501 m_layerViewMap.erase(layer);
Chris@45 502
Chris@45 503 } else {
Chris@45 504 return;
Chris@45 505 }
Chris@45 506 }
Chris@45 507
Chris@45 508 if (m_layers.find(layer) == m_layers.end()) {
Chris@45 509 std::cerr << "Document::deleteLayer: Layer "
Chris@45 510 << layer << " does not exist, or has already been deleted "
Chris@45 511 << "(this may not be as serious as it sounds)" << std::endl;
Chris@45 512 return;
Chris@45 513 }
Chris@45 514
Chris@45 515 m_layers.erase(layer);
Chris@45 516
Chris@52 517 std::cerr << "Document::deleteLayer: Removing, now have "
Chris@52 518 << m_layers.size() << " layers" << std::endl;
Chris@52 519
Chris@45 520 releaseModel(layer->getModel());
Chris@45 521 emit layerRemoved(layer);
Chris@45 522 emit layerAboutToBeDeleted(layer);
Chris@45 523 delete layer;
Chris@45 524 }
Chris@45 525
Chris@45 526 void
Chris@45 527 Document::setModel(Layer *layer, Model *model)
Chris@45 528 {
Chris@45 529 if (model &&
Chris@45 530 model != m_mainModel &&
Chris@45 531 m_models.find(model) == m_models.end()) {
Chris@45 532 std::cerr << "ERROR: Document::setModel: Layer " << layer
Chris@45 533 << " (\"" << layer->objectName().toStdString()
Chris@45 534 << "\") wants to use unregistered model " << model
Chris@45 535 << ": register the layer's model before setting it!"
Chris@45 536 << std::endl;
Chris@45 537 return;
Chris@45 538 }
Chris@45 539
Chris@45 540 Model *previousModel = layer->getModel();
Chris@45 541
Chris@45 542 if (previousModel == model) {
Chris@45 543 std::cerr << "WARNING: Document::setModel: Layer " << layer << " (\""
Chris@45 544 << layer->objectName().toStdString()
Chris@45 545 << "\") is already set to model "
Chris@45 546 << model << " (\""
Chris@45 547 << (model ? model->objectName().toStdString() : "(null)")
Chris@45 548 << "\")" << std::endl;
Chris@45 549 return;
Chris@45 550 }
Chris@45 551
Chris@45 552 if (model && model != m_mainModel) {
Chris@45 553 m_models[model].refcount ++;
Chris@45 554 }
Chris@45 555
Chris@45 556 if (model && previousModel) {
Chris@45 557 PlayParameterRepository::getInstance()->copyParameters
Chris@45 558 (previousModel, model);
Chris@45 559 }
Chris@45 560
Chris@45 561 LayerFactory::getInstance()->setModel(layer, model);
Chris@45 562
Chris@45 563 if (previousModel) {
Chris@45 564 releaseModel(previousModel);
Chris@45 565 }
Chris@45 566 }
Chris@45 567
Chris@45 568 void
Chris@45 569 Document::setChannel(Layer *layer, int channel)
Chris@45 570 {
Chris@45 571 LayerFactory::getInstance()->setChannel(layer, channel);
Chris@45 572 }
Chris@45 573
Chris@45 574 void
Chris@45 575 Document::addLayerToView(View *view, Layer *layer)
Chris@45 576 {
Chris@45 577 Model *model = layer->getModel();
Chris@45 578 if (!model) {
Chris@45 579 // std::cerr << "Document::addLayerToView: Layer (\""
Chris@45 580 // << layer->objectName().toStdString()
Chris@45 581 // << "\") with no model being added to view: "
Chris@45 582 // << "normally you want to set the model first" << std::endl;
Chris@45 583 } else {
Chris@45 584 if (model != m_mainModel &&
Chris@45 585 m_models.find(model) == m_models.end()) {
Chris@45 586 std::cerr << "ERROR: Document::addLayerToView: Layer " << layer
Chris@45 587 << " has unregistered model " << model
Chris@45 588 << " -- register the layer's model before adding the layer!" << std::endl;
Chris@45 589 return;
Chris@45 590 }
Chris@45 591 }
Chris@45 592
Chris@45 593 CommandHistory::getInstance()->addCommand
Chris@45 594 (new Document::AddLayerCommand(this, view, layer));
Chris@45 595 }
Chris@45 596
Chris@45 597 void
Chris@45 598 Document::removeLayerFromView(View *view, Layer *layer)
Chris@45 599 {
Chris@45 600 CommandHistory::getInstance()->addCommand
Chris@45 601 (new Document::RemoveLayerCommand(this, view, layer));
Chris@45 602 }
Chris@45 603
Chris@45 604 void
Chris@45 605 Document::addToLayerViewMap(Layer *layer, View *view)
Chris@45 606 {
Chris@45 607 bool firstView = (m_layerViewMap.find(layer) == m_layerViewMap.end() ||
Chris@45 608 m_layerViewMap[layer].empty());
Chris@45 609
Chris@45 610 if (m_layerViewMap[layer].find(view) !=
Chris@45 611 m_layerViewMap[layer].end()) {
Chris@45 612 std::cerr << "WARNING: Document::addToLayerViewMap:"
Chris@45 613 << " Layer " << layer << " -> view " << view << " already in"
Chris@45 614 << " layer view map -- internal inconsistency" << std::endl;
Chris@45 615 }
Chris@45 616
Chris@45 617 m_layerViewMap[layer].insert(view);
Chris@45 618
Chris@45 619 if (firstView) emit layerInAView(layer, true);
Chris@45 620 }
Chris@45 621
Chris@45 622 void
Chris@45 623 Document::removeFromLayerViewMap(Layer *layer, View *view)
Chris@45 624 {
Chris@45 625 if (m_layerViewMap[layer].find(view) ==
Chris@45 626 m_layerViewMap[layer].end()) {
Chris@45 627 std::cerr << "WARNING: Document::removeFromLayerViewMap:"
Chris@45 628 << " Layer " << layer << " -> view " << view << " not in"
Chris@45 629 << " layer view map -- internal inconsistency" << std::endl;
Chris@45 630 }
Chris@45 631
Chris@45 632 m_layerViewMap[layer].erase(view);
Chris@45 633
Chris@45 634 if (m_layerViewMap[layer].empty()) {
Chris@45 635 m_layerViewMap.erase(layer);
Chris@45 636 emit layerInAView(layer, false);
Chris@45 637 }
Chris@45 638 }
Chris@45 639
Chris@45 640 QString
Chris@45 641 Document::getUniqueLayerName(QString candidate)
Chris@45 642 {
Chris@45 643 for (int count = 1; ; ++count) {
Chris@45 644
Chris@45 645 QString adjusted =
Chris@45 646 (count > 1 ? QString("%1 <%2>").arg(candidate).arg(count) :
Chris@45 647 candidate);
Chris@45 648
Chris@45 649 bool duplicate = false;
Chris@45 650
Chris@45 651 for (LayerSet::iterator i = m_layers.begin(); i != m_layers.end(); ++i) {
Chris@45 652 if ((*i)->objectName() == adjusted) {
Chris@45 653 duplicate = true;
Chris@45 654 break;
Chris@45 655 }
Chris@45 656 }
Chris@45 657
Chris@45 658 if (!duplicate) return adjusted;
Chris@45 659 }
Chris@45 660 }
Chris@45 661
Chris@45 662 std::vector<Model *>
Chris@53 663 Document::getTransformerInputModels()
Chris@45 664 {
Chris@45 665 std::vector<Model *> models;
Chris@45 666
Chris@45 667 if (!m_mainModel) return models;
Chris@45 668
Chris@45 669 models.push_back(m_mainModel);
Chris@45 670
Chris@45 671 //!!! This will pick up all models, including those that aren't visible...
Chris@45 672
Chris@45 673 for (ModelMap::iterator i = m_models.begin(); i != m_models.end(); ++i) {
Chris@45 674
Chris@45 675 Model *model = i->first;
Chris@45 676 if (!model || model == m_mainModel) continue;
Chris@45 677 DenseTimeValueModel *dtvm = dynamic_cast<DenseTimeValueModel *>(model);
Chris@45 678
Chris@45 679 if (dtvm) {
Chris@45 680 models.push_back(dtvm);
Chris@45 681 }
Chris@45 682 }
Chris@45 683
Chris@45 684 return models;
Chris@45 685 }
Chris@45 686
Chris@50 687 bool
Chris@51 688 Document::canAlign()
Chris@50 689 {
Chris@53 690 TransformerId id = "vamp:match-vamp-plugin:match:path";
Chris@53 691 TransformerFactory *factory = TransformerFactory::getInstance();
Chris@53 692 return factory->haveTransformer(id);
Chris@50 693 }
Chris@50 694
Chris@45 695 void
Chris@45 696 Document::alignModel(Model *model)
Chris@45 697 {
Chris@45 698 if (!m_mainModel || model == m_mainModel) return;
Chris@45 699
Chris@45 700 RangeSummarisableTimeValueModel *rm =
Chris@45 701 dynamic_cast<RangeSummarisableTimeValueModel *>(model);
Chris@45 702 if (!rm) return;
Chris@48 703
Chris@48 704 if (rm->getAlignmentReference() == m_mainModel) return;
Chris@45 705
Chris@45 706 // This involves creating three new models:
Chris@45 707
Chris@45 708 // 1. an AggregateWaveModel to provide the mixdowns of the main
Chris@45 709 // model and the new model in its two channels, as input to the
Chris@45 710 // MATCH plugin
Chris@45 711
Chris@45 712 // 2. a SparseTimeValueModel, which is the model automatically
Chris@53 713 // created by FeatureExtractionPluginTransformer when running the
Chris@45 714 // MATCH plugin (thus containing the alignment path)
Chris@45 715
Chris@45 716 // 3. an AlignmentModel, which stores the path model and carries
Chris@45 717 // out alignment lookups on it.
Chris@45 718
Chris@45 719 // The first two of these are provided as arguments to the
Chris@45 720 // constructor for the third, which takes responsibility for
Chris@45 721 // deleting them. The AlignmentModel, meanwhile, is passed to the
Chris@45 722 // new model we are aligning, which also takes responsibility for
Chris@45 723 // it. We should not have to delete any of these new models here.
Chris@45 724
Chris@45 725 AggregateWaveModel::ChannelSpecList components;
Chris@45 726
Chris@45 727 components.push_back(AggregateWaveModel::ModelChannelSpec
Chris@45 728 (m_mainModel, -1));
Chris@45 729
Chris@45 730 components.push_back(AggregateWaveModel::ModelChannelSpec
Chris@45 731 (rm, -1));
Chris@45 732
Chris@45 733 Model *aggregate = new AggregateWaveModel(components);
Chris@45 734
Chris@53 735 TransformerId id = "vamp:match-vamp-plugin:match:path";
Chris@45 736
Chris@53 737 TransformerFactory *factory = TransformerFactory::getInstance();
Chris@45 738
Chris@45 739 Model *transformOutput = factory->transform
Chris@45 740 (id, aggregate,
Chris@53 741 factory->getDefaultContextForTransformer(id, aggregate),
Chris@45 742 "<plugin param-serialise=\"1\"/>");
Chris@45 743
Chris@45 744 SparseTimeValueModel *path = dynamic_cast<SparseTimeValueModel *>
Chris@45 745 (transformOutput);
Chris@45 746
Chris@45 747 if (!path) {
Chris@45 748 std::cerr << "Document::alignModel: ERROR: Failed to create alignment path (no MATCH plugin?)" << std::endl;
Chris@45 749 delete transformOutput;
Chris@45 750 delete aggregate;
Chris@45 751 return;
Chris@45 752 }
Chris@45 753
Chris@45 754 AlignmentModel *alignmentModel = new AlignmentModel
Chris@45 755 (m_mainModel, model, aggregate, path);
Chris@45 756
Chris@45 757 rm->setAlignment(alignmentModel);
Chris@45 758 }
Chris@45 759
Chris@45 760 void
Chris@45 761 Document::alignModels()
Chris@45 762 {
Chris@45 763 for (ModelMap::iterator i = m_models.begin(); i != m_models.end(); ++i) {
Chris@45 764 alignModel(i->first);
Chris@45 765 }
Chris@45 766 }
Chris@45 767
Chris@45 768 Document::AddLayerCommand::AddLayerCommand(Document *d,
Chris@45 769 View *view,
Chris@45 770 Layer *layer) :
Chris@45 771 m_d(d),
Chris@45 772 m_view(view),
Chris@45 773 m_layer(layer),
Chris@45 774 m_name(qApp->translate("AddLayerCommand", "Add %1 Layer").arg(layer->objectName())),
Chris@45 775 m_added(false)
Chris@45 776 {
Chris@45 777 }
Chris@45 778
Chris@45 779 Document::AddLayerCommand::~AddLayerCommand()
Chris@45 780 {
Chris@45 781 // std::cerr << "Document::AddLayerCommand::~AddLayerCommand" << std::endl;
Chris@45 782 if (!m_added) {
Chris@45 783 m_d->deleteLayer(m_layer);
Chris@45 784 }
Chris@45 785 }
Chris@45 786
Chris@45 787 void
Chris@45 788 Document::AddLayerCommand::execute()
Chris@45 789 {
Chris@45 790 for (int i = 0; i < m_view->getLayerCount(); ++i) {
Chris@45 791 if (m_view->getLayer(i) == m_layer) {
Chris@45 792 // already there
Chris@45 793 m_layer->setLayerDormant(m_view, false);
Chris@45 794 m_added = true;
Chris@45 795 return;
Chris@45 796 }
Chris@45 797 }
Chris@45 798
Chris@45 799 m_view->addLayer(m_layer);
Chris@45 800 m_layer->setLayerDormant(m_view, false);
Chris@45 801
Chris@45 802 m_d->addToLayerViewMap(m_layer, m_view);
Chris@45 803 m_added = true;
Chris@45 804 }
Chris@45 805
Chris@45 806 void
Chris@45 807 Document::AddLayerCommand::unexecute()
Chris@45 808 {
Chris@45 809 m_view->removeLayer(m_layer);
Chris@45 810 m_layer->setLayerDormant(m_view, true);
Chris@45 811
Chris@45 812 m_d->removeFromLayerViewMap(m_layer, m_view);
Chris@45 813 m_added = false;
Chris@45 814 }
Chris@45 815
Chris@45 816 Document::RemoveLayerCommand::RemoveLayerCommand(Document *d,
Chris@45 817 View *view,
Chris@45 818 Layer *layer) :
Chris@45 819 m_d(d),
Chris@45 820 m_view(view),
Chris@45 821 m_layer(layer),
Chris@45 822 m_name(qApp->translate("RemoveLayerCommand", "Delete %1 Layer").arg(layer->objectName())),
Chris@45 823 m_added(true)
Chris@45 824 {
Chris@45 825 }
Chris@45 826
Chris@45 827 Document::RemoveLayerCommand::~RemoveLayerCommand()
Chris@45 828 {
Chris@45 829 // std::cerr << "Document::RemoveLayerCommand::~RemoveLayerCommand" << std::endl;
Chris@45 830 if (!m_added) {
Chris@45 831 m_d->deleteLayer(m_layer);
Chris@45 832 }
Chris@45 833 }
Chris@45 834
Chris@45 835 void
Chris@45 836 Document::RemoveLayerCommand::execute()
Chris@45 837 {
Chris@45 838 bool have = false;
Chris@45 839 for (int i = 0; i < m_view->getLayerCount(); ++i) {
Chris@45 840 if (m_view->getLayer(i) == m_layer) {
Chris@45 841 have = true;
Chris@45 842 break;
Chris@45 843 }
Chris@45 844 }
Chris@45 845
Chris@45 846 if (!have) { // not there!
Chris@45 847 m_layer->setLayerDormant(m_view, true);
Chris@45 848 m_added = false;
Chris@45 849 return;
Chris@45 850 }
Chris@45 851
Chris@45 852 m_view->removeLayer(m_layer);
Chris@45 853 m_layer->setLayerDormant(m_view, true);
Chris@45 854
Chris@45 855 m_d->removeFromLayerViewMap(m_layer, m_view);
Chris@45 856 m_added = false;
Chris@45 857 }
Chris@45 858
Chris@45 859 void
Chris@45 860 Document::RemoveLayerCommand::unexecute()
Chris@45 861 {
Chris@45 862 m_view->addLayer(m_layer);
Chris@45 863 m_layer->setLayerDormant(m_view, false);
Chris@45 864
Chris@45 865 m_d->addToLayerViewMap(m_layer, m_view);
Chris@45 866 m_added = true;
Chris@45 867 }
Chris@45 868
Chris@45 869 void
Chris@45 870 Document::toXml(QTextStream &out, QString indent, QString extraAttributes) const
Chris@45 871 {
Chris@45 872 out << indent + QString("<data%1%2>\n")
Chris@45 873 .arg(extraAttributes == "" ? "" : " ").arg(extraAttributes);
Chris@45 874
Chris@45 875 if (m_mainModel) {
Chris@45 876 m_mainModel->toXml(out, indent + " ", "mainModel=\"true\"");
Chris@45 877 }
Chris@45 878
Chris@45 879 // Models that are not used in a layer that is in a view should
Chris@45 880 // not be written. Get our list of required models first.
Chris@45 881
Chris@45 882 std::set<const Model *> used;
Chris@45 883
Chris@45 884 for (LayerViewMap::const_iterator i = m_layerViewMap.begin();
Chris@45 885 i != m_layerViewMap.end(); ++i) {
Chris@45 886
Chris@45 887 if (i->first && !i->second.empty() && i->first->getModel()) {
Chris@45 888 used.insert(i->first->getModel());
Chris@45 889 }
Chris@45 890 }
Chris@45 891
Chris@45 892 for (ModelMap::const_iterator i = m_models.begin();
Chris@45 893 i != m_models.end(); ++i) {
Chris@45 894
Chris@45 895 const Model *model = i->first;
Chris@45 896 const ModelRecord &rec = i->second;
Chris@45 897
Chris@45 898 if (used.find(model) == used.end()) continue;
Chris@45 899
Chris@45 900 // We need an intelligent way to determine which models need
Chris@45 901 // to be streamed (i.e. have been edited, or are small) and
Chris@45 902 // which should not be (i.e. remain as generated by a
Chris@45 903 // transform, and are large).
Chris@45 904 //
Chris@45 905 // At the moment we can get away with deciding not to stream
Chris@45 906 // dense 3d models or writable wave file models, provided they
Chris@45 907 // were generated from a transform, because at the moment there
Chris@45 908 // is no way to edit those model types so it should be safe to
Chris@45 909 // regenerate them. That won't always work in future though.
Chris@45 910 // It would be particularly nice to be able to ask the user,
Chris@45 911 // as well as making an intelligent guess.
Chris@45 912
Chris@45 913 bool writeModel = true;
Chris@45 914 bool haveDerivation = false;
Chris@45 915
Chris@45 916 if (rec.source && rec.transform != "") {
Chris@45 917 haveDerivation = true;
Chris@45 918 }
Chris@45 919
Chris@45 920 if (haveDerivation) {
Chris@45 921 if (dynamic_cast<const WritableWaveFileModel *>(model)) {
Chris@45 922 writeModel = false;
Chris@45 923 } else if (dynamic_cast<const DenseThreeDimensionalModel *>(model)) {
Chris@45 924 writeModel = false;
Chris@45 925 }
Chris@45 926 }
Chris@45 927
Chris@45 928 if (writeModel) {
Chris@45 929 i->first->toXml(out, indent + " ");
Chris@45 930 }
Chris@45 931
Chris@45 932 if (haveDerivation) {
Chris@45 933
Chris@45 934 QString extentsAttributes;
Chris@45 935 if (rec.context.startFrame != 0 ||
Chris@45 936 rec.context.duration != 0) {
Chris@45 937 extentsAttributes = QString("startFrame=\"%1\" duration=\"%2\" ")
Chris@45 938 .arg(rec.context.startFrame)
Chris@45 939 .arg(rec.context.duration);
Chris@45 940 }
Chris@45 941
Chris@45 942 out << indent;
Chris@45 943 out << QString(" <derivation source=\"%1\" model=\"%2\" channel=\"%3\" domain=\"%4\" stepSize=\"%5\" blockSize=\"%6\" %7windowType=\"%8\" transform=\"%9\"")
Chris@45 944 .arg(XmlExportable::getObjectExportId(rec.source))
Chris@45 945 .arg(XmlExportable::getObjectExportId(i->first))
Chris@45 946 .arg(rec.context.channel)
Chris@45 947 .arg(rec.context.domain)
Chris@45 948 .arg(rec.context.stepSize)
Chris@45 949 .arg(rec.context.blockSize)
Chris@45 950 .arg(extentsAttributes)
Chris@45 951 .arg(int(rec.context.windowType))
Chris@45 952 .arg(XmlExportable::encodeEntities(rec.transform));
Chris@45 953
Chris@45 954 if (rec.configurationXml != "") {
Chris@45 955 out << ">\n " + indent + rec.configurationXml
Chris@45 956 + "\n" + indent + " </derivation>\n";
Chris@45 957 } else {
Chris@45 958 out << "/>\n";
Chris@45 959 }
Chris@45 960 }
Chris@45 961
Chris@45 962 //!!! We should probably own the PlayParameterRepository
Chris@45 963 PlayParameters *playParameters =
Chris@45 964 PlayParameterRepository::getInstance()->getPlayParameters(i->first);
Chris@45 965 if (playParameters) {
Chris@45 966 playParameters->toXml
Chris@45 967 (out, indent + " ",
Chris@45 968 QString("model=\"%1\"")
Chris@45 969 .arg(XmlExportable::getObjectExportId(i->first)));
Chris@45 970 }
Chris@45 971 }
Chris@45 972
Chris@45 973 for (LayerSet::const_iterator i = m_layers.begin();
Chris@45 974 i != m_layers.end(); ++i) {
Chris@45 975
Chris@45 976 (*i)->toXml(out, indent + " ");
Chris@45 977 }
Chris@45 978
Chris@45 979 out << indent + "</data>\n";
Chris@45 980 }
Chris@45 981
Chris@45 982