mirror of
https://github.com/eddyem/astrovideoguide_v3.git
synced 2025-12-06 02:35:11 +03:00
Add LocCorr_new: another variant of LocCorr (instead of pusirobot used multistepper; some errors fixed); I hope, sometimes I will finish loccorrNG
This commit is contained in:
parent
173bb8d0cc
commit
6ee58edc7d
BIN
LocCorr_new/15x12.png
Normal file
BIN
LocCorr_new/15x12.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 12 KiB |
9
LocCorr_new/CANSERVER_pusirobo
Normal file
9
LocCorr_new/CANSERVER_pusirobo
Normal file
@ -0,0 +1,9 @@
|
|||||||
|
MULTIPLY steps & speed by microsteps!!!
|
||||||
|
|
||||||
|
register U 0x581 stepper -> OK\nU maxspeed=OK
|
||||||
|
mesg U maxspeed 12800 -> OK\nU maxspeed=OK
|
||||||
|
register V 0x582 stepper -> OK\nV maxspeed=OK
|
||||||
|
mesg V maxspeed 12800 -> OK\nV maxspeed=OK
|
||||||
|
mesg U relmove 1600 -> OK\nU rotdir=OK\nU relsteps=OK
|
||||||
|
mesg U setzero -> OK\nU curpos=OK
|
||||||
|
mesg V setzero -> OK\nV curpos=OK
|
||||||
89
LocCorr_new/CMakeLists.txt
Normal file
89
LocCorr_new/CMakeLists.txt
Normal file
@ -0,0 +1,89 @@
|
|||||||
|
cmake_minimum_required(VERSION 3.0)
|
||||||
|
set(PROJ loccorr)
|
||||||
|
set(MINOR_VERSION "1")
|
||||||
|
set(MID_VERSION "0")
|
||||||
|
set(MAJOR_VERSION "0")
|
||||||
|
set(VERSION "${MAJOR_VERSION}.${MID_VERSION}.${MINOR_VERSION}")
|
||||||
|
|
||||||
|
project(${PROJ} VERSION ${VERSION} LANGUAGES C)
|
||||||
|
|
||||||
|
# default flags
|
||||||
|
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -std=gnu99")
|
||||||
|
|
||||||
|
set(CMAKE_COLOR_MAKEFILE ON)
|
||||||
|
|
||||||
|
# here is one of two variants: all .c in directory or .c files in list
|
||||||
|
aux_source_directory(${CMAKE_CURRENT_SOURCE_DIR} SOURCES)
|
||||||
|
|
||||||
|
set(CMAKE_VERBOSE_MAKEFILE "ON")
|
||||||
|
|
||||||
|
# list of options
|
||||||
|
option(DEBUG "Compile in debug mode" OFF)
|
||||||
|
|
||||||
|
# default flags
|
||||||
|
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -Wall -W -Wextra -std=gnu99")
|
||||||
|
set(CMAKE_COLOR_MAKEFILE ON)
|
||||||
|
|
||||||
|
# cmake -DDEBUG=yes -> debugging
|
||||||
|
if(DEBUG)
|
||||||
|
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -Og -g3 -ggdb -fno-builtin-strlen -Werror")
|
||||||
|
add_definitions(-DEBUG)
|
||||||
|
set(CMAKE_BUILD_TYPE DEBUG)
|
||||||
|
set(CMAKE_VERBOSE_MAKEFILE "ON")
|
||||||
|
else()
|
||||||
|
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -O3 -march=native -fdata-sections -ffunction-sections")
|
||||||
|
SET(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -Wl,--gc-sections")
|
||||||
|
set(CMAKE_BUILD_TYPE RELEASE)
|
||||||
|
endif()
|
||||||
|
|
||||||
|
message("Build type: ${CMAKE_BUILD_TYPE}")
|
||||||
|
|
||||||
|
###### pkgconfig ######
|
||||||
|
# pkg-config modules (for pkg-check-modules)
|
||||||
|
set(MODULES usefull_macros cfitsio improc)
|
||||||
|
|
||||||
|
# find packages:
|
||||||
|
SET(CMAKE_MODULE_PATH ${PROJECT_SOURCE_DIR})
|
||||||
|
find_package(PkgConfig REQUIRED)
|
||||||
|
find_package(FLYCAP REQUIRED)
|
||||||
|
find_package(BASLER REQUIRED)
|
||||||
|
pkg_check_modules(${PROJ} REQUIRED ${MODULES})
|
||||||
|
|
||||||
|
include(FindOpenMP)
|
||||||
|
if(OPENMP_FOUND)
|
||||||
|
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${OpenMP_C_FLAGS}")
|
||||||
|
set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} ${OpenMP_EXE_LINKER_FLAGS}")
|
||||||
|
add_definitions(-DOMP_FOUND)
|
||||||
|
endif()
|
||||||
|
if(NOT DEFINED PROCESSOR_COUNT)
|
||||||
|
set(PROCESSOR_COUNT 2) # by default 2 cores
|
||||||
|
execute_process(COMMAND getconf _NPROCESSORS_ONLN OUTPUT_VARIABLE PROCESSOR_COUNT OUTPUT_STRIP_TRAILING_WHITESPACE)
|
||||||
|
endif()
|
||||||
|
message("In multithreaded operations will use ${PROCESSOR_COUNT} threads")
|
||||||
|
|
||||||
|
|
||||||
|
# change wrong behaviour with install prefix
|
||||||
|
if(CMAKE_INSTALL_PREFIX_INITIALIZED_TO_DEFAULT AND CMAKE_INSTALL_PREFIX MATCHES "/usr/local")
|
||||||
|
else()
|
||||||
|
message("Change default install path to /usr/local")
|
||||||
|
set(CMAKE_INSTALL_PREFIX "/usr/local")
|
||||||
|
endif()
|
||||||
|
message("Install dir prefix: ${CMAKE_INSTALL_PREFIX}")
|
||||||
|
|
||||||
|
# exe file
|
||||||
|
add_executable(${PROJ} ${SOURCES})
|
||||||
|
# -I
|
||||||
|
include_directories(${${PROJ}_INCLUDE_DIRS} ${FLYCAP_INCLUDE_DIRS} ${BASLER_INCLUDE_DIRS})
|
||||||
|
# -L
|
||||||
|
link_directories(${${PROJ}_LIBRARY_DIRS} ${FLYCAP_LIBRARY_DIRS} ${BASLER_LIBRARY_DIRS})
|
||||||
|
# -D
|
||||||
|
add_definitions(${CFLAGS} -DLOCALEDIR=\"${LOCALEDIR}\"
|
||||||
|
-DPACKAGE_VERSION=\"${VERSION}\" -DGETTEXT_PACKAGE=\"${PROJ}\"
|
||||||
|
-DMINOR_VERSION=\"${MINOR_VERSION}\" -DMID_VERSION=\"${MID_VERSION}\"
|
||||||
|
-DMAJOR_VERSION=\"${MAJOR_VERSION}\" -DTHREAD_NUMBER=${PROCESSOR_COUNT})
|
||||||
|
|
||||||
|
# -l
|
||||||
|
target_link_libraries(${PROJ} ${${PROJ}_LIBRARIES} ${FLYCAP_LIBRARIES} ${BASLER_LIBRARIES} -lm)
|
||||||
|
|
||||||
|
# Installation of the program
|
||||||
|
INSTALL(TARGETS ${PROJ} DESTINATION "bin")
|
||||||
9
LocCorr_new/DEBUG.log.analyze
Executable file
9
LocCorr_new/DEBUG.log.analyze
Executable file
@ -0,0 +1,9 @@
|
|||||||
|
#!/bin/sh
|
||||||
|
|
||||||
|
ALLOC=$(grep ALLOCSZ DEBUG.log)
|
||||||
|
FREE=$(grep FREESZ DEBUG.log)
|
||||||
|
|
||||||
|
echo "Total: $(echo "$ALLOC" | wc -l) allocations and $(echo "$FREE" | wc -l) frees"
|
||||||
|
AL=$(echo "$ALLOC" | sed 's/ALLOCSZ(\([[:digit:]]*\))/\1/' | awk 'BEGIN{x=0;} {x=x+$1;} END{print x;}')
|
||||||
|
FR=$(echo "$FREE" | sed 's/FREESZ(\([[:digit:]]*\))/\1/' | awk 'BEGIN{x=0;} {x=x+$1;} END{print x;}')
|
||||||
|
echo "Allocated: $AL bytes, Freed: $FR bytes, difference: $((AL - FR)) bytes"
|
||||||
36
LocCorr_new/FindBASLER.cmake
Normal file
36
LocCorr_new/FindBASLER.cmake
Normal file
@ -0,0 +1,36 @@
|
|||||||
|
# - Try to find libpylonc.so
|
||||||
|
# Once done this will define
|
||||||
|
#
|
||||||
|
# BASLER_FOUND - system has lib
|
||||||
|
# BASLER_INCLUDE_DIR - include directory
|
||||||
|
# BASLER_LIBRARIES - Link these to use lib
|
||||||
|
|
||||||
|
# Copyright (c) 2021, Edward V. Emelianov <edward.emelianoff@gmail.com>
|
||||||
|
#
|
||||||
|
# Redistribution and use is allowed according to the terms of the GPLv2/GPLv3.
|
||||||
|
|
||||||
|
include(GNUInstallDirs)
|
||||||
|
|
||||||
|
find_path(BASLER_INCLUDE_DIR pylonc/PylonC.h
|
||||||
|
PATHS /usr/pylon/include /opt/pylon/include /usr/local/pylon/include /usr/include /usr/local/include /opt/include /opt/local/include
|
||||||
|
)
|
||||||
|
find_path(BASLER_LIBRARY_DIR libpylonc.so
|
||||||
|
PATHS /usr/pylon/lib /opt/pylon/lib /usr/local/pylon/lib /lib /lib64 /usr/lib /usr/lib64 /opt/lib /opt/lib64 /usr/local/lib /usr/local/lib64
|
||||||
|
)
|
||||||
|
find_library(BASLER_LIBRARY NAMES pylonc
|
||||||
|
PATHS /opt/pylon/lib /usr/pylon/lib /usr/local/pylon/lib /lib /lib64 /usr/lib /usr/lib64 /opt/lib /opt/lib64 /usr/local/lib /usr/local/lib64
|
||||||
|
)
|
||||||
|
|
||||||
|
find_package_handle_standard_args(BASLER DEFAULT_MSG BASLER_INCLUDE_DIR BASLER_LIBRARY BASLER_LIBRARY_DIR)
|
||||||
|
|
||||||
|
if(BASLER_FOUND)
|
||||||
|
set(BASLER_INCLUDE_DIRS ${BASLER_INCLUDE_DIR})
|
||||||
|
set(BASLER_LIBRARIES ${BASLER_LIBRARY})
|
||||||
|
set(BASLER_LIBRARY_DIRS ${BASLER_LIBRARY_DIR})
|
||||||
|
# message("BASLER include dir = ${BASLER_INCLUDE_DIRS}")
|
||||||
|
# message("BASLER lib = ${BASLER_LIBRARIES}")
|
||||||
|
# message("BASLER libdir = ${BASLER_LIBRARY_DIRS}")
|
||||||
|
mark_as_advanced(BASLER_INCLUDE_DIRS BASLER_LIBRARIES BASLER_LIBRARY_DIRS)
|
||||||
|
else()
|
||||||
|
message("BASLER not found")
|
||||||
|
endif(BASLER_FOUND)
|
||||||
40
LocCorr_new/FindFLYCAP.cmake
Normal file
40
LocCorr_new/FindFLYCAP.cmake
Normal file
@ -0,0 +1,40 @@
|
|||||||
|
# - Try to find libflycapture
|
||||||
|
# Once done this will define
|
||||||
|
#
|
||||||
|
# FLYCAP_FOUND - system has libflycapture
|
||||||
|
# FLYCAP_INCLUDE_DIR - include directory
|
||||||
|
# FLYCAP_LIBRARIES - Link these to use libflycapture
|
||||||
|
|
||||||
|
# Copyright (c) 2021, Edward V. Emelianov <edward.emelianoff@gmail.com>
|
||||||
|
#
|
||||||
|
# Redistribution and use is allowed according to the terms of the GPLv2/GPLv3.
|
||||||
|
|
||||||
|
include(GNUInstallDirs)
|
||||||
|
|
||||||
|
find_path(FLYCAP_INCLUDE_DIR FlyCapture2.h
|
||||||
|
PATH_SUFFIXES libflycapture flycapture
|
||||||
|
PATHS /usr/include /usr/local/include /opt/include /opt/local/include
|
||||||
|
)
|
||||||
|
find_path(FLYCAP_LIBRARY_DIR libflycapture.so
|
||||||
|
PATHS /lib /lib64 /usr/lib /usr/lib64 /opt/lib /opt/lib64 /usr/local/lib /usr/local/lib64
|
||||||
|
)
|
||||||
|
find_library(FLYCAP_LIBRARY NAMES flycapture
|
||||||
|
PATHS /lib /lib64 /usr/lib /usr/lib64 /opt/lib /opt/lib64 /usr/local/lib /usr/local/lib64
|
||||||
|
)
|
||||||
|
find_library(FLYCAP_LIBRARYC NAMES flycapture-c
|
||||||
|
PATHS /lib /lib64 /usr/lib /usr/lib64 /opt/lib /opt/lib64 /usr/local/lib /usr/local/lib64
|
||||||
|
)
|
||||||
|
|
||||||
|
find_package_handle_standard_args(FLYCAP DEFAULT_MSG FLYCAP_INCLUDE_DIR FLYCAP_LIBRARY FLYCAP_LIBRARYC FLYCAP_LIBRARY_DIR)
|
||||||
|
|
||||||
|
if(FLYCAP_FOUND)
|
||||||
|
set(FLYCAP_INCLUDE_DIRS ${FLYCAP_INCLUDE_DIR})
|
||||||
|
set(FLYCAP_LIBRARIES ${FLYCAP_LIBRARY} ${FLYCAP_LIBRARYC})
|
||||||
|
set(FLYCAP_LIBRARY_DIRS ${FLYCAP_LIBRARY_DIR})
|
||||||
|
# message("FLYCAP include dir = ${FLYCAP_INCLUDE_DIRS}")
|
||||||
|
# message("FLYCAP lib = ${FLYCAP_LIBRARIES}")
|
||||||
|
# message("FLYCAP libdir = ${FLYCAP_LIBRARY_DIRS}")
|
||||||
|
mark_as_advanced(FLYCAP_INCLUDE_DIRS FLYCAP_LIBRARIES FLYCAP_LIBRARY_DIRS)
|
||||||
|
else()
|
||||||
|
message("FLYCAP not found")
|
||||||
|
endif(FLYCAP_FOUND)
|
||||||
6
LocCorr_new/Readme.md
Normal file
6
LocCorr_new/Readme.md
Normal file
@ -0,0 +1,6 @@
|
|||||||
|
# Local corrector
|
||||||
|
|
||||||
|
Takes fresh images from CMOS, single file or directory using inotify, find objects, calculate their centroids and send messages to corrector over serial-proxy.
|
||||||
|
|
||||||
|
Used corrector controller - `multistepper`.
|
||||||
|
Used CMOS - Basler or Grasshopper.
|
||||||
335
LocCorr_new/basler.c
Normal file
335
LocCorr_new/basler.c
Normal file
@ -0,0 +1,335 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <pylonc/PylonC.h>
|
||||||
|
|
||||||
|
#include "basler.h"
|
||||||
|
#include "debug.h"
|
||||||
|
#include "imagefile.h"
|
||||||
|
|
||||||
|
static PYLON_DEVICE_HANDLE hDev;
|
||||||
|
static int isopened = FALSE;
|
||||||
|
static size_t payloadsize = 0; // size of imgBuf
|
||||||
|
static unsigned char *imgBuf = NULL;
|
||||||
|
static float expostime = 0.; // current exposition time
|
||||||
|
static PYLON_DEVICECALLBACK_HANDLE hCb;
|
||||||
|
|
||||||
|
typedef struct{
|
||||||
|
int64_t min;
|
||||||
|
int64_t max;
|
||||||
|
int64_t incr;
|
||||||
|
int64_t val;
|
||||||
|
} int64_values;
|
||||||
|
|
||||||
|
typedef struct{
|
||||||
|
double min;
|
||||||
|
double max;
|
||||||
|
double val;
|
||||||
|
} float_values;
|
||||||
|
|
||||||
|
static char* describeError(GENAPIC_RESULT reserr){
|
||||||
|
static char buf[1024];
|
||||||
|
char* errMsg, *bptr = buf;
|
||||||
|
size_t length, l = 1023;
|
||||||
|
GenApiGetLastErrorMessage(NULL, &length);
|
||||||
|
errMsg = MALLOC(char, length);
|
||||||
|
GenApiGetLastErrorMessage(errMsg, &length);
|
||||||
|
size_t ll = snprintf(bptr, l, "%s (%d); ", errMsg, (int)reserr);
|
||||||
|
if(ll > 0){l -= ll; bptr += ll;}
|
||||||
|
FREE(errMsg);
|
||||||
|
GenApiGetLastErrorDetail(NULL, &length);
|
||||||
|
errMsg = MALLOC(char, length);
|
||||||
|
GenApiGetLastErrorDetail(errMsg, &length);
|
||||||
|
snprintf(bptr, l, "%s", errMsg);
|
||||||
|
FREE(errMsg);
|
||||||
|
return buf;
|
||||||
|
}
|
||||||
|
|
||||||
|
#define PYLONFN(fn, ...) do{register GENAPIC_RESULT reserr; if(GENAPI_E_OK != (reserr=fn(__VA_ARGS__))){ \
|
||||||
|
WARNX(#fn "(): %s", describeError(reserr)); return FALSE;}}while(0)
|
||||||
|
|
||||||
|
static void disconnect(){
|
||||||
|
FNAME();
|
||||||
|
if(!isopened) return;
|
||||||
|
FREE(imgBuf);
|
||||||
|
PylonDeviceDeregisterRemovalCallback(hDev, hCb);
|
||||||
|
PylonDeviceClose(hDev);
|
||||||
|
PylonDestroyDevice(hDev);
|
||||||
|
PylonTerminate();
|
||||||
|
isopened = FALSE;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief chkNode - get node & check it for read/write
|
||||||
|
* @param phNode (io) - pointer to node
|
||||||
|
* @param nodeType - type of node
|
||||||
|
* @param wr - ==TRUE if need to check for writeable
|
||||||
|
* @return TRUE if node found & checked
|
||||||
|
*/
|
||||||
|
static int chkNode(NODE_HANDLE *phNode, const char *featureName, EGenApiNodeType nodeType, int wr){
|
||||||
|
if(!isopened || !phNode || !featureName) return FALSE;
|
||||||
|
NODEMAP_HANDLE hNodeMap;
|
||||||
|
EGenApiNodeType nt;
|
||||||
|
_Bool bv;
|
||||||
|
PYLONFN(PylonDeviceGetNodeMap, hDev, &hNodeMap);
|
||||||
|
PYLONFN(GenApiNodeMapGetNode, hNodeMap, featureName, phNode);
|
||||||
|
if(*phNode == GENAPIC_INVALID_HANDLE) return FALSE;
|
||||||
|
PYLONFN(GenApiNodeGetType, *phNode, &nt);
|
||||||
|
if(nodeType != nt) return FALSE;
|
||||||
|
PYLONFN(GenApiNodeIsReadable, *phNode, &bv);
|
||||||
|
if(!bv) return FALSE; // not readable
|
||||||
|
if(wr){
|
||||||
|
PYLONFN(GenApiNodeIsWritable, *phNode, &bv);
|
||||||
|
if(!bv) return FALSE; // not writeable
|
||||||
|
}
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
#pragma GCC diagnostic push
|
||||||
|
#pragma GCC diagnostic ignored "-Wunused-function"
|
||||||
|
|
||||||
|
// getters of different types of data
|
||||||
|
static int getBoolean(const char *featureName, _Bool *val){
|
||||||
|
if(!isopened || !featureName) return FALSE;
|
||||||
|
NODE_HANDLE hNode;
|
||||||
|
if(!chkNode(&hNode, featureName, BooleanNode, FALSE)) return FALSE;
|
||||||
|
if(!val) return TRUE;
|
||||||
|
PYLONFN(GenApiBooleanGetValue, hNode, val);
|
||||||
|
//DBG("Get boolean: %s = %s", featureName, val ? "true" : "false");
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
static int getInt(char *featureName, int64_values *val){
|
||||||
|
if(!isopened || !featureName) return FALSE;
|
||||||
|
NODE_HANDLE hNode;
|
||||||
|
if(!chkNode(&hNode, featureName, IntegerNode, FALSE)) return FALSE;
|
||||||
|
if(!val) return TRUE;
|
||||||
|
PYLONFN(GenApiIntegerGetMin, hNode, &val->min);
|
||||||
|
PYLONFN(GenApiIntegerGetMax, hNode, &val->max);
|
||||||
|
PYLONFN(GenApiIntegerGetInc, hNode, &val->incr);
|
||||||
|
PYLONFN(GenApiIntegerGetValue, hNode, &val->val);
|
||||||
|
//DBG("Get integer %s = %ld: min = %ld, max = %ld, incr = %ld", featureName, val->val, val->min, val->max, val->incr);
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
static int getFloat(char *featureName, float_values *val){
|
||||||
|
if(!isopened || !featureName) return FALSE;
|
||||||
|
NODE_HANDLE hNode;
|
||||||
|
if(!chkNode(&hNode, featureName, FloatNode, FALSE)) return FALSE;
|
||||||
|
if(!val) return TRUE;
|
||||||
|
PYLONFN(GenApiFloatGetMin, hNode, &val->min);
|
||||||
|
PYLONFN(GenApiFloatGetMax, hNode, &val->max);
|
||||||
|
PYLONFN(GenApiFloatGetValue, hNode, &val->val);
|
||||||
|
//DBG("Get float %s = %g: min = %g, max = %g", featureName, val->val, val->min, val->max);
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// setters of different types of data
|
||||||
|
static int setBoolean(char *featureName, _Bool val){
|
||||||
|
if(!isopened || !featureName) return FALSE;
|
||||||
|
NODE_HANDLE hNode;
|
||||||
|
if(!chkNode(&hNode, featureName, BooleanNode, TRUE)) return FALSE;
|
||||||
|
PYLONFN(GenApiBooleanSetValue, hNode, val);
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
static int setInt(char *featureName, int64_t val){
|
||||||
|
if(!isopened || !featureName) return FALSE;
|
||||||
|
NODE_HANDLE hNode;
|
||||||
|
if(!chkNode(&hNode, featureName, IntegerNode, TRUE)) return FALSE;
|
||||||
|
PYLONFN(GenApiIntegerSetValue, hNode, val);
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
static int setFloat(char *featureName, float val){
|
||||||
|
if(!isopened || !featureName) return FALSE;
|
||||||
|
NODE_HANDLE hNode;
|
||||||
|
if(!chkNode(&hNode, featureName, FloatNode, TRUE)) return FALSE;
|
||||||
|
PYLONFN(GenApiFloatSetValue, hNode, val);
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
#pragma GCC diagnostic pop
|
||||||
|
|
||||||
|
static void disableauto(){
|
||||||
|
if(!isopened) return;
|
||||||
|
const char *features[] = {"EnumEntry_TriggerSelector_AcquisitionStart",
|
||||||
|
"EnumEntry_TriggerSelector_FrameBurstStart",
|
||||||
|
"EnumEntry_TriggerSelector_FrameStart"};
|
||||||
|
const char *triggers[] = {"AcquisitionStart", "FrameBurstStart", "FrameStart"};
|
||||||
|
for(int i = 0; i < 3; ++i){
|
||||||
|
if(PylonDeviceFeatureIsAvailable(hDev, features[i])){
|
||||||
|
PylonDeviceFeatureFromString(hDev, "TriggerSelector", triggers[i]);
|
||||||
|
PylonDeviceFeatureFromString(hDev, "TriggerMode", "Off");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
PylonDeviceFeatureFromString(hDev, "GainAuto", "Off");
|
||||||
|
PylonDeviceFeatureFromString(hDev, "ExposureAuto", "Off");
|
||||||
|
PylonDeviceFeatureFromString(hDev, "ExposureMode", "Timed");
|
||||||
|
PylonDeviceFeatureFromString(hDev, "SequencerMode", "Off");
|
||||||
|
}
|
||||||
|
|
||||||
|
static void GENAPIC_CC removalCallbackFunction(_U_ PYLON_DEVICE_HANDLE hDevice){
|
||||||
|
disconnect();
|
||||||
|
}
|
||||||
|
|
||||||
|
static int connect(){
|
||||||
|
FNAME();
|
||||||
|
size_t numDevices;
|
||||||
|
disconnect();
|
||||||
|
PylonInitialize();
|
||||||
|
PYLONFN(PylonEnumerateDevices, &numDevices);
|
||||||
|
if(!numDevices){
|
||||||
|
WARNX("No cameras found");
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
PYLONFN(PylonCreateDeviceByIndex, 0, &hDev);
|
||||||
|
isopened = TRUE;
|
||||||
|
PYLONFN(PylonDeviceOpen, hDev, PYLONC_ACCESS_MODE_CONTROL | PYLONC_ACCESS_MODE_STREAM | PYLONC_ACCESS_MODE_EXCLUSIVE);
|
||||||
|
disableauto();
|
||||||
|
PYLONFN(PylonDeviceFeatureFromString, hDev, "PixelFormat", "Mono8");
|
||||||
|
PYLONFN(PylonDeviceFeatureFromString, hDev, "CameraOperationMode", "LongExposure");
|
||||||
|
PYLONFN(PylonDeviceFeatureFromString, hDev, "UserSetSelector", "HighGain"); // set high gain selector
|
||||||
|
PYLONFN(PylonDeviceFeatureFromString, hDev, "AcquisitionMode", "SingleFrame");
|
||||||
|
PYLONFN(PylonDeviceExecuteCommandFeature, hDev, "UserSetLoad"); // load high gain mode
|
||||||
|
PYLON_STREAMGRABBER_HANDLE hGrabber;
|
||||||
|
PYLONFN(PylonDeviceGetStreamGrabber, hDev, 0, &hGrabber);
|
||||||
|
PYLONFN(PylonStreamGrabberOpen, hGrabber);
|
||||||
|
// PYLON_WAITOBJECT_HANDLE hWaitStream;
|
||||||
|
// PYLONFN(PylonStreamGrabberGetWaitObject, hStreamGrabber, &hWaitStream);
|
||||||
|
PYLONFN(PylonStreamGrabberGetPayloadSize, hDev, hGrabber, &payloadsize);
|
||||||
|
PylonStreamGrabberClose(hGrabber);
|
||||||
|
PylonDeviceRegisterRemovalCallback(hDev, removalCallbackFunction, &hCb);
|
||||||
|
imgBuf = MALLOC(unsigned char, payloadsize);
|
||||||
|
//PYLONFN(PylonDeviceExecuteCommandFeature, hDev, "AcquisitionStart");
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static Image *capture(){
|
||||||
|
FNAME();
|
||||||
|
static int toohot = FALSE;
|
||||||
|
if(!isopened || !imgBuf) return NULL;
|
||||||
|
float_values f;
|
||||||
|
if(!getFloat("DeviceTemperature", &f)) WARNX("Can't get temperature");
|
||||||
|
else{
|
||||||
|
LOGDBG("Basler temperature: %.1f", f.val);
|
||||||
|
DBG("Temperature: %.1f", f.val);
|
||||||
|
if(f.val > 80.){
|
||||||
|
WARNX("Device too hot");
|
||||||
|
if(!toohot){
|
||||||
|
LOGWARN("Device too hot");
|
||||||
|
toohot = TRUE;
|
||||||
|
}
|
||||||
|
}else if(toohot && f.val < 75.){
|
||||||
|
LOGDBG("Device temperature is normal");
|
||||||
|
toohot = FALSE;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
PylonGrabResult_t grabResult;
|
||||||
|
_Bool bufferReady;
|
||||||
|
GENAPIC_RESULT res = PylonDeviceGrabSingleFrame(hDev, 0, imgBuf, payloadsize,
|
||||||
|
&grabResult, &bufferReady, 500 + (uint32_t)expostime);
|
||||||
|
if(res != GENAPI_E_OK || !bufferReady){
|
||||||
|
WARNX("res != GENAPI_E_OK || !bufferReady");
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
if(grabResult.Status != Grabbed){
|
||||||
|
WARNX("grabResult.Status != Grabbed");
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
Image *oIma = u8toImage(imgBuf, grabResult.SizeX, grabResult.SizeY, grabResult.SizeX + grabResult.PaddingX);
|
||||||
|
return oIma;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Basler have no "brightness" parameter
|
||||||
|
static int setbrightness(_U_ float b){
|
||||||
|
FNAME();
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int setexp(float e){
|
||||||
|
FNAME();
|
||||||
|
if(!isopened) return FALSE;
|
||||||
|
e *= 1000.;
|
||||||
|
if(!setFloat("ExposureTime", e)){
|
||||||
|
LOGWARN("Can't set expose time %g", e);
|
||||||
|
WARNX("Can't set expose time %g", e);
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
float_values f;
|
||||||
|
if(!getFloat("ExposureTime", &f)) return FALSE;
|
||||||
|
expostime = (float)f.val / 1000.;
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int setgain(_U_ float e){
|
||||||
|
FNAME();
|
||||||
|
if(!isopened) return FALSE;
|
||||||
|
if(!setFloat("Gain", e)){
|
||||||
|
LOGWARN("Can't set gain %g", e);
|
||||||
|
WARNX("Can't set gain %g", e);
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int changeformat(frameformat *fmt){
|
||||||
|
FNAME();
|
||||||
|
if(!isopened) return FALSE;
|
||||||
|
setInt("Width", fmt->w);
|
||||||
|
setInt("Height", fmt->h);
|
||||||
|
setInt("OffsetX", fmt->xoff);
|
||||||
|
setInt("OffsetY", fmt->yoff);
|
||||||
|
int64_values i;
|
||||||
|
if(getInt("Width", &i)) fmt->w = i.val;
|
||||||
|
if(getInt("Height", &i)) fmt->h = i.val;
|
||||||
|
if(getInt("OffsetX", &i)) fmt->xoff = i.val;
|
||||||
|
if(getInt("OffsetY", &i)) fmt->yoff = i.val;
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int geometrylimits(frameformat *max, frameformat *step){
|
||||||
|
FNAME();
|
||||||
|
if(!isopened || !max || !step) return FALSE;
|
||||||
|
int64_values i;
|
||||||
|
if(!getInt("Width", &i)) return FALSE;
|
||||||
|
max->w = i.max; step->w = i.incr;
|
||||||
|
if(!getInt("Height", &i)) return FALSE;
|
||||||
|
max->h = i.max; step->h = i.incr;
|
||||||
|
if(!getInt("OffsetX", &i)) return FALSE;
|
||||||
|
max->xoff = i.max; step->xoff = i.incr;
|
||||||
|
if(!getInt("OffsetY", &i)) return FALSE;
|
||||||
|
max->yoff = i.max; step->yoff = i.incr;
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static float gainmax(){
|
||||||
|
FNAME();
|
||||||
|
float_values v;
|
||||||
|
if(!getFloat("Gain", &v)) return 0.;
|
||||||
|
return (float)v.max;
|
||||||
|
}
|
||||||
|
|
||||||
|
// exported object
|
||||||
|
camera Basler = {
|
||||||
|
.disconnect = disconnect,
|
||||||
|
.connect = connect,
|
||||||
|
.capture = capture,
|
||||||
|
.setbrightness = setbrightness,
|
||||||
|
.setexp = setexp,
|
||||||
|
.setgain = setgain,
|
||||||
|
.setgeometry = changeformat,
|
||||||
|
.getgeomlimits = geometrylimits,
|
||||||
|
.getmaxgain = gainmax,
|
||||||
|
};
|
||||||
|
|
||||||
27
LocCorr_new/basler.h
Normal file
27
LocCorr_new/basler.h
Normal file
@ -0,0 +1,27 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
#pragma once
|
||||||
|
#ifndef BASLER_H__
|
||||||
|
#define BASLER_H__
|
||||||
|
|
||||||
|
#include "cameracapture.h" // `camera`
|
||||||
|
#define BASLER_CAPT_NAME "basler"
|
||||||
|
|
||||||
|
extern camera Basler;
|
||||||
|
|
||||||
|
#endif // BASLER_H__
|
||||||
500
LocCorr_new/binmorph.c
Normal file
500
LocCorr_new/binmorph.c
Normal file
@ -0,0 +1,500 @@
|
|||||||
|
/*
|
||||||
|
* binmorph.c - functions for morphological operations on binary image
|
||||||
|
*
|
||||||
|
* Copyright 2015 Edward V. Emelianoff <eddy@sao.ru>
|
||||||
|
*
|
||||||
|
* This program is free software; you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation; either version 2 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program; if not, write to the Free Software
|
||||||
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
||||||
|
* MA 02110-1301, USA.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <string.h> // memcpy
|
||||||
|
#include <math.h>
|
||||||
|
#include <sys/time.h>
|
||||||
|
|
||||||
|
#include "binmorph.h"
|
||||||
|
#include "debug.h"
|
||||||
|
#include "imagefile.h"
|
||||||
|
|
||||||
|
// global arrays for erosion/dilation masks
|
||||||
|
static uint8_t *ER = NULL, *DIL = NULL;
|
||||||
|
|
||||||
|
/*
|
||||||
|
* =================== AUXILIARY FUNCTIONS ===================>
|
||||||
|
*/
|
||||||
|
|
||||||
|
/**
|
||||||
|
* This function inits masks arrays for erosion and dilation
|
||||||
|
* You may call it yourself or it will be called when one of
|
||||||
|
* `erosion` or `dilation` functions will be ran first time
|
||||||
|
*/
|
||||||
|
static void morph_init(){
|
||||||
|
if(ER) return;
|
||||||
|
ER = MALLOC(uint8_t, 256);
|
||||||
|
DIL = MALLOC(uint8_t, 256);
|
||||||
|
OMP_FOR()
|
||||||
|
for(int i = 0; i < 256; i++){
|
||||||
|
ER[i] = i & ((i << 1) | 1) & ((i >> 1) | (0x80)); // don't forget that << and >> set borders to zero
|
||||||
|
DIL[i] = i | (i << 1) | (i >> 1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* =================== MORPHOLOGICAL OPERATIONS ===================>
|
||||||
|
*/
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Remove all non-4-connected pixels
|
||||||
|
* @param image (i) - input image
|
||||||
|
* @param W, H - size of binarized image (in pixels)
|
||||||
|
* @return allocated memory area with converted input image
|
||||||
|
*/
|
||||||
|
uint8_t *il_filter4(uint8_t *image, int W, int H){
|
||||||
|
//FNAME();
|
||||||
|
if(W < MINWIDTH || H < MINHEIGHT) return NULL;
|
||||||
|
uint8_t *ret = MALLOC(uint8_t, W*H);
|
||||||
|
int W0 = (W + 7) / 8; // width in bytes
|
||||||
|
int w = W0-1, h = H-1;
|
||||||
|
{
|
||||||
|
// top of image, y = 0
|
||||||
|
#define IM_UP
|
||||||
|
#include "fc_filter.h"
|
||||||
|
#undef IM_UP
|
||||||
|
}
|
||||||
|
{
|
||||||
|
// mid of image, y = 1..h-1
|
||||||
|
#include "fc_filter.h"
|
||||||
|
}
|
||||||
|
{
|
||||||
|
// image bottom, y = h
|
||||||
|
#define IM_DOWN
|
||||||
|
#include "fc_filter.h"
|
||||||
|
#undef IM_DOWN
|
||||||
|
}
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Remove all non-8-connected pixels (single points)
|
||||||
|
* @param image (i) - input image
|
||||||
|
* @param W, H - size of binarized image (in pixels)
|
||||||
|
* @return allocated memory area with converted input image
|
||||||
|
*/
|
||||||
|
uint8_t *il_filter8(uint8_t *image, int W, int H){
|
||||||
|
//FNAME();
|
||||||
|
if(W < MINWIDTH || H < MINHEIGHT) return NULL;
|
||||||
|
uint8_t *ret = MALLOC(uint8_t, W*H);
|
||||||
|
int W0 = (W + 7) / 8; // width in bytes
|
||||||
|
int w = W0-1, h = H-1;
|
||||||
|
{
|
||||||
|
#define IM_UP
|
||||||
|
#include "ec_filter.h"
|
||||||
|
#undef IM_UP
|
||||||
|
}
|
||||||
|
{
|
||||||
|
#include "ec_filter.h"
|
||||||
|
}
|
||||||
|
{
|
||||||
|
#define IM_DOWN
|
||||||
|
#include "ec_filter.h"
|
||||||
|
#undef IM_DOWN
|
||||||
|
}
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Make morphological operation of il_dilation
|
||||||
|
* @param image (i) - input image
|
||||||
|
* @param W, H - size of image (pixels)
|
||||||
|
* @return allocated memory area with il_dilation of input image
|
||||||
|
*/
|
||||||
|
uint8_t *il_dilation(uint8_t *image, int W, int H){
|
||||||
|
//FNAME();
|
||||||
|
if(W < MINWIDTH || H < MINHEIGHT) return NULL;
|
||||||
|
int W0 = (W + 7) / 8; // width in bytes
|
||||||
|
int w = W0-1, h = H-1, rest = 7 - (W - w*8);
|
||||||
|
uint8_t lastmask = ~(1<<rest);
|
||||||
|
if(!DIL) morph_init();
|
||||||
|
uint8_t *ret = MALLOC(uint8_t, W0*H);
|
||||||
|
{
|
||||||
|
// top of image, y = 0
|
||||||
|
#define IM_UP
|
||||||
|
#include "dilation.h"
|
||||||
|
#undef IM_UP
|
||||||
|
}
|
||||||
|
{
|
||||||
|
// mid of image, y = 1..h-1
|
||||||
|
#include "dilation.h"
|
||||||
|
}
|
||||||
|
{
|
||||||
|
// image bottom, y = h
|
||||||
|
#define IM_DOWN
|
||||||
|
#include "dilation.h"
|
||||||
|
#undef IM_DOWN
|
||||||
|
}
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Make morphological operation of il_erosion by cross 3x3 pixels
|
||||||
|
* @param image (i) - input image
|
||||||
|
* @param W, H - size of image (in pixels)
|
||||||
|
* @return allocated memory area with il_erosion of input image
|
||||||
|
*/
|
||||||
|
uint8_t *il_erosion(uint8_t *image, int W, int H){
|
||||||
|
//FNAME();
|
||||||
|
if(W < MINWIDTH || H < MINHEIGHT) return NULL;
|
||||||
|
if(!ER) morph_init();
|
||||||
|
int W0 = (W + 7) / 8; // width in bytes
|
||||||
|
int w = W0-1, h = H-1, rest = 8 - (W - w*8);
|
||||||
|
uint8_t lastmask = ~(1<<rest);
|
||||||
|
uint8_t *ret = MALLOC(uint8_t, W0*H);
|
||||||
|
//DBG("rest=%d, mask:0x%x", rest, lastmask);
|
||||||
|
OMP_FOR()
|
||||||
|
for(int y = 1; y < h; y++){ // reset first & last rows of image
|
||||||
|
uint8_t *iptr = &image[W0*y];
|
||||||
|
uint8_t *optr = &ret[W0*y];
|
||||||
|
register uint8_t p;
|
||||||
|
// x == 0, reset left bit
|
||||||
|
p = ER[*iptr] & *iptr & iptr[-W0] & iptr[W0];
|
||||||
|
if(!(*(++iptr) & 0x80)) p &= 0xfe;
|
||||||
|
*optr++ = p & 0x7f;
|
||||||
|
for(int x = 1; x < w; x++, iptr++, optr++){
|
||||||
|
p = ER[*iptr] & *iptr & iptr[-W0] & iptr[W0];
|
||||||
|
if(!(iptr[-1] & 1)) p &= 0x7f;
|
||||||
|
if(!(iptr[1] & 0x80)) p &= 0xfe;
|
||||||
|
*optr = p;
|
||||||
|
}
|
||||||
|
// x == w, reset right bit
|
||||||
|
p = ER[*iptr] & *iptr & iptr[-W0] & iptr[W0];
|
||||||
|
if(!(iptr[-1] & 1)) p &= 0x7f;
|
||||||
|
*optr = p & lastmask;
|
||||||
|
}
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Make il_erosion N times
|
||||||
|
uint8_t *il_erosionN(uint8_t *image, int W, int H, int N){
|
||||||
|
//FNAME();
|
||||||
|
if(W < 1 || H < 1) return NULL;
|
||||||
|
if(W < MINWIDTH || H < MINHEIGHT || N < 1){
|
||||||
|
uint8_t *copy = MALLOC(uint8_t, W*H);
|
||||||
|
memcpy(copy, image, W*H);
|
||||||
|
return copy;
|
||||||
|
}
|
||||||
|
uint8_t *cur = image, *next = NULL;
|
||||||
|
for(int i = 0; i < N; ++i){
|
||||||
|
next = il_erosion(cur, W, H);
|
||||||
|
if(cur != image) FREE(cur);
|
||||||
|
cur = next;
|
||||||
|
}
|
||||||
|
return next;
|
||||||
|
}
|
||||||
|
// Make il_dilation N times
|
||||||
|
uint8_t *il_dilationN(uint8_t *image, int W, int H, int N){
|
||||||
|
//FNAME();
|
||||||
|
if(W < 1 || H < 1) return NULL;
|
||||||
|
if(W < MINWIDTH || H < MINHEIGHT || N < 1){
|
||||||
|
uint8_t *copy = MALLOC(uint8_t, W*H);
|
||||||
|
memcpy(copy, image, W*H);
|
||||||
|
return copy;
|
||||||
|
}
|
||||||
|
uint8_t *cur = image, *next = NULL;
|
||||||
|
for(int i = 0; i < N; ++i){
|
||||||
|
next = il_dilation(cur, W, H);
|
||||||
|
if(cur != image) FREE(cur);
|
||||||
|
cur = next;
|
||||||
|
}
|
||||||
|
return next;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Ntimes opening
|
||||||
|
uint8_t *il_openingN(uint8_t *image, int W, int H, int N){
|
||||||
|
//FNAME();
|
||||||
|
if(W < MINWIDTH || H < MINHEIGHT || N < 1) return NULL;
|
||||||
|
uint8_t *er = il_erosionN(image, W, H, N);
|
||||||
|
uint8_t *op = il_dilationN(er, W, H, N);
|
||||||
|
FREE(er);
|
||||||
|
return op;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Ntimes closing
|
||||||
|
uint8_t *il_closingN(uint8_t *image, int W, int H, int N){
|
||||||
|
//FNAME();
|
||||||
|
if(W < MINWIDTH || H < MINHEIGHT || N < 1) return NULL;
|
||||||
|
uint8_t *di = il_dilationN(image, W, H, N);
|
||||||
|
uint8_t *cl = il_erosionN(di, W, H, N);
|
||||||
|
FREE(di);
|
||||||
|
return cl;
|
||||||
|
}
|
||||||
|
|
||||||
|
// top hat operation: image - opening(image)
|
||||||
|
uint8_t *il_topHat(uint8_t *image, int W, int H, int N){
|
||||||
|
//FNAME();
|
||||||
|
if(W < MINWIDTH || H < MINHEIGHT || N < 1) return NULL;
|
||||||
|
uint8_t *op = il_openingN(image, W, H, N);
|
||||||
|
int W0 = (W + 7) / 8; // width in bytes
|
||||||
|
int wh = W0 * H;
|
||||||
|
OMP_FOR()
|
||||||
|
for(int i = 0; i < wh; ++i)
|
||||||
|
op[i] = image[i] & (~op[i]);
|
||||||
|
return op;
|
||||||
|
}
|
||||||
|
|
||||||
|
// bottom hat operation: closing(image) - image
|
||||||
|
uint8_t *il_botHat(uint8_t *image, int W, int H, int N){
|
||||||
|
//FNAME();
|
||||||
|
if(W < MINWIDTH || H < MINHEIGHT || N < 1) return NULL;
|
||||||
|
uint8_t *op = il_closingN(image, W, H, N);
|
||||||
|
int W0 = (W + 7) / 8; // width in bytes
|
||||||
|
int wh = W0 * H;
|
||||||
|
OMP_FOR()
|
||||||
|
for(int i = 0; i < wh; ++i)
|
||||||
|
op[i] &= ~image[i];
|
||||||
|
return op;
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* <=================== MORPHOLOGICAL OPERATIONS ===================
|
||||||
|
*/
|
||||||
|
|
||||||
|
/*
|
||||||
|
* =================== LOGICAL OPERATIONS ===================>
|
||||||
|
*/
|
||||||
|
#if 0
|
||||||
|
/**
|
||||||
|
* Logical AND of two images
|
||||||
|
* @param im1, im2 (i) - two images
|
||||||
|
* @param W, H - their size (of course, equal for both images)
|
||||||
|
* @return allocated memory area with image = (im1 AND im2)
|
||||||
|
*/
|
||||||
|
uint8_t *imand(uint8_t *im1, uint8_t *im2, int W, int H){
|
||||||
|
uint8_t *ret = MALLOC(uint8_t, W*H);
|
||||||
|
int y;
|
||||||
|
OMP_FOR()
|
||||||
|
for(y = 0; y < H; y++){
|
||||||
|
int x, S = y*W;
|
||||||
|
uint8_t *rptr = &ret[S], *p1 = &im1[S], *p2 = &im2[S];
|
||||||
|
for(x = 0; x < W; x++)
|
||||||
|
*rptr++ = *p1++ & *p2++;
|
||||||
|
}
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Substitute image 2 from image 1: reset to zero all bits of image 1 which set to 1 on image 2
|
||||||
|
* @param im1, im2 (i) - two images
|
||||||
|
* @param W, H - their size (of course, equal for both images)
|
||||||
|
* @return allocated memory area with image = (im1 AND (!im2))
|
||||||
|
*/
|
||||||
|
uint8_t *substim(uint8_t *im1, uint8_t *im2, int W, int H){
|
||||||
|
uint8_t *ret = MALLOC(uint8_t, W*H);
|
||||||
|
int y;
|
||||||
|
OMP_FOR()
|
||||||
|
for(y = 0; y < H; y++){
|
||||||
|
int x, S = y*W;
|
||||||
|
uint8_t *rptr = &ret[S], *p1 = &im1[S], *p2 = &im2[S];
|
||||||
|
for(x = 0; x < W; x++)
|
||||||
|
*rptr++ = *p1++ & (~*p2++);
|
||||||
|
}
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
/*
|
||||||
|
* <=================== LOGICAL OPERATIONS ===================
|
||||||
|
*/
|
||||||
|
|
||||||
|
/*
|
||||||
|
* =================== CONNECTED COMPONENTS LABELING ===================>
|
||||||
|
*/
|
||||||
|
|
||||||
|
//#define TESTMSGS
|
||||||
|
#ifdef TESTMSGS
|
||||||
|
#define TEST(...) printf(__VA_ARGS__)
|
||||||
|
#else
|
||||||
|
#define TEST(...)
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// check table and rename all "oldval" into "newval"
|
||||||
|
static inline void remark(size_t newval, size_t oldval, size_t *assoc){
|
||||||
|
TEST("\tnew = %zd, old=%zd; ", newval, oldval);
|
||||||
|
// find the least values
|
||||||
|
do{newval = assoc[newval];}while(assoc[newval] != newval);
|
||||||
|
do{oldval = assoc[oldval];}while(assoc[oldval] != oldval);
|
||||||
|
TEST("\trealnew = %zd, realold=%zd ", newval, oldval);
|
||||||
|
// now change larger value to smaller
|
||||||
|
if(newval > oldval){
|
||||||
|
assoc[newval] = oldval;
|
||||||
|
TEST("change %zd to %zd\n", newval, oldval);
|
||||||
|
}else{
|
||||||
|
assoc[oldval] = newval;
|
||||||
|
TEST("change %zd to %zd\n", oldval, newval);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* label 4-connected components on image
|
||||||
|
* (slow algorythm, but easy to parallel)
|
||||||
|
*
|
||||||
|
* @param I (i) - image ("packed")
|
||||||
|
* @param W,H - size of the image (W - width in pixels)
|
||||||
|
* @param CC (o) - connected components boxes
|
||||||
|
* @return an array of labeled components
|
||||||
|
*/
|
||||||
|
size_t *il_cclabel4(uint8_t *Img, int W, int H, il_ConnComps **CC){
|
||||||
|
size_t *assoc;
|
||||||
|
if(W < MINWIDTH || H < MINHEIGHT) return NULL;
|
||||||
|
uint8_t *f = il_filter4(Img, W, H); // remove all non 4-connected pixels
|
||||||
|
//DBG("convert to size_t");
|
||||||
|
size_t *labels = bin2ST(f, W, H);
|
||||||
|
FREE(f);
|
||||||
|
//DBG("Calculate");
|
||||||
|
size_t Nmax = W*H/4; // max number of 4-connected labels
|
||||||
|
assoc = MALLOC(size_t, Nmax); // allocate memory for "remark" array
|
||||||
|
size_t last_assoc_idx = 1; // last index filled in assoc array
|
||||||
|
for(int y = 0; y < H; ++y){
|
||||||
|
bool found = false;
|
||||||
|
size_t *ptr = &labels[y*W];
|
||||||
|
size_t curmark = 0; // mark of pixel to the left
|
||||||
|
for(int x = 0; x < W; ++x, ++ptr){
|
||||||
|
if(!*ptr){found = false; continue;} // empty pixel
|
||||||
|
size_t U = (y) ? ptr[-W] : 0; // upper mark
|
||||||
|
if(found){ // there's a pixel to the left
|
||||||
|
if(U && U != curmark){ // meet old mark -> remark one of them in assoc[]
|
||||||
|
TEST("(%d, %d): remark %zd --> %zd\n", x, y, U, curmark);
|
||||||
|
remark(U, curmark, assoc);
|
||||||
|
curmark = U; // change curmark to upper mark (to reduce further checks)
|
||||||
|
}
|
||||||
|
}else{ // new mark -> change curmark
|
||||||
|
found = true;
|
||||||
|
if(U) curmark = U; // current mark will copy upper value
|
||||||
|
else{ // current mark is new value
|
||||||
|
curmark = last_assoc_idx++;
|
||||||
|
assoc[curmark] = curmark;
|
||||||
|
TEST("(%d, %d): new mark=%zd\n", x, y, curmark);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
*ptr = curmark;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
size_t *indexes = MALLOC(size_t, last_assoc_idx); // new indexes
|
||||||
|
size_t cidx = 1;
|
||||||
|
TEST("\n\n\nRebuild indexes\n\n");
|
||||||
|
for(size_t i = 1; i < last_assoc_idx; ++i){
|
||||||
|
TEST("%zd\t%zd ",i,assoc[i]);
|
||||||
|
// search new index
|
||||||
|
register size_t realnew = i, newval = 0;
|
||||||
|
do{
|
||||||
|
realnew = assoc[realnew];
|
||||||
|
TEST("->%zd ", realnew);
|
||||||
|
if(indexes[realnew]){ // find least index
|
||||||
|
newval = indexes[realnew];
|
||||||
|
TEST("real: %zd ", newval);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}while(assoc[realnew] != realnew);
|
||||||
|
if(newval){
|
||||||
|
TEST(" ==> %zd\n", newval);
|
||||||
|
indexes[i] = newval;
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
TEST("new index %zd\n", cidx);
|
||||||
|
// enter next label
|
||||||
|
indexes[i] = cidx++;
|
||||||
|
}
|
||||||
|
// cidx now is amount of detected objects + 1 - size of output array (0th idx is not used)
|
||||||
|
//DBG("amount after rebuild: %zd", cidx-1);
|
||||||
|
#ifdef TESTMSGS
|
||||||
|
printf("\n\n\nI\tASS[I]\tIDX[I]\n");
|
||||||
|
for(size_t i = 1; i < last_assoc_idx; ++i)
|
||||||
|
printf("%zd\t%zd\t%zd\n",i,assoc[i],indexes[i]);
|
||||||
|
#endif
|
||||||
|
il_Box *boxes = MALLOC(il_Box, cidx);
|
||||||
|
OMP_FOR()
|
||||||
|
for(size_t i = 1; i < cidx; ++i){ // init borders
|
||||||
|
boxes[i].xmin = W;
|
||||||
|
boxes[i].ymin = H;
|
||||||
|
}
|
||||||
|
#pragma omp parallel shared(boxes)
|
||||||
|
{
|
||||||
|
il_Box *l_boxes = MALLOC(il_Box, cidx);
|
||||||
|
for(size_t i = 1; i < cidx; ++i){ // init borders
|
||||||
|
l_boxes[i].xmin = W;
|
||||||
|
l_boxes[i].ymin = H;
|
||||||
|
}
|
||||||
|
#pragma omp for nowait
|
||||||
|
for(int y = 0; y < H; ++y){
|
||||||
|
size_t *lptr = &labels[y*W];
|
||||||
|
for(int x = 0; x < W; ++x, ++lptr){
|
||||||
|
if(!*lptr) continue;
|
||||||
|
register size_t mark = indexes[*lptr];
|
||||||
|
*lptr = mark;
|
||||||
|
il_Box *b = &l_boxes[mark];
|
||||||
|
++b->area;
|
||||||
|
if(b->xmax < x) b->xmax = x;
|
||||||
|
if(b->xmin > x) b->xmin = x;
|
||||||
|
if(b->ymax < y) b->ymax = y;
|
||||||
|
if(b->ymin > y) b->ymin = y;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#pragma omp critical
|
||||||
|
for(size_t i = 1; i < cidx; ++i){
|
||||||
|
il_Box *ob = &boxes[i], *ib = &l_boxes[i];
|
||||||
|
if(ob->xmax < ib->xmax) ob->xmax = ib->xmax;
|
||||||
|
if(ob->xmin > ib->xmin) ob->xmin = ib->xmin;
|
||||||
|
if(ob->ymax < ib->ymax) ob->ymax = ib->ymax;
|
||||||
|
if(ob->ymin > ib->ymin) ob->ymin = ib->ymin;
|
||||||
|
ob->area += ib->area;
|
||||||
|
}
|
||||||
|
FREE(l_boxes);
|
||||||
|
}
|
||||||
|
FREE(assoc);
|
||||||
|
FREE(indexes);
|
||||||
|
#ifdef TESTMSGS
|
||||||
|
for(size_t i = 1; i < cidx; ++i){
|
||||||
|
printf("%8zd\t%6d\t(%4d..%4d, %4d..%4d)\t%.2f\n", i, boxes[i].area,
|
||||||
|
boxes[i].xmin, boxes[i].xmax, boxes[i].ymin, boxes[i].ymax,
|
||||||
|
(1.+boxes[i].xmax-boxes[i].xmin)/(1.+boxes[i].ymax-boxes[i].ymin));
|
||||||
|
}printf("\n\n");
|
||||||
|
#endif
|
||||||
|
if(CC){
|
||||||
|
*CC = MALLOC(il_ConnComps, 1);
|
||||||
|
(*CC)->Nobj = cidx; (*CC)->boxes = boxes;
|
||||||
|
}else{
|
||||||
|
FREE(boxes);
|
||||||
|
}
|
||||||
|
return labels;
|
||||||
|
}
|
||||||
|
|
||||||
|
#if 0
|
||||||
|
// label 8-connected components, look cclabel4
|
||||||
|
size_t *cclabel8(size_t *labels, int W, int H, size_t *Nobj){
|
||||||
|
if(W < MINWIDTH || H < MINHEIGHT) return NULL;
|
||||||
|
#define LABEL_8
|
||||||
|
#include "cclabling.h"
|
||||||
|
#undef LABEL_8
|
||||||
|
return labels;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
/*
|
||||||
|
* <=================== CONNECTED COMPONENTS LABELING ===================
|
||||||
|
*/
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
* <=================== template ===================>
|
||||||
|
*/
|
||||||
64
LocCorr_new/binmorph.h
Normal file
64
LocCorr_new/binmorph.h
Normal file
@ -0,0 +1,64 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
#ifndef BINMORPH_H__
|
||||||
|
#define BINMORPH_H__
|
||||||
|
|
||||||
|
#include <stdbool.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <stdint.h>
|
||||||
|
#include "fits.h"
|
||||||
|
|
||||||
|
// minimal image size for morph operations
|
||||||
|
#define MINWIDTH (9)
|
||||||
|
#define MINHEIGHT (3)
|
||||||
|
|
||||||
|
// simple box with given borders
|
||||||
|
typedef struct{
|
||||||
|
uint16_t xmin;
|
||||||
|
uint16_t xmax;
|
||||||
|
uint16_t ymin;
|
||||||
|
uint16_t ymax;
|
||||||
|
uint32_t area; // total amount of object pixels inside the box
|
||||||
|
} il_Box;
|
||||||
|
|
||||||
|
typedef struct{
|
||||||
|
size_t Nobj;
|
||||||
|
il_Box *boxes;
|
||||||
|
} il_ConnComps;
|
||||||
|
|
||||||
|
|
||||||
|
// morphological operations:
|
||||||
|
uint8_t *il_dilation(uint8_t *image, int W, int H);
|
||||||
|
uint8_t *il_dilationN(uint8_t *image, int W, int H, int N);
|
||||||
|
uint8_t *il_erosion(uint8_t *image, int W, int H);
|
||||||
|
uint8_t *il_erosionN(uint8_t *image, int W, int H, int N);
|
||||||
|
uint8_t *il_openingN(uint8_t *image, int W, int H, int N);
|
||||||
|
uint8_t *il_closingN(uint8_t *image, int W, int H, int N);
|
||||||
|
uint8_t *il_topHat(uint8_t *image, int W, int H, int N);
|
||||||
|
uint8_t *il_botHat(uint8_t *image, int W, int H, int N);
|
||||||
|
|
||||||
|
// clear non 4-connected pixels
|
||||||
|
uint8_t *il_filter4(uint8_t *image, int W, int H);
|
||||||
|
// clear single pixels
|
||||||
|
uint8_t *il_filter8(uint8_t *image, int W, int H);
|
||||||
|
|
||||||
|
size_t *il_cclabel4(uint8_t *Img, int W, int H, il_ConnComps **CC);
|
||||||
|
|
||||||
|
#endif // BINMORPH_H__
|
||||||
275
LocCorr_new/cameracapture.c
Normal file
275
LocCorr_new/cameracapture.c
Normal file
@ -0,0 +1,275 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <float.h> // FLT_EPSILON
|
||||||
|
#include <math.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <string.h>
|
||||||
|
|
||||||
|
#include "cmdlnopts.h"
|
||||||
|
#include "config.h"
|
||||||
|
#include "debug.h"
|
||||||
|
#include "fits.h"
|
||||||
|
#include "grasshopper.h"
|
||||||
|
#include "imagefile.h"
|
||||||
|
#include "improc.h"
|
||||||
|
#include "median.h"
|
||||||
|
|
||||||
|
// pointer to selected camera
|
||||||
|
static camera *theCam = NULL;
|
||||||
|
|
||||||
|
static float gain = 0., gainmax = 0.;
|
||||||
|
static float exptime = 100.;
|
||||||
|
static float brightness = 0.;
|
||||||
|
static int connected = FALSE;
|
||||||
|
|
||||||
|
static frameformat curformat;
|
||||||
|
static frameformat maxformat;
|
||||||
|
static frameformat stepformat;
|
||||||
|
|
||||||
|
static void changeformat(){
|
||||||
|
if(!theCam) return;
|
||||||
|
if(maxformat.h < 1 || maxformat.w < 1){
|
||||||
|
WARNX("Bad max format data");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if(stepformat.h < 1 || stepformat.w < 1){
|
||||||
|
WARNX("Bad step format data");
|
||||||
|
LOGWARN("Bad step format data");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if(stepformat.xoff < 1) stepformat.xoff = 1;
|
||||||
|
if(stepformat.yoff < 1) stepformat.yoff = 1;
|
||||||
|
curformat.h = (theconf.height < maxformat.h) ? theconf.height : maxformat.h;
|
||||||
|
curformat.h -= curformat.h % stepformat.h;
|
||||||
|
curformat.w = (theconf.width < maxformat.w) ? theconf.width : maxformat.w;
|
||||||
|
curformat.w -= curformat.w % stepformat.w;
|
||||||
|
curformat.xoff = (theconf.xoff + curformat.w <= maxformat.w) ? theconf.xoff : maxformat.w - curformat.w;
|
||||||
|
curformat.xoff -= curformat.xoff % stepformat.xoff;
|
||||||
|
curformat.yoff = (theconf.yoff + curformat.h <= maxformat.h) ? theconf.yoff : maxformat.h - curformat.h;
|
||||||
|
curformat.yoff -= curformat.yoff % stepformat.yoff;
|
||||||
|
if(theCam->setgeometry(&curformat)){ // now we can change config values to real
|
||||||
|
theconf.height = curformat.h;
|
||||||
|
theconf.width = curformat.w;
|
||||||
|
theconf.xoff = curformat.xoff;
|
||||||
|
theconf.yoff = curformat.yoff;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief setCamera - set active camera & initialize it
|
||||||
|
* @param cptr - pointer to new camera
|
||||||
|
* @return FALSE if failed
|
||||||
|
*/
|
||||||
|
int setCamera(camera *cptr){
|
||||||
|
camdisconnect();
|
||||||
|
theCam = cptr;
|
||||||
|
connected = theCam->connect();
|
||||||
|
if(!connected) return FALSE;
|
||||||
|
gainmax = theCam->getmaxgain();
|
||||||
|
gain = theconf.gain;
|
||||||
|
brightness = theconf.brightness;
|
||||||
|
if(!theCam->getgeomlimits(&maxformat, &stepformat)){
|
||||||
|
WARNX("Can't detect camera format limits");
|
||||||
|
LOGWARN("Can't detect camera format limits");
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
changeformat();
|
||||||
|
LOGMSG("Camera connected, max gain: %.1f, max (W,H): (%d,%d)", gainmax, maxformat.w, maxformat.h);
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
void camdisconnect(){
|
||||||
|
if(!connected) return;
|
||||||
|
connected = FALSE;
|
||||||
|
if(theCam) theCam->disconnect();
|
||||||
|
}
|
||||||
|
|
||||||
|
static void calcexpgain(float newexp){
|
||||||
|
DBG("recalculate exposition: oldexp=%g, oldgain=%g, newexp=%g", exptime, gain, newexp);
|
||||||
|
while(newexp*1.25 > theconf.minexp){ // increase gain first
|
||||||
|
if(gain < gainmax - 0.9999){
|
||||||
|
gain += 1.;
|
||||||
|
newexp /= 1.25;
|
||||||
|
}else break;
|
||||||
|
}
|
||||||
|
while(newexp < theconf.minexp){
|
||||||
|
if(1.25*newexp < theconf.maxexp && gain > 0.9999){
|
||||||
|
gain -= 1.;
|
||||||
|
newexp *= 1.25;
|
||||||
|
}else break;
|
||||||
|
}
|
||||||
|
if(newexp < theconf.minexp) newexp = theconf.minexp;
|
||||||
|
else if(newexp > theconf.maxexp) newexp = theconf.maxexp;
|
||||||
|
exptime = newexp;
|
||||||
|
DBG("New values: exp=%g, gain=%g", exptime, gain);
|
||||||
|
}
|
||||||
|
|
||||||
|
//convertedImage.pData, convertedImage.cols, convertedImage.rows, convertedImage.stride
|
||||||
|
static void recalcexp(Image *I){
|
||||||
|
// check if user changed exposition values
|
||||||
|
if(exptime < theconf.minexp){
|
||||||
|
exptime = theconf.minexp;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
else if(exptime > theconf.maxexp){
|
||||||
|
exptime = theconf.maxexp;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
size_t *histogram = get_histogram(I);
|
||||||
|
if(!histogram){
|
||||||
|
WARNX("Can't calculate histogram");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
int idx100;
|
||||||
|
size_t sum100 = 0;
|
||||||
|
for(idx100 = 255; idx100 >= 0; --idx100){
|
||||||
|
sum100 += histogram[idx100];
|
||||||
|
if(sum100 > 100) break;
|
||||||
|
}
|
||||||
|
FREE(histogram);
|
||||||
|
DBG("Sum100=%zd, idx100=%d", sum100, idx100);
|
||||||
|
if(idx100 > 230 && idx100 < 253) return; // good values
|
||||||
|
if(idx100 > 253){ // exposure too long
|
||||||
|
calcexpgain(0.7*exptime);
|
||||||
|
}else{ // exposure too short
|
||||||
|
if(idx100 > 5)
|
||||||
|
calcexpgain(exptime * 230. / (float)idx100);
|
||||||
|
else
|
||||||
|
calcexpgain(exptime * 50.);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
int camcapture(void (*process)(Image*)){
|
||||||
|
FNAME();
|
||||||
|
static float oldexptime = 0.;
|
||||||
|
static float oldgain = -1.;
|
||||||
|
static float oldbrightness = -1.;
|
||||||
|
Image *oIma = NULL;
|
||||||
|
while(1){
|
||||||
|
if(stopwork){
|
||||||
|
DBG("STOP");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
if(!theCam){ // wait until camera be powered on
|
||||||
|
LOGERR("camcapture(): camera not initialized");
|
||||||
|
ERRX("Not initialized");
|
||||||
|
}
|
||||||
|
if(!connected){
|
||||||
|
DBG("Disconnected");
|
||||||
|
connected = theCam->connect();
|
||||||
|
sleep(1);
|
||||||
|
changeformat();
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
if(fabsf(oldbrightness - brightness) > FLT_EPSILON){ // new brightness
|
||||||
|
DBG("Change brightness to %g", brightness);
|
||||||
|
if(theCam->setbrightness(brightness)){
|
||||||
|
oldbrightness = brightness;
|
||||||
|
}else{
|
||||||
|
WARNX("Can't change brightness to %g", brightness);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if(exptime > theconf.maxexp) exptime = theconf.maxexp;
|
||||||
|
else if(exptime < theconf.minexp) exptime = theconf.minexp;
|
||||||
|
if(fabsf(oldexptime - exptime) > FLT_EPSILON){ // new exsposition value
|
||||||
|
DBG("Change exptime to %.2fms\n", exptime);
|
||||||
|
if(theCam->setexp(exptime)){
|
||||||
|
oldexptime = exptime;
|
||||||
|
}else{
|
||||||
|
WARNX("Can't change exposition time to %gms", exptime);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if(gain > gainmax) gain = gainmax;
|
||||||
|
if(fabsf(oldgain - gain) > FLT_EPSILON){ // change gain
|
||||||
|
DBG("Change gain to %g\n", gain);
|
||||||
|
if(theCam->setgain(gain)){
|
||||||
|
oldgain = gain;
|
||||||
|
}else{
|
||||||
|
WARNX("Can't change gain to %g", gain);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// change format
|
||||||
|
if(abs(curformat.h - theconf.height) || abs(curformat.w - theconf.width) || abs(curformat.xoff - theconf.xoff) || abs(curformat.yoff - theconf.yoff)){
|
||||||
|
changeformat();
|
||||||
|
}
|
||||||
|
if(!(oIma = theCam->capture())){
|
||||||
|
WARNX("Can't grab image");
|
||||||
|
camdisconnect();
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
if(theconf.expmethod == EXPAUTO) recalcexp(oIma);
|
||||||
|
else{
|
||||||
|
if(fabs(theconf.fixedexp - exptime) > FLT_EPSILON)
|
||||||
|
exptime = theconf.fixedexp;
|
||||||
|
if(fabs(theconf.gain - gain) > FLT_EPSILON)
|
||||||
|
gain = theconf.gain;
|
||||||
|
if(fabs(theconf.brightness - brightness) > FLT_EPSILON)
|
||||||
|
brightness = theconf.brightness;
|
||||||
|
}
|
||||||
|
if(process){
|
||||||
|
if(theconf.medfilt){
|
||||||
|
Image *X = get_median(oIma, theconf.medseed);
|
||||||
|
if(X){
|
||||||
|
FREE(oIma->data);
|
||||||
|
FREE(oIma);
|
||||||
|
oIma = X;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
process(oIma);
|
||||||
|
}
|
||||||
|
FREE(oIma->data);
|
||||||
|
FREE(oIma);
|
||||||
|
}
|
||||||
|
if(oIma){
|
||||||
|
FREE(oIma->data);
|
||||||
|
FREE(oIma);
|
||||||
|
}
|
||||||
|
camdisconnect();
|
||||||
|
DBG("CAMCAPTURE: out");
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief camstatus - return JSON with image status
|
||||||
|
* @param messageid - value of "messageid"
|
||||||
|
* @param buf - buffer for string
|
||||||
|
* @param buflen - length of `buf`
|
||||||
|
* @return buf
|
||||||
|
*/
|
||||||
|
char *camstatus(const char *messageid, char *buf, int buflen){
|
||||||
|
if(!buf || buflen < 2) return NULL;
|
||||||
|
if(!messageid) messageid = "unknown";
|
||||||
|
static char *impath = NULL;
|
||||||
|
if(!impath){
|
||||||
|
if(!(impath = realpath(GP->outputjpg, impath))){
|
||||||
|
WARN("realpath() (%s)", impath);
|
||||||
|
impath = strdup(GP->outputjpg);
|
||||||
|
}
|
||||||
|
DBG("path: %s", impath);
|
||||||
|
}
|
||||||
|
float xc, yc;
|
||||||
|
getcenter(&xc, &yc);
|
||||||
|
snprintf(buf, buflen, "{ \"%s\": \"%s\", \"camstatus\": \"%sconnected\", \"impath\": \"%s\", \"imctr\": %llu, "
|
||||||
|
"\"fps\": %.3f, \"expmethod\": \"%s\", \"exposition\": %g, \"gain\": %g, \"brightness\": %g, "
|
||||||
|
"\"xcenter\": %.1f, \"ycenter\": %.1f }\n",
|
||||||
|
MESSAGEID, messageid, connected ? "" : "dis", impath, ImNumber, getFramesPerS(),
|
||||||
|
(theconf.expmethod == EXPAUTO) ? "auto" : "manual", exptime, gain, brightness,
|
||||||
|
xc, yc);
|
||||||
|
return buf;
|
||||||
|
}
|
||||||
53
LocCorr_new/cameracapture.h
Normal file
53
LocCorr_new/cameracapture.h
Normal file
@ -0,0 +1,53 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
#pragma once
|
||||||
|
#ifndef CAMERACAPTURE_H__
|
||||||
|
#define CAMERACAPTURE_H__
|
||||||
|
|
||||||
|
#include "imagefile.h" // Image
|
||||||
|
|
||||||
|
|
||||||
|
// format of single frame
|
||||||
|
typedef struct{
|
||||||
|
int w; int h; // width & height
|
||||||
|
int xoff; int yoff; // X and Y offset
|
||||||
|
} frameformat;
|
||||||
|
|
||||||
|
typedef struct{
|
||||||
|
void (*disconnect)(); // disconnect & cleanup
|
||||||
|
int (*connect)(); // connect & init
|
||||||
|
Image* (*capture)(); // capture an image
|
||||||
|
// setters: brightness, exptime, gain
|
||||||
|
int (*setbrightness)(float b);
|
||||||
|
int (*setexp)(float e);
|
||||||
|
int (*setgain)(float g);
|
||||||
|
float (*getmaxgain)(); // get max available gain value
|
||||||
|
// geometry (if TRUE, all args are changed to suitable values)
|
||||||
|
int (*setgeometry)(frameformat *fmt);
|
||||||
|
// get limits of geometry: maximal values and steps
|
||||||
|
int (*getgeomlimits)(frameformat *max, frameformat *step);
|
||||||
|
//int (*getgeometry)(frameformat *fmt);
|
||||||
|
} camera;
|
||||||
|
|
||||||
|
int setCamera(camera *cptr);
|
||||||
|
void camdisconnect();
|
||||||
|
int camcapture(void (*process)(Image *));
|
||||||
|
char *camstatus(const char *messageid, char *buf, int buflen);
|
||||||
|
|
||||||
|
|
||||||
|
#endif // CAMERACAPTURE_H__
|
||||||
136
LocCorr_new/cclabling.h
Normal file
136
LocCorr_new/cclabling.h
Normal file
@ -0,0 +1,136 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
//#define TESTMSGS
|
||||||
|
|
||||||
|
#ifdef TESTMSGS
|
||||||
|
#define TEST(...) printf(__VA_ARGS__)
|
||||||
|
#else
|
||||||
|
#define TEST(...)
|
||||||
|
#endif
|
||||||
|
|
||||||
|
size_t Nmax = W*H/4; // max number of 4-connected labels
|
||||||
|
size_t *assoc = MALLOC(size_t, Nmax); // allocate memory for "remark" array
|
||||||
|
size_t last_assoc_idx = 1; // last index filled in assoc array
|
||||||
|
// was: 35
|
||||||
|
// check table and rename all "oldval" into "newval"
|
||||||
|
inline void remark(size_t newval, size_t oldval){
|
||||||
|
TEST("\tnew = %zd, old=%zd; ", newval, oldval);
|
||||||
|
// find the least values
|
||||||
|
do{newval = assoc[newval];}while(assoc[newval] != newval);
|
||||||
|
do{oldval = assoc[oldval];}while(assoc[oldval] != oldval);
|
||||||
|
TEST("\trealnew = %zd, realold=%zd ", newval, oldval);
|
||||||
|
// now change larger value to smaller
|
||||||
|
if(newval > oldval){
|
||||||
|
assoc[newval] = oldval;
|
||||||
|
TEST("change %zd to %zd\n", newval, oldval);
|
||||||
|
}else{
|
||||||
|
assoc[oldval] = newval;
|
||||||
|
TEST("change %zd to %zd\n", oldval, newval);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for(int y = 0; y < H; ++y){
|
||||||
|
bool found = false;
|
||||||
|
size_t *ptr = &labels[y*W];
|
||||||
|
size_t curmark = 0; // mark of pixel to the left
|
||||||
|
for(int x = 0; x < W; ++x, ++ptr){
|
||||||
|
if(!*ptr){found = false; continue;} // empty pixel
|
||||||
|
size_t U = (y) ? ptr[-W] : 0; // upper mark
|
||||||
|
if(found){ // there's a pixel to the left
|
||||||
|
if(U && U != curmark){ // meet old mark -> remark one of them in assoc[]
|
||||||
|
TEST("(%d, %d): remark %zd --> %zd\n", x, y, U, curmark);
|
||||||
|
remark(U, curmark);
|
||||||
|
curmark = U; // change curmark to upper mark (to reduce further checks)
|
||||||
|
}
|
||||||
|
}else{ // new mark -> change curmark
|
||||||
|
found = true;
|
||||||
|
if(U) curmark = U; // current mark will copy upper value
|
||||||
|
else{ // current mark is new value
|
||||||
|
curmark = last_assoc_idx++;
|
||||||
|
assoc[curmark] = curmark;
|
||||||
|
TEST("(%d, %d): new mark=%zd\n", x, y, curmark);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
*ptr = curmark;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
size_t *indexes = MALLOC(size_t, last_assoc_idx); // new indexes
|
||||||
|
size_t cidx = 1;
|
||||||
|
TEST("\n\n\nRebuild indexes\n\n");
|
||||||
|
for(size_t i = 1; i < last_assoc_idx; ++i){
|
||||||
|
TEST("%zd\t%zd ",i,assoc[i]);
|
||||||
|
// search new index
|
||||||
|
register size_t realnew = i, newval = 0;
|
||||||
|
do{
|
||||||
|
realnew = assoc[realnew];
|
||||||
|
TEST("->%zd ", realnew);
|
||||||
|
if(indexes[realnew]){ // find least index
|
||||||
|
newval = indexes[realnew];
|
||||||
|
TEST("real: %zd ", newval);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}while(assoc[realnew] != realnew);
|
||||||
|
if(newval){
|
||||||
|
TEST(" ==> %zd\n", newval);
|
||||||
|
indexes[i] = newval;
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
TEST("new index %zd\n", cidx);
|
||||||
|
// enter next label
|
||||||
|
indexes[i] = cidx++;
|
||||||
|
}
|
||||||
|
/*
|
||||||
|
// rebuild indexes
|
||||||
|
for(size_t i = 1; i < last_assoc_idx; ++i){
|
||||||
|
printf("%zd\t%zd ",i,assoc[i]);
|
||||||
|
if(i == assoc[i]){
|
||||||
|
if(i == cidx){
|
||||||
|
printf("- keep\n");
|
||||||
|
}else{
|
||||||
|
printf("- change to %zd\n", cidx);
|
||||||
|
size_t _2change = assoc[i];
|
||||||
|
assoc[i] = cidx;
|
||||||
|
for(size_t j = i+1; j < last_assoc_idx; ++j){
|
||||||
|
if(assoc[j] == _2change){
|
||||||
|
printf("\t%zd\t%zd -> %zd\n", j, assoc[j], cidx);
|
||||||
|
assoc[j] = cidx;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
++cidx;
|
||||||
|
}else printf("\n");
|
||||||
|
}*/
|
||||||
|
|
||||||
|
--cidx; // cidx now is amount of detected objects
|
||||||
|
DBG("amount after rebuild: %zd", cidx);
|
||||||
|
|
||||||
|
#ifdef TESTMSGS
|
||||||
|
printf("\n\n\nI\tASS[I]\tIDX[I]\n");
|
||||||
|
for(size_t i = 1; i < last_assoc_idx; ++i)
|
||||||
|
printf("%zd\t%zd\t%zd\n",i,assoc[i],indexes[i]);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
int wh = W*H;
|
||||||
|
OMP_FOR()
|
||||||
|
for(int i = 0; i < wh; ++i) labels[i] = indexes[labels[i]];
|
||||||
|
|
||||||
|
if(Nobj) *Nobj = cidx;
|
||||||
|
|
||||||
|
FREE(assoc);
|
||||||
|
FREE(indexes);
|
||||||
128
LocCorr_new/cclabling_ori.h
Normal file
128
LocCorr_new/cclabling_ori.h
Normal file
@ -0,0 +1,128 @@
|
|||||||
|
/*
|
||||||
|
* cclabling_1.h - inner part of functions cclabel4 and cclabel8
|
||||||
|
*
|
||||||
|
* Copyright 2015 Edward V. Emelianoff <eddy@sao.ru>
|
||||||
|
*
|
||||||
|
* This program is free software; you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation; either version 2 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program; if not, write to the Free Software
|
||||||
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
||||||
|
* MA 02110-1301, USA.
|
||||||
|
*/
|
||||||
|
|
||||||
|
//double t0 = dtime();
|
||||||
|
|
||||||
|
size_t N = 0, // current label
|
||||||
|
Nmax = W*H/4; // max number of labels
|
||||||
|
int w = W - 1, h = H - 1;
|
||||||
|
int y;
|
||||||
|
size_t *assoc = MALLOC(size_t, Nmax); // allocate memory for "remark" array
|
||||||
|
size_t last_assoc_idx = 0; // last index filled in assoc array
|
||||||
|
size_t currentnum = 0; // current pixel number
|
||||||
|
inline void remark(size_t old, size_t new){ // remark in assoc[] pixel with value old to assoc[new] or vice versa
|
||||||
|
size_t New = assoc[new], Old = assoc[old];
|
||||||
|
if(Old == New){
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
// now we must check Old: if Old<New we should swap them
|
||||||
|
if(Old < New){
|
||||||
|
register size_t _tmp_ = Old; Old = New; New = _tmp_; // swap values
|
||||||
|
_tmp_ = old; old = new; new = _tmp_;
|
||||||
|
}
|
||||||
|
// decrement counters for current value (because we make merging)
|
||||||
|
--currentnum;
|
||||||
|
for(size_t i = 1; i < last_assoc_idx; ++i){
|
||||||
|
size_t m = assoc[i];
|
||||||
|
if(m < Old) continue; // lower values
|
||||||
|
if(m == Old){ // change all old markers to new
|
||||||
|
assoc[i] = New;
|
||||||
|
}else{ // decrement all higher values
|
||||||
|
--assoc[i];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
size_t *ptr = labels;
|
||||||
|
for(y = 0; y < H; y++){
|
||||||
|
bool found = false;
|
||||||
|
size_t curmark = 0; // mark of pixel to the left
|
||||||
|
for(int x = 0; x < W; x++, ptr++){
|
||||||
|
size_t curval = *ptr;
|
||||||
|
if(!curval){ found = false; continue;}
|
||||||
|
size_t *U = (y) ? &ptr[-W] : NULL;
|
||||||
|
size_t upmark = 0; // mark from line above
|
||||||
|
if(!found){ // new pixel, check neighbours above
|
||||||
|
found = true;
|
||||||
|
// now check neighbours in upper string:
|
||||||
|
if(U){
|
||||||
|
#ifdef LABEL_8
|
||||||
|
if(x && U[-1]){ // there is something in upper left corner -> use its value
|
||||||
|
upmark = U[-1];
|
||||||
|
}else // check point above only if there's nothing in left up
|
||||||
|
#endif
|
||||||
|
if(U[0]) upmark = U[0];
|
||||||
|
#ifdef LABEL_8
|
||||||
|
if(x < w && U[1]){ // there's something in upper right
|
||||||
|
if(upmark){ // to the left of it was pixels
|
||||||
|
remark(U[1], upmark);
|
||||||
|
}else
|
||||||
|
upmark = U[1];
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
if(!upmark){ // there's nothing above - set current pixel to incremented counter
|
||||||
|
#ifdef LABEL_8 // check, whether pixel is not single
|
||||||
|
size_t *D = (y < h) ? &ptr[W] : NULL;
|
||||||
|
if( !(x && ((D && D[-1]) /*|| ptr[-1]*/)) // no left neighbours
|
||||||
|
&& !(x < w && ((D && D[1]) || ptr[1])) // no right neighbours
|
||||||
|
&& !(D && D[0])){ // no neighbour down
|
||||||
|
*ptr = 0; // erase this hermit!
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
// no neighbour down & neighbour to the right -> hermit
|
||||||
|
if((y < h && ptr[W] == 0) && (x < w && ptr[1] == 0)){
|
||||||
|
*ptr = 0; // erase this hermit!
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
upmark = ++N;
|
||||||
|
assoc[upmark] = ++currentnum; // refresh "assoc"
|
||||||
|
last_assoc_idx = upmark + 1;
|
||||||
|
}
|
||||||
|
*ptr = upmark;
|
||||||
|
curmark = upmark;
|
||||||
|
}else{ // there was something to the left -> we must chek only U[1]
|
||||||
|
if(U){
|
||||||
|
if(x < w && U[1]){ // there's something in upper right
|
||||||
|
remark(U[1], curmark);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
*ptr = curmark;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#if 0
|
||||||
|
// Step 2: rename markers
|
||||||
|
// first correct complex assotiations in assoc
|
||||||
|
OMP_FOR()
|
||||||
|
for(y = 0; y < H; y++){
|
||||||
|
size_t *ptr = &labels[y*W];
|
||||||
|
for(int x = 0; x < W; x++, ptr++){
|
||||||
|
size_t p = *ptr;
|
||||||
|
if(!p){continue;}
|
||||||
|
*ptr = assoc[p];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
FREE(assoc);
|
||||||
|
if(Nobj) *Nobj = currentnum;
|
||||||
|
//printf("%6.4f\t%zd\n", dtime()-t0, currentnum);
|
||||||
118
LocCorr_new/cmdlnopts.c
Normal file
118
LocCorr_new/cmdlnopts.c
Normal file
@ -0,0 +1,118 @@
|
|||||||
|
/* geany_encoding=koi8-r
|
||||||
|
* cmdlnopts.c - the only function that parse cmdln args and returns glob parameters
|
||||||
|
*
|
||||||
|
* Copyright 2013 Edward V. Emelianoff <eddy@sao.ru>
|
||||||
|
*
|
||||||
|
* This program is free software; you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation; either version 2 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program; if not, write to the Free Software
|
||||||
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
||||||
|
* MA 02110-1301, USA.
|
||||||
|
*/
|
||||||
|
#include <assert.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <strings.h>
|
||||||
|
#include <math.h>
|
||||||
|
|
||||||
|
#include "cmdlnopts.h"
|
||||||
|
#include "config.h"
|
||||||
|
#include "debug.h"
|
||||||
|
#include "improc.h"
|
||||||
|
|
||||||
|
/*
|
||||||
|
* here are global parameters initialisation
|
||||||
|
*/
|
||||||
|
static int help;
|
||||||
|
glob_pars *GP = NULL;
|
||||||
|
|
||||||
|
// DEFAULTS
|
||||||
|
// default global parameters
|
||||||
|
static glob_pars G = {
|
||||||
|
.pidfile = DEFAULT_PIDFILE,
|
||||||
|
.configname = DEFAULT_CONFFILE,
|
||||||
|
.steppersport = DEFAULT_STEPPERSPORT,
|
||||||
|
.throwpart = DEFAULT_THROWPART,
|
||||||
|
// .medradius = 1.,
|
||||||
|
.ndilations = DEFAULT_DILATIONS,
|
||||||
|
.nerosions = DEFAULT_EROSIONS,
|
||||||
|
.minarea = DEFAULT_MINAREA,
|
||||||
|
.maxarea = DEFAULT_MAXAREA,
|
||||||
|
.intensthres = DEFAULT_INTENSTHRES,
|
||||||
|
.maxexp = EXPOS_MAX,
|
||||||
|
.minexp = EXPOS_MIN,
|
||||||
|
.xtarget = -1.,
|
||||||
|
.ytarget = -1.,
|
||||||
|
.Naveraging = DEFAULT_NAVERAGE,
|
||||||
|
.ioport = DEFAULT_IOPORT,
|
||||||
|
.outputjpg = DEFAULT_OUTPJPEG,
|
||||||
|
};
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Define command line options by filling structure:
|
||||||
|
* name has_arg flag val type argptr help
|
||||||
|
*/
|
||||||
|
static myoption cmdlnopts[] = {
|
||||||
|
// common options
|
||||||
|
{"maxexp", NEED_ARG, NULL, 0, arg_double, APTR(&G.maxexp), _("maximal exposition time (ms), default: 500")},
|
||||||
|
{"minexp", NEED_ARG, NULL, 0, arg_double, APTR(&G.minexp), _("minimal exposition time (ms), default: 0.001")},
|
||||||
|
{"help", NO_ARGS, NULL, 'h', arg_int, APTR(&help), _("show this help")},
|
||||||
|
{"logfile", NEED_ARG, NULL, 'l', arg_string, APTR(&G.logfile), _("file to save logs (default: none)")},
|
||||||
|
{"pidfile", NEED_ARG, NULL, 'P', arg_string, APTR(&G.pidfile), _("pidfile (default: " DEFAULT_PIDFILE ")")},
|
||||||
|
{"verbose", NO_ARGS, NULL, 'v', arg_none, APTR(&G.verb), _("increase verbosity level of log file (each -v increased by 1)")},
|
||||||
|
{"input", NEED_ARG, NULL, 'i', arg_string, APTR(&G.inputname), _("file or directory name for monitoring (or grasshopper/basler for capturing)")},
|
||||||
|
{"blackp", NEED_ARG, NULL, 'b', arg_double, APTR(&G.throwpart), _("fraction of black pixels to throw away when make histogram eq")},
|
||||||
|
// {"radius", NEED_ARG, NULL, 'r', arg_int, APTR(&G.medradius), _("radius of median filter (r=1 -> 3x3, r=2 -> 5x5 etc.)")},
|
||||||
|
{"equalize", NO_ARGS, NULL, 'e', arg_int, APTR(&G.equalize), _("make historam equalization of saved jpeg")},
|
||||||
|
{"ndilat", NEED_ARG, NULL, 'D', arg_int, APTR(&G.ndilations),_("amount of dilations after thresholding (default: 2)")},
|
||||||
|
{"neros", NEED_ARG, NULL, 'E', arg_int, APTR(&G.nerosions), _("amount of erosions after dilations (default: 2)")},
|
||||||
|
{"minarea", NEED_ARG, NULL, 'I', arg_int, APTR(&G.minarea), _("minimal object pixels amount (default: 400)")},
|
||||||
|
{"maxarea", NEED_ARG, NULL, 'A', arg_int, APTR(&G.maxarea), _("maximal object pixels amount (default: 150000)")},
|
||||||
|
{"intthres",NEED_ARG, NULL, 'T', arg_double, APTR(&G.intensthres),_("threshold by total object intensity when sorting = |I1-I2|/(I1+I2), default: 0.01")},
|
||||||
|
{"xoff", NEED_ARG, NULL, 'x', arg_int, APTR(&G.xoff), _("X offset at grabbed image")},
|
||||||
|
{"yoff", NEED_ARG, NULL, 'y', arg_int, APTR(&G.yoff), _("Y offset at grabbed image")},
|
||||||
|
{"width", NEED_ARG, NULL, 'W', arg_int, APTR(&G.width), _("grabbed subimage width")},
|
||||||
|
{"height", NEED_ARG, NULL, 'H', arg_int, APTR(&G.height), _("grabbed subimage height")},
|
||||||
|
{"xtarget", NEED_ARG, NULL, 'X', arg_double, APTR(&G.xtarget), _("target point X coordinate")},
|
||||||
|
{"ytarget", NEED_ARG, NULL, 'Y', arg_double, APTR(&G.ytarget), _("target point Y coordinate")},
|
||||||
|
{"logXY", NEED_ARG, NULL, 'L', arg_string, APTR(&G.logXYname), _("file to log XY coordinates of selected star")},
|
||||||
|
{"confname",NEED_ARG, NULL, 'c', arg_string, APTR(&G.configname),_("name of configuration file (default: ./loccorr.conf)")},
|
||||||
|
{"canport", NEED_ARG, NULL, 'C', arg_string, APTR(&G.steppersport),_("port of local pusirobot CAN server (default: 4444)")},
|
||||||
|
{"naverage",NEED_ARG, NULL, 'N', arg_int, APTR(&G.Naveraging),_("amount of images to average processing (min 2, max 25)")},
|
||||||
|
{"ioport", NEED_ARG, NULL, 0, arg_int, APTR(&G.ioport), _("port for IO communication")},
|
||||||
|
{"jpegout", NEED_ARG, NULL, 'j', arg_string, APTR(&G.outputjpg), _("output jpeg file location (default: '" DEFAULT_OUTPJPEG "')")},
|
||||||
|
end_option
|
||||||
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Parse command line options and return dynamically allocated structure
|
||||||
|
* to global parameters
|
||||||
|
* @param argc - copy of argc from main
|
||||||
|
* @param argv - copy of argv from main
|
||||||
|
* @return allocated structure with global parameters
|
||||||
|
*/
|
||||||
|
glob_pars *parse_args(int argc, char **argv){
|
||||||
|
size_t hlen = 1024;
|
||||||
|
char helpstring[1024], *hptr = helpstring;
|
||||||
|
snprintf(hptr, hlen, "Usage: %%s [args]\n\n\tWhere args are:\n");
|
||||||
|
// format of help: "Usage: progname [args]\n"
|
||||||
|
change_helpstring(helpstring);
|
||||||
|
// parse arguments
|
||||||
|
parseargs(&argc, &argv, cmdlnopts);
|
||||||
|
if(help) showhelp(-1, cmdlnopts);
|
||||||
|
if(argc > 0){
|
||||||
|
WARNX("Extra parameters!");
|
||||||
|
showhelp(-1, cmdlnopts);
|
||||||
|
}
|
||||||
|
return &G;
|
||||||
|
}
|
||||||
|
|
||||||
77
LocCorr_new/cmdlnopts.h
Normal file
77
LocCorr_new/cmdlnopts.h
Normal file
@ -0,0 +1,77 @@
|
|||||||
|
/* geany_encoding=koi8-r
|
||||||
|
* cmdlnopts.h - comand line options for parceargs
|
||||||
|
*
|
||||||
|
* Copyright 2013 Edward V. Emelianoff <eddy@sao.ru>
|
||||||
|
*
|
||||||
|
* This program is free software; you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation; either version 2 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program; if not, write to the Free Software
|
||||||
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
||||||
|
* MA 02110-1301, USA.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
#ifndef CMDLNOPTS_H__
|
||||||
|
#define CMDLNOPTS_H__
|
||||||
|
|
||||||
|
// default values
|
||||||
|
#define DEFAULT_PIDFILE "/tmp/loccorr.pid"
|
||||||
|
#define DEFAULT_CONFFILE "./loccorr.conf"
|
||||||
|
#define DEFAULT_OUTPJPEG "./outpWcrosses.jpg"
|
||||||
|
#define DEFAULT_STEPPERSPORT (4444)
|
||||||
|
#define DEFAULT_IOPORT (12345)
|
||||||
|
#define DEFAULT_MAXAREA (150000)
|
||||||
|
#define DEFAULT_MINAREA (400)
|
||||||
|
#define DEFAULT_EROSIONS (2)
|
||||||
|
#define DEFAULT_DILATIONS (2)
|
||||||
|
#define DEFAULT_THROWPART (0.5)
|
||||||
|
#define DEFAULT_INTENSTHRES (0.01)
|
||||||
|
#define DEFAULT_NAVERAGE (5)
|
||||||
|
#define DEFAULT_MAXUSTEPS (16000)
|
||||||
|
#define DEFAULT_MAXVSTEPS (16000)
|
||||||
|
#define DEFAULT_NEROSIONS (3)
|
||||||
|
#define DEFAULT_NDILATIONS (3)
|
||||||
|
|
||||||
|
/*
|
||||||
|
* here are some typedef's for global data
|
||||||
|
*/
|
||||||
|
typedef struct{
|
||||||
|
char *pidfile; // name of PID file
|
||||||
|
char *logfile; // logging to this file
|
||||||
|
char *inputname; // input for monitor file or directory name
|
||||||
|
char *logXYname; // file to log XY coordinates of first point
|
||||||
|
char *configname; // name of configuration file (default: ./loccorr.conf)
|
||||||
|
char *outputjpg; // output jpeg name
|
||||||
|
int steppersport; // port of local motors CAN server
|
||||||
|
int equalize; // make historam equalization of saved jpeg
|
||||||
|
// int medradius; // radius of median filter (r=1 -> 3x3, r=2 -> 5x5 etc.)
|
||||||
|
int verb; // logfile verbosity level
|
||||||
|
int ndilations; // amount of erosions (default: 2)
|
||||||
|
int nerosions; // amount of dilations (default: 2)
|
||||||
|
int minarea; // minimal object pixels amount (default: 400)
|
||||||
|
int maxarea; // maximal object pixels amount (default: 150000)
|
||||||
|
int Naveraging; // amount of images to average processing (min 2, max 25, default 5)
|
||||||
|
int xoff; int yoff; // offset by X and Y axes
|
||||||
|
int width; int height; // target width and height of image
|
||||||
|
int ioport; // port for IO commands
|
||||||
|
double throwpart; // fraction of black pixels to throw away when make histogram eq
|
||||||
|
double intensthres; // threshold by total object intensity when sorting = |I1-I2|/(I1+I2), default: 0.01
|
||||||
|
double maxexp; // max exposition time (ms)
|
||||||
|
double minexp; // min exposition time (ms)
|
||||||
|
double xtarget; double ytarget;// target point coordinates
|
||||||
|
} glob_pars;
|
||||||
|
|
||||||
|
extern glob_pars *GP; // for GP->pidfile need in `signals`
|
||||||
|
|
||||||
|
glob_pars *parse_args(int argc, char **argv);
|
||||||
|
|
||||||
|
#endif // CMDLNOPTS_H__
|
||||||
407
LocCorr_new/config.c
Normal file
407
LocCorr_new/config.c
Normal file
@ -0,0 +1,407 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <limits.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <string.h>
|
||||||
|
|
||||||
|
#include "cmdlnopts.h"
|
||||||
|
#include "config.h"
|
||||||
|
#include "debug.h"
|
||||||
|
|
||||||
|
static char *conffile = NULL; // configuration file name
|
||||||
|
|
||||||
|
configuration theconf = {
|
||||||
|
.maxUpos=DEFAULT_MAXUSTEPS,
|
||||||
|
.minUpos=0,
|
||||||
|
.maxVpos=DEFAULT_MAXVSTEPS,
|
||||||
|
.minVpos=0,
|
||||||
|
.maxFpos=Fmaxsteps-1,
|
||||||
|
.minFpos=-Fmaxsteps+1,
|
||||||
|
.minarea=DEFAULT_MINAREA,
|
||||||
|
.maxarea=DEFAULT_MAXAREA,
|
||||||
|
.maxwh=1.1,
|
||||||
|
.minwh=0.9,
|
||||||
|
.Nerosions=DEFAULT_NEROSIONS,
|
||||||
|
.Ndilations=DEFAULT_NDILATIONS,
|
||||||
|
.xoff=0,
|
||||||
|
.yoff=0,
|
||||||
|
.width=0,
|
||||||
|
.height=0,
|
||||||
|
.equalize=1,
|
||||||
|
.naverage=DEFAULT_NAVERAGE,
|
||||||
|
.stpserverport=DEFAULT_STEPPERSPORT,
|
||||||
|
.starssort=0,
|
||||||
|
.Kxu=0,
|
||||||
|
.Kyu=0,
|
||||||
|
.Kxv=0,
|
||||||
|
.Kyv=0,
|
||||||
|
.xtarget=-1,
|
||||||
|
.ytarget=-1,
|
||||||
|
.throwpart=DEFAULT_THROWPART,
|
||||||
|
.maxexp=EXPOS_MAX - 1.,
|
||||||
|
.minexp=EXPOS_MIN,
|
||||||
|
.fixedexp=EXPOS_MIN*2,
|
||||||
|
.gain=20.,
|
||||||
|
.intensthres=DEFAULT_INTENSTHRES,
|
||||||
|
.medseed=MIN_MEDIAN_SEED,
|
||||||
|
};
|
||||||
|
|
||||||
|
// {"", PAR_DOUBLE, (void*)&theconf., 0},
|
||||||
|
static confparam parvals[] = {
|
||||||
|
{"maxarea", PAR_INT, (void*)&theconf.maxarea, 0, MINAREA, MAXAREA,
|
||||||
|
"maximal area (in square pixels) of recognized star image"},
|
||||||
|
{"minarea", PAR_INT, (void*)&theconf.minarea, 0, MINAREA, MAXAREA,
|
||||||
|
"minimal area (in square pixels) of recognized star image"},
|
||||||
|
{"minwh", PAR_DOUBLE, (void*)&theconf.minwh, 0, MINWH, 1.,
|
||||||
|
"minimal value of W/H roundness parameter"},
|
||||||
|
{"maxwh", PAR_DOUBLE, (void*)&theconf.maxwh, 0, 1., MAXWH,
|
||||||
|
"maximal value of W/H roundness parameter"},
|
||||||
|
{"ndilat", PAR_INT, (void*)&theconf.Ndilations, 0, 1., MAX_NDILAT,
|
||||||
|
"amount of dilations on binarized image"},
|
||||||
|
{"neros", PAR_INT, (void*)&theconf.Nerosions, 0, 1., MAX_NEROS,
|
||||||
|
"amount of erosions after dilations"},
|
||||||
|
{"xoffset", PAR_INT, (void*)&theconf.xoff, 0, 0., MAX_OFFSET,
|
||||||
|
"X offset of subimage"},
|
||||||
|
{"yoffset", PAR_INT, (void*)&theconf.yoff, 0, 0., MAX_OFFSET,
|
||||||
|
"Y offset of subimage"},
|
||||||
|
{"width", PAR_INT, (void*)&theconf.width, 0, 0., MAX_OFFSET,
|
||||||
|
"subimage width"},
|
||||||
|
{"height", PAR_INT, (void*)&theconf.height, 0, 0., MAX_OFFSET,
|
||||||
|
"subimage height"},
|
||||||
|
{"equalize", PAR_INT, (void*)&theconf.equalize, 0, 0., 1.,
|
||||||
|
"make histogram equalization"},
|
||||||
|
{"expmethod", PAR_INT, (void*)&theconf.expmethod, 0, 0., 1.,
|
||||||
|
"exposition method: 0 - auto, 1 - fixed"},
|
||||||
|
{"naverage", PAR_INT, (void*)&theconf.naverage, 0, 1., NAVER_MAX,
|
||||||
|
"calculate mean position by N images"},
|
||||||
|
{"umax", PAR_INT, (void*)&theconf.maxUpos, 0, -MAXSTEPS, MAXSTEPS,
|
||||||
|
"maximal value of steps on U semi-axe"},
|
||||||
|
{"umin", PAR_INT, (void*)&theconf.minUpos, 0, -MAXSTEPS, MAXSTEPS,
|
||||||
|
"minimal value of steps on U semi-axe"},
|
||||||
|
{"vmax", PAR_INT, (void*)&theconf.maxVpos, 0, -MAXSTEPS, MAXSTEPS,
|
||||||
|
"maximal value of steps on V semi-axe"},
|
||||||
|
{"vmin", PAR_INT, (void*)&theconf.minVpos, 0, -MAXSTEPS, MAXSTEPS,
|
||||||
|
"minimal value of steps on V semi-axe"},
|
||||||
|
{"focmax", PAR_INT, (void*)&theconf.maxFpos, 0, 0., Fmaxsteps,
|
||||||
|
"maximal focus position in microsteps"},
|
||||||
|
{"focmin", PAR_INT, (void*)&theconf.minFpos, 0, -Fmaxsteps, 0.,
|
||||||
|
"minimal focus position in microsteps"},
|
||||||
|
{"stpservport", PAR_INT, (void*)&theconf.stpserverport, 0, 0., 65535.,
|
||||||
|
"port number of steppers' server"},
|
||||||
|
{"Kxu", PAR_DOUBLE, (void*)&theconf.Kxu, 0, KUVMIN, KUVMAX,
|
||||||
|
"dU = Kxu*dX + Kyu*dY"},
|
||||||
|
{"Kyu", PAR_DOUBLE, (void*)&theconf.Kyu, 0, KUVMIN, KUVMAX,
|
||||||
|
"dU = Kxu*dX + Kyu*dY"},
|
||||||
|
{"Kxv", PAR_DOUBLE, (void*)&theconf.Kxv, 0, KUVMIN, KUVMAX,
|
||||||
|
"dV = Kxv*dX + Kyv*dY"},
|
||||||
|
{"Kyv", PAR_DOUBLE, (void*)&theconf.Kyv, 0, KUVMIN, KUVMAX,
|
||||||
|
"dV = Kxv*dX + Kyv*dY"},
|
||||||
|
{"xtarget", PAR_DOUBLE, (void*)&theconf.xtarget, 0, 1., MAX_OFFSET,
|
||||||
|
"X coordinate of target position"},
|
||||||
|
{"ytarget", PAR_DOUBLE, (void*)&theconf.ytarget, 0, 1., MAX_OFFSET,
|
||||||
|
"Y coordinate of target position"},
|
||||||
|
{"eqthrowpart", PAR_DOUBLE, (void*)&theconf.throwpart, 0, 0., MAX_THROWPART,
|
||||||
|
"a part of low intensity pixels to throw away when histogram equalized"},
|
||||||
|
{"minexp", PAR_DOUBLE, (void*)&theconf.minexp, 0, 0., EXPOS_MAX,
|
||||||
|
"minimal exposition time"},
|
||||||
|
{"maxexp", PAR_DOUBLE, (void*)&theconf.maxexp, 0, 0., EXPOS_MAX,
|
||||||
|
"maximal exposition time"},
|
||||||
|
{"fixedexp", PAR_DOUBLE, (void*)&theconf.fixedexp, 0, EXPOS_MIN, EXPOS_MAX,
|
||||||
|
"fixed (in manual mode) exposition time"},
|
||||||
|
{"intensthres", PAR_DOUBLE, (void*)&theconf.intensthres, 0, 0., 1.,
|
||||||
|
"threshold by total object intensity when sorting = |I1-I2|/(I1+I2)"},
|
||||||
|
{"gain", PAR_DOUBLE, (void*)&theconf.gain, 0, GAIN_MIN, GAIN_MAX,
|
||||||
|
"gain value in manual mode"},
|
||||||
|
{"brightness", PAR_DOUBLE, (void*)&theconf.brightness, 0, BRIGHT_MIN, BRIGHT_MAX,
|
||||||
|
"brightness value"},
|
||||||
|
{"starssort", PAR_INT, (void*)&theconf.starssort, 0, 0., 1.,
|
||||||
|
"stars sorting algorithm: by distance from target (0) or by intensity (1)"},
|
||||||
|
{"medfilt", PAR_INT, (void*)&theconf.medfilt, 0, 0., 1.,
|
||||||
|
"use median filter (1) or not (0)"},
|
||||||
|
{"medseed", PAR_INT, (void*)&theconf.medseed, 0, MIN_MEDIAN_SEED, MAX_MEDIAN_SEED,
|
||||||
|
"median filter radius"},
|
||||||
|
{"fixedbg", PAR_INT, (void*)&theconf.fixedbkg, 0, 0., 1.,
|
||||||
|
"don't calculate background, use fixed value instead"},
|
||||||
|
{"fbglevel", PAR_INT, (void*)&theconf.fixedbkgval, 0, FIXED_BK_MIN, FIXED_BK_MAX,
|
||||||
|
"fixed background level"},
|
||||||
|
{NULL, 0, NULL, 0, 0., 0., NULL}
|
||||||
|
};
|
||||||
|
|
||||||
|
// return pointer to buff with size l filled with list of all commands (+help messages + low/high values)
|
||||||
|
char *get_cmd_list(char *buff, int l){
|
||||||
|
if(!buff || l < 1) return NULL;
|
||||||
|
int L = l;
|
||||||
|
char *ptr = buff;
|
||||||
|
confparam *par = parvals;
|
||||||
|
while(L > 0 && par->name){
|
||||||
|
int s = snprintf(ptr, L, "%s=newval - %s (from %g to %g)\n", par->name, par->help, par->minval, par->maxval);
|
||||||
|
if(s < 1) break;
|
||||||
|
L -= s; ptr += s;
|
||||||
|
++par;
|
||||||
|
}
|
||||||
|
return buff;
|
||||||
|
}
|
||||||
|
|
||||||
|
static char *omitspaces(char *v){
|
||||||
|
if(!v) return NULL;
|
||||||
|
while(*v && (*v == ' ' || *v == '\t')) ++v;
|
||||||
|
char *ptr = strchr(v, ' ');
|
||||||
|
if(ptr) *ptr = 0;
|
||||||
|
ptr = strchr(v, '\t');
|
||||||
|
if(ptr) *ptr = 0;
|
||||||
|
return v;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Read key/value from `pair` (key = value)
|
||||||
|
// RETURNed value should be FREEd
|
||||||
|
char *get_keyval(const char *pair, char value[128]){
|
||||||
|
char key[128];
|
||||||
|
char val[128];
|
||||||
|
if(!pair || !*pair) return strdup("#"); // empty line
|
||||||
|
char *keyptr = key, *valptr = val;
|
||||||
|
int x = sscanf(pair, "%127[^=]=%127[^\n]%*c", key, val);
|
||||||
|
//DBG("x=%d, key='%s', val='%s'", x, key, val);
|
||||||
|
if(x < 0 || x > 2) return NULL; // wrong data or EOF
|
||||||
|
if(x == 0) return strdup("#"); // empty line
|
||||||
|
keyptr = omitspaces(key);
|
||||||
|
if(x == 2){ // param = value
|
||||||
|
valptr = omitspaces(val);
|
||||||
|
sprintf(value, "%s", valptr);
|
||||||
|
return strdup(keyptr);
|
||||||
|
}
|
||||||
|
if(*keyptr == '#' || *keyptr == '%'){ // comment
|
||||||
|
*value = 0;
|
||||||
|
return strdup("#");
|
||||||
|
}
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Read key/value from file
|
||||||
|
static char *read_key(FILE *file, char value[128]){
|
||||||
|
char *line = NULL;
|
||||||
|
size_t n = 0;
|
||||||
|
int got = getline(&line, &n, file);
|
||||||
|
if(!line) return NULL;
|
||||||
|
if(got < 0){
|
||||||
|
free(line);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
char *kv = get_keyval(line, value);
|
||||||
|
return kv;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int str2int(int *num, const char *str){
|
||||||
|
long res;
|
||||||
|
char *endptr;
|
||||||
|
if(!str) return 0;
|
||||||
|
res = strtol(str, &endptr, 0);
|
||||||
|
if(endptr == str || *str == '\0' || *endptr != '\0'){
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
if(res > INT_MAX || res < INT_MIN) return FALSE;
|
||||||
|
if(num) *num = (int)res;
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief chk_keyval - check key for presence in theconf and calculate its value
|
||||||
|
* @param key (i) - keyword
|
||||||
|
* @param val (i) - value
|
||||||
|
* @param result - result calculated from `val`
|
||||||
|
* @return pointer to confparam if found & checked
|
||||||
|
*/
|
||||||
|
confparam *chk_keyval(const char *key, const char *val, key_value *result){
|
||||||
|
if(!key || !val || !result) return NULL;
|
||||||
|
confparam *par = parvals;
|
||||||
|
while(par->name){
|
||||||
|
if(strcmp(key, par->name) == 0){
|
||||||
|
//DBG("key='%s', par->name='%s'", key, par->name);
|
||||||
|
result->type = par->type;
|
||||||
|
switch(par->type){
|
||||||
|
case PAR_INT:
|
||||||
|
//DBG("INTEGER");
|
||||||
|
if(!str2int(&result->val.intval, val)){
|
||||||
|
WARNX("Wrong integer value '%s' of parameter '%s'", val, key);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
if(result->val.intval < par->minval || result->val.intval > par->maxval)
|
||||||
|
WARNX("Value (%d) of parameter %s out of range %g..%g",
|
||||||
|
result->val.intval, par->name, par->minval, par->maxval);
|
||||||
|
else return par;
|
||||||
|
break;
|
||||||
|
case PAR_DOUBLE:
|
||||||
|
//DBG("DOUBLE");
|
||||||
|
if(!str2double(&result->val.dblval, val)){
|
||||||
|
WARNX("Wrong double value '%s' of parameter '%s'", val, key);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
//DBG("val: %g, min: %g, max: %g", result->val.dblval, par->minval, par->maxval);
|
||||||
|
if(result->val.dblval < par->minval || result->val.dblval > par->maxval)
|
||||||
|
WARNX("Value (%g) of parameter %s out of range %g..%g",
|
||||||
|
result->val.dblval, par->name, par->minval, par->maxval);
|
||||||
|
else return par;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
++par;
|
||||||
|
}
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief chkconfig - check configuration file and init variables
|
||||||
|
* @param confname - name of file
|
||||||
|
* @return FALSE if configuration file wrong or absent
|
||||||
|
*/
|
||||||
|
int chkconfig(const char *confname){
|
||||||
|
DBG("Config name: %s", confname);
|
||||||
|
if(conffile){ free(conffile); conffile = NULL; }
|
||||||
|
conffile = strdup(confname);
|
||||||
|
FILE *f = fopen(confname, "r");
|
||||||
|
int ret = TRUE;
|
||||||
|
if(!f){
|
||||||
|
WARN("Can't open %s", confname);
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
char *key, val[128];
|
||||||
|
confparam *par = parvals;
|
||||||
|
while(par->name){
|
||||||
|
par->got = 0;
|
||||||
|
++par;
|
||||||
|
}
|
||||||
|
while((key = read_key(f, val))){
|
||||||
|
if(*key == '#'){
|
||||||
|
free(key);
|
||||||
|
key = NULL;
|
||||||
|
continue; // comment
|
||||||
|
}
|
||||||
|
//DBG("key: %s", key);
|
||||||
|
key_value kv;
|
||||||
|
par = chk_keyval(key, val, &kv);
|
||||||
|
if(!par){
|
||||||
|
WARNX("Parameter '%s' is wrong or out of range", key);
|
||||||
|
free(key);
|
||||||
|
key = NULL;
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
switch(par->type){
|
||||||
|
case PAR_INT:
|
||||||
|
*((int*)par->ptr) = kv.val.intval;
|
||||||
|
break;
|
||||||
|
case PAR_DOUBLE:
|
||||||
|
*((double*)par->ptr) = kv.val.dblval;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
++par->got;
|
||||||
|
free(key);
|
||||||
|
key = NULL;
|
||||||
|
}
|
||||||
|
fclose(f);
|
||||||
|
int found = 0;
|
||||||
|
par = parvals;
|
||||||
|
while(par->name){
|
||||||
|
//DBG("parvals[]={%s, %d, %d(%g), %d}", par->name, par->type, *((int*)par->ptr), *((double*)par->ptr), par->got);
|
||||||
|
int k = par->got;
|
||||||
|
if(!k){
|
||||||
|
++par;
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
if(k > 1){
|
||||||
|
WARNX("parameter '%s' meets %d times", par->name, k);
|
||||||
|
ret = FALSE;
|
||||||
|
}
|
||||||
|
++found; ++par;
|
||||||
|
}
|
||||||
|
DBG("chkconfig(): found %d", found);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief saveconf - try to save configuration into file
|
||||||
|
* @param confname - config file name
|
||||||
|
* @return FALSE if failed
|
||||||
|
*/
|
||||||
|
int saveconf(const char *confname){
|
||||||
|
if(!confname){
|
||||||
|
if(!conffile){
|
||||||
|
WARNX("no conffile given");
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
confname = conffile;
|
||||||
|
}
|
||||||
|
FILE *f = fopen(confname, "w");
|
||||||
|
if(!f){
|
||||||
|
WARN("Can't open %s", confname);
|
||||||
|
LOGERR("Can't open %s to store configuration", confname);
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
confparam *par = parvals;
|
||||||
|
while(par->name){
|
||||||
|
par->got = 1;
|
||||||
|
switch(par->type){
|
||||||
|
case PAR_INT:
|
||||||
|
fprintf(f, "%s = %d\n", par->name, *((int*)par->ptr));
|
||||||
|
//DBG("%s = %d", par->name, *((int*)par->ptr));
|
||||||
|
break;
|
||||||
|
case PAR_DOUBLE:
|
||||||
|
fprintf(f, "%s = %.3f\n", par->name, *((double*)par->ptr));
|
||||||
|
//DBG("%s = %.3f", par->name, *((double*)par->ptr));
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
++par;
|
||||||
|
}
|
||||||
|
DBG("%s saved", confname);
|
||||||
|
LOGDBG("Configuration file '%s' saved (my PID=%zd)", confname, getpid());
|
||||||
|
fclose(f);
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// return buffer filled with current configuration
|
||||||
|
char *listconf(const char *messageid, char *buf, int buflen){
|
||||||
|
int L;
|
||||||
|
char *ptr = buf;
|
||||||
|
confparam *par = parvals;
|
||||||
|
L = snprintf(ptr, buflen, "{ \"%s\": \"%s\", ", MESSAGEID, messageid);
|
||||||
|
buflen -= L; ptr += L;
|
||||||
|
while(par->name && buflen > 0){
|
||||||
|
switch(par->type){
|
||||||
|
case PAR_INT:
|
||||||
|
L = snprintf(ptr, buflen, "\"%s\": %d", par->name, *((int*)par->ptr));
|
||||||
|
break;
|
||||||
|
case PAR_DOUBLE:
|
||||||
|
L = snprintf(ptr, buflen, "\"%s\": %.3f", par->name, *((double*)par->ptr));
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
L = 0;
|
||||||
|
}
|
||||||
|
++par;
|
||||||
|
if(L > -1){
|
||||||
|
buflen -= L; ptr += L;
|
||||||
|
}else{
|
||||||
|
buf[buflen-1] = 0; break;
|
||||||
|
}
|
||||||
|
if(par->name){ // put comma
|
||||||
|
L = snprintf(ptr, buflen, ", ");
|
||||||
|
if(L > -1){buflen -= L; ptr += L;}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
snprintf(ptr, buflen, " }\n");
|
||||||
|
return buf;
|
||||||
|
}
|
||||||
140
LocCorr_new/config.h
Normal file
140
LocCorr_new/config.h
Normal file
@ -0,0 +1,140 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
#pragma once
|
||||||
|
#ifndef CONFIG_H__
|
||||||
|
#define CONFIG_H__
|
||||||
|
|
||||||
|
// default configuration borders
|
||||||
|
// min/max total steps range
|
||||||
|
#define MAXSTEPS (50000)
|
||||||
|
#define Fmaxsteps (64000)
|
||||||
|
// steps per pixel
|
||||||
|
#define COEFMIN (0.1)
|
||||||
|
#define COEFMAX (10000)
|
||||||
|
// area
|
||||||
|
#define MINAREA (4)
|
||||||
|
#define MAXAREA (2500000)
|
||||||
|
#define MAX_NDILAT (100)
|
||||||
|
#define MAX_NEROS (100)
|
||||||
|
#define MAX_THROWPART (0.9)
|
||||||
|
#define MAX_OFFSET (10000)
|
||||||
|
// min/max exposition in ms
|
||||||
|
#define EXPOS_MIN (0.1)
|
||||||
|
#define EXPOS_MAX (4001.)
|
||||||
|
#define GAIN_MIN (0.)
|
||||||
|
#define GAIN_MAX (100.)
|
||||||
|
#define BRIGHT_MIN (0.)
|
||||||
|
#define BRIGHT_MAX (10.)
|
||||||
|
// max average images counter
|
||||||
|
#define NAVER_MAX (25)
|
||||||
|
// coefficients to convert dx,dy to du,dv
|
||||||
|
#define KUVMIN (-5000.)
|
||||||
|
#define KUVMAX (5000.)
|
||||||
|
// default coefficient for corrections (move to Kdu, Kdv instead of du, dv)
|
||||||
|
#define KCORR (0.90)
|
||||||
|
// min/max median seed
|
||||||
|
#define MIN_MEDIAN_SEED (1)
|
||||||
|
#define MAX_MEDIAN_SEED (7)
|
||||||
|
// fixed background
|
||||||
|
#define FIXED_BK_MIN (0)
|
||||||
|
#define FIXED_BK_MAX (250)
|
||||||
|
|
||||||
|
// exposition methods: 0 - auto, 1 - fixed
|
||||||
|
#define EXPAUTO (0)
|
||||||
|
#define EXPMANUAL (1)
|
||||||
|
|
||||||
|
// roundness parameter
|
||||||
|
#define MINWH (0.3)
|
||||||
|
#define MAXWH (3.)
|
||||||
|
|
||||||
|
// messageID field name
|
||||||
|
#define MESSAGEID "messageid"
|
||||||
|
|
||||||
|
typedef struct{
|
||||||
|
int maxUpos; // min/max positions
|
||||||
|
int minUpos;
|
||||||
|
int maxVpos;
|
||||||
|
int minVpos;
|
||||||
|
int maxFpos; // min/max F position (in microsteps) - user can't change this
|
||||||
|
int minFpos;
|
||||||
|
int minarea; // min/max area of star image
|
||||||
|
int maxarea;
|
||||||
|
int Nerosions; // amount of erosions/dilations
|
||||||
|
int Ndilations;
|
||||||
|
int xoff; // subimage offset
|
||||||
|
int yoff;
|
||||||
|
int width; // subimage size
|
||||||
|
int height;
|
||||||
|
int equalize; // !=0 to equalize output image histogram
|
||||||
|
int naverage; // amount of images for average calculation (>1)
|
||||||
|
int stpserverport; // steppers' server port
|
||||||
|
int starssort; // stars sorting algorithm: by distance from target (0) or by intensity (1)
|
||||||
|
int expmethod; // 0 - auto, 1 - fixed
|
||||||
|
int medfilt; // == 1 to make median filter before calculations
|
||||||
|
int medseed; // median seed
|
||||||
|
int fixedbkg; // don't calculate background, use fixed value instead
|
||||||
|
int fixedbkgval; // value of bk
|
||||||
|
// dU = Kxu*dX + Kyu*dY; dV = Kxv*dX + Kyv*dY
|
||||||
|
double Kxu; double Kyu;
|
||||||
|
double Kxv; double Kyv;
|
||||||
|
double minwh; double maxwh; // roundness parameters
|
||||||
|
double xtarget; // target (center) values (in absolute coordinates! screen coords = target - offset)
|
||||||
|
double ytarget;
|
||||||
|
double throwpart; // part of values to throw avay @ histogram equalisation
|
||||||
|
double maxexp; // minimal and maximal exposition (in ms)
|
||||||
|
double minexp;
|
||||||
|
double fixedexp; // exptime in manual mode
|
||||||
|
double gain; // gain value in manual mode
|
||||||
|
double brightness; // brightness @camera
|
||||||
|
double intensthres; // threshold for stars intensity comparison: fabs(Ia-Ib)/(Ia+Ib) > thres -> stars differs
|
||||||
|
} configuration;
|
||||||
|
|
||||||
|
typedef enum{
|
||||||
|
PAR_INT,
|
||||||
|
PAR_DOUBLE
|
||||||
|
} partype;
|
||||||
|
|
||||||
|
typedef struct{
|
||||||
|
const char *name; // parameter name
|
||||||
|
partype type; // type of parameter's value
|
||||||
|
void *ptr; // pointer to value in `theconf`
|
||||||
|
int got; // counter of parameter in config file
|
||||||
|
double minval; // min/max values
|
||||||
|
double maxval;
|
||||||
|
const char *help; // help message
|
||||||
|
} confparam;
|
||||||
|
|
||||||
|
typedef union{
|
||||||
|
double dblval;
|
||||||
|
int intval;
|
||||||
|
} dblint;
|
||||||
|
|
||||||
|
typedef struct{
|
||||||
|
dblint val;
|
||||||
|
partype type;
|
||||||
|
} key_value;
|
||||||
|
|
||||||
|
extern configuration theconf;
|
||||||
|
char *get_cmd_list(char *buff, int l);
|
||||||
|
int chkconfig(const char *confname);
|
||||||
|
int saveconf(const char *confname);
|
||||||
|
char *get_keyval(const char *pair, char value[128]);
|
||||||
|
confparam *chk_keyval(const char *key, const char *val, key_value *result);
|
||||||
|
char *listconf(const char *messageid, char *buf, int buflen);
|
||||||
|
|
||||||
|
#endif // CONFIG_H__
|
||||||
57
LocCorr_new/debug.c
Normal file
57
LocCorr_new/debug.c
Normal file
@ -0,0 +1,57 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifdef EBUG
|
||||||
|
|
||||||
|
#include <string.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
|
||||||
|
#include "debug.h"
|
||||||
|
|
||||||
|
#define DEBUGLOG "DEBUG.log"
|
||||||
|
|
||||||
|
sl_log *debuglog = NULL;
|
||||||
|
|
||||||
|
void makedebuglog(){
|
||||||
|
unlink(DEBUGLOG);
|
||||||
|
debuglog = sl_createlog(DEBUGLOG, LOGLEVEL_ANY, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief my_malloc - memory allocator for logger
|
||||||
|
* @param N - number of elements to allocate
|
||||||
|
* @param S - size of single element (typically sizeof)
|
||||||
|
* @return pointer to allocated memory area
|
||||||
|
*/
|
||||||
|
void *my_malloc(size_t N, size_t S){
|
||||||
|
size_t NS = N*S + sizeof(size_t);
|
||||||
|
sl_putlogt(0, debuglog, LOGLEVEL_ERR, "ALLOCSZ(%zd)", N*S);
|
||||||
|
void *p = malloc(NS);
|
||||||
|
if(!p) ERR("malloc");
|
||||||
|
memset(p, 0, NS);
|
||||||
|
*((size_t*)p) = N*S;
|
||||||
|
return (p + sizeof(size_t));
|
||||||
|
}
|
||||||
|
|
||||||
|
void my_free(void *ptr){
|
||||||
|
void *orig = ptr - sizeof(size_t);
|
||||||
|
sl_putlogt(0, debuglog, LOGLEVEL_ERR, "FREESZ(%zd)", *(size_t*)orig);
|
||||||
|
free(orig);
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif
|
||||||
57
LocCorr_new/debug.h
Normal file
57
LocCorr_new/debug.h
Normal file
@ -0,0 +1,57 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
#pragma once
|
||||||
|
#ifndef DEBUG_H__
|
||||||
|
#define DEBUG_H__
|
||||||
|
|
||||||
|
#include <usefull_macros.h>
|
||||||
|
|
||||||
|
#ifndef TRUE_INLINE
|
||||||
|
#define TRUE_INLINE __attribute__((always_inline)) static inline
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#ifdef EBUG
|
||||||
|
|
||||||
|
extern sl_log *debuglog;
|
||||||
|
void makedebuglog();
|
||||||
|
void *my_malloc(size_t N, size_t S);
|
||||||
|
void my_free(void *ptr);
|
||||||
|
|
||||||
|
//#undef FNAME
|
||||||
|
#undef ALLOC
|
||||||
|
#undef MALLOC
|
||||||
|
#undef FREE
|
||||||
|
#define _LOG(...) do{if(!debuglog) makedebuglog(); sl_putlogt(1, debuglog, LOGLEVEL_ERR, __VA_ARGS__);}while(0)
|
||||||
|
#define DBGLOG(...) do{_LOG("%s (%s, line %d)", __func__, __FILE__, __LINE__); \
|
||||||
|
sl_putlogt(0, debuglog, LOGLEVEL_ERR, __VA_ARGS__);}while(0)
|
||||||
|
//#define FNAME() _LOG("%s (%s, line %d)", __func__, __FILE__, __LINE__)
|
||||||
|
|
||||||
|
#define _str(x) #x
|
||||||
|
#define ALLOC(type, var, size) DBGLOG("%s *%s = ALLOC(%d)", _str(type), _str(var), size*sizeof(type)); \
|
||||||
|
type * var = ((type *)my_malloc(size, sizeof(type)))
|
||||||
|
#define MALLOC(type, size) ((type *)my_malloc(size, sizeof(type))); DBGLOG("ALLOC()")
|
||||||
|
#define FREE(ptr) do{if(ptr){DBGLOG("FREE(%s)", _str(ptr)); my_free(ptr); ptr = NULL;}}while(0);
|
||||||
|
|
||||||
|
#else
|
||||||
|
|
||||||
|
#define DBGLOG(...)
|
||||||
|
|
||||||
|
#endif
|
||||||
|
|
||||||
|
|
||||||
|
#endif // DEBUG_H__
|
||||||
79
LocCorr_new/dilation.h
Normal file
79
LocCorr_new/dilation.h
Normal file
@ -0,0 +1,79 @@
|
|||||||
|
/*
|
||||||
|
* dilation.h - inner part of function `dilation`
|
||||||
|
*
|
||||||
|
* Copyright 2013 Edward V. Emelianoff <eddy@sao.ru>
|
||||||
|
*
|
||||||
|
* This program is free software; you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation; either version 2 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program; if not, write to the Free Software
|
||||||
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
||||||
|
* MA 02110-1301, USA.
|
||||||
|
*/
|
||||||
|
|
||||||
|
/*
|
||||||
|
* HERE'S NO ANY "FILE-GUARDS" BECAUSE FILE IS MULTIPLY INCLUDED!
|
||||||
|
* I use this because don't want to dive into depths of BOOST
|
||||||
|
*
|
||||||
|
* multi-including with different defines before - is a simplest way to
|
||||||
|
* modify simple code for avoiding extra if-then-else
|
||||||
|
*/
|
||||||
|
|
||||||
|
|
||||||
|
#if ! defined IM_UP && ! defined IM_DOWN // image without upper and lower borders
|
||||||
|
OMP_FOR()
|
||||||
|
for(int y = 1; y < h; y++)
|
||||||
|
#endif
|
||||||
|
{
|
||||||
|
#ifdef IM_UP
|
||||||
|
int y = 0;
|
||||||
|
#endif
|
||||||
|
#ifdef IM_DOWN
|
||||||
|
int y = h;
|
||||||
|
#endif
|
||||||
|
uint8_t *iptr = &image[W0*y];
|
||||||
|
uint8_t *optr = &ret[W0*y];
|
||||||
|
uint8_t p = DIL[*iptr]
|
||||||
|
#ifndef IM_UP
|
||||||
|
| iptr[-W0]
|
||||||
|
#endif
|
||||||
|
#ifndef IM_DOWN
|
||||||
|
| iptr[W0]
|
||||||
|
#endif
|
||||||
|
;
|
||||||
|
if(iptr[1] & 0x80) p |= 1;
|
||||||
|
*optr = p;
|
||||||
|
optr++; iptr++;
|
||||||
|
for(int x = 1; x < w; x++, iptr++, optr++){
|
||||||
|
p = DIL[*iptr]
|
||||||
|
#ifndef IM_UP
|
||||||
|
| iptr[-W0]
|
||||||
|
#endif
|
||||||
|
#ifndef IM_DOWN
|
||||||
|
| iptr[W0]
|
||||||
|
#endif
|
||||||
|
;
|
||||||
|
if(iptr[1] & 0x80) p |= 1;
|
||||||
|
if(iptr[-1] & 1) p |= 0x80;
|
||||||
|
*optr = p;
|
||||||
|
}
|
||||||
|
p = DIL[*iptr]
|
||||||
|
#ifndef IM_UP
|
||||||
|
| iptr[-W0]
|
||||||
|
#endif
|
||||||
|
#ifndef IM_DOWN
|
||||||
|
| iptr[W0]
|
||||||
|
#endif
|
||||||
|
;
|
||||||
|
if(iptr[-1] & 1) p |= 0x80;
|
||||||
|
*optr = p & lastmask; // clear outern pixel
|
||||||
|
}
|
||||||
|
|
||||||
128
LocCorr_new/draw.c
Normal file
128
LocCorr_new/draw.c
Normal file
@ -0,0 +1,128 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
// simplest interface to draw lines & ellipses
|
||||||
|
#include "debug.h"
|
||||||
|
#include "draw.h"
|
||||||
|
#include "fits.h"
|
||||||
|
|
||||||
|
|
||||||
|
// base colors:
|
||||||
|
const uint8_t
|
||||||
|
C_R[] = {255, 0, 0},
|
||||||
|
C_G[] = {0, 255, 0},
|
||||||
|
C_B[] = {0, 0, 255},
|
||||||
|
C_K[] = {0, 0, 0},
|
||||||
|
C_W[] = {255,255,255};
|
||||||
|
|
||||||
|
void Img3_free(il_Img3 **im){
|
||||||
|
if(!im || !*im) return;
|
||||||
|
FREE((*im)->data);
|
||||||
|
FREE(*im);
|
||||||
|
}
|
||||||
|
|
||||||
|
void il_Pattern_free(il_Pattern **p){
|
||||||
|
if(!p || !*p) return;
|
||||||
|
FREE((*p)->data);
|
||||||
|
FREE(*p);
|
||||||
|
}
|
||||||
|
|
||||||
|
// make a single-channel (opaque) mask for cross; allocated here!!!
|
||||||
|
il_Pattern *il_Pattern_cross(int h, int w){
|
||||||
|
int s = h*w, hmid = h/2, wmid = w/2;
|
||||||
|
uint8_t *data = MALLOC(uint8_t, s);
|
||||||
|
il_Pattern *p = MALLOC(il_Pattern, 1);
|
||||||
|
p->data = data;
|
||||||
|
p->h = h; p->w = w;
|
||||||
|
uint8_t *ptr = &data[wmid];
|
||||||
|
for(int y = 0; y < h; ++y, ptr += w) *ptr = 255;
|
||||||
|
ptr = &data[hmid*w];
|
||||||
|
for(int x = 0; x < w; ++x, ++ptr) *ptr = 255;
|
||||||
|
return p;
|
||||||
|
}
|
||||||
|
|
||||||
|
// complicated cross
|
||||||
|
il_Pattern *il_Pattern_xcross(int h, int w){
|
||||||
|
int s = h*w, hmid = h/2, wmid = w/2;
|
||||||
|
uint8_t *data = MALLOC(uint8_t, s);
|
||||||
|
il_Pattern *p = MALLOC(il_Pattern, 1);
|
||||||
|
p->data = data;
|
||||||
|
p->h = h; p->w = w;
|
||||||
|
data[hmid*w + wmid] = 255; // point @ center
|
||||||
|
if(h < 7 || w < 7) return p;
|
||||||
|
int idxy1 = (hmid-3)*w, idxy2 = (hmid+3)*w;
|
||||||
|
int idxx1 = wmid-3, idxx2 = wmid+3;
|
||||||
|
for(int i = 0; i < wmid - 3; ++i){
|
||||||
|
data[idxy1+i] = data[idxy1+w-1-i] = 255;
|
||||||
|
data[idxy2+i] = data[idxy2+w-1-i] = 255;
|
||||||
|
}
|
||||||
|
for(int i = 0; i < hmid - 3; ++i){
|
||||||
|
data[idxx1 + i*w] = data[idxx1 + (h-1-i)*w] = 255;
|
||||||
|
data[idxx2 + i*w] = data[idxx2 + (h-1-i)*w] = 255;
|
||||||
|
}
|
||||||
|
return p;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief draw3_pattern - draw pattern @ 3-channel image
|
||||||
|
* @param img (io) - image
|
||||||
|
* @param p (i) - the pattern
|
||||||
|
* @param xc, yc - coordinates of pattern center @ image
|
||||||
|
* @param colr - color to draw pattern (when opaque == 255)
|
||||||
|
*/
|
||||||
|
void il_Pattern_draw3(il_Img3 *img, const il_Pattern *p, int xc, int yc, const uint8_t colr[]){
|
||||||
|
if(!img || !p) return;
|
||||||
|
int xul = xc - p->w/2, yul = yc - p->h/2;
|
||||||
|
int xdr = xul+p->w-1, ydr = yul+p->h-1;
|
||||||
|
int R = img->w, D = img->h; // right and down border coordinates + 1
|
||||||
|
if(ydr < 0 || xdr < 0 || xul > R-1 || yul > D-1) return; // box outside of image
|
||||||
|
|
||||||
|
int oxlow, oxhigh, oylow, oyhigh; // output limit coordinates
|
||||||
|
int ixlow, iylow; // intput limit coordinates
|
||||||
|
if(xul < 0){
|
||||||
|
oxlow = 0; ixlow = -xul;
|
||||||
|
}else{
|
||||||
|
oxlow = xul; ixlow = 0;
|
||||||
|
}
|
||||||
|
if(yul < 0){
|
||||||
|
oylow = 0; iylow = -yul;
|
||||||
|
}else{
|
||||||
|
oylow = yul; iylow = 0;
|
||||||
|
}
|
||||||
|
if(xdr < R){
|
||||||
|
oxhigh = xdr;
|
||||||
|
}else{
|
||||||
|
oxhigh = R;
|
||||||
|
}
|
||||||
|
if(ydr < D){
|
||||||
|
oyhigh = ydr;
|
||||||
|
}else{
|
||||||
|
oyhigh = D;
|
||||||
|
}
|
||||||
|
OMP_FOR()
|
||||||
|
for(int y = oylow; y < oyhigh; ++y){
|
||||||
|
uint8_t *in = &p->data[(iylow+y-oylow)*p->w + ixlow]; // opaque component
|
||||||
|
uint8_t *out = &img->data[(y*img->w + oxlow)*3]; // 3-colours
|
||||||
|
for(int x = oxlow; x < oxhigh; ++x, ++in, out += 3){
|
||||||
|
float opaque = ((float)*in)/255.;
|
||||||
|
for(int c = 0; c < 3; ++c){
|
||||||
|
out[c] = (uint8_t)(colr[c] * opaque + out[c]*(1.-opaque));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
46
LocCorr_new/draw.h
Normal file
46
LocCorr_new/draw.h
Normal file
@ -0,0 +1,46 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
#pragma once
|
||||||
|
#ifndef DRAW_H__
|
||||||
|
#define DRAW_H__
|
||||||
|
|
||||||
|
#include <stdint.h>
|
||||||
|
|
||||||
|
// 3-channel image for saving into jpg/png
|
||||||
|
typedef struct{
|
||||||
|
uint8_t *data; // image data
|
||||||
|
int w; // width
|
||||||
|
int h; // height
|
||||||
|
} il_Img3;
|
||||||
|
|
||||||
|
// opaque pattern for drawing
|
||||||
|
typedef struct{
|
||||||
|
uint8_t *data;
|
||||||
|
int w;
|
||||||
|
int h;
|
||||||
|
} il_Pattern;
|
||||||
|
|
||||||
|
// base colors
|
||||||
|
extern const uint8_t C_R[], C_G[], C_B[], C_K[], C_W[];
|
||||||
|
|
||||||
|
void il_Pattern_free(il_Pattern **p);
|
||||||
|
void il_Pattern_draw3(il_Img3 *img, const il_Pattern *p, int xc, int yc, const uint8_t colr[]);
|
||||||
|
il_Pattern *il_Pattern_cross(int h, int w);
|
||||||
|
il_Pattern *il_Pattern_xcross(int h, int w);
|
||||||
|
|
||||||
|
#endif // DRAW_H__
|
||||||
102
LocCorr_new/ec_filter.h
Normal file
102
LocCorr_new/ec_filter.h
Normal file
@ -0,0 +1,102 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
// inner part of function `filter8`
|
||||||
|
|
||||||
|
#if ! defined IM_UP && ! defined IM_DOWN
|
||||||
|
OMP_FOR()
|
||||||
|
for(int y = 1; y < h; y++)
|
||||||
|
#endif
|
||||||
|
{
|
||||||
|
#ifdef IM_UP
|
||||||
|
int y = 0;
|
||||||
|
#endif
|
||||||
|
#ifdef IM_DOWN
|
||||||
|
int y = h;
|
||||||
|
#endif
|
||||||
|
uint8_t *iptr = &image[W0*y];
|
||||||
|
uint8_t *optr = &ret[W0*y];
|
||||||
|
// x=0
|
||||||
|
register uint8_t inp = *iptr;
|
||||||
|
register uint8_t p = (inp << 1) | (inp >> 1)
|
||||||
|
#ifndef IM_UP
|
||||||
|
| iptr[-W0] | (iptr[-W0]<<1) | (iptr[-W0]>>1)
|
||||||
|
#endif
|
||||||
|
#ifndef IM_DOWN
|
||||||
|
| iptr[W0] | (iptr[W0]<<1) | (iptr[W0]>>1)
|
||||||
|
#endif
|
||||||
|
;
|
||||||
|
if((iptr[1]
|
||||||
|
#ifndef IM_UP
|
||||||
|
| iptr[-W0+1]
|
||||||
|
#endif
|
||||||
|
#ifndef IM_DOWN
|
||||||
|
| iptr[W0+1]
|
||||||
|
#endif
|
||||||
|
)& 0x80) p |= 1;
|
||||||
|
*optr++ = inp & p;
|
||||||
|
++iptr;
|
||||||
|
for(int x = 1; x < w; ++x){
|
||||||
|
inp = *iptr;
|
||||||
|
p = (inp << 1) | (inp >> 1)
|
||||||
|
#ifndef IM_UP
|
||||||
|
| iptr[-W0] | (iptr[-W0]<<1) | (iptr[-W0]>>1)
|
||||||
|
#endif
|
||||||
|
#ifndef IM_DOWN
|
||||||
|
| iptr[W0] | (iptr[W0]<<1) | (iptr[W0]>>1)
|
||||||
|
#endif
|
||||||
|
;
|
||||||
|
if((iptr[1]
|
||||||
|
#ifndef IM_UP
|
||||||
|
| iptr[-W0+1]
|
||||||
|
#endif
|
||||||
|
#ifndef IM_DOWN
|
||||||
|
| iptr[W0+1]
|
||||||
|
#endif
|
||||||
|
)& 0x80) p |= 1;
|
||||||
|
if((iptr[-1]
|
||||||
|
#ifndef IM_UP
|
||||||
|
| iptr[-W0-1]
|
||||||
|
#endif
|
||||||
|
#ifndef IM_DOWN
|
||||||
|
| iptr[W0-1]
|
||||||
|
#endif
|
||||||
|
)& 1) p |= 0x80;
|
||||||
|
*optr++ = inp & p;
|
||||||
|
++iptr;
|
||||||
|
}
|
||||||
|
// x = W0-1
|
||||||
|
inp = *iptr;
|
||||||
|
p = (inp << 1) | (inp >> 1)
|
||||||
|
#ifndef IM_UP
|
||||||
|
| iptr[-W0] | (iptr[-W0]<<1) | (iptr[-W0]>>1)
|
||||||
|
#endif
|
||||||
|
#ifndef IM_DOWN
|
||||||
|
| iptr[W0] | (iptr[W0]<<1) | (iptr[W0]>>1)
|
||||||
|
#endif
|
||||||
|
;
|
||||||
|
if((iptr[-1]
|
||||||
|
#ifndef IM_UP
|
||||||
|
| iptr[-W0-1]
|
||||||
|
#endif
|
||||||
|
#ifndef IM_DOWN
|
||||||
|
| iptr[W0-1]
|
||||||
|
#endif
|
||||||
|
)& 1) p |= 0x80;
|
||||||
|
*optr = inp & p;
|
||||||
|
}
|
||||||
74
LocCorr_new/fc_filter.h
Normal file
74
LocCorr_new/fc_filter.h
Normal file
@ -0,0 +1,74 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
// inner part of function `filter4`
|
||||||
|
|
||||||
|
#if ! defined IM_UP && ! defined IM_DOWN
|
||||||
|
OMP_FOR()
|
||||||
|
for(int y = 1; y < h; y++)
|
||||||
|
#endif
|
||||||
|
{
|
||||||
|
#ifdef IM_UP
|
||||||
|
int y = 0;
|
||||||
|
#endif
|
||||||
|
#ifdef IM_DOWN
|
||||||
|
int y = h;
|
||||||
|
#endif
|
||||||
|
uint8_t *iptr = &image[W0*y];
|
||||||
|
uint8_t *optr = &ret[W0*y];
|
||||||
|
// x=0
|
||||||
|
register uint8_t inp = *iptr;
|
||||||
|
register uint8_t p = (inp << 1) | (inp >> 1)
|
||||||
|
#ifndef IM_UP
|
||||||
|
| iptr[-W0]
|
||||||
|
#endif
|
||||||
|
#ifndef IM_DOWN
|
||||||
|
| iptr[W0]
|
||||||
|
#endif
|
||||||
|
;
|
||||||
|
if(iptr[1] & 0x80) p |= 1;
|
||||||
|
*optr++ = inp & p;
|
||||||
|
++iptr;
|
||||||
|
for(int x = 1; x < w; ++x){
|
||||||
|
inp = *iptr;
|
||||||
|
p = (inp << 1) | (inp >> 1)
|
||||||
|
#ifndef IM_UP
|
||||||
|
| iptr[-W0]
|
||||||
|
#endif
|
||||||
|
#ifndef IM_DOWN
|
||||||
|
| iptr[W0]
|
||||||
|
#endif
|
||||||
|
;
|
||||||
|
if(iptr[1] & 0x80) p |= 1;
|
||||||
|
if(iptr[-1] & 1) p |= 0x80;
|
||||||
|
*optr++ = inp & p;
|
||||||
|
++iptr;
|
||||||
|
}
|
||||||
|
// x = W0-1
|
||||||
|
inp = *iptr;
|
||||||
|
p = (inp << 1) | (inp >> 1)
|
||||||
|
#ifndef IM_UP
|
||||||
|
| iptr[-W0]
|
||||||
|
#endif
|
||||||
|
#ifndef IM_DOWN
|
||||||
|
| iptr[W0]
|
||||||
|
#endif
|
||||||
|
;
|
||||||
|
if(iptr[-1] & 1) p |= 0x80;
|
||||||
|
*optr = inp & p;
|
||||||
|
}
|
||||||
127
LocCorr_new/fits.c
Normal file
127
LocCorr_new/fits.c
Normal file
@ -0,0 +1,127 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
#include <stdint.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <math.h>
|
||||||
|
|
||||||
|
#include "debug.h"
|
||||||
|
#include "fits.h"
|
||||||
|
|
||||||
|
static int fitsstatus = 0;
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Macros for error processing when working with cfitsio functions
|
||||||
|
*/
|
||||||
|
#define TRYFITS(f, ...) \
|
||||||
|
do{ fitsstatus = 0; \
|
||||||
|
f(__VA_ARGS__, &fitsstatus); \
|
||||||
|
if(fitsstatus){ \
|
||||||
|
fits_report_error(stderr, fitsstatus); \
|
||||||
|
return FALSE;} \
|
||||||
|
}while(0)
|
||||||
|
#define FITSFUN(f, ...) \
|
||||||
|
do{ fitsstatus = 0; \
|
||||||
|
int ret = f(__VA_ARGS__, &fitsstatus); \
|
||||||
|
if(ret || fitsstatus) \
|
||||||
|
fits_report_error(stderr, fitsstatus); \
|
||||||
|
}while(0)
|
||||||
|
#define WRITEKEY(...) \
|
||||||
|
do{ fitsstatus = 0; \
|
||||||
|
fits_write_key(__VA_ARGS__, &fitsstatus); \
|
||||||
|
if(status) fits_report_error(stderr, status);\
|
||||||
|
}while(0)
|
||||||
|
|
||||||
|
void Image_free(Image **img){
|
||||||
|
FREE((*img)->data);
|
||||||
|
FREE(*img);
|
||||||
|
}
|
||||||
|
|
||||||
|
// I->data should be allocated!!!
|
||||||
|
static inline void convflt2ima(float *f, Image *I){
|
||||||
|
if(!I || !I->data || !f) return;
|
||||||
|
float min = *f, max = min;
|
||||||
|
int wh = I->height * I->width;
|
||||||
|
#pragma omp parallel shared(min, max)
|
||||||
|
{
|
||||||
|
float min_p = min, max_p = min;
|
||||||
|
#pragma omp for nowait
|
||||||
|
for(int i = 0; i < wh; ++i){
|
||||||
|
if(f[i] < min_p) min_p = f[i];
|
||||||
|
else if(f[i] > max_p) max_p = f[i];
|
||||||
|
}
|
||||||
|
#pragma omp critical
|
||||||
|
{
|
||||||
|
if(min > min_p) min = min_p;
|
||||||
|
if(max < max_p) max = max_p;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
float W = 255./(max - min);
|
||||||
|
#pragma omp for
|
||||||
|
for(int i = 0; i < wh; ++i){
|
||||||
|
I->data[i] = (Imtype)(W*(f[i] - min));
|
||||||
|
}
|
||||||
|
I->maxval = 255;
|
||||||
|
I->minval = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool FITS_read(const char *filename, Image **fits){
|
||||||
|
FNAME();
|
||||||
|
bool ret = TRUE;
|
||||||
|
fitsfile *fp;
|
||||||
|
int hdunum;
|
||||||
|
int naxis, dtype;
|
||||||
|
long naxes[2];
|
||||||
|
Image *img = MALLOC(Image, 1);
|
||||||
|
|
||||||
|
TRYFITS(fits_open_file, &fp, filename, READONLY);
|
||||||
|
FITSFUN(fits_get_num_hdus, fp, &hdunum);
|
||||||
|
DBG("Got %d HDUs", hdunum);
|
||||||
|
if(hdunum < 1){
|
||||||
|
WARNX(_("Can't read HDU"));
|
||||||
|
ret = FALSE;
|
||||||
|
goto returning;
|
||||||
|
}
|
||||||
|
// get image dimensions
|
||||||
|
TRYFITS(fits_get_img_param, fp, 2, &dtype, &naxis, naxes);
|
||||||
|
DBG("Image have %d axis", naxis);
|
||||||
|
if(naxis > 2){
|
||||||
|
WARNX(_("Images with > 2 dimensions are not supported"));
|
||||||
|
ret = FALSE;
|
||||||
|
goto returning;
|
||||||
|
}
|
||||||
|
img->width = naxes[0];
|
||||||
|
img->height = naxes[1];
|
||||||
|
DBG("got image %ldx%ld pix, bitpix=%d", naxes[0], naxes[1], dtype);
|
||||||
|
size_t sz = naxes[0] * naxes[1];
|
||||||
|
img->data = MALLOC(Imtype, sz);
|
||||||
|
float *targ = MALLOC(float, sz);
|
||||||
|
int stat = 0;
|
||||||
|
TRYFITS(fits_read_img, fp, TFLOAT, 1, sz, NULL, targ, &stat);
|
||||||
|
if(stat) WARNX(_("Found %d pixels with undefined value"), stat);
|
||||||
|
convflt2ima(targ, img);
|
||||||
|
FREE(targ);
|
||||||
|
DBG("ready");
|
||||||
|
|
||||||
|
returning:
|
||||||
|
FITSFUN(fits_close_file, fp);
|
||||||
|
if(!ret || !fits){
|
||||||
|
Image_free(&img);
|
||||||
|
}else *fits = img;
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
32
LocCorr_new/fits.h
Normal file
32
LocCorr_new/fits.h
Normal file
@ -0,0 +1,32 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
#ifndef FITS_H__
|
||||||
|
#define FITS_H__
|
||||||
|
|
||||||
|
#include <fitsio.h>
|
||||||
|
#include <omp.h>
|
||||||
|
#include <stdbool.h>
|
||||||
|
|
||||||
|
#include "imagefile.h"
|
||||||
|
|
||||||
|
void Image_free(Image **ima);
|
||||||
|
bool FITS_read(const char *filename, Image **fits);
|
||||||
|
|
||||||
|
#endif // FITS_H__
|
||||||
275
LocCorr_new/grasshopper.c
Normal file
275
LocCorr_new/grasshopper.c
Normal file
@ -0,0 +1,275 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <C/FlyCapture2_C.h>
|
||||||
|
#include <C/FlyCapture2Defs_C.h>
|
||||||
|
#include <float.h> // FLT_EPSILON
|
||||||
|
#include <math.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <string.h>
|
||||||
|
|
||||||
|
#include "config.h"
|
||||||
|
#include "debug.h"
|
||||||
|
#include "grasshopper.h"
|
||||||
|
#include "imagefile.h"
|
||||||
|
|
||||||
|
static fc2Context context;
|
||||||
|
static fc2PGRGuid guid;
|
||||||
|
static fc2Error err = FC2_ERROR_OK;
|
||||||
|
|
||||||
|
#ifndef Stringify
|
||||||
|
#define Stringify(x) #x
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#define FC2FN(fn, ...) do{err = FC2_ERROR_OK; if(FC2_ERROR_OK != (err=fn(context __VA_OPT__(,) __VA_ARGS__))){ \
|
||||||
|
WARNX(Stringify(fn) "(): %s", fc2ErrorToDescription(err)); return FALSE;}}while(0)
|
||||||
|
|
||||||
|
static void disconnect(){
|
||||||
|
fc2DestroyContext(context);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief setfloat - set absolute property value (float)
|
||||||
|
* @param t - type of property
|
||||||
|
* @param f - new value
|
||||||
|
* @return 1 if all OK
|
||||||
|
*/
|
||||||
|
static int setfloat(fc2PropertyType t, float f){
|
||||||
|
fc2Property prop = {0};
|
||||||
|
prop.type = t;
|
||||||
|
fc2PropertyInfo i = {0};
|
||||||
|
i.type = t;
|
||||||
|
FC2FN(fc2GetProperty, &prop);
|
||||||
|
FC2FN(fc2GetPropertyInfo, &i);
|
||||||
|
if(!prop.present || !i.present) return 0;
|
||||||
|
if(prop.autoManualMode){
|
||||||
|
if(!i.manualSupported){
|
||||||
|
WARNX("Can't set auto-only property");
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
prop.autoManualMode = false;
|
||||||
|
}
|
||||||
|
if(!prop.absControl){
|
||||||
|
if(!i.absValSupported){
|
||||||
|
WARNX("Can't set non-absolute property to absolute value");
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
prop.absControl = true;
|
||||||
|
}
|
||||||
|
if(!prop.onOff){
|
||||||
|
if(!i.onOffSupported){
|
||||||
|
WARNX("Can't set property ON");
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
prop.onOff = true;
|
||||||
|
}
|
||||||
|
if(prop.onePush && i.onePushSupported) prop.onePush = false;
|
||||||
|
prop.valueA = prop.valueB = 0;
|
||||||
|
prop.absValue = f;
|
||||||
|
FC2FN(fc2SetProperty, &prop);
|
||||||
|
// now check
|
||||||
|
FC2FN(fc2GetProperty, &prop);
|
||||||
|
if(fabsf(prop.absValue - f) > 0.02f){
|
||||||
|
WARNX("Can't set property! Got %g instead of %g.", prop.absValue, f);
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int setbrightness(float b){
|
||||||
|
return setfloat(FC2_BRIGHTNESS, b);
|
||||||
|
}
|
||||||
|
static int setexp(float e){
|
||||||
|
return setfloat(FC2_SHUTTER, e);
|
||||||
|
}
|
||||||
|
static int setgain(float e){
|
||||||
|
return setfloat(FC2_GAIN, e);
|
||||||
|
}
|
||||||
|
// TODO: +set saturation, hue, sharpness ?
|
||||||
|
|
||||||
|
|
||||||
|
static int propOnOff(fc2PropertyType t, BOOL onOff){
|
||||||
|
fc2Property prop = {0};
|
||||||
|
prop.type = t;
|
||||||
|
fc2PropertyInfo i = {0};
|
||||||
|
i.type = t;
|
||||||
|
FC2FN(fc2GetPropertyInfo, &i);
|
||||||
|
FC2FN(fc2GetProperty, &prop);
|
||||||
|
if(!prop.present || !i.present) return 0;
|
||||||
|
if(prop.onOff == onOff) return 0;
|
||||||
|
if(!i.onOffSupported){
|
||||||
|
WARNX("Property doesn't support state OFF");
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
prop.onOff = onOff;
|
||||||
|
FC2FN(fc2SetProperty, &prop);
|
||||||
|
FC2FN(fc2GetProperty, &prop);
|
||||||
|
if(prop.onOff != onOff){
|
||||||
|
WARNX("Can't change property OnOff state");
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
#define autoExpOff() propOnOff(FC2_AUTO_EXPOSURE, false)
|
||||||
|
#define whiteBalOff() propOnOff(FC2_WHITE_BALANCE, false)
|
||||||
|
#define gammaOff() propOnOff(FC2_GAMMA, false)
|
||||||
|
#define trigModeOff() propOnOff(FC2_TRIGGER_MODE, false)
|
||||||
|
#define trigDelayOff() propOnOff(FC2_TRIGGER_DELAY, false)
|
||||||
|
#define frameRateOff() propOnOff(FC2_FRAME_RATE, false)
|
||||||
|
|
||||||
|
static int geometrylimits(frameformat *max, frameformat *step){
|
||||||
|
fc2Format7Info f = {.mode = FC2_MODE_0};
|
||||||
|
BOOL b;
|
||||||
|
fc2Format7Info i = {0};
|
||||||
|
FC2FN(fc2GetFormat7Info, &i, &b);
|
||||||
|
if(!b || !max || !step) return FALSE;
|
||||||
|
max->h = f.maxHeight; max->w = f.maxWidth;
|
||||||
|
max->xoff = f.maxWidth - f.offsetHStepSize;
|
||||||
|
max->yoff = f.maxHeight - f.offsetVStepSize;
|
||||||
|
step->w = f.imageHStepSize;
|
||||||
|
step->h = f.imageVStepSize;
|
||||||
|
step->xoff = f.offsetHStepSize;
|
||||||
|
step->yoff = f.offsetVStepSize;
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int getformat(frameformat *fmt){
|
||||||
|
if(!fmt) return FALSE;
|
||||||
|
unsigned int packsz; float pc;
|
||||||
|
fc2Format7ImageSettings f7;
|
||||||
|
FC2FN(fc2GetFormat7Configuration, &f7, &packsz, &pc);
|
||||||
|
fmt->h = f7.height; fmt->w = f7.width;
|
||||||
|
fmt->xoff = f7.offsetX; fmt->yoff = f7.offsetY;
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int changeformat(frameformat *fmt){
|
||||||
|
if(!fmt) return FALSE;
|
||||||
|
BOOL b;
|
||||||
|
fc2Format7ImageSettings f7;
|
||||||
|
f7.mode = FC2_MODE_0;
|
||||||
|
f7.offsetX = fmt->xoff;
|
||||||
|
f7.offsetY = fmt->yoff;
|
||||||
|
f7.width = fmt->w;
|
||||||
|
f7.height = fmt->h;
|
||||||
|
DBG("offx=%d, offy=%d, w=%d, h=%d ", f7.offsetX, f7.offsetY, f7.width, f7.height);
|
||||||
|
f7.pixelFormat = FC2_PIXEL_FORMAT_MONO8;
|
||||||
|
fc2Format7PacketInfo f7p;
|
||||||
|
FC2FN(fc2ValidateFormat7Settings, &f7, &b, &f7p);
|
||||||
|
if(!b) return FALSE; // invalid
|
||||||
|
FC2FN(fc2SetFormat7Configuration, &f7, f7p.recommendedBytesPerPacket);
|
||||||
|
getformat(fmt); // change values to currently used
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int connect(){
|
||||||
|
FNAME();
|
||||||
|
unsigned int numCameras = 0;
|
||||||
|
if(FC2_ERROR_OK != (err = fc2CreateContext(&context))){
|
||||||
|
WARNX("fc2CreateContext(): %s", fc2ErrorToDescription(err));
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
FC2FN(fc2GetNumOfCameras, &numCameras);
|
||||||
|
if(numCameras == 0){
|
||||||
|
WARNX("No cameras detected!");
|
||||||
|
camdisconnect();
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
DBG("Found %d camera[s]", numCameras);
|
||||||
|
if(numCameras > 1){
|
||||||
|
WARNX("Found %d cameras, will use first", numCameras);
|
||||||
|
}
|
||||||
|
FC2FN(fc2GetCameraFromIndex, 0, &guid);
|
||||||
|
FC2FN(fc2Connect, &guid);
|
||||||
|
// turn off all shit
|
||||||
|
autoExpOff();
|
||||||
|
whiteBalOff();
|
||||||
|
gammaOff();
|
||||||
|
trigModeOff();
|
||||||
|
trigDelayOff();
|
||||||
|
frameRateOff();
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int GrabImage(fc2Image *convertedImage){
|
||||||
|
if(!convertedImage) return FALSE;
|
||||||
|
int ret = FALSE;
|
||||||
|
fc2Image rawImage;
|
||||||
|
// start capture
|
||||||
|
FC2FN(fc2StartCapture);
|
||||||
|
err = fc2CreateImage(&rawImage);
|
||||||
|
if(err != FC2_ERROR_OK){
|
||||||
|
WARNX("Error in fc2CreateImage: %s", fc2ErrorToDescription(err));
|
||||||
|
fc2StopCapture(context);
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
// Retrieve the image
|
||||||
|
err = fc2RetrieveBuffer(context, &rawImage);
|
||||||
|
if(err != FC2_ERROR_OK){
|
||||||
|
WARNX("Error in fc2RetrieveBuffer: %s", fc2ErrorToDescription(err));
|
||||||
|
goto rtn;
|
||||||
|
}
|
||||||
|
// Convert image to gray
|
||||||
|
err = fc2ConvertImageTo(FC2_PIXEL_FORMAT_MONO8, &rawImage, convertedImage);
|
||||||
|
if(err != FC2_ERROR_OK){
|
||||||
|
WARNX("Error in fc2ConvertImageTo: %s", fc2ErrorToDescription(err));
|
||||||
|
goto rtn;
|
||||||
|
}
|
||||||
|
ret = TRUE;
|
||||||
|
rtn:
|
||||||
|
fc2StopCapture(context);
|
||||||
|
fc2DestroyImage(&rawImage);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
// capture image, memory allocated HERE
|
||||||
|
static Image *capture(){
|
||||||
|
FNAME();
|
||||||
|
Image *oIma = NULL;
|
||||||
|
fc2Image convertedImage;
|
||||||
|
err = fc2CreateImage(&convertedImage);
|
||||||
|
if(err != FC2_ERROR_OK){
|
||||||
|
WARNX("capture_grasshopper(): can't create image, %s", fc2ErrorToDescription(err));
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
if(!GrabImage(&convertedImage)){
|
||||||
|
WARNX("Can't grab image");
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
oIma = u8toImage(convertedImage.pData, convertedImage.cols, convertedImage.rows, convertedImage.stride);
|
||||||
|
fc2DestroyImage(&convertedImage);
|
||||||
|
return oIma;
|
||||||
|
}
|
||||||
|
|
||||||
|
static float maxgain(){
|
||||||
|
return GAIN_MAX;
|
||||||
|
}
|
||||||
|
|
||||||
|
// exported object
|
||||||
|
camera GrassHopper = {
|
||||||
|
.disconnect = disconnect,
|
||||||
|
.connect = connect,
|
||||||
|
.capture = capture,
|
||||||
|
.setbrightness = setbrightness,
|
||||||
|
.setexp = setexp,
|
||||||
|
.setgain = setgain,
|
||||||
|
.setgeometry = changeformat,
|
||||||
|
.getgeomlimits = geometrylimits,
|
||||||
|
.getmaxgain = maxgain,
|
||||||
|
};
|
||||||
28
LocCorr_new/grasshopper.h
Normal file
28
LocCorr_new/grasshopper.h
Normal file
@ -0,0 +1,28 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
#pragma once
|
||||||
|
#ifndef GRASSHOPPER_H__
|
||||||
|
#define GRASSHOPPER_H__
|
||||||
|
|
||||||
|
#include "cameracapture.h" // `camera`
|
||||||
|
|
||||||
|
#define GRASSHOPPER_CAPT_NAME "grasshopper"
|
||||||
|
|
||||||
|
extern camera GrassHopper;
|
||||||
|
|
||||||
|
#endif // GRASSHOPPER_H__
|
||||||
582
LocCorr_new/imagefile.c
Normal file
582
LocCorr_new/imagefile.c
Normal file
@ -0,0 +1,582 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <dirent.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <sys/types.h>
|
||||||
|
#include <sys/stat.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
|
||||||
|
#include <stb/stb_image.h>
|
||||||
|
#include <stb/stb_image_write.h>
|
||||||
|
|
||||||
|
#include "basler.h"
|
||||||
|
#include "cameracapture.h"
|
||||||
|
#include "cmdlnopts.h"
|
||||||
|
#include "config.h"
|
||||||
|
#include "debug.h"
|
||||||
|
#include "draw.h"
|
||||||
|
#include "fits.h"
|
||||||
|
#include "grasshopper.h"
|
||||||
|
#include "imagefile.h"
|
||||||
|
#include "median.h"
|
||||||
|
|
||||||
|
typedef struct{
|
||||||
|
const char signature[8];
|
||||||
|
uint8_t len;
|
||||||
|
InputType it;
|
||||||
|
} imsign;
|
||||||
|
|
||||||
|
const imsign signatures[] = {
|
||||||
|
{"BM", 2, T_BMP},
|
||||||
|
{"SIMPLE", 6, T_FITS},
|
||||||
|
{{0x1f, 0x8b, 0x08}, 3, T_GZIP},
|
||||||
|
{"GIF8", 4, T_GIF},
|
||||||
|
{{0xff, 0xd8, 0xff, 0xdb}, 4, T_JPEG},
|
||||||
|
{{0xff, 0xd8, 0xff, 0xe0}, 4, T_JPEG},
|
||||||
|
{{0xff, 0xd8, 0xff, 0xe1}, 4, T_JPEG},
|
||||||
|
{{0x89, 0x50, 0x4e, 0x47}, 4, T_PNG},
|
||||||
|
// {{0x49, 0x49, 0x2a, 0x00}, 4, T_TIFF},
|
||||||
|
{"", 0, T_WRONG}
|
||||||
|
};
|
||||||
|
|
||||||
|
#ifdef EBUG
|
||||||
|
static char *hexdmp(const char sig[8]){
|
||||||
|
static char buf[128];
|
||||||
|
char *bptr = buf;
|
||||||
|
bptr += sprintf(bptr, "[ ");
|
||||||
|
for(int i = 0; i < 7; ++i){
|
||||||
|
bptr += sprintf(bptr, "%02X ", (uint8_t)sig[i]);
|
||||||
|
}
|
||||||
|
bptr += sprintf(bptr, "]");
|
||||||
|
return buf;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief imtype - check image type of given file
|
||||||
|
* @param f - opened image file structure
|
||||||
|
* @return image type or T_WRONG
|
||||||
|
*/
|
||||||
|
static InputType imtype(FILE *f){
|
||||||
|
char signature[8];
|
||||||
|
int x = fread(signature, 1, 7, f);
|
||||||
|
DBG("x=%d", x);
|
||||||
|
if(7 != x){
|
||||||
|
WARN("Can't read file signature");
|
||||||
|
return T_WRONG;
|
||||||
|
}
|
||||||
|
signature[7] = 0;
|
||||||
|
const imsign *s = signatures;
|
||||||
|
DBG("Got signature: %s (%s)", hexdmp(signature), signature);
|
||||||
|
while(s->len){
|
||||||
|
DBG("Check %s", s->signature);
|
||||||
|
if(0 == memcmp(s->signature, signature, s->len)){
|
||||||
|
DBG("Found signature %s", s->signature);
|
||||||
|
return s->it;
|
||||||
|
}
|
||||||
|
++s;
|
||||||
|
}
|
||||||
|
return T_WRONG;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief chkinput - check file/directory name
|
||||||
|
* @param name - name of file or directory
|
||||||
|
* @return type of `name`
|
||||||
|
*/
|
||||||
|
InputType chkinput(const char *name){
|
||||||
|
DBG("input name: %s", name);
|
||||||
|
if(0 == strcmp(name, GRASSHOPPER_CAPT_NAME)) return T_CAPT_GRASSHOPPER;
|
||||||
|
else if(0 == strcmp(name, BASLER_CAPT_NAME)) return T_CAPT_BASLER;
|
||||||
|
struct stat fd_stat;
|
||||||
|
stat(name, &fd_stat);
|
||||||
|
if(S_ISDIR(fd_stat.st_mode)){
|
||||||
|
DBG("%s is a directory", name);
|
||||||
|
DIR *d = opendir(name);
|
||||||
|
if(!d){
|
||||||
|
WARN("Can't open directory %s", name);
|
||||||
|
return T_WRONG;
|
||||||
|
}
|
||||||
|
closedir(d);
|
||||||
|
return T_DIRECTORY;
|
||||||
|
}
|
||||||
|
FILE *f = fopen(name, "r");
|
||||||
|
if(!f){
|
||||||
|
WARN("Can't open file %s", name);
|
||||||
|
return T_WRONG;
|
||||||
|
}
|
||||||
|
InputType tp = imtype(f);
|
||||||
|
DBG("Image type of %s is %d", name, tp);
|
||||||
|
fclose(f);
|
||||||
|
return tp;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief u8toImage - convert uint8_t data to Image structure (flipping upside down for FITS coordinates)
|
||||||
|
* @param data - original image data
|
||||||
|
* @param width - image width
|
||||||
|
* @param height - image height
|
||||||
|
* @param stride - image width with alignment
|
||||||
|
* @return Image structure (fully allocated, you can FREE(data) after it)
|
||||||
|
*/
|
||||||
|
Image *u8toImage(const uint8_t *data, int width, int height, int stride){
|
||||||
|
FNAME();
|
||||||
|
Image *outp = Image_new(width, height);
|
||||||
|
// flip image updown for FITS coordinate system
|
||||||
|
OMP_FOR()
|
||||||
|
for(int y = 0; y < height; ++y){
|
||||||
|
Imtype *Out = &outp->data[(height-1-y)*width];
|
||||||
|
const uint8_t *In = &data[y*stride];
|
||||||
|
for(int x = 0; x < width; ++x){
|
||||||
|
*Out++ = (Imtype)(*In++);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Image_minmax(outp);
|
||||||
|
return outp;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief im_load - load image file
|
||||||
|
* @param name - filename
|
||||||
|
* @return Image structure or NULL
|
||||||
|
*/
|
||||||
|
static inline Image *im_load(const char *name){
|
||||||
|
int width, height, channels;
|
||||||
|
uint8_t *img = stbi_load(name, &width, &height, &channels, 1);
|
||||||
|
if(!img){
|
||||||
|
WARNX("Error in loading the image %s\n", name);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
Image *I = u8toImage(img, width, height, width);
|
||||||
|
free(img);
|
||||||
|
return I;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Image_read - read image from any supported file type
|
||||||
|
* @param name - path to image
|
||||||
|
* @return image or NULL if failed
|
||||||
|
*/
|
||||||
|
Image *Image_read(const char *name){
|
||||||
|
InputType tp = chkinput(name);
|
||||||
|
if(tp == T_DIRECTORY || tp == T_WRONG){
|
||||||
|
WARNX("Bad file type to read");
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
Image *outp = NULL;
|
||||||
|
if(tp == T_FITS || tp == T_GZIP){
|
||||||
|
if(!FITS_read(name, &outp)){
|
||||||
|
WARNX("Can't read %s", name);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
}else outp = im_load(name);
|
||||||
|
return outp;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Image_new - allocate memory for new struct Image & Image->data
|
||||||
|
* @param w, h - image size
|
||||||
|
* @return data allocated here
|
||||||
|
*/
|
||||||
|
Image *Image_new(int w, int h){
|
||||||
|
if(w < 1 || h < 1) return NULL;
|
||||||
|
Image *outp = MALLOC(Image, 1);
|
||||||
|
outp->width = w;
|
||||||
|
outp->height = h;
|
||||||
|
outp->data = MALLOC(Imtype, w*h);
|
||||||
|
return outp;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Image_sim - allocate memory for new empty Image with similar size & data type
|
||||||
|
* @param i - sample image
|
||||||
|
* @return data allocated here (with empty keylist & zeros in data)
|
||||||
|
*/
|
||||||
|
Image *Image_sim(const Image *i){
|
||||||
|
if(!i) return NULL;
|
||||||
|
Image *outp = Image_new(i->width, i->height);
|
||||||
|
return outp;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief get_histogram - calculate image histogram
|
||||||
|
* @param I - orig
|
||||||
|
* @return
|
||||||
|
*/
|
||||||
|
size_t *get_histogram(const Image *I){
|
||||||
|
if(!I || !I->data) return NULL;
|
||||||
|
size_t *histogram = MALLOC(size_t, 256);
|
||||||
|
int wh = I->width * I->height;
|
||||||
|
#pragma omp parallel
|
||||||
|
{
|
||||||
|
size_t histogram_private[256] = {0};
|
||||||
|
#pragma omp for nowait
|
||||||
|
for(int i = 0; i < wh; ++i){
|
||||||
|
++histogram_private[I->data[i]];
|
||||||
|
}
|
||||||
|
#pragma omp critical
|
||||||
|
{
|
||||||
|
for(int i = 0; i < 256; ++i) histogram[i] += histogram_private[i];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return histogram;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief calc_background - Simple background calculation by histogram
|
||||||
|
* @param img (i) - input image (here will be modified its top2proc field)
|
||||||
|
* @param bk (o) - background value
|
||||||
|
* @return 0 if error
|
||||||
|
*/
|
||||||
|
int calc_background(const Image *img, Imtype *bk){
|
||||||
|
if(!img || !img->data || !bk) return FALSE;
|
||||||
|
if(img->maxval == img->minval){
|
||||||
|
WARNX("Zero or overilluminated image!");
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
if(theconf.fixedbkg){
|
||||||
|
if(theconf.fixedbkg > img->minval){
|
||||||
|
WARNX("Image values too small");
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
*bk = theconf.fixedbkg;
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
size_t *histogram = get_histogram(img);
|
||||||
|
|
||||||
|
size_t modeidx = 0, modeval = 0;
|
||||||
|
for(int i = 0; i < 256; ++i)
|
||||||
|
if(modeval < histogram[i]){
|
||||||
|
modeval = histogram[i];
|
||||||
|
modeidx = i;
|
||||||
|
}
|
||||||
|
//DBG("Mode=%g @ idx%d (N=%d)", ((Imtype)modeidx / 255.)*ampl, modeidx, modeval);
|
||||||
|
ssize_t diff2[256] = {0};
|
||||||
|
for(int i = 2; i < 254; ++i) diff2[i] = (histogram[i+2]+histogram[i-2]-2*histogram[i])/4;
|
||||||
|
//green("HISTO:\n");
|
||||||
|
//for(int i = 0; i < 256; ++i) printf("%d:\t%d\t%d\n", i, histogram[i], diff2[i]);
|
||||||
|
FREE(histogram);
|
||||||
|
if(modeidx < 2) modeidx = 2;
|
||||||
|
if(modeidx > 253){
|
||||||
|
WARNX("Overilluminated image");
|
||||||
|
return FALSE; // very bad image: overilluminated
|
||||||
|
}
|
||||||
|
size_t borderidx = modeidx;
|
||||||
|
for(int i = modeidx; i < 254; ++i){ // search bend-point by second derivate
|
||||||
|
if(diff2[i] <= 0 && diff2[i+1] <= 0){
|
||||||
|
borderidx = i; break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
//DBG("borderidx=%d -> %d", borderidx, (borderidx+modeidx)/2);
|
||||||
|
//*bk = (borderidx + modeidx) / 2;
|
||||||
|
*bk = borderidx;
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief linear - linear transform for preparing file to save as JPEG or other type (mirror image upside down!)
|
||||||
|
* @param I - input image
|
||||||
|
* @param nchannels - 1 or 3 colour channels
|
||||||
|
* @return allocated here image for jpeg/png storing
|
||||||
|
*/
|
||||||
|
uint8_t *linear(const Image *I, int nchannels){ // only 1 and 3 channels supported!
|
||||||
|
if(!I || !I->data || (nchannels != 1 && nchannels != 3)) return NULL;
|
||||||
|
int width = I->width, height = I->height;
|
||||||
|
size_t stride = width*nchannels, S = height*stride;
|
||||||
|
uint8_t *outp = MALLOC(uint8_t, S);
|
||||||
|
float min = (float)I->minval, max = (float)I->maxval, W = 255./(max - min);
|
||||||
|
//DBG("make linear transform %dx%d, %d channels", I->width, I->height, nchannels);
|
||||||
|
if(nchannels == 3){
|
||||||
|
OMP_FOR()
|
||||||
|
for(int y = 0; y < height; ++y){
|
||||||
|
uint8_t *Out = &outp[(height-1-y)*stride];
|
||||||
|
Imtype *In = &I->data[y*width];
|
||||||
|
for(int x = 0; x < width; ++x){
|
||||||
|
Out[0] = Out[1] = Out[2] = (uint8_t)(W*((float)(*In++) - min));
|
||||||
|
Out += 3;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}else{
|
||||||
|
OMP_FOR()
|
||||||
|
for(int y = 0; y < height; ++y){
|
||||||
|
uint8_t *Out = &outp[(height-1-y)*width];
|
||||||
|
Imtype *In = &I->data[y*width];
|
||||||
|
for(int x = 0; x < width; ++x){
|
||||||
|
*Out++ = (uint8_t)(W*((float)(*In++) - min));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return outp;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief equalize - hystogram equalization (mirror image upside down!)
|
||||||
|
* @param I - input image
|
||||||
|
* @param nchannels - 1 or 3 colour channels
|
||||||
|
* @param throwpart - which part of black pixels (from all amount) to throw away
|
||||||
|
* @return allocated here image for jpeg/png storing
|
||||||
|
*/
|
||||||
|
uint8_t *equalize(const Image *I, int nchannels, double throwpart){
|
||||||
|
if(!I || !I->data || (nchannels != 1 && nchannels != 3)) return NULL;
|
||||||
|
int width = I->width, height = I->height;
|
||||||
|
size_t stride = width*nchannels, S = height*stride;
|
||||||
|
size_t *orig_histo = get_histogram(I); // original hystogram (linear)
|
||||||
|
if(!orig_histo) return NULL;
|
||||||
|
uint8_t *outp = MALLOC(uint8_t, S);
|
||||||
|
uint8_t eq_levls[256] = {0}; // levels to convert: newpix = eq_levls[oldpix]
|
||||||
|
int s = width*height;
|
||||||
|
int Nblack = 0, bpart = (int)(throwpart * (double)s);
|
||||||
|
int startidx;
|
||||||
|
// remove first part of black pixels
|
||||||
|
for(startidx = 0; startidx < 256; ++startidx){
|
||||||
|
Nblack += orig_histo[startidx];
|
||||||
|
if(Nblack >= bpart) break;
|
||||||
|
}
|
||||||
|
++startidx;
|
||||||
|
/* remove last part of white pixels
|
||||||
|
for(stopidx = 255; stopidx > startidx; --stopidx){
|
||||||
|
Nwhite += orig_hysto[stopidx];
|
||||||
|
if(Nwhite >= wpart) break;
|
||||||
|
}*/
|
||||||
|
//DBG("Throw %d (real: %d black) pixels, startidx=%d", bpart, Nblack, startidx);
|
||||||
|
double part = (double)(s + 1. - Nblack) / 256., N = 0.;
|
||||||
|
for(int i = startidx; i < 256; ++i){
|
||||||
|
N += orig_histo[i];
|
||||||
|
eq_levls[i] = (uint8_t)(N/part);
|
||||||
|
}
|
||||||
|
//for(int i = stopidx; i < 256; ++i) eq_levls[i] = 255;
|
||||||
|
#if 0
|
||||||
|
DBG("Original / new histogram");
|
||||||
|
for(int i = 0; i < 256; ++i) printf("%d\t%d\t%d\n", i, orig_hysto[i], eq_levls[i]);
|
||||||
|
#endif
|
||||||
|
if(nchannels == 3){
|
||||||
|
OMP_FOR()
|
||||||
|
for(int y = 0; y < height; ++y){
|
||||||
|
uint8_t *Out = &outp[(height-1-y)*stride];
|
||||||
|
Imtype *In = &I->data[y*width];
|
||||||
|
for(int x = 0; x < width; ++x){
|
||||||
|
Out[0] = Out[1] = Out[2] = eq_levls[*In++];
|
||||||
|
Out += 3;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}else{
|
||||||
|
OMP_FOR()
|
||||||
|
for(int y = 0; y < height; ++y){
|
||||||
|
uint8_t *Out = &outp[(height-1-y)*width];
|
||||||
|
Imtype *In = &I->data[y*width];
|
||||||
|
for(int x = 0; x < width; ++x){
|
||||||
|
*Out++ = eq_levls[*In++];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
FREE(orig_histo);
|
||||||
|
return outp;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Image_write_jpg - save image as JPG file (flipping upside down)
|
||||||
|
* @param I - image
|
||||||
|
* @param name - filename
|
||||||
|
* @param eq == 0 to write linear, != 0 to write equalized image
|
||||||
|
* @return 0 if failed
|
||||||
|
*/
|
||||||
|
int Image_write_jpg(const Image *I, const char *name, int eq){
|
||||||
|
if(!I || !I->data) return 0;
|
||||||
|
uint8_t *outp = NULL;
|
||||||
|
if(eq)
|
||||||
|
outp = equalize(I, 1, theconf.throwpart);
|
||||||
|
else
|
||||||
|
outp = linear(I, 1);
|
||||||
|
if(!outp) return 0;
|
||||||
|
//DBG("Try to write %s", name);
|
||||||
|
char *tmpnm = MALLOC(char, strlen(name) + 5);
|
||||||
|
sprintf(tmpnm, "%s-tmp", name);
|
||||||
|
int r = stbi_write_jpg(tmpnm, I->width, I->height, 1, outp, 95);
|
||||||
|
if(r){
|
||||||
|
if(rename(tmpnm, name)){
|
||||||
|
WARN("rename()");
|
||||||
|
r = 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
FREE(tmpnm);
|
||||||
|
FREE(outp);
|
||||||
|
return r;
|
||||||
|
}
|
||||||
|
|
||||||
|
// calculate extremal values of image data and store them in it
|
||||||
|
void Image_minmax(Image *I){
|
||||||
|
if(!I || !I->data) return;
|
||||||
|
Imtype min = *(I->data), max = min;
|
||||||
|
int wh = I->width * I->height;
|
||||||
|
#ifdef EBUG
|
||||||
|
double t0 = dtime();
|
||||||
|
#endif
|
||||||
|
#pragma omp parallel shared(min, max)
|
||||||
|
{
|
||||||
|
int min_p = min, max_p = min;
|
||||||
|
#pragma omp for nowait
|
||||||
|
for(int i = 0; i < wh; ++i){
|
||||||
|
Imtype pixval = I->data[i];
|
||||||
|
if(pixval < min_p) min_p = pixval;
|
||||||
|
else if(pixval > max_p) max_p = pixval;
|
||||||
|
}
|
||||||
|
#pragma omp critical
|
||||||
|
{
|
||||||
|
if(min > min_p) min = min_p;
|
||||||
|
if(max < max_p) max = max_p;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
I->maxval = max;
|
||||||
|
I->minval = min;
|
||||||
|
DBG("Image_minmax(): Min=%d, Max=%d, time: %gms", min, max, (dtime()-t0)*1e3);
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* =================== CONVERT IMAGE TYPES ===================>
|
||||||
|
*/
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief bin2Im - convert binarized image into floating
|
||||||
|
* @param image - binarized image
|
||||||
|
* @param W, H - its size (in pixels!)
|
||||||
|
* @return Image structure
|
||||||
|
*/
|
||||||
|
Image *bin2Im(const uint8_t *image, int W, int H){
|
||||||
|
Image *ret = Image_new(W, H);
|
||||||
|
int stride = (W + 7) / 8, s1 = (stride*8 == W) ? stride : stride - 1;
|
||||||
|
OMP_FOR()
|
||||||
|
for(int y = 0; y < H; y++){
|
||||||
|
Imtype *optr = &ret->data[y*W];
|
||||||
|
const uint8_t *iptr = &image[y*stride];
|
||||||
|
for(int x = 0; x < s1; x++){
|
||||||
|
register uint8_t inp = *iptr++;
|
||||||
|
for(int i = 0; i < 8; ++i){
|
||||||
|
*optr++ = (inp & 0x80) ? 1. : 0;
|
||||||
|
inp <<= 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
int rest = W - s1*8;
|
||||||
|
if(rest){
|
||||||
|
register uint8_t inp = *iptr;
|
||||||
|
for(int i = 0; i < rest; ++i){
|
||||||
|
*optr++ = (inp & 0x80) ? 1. : 0;
|
||||||
|
inp <<= 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
ret->minval = 0;
|
||||||
|
ret->maxval = 1;
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Convert floatpoint image into pseudo-packed (1 char == 8 pixels), all values > bk will be 1, else - 0
|
||||||
|
* @param im (i) - image to convert
|
||||||
|
* @param stride (o) - new width of image
|
||||||
|
* @param bk - background level (all values < bk will be 0, other will be 1)
|
||||||
|
* @return allocated memory area with "packed" image
|
||||||
|
*/
|
||||||
|
uint8_t *Im2bin(const Image *im, Imtype bk){
|
||||||
|
if(!im) return NULL;
|
||||||
|
int W = im->width, H = im->height;
|
||||||
|
if(W < 2 || H < 2) return NULL;
|
||||||
|
int y, W0 = (W + 7) / 8, s1 = (W/8 == W0) ? W0 : W0 - 1;
|
||||||
|
uint8_t *ret = MALLOC(uint8_t, W0 * H);
|
||||||
|
OMP_FOR()
|
||||||
|
for(y = 0; y < H; ++y){
|
||||||
|
Imtype *iptr = &im->data[y*W];
|
||||||
|
uint8_t *optr = &ret[y*W0];
|
||||||
|
for(int x = 0; x < s1; ++x){
|
||||||
|
register uint8_t o = 0;
|
||||||
|
for(int i = 0; i < 8; ++i){
|
||||||
|
o <<= 1;
|
||||||
|
if(*iptr++ > bk) o |= 1;
|
||||||
|
}
|
||||||
|
*optr++ = o;
|
||||||
|
}
|
||||||
|
int rest = W - s1*8;
|
||||||
|
if(rest){
|
||||||
|
register uint8_t o = 0;
|
||||||
|
for(int x = 0; x < rest; ++x){
|
||||||
|
o <<= 1;
|
||||||
|
if(*iptr++ > bk) o |= 1;
|
||||||
|
}
|
||||||
|
*optr = o << (8 - rest);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
#if 0
|
||||||
|
UNUSED function! Need to be refactored
|
||||||
|
// convert size_t labels into Image
|
||||||
|
Image *ST2Im(const size_t *image, int W, int H){
|
||||||
|
Image *ret = Image_new(W, H);
|
||||||
|
OMP_FOR()
|
||||||
|
for(int y = 0; y < H; ++y){
|
||||||
|
Imtype *optr = &ret->data[y*W];
|
||||||
|
const size_t *iptr = &image[y*W];
|
||||||
|
for(int x = 0; x < W; ++x){
|
||||||
|
*optr++ = (Imtype)*iptr++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Image_minmax(ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Convert "packed" image into size_t array for conncomp procedure
|
||||||
|
* @param image (i) - input image
|
||||||
|
* @param W, H - size of image in pixels
|
||||||
|
* @return allocated memory area with copy of an image
|
||||||
|
*/
|
||||||
|
size_t *bin2ST(const uint8_t *image, int W, int H){
|
||||||
|
size_t *ret = MALLOC(size_t, W * H);
|
||||||
|
int W0 = (W + 7) / 8, s1 = W0 - 1;
|
||||||
|
OMP_FOR()
|
||||||
|
for(int y = 0; y < H; y++){
|
||||||
|
size_t *optr = &ret[y*W];
|
||||||
|
const uint8_t *iptr = &image[y*W0];
|
||||||
|
for(int x = 0; x < s1; ++x){
|
||||||
|
register uint8_t inp = *iptr++;
|
||||||
|
for(int i = 0; i < 8; ++i){
|
||||||
|
*optr++ = (inp & 0x80) ? 1 : 0;
|
||||||
|
inp <<= 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
int rest = W - s1*8;
|
||||||
|
if(rest){
|
||||||
|
register uint8_t inp = *iptr;
|
||||||
|
for(int i = 0; i < rest; ++i){
|
||||||
|
*optr++ = (inp & 0x80) ? 1 : 0;
|
||||||
|
inp <<= 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
* <=================== CONVERT IMAGE TYPES ===================
|
||||||
|
*/
|
||||||
72
LocCorr_new/imagefile.h
Normal file
72
LocCorr_new/imagefile.h
Normal file
@ -0,0 +1,72 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
#pragma once
|
||||||
|
#ifndef IMAGEFILE_H__
|
||||||
|
#define IMAGEFILE_H__
|
||||||
|
|
||||||
|
#include <stdint.h>
|
||||||
|
#include <stdlib.h> // size_t
|
||||||
|
|
||||||
|
#ifndef OMP_FOR
|
||||||
|
#define Stringify(x) #x
|
||||||
|
#define OMP_FOR(x) _Pragma(Stringify(omp parallel for x))
|
||||||
|
#endif
|
||||||
|
|
||||||
|
typedef uint8_t Imtype; // maybe float or double only
|
||||||
|
|
||||||
|
typedef struct{
|
||||||
|
int width; // width
|
||||||
|
int height; // height
|
||||||
|
Imtype *data; // picture data
|
||||||
|
Imtype minval; // extremal data values
|
||||||
|
Imtype maxval;
|
||||||
|
} Image;
|
||||||
|
|
||||||
|
// input file/directory type
|
||||||
|
typedef enum{
|
||||||
|
T_WRONG,
|
||||||
|
T_DIRECTORY,
|
||||||
|
T_BMP,
|
||||||
|
T_FITS,
|
||||||
|
T_GZIP,
|
||||||
|
T_GIF,
|
||||||
|
T_JPEG,
|
||||||
|
T_PNG,
|
||||||
|
T_CAPT_GRASSHOPPER, // capture grasshopper
|
||||||
|
T_CAPT_BASLER
|
||||||
|
} InputType;
|
||||||
|
|
||||||
|
void Image_minmax(Image *I);
|
||||||
|
uint8_t *linear(const Image *I, int nchannels);
|
||||||
|
uint8_t *equalize(const Image *I, int nchannels, double throwpart);
|
||||||
|
InputType chkinput(const char *name);
|
||||||
|
Image *Image_read(const char *name);
|
||||||
|
Image *Image_new(int w, int h);
|
||||||
|
Image *Image_sim(const Image *i);
|
||||||
|
void Image_free(Image **I);
|
||||||
|
int Image_write_jpg(const Image *I, const char *name, int equalize);
|
||||||
|
size_t *get_histogram(const Image *I);
|
||||||
|
int calc_background(const Image *img, Imtype *bk);
|
||||||
|
|
||||||
|
Image *u8toImage(const uint8_t *data, int width, int height, int stride);
|
||||||
|
Image *bin2Im(const uint8_t *image, int W, int H);
|
||||||
|
uint8_t *Im2bin(const Image *im, Imtype bk);
|
||||||
|
size_t *bin2ST(const uint8_t *image, int W, int H);
|
||||||
|
//Image *ST2Im(const size_t *image, int W, int H);
|
||||||
|
|
||||||
|
#endif // IMAGEFILE_H__
|
||||||
351
LocCorr_new/improc.c
Normal file
351
LocCorr_new/improc.c
Normal file
@ -0,0 +1,351 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <math.h>
|
||||||
|
#include <stb/stb_image_write.h>
|
||||||
|
#include <stdbool.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <time.h>
|
||||||
|
|
||||||
|
#include "basler.h"
|
||||||
|
#include "binmorph.h"
|
||||||
|
#include "cameracapture.h"
|
||||||
|
#include "cmdlnopts.h"
|
||||||
|
#include "config.h"
|
||||||
|
#include "debug.h"
|
||||||
|
#include "draw.h"
|
||||||
|
#include "grasshopper.h"
|
||||||
|
#include "fits.h"
|
||||||
|
#include "improc.h"
|
||||||
|
#include "inotify.h"
|
||||||
|
#include "median.h"
|
||||||
|
#include "steppers.h"
|
||||||
|
|
||||||
|
volatile atomic_ullong ImNumber = 0; // GLOBAL: counter of processed images
|
||||||
|
volatile atomic_bool stopwork = FALSE; // GLOBAL: suicide
|
||||||
|
|
||||||
|
// GLOBAL: get image information
|
||||||
|
char *(*imagedata)(const char *messageid, char *buf, int buflen) = NULL;
|
||||||
|
|
||||||
|
static FILE *fXYlog = NULL;
|
||||||
|
static double tstart = 0.; // time of logging start
|
||||||
|
static double FPS = 0.; // frames per second
|
||||||
|
static float xc = -1., yc = -1.; // center coordinates
|
||||||
|
|
||||||
|
typedef struct{
|
||||||
|
uint32_t area; // object area in pixels
|
||||||
|
double Isum; // total object's intensity over background
|
||||||
|
double WdivH; // width of object's box divided by height
|
||||||
|
double xc; // centroid coordinates
|
||||||
|
double yc;
|
||||||
|
double xsigma; // STD by horizontal and vertical axes
|
||||||
|
double ysigma;
|
||||||
|
} object;
|
||||||
|
|
||||||
|
// functions for Qsort
|
||||||
|
static int compIntens(const void *a, const void *b){ // compare by intensity
|
||||||
|
const object *oa = (const object*)a;
|
||||||
|
const object *ob = (const object*)b;
|
||||||
|
double idiff = (oa->Isum - ob->Isum)/(oa->Isum + ob->Isum);
|
||||||
|
if(fabs(idiff) > theconf.intensthres) return (idiff > 0) ? -1:1;
|
||||||
|
double r2a = oa->xc * oa->xc + oa->yc * oa->yc;
|
||||||
|
double r2b = ob->xc * ob->xc + ob->yc * ob->yc;
|
||||||
|
return (r2a < r2b) ? -1 : 1;
|
||||||
|
}
|
||||||
|
static int compDist(const void *a, const void *b){ // compare by distanse from target
|
||||||
|
const object *oa = (const object*)a;
|
||||||
|
const object *ob = (const object*)b;
|
||||||
|
double xtg = theconf.xtarget - theconf.xoff, ytg = theconf.ytarget - theconf.yoff;
|
||||||
|
double xa = oa->xc - xtg, xb = ob->xc - xtg,
|
||||||
|
ya = oa->yc - ytg, yb = ob->yc - ytg;
|
||||||
|
double r2a = xa*xa + ya*ya;
|
||||||
|
double r2b = xb*xb + yb*yb;
|
||||||
|
return (r2a < r2b) ? -1 : 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void XYnewline(){
|
||||||
|
if(!fXYlog) return;
|
||||||
|
fprintf(fXYlog, "\n");
|
||||||
|
fflush(fXYlog);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void getDeviation(object *curobj){
|
||||||
|
int averflag = 0;
|
||||||
|
static double Xc[NAVER_MAX+1], Yc[NAVER_MAX+1];
|
||||||
|
double xx = curobj->xc, yy = curobj->yc, xsum2 = 0., ysum2 = 0.;
|
||||||
|
double Sx = 0., Sy = 0.;
|
||||||
|
static int counter = 0;
|
||||||
|
Xc[counter] = curobj->xc; Yc[counter] = curobj->yc;
|
||||||
|
if(fXYlog){ // make log record
|
||||||
|
fprintf(fXYlog, "%.2f\t%.1f\t%.1f\t%.1f\t%.1f\t%.1f\t",
|
||||||
|
dtime(), curobj->xc, curobj->yc,
|
||||||
|
curobj->xsigma, curobj->ysigma, curobj->WdivH);
|
||||||
|
}
|
||||||
|
//DBG("counter = %d", counter);
|
||||||
|
if(++counter < theconf.naverage){
|
||||||
|
goto process_corrections;
|
||||||
|
}
|
||||||
|
// it's time to calculate average deviations
|
||||||
|
xx = 0.; yy = 0.;
|
||||||
|
for(int i = 0; i < counter; ++i){
|
||||||
|
double x = Xc[i], y = Yc[i];
|
||||||
|
xx += x; yy += y;
|
||||||
|
xsum2 += x*x; ysum2 += y*y;
|
||||||
|
}
|
||||||
|
xx /= counter; yy /= counter;
|
||||||
|
Sx = sqrt(xsum2/counter - xx*xx);
|
||||||
|
Sy = sqrt(ysum2/counter - yy*yy);
|
||||||
|
counter = 0;
|
||||||
|
#ifdef EBUG
|
||||||
|
green("\n Average centroid: X=%.1f (+-%.1f), Y=%.1f (+-%.1f)\n", xx, Sx, yy, Sy);
|
||||||
|
#endif
|
||||||
|
LOGDBG("getDeviation(): Average centroid: X=%.1f (+-%.1f), Y=%.1f (+-%.1f)", xx, Sx, yy, Sy);
|
||||||
|
averflag = 1;
|
||||||
|
if(fXYlog) fprintf(fXYlog, "%.1f\t%.1f\t%.1f\t%.1f", xx, yy, Sx, Sy);
|
||||||
|
process_corrections:
|
||||||
|
if(theSteppers && theSteppers->proc_corr && averflag){
|
||||||
|
if(Sx > XY_TOLERANCE || Sy > XY_TOLERANCE){
|
||||||
|
LOGDBG("Bad value - not process"); // don't run processing for bad data
|
||||||
|
}else
|
||||||
|
theSteppers->proc_corr(xx, yy);
|
||||||
|
}else{
|
||||||
|
LOGERR("Lost connection with stepper server");
|
||||||
|
ERRX("Lost connection with stepper server");
|
||||||
|
}
|
||||||
|
XYnewline();
|
||||||
|
}
|
||||||
|
|
||||||
|
void process_file(Image *I){
|
||||||
|
static double lastTproc = 0.;
|
||||||
|
/*
|
||||||
|
#ifdef EBUG
|
||||||
|
double t0 = dtime(), tlast = t0;
|
||||||
|
#define DELTA(p) do{double t = dtime(); DBG("---> %s @ %gms (delta: %gms)", p, (t-t0)*1e3, (t-tlast)*1e3); tlast = t;}while(0)
|
||||||
|
#else
|
||||||
|
*/
|
||||||
|
#define DELTA(x)
|
||||||
|
//#endif
|
||||||
|
// I - original image
|
||||||
|
// mean - local mean
|
||||||
|
// std - local STD
|
||||||
|
/**** read original image ****/
|
||||||
|
DELTA("Imread");
|
||||||
|
if(!I){
|
||||||
|
WARNX("No image");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
int W = I->width, H = I->height;
|
||||||
|
//save_fits(I, "fitsout.fits");
|
||||||
|
//DELTA("Save original");
|
||||||
|
Imtype bk;
|
||||||
|
if(calc_background(I, &bk)){
|
||||||
|
DBG("backgr = %d", bk);
|
||||||
|
DELTA("Got background");
|
||||||
|
uint8_t *ibin = Im2bin(I, bk);
|
||||||
|
DELTA("Made binary");
|
||||||
|
if(ibin){
|
||||||
|
//savebin(ibin, W, H, "binary.fits");
|
||||||
|
//DELTA("save binary.fits");
|
||||||
|
uint8_t *er = il_erosionN(ibin, W, H, theconf.Nerosions);
|
||||||
|
FREE(ibin);
|
||||||
|
DELTA("Erosion");
|
||||||
|
//savebin(er, W, H, "erosion.fits");
|
||||||
|
//DELTA("Save erosion");
|
||||||
|
uint8_t *opn = il_dilationN(er, W, H, theconf.Ndilations);
|
||||||
|
FREE(er);
|
||||||
|
DELTA("Opening");
|
||||||
|
//savebin(opn, W, H, "opening.fits");
|
||||||
|
//DELTA("Save opening");
|
||||||
|
il_ConnComps *cc = NULL;
|
||||||
|
size_t *S = il_cclabel4(opn, W, H, &cc);
|
||||||
|
FREE(opn);
|
||||||
|
if(cc->Nobj > 1){ // Nobj = amount of objects + 1
|
||||||
|
DBGLOG("Nobj=%d", cc->Nobj-1);
|
||||||
|
object *Objects = MALLOC(object, cc->Nobj-1);
|
||||||
|
int objctr = 0;
|
||||||
|
for(size_t i = 1; i < cc->Nobj; ++i){
|
||||||
|
il_Box *b = &cc->boxes[i];
|
||||||
|
double wh = ((double)b->xmax - b->xmin)/(b->ymax - b->ymin);
|
||||||
|
//DBG("Obj# %zd: wh=%g, area=%d", i, wh, b->area);
|
||||||
|
if(wh < theconf.minwh || wh > theconf.maxwh) continue;
|
||||||
|
if((int)b->area < theconf.minarea || (int)b->area > theconf.maxarea) continue;
|
||||||
|
double xc = 0., yc = 0.;
|
||||||
|
double x2c = 0., y2c = 0., Isum = 0.;
|
||||||
|
for(size_t y = b->ymin; y <= b->ymax; ++y){
|
||||||
|
size_t idx = y*W + b->xmin;
|
||||||
|
size_t *maskptr = &S[idx];
|
||||||
|
Imtype *Iptr = &I->data[idx];
|
||||||
|
for(size_t x = b->xmin; x <= b->xmax; ++x, ++maskptr, ++Iptr){
|
||||||
|
if(*maskptr != i) continue;
|
||||||
|
double intens = (double) (*Iptr - bk);
|
||||||
|
if(intens < 0.) continue;
|
||||||
|
double xw = x * intens, yw = y * intens;
|
||||||
|
xc += xw;
|
||||||
|
yc += yw;
|
||||||
|
x2c += xw * x;
|
||||||
|
y2c += yw * y;
|
||||||
|
Isum += intens;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
xc /= Isum; yc /= Isum;
|
||||||
|
x2c = x2c/Isum - xc*xc;
|
||||||
|
y2c = y2c/Isum - yc*yc;
|
||||||
|
Objects[objctr++] = (object){
|
||||||
|
.area = b->area, .Isum = Isum,
|
||||||
|
.WdivH = wh, .xc = xc, .yc = yc,
|
||||||
|
.xsigma = sqrt(x2c), .ysigma = sqrt(y2c)
|
||||||
|
};
|
||||||
|
}
|
||||||
|
DELTA("Labeling");
|
||||||
|
DBGLOG("T%.2f, N=%d\n", dtime(), objctr);
|
||||||
|
if(objctr > 1){
|
||||||
|
if(theconf.starssort)
|
||||||
|
qsort(Objects, objctr, sizeof(object), compIntens);
|
||||||
|
else
|
||||||
|
qsort(Objects, objctr, sizeof(object), compDist);
|
||||||
|
}
|
||||||
|
if(objctr){
|
||||||
|
#ifdef EBUG
|
||||||
|
object *o = Objects;
|
||||||
|
green("%6s\t%6s\t%6s\t%6s\t%6s\t%6s\t%6s\t%6s\t%8s\n",
|
||||||
|
"N", "Area", "Mv", "W/H", "Xc", "Yc", "Sx", "Sy", "Area/r^2");
|
||||||
|
for(int i = 0; i < objctr; ++i, ++o){
|
||||||
|
// 1.0857 = 2.5/ln(10)
|
||||||
|
printf("%6d\t%6d\t%6.1f\t%6.1f\t%6.1f\t%6.1f\t%6.1f\t%6.1f\t%8.1f\n",
|
||||||
|
i, o->area, 20.-1.0857*log(o->Isum), o->WdivH, o->xc, o->yc,
|
||||||
|
o->xsigma, o->ysigma, o->area/o->xsigma/o->ysigma);
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
getDeviation(Objects); // calculate dX/dY and process corrections
|
||||||
|
}
|
||||||
|
{ // prepare image and save jpeg
|
||||||
|
uint8_t *outp = NULL;
|
||||||
|
if(theconf.equalize)
|
||||||
|
outp = equalize(I, 3, theconf.throwpart);
|
||||||
|
else
|
||||||
|
outp = linear(I, 3);
|
||||||
|
static il_Pattern *cross = NULL, *crossL = NULL;
|
||||||
|
if(!cross) cross = il_Pattern_xcross(33, 33);
|
||||||
|
if(!crossL) crossL = il_Pattern_xcross(51, 51);
|
||||||
|
il_Img3 i3 = {.data = outp, .w = I->width, .h = H};
|
||||||
|
// draw fiber center position
|
||||||
|
il_Pattern_draw3(&i3, crossL, theconf.xtarget-theconf.xoff, H-(theconf.ytarget-theconf.yoff), C_R);
|
||||||
|
if(objctr){ // draw crosses @ objects' centers
|
||||||
|
int H = I->height;
|
||||||
|
// draw current star centroid
|
||||||
|
il_Pattern_draw3(&i3, cross, Objects[0].xc, H-Objects[0].yc, C_G);
|
||||||
|
// add offset to show in target system
|
||||||
|
xc = Objects[0].xc + theconf.xoff;
|
||||||
|
yc = Objects[0].yc + theconf.yoff;
|
||||||
|
// draw other centroids
|
||||||
|
for(int i = 1; i < objctr; ++i)
|
||||||
|
il_Pattern_draw3(&i3, cross, Objects[i].xc, H-Objects[i].yc, C_B);
|
||||||
|
}else{xc = -1.; yc = -1.;}
|
||||||
|
char *tmpnm = MALLOC(char, strlen(GP->outputjpg) + 5);
|
||||||
|
sprintf(tmpnm, "%s-tmp", GP->outputjpg);
|
||||||
|
if(stbi_write_jpg(tmpnm, I->width, I->height, 3, outp, 95)){
|
||||||
|
if(rename(tmpnm, GP->outputjpg)){
|
||||||
|
WARN("rename()");
|
||||||
|
LOGWARN("can't save %s", GP->outputjpg);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
FREE(tmpnm);
|
||||||
|
FREE(outp);
|
||||||
|
}
|
||||||
|
FREE(Objects);
|
||||||
|
}else{
|
||||||
|
xc = -1.; yc = -1.;
|
||||||
|
Image_write_jpg(I, GP->outputjpg, theconf.equalize);
|
||||||
|
}
|
||||||
|
DBGLOG("Image saved");
|
||||||
|
FREE(S);
|
||||||
|
FREE(cc);
|
||||||
|
}
|
||||||
|
}else Image_write_jpg(I, GP->outputjpg, theconf.equalize);
|
||||||
|
++ImNumber;
|
||||||
|
if(lastTproc > 1.) FPS = 1. / (dtime() - lastTproc);
|
||||||
|
lastTproc = dtime();
|
||||||
|
DELTA("End");
|
||||||
|
}
|
||||||
|
|
||||||
|
static char *localimages(const char *messageid, int isdir, char *buf, int buflen){
|
||||||
|
static char *impath = NULL;
|
||||||
|
if(!impath){
|
||||||
|
if(!realpath(GP->outputjpg, impath)) impath = strdup(GP->outputjpg);
|
||||||
|
}
|
||||||
|
snprintf(buf, buflen, "{ \"%s\": \"%s\", \"camstatus\": \"watch %s\", \"impath\": \"%s\", \"xcenter\": %.1f, \"ycenter\": %.1f }",
|
||||||
|
MESSAGEID, messageid, isdir ? "directory" : "file", impath, xc, yc);
|
||||||
|
return buf;
|
||||||
|
}
|
||||||
|
static char *watchdr(const char *messageid, char *buf, int buflen){
|
||||||
|
return localimages(messageid, 1, buf, buflen);
|
||||||
|
}
|
||||||
|
static char *watchfl(const char *messageid, char *buf, int buflen){
|
||||||
|
return localimages(messageid, 0, buf, buflen);
|
||||||
|
}
|
||||||
|
|
||||||
|
int process_input(InputType tp, char *name){
|
||||||
|
DBG("process_input(%d, %s)", tp, name);
|
||||||
|
if(tp == T_DIRECTORY){
|
||||||
|
imagedata = watchdr;
|
||||||
|
return watch_directory(name, process_file);
|
||||||
|
}else if(tp == T_CAPT_GRASSHOPPER || tp == T_CAPT_BASLER){
|
||||||
|
camera *cam = &GrassHopper;
|
||||||
|
if(tp == T_CAPT_BASLER) cam = &Basler;
|
||||||
|
if(!setCamera(cam)){
|
||||||
|
WARNX("The camera disconnected");
|
||||||
|
LOGWARN("The camera disconnected");
|
||||||
|
}
|
||||||
|
imagedata = camstatus;
|
||||||
|
return camcapture(process_file);
|
||||||
|
}
|
||||||
|
imagedata = watchfl;
|
||||||
|
return watch_file(name, process_file);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief openXYlog - open file to log XY values
|
||||||
|
* @param name - filename
|
||||||
|
*/
|
||||||
|
void openXYlog(const char *name){
|
||||||
|
closeXYlog();
|
||||||
|
fXYlog = fopen(name, "a");
|
||||||
|
if(!fXYlog){
|
||||||
|
char *e = strerror(errno);
|
||||||
|
WARNX("Can't create file %s: %s", name, e);
|
||||||
|
LOGERR("Can't create file %s: %s", name, e);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
time_t t = time(NULL);
|
||||||
|
fprintf(fXYlog, "# Start at: %s", ctime(&t));
|
||||||
|
fprintf(fXYlog, "# time Xc\tYc\t\tSx\tSy\tW/H\taverX\taverY\tSX\tSY\n");
|
||||||
|
fflush(fXYlog);
|
||||||
|
tstart = dtime();
|
||||||
|
}
|
||||||
|
void closeXYlog(){
|
||||||
|
if(!fXYlog) return;
|
||||||
|
fclose(fXYlog);
|
||||||
|
fXYlog = NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
double getFramesPerS(){ return FPS; }
|
||||||
|
|
||||||
|
void getcenter(float *x, float *y){
|
||||||
|
if(x) *x = xc;
|
||||||
|
if(y) *y = yc;
|
||||||
|
}
|
||||||
41
LocCorr_new/improc.h
Normal file
41
LocCorr_new/improc.h
Normal file
@ -0,0 +1,41 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
#pragma once
|
||||||
|
#ifndef IMPROC_H__
|
||||||
|
#define IMPROC_H__
|
||||||
|
|
||||||
|
#include <stdatomic.h>
|
||||||
|
|
||||||
|
#include "imagefile.h"
|
||||||
|
|
||||||
|
// tolerance of deviations by X and Y axis (if sigmaX or sigmaY greater, values considered to be wrong)
|
||||||
|
#define XY_TOLERANCE (5.)
|
||||||
|
|
||||||
|
extern volatile atomic_bool stopwork;
|
||||||
|
extern volatile atomic_ullong ImNumber;
|
||||||
|
|
||||||
|
extern char *(*imagedata)(const char *messageid, char *buf, int buflen);
|
||||||
|
|
||||||
|
void process_file(Image *I);
|
||||||
|
int process_input(InputType tp, char *name);
|
||||||
|
void openXYlog(const char *name);
|
||||||
|
void closeXYlog();
|
||||||
|
double getFramesPerS();
|
||||||
|
void getcenter(float *x, float *y);
|
||||||
|
|
||||||
|
#endif // IMPROC_H__
|
||||||
129
LocCorr_new/inotify.c
Normal file
129
LocCorr_new/inotify.c
Normal file
@ -0,0 +1,129 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <errno.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <string.h> // strlen
|
||||||
|
#include <sys/inotify.h>
|
||||||
|
#include <sys/select.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
|
||||||
|
#include "debug.h"
|
||||||
|
#include "imagefile.h"
|
||||||
|
|
||||||
|
static char filenm[FILENAME_MAX];
|
||||||
|
|
||||||
|
// handle inotify event IN_CLOSE_WRITE
|
||||||
|
/**
|
||||||
|
* @brief handle_events - handle inotify event IN_CLOSE_WRITE
|
||||||
|
* @param name - directory or file name
|
||||||
|
* @param fd - inotify init descriptor
|
||||||
|
* @return 1 if file was changed and ready to be processed, 0 if not changed, -1 if removed
|
||||||
|
*/
|
||||||
|
static int changed(const char *name, int fd, uint32_t mask){
|
||||||
|
struct inotify_event buf[10] = {0};
|
||||||
|
ssize_t len = read(fd, buf, sizeof buf);
|
||||||
|
if(len == -1 && errno != EAGAIN){
|
||||||
|
WARN("inotify read()");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
if(len < 1) return 0; // not ready
|
||||||
|
uint32_t bm = buf[0].mask;
|
||||||
|
buf[0].mask = 0;
|
||||||
|
for(int i = 0; i < 10; ++i)
|
||||||
|
DBG("CHANGED: %s (%d)\n", buf[i].name, bm);
|
||||||
|
if(bm & mask){
|
||||||
|
if(name){
|
||||||
|
snprintf(filenm, FILENAME_MAX, "%s/%s", name, buf[0].name); // full path
|
||||||
|
}else{
|
||||||
|
snprintf(filenm, FILENAME_MAX, "%s", buf[0].name); // file name
|
||||||
|
}
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
DBG("IN_IGNORED");
|
||||||
|
return -1; // IN_IGNORED (file removed)
|
||||||
|
}
|
||||||
|
|
||||||
|
static int initinot(const char *name, int *filed, uint32_t mask){
|
||||||
|
static int fd = -1, wd = -1;
|
||||||
|
if(wd > -1){
|
||||||
|
inotify_rm_watch(fd, wd);
|
||||||
|
wd = -1; fd = -1;
|
||||||
|
}
|
||||||
|
if(fd == -1) fd = inotify_init();//1(IN_NONBLOCK);
|
||||||
|
if(fd == -1){
|
||||||
|
WARN("inotify_init()");
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
if(-1 == (wd = inotify_add_watch(fd, name, mask))){
|
||||||
|
WARN("inotify_add_watch()");
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
if(filed) *filed = fd;
|
||||||
|
return wd;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int watch_any(const char *name, void (*process)(Image*), uint32_t mask){
|
||||||
|
int fd = -1, wd = -1;
|
||||||
|
while(1){
|
||||||
|
if(wd < 1 || fd < 1){
|
||||||
|
wd = initinot(name, &fd, mask);
|
||||||
|
if(wd < 1){
|
||||||
|
usleep(1000);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
int ch;
|
||||||
|
if(mask == IN_CLOSE_WRITE)
|
||||||
|
ch = changed(NULL, fd, mask); // only file name
|
||||||
|
else
|
||||||
|
ch = changed(name, fd, mask); // full path
|
||||||
|
if(ch == -1){ sleep(1); wd = -1; continue; } // removed
|
||||||
|
if(ch == 1){ // changed
|
||||||
|
if(process){
|
||||||
|
Image *I = Image_read(filenm);
|
||||||
|
process(I);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
close(fd);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
int watch_file(const char *name, void (*process)(Image*)){
|
||||||
|
FNAME();
|
||||||
|
if(!name){
|
||||||
|
WARNX("Need filename");
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
return watch_any(name, process, IN_CLOSE_WRITE);
|
||||||
|
}
|
||||||
|
|
||||||
|
int watch_directory(char *name, void (*process)(Image*)){
|
||||||
|
FNAME();
|
||||||
|
if(!name){
|
||||||
|
WARNX("Need directory name");
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
int l = strlen(name) - 1;
|
||||||
|
if(name[l] == '/') name[l] = 0;
|
||||||
|
return watch_any(name, process, IN_CLOSE_WRITE|IN_ISDIR);
|
||||||
|
}
|
||||||
28
LocCorr_new/inotify.h
Normal file
28
LocCorr_new/inotify.h
Normal file
@ -0,0 +1,28 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
#ifndef INOTIFY_H__
|
||||||
|
#define INOTIFY_H__
|
||||||
|
|
||||||
|
#include "fits.h" // Image*
|
||||||
|
|
||||||
|
int watch_file(const char *name, void (*process)(Image*));
|
||||||
|
int watch_directory(char *name, void (*process)(Image*));
|
||||||
|
|
||||||
|
#endif // INOTIFY_H__
|
||||||
36
LocCorr_new/loccorr.conf
Normal file
36
LocCorr_new/loccorr.conf
Normal file
@ -0,0 +1,36 @@
|
|||||||
|
maxarea = 4000
|
||||||
|
minarea = 100
|
||||||
|
minwh = 0.800
|
||||||
|
maxwh = 1.300
|
||||||
|
ndilat = 3
|
||||||
|
neros = 3
|
||||||
|
xoffset = 0
|
||||||
|
yoffset = 0
|
||||||
|
width = 1920
|
||||||
|
height = 1200
|
||||||
|
equalize = 0
|
||||||
|
expmethod = 0
|
||||||
|
naverage = 3
|
||||||
|
umax = 400
|
||||||
|
vmax = 230
|
||||||
|
focmax = 4500
|
||||||
|
focmin = 20
|
||||||
|
stpservport = 4444
|
||||||
|
Kxu = 45.155
|
||||||
|
Kyu = 68.838
|
||||||
|
Kxv = 43.771
|
||||||
|
Kyv = -70.428
|
||||||
|
xtarget = 1039.600
|
||||||
|
ytarget = 602.820
|
||||||
|
eqthrowpart = 0.900
|
||||||
|
minexp = 50.000
|
||||||
|
maxexp = 1000.000
|
||||||
|
fixedexp = 1000.000
|
||||||
|
intensthres = 0.010
|
||||||
|
gain = 36.000
|
||||||
|
brightness = 0.000
|
||||||
|
starssort = 0
|
||||||
|
medfilt = 0
|
||||||
|
medseed = 1
|
||||||
|
fixedbg = 0
|
||||||
|
fbglevel = 100
|
||||||
234
LocCorr_new/main.c
Normal file
234
LocCorr_new/main.c
Normal file
@ -0,0 +1,234 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <math.h>
|
||||||
|
#include <pthread.h>
|
||||||
|
#include <signal.h> // signal
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <string.h> // strdup
|
||||||
|
#include <sys/prctl.h> //prctl
|
||||||
|
#include <sys/wait.h> // wait
|
||||||
|
|
||||||
|
#include "cmdlnopts.h"
|
||||||
|
#include "config.h"
|
||||||
|
#include "debug.h"
|
||||||
|
#include "improc.h"
|
||||||
|
#include "steppers.h"
|
||||||
|
#include "socket.h"
|
||||||
|
|
||||||
|
static InputType tp;
|
||||||
|
static pid_t childpid;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* We REDEFINE the default WEAK function of signal processing
|
||||||
|
*/
|
||||||
|
void signals(int sig){
|
||||||
|
DBG("SIGN: %d, child PID: %d", sig, childpid);
|
||||||
|
if(childpid){
|
||||||
|
DBG("Father -> just exit");
|
||||||
|
exit(sig); // father -> do nothin @ end
|
||||||
|
}
|
||||||
|
if(sig){
|
||||||
|
signal(sig, SIG_IGN);
|
||||||
|
DBG("Get signal %d.", sig);
|
||||||
|
}
|
||||||
|
stopwork = TRUE;
|
||||||
|
saveconf(NULL);
|
||||||
|
if(GP && GP->pidfile){ // remove unnesessary PID file
|
||||||
|
DBG("unlink(GP->pidfile)");
|
||||||
|
unlink(GP->pidfile);
|
||||||
|
}
|
||||||
|
if(theSteppers && theSteppers->stepdisconnect) theSteppers->stepdisconnect();
|
||||||
|
DBG("closeXYlog()");
|
||||||
|
closeXYlog();
|
||||||
|
DBG("EXIT %d", sig);
|
||||||
|
LOGERR("Exit with status %d", sig);
|
||||||
|
exit(sig);
|
||||||
|
}
|
||||||
|
|
||||||
|
void iffound_default(pid_t pid){
|
||||||
|
ERRX("Another copy of this process found, pid=%d. Exit.", pid);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void *procinp_thread(_U_ void* arg){
|
||||||
|
int p = process_input(tp, GP->inputname);
|
||||||
|
LOGDBG("process_input=%d", p);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
static InputType chk_inp(const char *name){
|
||||||
|
if(!name) ERRX("Point file or directory name to monitor");
|
||||||
|
InputType itp = chkinput(GP->inputname);
|
||||||
|
if(T_WRONG == itp) return T_WRONG;
|
||||||
|
green("\n%s is a ", name);
|
||||||
|
switch(itp){
|
||||||
|
case T_DIRECTORY:
|
||||||
|
printf("directory");
|
||||||
|
break;
|
||||||
|
case T_JPEG:
|
||||||
|
printf("jpeg");
|
||||||
|
break;
|
||||||
|
case T_PNG:
|
||||||
|
printf("png");
|
||||||
|
break;
|
||||||
|
case T_GIF:
|
||||||
|
printf("gif");
|
||||||
|
break;
|
||||||
|
case T_FITS:
|
||||||
|
printf("fits");
|
||||||
|
break;
|
||||||
|
case T_BMP:
|
||||||
|
printf("bmp");
|
||||||
|
break;
|
||||||
|
case T_GZIP:
|
||||||
|
printf("maybe fits.gz?");
|
||||||
|
break;
|
||||||
|
case T_CAPT_GRASSHOPPER:
|
||||||
|
printf("capture grasshopper camera");
|
||||||
|
break;
|
||||||
|
case T_CAPT_BASLER:
|
||||||
|
printf("capture basler camera");
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
printf("Unsupported type\n");
|
||||||
|
return T_WRONG;
|
||||||
|
}
|
||||||
|
printf("\n");
|
||||||
|
return itp;
|
||||||
|
}
|
||||||
|
|
||||||
|
int main(int argc, char *argv[]){
|
||||||
|
initial_setup();
|
||||||
|
char *self = strdup(argv[0]);
|
||||||
|
GP = parse_args(argc, argv);
|
||||||
|
if(GP->throwpart < 0. || GP->throwpart > 0.99){
|
||||||
|
ERRX("Fraction of black pixels should be in [0., 0.99]");
|
||||||
|
}
|
||||||
|
if(GP->Naveraging < 1 || GP->Naveraging > NAVER_MAX)
|
||||||
|
ERRX("Averaging amount should be from 1 to %d", NAVER_MAX);
|
||||||
|
tp = chk_inp(GP->inputname);
|
||||||
|
if(tp == T_WRONG) ERRX("Enter correct image file or directory name");
|
||||||
|
// check ability of saving file
|
||||||
|
{
|
||||||
|
FILE *f = fopen(GP->outputjpg, "w");
|
||||||
|
if(!f) ERR("Can't create %s", GP->outputjpg);
|
||||||
|
fclose(f);
|
||||||
|
}
|
||||||
|
if(GP->logfile){
|
||||||
|
sl_loglevel lvl = LOGLEVEL_ERR; // default log level - errors
|
||||||
|
int v = GP->verb;
|
||||||
|
while(v--){ // increase loglevel for each "-v"
|
||||||
|
if(++lvl == LOGLEVEL_ANY) break;
|
||||||
|
}
|
||||||
|
OPENLOG(GP->logfile, lvl, 1);
|
||||||
|
DBG("Opened log file @ level %d", lvl);
|
||||||
|
}
|
||||||
|
int C = chkconfig(GP->configname);
|
||||||
|
if(!C){
|
||||||
|
LOGWARN("Wrong/absent configuration file");
|
||||||
|
WARNX("Wrong/absent configuration file");
|
||||||
|
}
|
||||||
|
// change `theconf` parameters to user values
|
||||||
|
{
|
||||||
|
if(GP->maxarea != DEFAULT_MAXAREA || theconf.maxarea == 0) theconf.maxarea = GP->maxarea;
|
||||||
|
if(GP->minarea != DEFAULT_MINAREA || theconf.minarea == 0) theconf.minarea = GP->minarea;
|
||||||
|
if(GP->xtarget > 0.) theconf.xtarget = GP->xtarget;
|
||||||
|
if(GP->ytarget > 0.) theconf.ytarget = GP->ytarget;
|
||||||
|
if(GP->nerosions != DEFAULT_EROSIONS || theconf.Nerosions == 0){
|
||||||
|
if(GP->nerosions < 1 || GP->nerosions > MAX_NEROS) ERRX("Amount of erosions should be from 1 to %d", MAX_NEROS);
|
||||||
|
theconf.Nerosions = GP->nerosions;
|
||||||
|
}
|
||||||
|
if(GP->ndilations != DEFAULT_DILATIONS || theconf.Ndilations == 0){
|
||||||
|
if(GP->ndilations < 1 || GP->ndilations > MAX_NDILAT) ERRX("Amount of erosions should be from 1 to %d", MAX_NDILAT);
|
||||||
|
theconf.Ndilations = GP->ndilations;
|
||||||
|
}
|
||||||
|
if(fabs(GP->throwpart - DEFAULT_THROWPART) > DBL_EPSILON || theconf.throwpart < DBL_EPSILON){
|
||||||
|
if(GP->throwpart < 0. || GP->throwpart > MAX_THROWPART) ERRX("'thworpart' should be from 0 to %g", MAX_THROWPART);
|
||||||
|
theconf.throwpart = GP->throwpart;
|
||||||
|
}
|
||||||
|
if(GP->xoff && GP->xoff < MAX_OFFSET) theconf.xoff = GP->xoff;
|
||||||
|
if(GP->yoff && GP->yoff < MAX_OFFSET) theconf.yoff = GP->yoff;
|
||||||
|
if(GP->width && GP->width < MAX_OFFSET) theconf.width = GP->width;
|
||||||
|
if(GP->height && GP->height < MAX_OFFSET) theconf.height = GP->height;
|
||||||
|
if(fabs(GP->minexp - EXPOS_MIN) > DBL_EPSILON || theconf.minexp < DBL_EPSILON){
|
||||||
|
if(GP->minexp < DBL_EPSILON || GP->minexp > EXPOS_MAX) ERRX("Minimal exposition should be > 0 and < %g", EXPOS_MAX);
|
||||||
|
theconf.minexp = GP->minexp;
|
||||||
|
}
|
||||||
|
if(fabs(GP->maxexp - EXPOS_MAX) > DBL_EPSILON || theconf.maxexp < theconf.minexp){
|
||||||
|
if(GP->maxexp < theconf.minexp) ERRX("Maximal exposition should be greater than minimal");
|
||||||
|
theconf.maxexp = GP->maxexp;
|
||||||
|
}
|
||||||
|
if(GP->equalize && theconf.equalize == 0) theconf.equalize = 1;
|
||||||
|
if(fabs(GP->intensthres - DEFAULT_INTENSTHRES) > DBL_EPSILON){
|
||||||
|
if(GP->intensthres < DBL_EPSILON || GP->intensthres > 1.-DBL_EPSILON) ERRX("'intensthres' should be from 0 to 1");
|
||||||
|
theconf.intensthres = GP->intensthres;
|
||||||
|
}
|
||||||
|
if(GP->Naveraging != DEFAULT_NAVERAGE || theconf.naverage < 1){
|
||||||
|
if(GP->Naveraging < 1 || GP->Naveraging > NAVER_MAX) ERRX("N images for averaging should be from 1 to %d", NAVER_MAX);
|
||||||
|
theconf.naverage = GP->Naveraging;
|
||||||
|
}
|
||||||
|
if(GP->steppersport != DEFAULT_STEPPERSPORT || theconf.stpserverport == 0){
|
||||||
|
if(GP->steppersport < 1 || GP->steppersport > 65535) ERRX("Wrong steppers' server port: %d", GP->steppersport);
|
||||||
|
theconf.stpserverport = GP->steppersport;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
check4running(self, GP->pidfile);
|
||||||
|
DBG("%s started, snippets library version is %s\n", self, sl_libversion());
|
||||||
|
free(self); self = NULL;
|
||||||
|
signal(SIGTERM, signals); // kill (-15) - quit
|
||||||
|
signal(SIGHUP, SIG_IGN); // hup - ignore
|
||||||
|
signal(SIGINT, signals); // ctrl+C - quit
|
||||||
|
signal(SIGQUIT, signals); // ctrl+\ - quit
|
||||||
|
signal(SIGTSTP, SIG_IGN); // ignore ctrl+Z
|
||||||
|
while(1){ // guard for dead processes
|
||||||
|
childpid = fork();
|
||||||
|
if(childpid){ // father
|
||||||
|
LOGMSG("create child with PID %d\n", childpid);
|
||||||
|
DBG("Created child with PID %d\n", childpid);
|
||||||
|
wait(NULL);
|
||||||
|
LOGMSG("child %d died\n", childpid);
|
||||||
|
WARNX("Child %d died\n", childpid);
|
||||||
|
sleep(5);
|
||||||
|
}else{ // son
|
||||||
|
//prctl(PR_SET_PDEATHSIG, SIGTERM); // send SIGTERM to child when parent dies
|
||||||
|
break; // go out to normal functional
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if(!(theSteppers = steppers_connect())){
|
||||||
|
LOGERR("Steppers server unavailable, can't run");
|
||||||
|
ERRX("Steppers server unavailable, can't run");
|
||||||
|
}
|
||||||
|
if(GP->logXYname) openXYlog(GP->logXYname);
|
||||||
|
LOGMSG("Start application...");
|
||||||
|
LOGDBG("xtag=%g, ytag=%g", theconf.xtarget, theconf.ytarget);
|
||||||
|
openIOport(GP->ioport);
|
||||||
|
pthread_t inp_thread;
|
||||||
|
if(pthread_create(&inp_thread, NULL, procinp_thread, NULL)){
|
||||||
|
LOGERR("pthread_create() for image input failed");
|
||||||
|
ERR("pthread_create()");
|
||||||
|
}
|
||||||
|
while(1){
|
||||||
|
if(stopwork || pthread_kill(inp_thread, 0) == ESRCH){
|
||||||
|
DBG("close");
|
||||||
|
pthread_join(inp_thread, NULL);
|
||||||
|
DBG("out");
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
446
LocCorr_new/median.c
Normal file
446
LocCorr_new/median.c
Normal file
@ -0,0 +1,446 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
// FOR MEDIATOR:
|
||||||
|
// Copyright (c) 2011 ashelly.myopenid.com under <http://www.opensource.org/licenses/mit-license>
|
||||||
|
|
||||||
|
|
||||||
|
// TODO: resolve problem with borders
|
||||||
|
|
||||||
|
#include <math.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <string.h>
|
||||||
|
|
||||||
|
#include "config.h"
|
||||||
|
#include "debug.h"
|
||||||
|
#include "imagefile.h"
|
||||||
|
#include "median.h"
|
||||||
|
|
||||||
|
|
||||||
|
#define ELEM_SWAP(a, b) {register Imtype t = a; a = b; b = t;}
|
||||||
|
#define PIX_SORT(a, b) {if (p[a] > p[b]) ELEM_SWAP(p[a], p[b]);}
|
||||||
|
|
||||||
|
static inline Imtype mean(Imtype a, Imtype b){
|
||||||
|
register uint16_t x = ((uint16_t)a + (uint16_t)b) / 2;
|
||||||
|
return (Imtype)x;
|
||||||
|
}
|
||||||
|
|
||||||
|
static Imtype opt_med2(Imtype *p){
|
||||||
|
return mean(p[0], p[1]);
|
||||||
|
}
|
||||||
|
static Imtype opt_med3(Imtype *p){
|
||||||
|
PIX_SORT(0, 1); PIX_SORT(1, 2); PIX_SORT(0, 1);
|
||||||
|
return(p[1]) ;
|
||||||
|
}
|
||||||
|
static Imtype opt_med4(Imtype *p){
|
||||||
|
PIX_SORT(0, 2); PIX_SORT(1, 3);
|
||||||
|
PIX_SORT(0, 1); PIX_SORT(2, 3);
|
||||||
|
return mean(p[1], p[2]);
|
||||||
|
}
|
||||||
|
static Imtype opt_med5(Imtype *p){
|
||||||
|
PIX_SORT(0, 1); PIX_SORT(3, 4); PIX_SORT(0, 3);
|
||||||
|
PIX_SORT(1, 4); PIX_SORT(1, 2); PIX_SORT(2, 3) ;
|
||||||
|
PIX_SORT(1, 2);
|
||||||
|
return(p[2]) ;
|
||||||
|
}
|
||||||
|
// even values are from "FAST, EFFICIENT MEDIAN FILTERS WITH EVEN LENGTH WINDOWS", J.P. HAVLICEK, K.A. SAKADY, G.R.KATZ
|
||||||
|
static Imtype opt_med6(Imtype *p){
|
||||||
|
PIX_SORT(1, 2); PIX_SORT(3, 4);
|
||||||
|
PIX_SORT(0, 1); PIX_SORT(2, 3); PIX_SORT(4, 5);
|
||||||
|
PIX_SORT(1, 2); PIX_SORT(3, 4);
|
||||||
|
PIX_SORT(0, 1); PIX_SORT(2, 3); PIX_SORT(4, 5);
|
||||||
|
PIX_SORT(1, 2); PIX_SORT(3, 4);
|
||||||
|
return mean(p[2], p[3]);
|
||||||
|
}
|
||||||
|
static Imtype opt_med7(Imtype *p){
|
||||||
|
PIX_SORT(0, 5); PIX_SORT(0, 3); PIX_SORT(1, 6);
|
||||||
|
PIX_SORT(2, 4); PIX_SORT(0, 1); PIX_SORT(3, 5);
|
||||||
|
PIX_SORT(2, 6); PIX_SORT(2, 3); PIX_SORT(3, 6);
|
||||||
|
PIX_SORT(4, 5); PIX_SORT(1, 4); PIX_SORT(1, 3);
|
||||||
|
PIX_SORT(3, 4); return (p[3]);
|
||||||
|
}
|
||||||
|
// optimal Batcher's sort for 8 elements (http://myopen.googlecode.com/svn/trunk/gtkclient_tdt/include/fast_median.h)
|
||||||
|
static Imtype opt_med8(Imtype *p){
|
||||||
|
PIX_SORT(0, 4); PIX_SORT(1, 5); PIX_SORT(2, 6);
|
||||||
|
PIX_SORT(3, 7); PIX_SORT(0, 2); PIX_SORT(1, 3);
|
||||||
|
PIX_SORT(4, 6); PIX_SORT(5, 7); PIX_SORT(2, 4);
|
||||||
|
PIX_SORT(3, 5); PIX_SORT(0, 1); PIX_SORT(2, 3);
|
||||||
|
PIX_SORT(4, 5); PIX_SORT(6, 7); PIX_SORT(1, 4);
|
||||||
|
PIX_SORT(3, 6);
|
||||||
|
return mean(p[3], p[4]);
|
||||||
|
}
|
||||||
|
static Imtype opt_med9(Imtype *p){
|
||||||
|
PIX_SORT(1, 2); PIX_SORT(4, 5); PIX_SORT(7, 8);
|
||||||
|
PIX_SORT(0, 1); PIX_SORT(3, 4); PIX_SORT(6, 7);
|
||||||
|
PIX_SORT(1, 2); PIX_SORT(4, 5); PIX_SORT(7, 8);
|
||||||
|
PIX_SORT(0, 3); PIX_SORT(5, 8); PIX_SORT(4, 7);
|
||||||
|
PIX_SORT(3, 6); PIX_SORT(1, 4); PIX_SORT(2, 5);
|
||||||
|
PIX_SORT(4, 7); PIX_SORT(4, 2); PIX_SORT(6, 4);
|
||||||
|
PIX_SORT(4, 2); return(p[4]);
|
||||||
|
}
|
||||||
|
static Imtype opt_med16(Imtype *p){
|
||||||
|
PIX_SORT(0, 8); PIX_SORT(1, 9); PIX_SORT(2, 10); PIX_SORT(3, 11);
|
||||||
|
PIX_SORT(4, 12); PIX_SORT(5, 13); PIX_SORT(6, 14); PIX_SORT(7, 15);
|
||||||
|
PIX_SORT(0, 4); PIX_SORT(1, 5); PIX_SORT(2, 6); PIX_SORT(3, 7);
|
||||||
|
PIX_SORT(8, 12); PIX_SORT(9, 13); PIX_SORT(10, 14); PIX_SORT(11, 15);
|
||||||
|
PIX_SORT(4, 8); PIX_SORT(5, 9); PIX_SORT(6, 10); PIX_SORT(7, 11);
|
||||||
|
PIX_SORT(0, 2); PIX_SORT(1, 3); PIX_SORT(4, 6); PIX_SORT(5, 7);
|
||||||
|
PIX_SORT(8, 10); PIX_SORT(9, 11); PIX_SORT(12, 14); PIX_SORT(13, 15);
|
||||||
|
PIX_SORT(2, 8); PIX_SORT(3, 9); PIX_SORT(6, 12); PIX_SORT(7, 13);
|
||||||
|
PIX_SORT(2, 4); PIX_SORT(3, 5); PIX_SORT(6, 8); PIX_SORT(7, 9);
|
||||||
|
PIX_SORT(10, 12); PIX_SORT(11, 13); PIX_SORT(0, 1); PIX_SORT(2, 3);
|
||||||
|
PIX_SORT(4, 5); PIX_SORT(6, 7); PIX_SORT(8, 9); PIX_SORT(10, 11);
|
||||||
|
PIX_SORT(12, 13); PIX_SORT(14, 15); PIX_SORT(1, 8); PIX_SORT(3, 10);
|
||||||
|
PIX_SORT(5, 12); PIX_SORT(7, 14); PIX_SORT(5, 8); PIX_SORT(7, 10);
|
||||||
|
return mean(p[7], p[8]);
|
||||||
|
}
|
||||||
|
static Imtype opt_med25(Imtype *p){
|
||||||
|
PIX_SORT(0, 1) ; PIX_SORT(3, 4) ; PIX_SORT(2, 4) ;
|
||||||
|
PIX_SORT(2, 3) ; PIX_SORT(6, 7) ; PIX_SORT(5, 7) ;
|
||||||
|
PIX_SORT(5, 6) ; PIX_SORT(9, 10) ; PIX_SORT(8, 10) ;
|
||||||
|
PIX_SORT(8, 9) ; PIX_SORT(12, 13); PIX_SORT(11, 13) ;
|
||||||
|
PIX_SORT(11, 12); PIX_SORT(15, 16); PIX_SORT(14, 16) ;
|
||||||
|
PIX_SORT(14, 15); PIX_SORT(18, 19); PIX_SORT(17, 19) ;
|
||||||
|
PIX_SORT(17, 18); PIX_SORT(21, 22); PIX_SORT(20, 22) ;
|
||||||
|
PIX_SORT(20, 21); PIX_SORT(23, 24); PIX_SORT(2, 5) ;
|
||||||
|
PIX_SORT(3, 6) ; PIX_SORT(0, 6) ; PIX_SORT(0, 3) ;
|
||||||
|
PIX_SORT(4, 7) ; PIX_SORT(1, 7) ; PIX_SORT(1, 4) ;
|
||||||
|
PIX_SORT(11, 14); PIX_SORT(8, 14) ; PIX_SORT(8, 11) ;
|
||||||
|
PIX_SORT(12, 15); PIX_SORT(9, 15) ; PIX_SORT(9, 12) ;
|
||||||
|
PIX_SORT(13, 16); PIX_SORT(10, 16); PIX_SORT(10, 13) ;
|
||||||
|
PIX_SORT(20, 23); PIX_SORT(17, 23); PIX_SORT(17, 20) ;
|
||||||
|
PIX_SORT(21, 24); PIX_SORT(18, 24); PIX_SORT(18, 21) ;
|
||||||
|
PIX_SORT(19, 22); PIX_SORT(8, 17) ; PIX_SORT(9, 18) ;
|
||||||
|
PIX_SORT(0, 18) ; PIX_SORT(0, 9) ; PIX_SORT(10, 19) ;
|
||||||
|
PIX_SORT(1, 19) ; PIX_SORT(1, 10) ; PIX_SORT(11, 20) ;
|
||||||
|
PIX_SORT(2, 20) ; PIX_SORT(2, 11) ; PIX_SORT(12, 21) ;
|
||||||
|
PIX_SORT(3, 21) ; PIX_SORT(3, 12) ; PIX_SORT(13, 22) ;
|
||||||
|
PIX_SORT(4, 22) ; PIX_SORT(4, 13) ; PIX_SORT(14, 23) ;
|
||||||
|
PIX_SORT(5, 23) ; PIX_SORT(5, 14) ; PIX_SORT(15, 24) ;
|
||||||
|
PIX_SORT(6, 24) ; PIX_SORT(6, 15) ; PIX_SORT(7, 16) ;
|
||||||
|
PIX_SORT(7, 19) ; PIX_SORT(13, 21); PIX_SORT(15, 23) ;
|
||||||
|
PIX_SORT(7, 13) ; PIX_SORT(7, 15) ; PIX_SORT(1, 9) ;
|
||||||
|
PIX_SORT(3, 11) ; PIX_SORT(5, 17) ; PIX_SORT(11, 17) ;
|
||||||
|
PIX_SORT(9, 17) ; PIX_SORT(4, 10) ; PIX_SORT(6, 12) ;
|
||||||
|
PIX_SORT(7, 14) ; PIX_SORT(4, 6) ; PIX_SORT(4, 7) ;
|
||||||
|
PIX_SORT(12, 14); PIX_SORT(10, 14); PIX_SORT(6, 7) ;
|
||||||
|
PIX_SORT(10, 12); PIX_SORT(6, 10) ; PIX_SORT(6, 17) ;
|
||||||
|
PIX_SORT(12, 17); PIX_SORT(7, 17) ; PIX_SORT(7, 10) ;
|
||||||
|
PIX_SORT(12, 18); PIX_SORT(7, 12) ; PIX_SORT(10, 18) ;
|
||||||
|
PIX_SORT(12, 20); PIX_SORT(10, 20); PIX_SORT(10, 12) ;
|
||||||
|
return (p[12]);
|
||||||
|
}
|
||||||
|
#undef PIX_SORT
|
||||||
|
#define PIX_SORT(a, b) {if (a > b) ELEM_SWAP(a, b);}
|
||||||
|
/**
|
||||||
|
* quick select - algo for approximate median calculation for array idata of size n
|
||||||
|
*/
|
||||||
|
static Imtype quick_select(Imtype *idata, int n){
|
||||||
|
int low, high;
|
||||||
|
int median;
|
||||||
|
int middle, ll, hh;
|
||||||
|
Imtype *arr = MALLOC(Imtype, n);
|
||||||
|
memcpy(arr, idata, n*sizeof(Imtype));
|
||||||
|
low = 0 ; high = n-1 ; median = (low + high) / 2;
|
||||||
|
for(;;){
|
||||||
|
if(high <= low) // One element only
|
||||||
|
break;
|
||||||
|
if(high == low + 1){ // Two elements only
|
||||||
|
PIX_SORT(arr[low], arr[high]) ;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
// Find median of low, middle and high Imtypes; swap into position low
|
||||||
|
middle = (low + high) / 2;
|
||||||
|
PIX_SORT(arr[middle], arr[high]) ;
|
||||||
|
PIX_SORT(arr[low], arr[high]) ;
|
||||||
|
PIX_SORT(arr[middle], arr[low]) ;
|
||||||
|
// Swap low Imtype (now in position middle) into position (low+1)
|
||||||
|
ELEM_SWAP(arr[middle], arr[low+1]) ;
|
||||||
|
// Nibble from each end towards middle, swapping Imtypes when stuck
|
||||||
|
ll = low + 1;
|
||||||
|
hh = high;
|
||||||
|
for(;;){
|
||||||
|
do ll++; while (arr[low] > arr[ll]);
|
||||||
|
do hh--; while (arr[hh] > arr[low]);
|
||||||
|
if(hh < ll) break;
|
||||||
|
ELEM_SWAP(arr[ll], arr[hh]) ;
|
||||||
|
}
|
||||||
|
// Swap middle Imtype (in position low) back into correct position
|
||||||
|
ELEM_SWAP(arr[low], arr[hh]) ;
|
||||||
|
// Re-set active partition
|
||||||
|
if (hh <= median) low = ll;
|
||||||
|
if (hh >= median) high = hh - 1;
|
||||||
|
}
|
||||||
|
Imtype ret = arr[median];
|
||||||
|
FREE(arr);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
#undef PIX_SORT
|
||||||
|
#undef ELEM_SWAP
|
||||||
|
|
||||||
|
/**
|
||||||
|
* calculate median of array idata with size n
|
||||||
|
*/
|
||||||
|
Imtype calc_median(Imtype *idata, int n){
|
||||||
|
if(!idata || n < 1){
|
||||||
|
WARNX("calc_median(): wrong dta"); return 0.;
|
||||||
|
}
|
||||||
|
typedef Imtype (*medfunc)(Imtype *p);
|
||||||
|
medfunc fn = NULL;
|
||||||
|
const medfunc fnarr[] = {opt_med2, opt_med3, opt_med4, opt_med5, opt_med6,
|
||||||
|
opt_med7, opt_med8, opt_med9};
|
||||||
|
if(n == 1) return *idata;
|
||||||
|
if(n < 10) fn = fnarr[n - 2];
|
||||||
|
else if(n == 16) fn = opt_med16;
|
||||||
|
else if(n == 25) fn = opt_med25;
|
||||||
|
if(fn){
|
||||||
|
return fn(idata);
|
||||||
|
}else{
|
||||||
|
return quick_select(idata, n);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#define ImtypeLess(a,b) ((a)<(b))
|
||||||
|
#define ImtypeMean(a,b) (((a)+(b))/2)
|
||||||
|
|
||||||
|
typedef struct Mediator_t{
|
||||||
|
Imtype* data; // circular queue of values
|
||||||
|
int* pos; // index into `heap` for each value
|
||||||
|
int* heap; // max/median/min heap holding indexes into `data`.
|
||||||
|
int N; // allocated size.
|
||||||
|
int idx; // position in circular queue
|
||||||
|
int ct; // count of Imtypes in queue
|
||||||
|
} Mediator;
|
||||||
|
|
||||||
|
/*--- Helper Functions ---*/
|
||||||
|
|
||||||
|
#define minCt(m) (((m)->ct-1)/2) //count of Imtypes in minheap
|
||||||
|
#define maxCt(m) (((m)->ct)/2) //count of Imtypes in maxheap
|
||||||
|
|
||||||
|
//returns 1 if heap[i] < heap[j]
|
||||||
|
static inline int mmless(Mediator* m, int i, int j){
|
||||||
|
return ImtypeLess(m->data[m->heap[i]],m->data[m->heap[j]]);
|
||||||
|
}
|
||||||
|
|
||||||
|
//swaps Imtypes i&j in heap, maintains indexes
|
||||||
|
static inline int mmexchange(Mediator* m, int i, int j){
|
||||||
|
int t = m->heap[i];
|
||||||
|
m->heap[i] = m->heap[j];
|
||||||
|
m->heap[j] = t;
|
||||||
|
m->pos[m->heap[i]] = i;
|
||||||
|
m->pos[m->heap[j]] = j;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
//swaps Imtypes i&j if i<j; returns true if swapped
|
||||||
|
static inline int mmCmpExch(Mediator* m, int i, int j){
|
||||||
|
return (mmless(m,i,j) && mmexchange(m,i,j));
|
||||||
|
}
|
||||||
|
|
||||||
|
//maintains minheap property for all Imtypes below i/2.
|
||||||
|
static inline void minSortDown(Mediator* m, int i){
|
||||||
|
for(; i <= minCt(m); i*=2){
|
||||||
|
if(i>1 && i < minCt(m) && mmless(m, i+1, i)) ++i;
|
||||||
|
if(!mmCmpExch(m,i,i/2)) break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//maintains maxheap property for all Imtypes below i/2. (negative indexes)
|
||||||
|
static inline void maxSortDown(Mediator* m, int i){
|
||||||
|
for(; i >= -maxCt(m); i*=2){
|
||||||
|
if(i<-1 && i > -maxCt(m) && mmless(m, i, i-1)) --i;
|
||||||
|
if(!mmCmpExch(m,i/2,i)) break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//maintains minheap property for all Imtypes above i, including median
|
||||||
|
//returns true if median changed
|
||||||
|
static inline int minSortUp(Mediator* m, int i){
|
||||||
|
while (i > 0 && mmCmpExch(m, i, i/2)) i /= 2;
|
||||||
|
return (i == 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
//maintains maxheap property for all Imtypes above i, including median
|
||||||
|
//returns true if median changed
|
||||||
|
static inline int maxSortUp(Mediator* m, int i){
|
||||||
|
while (i < 0 && mmCmpExch(m, i/2, i)) i /= 2;
|
||||||
|
return (i == 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
/*--- Public Interface ---*/
|
||||||
|
|
||||||
|
//creates new Mediator: to calculate `nImtypes` running median.
|
||||||
|
//mallocs single block of memory, caller must free.
|
||||||
|
static Mediator* MediatorNew(int nImtypes){
|
||||||
|
int size = sizeof(Mediator) + nImtypes*(sizeof(Imtype)+sizeof(int)*2);
|
||||||
|
Mediator* m = malloc(size);
|
||||||
|
m->data = (Imtype*)(m + 1);
|
||||||
|
m->pos = (int*) (m->data + nImtypes);
|
||||||
|
m->heap = m->pos + nImtypes + (nImtypes / 2); //points to middle of storage.
|
||||||
|
m->N = nImtypes;
|
||||||
|
m->ct = m->idx = 0;
|
||||||
|
while (nImtypes--){ //set up initial heap fill pattern: median,max,min,max,...
|
||||||
|
m->pos[nImtypes] = ((nImtypes+1)/2) * ((nImtypes&1)? -1 : 1);
|
||||||
|
m->heap[m->pos[nImtypes]] = nImtypes;
|
||||||
|
}
|
||||||
|
return m;
|
||||||
|
}
|
||||||
|
|
||||||
|
//Inserts Imtype, maintains median in O(lg nImtypes)
|
||||||
|
static void MediatorInsert(Mediator* m, Imtype v){
|
||||||
|
int isNew=(m->ct<m->N);
|
||||||
|
int p = m->pos[m->idx];
|
||||||
|
Imtype old = m->data[m->idx];
|
||||||
|
m->data[m->idx]=v;
|
||||||
|
m->idx = (m->idx+1) % m->N;
|
||||||
|
m->ct+=isNew;
|
||||||
|
if(p>0){ //new Imtype is in minHeap
|
||||||
|
if (!isNew && ImtypeLess(old,v)) minSortDown(m,p*2);
|
||||||
|
else if (minSortUp(m,p)) maxSortDown(m,-1);
|
||||||
|
}else if (p<0){ //new Imtype is in maxheap
|
||||||
|
if (!isNew && ImtypeLess(v,old)) maxSortDown(m,p*2);
|
||||||
|
else if (maxSortUp(m,p)) minSortDown(m, 1);
|
||||||
|
}else{ //new Imtype is at median
|
||||||
|
if (maxCt(m)) maxSortDown(m,-1);
|
||||||
|
if (minCt(m)) minSortDown(m, 1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//returns median Imtype (or average of 2 when Imtype count is even)
|
||||||
|
static Imtype MediatorMedian(Mediator* m){
|
||||||
|
Imtype v = m->data[m->heap[0]];
|
||||||
|
if ((m->ct&1) == 0) v = ImtypeMean(v, m->data[m->heap[-1]]);
|
||||||
|
return v;
|
||||||
|
}
|
||||||
|
|
||||||
|
#if 0
|
||||||
|
// median + min/max
|
||||||
|
static Imtype MediatorStat(Mediator* m, Imtype *minval, Imtype *maxval){
|
||||||
|
Imtype v = m->data[m->heap[0]];
|
||||||
|
if ((m->ct&1) == 0) v = ImtypeMean(v, m->data[m->heap[-1]]);
|
||||||
|
Imtype min = v, max = v;
|
||||||
|
int i;
|
||||||
|
for(i = -maxCt(m); i < 0; ++i){
|
||||||
|
int v = m->data[m->heap[i]];
|
||||||
|
if(v < min) min = v;
|
||||||
|
}
|
||||||
|
*minval = min;
|
||||||
|
for(i = 1; i <= minCt(m); ++i){
|
||||||
|
int v = m->data[m->heap[i]];
|
||||||
|
if(v > max) max = v;
|
||||||
|
}
|
||||||
|
*maxval = max;
|
||||||
|
return v;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
/**
|
||||||
|
* filter image by median (seed*2 + 1) x (seed*2 + 1)
|
||||||
|
*/
|
||||||
|
Image *get_median(const Image *img, int seed){
|
||||||
|
if(seed < 1) return NULL;
|
||||||
|
size_t w = img->width, h = img->height;
|
||||||
|
Image *out = Image_sim(img);
|
||||||
|
Imtype *med = out->data, *inputima = img->data;
|
||||||
|
|
||||||
|
size_t blksz = seed * 2 + 1, fullsz = blksz * blksz;
|
||||||
|
#ifdef EBUG
|
||||||
|
double t0 = dtime();
|
||||||
|
#endif
|
||||||
|
OMP_FOR(shared(inputima, med))
|
||||||
|
for(size_t x = seed; x < w - seed; ++x){
|
||||||
|
size_t xx, yy, xm = x + seed + 1, y, ymax = blksz - 1, xmin = x - seed;
|
||||||
|
Mediator* m = MediatorNew(fullsz);
|
||||||
|
// initial fill
|
||||||
|
for(yy = 0; yy < ymax; ++yy)
|
||||||
|
for(xx = xmin; xx < xm; ++xx)
|
||||||
|
MediatorInsert(m, inputima[xx + yy*w]);
|
||||||
|
ymax = 2*seed*w;
|
||||||
|
xmin += ymax;
|
||||||
|
xm += ymax;
|
||||||
|
ymax = h - seed;
|
||||||
|
size_t medidx = x + seed * w;
|
||||||
|
for(y = seed; y < ymax; ++y, xmin += w, xm += w, medidx += w){
|
||||||
|
for(xx = xmin; xx < xm; ++xx)
|
||||||
|
MediatorInsert(m, inputima[xx]);
|
||||||
|
med[medidx] = MediatorMedian(m);
|
||||||
|
}
|
||||||
|
free(m);
|
||||||
|
}
|
||||||
|
Image_minmax(out);
|
||||||
|
DBG("time for median filtering %zdx%zd of image %zdx%zd: %gs", blksz, blksz, w, h,
|
||||||
|
dtime() - t0);
|
||||||
|
return out;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief get_stat - calculate floating statistics in (seed*2+1)^2
|
||||||
|
* @param in (i) - input image
|
||||||
|
* @param seed - radius of box
|
||||||
|
* @param mean (o) - mean by box (excluding borders)
|
||||||
|
* @param std (o) - STD by box (excluding borders)
|
||||||
|
* @retur 0 if error
|
||||||
|
*/
|
||||||
|
int get_stat(const Image *in, int seed, Image **mean, Image **std){
|
||||||
|
if(!in) return FALSE;
|
||||||
|
if(seed < 1 || seed > (in->width - 1)/2 || seed > (in->height - 1)/2) return FALSE;
|
||||||
|
#ifdef EBUG
|
||||||
|
double t0 = dtime();
|
||||||
|
#endif
|
||||||
|
Image *M = NULL, *S = NULL;
|
||||||
|
if(mean) M = Image_sim(in);
|
||||||
|
if(std) S = Image_sim(in);
|
||||||
|
int ymax = in->height - seed, xmax = in->width - seed;
|
||||||
|
int hsz = (seed*2 + 1), sz = hsz * hsz, w = in->width;
|
||||||
|
OMP_FOR()
|
||||||
|
for(int y = seed; y < ymax; ++y){ // dumb calculations
|
||||||
|
int startidx = y*w + seed;
|
||||||
|
Imtype *om = (M) ? &M->data[startidx] : NULL;
|
||||||
|
Imtype *os = (S) ? &S->data[startidx] : NULL;
|
||||||
|
for(int x = seed; x < xmax; ++x){
|
||||||
|
double sum = 0, sum2 = 0;
|
||||||
|
int yb = y + seed + 1, xm = x - seed;
|
||||||
|
for(int yy = y - seed; yy < yb; ++yy){
|
||||||
|
Imtype *ptr = &in->data[yy * w + xm];
|
||||||
|
for(int xx = 0; xx < hsz; ++xx){
|
||||||
|
double d = *ptr++;
|
||||||
|
sum += d;
|
||||||
|
sum2 += d*d;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
//DBG("sum=%g, sum2=%g, sz=%d", sum, sum2, sz);
|
||||||
|
sum /= sz;
|
||||||
|
if(om){
|
||||||
|
*om++ = (Imtype)sum;
|
||||||
|
//DBG("mean (%d, %d): %g", x, y, sum);
|
||||||
|
}
|
||||||
|
if(os) *os++ = (Imtype)sqrt(sum2/sz - sum*sum);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if(mean){
|
||||||
|
Image_minmax(M);
|
||||||
|
*mean = M;
|
||||||
|
}
|
||||||
|
if(std){
|
||||||
|
Image_minmax(S);
|
||||||
|
*std = S;
|
||||||
|
}
|
||||||
|
DBG("time for mean/sigma computation: %gs", dtime() - t0);
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
31
LocCorr_new/median.h
Normal file
31
LocCorr_new/median.h
Normal file
@ -0,0 +1,31 @@
|
|||||||
|
/*
|
||||||
|
* median.h
|
||||||
|
*
|
||||||
|
* Copyright 2015 Edward V. Emelianov <eddy@sao.ru, edward.emelianoff@gmail.com>
|
||||||
|
*
|
||||||
|
* This program is free software; you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation; either version 2 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program; if not, write to the Free Software
|
||||||
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
||||||
|
* MA 02110-1301, USA.
|
||||||
|
*/
|
||||||
|
#pragma once
|
||||||
|
#ifndef __MEDIAN_H__
|
||||||
|
#define __MEDIAN_H__
|
||||||
|
|
||||||
|
#include "fits.h"
|
||||||
|
|
||||||
|
Imtype calc_median(Imtype *idata, int n);
|
||||||
|
Image *get_median(const Image *img, int seed);
|
||||||
|
int get_stat(const Image *in, int seed, Image **mean, Image **std);
|
||||||
|
|
||||||
|
#endif // __MEDIAN_H__
|
||||||
421
LocCorr_new/socket.c
Normal file
421
LocCorr_new/socket.c
Normal file
@ -0,0 +1,421 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
|
||||||
|
#include <arpa/inet.h> // inet_ntop
|
||||||
|
#include <libgen.h> // basename
|
||||||
|
#include <limits.h> // INT_xxx
|
||||||
|
#include <netdb.h> // addrinfo
|
||||||
|
#include <poll.h>
|
||||||
|
#include <pthread.h>
|
||||||
|
#include <signal.h> // pthread_kill
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <sys/ioctl.h>
|
||||||
|
#include <sys/syscall.h> // syscall
|
||||||
|
#include <unistd.h> // daemon
|
||||||
|
|
||||||
|
#include "cmdlnopts.h"
|
||||||
|
#include "config.h"
|
||||||
|
#include "debug.h"
|
||||||
|
#include "improc.h"
|
||||||
|
#include "socket.h"
|
||||||
|
#include "steppers.h"
|
||||||
|
|
||||||
|
// buffer size for received data
|
||||||
|
#define BUFLEN (1024)
|
||||||
|
// buffer size for answer
|
||||||
|
#define ANSBUFLEN (32768)
|
||||||
|
// Max amount of connections
|
||||||
|
#define BACKLOG (10)
|
||||||
|
|
||||||
|
/*
|
||||||
|
TODO3: add 'FAIL error text' if not OK and instead all "wrong message"
|
||||||
|
*/
|
||||||
|
|
||||||
|
|
||||||
|
// additional commands list - getters
|
||||||
|
typedef struct{
|
||||||
|
const char *command;
|
||||||
|
char *(*handler)(const char *messageid, char *buf, int buflen);
|
||||||
|
const char *help;
|
||||||
|
} getter;
|
||||||
|
// setters
|
||||||
|
typedef struct{
|
||||||
|
const char *command;
|
||||||
|
char *(*handler)(const char *val, char *buf, int buflen);
|
||||||
|
const char *help;
|
||||||
|
} setter;
|
||||||
|
|
||||||
|
static char *helpmsg(const char *messageid, char *buf, int buflen);
|
||||||
|
static char *stepperstatus(const char *messageid, char *buf, int buflen);
|
||||||
|
static char *getimagedata(const char *messageid, char *buf, int buflen);
|
||||||
|
static getter getterHandlers[] = {
|
||||||
|
{"help", helpmsg, "List avaiable commands"},
|
||||||
|
{"settings", listconf, "List current configuration"},
|
||||||
|
{"stpserv", stepperstatus, "Get status of steppers server"},
|
||||||
|
{"imdata", getimagedata, "Get image data (status, path, FPS, counter)"},
|
||||||
|
{NULL, NULL, NULL}
|
||||||
|
};
|
||||||
|
|
||||||
|
static char *setstepperstate(const char *state, char *buf, int buflen);
|
||||||
|
static char *setfocusstate(const char *state, char *buf, int buflen);
|
||||||
|
static char *moveU(const char *val, char *buf, int buflen);
|
||||||
|
static char *moveV(const char *val, char *buf, int buflen);
|
||||||
|
|
||||||
|
static setter setterHandlers[] = {
|
||||||
|
{"stpstate", setstepperstate, "Set given steppers' server state"},
|
||||||
|
{"focus", setfocusstate, "Move focus to given value"},
|
||||||
|
{"moveU", moveU, "Relative moving by U axe"},
|
||||||
|
{"moveV", moveV, "Relative moving by V axe"},
|
||||||
|
{NULL, NULL, NULL}
|
||||||
|
};
|
||||||
|
|
||||||
|
static char *retFAIL(char *buf, int buflen){
|
||||||
|
snprintf(buf, buflen, FAIL);
|
||||||
|
return buf;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**************** functions to process commands ****************/
|
||||||
|
// getters
|
||||||
|
static char *helpmsg(_U_ const char *messageid, char *buf, int buflen){
|
||||||
|
if(get_cmd_list(buf, buflen)){
|
||||||
|
int l = strlen(buf), L = buflen - l;
|
||||||
|
char *ptr = buf + l;
|
||||||
|
getter *g = getterHandlers;
|
||||||
|
while(L > 0 && g->command){
|
||||||
|
int s = snprintf(ptr, L, "%s - %s\n", g->command, g->help);
|
||||||
|
if(s < 1) break;
|
||||||
|
L -= s; ptr += s;
|
||||||
|
++g;
|
||||||
|
}
|
||||||
|
setter *sh = setterHandlers;
|
||||||
|
while(L > 0 && sh->command){
|
||||||
|
int s = snprintf(ptr, L, "%s=newval - %s\n", sh->command, sh->help);
|
||||||
|
if(s < 1) break;
|
||||||
|
L -= s; ptr += s;
|
||||||
|
++sh;
|
||||||
|
}
|
||||||
|
return buf;
|
||||||
|
}
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
static char *stepperstatus(const char *messageid, char *buf, int buflen){
|
||||||
|
if(theSteppers && theSteppers->stepstatus) return theSteppers->stepstatus(messageid, buf, buflen);
|
||||||
|
return retFAIL(buf, buflen);
|
||||||
|
}
|
||||||
|
static char *getimagedata(const char *messageid, char *buf, int buflen){
|
||||||
|
if(imagedata) return imagedata(messageid, buf, buflen);
|
||||||
|
return retFAIL(buf, buflen);
|
||||||
|
}
|
||||||
|
|
||||||
|
// setters
|
||||||
|
static char *setstepperstate(const char *state, char *buf, int buflen){
|
||||||
|
DBG("set steppersstate to %s", state);
|
||||||
|
if(theSteppers && theSteppers->setstepstatus) return theSteppers->setstepstatus(state, buf, buflen);
|
||||||
|
return retFAIL(buf, buflen);
|
||||||
|
}
|
||||||
|
static char *setfocusstate(const char *state, char *buf, int buflen){
|
||||||
|
DBG("move focus to %s", state);
|
||||||
|
if(theSteppers && theSteppers->movefocus) return theSteppers->movefocus(state, buf, buflen);
|
||||||
|
return retFAIL(buf, buflen);
|
||||||
|
}
|
||||||
|
static char *moveU(const char *val, char *buf, int buflen){
|
||||||
|
if(theSteppers && theSteppers->moveByU) return theSteppers->moveByU(val, buf, buflen);
|
||||||
|
return retFAIL(buf, buflen);
|
||||||
|
}
|
||||||
|
static char *moveV(const char *val, char *buf, int buflen){
|
||||||
|
if(theSteppers && theSteppers->moveByV) return theSteppers->moveByV(val, buf, buflen);
|
||||||
|
return retFAIL(buf, buflen);
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
static char *rmnl(const char *msg, char *buf, int buflen){
|
||||||
|
strncpy(buf, msg, buflen);
|
||||||
|
char *nl = strchr(buf, '\n');
|
||||||
|
if(nl) *nl = 0;
|
||||||
|
return buf;
|
||||||
|
}
|
||||||
|
*/
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief processCommand - command parser
|
||||||
|
* @param msg - incoming message
|
||||||
|
* @param ans - buffer for answer
|
||||||
|
* @param anslen - length of `ans`
|
||||||
|
* @return NULL if no answer or pointer to ans
|
||||||
|
*/
|
||||||
|
static char *processCommand(const char msg[BUFLEN], char *ans, int anslen){
|
||||||
|
char value[BUFLEN];
|
||||||
|
char *kv = get_keyval(msg, value);
|
||||||
|
confparam *par;
|
||||||
|
if(kv){
|
||||||
|
DBG("got KEY '%s' with value '%s'", kv, value);
|
||||||
|
key_value result;
|
||||||
|
par = chk_keyval(kv, value, &result);
|
||||||
|
free(kv); kv = NULL;
|
||||||
|
if(par){
|
||||||
|
switch(par->type){
|
||||||
|
case PAR_INT:
|
||||||
|
DBG("FOUND! Integer, old=%d, new=%d", *((int*)par->ptr), result.val.intval);
|
||||||
|
*((int*)par->ptr) = result.val.intval;
|
||||||
|
break;
|
||||||
|
case PAR_DOUBLE:
|
||||||
|
DBG("FOUND! Double, old=%g, new=%g", *((double*)par->ptr), result.val.dblval);
|
||||||
|
*((double*)par->ptr) = result.val.dblval;
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
snprintf(ans, anslen, FAIL);
|
||||||
|
return ans;
|
||||||
|
}
|
||||||
|
snprintf(ans, anslen, OK);
|
||||||
|
return ans;
|
||||||
|
}else{
|
||||||
|
setter *s = setterHandlers;
|
||||||
|
while(s->command){
|
||||||
|
int l = strlen(s->command);
|
||||||
|
if(strncasecmp(msg, s->command, l) == 0)
|
||||||
|
return s->handler(value, ans, anslen);
|
||||||
|
++s;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}else{
|
||||||
|
getter *g = getterHandlers;
|
||||||
|
while(g->command){
|
||||||
|
int l = strlen(g->command);
|
||||||
|
if(strncasecmp(msg, g->command, l) == 0)
|
||||||
|
return g->handler(g->command, ans, anslen);
|
||||||
|
++g;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
snprintf(ans, anslen, FAIL);
|
||||||
|
return ans;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**************** SERVER FUNCTIONS ****************/
|
||||||
|
/**
|
||||||
|
* Send data over socket (and add trailing '\n' if absent)
|
||||||
|
* @param sock - socket fd
|
||||||
|
* @param textbuf - zero-trailing buffer with data to send
|
||||||
|
* @return amount of sent bytes
|
||||||
|
*/
|
||||||
|
static size_t send_data(int sock, const char *textbuf){
|
||||||
|
ssize_t Len = strlen(textbuf);
|
||||||
|
if(Len != write(sock, textbuf, Len)){
|
||||||
|
WARN("write()");
|
||||||
|
LOGERR("send_data(): write() failed");
|
||||||
|
return 0;
|
||||||
|
}else{
|
||||||
|
LOGDBG("send_data(): sent '%s'", textbuf);
|
||||||
|
}
|
||||||
|
if(textbuf[Len-1] != '\n') Len += write(sock, "\n", 1);
|
||||||
|
return (size_t)Len;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief handle_socket - read and process data from socket
|
||||||
|
* @param sock - socket fd
|
||||||
|
* @return 0 if all OK, 1 if socket closed
|
||||||
|
*/
|
||||||
|
static int handle_socket(int sock){
|
||||||
|
FNAME();
|
||||||
|
char buff[BUFLEN];
|
||||||
|
char ansbuff[ANSBUFLEN];
|
||||||
|
ssize_t rd = read(sock, buff, BUFLEN-1);
|
||||||
|
if(rd < 1){
|
||||||
|
DBG("read() == %zd", rd);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
// add trailing zero to be on the safe side
|
||||||
|
buff[rd] = 0;
|
||||||
|
// now we should check what do user want
|
||||||
|
// here we can process user data
|
||||||
|
DBG("user %d send '%s'", sock, buff);
|
||||||
|
LOGDBG("user %d send '%s'", sock, buff);
|
||||||
|
//pthread_mutex_lock(&mutex);
|
||||||
|
char *ans = processCommand(buff, ansbuff, ANSBUFLEN-1); // run command parser
|
||||||
|
if(ans){
|
||||||
|
send_data(sock, ans); // send answer
|
||||||
|
}
|
||||||
|
//pthread_mutex_unlock(&mutex);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
// main socket server
|
||||||
|
static void *server(void *asock){
|
||||||
|
DBG("server(): getpid: %d, tid: %lu",getpid(), syscall(SYS_gettid));
|
||||||
|
int sock = *((int*)asock);
|
||||||
|
if(listen(sock, BACKLOG) == -1){
|
||||||
|
LOGERR("server(): listen() failed");
|
||||||
|
WARN("listen");
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
int nfd = 1;
|
||||||
|
struct pollfd poll_set[BACKLOG+1];
|
||||||
|
memset(poll_set, 0, sizeof(poll_set));
|
||||||
|
poll_set[0].fd = sock;
|
||||||
|
poll_set[0].events = POLLIN;
|
||||||
|
while(1){
|
||||||
|
if(stopwork){
|
||||||
|
DBG("server() exit @ global stop");
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
poll(poll_set, nfd, 1); // poll for 1ms
|
||||||
|
for(int fdidx = 0; fdidx < nfd; ++fdidx){ // poll opened FDs
|
||||||
|
if((poll_set[fdidx].revents & POLLIN) == 0) continue;
|
||||||
|
poll_set[fdidx].revents = 0;
|
||||||
|
if(fdidx){ // client
|
||||||
|
int fd = poll_set[fdidx].fd;
|
||||||
|
//int nread = 0;
|
||||||
|
//ioctl(fd, FIONREAD, &nread);
|
||||||
|
if(handle_socket(fd)){ // socket closed - remove it from list
|
||||||
|
close(fd);
|
||||||
|
DBG("Client with fd %d closed", fd);
|
||||||
|
LOGMSG("Client %d disconnected", fd);
|
||||||
|
// move last to free space
|
||||||
|
poll_set[fdidx] = poll_set[nfd - 1];
|
||||||
|
--nfd;
|
||||||
|
}
|
||||||
|
}else{ // server
|
||||||
|
socklen_t size = sizeof(struct sockaddr_in);
|
||||||
|
struct sockaddr_in their_addr;
|
||||||
|
int newsock = accept(sock, (struct sockaddr*)&their_addr, &size);
|
||||||
|
if(newsock <= 0){
|
||||||
|
LOGERR("server(): accept() failed");
|
||||||
|
WARN("accept()");
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
struct in_addr ipAddr = their_addr.sin_addr;
|
||||||
|
char str[INET_ADDRSTRLEN];
|
||||||
|
inet_ntop(AF_INET, &ipAddr, str, INET_ADDRSTRLEN);
|
||||||
|
DBG("Connection from %s, give fd=%d", str, newsock);
|
||||||
|
LOGMSG("Got connection from %s, fd=%d", str, newsock);
|
||||||
|
if(nfd == BACKLOG + 1){
|
||||||
|
LOGWARN("Max amount of connections: disconnect %s (%d)", str, newsock);
|
||||||
|
send_data(newsock, "Max amount of connections reached!");
|
||||||
|
WARNX("Limit of connections reached");
|
||||||
|
close(newsock);
|
||||||
|
}else{
|
||||||
|
memset(&poll_set[nfd], 0, sizeof(struct pollfd));
|
||||||
|
poll_set[nfd].fd = newsock;
|
||||||
|
poll_set[nfd].events = POLLIN;
|
||||||
|
++nfd;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} // endfor
|
||||||
|
}
|
||||||
|
LOGERR("server(): UNREACHABLE CODE REACHED!");
|
||||||
|
}
|
||||||
|
|
||||||
|
// data gathering & socket management
|
||||||
|
static void daemon_(int sock){
|
||||||
|
if(sock < 0) return;
|
||||||
|
pthread_t sock_thread;//, canserver_thread;
|
||||||
|
if(pthread_create(&sock_thread, NULL, server, (void*) &sock)){
|
||||||
|
LOGERR("daemon_(): pthread_create() failed");
|
||||||
|
ERR("pthread_create()");
|
||||||
|
}
|
||||||
|
do{
|
||||||
|
if(stopwork){
|
||||||
|
DBG("kill");
|
||||||
|
pthread_join(sock_thread, NULL);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if(pthread_kill(sock_thread, 0) == ESRCH){ // died
|
||||||
|
WARNX("Sockets thread died");
|
||||||
|
LOGERR("Sockets thread died");
|
||||||
|
pthread_join(sock_thread, NULL);
|
||||||
|
if(pthread_create(&sock_thread, NULL, server, (void*) &sock)){
|
||||||
|
LOGERR("daemon_(): new pthread_create(sock_thread) failed");
|
||||||
|
ERR("pthread_create(sock_thread)");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
usleep(1000); // sleep a little or thread's won't be able to lock mutex
|
||||||
|
}while(1);
|
||||||
|
LOGERR("daemon_(): UNREACHABLE CODE REACHED!");
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* open sockets
|
||||||
|
* // should be called only once!!!
|
||||||
|
*/
|
||||||
|
static void *connect2sock(void *data){
|
||||||
|
FNAME();
|
||||||
|
char port[10];
|
||||||
|
int portN = *((int*)data);
|
||||||
|
snprintf(port, 10, "%d", portN);
|
||||||
|
DBG("get port: %s", port);
|
||||||
|
int sock = -1;
|
||||||
|
struct addrinfo hints, *res, *p;
|
||||||
|
memset(&hints, 0, sizeof(hints));
|
||||||
|
hints.ai_family = AF_INET;
|
||||||
|
hints.ai_socktype = SOCK_STREAM;
|
||||||
|
hints.ai_flags = AI_PASSIVE;
|
||||||
|
if(getaddrinfo("127.0.0.1", port, &hints, &res) != 0){ // accept only local connections
|
||||||
|
LOGERR("daemonize(): getaddrinfo() failed");
|
||||||
|
ERR("getaddrinfo");
|
||||||
|
}
|
||||||
|
struct sockaddr_in *ia = (struct sockaddr_in*)res->ai_addr;
|
||||||
|
char str[INET_ADDRSTRLEN];
|
||||||
|
inet_ntop(AF_INET, &(ia->sin_addr), str, INET_ADDRSTRLEN);
|
||||||
|
// loop through all the results and bind to the first we can
|
||||||
|
for(p = res; p != NULL; p = p->ai_next){
|
||||||
|
if((sock = socket(p->ai_family, p->ai_socktype, p->ai_protocol)) == -1){
|
||||||
|
LOGWARN("openIOport(): socket() failed");
|
||||||
|
WARN("socket");
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
int reuseaddr = 1;
|
||||||
|
if(setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &reuseaddr, sizeof(int)) == -1){
|
||||||
|
LOGERR("openIOport(): setsockopt() failed");
|
||||||
|
ERR("setsockopt");
|
||||||
|
}
|
||||||
|
if(bind(sock, p->ai_addr, p->ai_addrlen) == -1){
|
||||||
|
close(sock);
|
||||||
|
LOGERR("openIOport(): bind() failed");
|
||||||
|
WARN("bind");
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
break; // if we get here, we have a successfull connection
|
||||||
|
}
|
||||||
|
if(p == NULL){
|
||||||
|
LOGERR("openIOport(): failed to bind socket, exit");
|
||||||
|
// looped off the end of the list with no successful bind
|
||||||
|
ERRX("failed to bind socket");
|
||||||
|
}
|
||||||
|
freeaddrinfo(res);
|
||||||
|
daemon_(sock);
|
||||||
|
close(sock);
|
||||||
|
LOGWARN("openIOport(): close @ global stop");
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
// run socket thread
|
||||||
|
void openIOport(int portN){
|
||||||
|
static int portnum = 0;
|
||||||
|
if(portnum) return;
|
||||||
|
portnum = portN;
|
||||||
|
pthread_t connthread;
|
||||||
|
DBG("open port: %d", portN);
|
||||||
|
if(pthread_create(&connthread, NULL, connect2sock, (void*) &portnum)){
|
||||||
|
LOGERR("openIOport(): pthread_create() failed");
|
||||||
|
ERR("pthread_create()");
|
||||||
|
}
|
||||||
|
pthread_detach(connthread);
|
||||||
|
}
|
||||||
29
LocCorr_new/socket.h
Normal file
29
LocCorr_new/socket.h
Normal file
@ -0,0 +1,29 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
#ifndef __SOCKET_H__
|
||||||
|
#define __SOCKET_H__
|
||||||
|
|
||||||
|
// standard answers
|
||||||
|
#define OK "OK\n"
|
||||||
|
#define FAIL "FAILED\n"
|
||||||
|
|
||||||
|
void openIOport(int portN);
|
||||||
|
|
||||||
|
#endif // __SOCKET_H__
|
||||||
29
LocCorr_new/stbimpl.c
Normal file
29
LocCorr_new/stbimpl.c
Normal file
@ -0,0 +1,29 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2021 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
// implementation of STB library (to accelerate compilation)
|
||||||
|
#define STBI_NO_PSD
|
||||||
|
#define STBI_NO_TGA
|
||||||
|
#define STBI_NO_HDR
|
||||||
|
#define STBI_NO_PIC
|
||||||
|
#define STBI_NO_PNM
|
||||||
|
|
||||||
|
#define STB_IMAGE_IMPLEMENTATION
|
||||||
|
#include <stb/stb_image.h>
|
||||||
|
#define STB_IMAGE_WRITE_IMPLEMENTATION
|
||||||
|
#include <stb/stb_image_write.h>
|
||||||
1068
LocCorr_new/steppers.c
Normal file
1068
LocCorr_new/steppers.c
Normal file
File diff suppressed because it is too large
Load Diff
39
LocCorr_new/steppers.h
Normal file
39
LocCorr_new/steppers.h
Normal file
@ -0,0 +1,39 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the loccorr project.
|
||||||
|
* Copyright 2024 Edward V. Emelianov <edward.emelianoff@gmail.com>.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
// set state to `disconnect` after this amount of errors in `moving_finished`
|
||||||
|
#define MAX_ERR_CTR (15)
|
||||||
|
|
||||||
|
// amount of ALL motors
|
||||||
|
#define NMOTORS (8)
|
||||||
|
|
||||||
|
typedef struct{
|
||||||
|
void (*proc_corr)(double, double);
|
||||||
|
char *(*stepstatus)(const char *messageid, char *buf, int buflen);
|
||||||
|
char *(*setstepstatus)(const char *newstatus, char *buf, int buflen);
|
||||||
|
char *(*movefocus)(const char *newstatus, char *buf, int buflen);
|
||||||
|
char *(*moveByU)(const char *val, char *buf, int buflen);
|
||||||
|
char *(*moveByV)(const char *val, char *buf, int buflen);
|
||||||
|
void (*stepdisconnect)();
|
||||||
|
} steppersproc;
|
||||||
|
|
||||||
|
steppersproc *steppers_connect();
|
||||||
|
extern steppersproc* theSteppers;
|
||||||
|
|
||||||
Loading…
x
Reference in New Issue
Block a user