RichardsFastSlow/src/qt/output/output_view.cpp

#include "qt/output/output_view.hpp"

QtOutputView::QtOutputView(QtKernel* k){
  kernel=k;
  nX=kernel->geometry.nX;
  dX=kernel->geometry.dX;
  lX=kernel->geometry.lX;
  factor=kernel->factor;
};

void
QtOutputView::initializeGL(){
  glClearColor(1,1,1,1);
  glEnable(GL_DEPTH_TEST);
  glEnable(GL_LIGHT0);
  glEnable(GL_LIGHTING);
  glColorMaterial(GL_FRONT_AND_BACK, GL_AMBIENT_AND_DIFFUSE);
  glEnable(GL_COLOR_MATERIAL);
}

void
QtOutputView::paintGL(){
  glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
  if(step>kernel->step) return;
  double h=height();
  double w=width();
  glBegin(GL_TRIANGLES);
  for(size_t i=0;i<nX-1;++i){
    size_t max_left=kernel->nZ(i);
    size_t max_right=kernel->nZ(i+1);
    size_t left=0;
    size_t right=0;
    while(left<max_left-1 and right<max_right-1){
      if(kernel->Z(i,left+1)<=kernel->Z(i+1,right+1)){
        drawTriangle(i,left,i+1,right,i,left+1);
        ++left;
      }
      else{
        drawTriangle(i,left,i+1,right,i+1,right+1);
        ++right;
      }
    }
    if(left==max_left-1){
      while(right<max_right-1){
        drawTriangle(i,left,i+1,right,i+1,right+1);
        ++right;
      }
    }
    if(right==max_right-1){
      while(left<max_left-1){
        drawTriangle(i,left,i+1,right,i,left+1);
        ++left;
      }
    }
  }
  glEnd();
  //Draw area under hbot
  glColor3f(0.6,0.6,0.6);
  glLineWidth(3);
  glBegin(GL_QUADS);
  for(size_t ix=0;ix<nX-1;ix+=1){
    double x=ix*dX;
    glVertex3f(x,0,1);
    glVertex3f(x,kernel->geometry.hbot[ix],0);
    glVertex3f(x+dX,kernel->geometry.hbot[ix+1],0);
    glVertex3f(x+dX,0,0);

  }
  glEnd();

  //Draw hsoil
  glColor3f(0.6,0.3,0);
  glLineWidth(1);
  glBegin(GL_LINE_STRIP);
  for(size_t ix=0;ix<nX;ix+=1){
    double x=ix*dX;
    glVertex3f(x,kernel->geometry.hsoil[ix],0.8);
  }
  glEnd();

  //Draw hydr
  glColor3f(1,0,0);
  glLineWidth(3);
  glBegin(GL_LINE_STRIP);
  for(size_t ix=0;ix<nX;ix+=1){
    double x=ix*dX;
    glVertex3f(x,kernel->solution[step]->hydr[ix]-Psat/(Physics::rho*Physics::g),1);
  }
  glEnd();

  //Draw hov
  glColor3f(0,0,0);
  glLineWidth(1);
  glBegin(GL_LINE_STRIP);
  for(size_t ix=0;ix<nX;ix+=1){
    double x=ix*dX;
    double y=kernel->solution[step]->hov[ix];
    glVertex3f(x,y,1);
  }
  glEnd();

  drawVectors();
}

void
QtOutputView::resizeGL(int w,int h){
  glViewport(0,0,w,h);
  glMatrixMode(GL_PROJECTION);
  glLoadIdentity();
  gluOrtho2D(0,lX,0,factor);
  glMatrixMode(GL_MODELVIEW);
  glLoadIdentity();
}

void
QtOutputView::drawTriangle(size_t ix1,size_t iz1,size_t ix2,size_t iz2,size_t ix3,size_t iz3){
  double factor_inv=1;//1/kernel->solution.factor;
  double h=height();
  double w=width();
  double x1=ix1*dX;
  double x2=ix2*dX;
  double x3=ix3*dX;
  double y1=kernel->Z(ix1,iz1)*factor_inv;;
  double y2=kernel->Z(ix2,iz2)*factor_inv;;
  double y3=kernel->Z(ix3,iz3)*factor_inv;;
  setColor(ix1,iz1);
  glVertex3f(x1,y1,0);
  setColor(ix2,iz2);
  glVertex3f(x2,y2,0);
  setColor(ix3,iz3);
  glVertex3f(x3,y3,0);
}

