source: proiecte/pmake3d/make3d_original/Make3dSingleImageStanford_version0.1/LearningCode/Rendering/Oldversion/vrml_test_faceset_goodSkyBoundary.m @ 37

% *  This code was used in the following articles:
% *  [1] Learning 3-D Scene Structure from a Single Still Image, 
% *      Ashutosh Saxena, Min Sun, Andrew Y. Ng, 
% *      In ICCV workshop on 3D Representation for Recognition (3dRR-07), 2007.
% *      (best paper)
% *  [2] 3-D Reconstruction from Sparse Views using Monocular Vision, 
% *      Ashutosh Saxena, Min Sun, Andrew Y. Ng, 
% *      In ICCV workshop on Virtual Representations and Modeling 
% *      of Large-scale environments (VRML), 2007. 
% *  [3] 3-D Depth Reconstruction from a Single Still Image, 
% *      Ashutosh Saxena, Sung H. Chung, Andrew Y. Ng. 
% *      International Journal of Computer Vision (IJCV), Aug 2007. 
% *  [6] Learning Depth from Single Monocular Images, 
% *      Ashutosh Saxena, Sung H. Chung, Andrew Y. Ng. 
% *      In Neural Information Processing Systems (NIPS) 18, 2005.
% *
% *  These articles are available at:
% *  http://make3d.stanford.edu/publications
% * 
% *  We request that you cite the papers [1], [3] and [6] in any of
% *  your reports that uses this code. 
% *  Further, if you use the code in image3dstiching/ (multiple image version),
% *  then please cite [2].
% *  
% *  If you use the code in third_party/, then PLEASE CITE and follow the
% *  LICENSE OF THE CORRESPONDING THIRD PARTY CODE.
% *
% *  Finally, this code is for non-commercial use only.  For further 
% *  information and to obtain a copy of the license, see 
% *
% *  http://make3d.stanford.edu/publications/code
% *
% *  Also, the software distributed under the License is distributed on an 
% * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either 
% *  express or implied.   See the License for the specific language governing 
% *  permissions and limitations under the License.
% *
% */
function [vrml_filename] = vrml_test_faceset_goodSkyBoundary(ScratchDataFolder,...
          filename, Position3D, depth, ray, DepthDirectory,...
          HBrokeBook, VBrokeBook, Grid, maskSky, maskG,...
          Np, Sep, a, b, Ox,...
          Oy)
%function [] = vrml_test_faceset_triangle(filename,PlaneParameterTure,LowResImgIndexSuperpixelSep,DepthDirectory,a,b,Ox,Oy)
% this function play with the VRML
% using only FaceSet

%%% Param descriptions from Rajiv
% filename - the name of the image file
% Position3D - the matrix of 3d coordinates for all the point (55x305)
% depth - the matrix of depth values for each point (55x305)
% ray - the ray vector for each point in the image (55x305) 
% DepthDirectory - the name of the directory used to store final
%   vrml files in a subfolder vrml
% HBrokeBook - an array that contains the information if the
%   triangle made during writing the vrml file should be reversed 
%   in order to align to a straight line 
% VBrokeBook - same as HBrokeBook but in the vertical direction
% Grid - boolean variable which decides if there shud be a grid drawn in the vrml
% maskSky - the skymask cell array, we need this not to draw the sky from the image 
% maskG - ground mask
% Np - parameter which is used when deciding if Np neighbor pixels
%   of a pixel are sky, then the pixel being considered should also
%   be declared to be sky, this is because the sky mask is not accurate 
% Sep - parameter not used any more, shud be removed
% a, -->
% b, -->
% Ox,-->
% Oy,--> camera parameters used to generate the Ray, not used anymore as we directly pass the ray, should be removed
%%%

displayFlag = true;

global GeneralDataFolder  LocalFolder ClusterExecutionDirectory...
    ImgFolder VertYNuPatch VertYNuDepth HoriXNuPatch HoriXNuDepth a_default b_default Ox_default Oy_default...
    Horizon_default batchSize NuRow_default SegVertYSize SegHoriXSize WeiBatchSize;

