#pragma once
#include "BaseDlg.h"
#include "afxcmn.h"
#include "afxwin.h"
#include "NavHelper.h"

#define MAX_HOLE_SPINNERS 15

// CMultiShotDlg dialog

class CMultiShotDlg : public CBaseDlg
{

public:
	CMultiShotDlg(CWnd* pParent = NULL);   // standard constructor
	virtual ~CMultiShotDlg();
  bool RecordingISValues() {return mRecordingCustom || mRecordingRegular;};
  bool AdjustingISValues() {return mSteppingAdjusting > 0; };
  static bool DoingAutoStepAdj() { return mDoingAutoStepAdj; };
  static void SetSteppingAdjusting(int inVal) { mSteppingAdjusting = inVal; };

// Dialog Data
	enum { IDD = IDD_MULTI_SHOT_SETUP };

protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
	virtual void OnOK();
	virtual void OnCancel();
	virtual BOOL OnInitDialog();
  virtual BOOL PreTranslateMessage(MSG* pMsg);
  virtual void PostNcDestroy();

	DECLARE_MESSAGE_MAP()

private:
  static MultiShotParams *mActiveParams;
  MultiShotParams mSavedParams;
  static CShiftManager *mShiftManager;
  int mPanelStart[7];
  int mNumInPanel[7];
  static bool mRecordingRegular;
  static bool mRecordingCustom;
  static int mSteppingAdjusting;
  static std::vector<double> mSavedISX, mSavedISY;
  static BOOL mSavedMouseStage;
  static double mRecordStageX, mRecordStageY;
  static double mStartingISX, mStartingISY;
  bool mDisabledDialog;
  static int mNavPointIncrement;
  bool mWarnedIStoNav;
  int mLastMagIndex;
  double mLastIntensity;
  double mLastDrawTime;
  static int mSavedLDForCamera;
  static int mAreaSaved;
  static int mStepConsNum;
  static float mSavedDefOffset;
  static LowDoseParams mSavedLDParams;
  static MontParam *mSavedMontParam;
  static CSerialEMApp *mWinApp;
  static bool mDoingAutoStepAdj;
  static float mSavedPrevExp;
  static double mDeferredISX;
  static double mDeferredISY;
  static int mAutoAdjCropSize;
  static int mSavedShiftsOnAcq;
  static bool mAutoAdjDoCenter;

public:
  bool mCanReturnEarly;
  BOOL mHasIlluminatedArea;
  CString m_strNumShots;
  CSpinButtonCtrl m_sbcNumShots;
  afx_msg void OnDeltaposSpinNumShots(NMHDR *pNMHDR, LRESULT *pResult);
  float m_fSpokeDist;
  afx_msg void OnCheckEarlyReturn();
  CEdit m_editEarlyFrames;
  int m_iEarlyFrames;
  BOOL m_bSaveRecord;
  CButton m_butUseIllumArea;
  BOOL m_bUseIllumArea;
  CStatic m_statBeamDiam;
  CStatic m_statBeamMicrons;
  CEdit m_editBeamDiam;
  float m_fBeamDiam;
  afx_msg void OnRnoCenter();
  afx_msg void OnKillfocusEditSpokeDist();
  afx_msg void OnKillfocusEditBeamDiam();
  afx_msg void OnCheckUseIllumArea();
  int m_iCenterShot;
  CStatic m_statNumEarly;
  CButton m_butNoEarly;
  int m_iEarlyReturn;
  afx_msg void OnRnoEarly();
  void UpdateAndUseMSparams(bool draw = true);
  float m_fExtraDelay;
  CButton m_butSaveRecord;
  afx_msg void OnKillfocusEditEarlyFrames();
  void ManageEnables(void);
  void ManageHexGrid();
  CButton m_butAdjustBeamTilt;
  BOOL m_bAdjustBeamTilt;
  CStatic m_statComaIScal;
  CStatic m_statComaConditions;
  BOOL m_bDoShotsInHoles;
  BOOL m_bDoMultipleHoles;
  CStatic m_statNumXholes;
  CStatic m_statNumYholes;
  CString m_strNumYholes;
  CString m_strNumXholes;
  CSpinButtonCtrl m_sbcNumXholes;
  CSpinButtonCtrl m_sbcNumYholes;
  CButton m_butSetRegular;
  CButton m_butSetCustom;
  CButton m_butUseCustom;
  BOOL m_bUseCustom;
  CStatic m_statRegular;
  CStatic m_statSpacing;
  afx_msg void OnDoMultipleHoles();
  afx_msg void OnUseCustom();
  afx_msg void OnDeltaposSpinNumXHoles(NMHDR *pNMHDR, LRESULT *pResult);
  afx_msg void OnDeltaposSpinNumYHoles(NMHDR *pNMHDR, LRESULT *pResult);
  afx_msg void OnSetRegular();
  afx_msg void OnSetCustom();
  afx_msg void OnDoShotsInHole();
  CButton m_butNoCenter;
  CButton m_butCenterBefore;
  CButton m_butCenterAfter;
  CEdit m_editSpokeDist;
  CStatic m_statNumPeripheral;
  CStatic m_statCenterGroup;
  CStatic m_statNumShots;
  CStatic m_statCenterDist;
  CStatic m_statCenterUm;
  CStatic m_statHoleDelayFac;
  CEdit m_editHoleDelayFac;
  float m_fHoleDelayFac;
  void ManagePanels(bool adjust = false);
  CButton m_butSaveIS;
  CButton m_butEndPattern;
  CButton m_butAbort;
  afx_msg void OnButSaveIs();
  static int DoSaveIS(CString &str);
  static void AutoStepAdjNextTask(int param);
  static int AutoStepBusy();
  static ScaleMat ISfocusAdjustmentForBufOrArea(EMimageBuffer *imBufs, int area);
  afx_msg void OnButEndPattern();
  afx_msg void OnButAbort();
  CStatic m_statSaveInstructions;
  bool RecordingHoles() {return mRecordingRegular || mRecordingCustom || mSteppingAdjusting;};
  static void StopRecording(void);
  static void CloseTempPrevMontage(const char *filename);
  void StartRecording(const char *instruct);
  static int DoStartRecording();
  static int SetupTempPrevMontage(const char *filename, int xNframes, int yNframes, 
    int overviewBin, int overlapDiv);
  static int StartImageIfNeeded();
  static bool CanDoAutoAdjust(int magType, int areaToUse,
    int otherMag, int prevMag, float holeSize, BOOL hexGrid, BOOL custom, int &cropSize, CString &str);
  void UpdateSettings(void);
  CEdit m_editExtraDelay;
  CButton m_butCancel;
  CButton m_butIStoPt;
  afx_msg void OnButIsToPt();
  void UpdateMultiDisplay(int magInd, double intensity);
  BOOL mLastPanelStates[7];
  afx_msg void OnSaveRecord();
  afx_msg void OnAdjustBeamTilt();
  CButton m_butOmit3x3Corners;
  BOOL m_bOmit3x3Corners;
  afx_msg void OnOmit3x3Corners();
  BOOL m_bDoSecondRing;
  CStatic m_statNum2Shots;
  CStatic m_statRing2Shots;
  CStatic m_statRing2Um;
  CEdit m_editRing2Dist;
  CSpinButtonCtrl m_sbcRing2Num;
  afx_msg void OnDoSecondRing();
  afx_msg void OnDeltaposSpinRing2Num(NMHDR *pNMHDR, LRESULT *pResult);
  float m_fRing2Dist;
  afx_msg void OnKillfocusEditRing2Dist();
  CString m_strNum2Shots;
  CButton m_butDoSecondRing;
  CButton m_butUseLastHoleVecs;
  CButton m_butStepAdjust;
  afx_msg void OnButUseLastHoleVecs();
  afx_msg void OnButStepAdjust();
  CButton m_butCalBTvsIS;
  CButton m_butTestMultishot;
  afx_msg void OnButCalBtVsIs();
  afx_msg void OnButTestMultishot();
  CButton m_butHexGrid;
  static BOOL m_bHexGrid;
  afx_msg void OnCheckHexGrid();
  CString m_strAdjustStatus;
  CButton m_butUseMapVectors;
  afx_msg void OnButUseMapVectors();
  int ConfirmReplacingShiftVectors(int kind, int vecType);
  CButton m_butApplyAdjustment;
  afx_msg void OnButApplyAdjustment();
  afx_msg void OnKillfocusEditHoleDelayFac();
  CButton m_butUseNavPts;
  afx_msg void OnButUseNavPts();
  afx_msg void OnButUndoAuto();
};