annotate transform/ModelTransformer.h @ 1197:fbe0fd84cb50 spectrogram-minor-refactor

Float/double conversion fixes
author Chris Cannam
date Mon, 01 Aug 2016 16:25:06 +0100
parents 6ea7761a418b
children 5a1198083d9a
rev   line source
Chris@320 1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
Chris@320 2
Chris@320 3 /*
Chris@320 4 Sonic Visualiser
Chris@320 5 An audio file viewer and annotation editor.
Chris@320 6 Centre for Digital Music, Queen Mary, University of London.
Chris@320 7 This file copyright 2006 Chris Cannam.
Chris@320 8
Chris@320 9 This program is free software; you can redistribute it and/or
Chris@320 10 modify it under the terms of the GNU General Public License as
Chris@320 11 published by the Free Software Foundation; either version 2 of the
Chris@320 12 License, or (at your option) any later version. See the file
Chris@320 13 COPYING included with this distribution for more information.
Chris@320 14 */
Chris@320 15
Chris@329 16 #ifndef _TRANSFORMER_H_
Chris@329 17 #define _TRANSFORMER_H_
Chris@320 18
Chris@320 19 #include "base/Thread.h"
Chris@320 20
Chris@320 21 #include "data/model/Model.h"
Chris@320 22
Chris@350 23 #include "Transform.h"
Chris@350 24
Chris@320 25 /**
Chris@331 26 * A ModelTransformer turns one data model into another.
Chris@320 27 *
Chris@331 28 * Typically in this application, a ModelTransformer might have a
Chris@320 29 * DenseTimeValueModel as its input (e.g. an audio waveform) and a
Chris@320 30 * SparseOneDimensionalModel (e.g. detected beats) as its output.
Chris@320 31 *
Chris@331 32 * The ModelTransformer typically runs in the background, as a
Chris@331 33 * separate thread populating the output model. The model is
Chris@331 34 * available to the user of the ModelTransformer immediately, but may
Chris@331 35 * be initially empty until the background thread has populated it.
Chris@320 36 */
Chris@320 37
Chris@331 38 class ModelTransformer : public Thread
Chris@320 39 {
Chris@320 40 public:
Chris@331 41 virtual ~ModelTransformer();
Chris@320 42
Chris@849 43 typedef std::vector<Model *> Models;
Chris@849 44
Chris@350 45 class Input {
Chris@350 46 public:
Chris@350 47 Input(Model *m) : m_model(m), m_channel(-1) { }
Chris@350 48 Input(Model *m, int c) : m_model(m), m_channel(c) { }
Chris@350 49
Chris@350 50 Model *getModel() const { return m_model; }
Chris@350 51 void setModel(Model *m) { m_model = m; }
Chris@350 52
Chris@350 53 int getChannel() const { return m_channel; }
Chris@350 54 void setChannel(int c) { m_channel = c; }
Chris@350 55
Chris@350 56 protected:
Chris@350 57 Model *m_model;
Chris@350 58 int m_channel;
Chris@350 59 };
Chris@350 60
Chris@361 61 /**
Chris@361 62 * Hint to the processing thread that it should give up, for
Chris@361 63 * example because the process is going to exit or we want to get
Chris@361 64 * rid of the input model. Caller should still wait() and/or
Chris@361 65 * delete the transform before assuming its input and output
Chris@361 66 * models are no longer required.
Chris@361 67 */
Chris@320 68 void abandon() { m_abandoned = true; }
Chris@320 69
Chris@361 70 /**
Chris@1079 71 * Return true if the processing thread is being or has been
Chris@1079 72 * abandoned, i.e. if abandon() has been called.
Chris@1079 73 */
Chris@1079 74 bool isAbandoned() const { return m_abandoned; }
Chris@1079 75
Chris@1079 76 /**
Chris@361 77 * Return the input model for the transform.
Chris@361 78 */
Chris@350 79 Model *getInputModel() { return m_input.getModel(); }
Chris@361 80
Chris@361 81 /**
Chris@361 82 * Return the input channel spec for the transform.
Chris@361 83 */
Chris@350 84 int getInputChannel() { return m_input.getChannel(); }
Chris@350 85
Chris@361 86 /**
Chris@849 87 * Return the set of output models created by the transform or
Chris@849 88 * transforms. Returns an empty list if any transform could not
Chris@849 89 * be initialised; an error message may be available via
Chris@849 90 * getMessage() in this situation.
Chris@361 91 */
Chris@877 92 Models getOutputModels() { return m_outputs; }
Chris@361 93
Chris@361 94 /**
Chris@849 95 * Return the set of output models, also detaching them from the
Chris@849 96 * transformer so that they will not be deleted when the
Chris@849 97 * transformer is. The caller takes ownership of the models.
Chris@361 98 */
Chris@877 99 Models detachOutputModels() {
Chris@877 100 m_detached = true;
Chris@877 101 return getOutputModels();
Chris@877 102 }
Chris@877 103
Chris@877 104 /**
Chris@877 105 * Return any additional models that were created during
Chris@877 106 * processing. This might happen if, for example, a transform was
Chris@877 107 * configured to split a multi-bin output into separate single-bin
Chris@877 108 * models as it processed. These should not be queried until after
Chris@877 109 * the transform has completed.
Chris@877 110 */
Chris@877 111 virtual Models getAdditionalOutputModels() { return Models(); }
Chris@877 112
Chris@877 113 /**
Chris@877 114 * Return true if the current transform is one that may produce
Chris@877 115 * additional models (to be retrieved through
Chris@877 116 * getAdditionalOutputModels above).
Chris@877 117 */
Chris@877 118 virtual bool willHaveAdditionalOutputModels() { return false; }
Chris@877 119
Chris@877 120 /**
Chris@877 121 * Return the set of additional models, also detaching them from
Chris@877 122 * the transformer. The caller takes ownership of the models.
Chris@877 123 */
Chris@877 124 virtual Models detachAdditionalOutputModels() {
Chris@877 125 m_detachedAdd = true;
Chris@877 126 return getAdditionalOutputModels();
Chris@877 127 }
Chris@320 128
Chris@361 129 /**
Chris@361 130 * Return a warning or error message. If getOutputModel returned
Chris@361 131 * a null pointer, this should contain a fatal error message for
Chris@361 132 * the transformer; otherwise it may contain a warning to show to
Chris@361 133 * the user about e.g. suboptimal block size or whatever.
Chris@361 134 */
Chris@361 135 QString getMessage() const { return m_message; }
Chris@361 136
Chris@320 137 protected:
Chris@350 138 ModelTransformer(Input input, const Transform &transform);
Chris@849 139 ModelTransformer(Input input, const Transforms &transforms);
Chris@320 140
Chris@849 141 Transforms m_transforms;
Chris@350 142 Input m_input; // I don't own the model in this
Chris@849 143 Models m_outputs; // I own this, unless...
Chris@320 144 bool m_detached; // ... this is true.
Chris@877 145 bool m_detachedAdd;
Chris@320 146 bool m_abandoned;
Chris@361 147 QString m_message;
Chris@320 148 };
Chris@320 149
Chris@320 150 #endif