if nargin < 13
    a = 0.70783777 %0.129; % horizontal physical size of image plane normalized to focal length (in meter)
    b = 0.946584169%0.085; % vertical physical size of image plane normalized to focal length (in meter)
    Ox = -0.010727086; % camera origin offset from the image center in horizontal direction
    Oy = -0.0111130176; % camera origin offset from the image center in vertical direction
elseif nargin < 14
    b = a;
    Ox = -0.010727086; % camera origin offset from the image center in horizontal direction
    Oy = -0.0111130176; % camera origin offset from the image center in vertical direction
elseif nargin < 15
    Ox = -0.010727086; % camera origin offset from the image center in horizontal direction
    Oy = -0.0111130176; % camera origin offset from the image center in vertical direction
elseif nargin < 16
    Oy = Ox;
end

if displayFlag
disp('In VRML generation Code');
end

% define global variable
global GeneralDataFolder  LocalFolder;

%maskSky = imerode(maskSky, strel('disk', 3) );
%maskG = imerode(maskSky, strel('disk', 3) );

%calculating the mean of the sky color
% imageActual = imread([GeneralDataFolder '/' ImgFolder '/' filename '.jpg']);
% meanSkyColor = permute( sum( sum( repmat(maskSky, [1 1 3]) .* ...
%                               double( imresize(imageActual, size(maskSky) ) ), 1), 2), [3 1 2]) ;
% meanGroundColor = permute( sum( sum( repmat(maskG, [1 1 3]) .* ...
%                               double( imresize(imageActual, size(maskG) ) ), 1), 2), [3 1 2]) ;
% meanSkyColor = meanSkyColor / sum(sum( maskSky ) );
% if any(isnan(meanSkyColor))
%    if displayFlag
%       disp('meanSkyColor NaN');
%    end
%    meanSkyColor = [0.31 0.54 0.76];
% end
% meanGroundColor = meanGroundColor / sum(sum( maskG ));
% if any(isnan(meanGroundColor))
%    if displayFlag
%       disp('meanGroundColor NaN');
%    end
%    meanGroundColor = [0 0 0];
% end

vrml_filename = [DepthDirectory '.wrl'];
%LowResImgIndexSuperpixelSepTemp = LowResImgIndexSuperpixelSep;
%[VertYSize HoriXSize] = size(LowResImgIndexSuperpixelSepTemp);
[dum VertYSize HoriXSize] = size(Position3D);
%nu_patch = VertYSize* HoriXSize;

Position3DCoord = reshape(Position3D,3,VertYSize*HoriXSize);    
%Position3DCoord(3,:) = -Position3DCoord(3,:); % important to make z direction negative

table = reshape(0:(VertYSize*HoriXSize-1),VertYSize,[]);
Position3DCoordOrdeerIndex = zeros(1,92568); %RAJIV
GridPosition3DCoordOrdeerIndex = zeros(1,92576);

% initialize the HBrokeBook and VBrokeBook
if isempty(HBrokeBook) | isempty(VBrokeBook)
   disp('empty HBrokeBook VBrokeBook in vrml_goodSkyBoundary')
   HBrokeBook = zeros(VertYSize,HoriXSize-1);
   VBrokeBook = zeros(VertYSize-1,HoriXSize);
end
HBrokeBook = HBrokeBook | maskSky(:,1:(end-1)) | maskSky(:,2:(end));
VBrokeBook = VBrokeBook | maskSky(1:(end-1),:) | maskSky(2:(end),:);

if displayFlag,
disp('Going into FOR LOOP, which is bad in Matlab. This code should not take longer than 3 seconds');
end

for i = 1:VertYSize-1
    for j = 1:HoriXSize-1
        % check which diagnoal to render
