ENH: Run time post-processing updates and bug fixing. Fixes #128 #121 #99

This commit is contained in:
Andrew Heather
2016-06-10 15:23:56 +01:00
parent 430b6208e6
commit 151bcc8018
3 changed files with 68 additions and 78 deletions

View File

@ -3,7 +3,7 @@
\\ / F ield | OpenFOAM: The Open Source CFD Toolbox \\ / F ield | OpenFOAM: The Open Source CFD Toolbox
\\ / O peration | \\ / O peration |
\\ / A nd | Copyright (C) 2015 OpenFOAM Foundation \\ / A nd | Copyright (C) 2015 OpenFOAM Foundation
\\/ M anipulation | Copyright (C) 2015 OpenCFD Ltd. \\/ M anipulation | Copyright (C) 2015-2016 OpenCFD Ltd.
------------------------------------------------------------------------------- -------------------------------------------------------------------------------
License License
This file is part of OpenFOAM. This file is part of OpenFOAM.
@ -453,31 +453,32 @@ Foam::fieldVisualisationBase::fieldVisualisationBase
{ {
case cbColour: case cbColour:
{ {
scalarBar_.visible_ = false;
break; break;
} }
case cbField: case cbField:
{ {
dict.lookup("range") >> range_; dict.lookup("range") >> range_;
if (dict.found("colourMap"))
{
colourMap_ = colourMapTypeNames.read(dict.lookup("colourMap"));
}
const dictionary& sbarDict = dict.subDict("scalarBar");
sbarDict.lookup("visible") >> scalarBar_.visible_;
if (scalarBar_.visible_)
{
sbarDict.lookup("vertical") >> scalarBar_.vertical_;
sbarDict.lookup("position") >> scalarBar_.position_;
sbarDict.lookup("title") >> scalarBar_.title_;
sbarDict.lookup("fontSize") >> scalarBar_.fontSize_;
sbarDict.lookup("labelFormat") >> scalarBar_.labelFormat_;
sbarDict.lookup("numberOfLabels") >> scalarBar_.numberOfLabels_;
}
break; break;
} }
} }
if (dict.found("colourMap"))
{
colourMap_ = colourMapTypeNames.read(dict.lookup("colourMap"));
}
const dictionary& sbarDict = dict.subDict("scalarBar");
sbarDict.lookup("visible") >> scalarBar_.visible_;
if (scalarBar_.visible_)
{
sbarDict.lookup("vertical") >> scalarBar_.vertical_;
sbarDict.lookup("position") >> scalarBar_.position_;
sbarDict.lookup("title") >> scalarBar_.title_;
sbarDict.lookup("fontSize") >> scalarBar_.fontSize_;
sbarDict.lookup("labelFormat") >> scalarBar_.labelFormat_;
sbarDict.lookup("numberOfLabels") >> scalarBar_.numberOfLabels_;
}
} }

View File

