source: flair-src/trunk/lib/FlairSensorActuator/src/Camera.cpp@ 155

// %flair:license{
// This file is part of the Flair framework distributed under the
// CECILL-C License, Version 1.0.
// %flair:license}
//  created:    2014/03/06
//  filename:   Camera.cpp
//
//  author:     Guillaume Sanahuja
//              Copyright Heudiasyc UMR UTC/CNRS 7253
//
//  version:    $Id: $
//
//  purpose:    Virtual class for Camera
//
//
/*********************************************************************/

#include "Camera.h"
#include <FrameworkManager.h>
#include <Tab.h>
#include <TabWidget.h>
#include <GroupBox.h>
#include <GridLayout.h>
#include <DataPlot1D.h>
#include <Picture.h>
#include <VisionFilter.h>
#include <highgui.h>
#include <fstream>

using std::string;
using namespace flair::core;
using namespace flair::gui;

namespace flair {
namespace sensor {

Camera::Camera(string name, uint16_t width,
               uint16_t height, cvimage::Type::Format format)
    : IODevice(getFrameworkManager(), name) {
  plot_tab = NULL;
        logFormat=LogFormat::NONE;

  // do not allocate imagedata, allocation is done by the camera
  output = new cvimage((IODevice *)this, width, height, format, "out", false);
        
  // station sol
  main_tab = new Tab(getFrameworkManager()->GetTabWidget(), name);
  tab = new TabWidget(main_tab->NewRow(), name);
  sensor_tab = new Tab(tab, "Setup");
  setup_groupbox = new GroupBox(sensor_tab->NewRow(), name);
  setup_layout = new GridLayout(sensor_tab->NewRow(), "setup");
}

//This contructor must only be called for a simulated device.
Camera::Camera(const IODevice *parent, std::string name)
    : IODevice(parent, name) {
  plot_tab = NULL;
  main_tab = NULL;
  tab = NULL;
  sensor_tab = NULL;
  setup_groupbox = NULL;

  output = NULL;
        logFormat=LogFormat::NONE;
}

Camera::~Camera() {
  if (main_tab != NULL)
    delete main_tab;
}

void Camera::SetLogFormat(LogFormat logFormat) {
        this->logFormat=logFormat;
        switch(logFormat) {
                case LogFormat::RAW:
                        AddDataToLog(output);
                        Warn("raw log of cvimage is not yet implemented\n");
                        break;
                case LogFormat::JPG:
                                Warn("logging cvimage to jpeg\n");
                                Warn("jpeg are not included in classical dbt file, as dbt does not handle variable length\n");
                        break;
                default:
                        Warn("LogFormat unknown\n");
        }
}

DataType const &Camera::GetOutputDataType() const {
  return output->GetDataType();
}

GroupBox *Camera::GetGroupBox(void) const { return setup_groupbox; }

GridLayout *Camera::GetLayout(void) const { return setup_layout; }

void Camera::UseDefaultPlot(const core::cvimage *image) {
  if (tab == NULL) {
    Err("not applicable for simulation part.\n");
    return;
  }

  plot_tab = new Tab(tab, "Picture");
  Picture *plot = new Picture(plot_tab->NewRow(), ObjectName(), image);
}

Tab *Camera::GetPlotTab(void) const { return plot_tab; }

uint16_t Camera::Width(void) const { return output->GetDataType().GetWidth(); }

uint16_t Camera::Height(void) const {
  return output->GetDataType().GetHeight();
}

core::cvimage *Camera::Output(void) { return output; }

void Camera::ProcessUpdate(core::io_data* data) {
        if(getFrameworkManager()->IsLogging() && getFrameworkManager()->IsDeviceLogged(this)) {
                if(logFormat==LogFormat::JPG) {
                        data->GetMutex();
                        IplImage *img=((cvimage*)data)->img;
                        string filename=getFrameworkManager()->GetLogPath()+"/"+ObjectName()+"_"+std::to_string(data->DataTime())+".jpg";
                        switch(((cvimage*)data)->GetDataType().GetFormat()) {
                                case cvimage::Type::Format::Gray:
                                        saveToJpeg(img,filename,PictureFormat_t::Gray,PictureFormat_t::Gray);
                                        break;
                                case cvimage::Type::Format::BGR:
                                        saveToJpeg(img,filename,PictureFormat_t::RGB,PictureFormat_t::RGB);
                                        break;
                                case cvimage::Type::Format::UYVY:
                                        saveToJpeg(img,filename,PictureFormat_t::YUV_422ile,PictureFormat_t::YUV_422p);
                                        break;
                                default:
                                        Warn("cannot log to this format\n");
                                        break;
                        }
                        data->ReleaseMutex();
                }
        }
        IODevice::ProcessUpdate(data);
}

void Camera::SavePictureToFile(string filename) const {
        if(filename=="") filename="./"+ObjectName()+"_"+std::to_string(GetTime())+".jpg";
    string::size_type idx = filename.rfind('.');

        if(idx != string::npos) {
                Printf("saving %s\n", filename.c_str());
                string extension = filename.substr(idx+1);
                
                output->GetMutex();
                if(extension=="jpg") {
                        if(output->GetDataType().GetFormat()==cvimage::Type::Format::Gray) saveToJpeg(output->img,filename,PictureFormat_t::Gray,PictureFormat_t::Gray);
                        if(output->GetDataType().GetFormat()==cvimage::Type::Format::BGR) saveToJpeg(output->img,filename,PictureFormat_t::RGB,PictureFormat_t::RGB);
                        if(output->GetDataType().GetFormat()==cvimage::Type::Format::UYVY) saveToJpeg(output->img,filename,PictureFormat_t::YUV_422ile,PictureFormat_t::YUV_422p);
                } else {
                        cvSaveImage(filename.c_str(),output->img);
                }
                output->ReleaseMutex();
                        
        } else {
                Warn("saving %s no file extension!\n", filename.c_str());
        }
}

void Camera::SaveRawPictureToFile(string filename) const {
  Printf("saving %s, size %i\n", filename.c_str(), output->img->imageSize);
  std::ofstream pFile;
  pFile.open(filename);
  output->GetMutex();
  pFile.write(output->img->imageData, output->img->imageSize);
  output->ReleaseMutex();

  pFile.close();
}

} // end namespace sensor
} // end namespace flair