#pragma once
#include "paddle/fluid/train/custom_trainer/feed/common/registerer.h"
#include "paddle/fluid/train/custom_trainer/feed/trainer_context.h"

namespace paddle {
namespace custom_trainer {
namespace feed {

class Process {
public:
    Process() {}
    virtual ~Process() {}
    virtual int initialize(std::shared_ptr<TrainerContext> context_ptr) {
        _context_ptr = context_ptr.get();
        return 0;
    }
    virtual int run();
protected:
    TrainerContext* _context_ptr = NULL;
};
REGIST_REGISTERER(Process);

}  // namespace feed
}  // namespace custom_trainer
}  // namespace paddle