https://github.com/lingqi/WaveOpticsBrdf
Revision 73853c6dfa6152945bfc02eb94349ca510ca7d72 authored by Milos Hasan on 08 January 2019, 00:26:32 UTC, committed by Milos Hasan on 08 January 2019, 00:26:32 UTC
1 parent a92800f
Tip revision: 73853c6dfa6152945bfc02eb94349ca510ca7d72 authored by Milos Hasan on 08 January 2019, 00:26:32 UTC
print ranges
print ranges
Tip revision: 73853c6
genBrdf.cpp
/*
Copyright 2018 Lingqi Yan
This file is part of WaveOpticsBrdf.
WaveOpticsBrdf is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
WaveOpticsBrdf is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with WaveOpticsBrdf. If not, see <https://www.gnu.org/licenses/>.
*/
#include <iostream>
#include <unistd.h>
#include <chrono>
#include <string>
#include <sstream>
#include "waveBrdf.h"
#include "waveNdf.h"
#include "spectrum.h"
using namespace std;
using namespace Eigen;
Float* mkCrop(Float* img, int n, int c) {
Float* result = new Float[c * c * 3 * sizeof(Float)];
Map<ColorImage> src((Color*) img, n, n);
Map<ColorImage> dst((Color*) result, c, c);
int i = (n - c) / 2;
dst = src.block(i, i, c, c);
delete[] img;
return result;
}
void printRange(float radius) {
cout << "NDF image range: [-" << radius << ", " << radius << "]^2\n";
}
int main(int argc, char **argv) {
srand(time(NULL));
char *heightfieldFilename = NULL;
double texelWidth = 1.0;
double vertScale = 1.0;
double mu_x = 0.0;
double mu_y = 0.0;
double sigma_p = 10.0;
string method = "Wave";
int sampleNum = 500000;
string diffModel = "OHS";
double lambda = 0.5;
double omega_i_x = 0.0;
double omega_i_y = 0.0;
char *outputFilename = NULL;
int resolution = 256;
int crop = 0;
int footprint_k = 3;
int opt;
while ((opt = getopt(argc, argv, "i:o:l:r:e:w:v:x:y:p:m:g:d:s:t:n:c:k:")) != -1) {
switch (opt) {
case 'i': heightfieldFilename = optarg; break; // heightfield filename.
case 'w': texelWidth = atof(optarg); break; // The width of a texel in microns on the heightfield.
case 'v': vertScale = atof(optarg); break; // The vertical scaling factor of the heightfield.
case 'x': mu_x = atof(optarg); break; // Center x of the Gaussian footprint.
case 'y': mu_y = atof(optarg); break; // Center y of the Gaussian footprint.
case 'p': sigma_p = atof(optarg); break; // Size (1 sigma) of the Gaussian footprint.
case 'm': method = optarg; break; // Method. Choose between "Geom" and "Wave".
case 'n': sampleNum = atoi(optarg); break; // Number of binning samples. Only valid for geometric optics.
case 'd': diffModel = optarg; break; // Diffraction model. Choose between "OHS", "GHS", "ROHS", "RGHS" and "Kirchhoff". And expect only subtle difference.
case 'l': lambda = atof(optarg); break; // Wavelength in microns. Once set, single wavelength mode is ON.
case 's': omega_i_x = atof(optarg); break; // Incoming light's x coordinate (assuming z = 1).
case 't': omega_i_y = atof(optarg); break; // Incoming light's y coordinate (assuming z = 1).
case 'o': outputFilename = optarg; break; // output filename.
case 'r': resolution = atoi(optarg); break; // output resolution.
case 'c': crop = atoi(optarg); break; // crop resolution (no crop if value <= 0)
case 'k': footprint_k = atoi(optarg); break; // Gaussian footprint cutoff at k sigma
}
}
if (heightfieldFilename == NULL) {
cout << "A heightfield file must be specified." << endl;
return -1;
}
if (outputFilename == NULL) {
cout << "An output file must be specified." << endl;
return -1;
}
SpectrumInit();
EXRImage heightfieldImage(heightfieldFilename);
Heightfield heightfield(&heightfieldImage, texelWidth, vertScale);
Query query;
query.mu_p = Vector2(mu_x, mu_y);
query.sigma_p = sigma_p;
query.omega_i = Vector3(omega_i_x, omega_i_y, 1.0).normalized().head(2);
query.lambda = lambda;
int n = resolution;
if (method == "Geom") {
GeometricBrdf geometricBrdf(&heightfield, sampleNum);
Float *brdfImage = geometricBrdf.genBrdfImage(query, n);
EXRImage::writeImage(brdfImage, outputFilename, n, n);
delete[] brdfImage;
} else if (method == "Wave") {
WaveBrdfAccel waveBrdfAccel(&heightfield, diffModel);
Float *brdfImage = waveBrdfAccel.genBrdfImage(query, n);
EXRImage::writeImage(brdfImage, outputFilename, n, n);
delete[] brdfImage;
} else if (method == "GeomNdf") {
GeometricBrdf geometricBrdf(&heightfield, sampleNum);
Float *ndfImage = geometricBrdf.genNdfImage(query, n);
float radius = 1;
if (crop > 0) { ndfImage = mkCrop(ndfImage, n, crop); radius *= float(crop) / n; n = crop; }
EXRImage::writeImage(ndfImage, outputFilename, n, n);
delete[] ndfImage;
printRange(radius);
}
else if (method == "GeomNdfMany") {
GeometricBrdf geometricBrdf(&heightfield, sampleNum);
int N = 256;
float delta = 16;
for (int i = 0; i < N; i++) {
for (int j = 0; j < N; j++) {
query.mu_p = delta * Vector2(i, j);
Float *ndfImage = geometricBrdf.genNdfImage(query, n);
int s = n;
if (crop > 0) { ndfImage = mkCrop(ndfImage, n, crop); s = crop; }
stringstream ss;
ss << outputFilename << i << '_' << j << ".exr";
EXRImage::writeImage(ndfImage, ss.str().c_str(), s, s);
delete[] ndfImage;
}
}
}
else if (method == "WaveNdf") {
WaveNDF waveNdf(heightfield, n, crop, footprint_k, lambda > 0);
waveNdf.generate(query, outputFilename);
if (crop > 0) n = crop;
float lambda = query.lambda;
if (lambda <= 0) lambda = 0.5f;
float radius = lambda * n / (4 * footprint_k * query.sigma_p);
printRange(radius);
}
else if (method == "WaveNdfMany") {
int N = 256;
float delta = 16;
#pragma omp parallel for schedule(dynamic)
for (int i = 0; i < N; i++) {
WaveNDF waveNdf(heightfield, n, crop, footprint_k);
Query q = query;
for (int j = 0; j < N; j++) {
q.mu_p = delta * Vector2(i, j);
stringstream ss;
ss << outputFilename << i << '_' << j << ".exr";
waveNdf.generate(q, ss.str().c_str());
}
}
}
return 0;
}
Computing file changes ...