%        if isempty(HBrokeBook) | isempty(VBrokeBook)
%           Index = [ table(i,j) table(i+1,j) table(i+1,j+1) table(i,j) table(i+1,j+1) table(i,j+1)];
%           GridIndex = [table(i,j) table(i+1,j) table(i+1,j+1) table(i,j) table(i,j) table(i+1,j+1) table(i,j+1) table(i,j)];
%        else 
           target = [HBrokeBook(i,j) VBrokeBook(i,j+1) HBrokeBook(i+1,j) VBrokeBook(i,j)];
           if all(target == [0 0 1 1]);
              Index = [ table(i,j) table(i+1,j+1) table(i,j+1) ];%table(i,j) table(i+1,j) table(i+1,j+1)];
              GridIndex = [table(i,j) table(i+1,j+1) table(i,j+1) table(i,j)];
           elseif all(target == [1 0 0 1]);
              Index = [ table(i+1,j) table(i+1,j+1) table(i,j+1) ];%table(i,j) table(i+1,j) table(i,j+1)];
              GridIndex = [table(i+1,j) table(i+1,j+1) table(i,j+1) table(i+1,j)];
           elseif all(target == [1 1 0 0]);
              Index = [ table(i,j) table(i+1,j) table(i+1,j+1) ];%table(i,j) table(i+1,j+1) table(i,j+1)];
              GridIndex = [ table(i,j) table(i+1,j) table(i+1,j+1) table(i,j)];
           elseif all(target == [0 1 1 0]);
              Index = [ table(i,j) table(i+1,j) table(i,j+1) ];%table(i+1,j) table(i+1,j+1) table(i,j+1)];
              GridIndex = [ table(i,j) table(i+1,j) table(i,j+1) table(i,j)];
           elseif sum(target) <3 && ~all(target == [0 1 0 1])
              Index = [ table(i,j) table(i+1,j) table(i+1,j+1) table(i,j) table(i+1,j+1) table(i,j+1)];
              GridIndex = [table(i,j) table(i+1,j) table(i+1,j+1) table(i,j) table(i,j) table(i+1,j+1) table(i,j+1) table(i,j)];
           else
%              disp('sky');
               Index = [];
               GridIndex = [];
           end
%        end
           Position3DCoordOrdeerIndex = [Position3DCoordOrdeerIndex Index];
           GridPosition3DCoordOrdeerIndex = [Position3DCoordOrdeerIndex GridIndex];

    end
end

% calculate PositionTexCoord
%PositionTex = permute(ray(:,:,1:2)./repmat(cat(3,a,b),[VertYSize HoriXSize 1])+0.5,[3 1 2]);
temp = ray(:,:,1:2)./repmat(ray(:,:,3),[1 1 2]);
PositionTex = permute(temp./repmat(cat(3,a,b),[VertYSize HoriXSize 1])+repmat(cat(3,Ox,Oy),[VertYSize HoriXSize 1]),[3 1 2]);
PositionTexCoord = PositionTex;   

