// YourApp.h or YourApp-BrindgingHeader.h
@import AugmentPlayerSDK;

// These are demo credentials linked to the demo catalog below
NSString* APP_ID = @"357fee36746668573ceb2f5957c4869ee1a62a112639bac9b0fae43c7c431692";
NSString* APP_SECRET = @"80ae1420e164e0440d5329067bcdd953e9fa6c63b75c001c06d169a4f11268c5";
NSString* VUFORIA_KEY = @"ATQqCM7/////AAAAGXLs+GRi0UwXh0X+/qQL49dbZGym8kKo+iRtgC95tbJoCWjXXZihDl5pzxoca2JxLcYxBJ2pIeIE4dNcK0etMeb1746L7lq6vSFen43cS7P1P/HXjwHtUouV5Xus2U0F7WHUTKuO629jKFO13fBQczuY52UJcSEhsu9jHPMaupo5CpqQT3TFTQjlhzHhVXiVMEqq7RI+Edwh8TCSfGAbNRdbIELTfK+8YDYqwEHDbp62mFrs68YnCEQZDrpcLyC8WzFCVZtnUq3Cj3YBUfQ6gNnENYiuLf06gAAF/FcaF65VYveGRBbp3hpkqolX28bxPiUYNVknCSFXICPHciVntxF+rcHW5rrX7Cg/IVFGdNRF";

@interface ViewController ()

@property (weak, nonatomic) IBOutlet UIActivityIndicatorView* activityIndicator;
@property (nonatomic, weak) NSURLSessionDownloadTask* activeDownloadTask;

// AugmentPlayerSDK part
@property (nonatomic, weak) IBOutlet AGTView* ARView;
@property (nonatomic, strong) id<AGTAugmentPlayer> augmentPlayer;
@property (nonatomic, strong) AGTAugmentSDK* augmentSDK;

@property (nonatomic, weak) id<AGTCancellable> activeNetworkTask;

@end

// ...

- (void)viewDidLoad {
    [super viewDidLoad];

    self.augmentSDK = [AGTAugmentSDK alloc] initWithClientID: APP_ID
                                                clientSecret: APP_SECRET];

    // WARNING: [augmentSDK augmentPlayer] will now create a NEW instance for every call
    self.augmentPlayer = [augmentSDK createAugmentPlayer];

    // AGTView needs to know its renderer. if you do not set its augmentPlayer
    // property, it won't display anything
    [self.ARView setAugmentPlayer:self.augmentPlayer];
    [self prepareARSession];
}

- (void)viewWillDisappear:(BOOL)animated {
    [super viewWillDisappear: animated];
    NSError* error = nil;
    [self.augmentPlayer pause:&error];
    if (error != nil) {
        // Handle error
    } else {
        [self.augmentPlayer unloadAll];
    }
}

- (void)prepareARSession {
    // You need to grant camera permission before starting AugmentPlayerSDK
    // otherwise you will have an error during initialization: "Cannot access the camera"
    AVAuthorizationStatus authStatus = [AVCaptureDevice authorizationStatusForMediaType:AVMediaTypeVideo];
    if (authStatus == AVAuthorizationStatusAuthorized) {
        [self loadMyModel];
    }
    else if (authStatus == AVAuthorizationStatusNotDetermined) {
        [AVCaptureDevice requestAccessForMediaType:AVMediaTypeVideo
            completionHandler:^(BOOL granted) {
                if (granted) {
                    dispatch_async(dispatch_get_main_queue(), ^{
                        [self loadMyModel];
                    });
                }
            }];
    }
}

