def regenerateSeefloor(self):
driftsStepSize = self.step_size()
print "Using driftsStepSize: " + str(driftsStepSize)
#TODO: extract logic in few rows into a "regenerate drafts" module/class
print("regenerating/interpolating Drafts")
manualDrifts = DriftManualData.createFromFile(self.__folderStruct)
rawDrifts = DriftRawData(self.__folderStruct)
df = rawDrifts.interpolate(manualDrifts, driftsStepSize)
drifts = DriftData.createFromFolderStruct(self.__folderStruct)
drifts.setDF(df)
drifts.saveToFile(self.__folderStruct.getDriftsFilepath())
print("regenerating/interpolating RedDots")
rdd = RedDotsData.createFromFolderStruct(self.__folderStruct)
rdd.saveInterpolatedDFToFile(drifts.minFrameID(),
drifts.maxFrameID() + 1)
print("regenerating SeeFloor")
sf = SeeFloor.createFromFolderStruct(self.__folderStruct)
sf.saveToFile()
print("regenerating crabs_on_seefloor")
crabs = CrabsData(self.__folderStruct)
crabs_on_seefloor_df = crabs.generate_crabs_on_seefloor(sf)
crabs_on_seefloor_df.save_file_csv(
self.__folderStruct.getCrabsOnSeefloorFilepath())
def regenerateGraphs(self):
print("drawign graphs for RedDots")
rdd = RedDotsData.createFromFolderStruct(self.__folderStruct)
rdd.saveGraphOfAngle()
rdd.saveGraphOfDistance()
print("drawing graphs for SeeFloor")
sf = SeeFloor.createFromFolderStruct(self.__folderStruct)
sf.saveGraphSeefloorY()
sf.saveGraphSeefloorX()
sf.saveGraphSeefloorXY()
sf.saveGraphDriftsMillimeters()
sf.saveGraphDriftsPixels()
def __init__(self, imageWin, folderStruct, videoStream, driftData):
# type: (ImageWindow, FolderStructure, VideoStream, DriftData) -> ScientistUI
self.__zoom = False
self.__markingDrift = False
self.__contrastLevel = self.CONTRAST_NORMAL
self.__imageWin = imageWin
self.__folderStruct = folderStruct
self.__videoStream = videoStream
self.__driftData = driftData
self.__badFramesData = BadFramesData.createFromFolderStruct(
folderStruct)
self.__redDotsData = RedDotsData.createFromFolderStruct(folderStruct)
self.__seeFloor = SeeFloor.createFromFolderStruct(folderStruct)
self.__seeFloorNoBadBlocks = SeeFloorNoBadBlocks.createFromFolderStruct(
folderStruct)
self.__crabData = CrabsData(self.__folderStruct)
self.__redDotsUI = RedDotsUI(self.__videoStream)
self.__markersData = MarkersData(folderStruct)
self.__marker_id = "0"
rootDir = "C:/workspaces/AnjutkaVideo/2020-Kara/2020.09.16_6922"
videoFileName = "R_20200916_194953_20200916_195355"
# rootDir = "C:/workspaces/AnjutkaVideo/2020-Kara/2020.09.18_6923"
# videoFileName = "R_20200918_111643_20200918_112107"
folderStruct = FolderStructure(rootDir, videoFileName)
StreamToLogger(folderStruct.getLogFilepath())
print("Starting to cut video into frames")
interpolator = InterpolateController(folderStruct)
interpolator.regenerateSeefloor()
seefloorGeometry = SeeFloor.createFromFolderStruct(folderStruct)
videoStream = VideoStream(folderStruct.getVideoFilepath())
framesStitcher = VideoToImages(seefloorGeometry, videoStream)
#framesToSaveToFile = framesStitcher.determineFrames()
#print("Dataframe Contains:", framesToSaveToFile)
frameIDs = framesStitcher.getListOfFrameIDs()
framesStitcher.saveFramesToFile(frameIDs, folderStruct.getFramesDirpath())
loggerFramesCSV = Logger.openInOverwriteMode(folderStruct.getFramesFilepath())
framesStitcher.saveFramesToCSVFile(frameIDs, loggerFramesCSV)
crabsData = CrabsData(folderStruct)
lst = crabsData.allFramesWithCrabs()