Mercurial Hosting > traffic-intelligence
annotate include/InputFrameListModule.h @ 417:a2ff03a52b73
added basic logging capability for debugging
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Thu, 26 Sep 2013 17:07:44 -0400 |
parents | b829ebdc18e6 |
children |
rev | line source |
---|---|
400
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
1 #ifndef INPUT_FRAME_LIST_MODULE_H |
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
2 #define INPUT_FRAME_LIST_MODULE_H |
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
3 |
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
4 #include "InputFrameProviderIface.h" |
401
b829ebdc18e6
simplified input of directories of video frames (simply use the video filename parameter to point at the directory)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
400
diff
changeset
|
5 |
400
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
6 #include <string> |
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
7 #include <vector> |
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
8 |
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
9 class InputFrameListModule : public InputFrameProviderIface |
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
10 { |
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
11 public: |
401
b829ebdc18e6
simplified input of directories of video frames (simply use the video filename parameter to point at the directory)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
400
diff
changeset
|
12 InputFrameListModule(const std::string& _dirname); |
400
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
13 ~InputFrameListModule(); |
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
14 |
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
15 bool getNextFrame(cv::Mat&); |
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
16 unsigned int getNbFrames(); |
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
17 bool isOpen() const { return mInit;} |
401
b829ebdc18e6
simplified input of directories of video frames (simply use the video filename parameter to point at the directory)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
400
diff
changeset
|
18 void setFrameNumber(const unsigned int& frameNumber); |
400
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
19 |
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
20 virtual const cv::Size& getSize() const { return mSize;} |
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
21 private: |
401
b829ebdc18e6
simplified input of directories of video frames (simply use the video filename parameter to point at the directory)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
400
diff
changeset
|
22 void loadImageList(void); |
b829ebdc18e6
simplified input of directories of video frames (simply use the video filename parameter to point at the directory)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
400
diff
changeset
|
23 std::vector<std::string> filenames; |
400
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
24 unsigned int mCurrentIdx; |
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
25 bool mInit; |
401
b829ebdc18e6
simplified input of directories of video frames (simply use the video filename parameter to point at the directory)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
400
diff
changeset
|
26 std::string dirname; |
400
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
27 cv::Size mSize; |
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
28 }; |
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
29 |
7ef1071e3cc3
clean up of input classes for list of images and video files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
399
diff
changeset
|
30 #endif |