/Code/core/ComponentColorMask.cpp

https://github.com/roboshepherd/swistrackplus · C++ · 104 lines · 79 code · 15 blank · 10 comment · 29 complexity · 5480cf60bd3db580c3d7693c4707f290 MD5 · raw file 

    

  1. #include "ComponentColorMask.h"

    2. 

  2. #define THISCLASS ComponentColorMask

    3. 

  3. 

    4. 

  4. #include <highgui.h>

    5. 

  5. #include "DisplayEditor.h"

    6. 

  6. 

    7. 

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

    8. 

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

    9. 

  9. 		mMaskImage(0), mMode(cMode_BlackBlack),

    10. 

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

    11. 

  11. 

    12. 

  12. 	// Data structure relations

    13. 

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

    14. 

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

    15. 

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

    16. 

  16. 	AddDisplay(&mDisplayOutput);

    17. 

  17. 

    18. 

  18. 	// Read the XML configuration file

    19. 

  19. 	Initialize();

    20. 

  20. }

    21. 

  21. 

    22. 

  22. THISCLASS::~ComponentColorMask() {

    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. 		// Already in BGR, do nothing

    40. 

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

    41. 

  41. 		// Grayscale image, we convert to BGR

    42. 

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

    43. 

  43. 		cvCvtColor(mMaskImage, img, CV_GRAY2BGR);

    44. 

  44. 		cvReleaseImage(&mMaskImage);

    45. 

  45. 		mMaskImage = img;

    46. 

  46. 	} else {

    47. 

  47. 		AddError(wxT("Invalid mask file. The mask file must be a grayscale or color image."));

    48. 

  48. 		return;

    49. 

  49. 	}

    50. 

  50. 

    51. 

  51. 	// load other parameters:

    52. 

  52. 	OnReloadConfiguration();

    53. 

  53. 	return;

    54. 

  54. }

    55. 

  55. 

    56. 

  56. void THISCLASS::OnReloadConfiguration() {

    57. 

  57. 	// Mask mode

    58. 

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

    59. 

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

    60. 

  60. 		mMode = cMode_WhiteWhite;

    61. 

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

    62. 

  62. 		mMode = cMode_WhiteBlack;

    63. 

  63. 		cvNot(mMaskImage, mMaskImage);

    64. 

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

    65. 

  65. 		mMode = cMode_BlackWhite;

    66. 

  66. 		cvNot(mMaskImage, mMaskImage);

    67. 

  67. 	} else {

    68. 

  68. 		mMode = cMode_BlackBlack;

    69. 

  69. 	}

    70. 

  70. }

    71. 

  71. 

    72. 

  72. void THISCLASS::OnStep() {

    73. 

  73. 	if (! mCore->mDataStructureImageColor.mImage) {

    74. 

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

    75. 

  75. 		return;

    76. 

  76. 	}

    77. 

  77. 

    78. 

  78. 	// Mask the image

    79. 

  79. 	if (mMaskImage) {

    80. 

  80. 		if ((mCore->mDataStructureImageColor.mImage->width != mMaskImage->width) || (mCore->mDataStructureImageColor.mImage->height != mMaskImage->height)) {

    81. 

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

    82. 

  82. 			return;

    83. 

  83. 		}

    84. 

  84. 

    85. 

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

    86. 

  86. 			cvOr(mCore->mDataStructureImageColor.mImage, mMaskImage, mCore->mDataStructureImageColor.mImage);

    87. 

  87. 		} else {

    88. 

  88. 			cvAnd(mCore->mDataStructureImageColor.mImage, mMaskImage, mCore->mDataStructureImageColor.mImage);

    89. 

  89. 		}

    90. 

  90. 	}

    91. 

  91. 

    92. 

  92. 	// Set the display

    93. 

  93. 	DisplayEditor de(&mDisplayOutput);

    94. 

  94. 	if (de.IsActive()) {

    95. 

  95. 		de.SetMainImage(mCore->mDataStructureImageColor.mImage);

    96. 

  96. 	}

    97. 

  97. }

    98. 

  98. 

    99. 

  99. void THISCLASS::OnStepCleanup() {

    100. 

  100. }

    101. 

  101. 

    102. 

  102. void THISCLASS::OnStop() {

    103. 

  103. 	cvReleaseImage(&mMaskImage);

    104. 

  104. }
    105.