% find the Sky Wall point
Xmin=min(Position3DCoord(1,:))-Sep;
Xmax=max(Position3DCoord(1,:))+Sep;
Ymin=min(Position3DCoord(2,:))-Sep;
Ymax=max(Position3DCoord(2,:))+Sep;
Zmin=min(Position3DCoord(3,:))-Sep;
Zmax=max(Position3DCoord(3,:))+Sep;
%% the eight corners of the cube would then be
cc=[];
cc = [cc [Xmin Ymin Zmax]']; %% 1
cc = [cc [Xmax Ymin Zmax]']; %% 2
cc = [cc [Xmax Ymin Zmin]']; %% 3
cc = [cc [Xmin Ymin Zmin]']; %% 4

cc = [cc [Xmin Ymax Zmax]']; %% 5
cc = [cc [Xmax Ymax Zmax]']; %% 6
cc = [cc [Xmax Ymax Zmin]']; %% 7
cc = [cc [Xmin Ymax Zmin]']; %% 8
% wallIndices = [front_wall right_wall back_wall left_wall];
wallIndices = [2 6 7 3 3 7 8 4 1 4 8 5 5 8 7 6 1 2 3 4 1 5 6 2]-1;

% inital header
disp('writing vrml..');
disp([ScratchDataFolder '/' vrml_filename])
fp = fopen([ScratchDataFolder '/' vrml_filename],'w+');

fprintf(fp, '#VRML V2.0 utf8\n');

% add navigate_info
fprintf(fp, 'NavigationInfo {\n');
fprintf(fp, '  headlight TRUE\n');
fprintf(fp, '  type ["FLY", "ANY"]}\n\n');

% add viewpoint
fprintf(fp, 'Viewpoint {\n');
fprintf(fp, '    position        0 0.0 0.0\n');
fprintf(fp, '    orientation     0 0 0 0\n');
fprintf(fp, '    fieldOfView     0.7\n');
fprintf(fp, '    description "Original"}\n');

%============== add background color======
fprintf(fp, 'DEF Back1 Background {\n');
fprintf(fp, 'groundColor [.3 .29 .27]\n');
% fprintf(fp, 'groundColor [%f %f %f]\n', meanGroundColor/255);
fprintf(fp, 'skyColor [0.31 0.54 0.76]}\n');
% fprintf(fp, 'skyColor [%f %f %f]}\n', meanSkyColor/255);
%=========================================

% add Shape for texture faceset
fprintf(fp, 'Shape{\n');
fprintf(fp, '  appearance Appearance {\n');
fprintf(fp, ['   texture ImageTexture { url "./' filename '.jpg' '" }\n']);
fprintf(fp, '  }\n');
fprintf(fp, '  geometry IndexedFaceSet {\n');
fprintf(fp, '    coord Coordinate {\n');

% insert coordinate in 3d
% =======================
fprintf(fp, '      point [ \n');
fprintf(fp, '        %.2f %.2f %.2f,\n',Position3DCoord);
fprintf(fp, '      ]\n');
fprintf(fp, '    }\n');

% insert coordinate index in 3d
fprintf(fp, '    coordIndex [\n');
fprintf(fp, '              %g %g %g -1,\n',Position3DCoordOrdeerIndex);
fprintf(fp, '    ]\n');

% insert texture coordinate
fprintf(fp, '    texCoord TextureCoordinate {\n');
fprintf(fp, '      point [\n');
fprintf(fp, '              %.4g %.4g,\n',PositionTexCoord);
fprintf(fp, '        ]\n');
fprintf(fp, '    }\n');
fprintf(fp, '    texCoordIndex [\n');
fprintf(fp, '              %g %g %g -1,\n',Position3DCoordOrdeerIndex);
fprintf(fp, '    ]\n');
fprintf(fp, '  }\n');
fprintf(fp, '}\n');

if Grid == 1
% ========================================
fprintf(fp, 'Shape{\n');
fprintf(fp, '  appearance Appearance { material Material {emissiveColor 1 0 0  }}\n');
fprintf(fp, '    geometry IndexedLineSet {\n');
fprintf(fp, '    coord Coordinate {\n');
fprintf(fp, '      point [ \n');
fprintf(fp, '        %.2f %.2f %.2f,\n',Position3DCoord);
fprintf(fp, '      ]\n');
fprintf(fp, '    }\n');
fprintf(fp, '    coordIndex [\n');
fprintf(fp, '              %g %g %g %g -1,\n',GridPosition3DCoordOrdeerIndex);
fprintf(fp, '    ]\n');
%fprintf(fp, '   color Color { color [1 0 0, 0 1 0, 0 0 1]}\n');
%fprintf(fp, '   colorIndex [0 0 0 0 0 ]\n');
fprintf(fp, '    colorPerVertex FALSE\n');
fprintf(fp, '    }\n');
%==================================
fprintf(fp, '  }\n');
fprintf(fp, '}\n');
end
% close the file
fclose(fp);

disp('Finished writing vrml');