#ifndef INCLUDED_MEMORY_ITERATOR
#define INCLUDED_MEMORY_ITERATOR

#include "MemList.h"
#include "MemorySegment.h"
#include "AddrList.h"
#include "SchedTask.h"

enum {

  READ_WRITE,
  READ,
  WRITE,

};

class MemIterator {
public:

    // 走査する アドレスリスト
    AddrListPtr addr_list;

    // 扱う MemorySegment のリスト
    MemList *mem_list;

    // ms の stage
    MemorySegmentPtr read_ms;
    MemorySegmentPtr exec_ms;
    MemorySegmentPtr write_ms;
    MemorySegmentPtr free_ms;

    SchedTask *smanager;

    MemIterator(AddrListPtr add_list, MemList *mem_list, int cmd, SchedTask *smanager);
    ~MemIterator();


    MemorySegmentPtr (MemIterator::*get_ms)(void);

    int hasNext(void);

    MemorySegmentPtr get_read_write_ms(void);
    MemorySegmentPtr get_read_ms(void);
    MemorySegmentPtr get_write_ms(void);

    void collect_ms(void);
    void overwrite_list(AddrListPtr list);

};

typedef MemIterator *MemIteratorPtr;

#endif