#import "MyDocument.h"

#import "BasicVTKView.h"

#import "vtkInteractorStyleSwitch.h"
#import "vtkCocoaRenderWindowInteractor.h"
#import "vtkConeSource.h"
#import "vtkCylinderSource.h"
#import "vtkPolyDataMapper.h"
#import "vtkSmartPointer.h"
#import "vtkDebugLeaks.h"

@implementation MyDocument

- (void)setupLeftVTKView
{
    [leftVTKView initializeVTKSupport];

    // 'smart pointers' are used because they are very similiar to creating auto-realeased objects in Cocoa.

    // Personal Taste Section. I like to use a trackball interactor
    vtkSmartPointer<vtkInteractorStyleSwitch>   intStyle = vtkSmartPointer<vtkInteractorStyleSwitch>::New();
    intStyle->SetCurrentStyleToTrackballCamera();
    [leftVTKView getInteractor]->SetInteractorStyle(intStyle);

    // Create a cone, see the "VTK User's Guide" for details
    vtkSmartPointer<vtkConeSource> cone = vtkSmartPointer<vtkConeSource>::New();
        cone->SetHeight(3.0);
        cone->SetRadius(1.0);
        cone->SetResolution(100);
    vtkSmartPointer<vtkPolyDataMapper>  coneMapper = vtkSmartPointer<vtkPolyDataMapper>::New();
        coneMapper->SetInputConnection(cone->GetOutputPort());
    vtkSmartPointer<vtkActor>   coneActor = vtkSmartPointer<vtkActor>::New();
        coneActor->SetMapper(coneMapper);
    [leftVTKView getRenderer]->AddActor(coneActor);

    // Tell the system that the view needs to be redrawn
    [leftVTKView setNeedsDisplay:YES];
}

- (void)setupRightVTKView
{
    [rightVTKView initializeVTKSupport];

    // 'smart pointers' are used because they are very similiar to creating auto-realeased objects in Cocoa.

    // Personal Taste Section. I like to use a trackball interactor
    vtkSmartPointer<vtkInteractorStyleSwitch>   intStyle = vtkSmartPointer<vtkInteractorStyleSwitch>::New();
    intStyle->SetCurrentStyleToTrackballCamera();
    [rightVTKView getInteractor]->SetInteractorStyle(intStyle);

    // Create a cyclinder, see the "VTK User's Guide" for details
    vtkSmartPointer<vtkCylinderSource>  cylinder = vtkSmartPointer<vtkCylinderSource>::New();
        cylinder->SetResolution(100);
    vtkSmartPointer<vtkPolyDataMapper>  cylinderMapper = vtkSmartPointer<vtkPolyDataMapper>::New();
        cylinderMapper->SetInputConnection(cylinder->GetOutputPort());
    vtkSmartPointer<vtkActor>   cylinderActor = vtkSmartPointer<vtkActor>::New();
        cylinderActor->SetMapper(cylinderMapper);
    [rightVTKView getRenderer]->AddActor(cylinderActor);

    // Tell the system that the view needs to be redrawn
    [rightVTKView setNeedsDisplay:YES];
}

#pragma mark -

- (id)init 
{
    self = [super init];
    if (self != nil) {
        // Add your subclass-specific initialization here.
        // If an error occurs here, send a [self release] message and return nil.
    }
    return self;
}

- (void)close
{
    [super close];

    // Releases memory allocated in initializeVTKSupport.
    // In a way, calling these is pointless since the application is quitting anyway.
    [leftVTKView cleanUpVTKSupport];
    [rightVTKView cleanUpVTKSupport];

    // If you have built vtk with VTK_DEBUG_LEAKS on then this method will print out any leaks
    // that exist.  The sample has been careful to cleanup after itself, so there should be no leaks.
    vtkDebugLeaks::PrintCurrentLeaks();
}

- (NSString *)windowNibName
{
    // Override returning the nib file name of the document
    // If you need to use a subclass of NSWindowController or if your document supports multiple NSWindowControllers, you should remove this method and override -makeWindowControllers instead.
    return @"MyDocument";
}

- (void)windowControllerDidLoadNib:(NSWindowController *)windowController
{
    [super windowControllerDidLoadNib:windowController];

    // vtk stuff
    [self setupLeftVTKView];
    [self setupRightVTKView];
}


- (NSData *)dataRepresentationOfType:(NSString *)aType
{
    // Insert code here to write your document from the given data.  You can also choose to override -fileWrapperRepresentationOfType: or -writeToFile:ofType: instead.

    // For applications targeted for Tiger or later systems, you should use the new Tiger API -dataOfType:error:.  In this case you can also choose to override -writeToURL:ofType:error:, -fileWrapperOfType:error:, or -writeToURL:ofType:forSaveOperation:originalContentsURL:error: instead.

    return nil;
}

- (BOOL)loadDataRepresentation:(NSData *)data ofType:(NSString *)aType
{
    // Insert code here to read your document from the given data.  You can also choose to override -loadFileWrapperRepresentation:ofType: or -readFromFile:ofType: instead.

    // For applications targeted for Tiger or later systems, you should use the new Tiger API readFromData:ofType:error:.  In this case you can also choose to override -readFromURL:ofType:error: or -readFromFileWrapper:ofType:error: instead.

    return YES;
}

@end