@ -138,15 +138,6 @@ void Foam::scene::readCamera(const dictionary& dict)
} }
} }
if (dict.found("zoom"))
{
cameraZoom_.reset(DataEntry<scalar>::New("zoom", dict).ptr());
}
else
{
cameraZoom_.reset(new Constant<scalar>("zoom", 1.0));
}
if (dict.found("viewAngle")) if (dict.found("viewAngle"))
{ {
cameraViewAngle_.reset(DataEntry<scalar>::New("viewAngle", dict).ptr()); cameraViewAngle_.reset(DataEntry<scalar>::New("viewAngle", dict).ptr());
@ -172,11 +163,12 @@ void Foam::scene::readColours(const dictionary& dict)
void Foam::scene::initialise(vtkRenderer* renderer, const word& outputName) void Foam::scene::initialise(vtkRenderer* renderer, const word& outputName)
{ {
currentFrameI_ = 0; currentFrameI_ = 0;
position_ = startPosition_;
outputName_ = outputName; outputName_ = outputName;
// Set the background // Set the background
const vector backgroundColour = colours_["background"]->value(position()); const vector backgroundColour = colours_["background"]->value(position_);
renderer->SetBackground renderer->SetBackground
( (
backgroundColour.x(), backgroundColour.x(),
@ -188,7 +180,7 @@ void Foam::scene::initialise(vtkRenderer* renderer, const word& outputName)
if (colours_.found("background2")) if (colours_.found("background2"))
{ {
renderer->GradientBackgroundOn(); renderer->GradientBackgroundOn();
vector backgroundColour2 = colours_["background2"]->value(position()); vector backgroundColour2 = colours_["background2"]->value(position_);
renderer->SetBackground2 renderer->SetBackground2
( (
@ -208,9 +200,24 @@ void Foam::scene::initialise(vtkRenderer* renderer, const word& outputName)
camera->SetParallelProjection(parallelProjection_); camera->SetParallelProjection(parallelProjection_);
renderer->SetActiveCamera(camera); renderer->SetActiveCamera(camera);
setCamera(renderer, true);
// Initialise the extents // Initialise the camera
const vector up = cameraUp_->value(position_);
const vector pos = cameraPosition_->value(position_);
const point focalPoint = cameraFocalPoint_->value(position_);
camera->SetViewUp(up.x(), up.y(), up.z());
camera->SetPosition(pos.x(), pos.y(), pos.z());
camera->SetFocalPoint(focalPoint.x(), focalPoint.y(), focalPoint.z());
camera->Modified();
// Add the lights
vtkSmartPointer<vtkLightKit> lightKit = vtkSmartPointer<vtkLightKit>::New();
lightKit->AddLightsToRenderer(renderer);
// For static mode initialise the clip box
if (mode_ == mtStatic) if (mode_ == mtStatic)
{ {
const point& min = clipBox_.min(); const point& min = clipBox_.min();
@ -232,48 +239,38 @@ void Foam::scene::initialise(vtkRenderer* renderer, const word& outputName)
vtkSmartPointer<vtkActor> clipActor = vtkSmartPointer<vtkActor>::New(); vtkSmartPointer<vtkActor> clipActor = vtkSmartPointer<vtkActor>::New();
clipActor->SetMapper(clipMapper); clipActor->SetMapper(clipMapper);
clipActor->VisibilityOn(); clipActor->VisibilityOff();
renderer->AddActor(clipActor); renderer->AddActor(clipActor);
// Call resetCamera to fit clip box in view
clipActor->VisibilityOn();
renderer->ResetCamera(); renderer->ResetCamera();
// Restore viewAngle (it might be reset by clipping)
vtkCamera* camera = renderer->GetActiveCamera();
if (!parallelProjection_)
{
camera->SetViewAngle(cameraViewAngle_->value(position()));
}
camera->Modified();
clipActor->VisibilityOff(); clipActor->VisibilityOff();
} }
} }
void Foam::scene::setCamera(vtkRenderer* renderer, const bool override) const void Foam::scene::setCamera(vtkRenderer* renderer) const
{ {
if (mode_ == mtFlightPath || override) if (mode_ == mtFlightPath)
{ {
const vector up = cameraUp_->value(position_);
const vector pos = cameraPosition_->value(position_);
const point focalPoint = cameraFocalPoint_->value(position_);
vtkCamera* camera = renderer->GetActiveCamera(); vtkCamera* camera = renderer->GetActiveCamera();
if (!parallelProjection_)
{
camera->SetViewAngle(cameraViewAngle_->value(position()));
}
const vector up = cameraUp_->value(position());
const vector pos = cameraPosition_->value(position());
const point focalPoint = cameraFocalPoint_->value(position());
camera->SetViewUp(up.x(), up.y(), up.z()); camera->SetViewUp(up.x(), up.y(), up.z());
camera->SetPosition(pos.x(), pos.y(), pos.z()); camera->SetPosition(pos.x(), pos.y(), pos.z());
camera->SetFocalPoint(focalPoint.x(), focalPoint.y(), focalPoint.z()); camera->SetFocalPoint(focalPoint.x(), focalPoint.y(), focalPoint.z());
camera->Modified(); camera->Modified();
}
vtkSmartPointer<vtkLightKit> lightKit = if (!parallelProjection_)
vtkSmartPointer<vtkLightKit>::New(); {
lightKit->AddLightsToRenderer(renderer); // Restore viewAngle (it might be reset by clipping)
vtkCamera* camera = renderer->GetActiveCamera();
camera->SetViewAngle(cameraViewAngle_->value(position_));
camera->Modified();
} }
} }
@ -298,7 +295,6 @@ Foam::scene::scene(const objectRegistry& obr, const word& name)
cameraPosition_(NULL), cameraPosition_(NULL),
cameraFocalPoint_(NULL), cameraFocalPoint_(NULL),
cameraUp_(NULL), cameraUp_(NULL),
cameraZoom_(NULL),
cameraViewAngle_(NULL), cameraViewAngle_(NULL),
clipBox_(), clipBox_(),
parallelProjection_(true), parallelProjection_(true),
@ -348,8 +344,7 @@ void Foam::scene::read(const dictionary& dict)
bool Foam::scene::loop(vtkRenderer* renderer) bool Foam::scene::loop(vtkRenderer* renderer)
{ {
static bool initialised = false; static bool initialised = false;
setCamera(renderer);
setCamera(renderer, false);
if (!initialised) if (!initialised)
{ {
@ -357,19 +352,15 @@ bool Foam::scene::loop(vtkRenderer* renderer)
return true; return true;
} }
// Ensure that all objects can be seen without clipping
// Note: can only be done after all objects have been added!
renderer->ResetCameraClippingRange();
// Save image from last iteration // Save image from last iteration
saveImage(renderer->GetRenderWindow()); saveImage(renderer->GetRenderWindow());
currentFrameI_++; currentFrameI_++;
// Warning only if camera is in flight mode
if ((mode_ == mtFlightPath) && (position_ > (1 + 0.5*dPosition_)))
{
WarningInFunction
<< "Current position "<< position_ <<" exceeded 1 - please check your setup"
<< endl;
}
position_ = startPosition_ + currentFrameI_*dPosition_; position_ = startPosition_ + currentFrameI_*dPosition_;
if (currentFrameI_ < nFrameTotal_) if (currentFrameI_ < nFrameTotal_)
@ -378,6 +369,7 @@ bool Foam::scene::loop(vtkRenderer* renderer)
} }
else else
{ {
initialised = false;
return false; return false;
} }
} }
@ -392,9 +384,9 @@ void Foam::scene::saveImage(vtkRenderWindow* renderWindow) const
const Time& runTime = obr_.time(); const Time& runTime = obr_.time();
fileName prefix(Pstream::parRun() fileName prefix(Pstream::parRun() ?
? runTime.path()/".."/"postProcessing"/name_/obr_.time().timeName() runTime.path()/".."/"postProcessing"/name_/obr_.time().timeName() :
: runTime.path()/"postProcessing"/name_/obr_.time().timeName()); runTime.path()/"postProcessing"/name_/obr_.time().timeName());
mkDir(prefix); mkDir(prefix);

View File

@ -3,7 +3,7 @@
\\ / F ield | OpenFOAM: The Open Source CFD Toolbox \\ / F ield | OpenFOAM: The Open Source CFD Toolbox
\\ / O peration | \\ / O peration |
\\ / A nd | Copyright (C) 2015 OpenFOAM Foundation \\ / A nd | Copyright (C) 2015 OpenFOAM Foundation
\\/ M anipulation | \\/ M anipulation | Copyright (C) 2016 OpenCFD Ltd.
------------------------------------------------------------------------------- -------------------------------------------------------------------------------
License License
This file is part of OpenFOAM. This file is part of OpenFOAM.
@ -22,7 +22,7 @@ License
along with OpenFOAM. If not, see <http://www.gnu.org/licenses/>. along with OpenFOAM. If not, see <http://www.gnu.org/licenses/>.
Class Class
Foam::camera Foam::scene
Description Description
@ -116,9 +116,6 @@ protected:
//- Up direction //- Up direction
autoPtr<DataEntry<vector> > cameraUp_; autoPtr<DataEntry<vector> > cameraUp_;
//- Zoom level
autoPtr<DataEntry<scalar> > cameraZoom_;
//- View angle //- View angle
autoPtr<DataEntry<scalar> > cameraViewAngle_; autoPtr<DataEntry<scalar> > cameraViewAngle_;
@ -134,7 +131,7 @@ protected:
//- Number of frames //- Number of frames
label nFrameTotal_; label nFrameTotal_;
//- startPosition [0-1] //- Start position [0-1]
scalar startPosition_; scalar startPosition_;
//- Position [0-1] //- Position [0-1]
@ -152,7 +149,7 @@ protected:
// Protected Member Functions // Protected Member Functions
void setCamera(vtkRenderer* renderer, const bool override) const; void setCamera(vtkRenderer* renderer) const;
string frameIndexStr() const; string frameIndexStr() const;