|| || || || || || || ||

---

CameraMovementReader.h

Go to the documentation of this file.

#ifndef Impala_Application_SDash_CameraMovementReader_h
#define Impala_Application_SDash_CameraMovementReader_h

#include <string>
#include <map>
#include <vector>

#include "Util/StringParser.h"
#include "Basis/FileExists.h"
#include "Util/IOBufferFile.h"

#include "Visualization/Video.h"

#include "Application/sdash/CameraEvent.h"

namespace Impala {
namespace Application {
namespace SDash {

class CameraMovementReader
{

public:

    // reads from the base path camera movements from files, for the video's specified;
    // adds as a side effect every frame nr involved in movement, to the video's movement annotations
    void Read(
        const std::string& basePath, 
        std::vector<Visualization::Video>& videos, 
        std::map<int, std::vector<CameraEvent> >& camEventSchedule)
    {

        int totalEventsFound = 0;
        int totalFramesFound = 0;
        std::vector<Visualization::Video>::iterator videoIter = videos.begin();
        while (videoIter != videos.end())
        {
            int videoId = videoIter->mId;
            int eventsFound = 0;
            int framesFound = 0;
            std::string filePath = basePath + "cam_moving_" + MakeString(videoId) + ".csv";
            if (FileExists(filePath))
            {
                // open the file
                Util::IOBufferFile file(filePath, true, true);

                int prevFrameNr = -2;
                int frameNr = -1;
                CameraEvent camEvent;
                do
                {
                    // read a line, parse into frame nr
                    std::string line = file.ReadLine();
                    if (line.empty())
                    {
                        if (prevFrameNr >= 0)
                        {
                            // end last sequence
                            int firstFrameOfSeq = camEvent.FrameNr();
                            int eventTimeSec = firstFrameOfSeq * videoIter->GetTimePerFrame();
                            camEvent.SetComments(MakeString(2 + prevFrameNr - firstFrameOfSeq) + " frames");
                            camEventSchedule[eventTimeSec].push_back(camEvent);
                        }

                        break;
                    }

                    Util::StringParser parser(line);
                    frameNr = parser.GetInt(';');
                    if (frameNr > prevFrameNr + 1)
                    {
                        if (prevFrameNr >= 0)
                        {
                            // end previous sequence
                            int firstFrameOfSeq = camEvent.FrameNr();
                            int eventTimeSec = firstFrameOfSeq * videoIter->GetTimePerFrame();
                            camEvent.SetComments(MakeString(2 + prevFrameNr - firstFrameOfSeq) + " frames");
                            camEventSchedule[eventTimeSec].push_back(camEvent);
                        }

                        // start of a movement sequence
                        int camNr = videoId + 1;
                        camEvent = CameraEvent(camNr, 3, frameNr);
                        eventsFound++;
                        videoIter->AddMovementFrame(frameNr); // first frame of seq.
                        framesFound++;
                    }
                    //lastCamEventP->SetComments(MakeString(frameNr - lastCamEventP->FrameNr()) + " frames");
                    videoIter->AddMovementFrame(frameNr + 1); // movement ends at earliest at this frame
                    framesFound++;
                    prevFrameNr = frameNr;
                } while (true);
            }

            std::cout << eventsFound << " camera movements found for video #" << videoId << std::endl;
            std::cout << framesFound << " camera movement frames found for video #" << videoId << std::endl;
            totalEventsFound += eventsFound;
            totalFramesFound += framesFound;
            videoIter++;
        }

        std::cout << totalEventsFound << " camera movements found in total" << std::endl;
        std::cout << totalFramesFound << " camera movement frames found in total" << std::endl;
        //return 0;
    }

private:


}; //class

}
}
}

#endif

---