AGTTrackingStatusInitializing, //The underlying AR system is getting set-up.
AGTTrackingStatusNormal, //The AR tracking is working properly and the discovery mode is active.
AGTTrackingStatusNotAvailable, //The AR tracking is not available; try reseting the player.
AGTTrackingStatusLimitedExcessiveMotion, //Tracking is limited due to a excessive motion of the camera.
AGTTrackingStatusLimitedInsufficientFeatures, //Tracking is limited due to a lack of features (illumination or details) visible to the camera.
AGTTrackingStatusFeaturesDetected, //Some features has been detected. Invite the user to explorer a little more to detected a plane to place the product.
AGTTrackingStatusPlaneDetected, //Your camera is pointing right over a surface. Invite the user to tap to place the product on the plane.
AGTTrackingStatusError, //Something went wrong with the player, this message will tell you what.
AGTTrackingStatusTrackerDetected, //The user has detected a tracker (non-ArKit devices only).

- (void)trackStatus:(AGTTrackingStatus)status withErrorMessage:(NSString*)errorMessage {
    // handle events
}

- (void)viewDidLoad {
    // ....
    self.augmentPlayer.trackingStatusDelegate = self;
}

- (void)loadMyModel {
    id<AGTProduct> cachedProduct = [self.augmentSDK.productsDataController productForIdentifier: @"84"]; // From the catalog
    if (cachedProduct != nil) {
        [self addProductToAugmentPlayer:cachedProduct];
    }
    else {
        [self queryAndAddProductToAugmentPlayerWithIdentifier: @"84"];
    }
}

- (void)queryAndAddProductToAugmentPlayerWithIdentifier:(NSString*)productIdentifier {
    // you cannot directly add a product to AugmentPlayer
    // first, you need to check if your product has a model3D in Augment's database
    [self.activityIndicator startAnimating];
    __weak typeof(self) weakSelf = self; //__
    [self.augmentSDK.productsDataController
     checkIfModelDoesExistForProductIdentifier: @"84" brand: @"Whirlpool" name: @"Fridge" ean: nil completion:^(id<AGTProduct> product, NSError* queryError) {
         [weakSelf.activityIndicator stopAnimating];
         if (product != nil && queryError == nil) {
             NSLog(@"ViewController: Product has a model3D in Augment database!");
             [weakSelf addProductToAugmentPlayer:product];
         }
         else {
             // this error might be "model 3D not found in Augment database or anything else
             NSLog(@"ViewController: Query Error %@", queryError.localizedDescription);
         }
     }];
}

- (void)addProductToAugmentPlayer:(id<AGTProduct>)product {
    // after making sure that your product has a model3D, now you can add it to
    // AugmentPlayer via AugmentSDK
    [self.activityIndicator startAnimating];
    __weak typeof(self) weakSelf = self; //__
    [self.augmentSDK
     addProductToAugmentPlayer:product
     downloadProgress:^(NSProgress* progress) {
         NSLog(@"Download progress: %%%f", progress.fractionCompleted);
     }
     operationCompletionWithModelIdentifier:^(NSUUID *itemIdentifier, NSArray<NSError*> * errors) {
         [weakSelf.activityIndicator stopAnimating];
         if (errors.count <= 0 && itemIdentifier != nil) {
             NSLog(@"ViewController: model with id %@ added", itemIdentifier);
             // At that moment you made sure that a model 3D is added to
             // AugmentPlayer and now you can start AugmentPlayer
            [weakSelf startAR];
         }
         else {
             // As addProductToAugment method does more than one thing under the
             // hood, it may return multiple errors
             NSMutableString* errorString = [[NSMutableString alloc] init];
             [errors enumerateObjectsUsingBlock:^(NSError * obj, NSUInteger idx, BOOL * stop) {
                 [errorString appendString:obj.localizedDescription];
                 [errorString appendString:@" "];
             }];
             NSLog(@"ViewController: an error occurred %@", errorString);
         }
     }];
}

- (void)startAR {
    [self.augmentSDK.augmentPlayer resume];
}

- (IBAction)recenterProducts:(id)sender {
    [self.augmentSDK.augmentPlayer recenterProducts];
}

- (void)onModelAdded:(NSString*)model3DUuid;
- (void)onModelTranslated:(NSString*)model3DUuid;
- (void)onModelRotated:(NSString*)model3DUuid;