source: trunk/VUT/GtpVisibilityPreprocessor/src/Camera.cpp @ 182

#include <algorithm>
#include <qstring.h>
#include <qimage.h>
#include "Camera.h"
#include "Ray.h"
#include "KdTree.h"
#include "Mesh.h"
#include "Exporter.h"

bool
Camera::SnapImage(string filename,
                  KdTree *tree
                  )
{
  int x;
  int y;

  bool exportRays = true;

  
#if QT_VERSION < 0x040000
  QImage image(mWidth, mHeight, 32);
#else
  //QImage image(mWidth, mHeight, QImage::Format_RGB32);
#endif
  
  vector<Ray> rays;

  long t1 = GetTime();

  Ray ray;
  for (y = 0; y < mHeight; y++) {
    cout<<"+";
    for (x = 0; x < mWidth; x++) {
      SetupRay(ray, x, y);
      if (tree->CastRay(ray))
        if (ray.intersections.size()) {
          sort(ray.intersections.begin(), ray.intersections.end());
          MeshInstance *mesh = (MeshInstance*)ray.intersections[0].mObject;
          RgbColor color(1,1,1);
          if (mesh->GetMesh()->mMaterial)
            color = mesh->GetMesh()->mMaterial->mDiffuseColor;
#if QT_VERSION < 0x040000
          image.setPixel(x, y, qRgb(color.r*255,
                                    color.g*255,
                                    color.b*255
                                    ));
#endif

        }
      
      if (exportRays && (x==222) && (y==97))
        rays.push_back(ray);
    }
  }
  
  long t2 = GetTime();
  cout<<"#RAY_CAST_TIME\n";
  cout<<TimeDiff(t1, t2)<<"\n";
  
  cout<<"Saving image"<<endl;
#if QT_VERSION < 0x040000
  image.save(filename.c_str(), "PNG");
#endif
  Exporter *exporter = NULL;
  if (exportRays) {
    exporter = Exporter::GetExporter(filename + "-rays" + ".x3d");
    exporter->SetWireframe();
    exporter->ExportKdTree(*tree);
    exporter->ExportRays(rays, 10000);
    exporter->SetFilled();
    int k =0;
    for (int j=0; j < rays.size(); j++)
      if (rays[j].leaves.size()) {
        Ray *ray = &(rays[j]);
        int i;
        if (1)
        for (i= 0; i < ray->leaves.size(); i++)
          exporter->ExportBox(tree->GetBox(ray->leaves[i]));
        if (0)
        for (i= 0; i < ray->meshes.size(); i++)
          exporter->ExportIntersectable(ray->meshes[i]);

      }

    delete exporter;
  }
  
  
  return true;
}


void
Camera::SetupRay(Ray &ray, const int x, const int y)
{
  Vector3 xv = mRight*((x - mWidth/2)/(float)mWidth);
  Vector3 yv = mUp*((y - mHeight/2)/(float)mHeight);
  Vector3 target = xv + yv + mDirection;

  ray.intersections.clear();
  ray.leaves.clear();
  ray.meshes.clear();
  
  ray.Init(mPosition, target, Ray::LOCAL_RAY);
  
}