/Code/core/ComponentBinaryMask.cpp

https://github.com/roboshepherd/swistrackplus · C++ · 107 lines · 81 code · 15 blank · 11 comment · 29 complexity · 01a037be9e71e934d8503f4150409b46 MD5 · raw file 

    

  1. #include "ComponentBinaryMask.h"

    2. 

  2. #define THISCLASS ComponentBinaryMask

    3. 

  3. 

    4. 

  4. #include <highgui.h>

    5. 

  5. #include "DisplayEditor.h"

    6. 

  6. 

    7. 

  7. THISCLASS::ComponentBinaryMask(SwisTrackCore *stc):

    8. 

  8. 		Component(stc, wxT("BinaryMask")),

    9. 

  9. 		mMaskImage(0),

    10. 

  10. 		mDisplayOutput(wxT("Output"), wxT("After applying mask")) {

    11. 

  11. 

    12. 

  12. 	// Data structure relations

    13. 

  13. 	mCategory = &(mCore->mCategoryPreprocessingBinary);

    14. 

  14. 	AddDataStructureRead(&(mCore->mDataStructureImageBinary));

    15. 

  15. 	AddDataStructureWrite(&(mCore->mDataStructureImageBinary));

    16. 

  16. 	AddDisplay(&mDisplayOutput);

    17. 

  17. 

    18. 

  18. 	// Read the XML configuration file

    19. 

  19. 	Initialize();

    20. 

  20. }

    21. 

  21. 

    22. 

  22. THISCLASS::~ComponentBinaryMask() {

    23. 

  23. }

    24. 

  24. 

    25. 

  25. void THISCLASS::OnStart() {

    26. 

  26. 	// Load mask image

    27. 

  27. 	wxString filename_string = GetConfigurationString(wxT("MaskImage"), wxT(""));

    28. 

  28. 	wxFileName filename = mCore->GetProjectFileName(filename_string);

    29. 

  29. 	if (filename.IsOk()) {

    30. 

  30. 		mMaskImage = cvLoadImage(filename.GetFullPath().mb_str(wxConvFile), CV_LOAD_IMAGE_ANYDEPTH | CV_LOAD_IMAGE_ANYCOLOR);

    31. 

  31. 	}

    32. 

  32. 	if (! mMaskImage) {

    33. 

  33. 		AddError(wxT("Cannot open mask file."));

    34. 

  34. 		return;

    35. 

  35. 	}

    36. 

  36. 

    37. 

  37. 	// Convert mask image

    38. 

  38. 	if (mMaskImage->nChannels == 3) {

    39. 

  39. 		// BGR case, we convert to gray

    40. 

  40. 		IplImage *img = cvCreateImage(cvSize(mMaskImage->width, mMaskImage->height), mMaskImage->depth, 1);

    41. 

  41. 		cvCvtColor(mMaskImage, img, CV_BGR2GRAY);

    42. 

  42. 		cvReleaseImage(&mMaskImage);

    43. 

  43. 		mMaskImage = img;

    44. 

  44. 	} else if (mMaskImage->nChannels == 1) {

    45. 

  45. 		// Already in gray, do nothing

    46. 

  46. 	} else {

    47. 

  47. 		// Other cases, we take the first channel

    48. 

  48. 		IplImage *img = cvCreateImage(cvSize(mMaskImage->width, mMaskImage->height), mMaskImage->depth, 1);

    49. 

  49. 		cvCvtPixToPlane(mMaskImage, img, NULL, NULL, NULL);

    50. 

  50. 		cvReleaseImage(&mMaskImage);

    51. 

  51. 		mMaskImage = img;

    52. 

  52. 	}

    53. 

  53. 

    54. 

  54. 	// load other parameters:

    55. 

  55. 	OnReloadConfiguration();

    56. 

  56. 	return;

    57. 

  57. }

    58. 

  58. 

    59. 

  59. void THISCLASS::OnReloadConfiguration() {

    60. 

  60. 	// Mask mode

    61. 

  61. 	wxString mode = GetConfigurationString(wxT("Mode"), wxT("black-black"));

    62. 

  62. 	if (mode == wxT("white-white")) {

    63. 

  63. 		mMode = cMode_WhiteWhite;

    64. 

  64. 	} else if (mode == wxT("white-black")) {

    65. 

  65. 		mMode = cMode_WhiteBlack;

    66. 

  66. 		cvNot(mMaskImage, mMaskImage);

    67. 

  67. 	} else if (mode == wxT("black-white")) {

    68. 

  68. 		mMode = cMode_BlackWhite;

    69. 

  69. 		cvNot(mMaskImage, mMaskImage);

    70. 

  70. 	} else {

    71. 

  71. 		mMode = cMode_BlackBlack;

    72. 

  72. 	}

    73. 

  73. }

    74. 

  74. 

    75. 

  75. void THISCLASS::OnStep() {

    76. 

  76. 	if (! mCore->mDataStructureImageBinary.mImage) {

    77. 

  77. 		AddError(wxT("No input image."));

    78. 

  78. 		return;

    79. 

  79. 	}

    80. 

  80. 

    81. 

  81. 	// Mask the image

    82. 

  82. 	if (mMaskImage) {

    83. 

  83. 		if ((mCore->mDataStructureImageBinary.mImage->width != mMaskImage->width) || (mCore->mDataStructureImageBinary.mImage->height != mMaskImage->height)) {

    84. 

  84. 			AddError(wxT("Wrong mask size."));

    85. 

  85. 			return;

    86. 

  86. 		}

    87. 

  87. 

    88. 

  88. 		if ((mMode == cMode_WhiteWhite) || (mMode == cMode_BlackWhite)) {

    89. 

  89. 			cvOr(mCore->mDataStructureImageBinary.mImage, mMaskImage, mCore->mDataStructureImageBinary.mImage);

    90. 

  90. 		} else {

    91. 

  91. 			cvAnd(mCore->mDataStructureImageBinary.mImage, mMaskImage, mCore->mDataStructureImageBinary.mImage);

    92. 

  92. 		}

    93. 

  93. 	}

    94. 

  94. 

    95. 

  95. 	// Set the display

    96. 

  96. 	DisplayEditor de(&mDisplayOutput);

    97. 

  97. 	if (de.IsActive()) {

    98. 

  98. 		de.SetMainImage(mCore->mDataStructureImageBinary.mImage);

    99. 

  99. 	}

    100. 

  100. }

    101. 

  101. 

    102. 

  102. void THISCLASS::OnStepCleanup() {

    103. 

  103. }

    104. 

  104. 

    105. 

  105. void THISCLASS::OnStop() {

    106. 

  106. 	cvReleaseImage(&mMaskImage);

    107. 

  107. }
    108.