CEBL  2.1
DataProcess.hpp
Go to the documentation of this file.
1 /*
2 * CEBL : CSU EEG Brain-Computer Interface Lab
3 *
4 * Author: Jeshua Bratman - jeshuabratman@gmail.com
5 *
6 * This file is part of CEBL.
7 *
8 * CEBL is free software; you can redistribute it and/or modify it.
9 * We only ask that if you use our code that you cite the source in
10 * your project or publication.
11 *
12 * EEG Group (www.cs.colostate.edu/eeg)
13 * Department of Computer Science
14 * Colorado State University
15 *
16 */
17 
26 #ifndef DATAPROCESS_H
27 #define DATAPROCESS_H
28 
29 #include "EEGTrainingData.hpp"
30 #include "EEGData.hpp"
31 
32 //forward declarations
33 class CEBLModel;
34 
36 {
37 private:
38  CEBLModel * model;
39 
40  //options to process
41  bool remove_disabled;
42  bool filter;
43  bool reference;
44 
45 public:
46  DataProcess(CEBLModel * model);
47  ~DataProcess();
48 
50  EEGData & process(EEGData & data);
51  EEGData & process(EEGData & data,
52  bool remove_disabled, bool reference, bool filter);
53 
54  //SETTERS
55  void setRemoveEnabled(bool r) { remove_disabled = r; }
56  void setFilterEnabled(bool f) { filter = f; }
57  void setReferenceEnabled(bool r) { reference = r; }
58 
59  //GETTERS
60  bool getRemoveEnabled() { return remove_disabled; }
61  bool getFilterEnabled() { return filter; }
62  bool getReferenceEnabled() { return reference; }
63 
64 };
65 
66 
67 #endif