void
QtOutputView::draw(size_t _step){
  step=_step;
  update();
}

void
QtOutputView::mousePressEvent(QMouseEvent* event){
  double x=double(event->x())/width()*lX;
  double y=1-(double(event->y())/height());
  displayInfos(x,y*factor);
}

void
QtOutputView::setColor(size_t ix,size_t iz){
  double p=getP(ix,iz);
  double s=Physics::s(p);

  double r=(1-s);
  double g=0.7*(1-s);
  double b=0.8*s+0.4*(1-s);
  glColor3f(r,g,b);
}

void
QtOutputView::drawVector(size_t ix,size_t iz,double u,double v){
  double x1=ix*dX;
  double y1=kernel->Z(ix,iz);
  double x2=x1+u;
  double y2=y1+v;
  double x3=x2+(-0.866*u+0.5*v)*scale_arrow_head*lX;
  double y3=y2+(-0.5*u-0.866*v)*scale_arrow_head*factor;
  double x4=x2+(-0.866*u-0.5*v)*scale_arrow_head*lX;
  double y4=y2+(0.5*u-0.866*v)*scale_arrow_head*factor;

  glColor3f(1,1,0);
  glBegin(GL_LINE_STRIP);
  glVertex3f(x1,y1,0.8);
  glVertex3f(x2,y2,0.8);
  glEnd();
  glBegin(GL_LINE_STRIP);
  glVertex3f(x3,y3,0.8);
  glVertex3f(x2,y2,0.8);
  glVertex3f(x4,y4,0.8);
  glEnd();

}

void
QtOutputView::drawVectors(){
  for(size_t ix=1;ix<nX-1;ix+=arrow_step_x){
    //Case iz=0
    double p1=getP(ix,0);
    double p2=getP(ix,1);
    double u=-Physics::k0*Physics::kr(getPl(ix)+Physics::rho*Physics::g*(getl(ix)-kernel->Z(ix,0)))*(getHydr(ix+1)-getHydr(ix-1))/(2*dX);
    double v=-Physics::k0*Physics::kr(p1)*((p2-p1)/(Physics::rho*Physics::g*kernel->geometry.dZ[ix])+1);
    drawVector(ix,0,scale_arrow*u,scale_arrow*v);
    for(size_t iz=arrow_step_z;iz<kernel->nZ(ix)-1;iz+=arrow_step_z){
      double p0=getP(ix,iz-1);
      double p1=getP(ix,iz);
      double p2=getP(ix,iz+1);
      double u=-Physics::k0*Physics::kr(getPl(ix)+Physics::rho*Physics::g*(getl(ix)-kernel->Z(ix,iz)))*(getHydr(ix+1)-getHydr(ix-1))/(2*dX);
      double v=-Physics::k0*Physics::kr(p1)*((p2-p0)/(2*Physics::rho*Physics::g*kernel->geometry.dZ[ix])+1);
      drawVector(ix,iz,scale_arrow*u,scale_arrow*v);
    }
  }
}

void
QtOutputView::displayInfos(double x,double z){
  cout<<"---------------------------------"<<endl;
  cout<<"x = "<<x<<endl;
  cout<<"z = "<<z<<endl;
  size_t ix=floor(x/dX+0.5);
  if(ix==nX) --ix;
  cout<<"ix = "<<ix<<endl;

  double hbot=kernel->geometry.hbot[ix];
  cout<<"hbot = "<<hbot<<endl;
  double hsoil=kernel->geometry.hsoil[ix];
  cout<<"hsoil = "<<hsoil<<endl;
  cout<<"l = "<<getl(ix)<<endl;
    if(z>hbot and z<hsoil){
    double dZ=kernel->geometry.dZ[ix];
    cout<<"dZ = "<<dZ<<endl;
    size_t nz=kernel->geometry.nZ[ix];
    cout<<"nZ = "<<nz<<endl;
    size_t iz=floor((z-hbot)/dZ+0.5);
    if(iz==nz) --iz;
    cout<<"iz = "<<iz<<endl;
    cout<<"P = "<<getP(ix,iz)<<endl;
    }
}