windows support

Author: Tieqiong

setup.py

@@ -17,15 +17,36 @@
 
 
 def get_boost_libraries():
-    base_lib = "boost_python"
-    major, minor = str(sys.version_info[0]), str(sys.version_info[1])
-    tags = [f"{major}{minor}", major, ""]
-    mttags = ["", "-mt"]
-    candidates = [base_lib + tag for tag in tags for mt in mttags] + [base_lib]
-    for lib in candidates:
-        if find_library(lib):
-            return [lib]
-    raise RuntimeError("Cannot find a suitable Boost.Python library.")
+    """
+    Link against both Boost.Python and Boost.Serialization
+    from the active conda env’s lib/ directory.
+    """
+    conda_prefix = os.environ.get("CONDA_PREFIX")
+    if not conda_prefix:
+        raise EnvironmentError("CONDA_PREFIX is not set; please activate your conda environment.")
+
+    libdir = Path(conda_prefix) / "lib"
+    major, minor = sys.version_info[:2]
+
+    # look for the _python_ library
+    py_cands = [f"boost_python{major}{minor}", f"boost_python{major}", "boost_python"]
+    for lib in py_cands:
+        if (libdir / f"lib{lib}.so").exists():
+            boost_py = lib
+            break
+    else:
+        raise RuntimeError(f"Could not find any of {py_cands!r} in {libdir}")
+
+    # look for the _serialization_ library
+    ser_cands = ["boost_serialization"]
+    for lib in ser_cands:
+        if (libdir / f"lib{lib}.so").exists():
+            boost_ser = lib
+            break
+    else:
+        raise RuntimeError(f"Could not find any of {ser_cands!r} in {libdir}")
+
+    return [boost_ser, boost_py]
 
 
 def get_boost_config():
@@ -48,20 +69,46 @@ def get_boost_config():
     return {"include_dirs": [str(inc)], "library_dirs": [str(lib)]}
 
 
+def get_objcryst_libraries():
+    conda_prefix = os.environ.get("CONDA_PREFIX")
+    if not conda_prefix:
+        raise EnvironmentError(
+            "CONDA_PREFIX is not set. Please install ObjCryst using conda and activate the environment."
+        )
+    if os.name == "nt":
+        libdir = Path(conda_prefix) / "Library" / "lib"
+    else:
+        libdir = Path(conda_prefix) / "lib"
+
+    libs = []
+    for fn in os.listdir(libdir):
+        low = fn.lower()
+        if low.endswith(".lib") and "objcryst" in low:
+            libs.append(os.path.splitext(fn)[0])
+    if not libs:
+        raise RuntimeError(f"No ObjCryst libraries found in {libdir}")
+    return libs
+
+
 if os.name == "nt":
     compile_args = ["/std:c++14"]
     macros = [("_USE_MATH_DEFINES", None)]
+    extra_link_args  = ["/FORCE:MULTIPLE"]
 else:
     compile_args = ["-std=c++11"]
     macros = []
+    extra_link_args = []
 
 boost_cfg = get_boost_config()
+objcryst_libs = get_objcryst_libraries()
+
 ext_kws = {
-    "libraries": ["diffpy"] + get_boost_libraries(),
+    "libraries": ["diffpy"] + get_boost_libraries() + objcryst_libs,
     "extra_compile_args": compile_args,
-    "extra_link_args": [],
+    "extra_link_args": extra_link_args,
     "include_dirs": [numpy.get_include()] + boost_cfg["include_dirs"],
     "library_dirs": boost_cfg["library_dirs"],
+    "runtime_library_dirs": boost_cfg["library_dirs"],
     "define_macros": macros,
 }