X-lanni
156 lines
2.8 KiB
C++
156 lines
2.8 KiB
C++
/*
|
|
* ProgressionStatus.h
|
|
*
|
|
* Created on: Feb 28, 2017
|
|
* Author: mathieu
|
|
*/
|
|
|
|
#ifndef APP_ANDROID_JNI_PROGRESSIONSTATUS_H_
|
|
#define APP_ANDROID_JNI_PROGRESSIONSTATUS_H_
|
|
|
|
#include <rtabmap/core/ProgressState.h>
|
|
#include <rtabmap/utilite/ULogger.h>
|
|
#include <rtabmap/utilite/UEventsManager.h>
|
|
#ifdef __ANDROID__
|
|
#include <jni.h>
|
|
#endif
|
|
|
|
namespace rtabmap {
|
|
|
|
class ProgressEvent : public UEvent
|
|
{
|
|
public:
|
|
ProgressEvent(int count = 1) : count_(count){}
|
|
virtual std::string getClassName() const {return "ProgressEvent";}
|
|
|
|
int count_;
|
|
};
|
|
|
|
class ProgressionStatus: public ProgressState, public UEventsHandler
|
|
{
|
|
public:
|
|
ProgressionStatus() : count_(0), max_(100)
|
|
#ifdef __ANDROID__
|
|
, jvm_(0), rtabmap_(0)
|
|
#else
|
|
, swiftClassPtr_(0)
|
|
#endif
|
|
{
|
|
registerToEventsManager();
|
|
}
|
|
|
|
#ifdef __ANDROID__
|
|
void setJavaObjects(JavaVM * jvm, jobject rtabmap)
|
|
{
|
|
jvm_ = jvm;
|
|
rtabmap_ = rtabmap;
|
|
}
|
|
#else
|
|
void setSwiftCallback(void * classPtr, void(*callback)(void *, int, int))
|
|
{
|
|
swiftClassPtr_ = classPtr;
|
|
swiftCallback = callback;
|
|
}
|
|
#endif
|
|
|
|
void reset(int max)
|
|
{
|
|
count_=-1;
|
|
max_ = max;
|
|
setCanceled(false);
|
|
|
|
increment();
|
|
}
|
|
|
|
void setMax(int max)
|
|
{
|
|
max_ = max;
|
|
}
|
|
int getMax() const {return max_;}
|
|
|
|
void increment(int count = 1) const
|
|
{
|
|
UEventsManager::post(new ProgressEvent(count));
|
|
}
|
|
|
|
void finish()
|
|
{
|
|
|
|
}
|
|
|
|
virtual bool callback(const std::string & msg) const
|
|
{
|
|
if(!isCanceled())
|
|
{
|
|
increment();
|
|
}
|
|
|
|
return ProgressState::callback(msg);
|
|
}
|
|
virtual ~ProgressionStatus(){}
|
|
|
|
protected:
|
|
virtual bool handleEvent(UEvent * event)
|
|
{
|
|
if(event->getClassName().compare("ProgressEvent") == 0)
|
|
{
|
|
count_ += ((ProgressEvent*)event)->count_;
|
|
// Call JAVA callback
|
|
bool success = false;
|
|
#ifdef __ANDROID__
|
|
if(jvm_ && rtabmap_)
|
|
{
|
|
JNIEnv *env = 0;
|
|
jint rs = jvm_->AttachCurrentThread(&env, NULL);
|
|
if(rs == JNI_OK && env)
|
|
{
|
|
jclass clazz = env->GetObjectClass(rtabmap_);
|
|
if(clazz)
|
|
{
|
|
jmethodID methodID = env->GetMethodID(clazz, "updateProgressionCallback", "(II)V" );
|
|
if(methodID)
|
|
{
|
|
env->CallVoidMethod(rtabmap_, methodID,
|
|
count_,
|
|
max_);
|
|
success = true;
|
|
}
|
|
}
|
|
}
|
|
jvm_->DetachCurrentThread();
|
|
}
|
|
#else // APPLE
|
|
if(swiftClassPtr_)
|
|
{
|
|
std::function<void()> actualCallback = [&](){
|
|
swiftCallback(swiftClassPtr_, count_, max_);
|
|
};
|
|
actualCallback();
|
|
success = true;
|
|
}
|
|
#endif
|
|
if(!success)
|
|
{
|
|
UERROR("Failed to call rtabmap::updateProgressionCallback");
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
private:
|
|
int count_;
|
|
int max_;
|
|
#ifdef __ANDROID__
|
|
JavaVM *jvm_;
|
|
jobject rtabmap_;
|
|
#else
|
|
void * swiftClassPtr_;
|
|
void(*swiftCallback)(void *, int, int);
|
|
#endif
|
|
};
|
|
|
|
}
|
|
|
|
|
|
#endif /* APP_ANDROID_JNI_PROGRESSIONSTATUS_H_ */
|