summaryrefslogtreecommitdiff
path: root/src/cachemanager.h
blob: 1660ecfde3c8f0aa58a2373f0a34d937cdd13b35 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/***************************************************************************
 *            cachemanager.h
 *
 *  Fri Apr 10 10:39:24 CEST 2015
 *  Copyright 2015 Jonas Suhr Christensen
 *  jsc@umbraculum.org
 ****************************************************************************/

/*
 *  This file is part of DrumGizmo.
 *
 *  DrumGizmo 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.
 *
 *  DrumGizmo 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 DrumGizmo; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA.
 */
#ifndef __DRUMGIZMO_CACHEMANAGER_H__
#define __DRUMGIZMO_CACHEMANAGER_H__

#include <string>
#include <list>
#include <vector>

#include "thread.h"
#include "semaphore.h"
#include "mutex.h"

#include "audiotypes.h"
#include "audiofile.h"

#define CACHE_DUMMYID -2
#define CACHE_NOID -1

#define CHUNK_MULTIPLIER 16

class AudioFile;
typedef int cacheid_t;
class AFile;


//TODO:
// 1: Move nodata initialisation to init method.
// 2: Make semaphore in thread to block init call until thread has been started.

//// next
// Pre: preloaded contains 2 x framesize. chunk size is framesize.
// allocate 2 chunks and copy initial_samples_needed to first buffer from
// preloaded data and enough to fill up the second buffer from preloaded
// returns the first buffer and its size in &size.
// get id from "free stack" and store pointers to buffers in id vector.
// event: open sndfile handle (if not already open) and increase refcount

//// next
// Return which ever buffer is the front, swap them and add event to load the
// next chunk.

//// close
// decrement file handle refcounter and close file if it is 0.
// free the 2 buffers
// (do not erase from the id vector), push index to
// "free stack" for reuse.

class CacheManager : public Thread {
public:
  /**
   * Empty constructor...
   */
  CacheManager();

  /**
   * Destroy object and stop thread if needed.
   */
  ~CacheManager();

  /**
   * Initialise cache manager and allocate needed resources
   * This method starts the cache manager thread.
   * This method blocks until the thread has been started.
   * @param poolsize The maximum number of parellel events supported. 
   */
  void init(size_t poolsize, bool threaded);

  /**
   * Stop thread and clean up resources.
   * This method blocks until the thread has stopped.
   */
  void deinit();

  /**
   * Register new cache entry.
   * Prepares an entry in the cache manager for future disk streaming.
   * @param file A pointer to the file which is to be streamed from.
   * @param initial_samples_needed The number of samples needed in the first
   *  read that is not nessecarily of framesize. This is the number of samples
   *  from the input event offset to the end of the frame in which it resides.
   *  initial_samples_needed <= framesize.
   * @param channel The channel to which the cache id will be bound.
   * @param [out] new_id The newly created cache id.
   * @return A pointer to the first buffer containing the
   *  'initial_samples_needed' number of samples.
   */
  sample_t *open(AudioFile *file, size_t initial_samples_needed, int channel,
                 cacheid_t &new_id);

  /**
   * Get next buffer.
   * Returns the next buffer for reading based on cache id.
   * This function will (if needed) schedule a new disk read to make sure that
   * data is available in the next call to this method.
   * @param id The cache id to read from.
   * @param [out] size The size of the returned buffer.
   * @return A pointer to the buffer.
   */
  sample_t *next(cacheid_t id, size_t &size);

  /**
   * Unregister cache entry.
   * Close associated file handles and free associated buffers.
   * @param id The cache id to close.
   */
  void close(cacheid_t id);

  /**
   * Set internal framesize used when iterating through cache buffers.
   */
  void setFrameSize(size_t framesize);

  /**
   * Control reader thread.
   * Set to true to make reading happen threaded, false to do all reading sync.
   */
  void setAsyncMode(bool async);

  ///! Internal thread main method - needs to be public.
  void thread_main();

  class Channel {
  public:
    size_t channel;
    sample_t* samples;
    size_t num_samples;
    volatile bool* ready;
  };

private:
  size_t framesize;
  sample_t *nodata;

  typedef struct {
    AFile *afile;
    size_t channel;
    size_t pos; //< File possition
    volatile bool ready;
    sample_t *front;
    sample_t *back;
    size_t localpos; //< Intra buffer (front) position.

    sample_t* preloaded_samples; // NULL means not active.
    size_t preloaded_samples_size;

  } cache_t;

  typedef enum {
    LOADNEXT = 0,
    CLOSE = 1
  } cmd_t;

  typedef struct {
    cmd_t cmd;

    // For close event:
    cacheid_t id;

    // For load next event:
    size_t pos;
    AFile *afile;
    std::list<CacheManager::Channel> channels;
  } cevent_t;

  cevent_t createLoadNextEvent(AFile *afile, size_t channel, size_t pos,
                               sample_t* buffer, volatile bool* ready);
  cevent_t createCloseEvent(cacheid_t id);

  void handleLoadNextEvent(cevent_t& e);
  void handleCloseEvent(cevent_t& e);

  void handleEvent(cevent_t& e);
  void pushEvent(cevent_t& e);

  std::vector<cache_t> id2cache; 

  // Protected by mutex:
  std::list<cevent_t> eventqueue;
  std::list<cacheid_t> availableids; 
  
  Mutex m_events;
  Mutex m_ids;

  bool threaded; // Indicates if we are running in thread mode or offline mode.
  Semaphore sem;
  Semaphore sem_run;
  bool running;

  std::map<std::string, AFile*> files;
};

#endif/*__DRUMGIZMO_CACHEMANAGER_H__*/